#pragma once
#include <map>
#include <functional>
#include <memory>
#include "MessageIssue.h"
struct LspMessage;
struct NotificationInMessage;
struct lsBaseOutMessage;
struct  RequestInMessage;

using GenericResponseHandler = std::function< bool(std::unique_ptr<LspMessage>) >;
using GenericRequestHandler = std::function< bool(std::unique_ptr<LspMessage>) >;
using GenericNotificationHandler = std::function< bool(std::unique_ptr<LspMessage>) >;

class Endpoint
{
public:
        virtual  ~Endpoint() = default;
        virtual  bool onRequest(std::unique_ptr<LspMessage>) = 0;
        virtual  bool notify(std::unique_ptr<LspMessage>) = 0;

        virtual  bool onResponse(const std::string&, std::unique_ptr<LspMessage>) = 0;
        virtual  void registerRequestHandler(const std::string&, GenericResponseHandler ) = 0;
        virtual  void registerNotifyHandler(const std::string&,  GenericNotificationHandler ) = 0;
};

class GenericEndpoint :public Endpoint
{

public:
        GenericEndpoint(lsp::Log& l):log(l){}
        bool notify(std::unique_ptr<LspMessage>) override;
        bool onResponse(const std::string&, std::unique_ptr<LspMessage>) override;

        bool onRequest(std::unique_ptr<LspMessage>) override;
        std::map< std::string, GenericRequestHandler > method2request;
        std::map< std::string, GenericResponseHandler  > method2response;
        std::map< std::string, GenericNotificationHandler  > method2notification;

          void registerRequestHandler(const std::string& method, GenericResponseHandler cb) override
        {
                  method2request[method] = cb;
        }

        void registerNotifyHandler(const std::string& method, GenericNotificationHandler cb) override
          {
                method2notification[method] = cb;
          }
        lsp::Log& log;



};