1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
|
#pragma once
#include <string>
#include <map>
#include <functional>
#include <LibLsp/JsonRpc/message.h>
class Reader;
using GenericRequestJsonHandler = std::function< std::unique_ptr<LspMessage>(Reader&) >;
using GenericResponseJsonHandler = std::function< std::unique_ptr<LspMessage>(Reader&) >;
using GenericNotificationJsonHandler = std::function< std::unique_ptr<LspMessage>(Reader&) >;
class MessageJsonHandler
{
public:
std::map< std::string, GenericRequestJsonHandler > method2request;
std::map< std::string, GenericResponseJsonHandler > method2response;
std::map< std::string, GenericNotificationJsonHandler > method2notification;
const GenericRequestJsonHandler* GetRequestJsonHandler(const char* methodInfo) const
{
const auto findIt = method2request.find(methodInfo);
return findIt == method2request.end() ? nullptr : &findIt->second;
}
void SetRequestJsonHandler(const std::string& methodInfo, GenericRequestJsonHandler handler)
{
method2request[methodInfo] = handler;
}
const GenericResponseJsonHandler* GetResponseJsonHandler(const char* methodInfo) const
{
const auto findIt = method2response.find(methodInfo);
return findIt == method2response.end() ? nullptr : &findIt->second;
}
void SetResponseJsonHandler(const std::string& methodInfo,GenericResponseJsonHandler handler)
{
method2response[methodInfo] = handler;
}
const GenericNotificationJsonHandler* GetNotificationJsonHandler(const char* methodInfo) const
{
const auto findIt = method2notification.find(methodInfo);
return findIt == method2notification.end() ? nullptr : &findIt->second;
}
void SetNotificationJsonHandler(const std::string& methodInfo, GenericNotificationJsonHandler handler)
{
method2notification[methodInfo] = handler;
}
std::unique_ptr<LspMessage> parseResponseMessage(const std::string&, Reader&);
std::unique_ptr<LspMessage> parseRequstMessage(const std::string&, Reader&);
bool resovleResponseMessage(Reader&, std::pair<std::string, std::unique_ptr<LspMessage>>& result);
std::unique_ptr<LspMessage> parseNotificationMessage(const std::string&, Reader&);
};
|