summaryrefslogtreecommitdiff
path: root/lsp.c
diff options
context:
space:
mode:
Diffstat (limited to 'lsp.c')
-rw-r--r--lsp.c313
1 files changed, 232 insertions, 81 deletions
diff --git a/lsp.c b/lsp.c
index bff1ed0..94978ec 100644
--- a/lsp.c
+++ b/lsp.c
@@ -1,6 +1,8 @@
-// @TODO: documentation
-// @TODO : make sure offsets are utf-16!
-// @TODO: maximum queue size for requests/responses just in case?
+// @TODO:
+// - make json_object_get take value not pointer
+// - documentation
+// - make sure offsets are utf-16!
+// - maximum queue size for requests/responses just in case?
typedef enum {
LSP_REQUEST,
@@ -66,11 +68,45 @@ typedef struct {
} data;
} LSPRequest;
+// info we want to keep track of about a request,
+// so we can deal with the response appropriately.
+typedef struct {
+ u64 id;
+ LSPRequestType type;
+} LSPRequestTrackedInfo;
+
+typedef struct {
+ u32 offset;
+} LSPResponseString;
+
+typedef struct {
+ LSPResponseString label;
+ // note: the items are sorted here in this file.
+ LSPResponseString sort_by;
+} LSPCompletionItem;
+
+typedef struct {
+ // dynamic array
+ LSPCompletionItem *items;
+} LSPResponseCompletion;
+
+typedef LSPRequestType LSPResponseType;
+typedef struct {
+ LSPResponseType type;
+ // LSP responses tend to have a lot of strings.
+ // to avoid doing a ton of allocations+frees,
+ // they're all stored here.
+ char *string_data;
+ union {
+ LSPResponseCompletion completion;
+ } data;
+} LSPResponse;
+
typedef struct {
LSPMessageType type;
union {
LSPRequest request;
- JSON response;
+ LSPResponse response;
} u;
} LSPMessage;
@@ -81,6 +117,8 @@ typedef struct {
SDL_mutex *messages_mutex;
LSPRequest *requests_client2server;
LSPRequest *requests_server2client;
+ // only applicable for client-to-server requests
+ LSPRequestTrackedInfo *requests_tracked_info;
SDL_mutex *requests_mutex;
bool initialized; // has the response to the initialize request been sent?
SDL_Thread *communication_thread;
@@ -144,12 +182,65 @@ static const char *lsp_language_id(Language lang) {
return "text";
}
+
+static void lsp_position_free(LSPDocumentPosition *position) {
+ free(position->path);
+}
+
+static void lsp_request_free(LSPRequest *r) {
+ switch (r->type) {
+ case LSP_NONE:
+ assert(0);
+ break;
+ case LSP_INITIALIZE:
+ case LSP_INITIALIZED:
+ case LSP_SHUTDOWN:
+ case LSP_EXIT:
+ break;
+ case LSP_COMPLETION: {
+ LSPRequestCompletion *completion = &r->data.completion;
+ lsp_position_free(&completion->position);
+ } break;
+ case LSP_OPEN: {
+ LSPRequestOpen *open = &r->data.open;
+ free(open->filename);
+ free(open->file_contents);
+ } break;
+ case LSP_SHOW_MESSAGE:
+ case LSP_LOG_MESSAGE:
+ free(r->data.message.message);
+ break;
+ }
+}
+
+static void lsp_response_free(LSPResponse *r) {
+ arr_free(r->string_data);
+ switch (r->type) {
+ case LSP_COMPLETION:
+ arr_free(r->data.completion.items);
+ break;
+ default:
+ break;
+ }
+}
+
+void lsp_message_free(LSPMessage *message) {
+ switch (message->type) {
+ case LSP_REQUEST:
+ lsp_request_free(&message->u.request);
+ break;
+ case LSP_RESPONSE:
+ lsp_response_free(&message->u.response);
+ break;
+ }
+ memset(message, 0, sizeof *message);
+}
+
// technically there are "requests" and "notifications"
// notifications are different in that they don't have IDs and don't return responses.
-// the distinction isn't super important to us though.
+// this function handles both.
// returns the ID of the request
-static u64 write_request(LSP *lsp, const LSPRequest *request) {
- unsigned long long id = lsp->request_id++;
+static void write_request(LSP *lsp, const LSPRequest *request) {
StrBuilder builder = str_builder_new();
@@ -159,6 +250,20 @@ static u64 write_request(LSP *lsp, const LSPRequest *request) {
str_builder_append(&builder, "{\"jsonrpc\":\"2.0\",");
+ bool is_notification = request->type == LSP_INITIALIZED
+ || request->type == LSP_EXIT;
+ if (!is_notification) {
+ unsigned long long id = lsp->request_id++;
+ str_builder_appendf(&builder, "\"id\":%llu,", id);
+ LSPRequestTrackedInfo info = {
+ .id = id,
+ .type = request->type
+ };
+ SDL_LockMutex(lsp->requests_mutex);
+ arr_add(lsp->requests_tracked_info, info);
+ SDL_UnlockMutex(lsp->requests_mutex);
+ }
+
switch (request->type) {
case LSP_NONE:
// these are server-to-client-only requests
@@ -168,10 +273,10 @@ static u64 write_request(LSP *lsp, const LSPRequest *request) {
break;
case LSP_INITIALIZE: {
str_builder_appendf(&builder,
- "\"id\":%llu,\"method\":\"initialize\",\"params\":{"
+ "\"method\":\"initialize\",\"params\":{"
"\"processId\":%d,"
"\"capabilities\":{}"
- "}", id, process_get_id());
+ "}", process_get_id());
} break;
case LSP_INITIALIZED:
str_builder_append(&builder, "\"method\":\"initialized\"");
@@ -182,13 +287,12 @@ static u64 write_request(LSP *lsp, const LSPRequest *request) {
char *escaped_text = json_escape(open->file_contents);
str_builder_appendf(&builder,
- "\"id\":%llu,\"method\":\"textDocument/open\",\"params\":{"
+ "\"method\":\"textDocument/open\",\"params\":{"
"textDocument:{"
"uri:\"file://%s\","
"languageId:\"%s\","
"version:1,"
"text:\"%s\"}}",
- id,
escaped_filename,
lsp_language_id(open->language),
escaped_text);
@@ -198,24 +302,23 @@ static u64 write_request(LSP *lsp, const LSPRequest *request) {
case LSP_COMPLETION: {
const LSPRequestCompletion *completion = &request->data.completion;
char *escaped_path = json_escape(completion->position.path);
- str_builder_appendf(&builder,"\"id\":%llu,\"method\":\"textDocument/completion\",\"params\":{"
- "textDocument:\"%s\","
- "position:{"
- "line:%lu,"
- "character:%lu"
+ str_builder_appendf(&builder,"\"method\":\"textDocument/completion\",\"params\":{"
+ "\"textDocument\":{\"uri\":\"file://%s\"},"
+ "\"position\":{"
+ "\"line\":%lu,"
+ "\"character\":%lu"
"}"
"}",
- id,
escaped_path,
(ulong)completion->position.line,
(ulong)completion->position.character);
free(escaped_path);
} break;
case LSP_SHUTDOWN:
- str_builder_appendf(&builder, "\"id\":%llu,\"method\":\"shutdown\"", id);
+ str_builder_append(&builder, "\"method\":\"shutdown\"");
break;
case LSP_EXIT:
- str_builder_appendf(&builder, "\"method\":\"exit\"");
+ str_builder_append(&builder, "\"method\":\"exit\"");
break;
}
@@ -242,8 +345,6 @@ static u64 write_request(LSP *lsp, const LSPRequest *request) {
process_write(&lsp->process, content, strlen(content));
str_builder_free(&builder);
-
- return (u64)id;
}
// figure out if data begins with a complete LSP response.
@@ -276,7 +377,7 @@ static bool parse_server2client_request(LSP *lsp, JSON *json, LSPRequest *reques
}
char str[64] = {0};
- json_string_get(json, &method.val.string, str, sizeof str);
+ json_string_get(json, method.val.string, str, sizeof str);
if (streq(str, "window/showMessage")) {
request->type = LSP_SHOW_MESSAGE;
@@ -303,7 +404,7 @@ static bool parse_server2client_request(LSP *lsp, JSON *json, LSPRequest *reques
LSPRequestMessage *m = &request->data.message;
m->type = (LSPWindowMessageType)mtype;
- m->message = json_string_get_alloc(json, &message.val.string);
+ m->message = json_string_get_alloc(json, message.val.string);
return true;
} else if (str_has_prefix(str, "$/")) {
// we can safely ignore this
@@ -313,19 +414,87 @@ static bool parse_server2client_request(LSP *lsp, JSON *json, LSPRequest *reques
return false;
}
+const char *lsp_response_string(const LSPResponse *response, LSPResponseString string) {
+ assert(string.offset < arr_len(response->string_data));
+ return &response->string_data[string.offset];
+}
+
+static LSPResponseString lsp_response_add_json_string(LSPResponse *response, const JSON *json, JSONString string) {
+ u32 offset = arr_len(response->string_data);
+ arr_set_len(response->string_data, offset + string.len + 1);
+ json_string_get(json, string, response->string_data + offset, string.len + 1);
+ return (LSPResponseString){
+ .offset = offset
+ };
+}
+
+static bool parse_completion(LSP *lsp, const JSON *json, LSPResponse *response) {
+ // deal with textDocument/completion response.
+ // result: CompletionItem[] | CompletionList | null
+ response->type = LSP_COMPLETION;
+ LSPResponseCompletion *completion = &response->data.completion;
+
+ JSONValue result = json_get(json, "result");
+ JSONValue items_value = {0};
+ switch (result.type) {
+ case JSON_NULL:
+ // no completions
+ return true;
+ case JSON_ARRAY:
+ items_value = result;
+ break;
+ case JSON_OBJECT:
+ items_value = json_object_get(json, &result.val.object, "items");
+ break;
+ default:
+ lsp_set_error(lsp, "Weird result type for textDocument/completion response: %s.", json_type_to_str(result.type));
+ break;
+ }
+
+ if (items_value.type != JSON_ARRAY) {
+ lsp_set_error(lsp, "Expected array for completion list, got %s", json_type_to_str(items_value.type));
+ return false;
+ }
+
+ JSONArray items = items_value.val.array;
+
+ arr_set_len(completion->items, items.len);
+
+ for (u32 i = 0; i < items.len; ++i) {
+ LSPCompletionItem *item = &completion->items[i];
+
+ JSONValue item_value = json_array_get(json, &items, i);
+ if (item_value.type != JSON_OBJECT) {
+ lsp_set_error(lsp, "Expected array for completion list, got %s", json_type_to_str(items_value.type));
+ return false;
+ }
+ JSONObject item_object = item_value.val.object;
+ JSONValue label_value = json_object_get(json, &item_object, "label");
+ if (label_value.type != JSON_STRING) {
+ lsp_set_error(lsp, "Expected string for completion label, got %s", json_type_to_str(label_value.type));
+ return false;
+ }
+ JSONString label = label_value.val.string;
+ item->label = lsp_response_add_json_string(response, json, label);
+ printf("%s\n",lsp_response_string(response,item->label));//@TODO
+ }
+
+ return true;
+}
+
+
static void process_message(LSP *lsp, JSON *json) {
- #if 1
+ #if 0
printf("\x1b[3m");
json_debug_print(json);
printf("\x1b[0m\n");
#endif
- bool keep_json = false;
-
+
JSONValue error = json_get(json, "error.message");
if (error.type == JSON_STRING) {
char err[256] = {0};
- json_string_get(json, &error.val.string, err, sizeof err);;
+ json_string_get(json, error.val.string, err, sizeof err);;
lsp_set_error(lsp, "%s", err);
goto ret;
}
@@ -347,19 +516,42 @@ static void process_message(LSP *lsp, JSON *json) {
.type = LSP_INITIALIZED,
.data = {0},
};
- u64 initialized_id = write_request(lsp, &initialized);
- // this should be the second request.
- (void)initialized_id;
- assert(initialized_id == 1);
+ write_request(lsp, &initialized);
// we can now send requests which have nothing to do with initialization
lsp->initialized = true;
} else {
- SDL_LockMutex(lsp->messages_mutex);
- LSPMessage *message = arr_addp(lsp->messages);
- message->type = LSP_RESPONSE;
- message->u.response = *json;
- SDL_UnlockMutex(lsp->messages_mutex);
- keep_json = true;
+ u64 id_no = (u64)id.val.number;
+ LSPRequestTrackedInfo tracked_info = {0};
+ SDL_LockMutex(lsp->requests_mutex);
+ arr_foreach_ptr(lsp->requests_tracked_info, LSPRequestTrackedInfo, info) {
+ if (info->id == id_no) {
+ // hey its the thing
+ tracked_info = *info;
+ arr_remove(lsp->requests_tracked_info, (u32)(info - lsp->requests_tracked_info));
+ break;
+ }
+ }
+ SDL_UnlockMutex(lsp->requests_mutex);
+
+ LSPResponse response = {0};
+ bool success = false;
+ switch (tracked_info.type) {
+ case LSP_COMPLETION:
+ success = parse_completion(lsp, json, &response);
+ break;
+ default:
+ // it's some response we don't care about
+ break;
+ }
+ if (success) {
+ SDL_LockMutex(lsp->messages_mutex);
+ LSPMessage *message = arr_addp(lsp->messages);
+ message->type = LSP_RESPONSE;
+ message->u.response = response;
+ SDL_UnlockMutex(lsp->messages_mutex);
+ } else {
+ lsp_response_free(&response);
+ }
}
} else if (json_has(json, "method")) {
LSPRequest request = {0};
@@ -374,8 +566,7 @@ static void process_message(LSP *lsp, JSON *json) {
lsp_set_error(lsp, "Bad message from server (no result, no method).");
}
ret:
- if (!keep_json)
- json_free(json);
+ json_free(json);
}
@@ -408,7 +599,7 @@ static void lsp_receive(LSP *lsp, size_t max_size) {
// kind of a hack. this is needed because arr_set_len zeroes the data.
arr_hdr_(lsp->received_data)->len = (u32)received_so_far;
lsp->received_data[received_so_far] = '\0';// null terminate
- #if 0
+ #if 1
printf("\x1b[3m%s\x1b[0m\n",lsp->received_data);
#endif
@@ -433,47 +624,6 @@ static void lsp_receive(LSP *lsp, size_t max_size) {
}
}
-static void lsp_position_free(LSPDocumentPosition *position) {
- free(position->path);
-}
-
-static void lsp_request_free(LSPRequest *r) {
- switch (r->type) {
- case LSP_NONE:
- assert(0);
- break;
- case LSP_INITIALIZE:
- case LSP_INITIALIZED:
- case LSP_SHUTDOWN:
- case LSP_EXIT:
- break;
- case LSP_COMPLETION: {
- LSPRequestCompletion *completion = &r->data.completion;
- lsp_position_free(&completion->position);
- } break;
- case LSP_OPEN: {
- LSPRequestOpen *open = &r->data.open;
- free(open->filename);
- free(open->file_contents);
- } break;
- case LSP_SHOW_MESSAGE:
- case LSP_LOG_MESSAGE:
- free(r->data.message.message);
- break;
- }
-}
-
-void lsp_message_free(LSPMessage *message) {
- switch (message->type) {
- case LSP_REQUEST:
- lsp_request_free(&message->u.request);
- break;
- case LSP_RESPONSE:
- json_free(&message->u.response);
- break;
- }
-}
-
// send requests.
static bool lsp_send(LSP *lsp) {
if (!lsp->initialized) {
@@ -574,6 +724,7 @@ bool lsp_create(LSP *lsp, const char *analyzer_command) {
lsp->quit_sem = SDL_CreateSemaphore(0);
lsp->messages_mutex = SDL_CreateMutex();
+ lsp->requests_mutex = SDL_CreateMutex();
lsp->communication_thread = SDL_CreateThread(lsp_communication_thread, "LSP communicate", lsp);
return true;
}
generated by cgit v1.2.3 (git 2.39.1) at 2024-12-26 15:50:38 +0000