﻿
#include "LibLsp/lsp/ProcessIoService.h"
#ifdef STDIO_CLIENT_EXAMPLE

#include <boost/program_options.hpp>
#include "LibLsp/lsp/general/exit.h"
#include "LibLsp/lsp/textDocument/declaration_definition.h"

#include "LibLsp/lsp/textDocument/signature_help.h"
#include "LibLsp/lsp/general/initialize.h"
#include "LibLsp/lsp/ProtocolJsonHandler.h"
#include "LibLsp/lsp/textDocument/typeHierarchy.h"
#include "LibLsp/lsp/AbsolutePath.h"
#include "LibLsp/lsp/textDocument/resolveCompletionItem.h"
#include <network/uri.hpp>
#include "LibLsp/JsonRpc/Endpoint.h"
#include "LibLsp/JsonRpc/stream.h"
#include "LibLsp/JsonRpc/TcpServer.h"
#include "LibLsp/lsp/textDocument/document_symbol.h"
#include "LibLsp/lsp/workspace/execute_command.h"
#include <boost/process.hpp>
#include <boost/process/windows.hpp>
#include <boost/filesystem.hpp>
#include <boost/asio.hpp>
#include <iostream>


using namespace boost::asio::ip;
using namespace std;
class DummyLog :public lsp::Log
{
public:

	void log(Level level, std::wstring&& msg)
	{
		
		std::wcerr << msg << std::endl;
	};
	void log(Level level, const std::wstring& msg)
	{
		std::wcerr << msg << std::endl;
	};
	void log(Level level, std::string&& msg)
	{
		std::cerr << msg << std::endl;
	};
	void log(Level level, const std::string& msg)
	{
		std::cerr << msg << std::endl;
	};
};

struct boost_process_ipstream : lsp::base_istream< boost::process::ipstream >
{
	explicit boost_process_ipstream(boost::process::ipstream& _t)
		: base_istream<boost::process::ipstream>(_t)
	{
	}

	std::string what() override
	{
		return {};
	}
	void clear() override
	{
		
	}
};
struct boost_process_opstream : lsp::base_ostream< boost::process::opstream >
{
	explicit boost_process_opstream(boost::process::opstream& _t)
		: lsp::base_ostream<boost::process::opstream>(_t)
	{
	}

	std::string what() override
	{
		return {};
	}
	void clear() override
	{

	}
};
class Client
{
public:
	Client(std::string& execPath,const std::vector<std::string>& args) 
		:point(protocol_json_handler, endpoint, _log)
	{
		std::error_code ec;
		namespace bp = boost::process;
		c = std::make_shared<bp::child>(asio_io.getIOService(), execPath, 
			bp::args = args,
			ec,
			bp::windows::show_normal,
			bp::std_out > read_from_service,
			bp::std_in < write_to_service,
			bp::on_exit([&](int exit_code, const std::error_code& ec_in){
				
			}));
		if (ec)
		{
			// fail
			_log.log(lsp::Log::Level::SEVERE, "Start server failed.");
		}
		else {
			//success
			point.startProcessingMessages(read_from_service_proxy, write_to_service_proxy);
		}
	}
	~Client()
	{
		point.Stop();
		::Sleep(1000);
		
	}

	lsp::ProcessIoService asio_io;
	std::shared_ptr < lsp::ProtocolJsonHandler >  protocol_json_handler = std::make_shared< lsp::ProtocolJsonHandler>();
	DummyLog _log;

	std::shared_ptr<GenericEndpoint>  endpoint = std::make_shared<GenericEndpoint>(_log);

	boost::process::opstream write_to_service;
	boost::process::ipstream   read_from_service;

	std::shared_ptr<lsp::ostream> write_to_service_proxy = std::make_shared<boost_process_opstream>(write_to_service);
	std::shared_ptr<lsp::istream>  read_from_service_proxy = std::make_shared< boost_process_ipstream >(read_from_service);

	std::shared_ptr<boost::process::child> c;
	RemoteEndPoint point;
};

int main(int argc, char* argv[])
{
	using namespace  boost::program_options;
	options_description desc("Allowed options");
	desc.add_options()
		("help,h", "produce help message")
		("execPath", value<string>(), "LSP server's path");
	


	variables_map vm;
	try {
		store(parse_command_line(argc, argv, desc), vm);
	}
	catch (std::exception& e) {
		std::cout << "Undefined input.Reason:" << e.what() << std::endl;
		return 0;
	}
	notify(vm);


	if (vm.count("help"))
	{
		cout << desc << endl;
		return 1;
	}
	string execPath;
	if (vm.count("execPath"))
	{
		execPath = vm["execPath"].as<string>();
	}
	else
	{
		execPath = "STDIO_SERVER_EXAMPLE.exe";
	}

	Client client(execPath, {});
	{
		td_initialize::request req;
		auto rsp = client.point.waitResponse(req);
		if (rsp)
		{
			std::cerr << rsp->ToJson() << std::endl;
		}
		else
		{
			std::cerr << "get initialze  response time out" << std::endl;
		}
	}
	{
		td_definition::request req;
		auto future_rsp = client.point.send(req);
		auto state = future_rsp.wait_for(std::chrono::seconds(4));
		if (std::future_status::timeout == state)
		{
			std::cerr << "get textDocument/definition  response time out" << std::endl;
			return 0;
		}
		auto rsp = future_rsp.get();
		if (rsp.is_error)
		{
			std::cerr << "get textDocument/definition  response error" << std::endl;

		}
		else
		{
			std::cerr << rsp.response.ToJson() << std::endl;
		}
	}
	{
		td_initialize::request req;
		auto rsp = client.point.waitResponse(req);
		if (rsp)
		{
			std::cerr << rsp->ToJson() << std::endl;
		}
		else
		{
			std::cerr << "get initialze  response time out" << std::endl;
		}
	}
	Notify_Exit::notify notify;
	client.point.send(notify);
	return 0;
}
#endif

