#include "LibLsp/lsp/ProcessIoService.h"

#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/filesystem.hpp>
#include <boost/asio.hpp>
#include <iostream>

#include <thread>

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::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();
       std::this_thread::sleep_for(std::chrono::milliseconds (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 = "StdIOServerExample";
       }

       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 (lsp::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;
}