diff options
Diffstat (limited to 'lsp-write.c')
| -rw-r--r-- | lsp-write.c | 120 |
1 files changed, 88 insertions, 32 deletions
diff --git a/lsp-write.c b/lsp-write.c index b66def9..ef60b91 100644 --- a/lsp-write.c +++ b/lsp-write.c @@ -217,6 +217,17 @@ static void write_key_range(JSONWriter *o, const char *key, LSPRange range) { write_range(o, range); } +static void write_workspace_folders(JSONWriter *o, char **workspace_folders) { + write_arr_start(o); + arr_foreach_ptr(workspace_folders, char *, folder) { + write_arr_elem_obj_start(o); + write_key_file_uri_direct(o, "uri", *folder); + write_key_string(o, "name", *folder); + write_obj_end(o); + } + write_arr_end(o); +} + static const char *lsp_request_method(LSPRequest *request) { switch (request->type) { case LSP_REQUEST_NONE: break; @@ -240,6 +251,8 @@ static const char *lsp_request_method(LSPRequest *request) { return "shutdown"; case LSP_REQUEST_EXIT: return "exit"; + case LSP_REQUEST_WORKSPACE_FOLDERS: + return "workspace/workspaceFolders"; } assert(0); return "$/ignore"; @@ -259,22 +272,54 @@ static bool request_type_is_notification(LSPRequestType type) { case LSP_REQUEST_SHOW_MESSAGE: case LSP_REQUEST_LOG_MESSAGE: case LSP_REQUEST_COMPLETION: + case LSP_REQUEST_WORKSPACE_FOLDERS: return false; } assert(0); return false; } + +static const size_t max_header_size = 64; +static JSONWriter message_writer_new(LSP *lsp) { + JSONWriter writer = json_writer_new(lsp); + // this is where our header will go + str_builder_append_null(&writer.builder, max_header_size); + return writer; +} + +static void message_writer_write_and_free(LSP *lsp, JSONWriter *o) { + StrBuilder builder = o->builder; + + // this is kind of hacky but it lets us send the whole request with one write call. + // probably not *actually* needed. i thought it would help fix an error but it didn't. + size_t content_length = str_builder_len(&builder) - max_header_size; + char content_length_str[32]; + sprintf(content_length_str, "%zu", content_length); + size_t header_size = strlen("Content-Length: \r\n\r\n") + strlen(content_length_str); + char *header = &builder.str[max_header_size - header_size]; + strcpy(header, "Content-Length: "); + strcat(header, content_length_str); + // we specifically DON'T want a null byte + memcpy(header + strlen(header), "\r\n\r\n", 4); + + char *content = header; + #if LSP_SHOW_C2S + printf("%s%s%s\n",term_bold(stdout),content,term_clear(stdout)); + #endif + + // @TODO: does write always write the full amount? probably not. this should be fixed. + process_write(&lsp->process, content, strlen(content)); + + str_builder_free(&builder); +} + // NOTE: don't call lsp_request_free after calling this function. // I will do it for you. static void write_request(LSP *lsp, LSPRequest *request) { - JSONWriter writer = json_writer_new(lsp); + JSONWriter writer = message_writer_new(lsp); JSONWriter *o = &writer; - u32 max_header_size = 64; - // this is where our header will go - str_builder_append_null(&o->builder, max_header_size); - write_obj_start(o); write_key_string(o, "jsonrpc", "2.0"); @@ -291,6 +336,7 @@ static void write_request(LSP *lsp, LSPRequest *request) { // these are server-to-client-only requests case LSP_REQUEST_SHOW_MESSAGE: case LSP_REQUEST_LOG_MESSAGE: + case LSP_REQUEST_WORKSPACE_FOLDERS: assert(0); break; case LSP_REQUEST_INITIALIZED: @@ -335,11 +381,13 @@ static void write_request(LSP *lsp, LSPRequest *request) { write_key_bool(o, "contextSupport", true); write_obj_end(o); write_obj_end(o); + write_key_obj_start(o, "workspace"); + write_key_bool(o, "workspaceFolders", true); + write_obj_end(o); write_obj_end(o); - write_key_file_uri_direct(o, "rootUri", lsp->root_dir); -// write_key_arr_start(o, "workspaceFolders"); -// write_arr_elem_obj_start(o); -// write_arr_end(o); + write_key_file_uri_direct(o, "rootUri", lsp->workspace_folders[0]); + write_key(o, "workspaceFolders"); + write_workspace_folders(o, lsp->workspace_folders); write_key_obj_start(o, "clientInfo"); write_key_string(o, "name", "ted"); write_obj_end(o); @@ -410,29 +458,7 @@ static void write_request(LSP *lsp, LSPRequest *request) { write_obj_end(o); - StrBuilder builder = o->builder; - - // this is kind of hacky but it lets us send the whole request with one write call. - // probably not *actually* needed. i thought it would help fix an error but it didn't. - size_t content_length = str_builder_len(&builder) - max_header_size; - char content_length_str[32]; - sprintf(content_length_str, "%zu", content_length); - size_t header_size = strlen("Content-Length: \r\n\r\n") + strlen(content_length_str); - char *header = &builder.str[max_header_size - header_size]; - strcpy(header, "Content-Length: "); - strcat(header, content_length_str); - // we specifically DON'T want a null byte - memcpy(header + strlen(header), "\r\n\r\n", 4); - - char *content = header; - #if LSP_SHOW_C2S - printf("%s%s%s\n",term_bold(stdout),content,term_clear(stdout)); - #endif - - // @TODO: does write always write the full amount? probably not. this should be fixed. - process_write(&lsp->process, content, strlen(content)); - - str_builder_free(&builder); + message_writer_write_and_free(lsp, o); if (is_notification) { lsp_request_free(request); @@ -442,3 +468,33 @@ static void write_request(LSP *lsp, LSPRequest *request) { SDL_UnlockMutex(lsp->requests_mutex); } } + +// NOTE: don't call lsp_response_free after calling this function. +// I will do it for you. +static void write_response(LSP *lsp, LSPResponse *response) { + + JSONWriter writer = message_writer_new(lsp); + JSONWriter *o = &writer; + LSPRequest *request = &response->request; + + if (request->id_string) + write_key_string(o, "id", request->id_string); + else + write_key_number(o, "id", request->id); + write_key_obj_start(o, "result"); + + switch (response->request.type) { + case LSP_REQUEST_WORKSPACE_FOLDERS: + write_workspace_folders(o, lsp->workspace_folders); + break; + default: + // this is not a valid client-to-server response. + assert(0); + break; + } + write_obj_end(o); + + message_writer_write_and_free(lsp, o); + + lsp_response_free(response); +} |