mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 11:26:18 -04:00
139 lines
4.9 KiB
C++
139 lines
4.9 KiB
C++
#include "lsp_router.h"
|
|
|
|
#include "common/log/log.h"
|
|
|
|
#include "lsp/handlers/initialize.h"
|
|
#include "lsp/protocol/error_codes.h"
|
|
#include "text_document/completion.h"
|
|
#include "text_document/document_symbol.h"
|
|
#include "text_document/document_synchronization.h"
|
|
#include "text_document/go_to.h"
|
|
#include "text_document/hover.h"
|
|
|
|
#include "third-party/fmt/core.h"
|
|
|
|
LSPRoute::LSPRoute() : m_route_type(LSPRouteType::NOOP) {}
|
|
|
|
LSPRoute::LSPRoute(std::function<void(Workspace&, json)> notification_handler)
|
|
: m_route_type(LSPRouteType::NOTIFICATION), m_notification_handler(notification_handler) {}
|
|
|
|
LSPRoute::LSPRoute(std::function<void(Workspace&, json)> notification_handler,
|
|
std::function<std::optional<json>(Workspace&, json)> post_notification_publish)
|
|
: m_route_type(LSPRouteType::NOTIFICATION),
|
|
m_notification_handler(notification_handler),
|
|
m_post_notification_publish(post_notification_publish) {}
|
|
|
|
LSPRoute::LSPRoute(std::function<std::optional<json>(Workspace&, int, json)> request_handler)
|
|
: m_route_type(LSPRouteType::REQUEST_RESPONSE), m_request_handler(request_handler) {}
|
|
|
|
void LSPRouter::init_routes() {
|
|
m_routes["initialize"] = LSPRoute(initialize_handler);
|
|
m_routes["initialize"].m_generic_post_action = [](Workspace& workspace) {
|
|
workspace.set_initialized(true);
|
|
};
|
|
m_routes["initialized"] = LSPRoute();
|
|
m_routes["textDocument/documentSymbol"] = LSPRoute(document_symbols_handler);
|
|
m_routes["textDocument/didOpen"] = LSPRoute(did_open_handler, did_open_push_diagnostics);
|
|
m_routes["textDocument/didChange"] = LSPRoute(did_change_handler, did_change_push_diagnostics);
|
|
m_routes["textDocument/didClose"] = LSPRoute(did_close_handler);
|
|
m_routes["textDocument/hover"] = LSPRoute(hover_handler);
|
|
m_routes["textDocument/definition"] = LSPRoute(go_to_definition_handler);
|
|
m_routes["textDocument/completion"] = LSPRoute(get_completions_handler);
|
|
}
|
|
|
|
json error_resp(ErrorCodes error_code, const std::string& error_message) {
|
|
json error{
|
|
{"code", static_cast<int>(error_code)},
|
|
{"message", error_message},
|
|
};
|
|
return json{{"error", error}};
|
|
}
|
|
|
|
std::string LSPRouter::make_response(const json& result) {
|
|
json content = result;
|
|
content["jsonrpc"] = "2.0";
|
|
|
|
std::string header;
|
|
header.append("Content-Length: " + std::to_string(content.dump().size()) + "\r\n");
|
|
header.append("Content-Type: application/vscode-jsonrpc;charset=utf-8\r\n");
|
|
header.append("\r\n");
|
|
return header + content.dump();
|
|
}
|
|
|
|
std::optional<std::vector<std::string>> LSPRouter::route_message(
|
|
const MessageBuffer& message_buffer,
|
|
AppState& appstate) {
|
|
const json& body = message_buffer.body();
|
|
const auto method = body.at("method").get<std::string>();
|
|
|
|
// If the workspace has not yet been initialized but the client sends a
|
|
// message that doesn't have method "initialize" then we'll return an error
|
|
// as per the LSP spec.
|
|
if (method != "initialize" && !appstate.workspace.is_initialized()) {
|
|
auto error = {
|
|
make_response(error_resp(ErrorCodes::ServerNotInitialized, "Server not yet initialized."))};
|
|
return std::make_optional(error);
|
|
}
|
|
|
|
// Exit early if we can't handle the route
|
|
if (m_routes.find(method) == m_routes.end()) {
|
|
lg::warn("Method not supported '{}'", method);
|
|
auto error = {make_response(
|
|
error_resp(ErrorCodes::MethodNotFound, fmt::format("Method '{}' not supported", method)))};
|
|
return std::make_optional(error);
|
|
}
|
|
|
|
try {
|
|
auto& route = m_routes.at(method);
|
|
std::vector<json> resp_bodies;
|
|
// Handle the request/notificiation
|
|
switch (route.m_route_type) {
|
|
case LSPRouteType::NOOP:
|
|
break;
|
|
case LSPRouteType::NOTIFICATION:
|
|
route.m_notification_handler(appstate.workspace, body["params"]);
|
|
break;
|
|
case LSPRouteType::REQUEST_RESPONSE:
|
|
auto resp_body = route.m_request_handler(appstate.workspace, body["id"], body["params"]);
|
|
json resp;
|
|
resp["id"] = body["id"];
|
|
if (resp_body) {
|
|
resp["result"] = resp_body.value();
|
|
} else {
|
|
resp["result"] = nullptr;
|
|
}
|
|
resp_bodies.push_back(resp);
|
|
break;
|
|
}
|
|
|
|
// Run any publish we need to do after the fact
|
|
if (route.m_post_notification_publish) {
|
|
auto resp = route.m_post_notification_publish.value()(appstate.workspace, body["params"]);
|
|
if (resp) {
|
|
resp_bodies.push_back(resp.value());
|
|
}
|
|
}
|
|
|
|
// Run any generic post action
|
|
if (route.m_generic_post_action) {
|
|
route.m_generic_post_action.value()(appstate.workspace);
|
|
}
|
|
|
|
// Serialize all payloads with headers
|
|
std::vector<std::string> resps;
|
|
for (const auto& body : resp_bodies) {
|
|
resps.push_back(make_response(body));
|
|
}
|
|
|
|
// Return
|
|
if (resps.empty()) {
|
|
return {};
|
|
}
|
|
return resps;
|
|
} catch (std::exception& e) {
|
|
lg::error("Unexpected exception occurred - {} | {}", e.what(), body.dump());
|
|
// TODO - return an error with the message
|
|
return {};
|
|
}
|
|
}
|