EmmyLuaCodeStyle/CodeFormatServer/src/CodeFormatServer.cpp at master · Endll/EmmyLuaCodeStyle · GitHub

Repository navigation

35 lines (31 loc) · 980 Bytes

#include <asio.hpp>

#include "CodeFormatServer/LanguageClient.h"

#include "CodeFormatServer/Session/SocketIOSession.h"

#include "CodeFormatServer/Session/StandardIOSession.h"

// https://stackoverflow.com/questions/1598985/c-read-binary-stdin

#ifdef _WIN32

# include <io.h>

# include <fcntl.h>

# define SET_BINARY_MODE() _setmode(_fileno(stdin), _O_BINARY);\

_setmode(_fileno(stdout), _O_BINARY)

#else

# define SET_BINARY_MODE() (void)0

#endif

using namespace asio::ip;

int main(int argc, char** argv)

{

auto& ioc = LanguageClient::GetInstance().GetIOContext();

if (argc > 1)

{

int port = std::stoi(argv[1]);

tcp::acceptor acceptor(ioc, tcp::endpoint(tcp::v4(), port));

auto socket = acceptor.accept();

LanguageClient::GetInstance().SetSession(std::make_shared<SocketIOSession>(std::move(socket)));

}

else

{

SET_BINARY_MODE();

LanguageClient::GetInstance().SetSession(std::make_shared<StandardIOSession>());

}

return LanguageClient::GetInstance().Run();

}