This sets our language server up to handle the codeLens requests coming from the language client. This just ensures that the server tells the client that sending codeLens requests is fine. Next we'll implement the actual functionality that resolves the code leneses.

Author: zth

server/src/server.ts

@@ -13,6 +13,7 @@ import {
   DidChangeConfigurationNotification,
   InitializeParams,
   InlayHintParams,
+  CodeLensParams,
 } from "vscode-languageserver-protocol";
 import * as utils from "./utils";
 import * as codeActions from "./codeActions";
@@ -411,6 +412,18 @@ function sendInlayHintsRefresh() {
   send(request);
 }
 
+function codeLens(msg: p.RequestMessage) {
+  const params = msg.params as p.CodeLensParams;
+  const filePath = fileURLToPath(params.textDocument.uri);
+
+  const response = utils.runAnalysisCommand(
+    filePath,
+    ["codeLens", filePath],
+    msg
+  );
+  return response;
+}
+
 function definition(msg: p.RequestMessage) {
   // https://microsoft.github.io/language-server-protocol/specifications/specification-current/#textDocument_definition
   let params = msg.params as p.DefinitionParams;
@@ -1001,6 +1014,9 @@ function onMessage(msg: p.Message) {
             full: true,
           },
           inlayHintProvider: extensionConfiguration.inlayHints.enable,
+          codeLensProvider: {
+            workDoneProgress: false
+          },
         },
       };
       let response: p.ResponseMessage = {
@@ -1086,6 +1102,12 @@ function onMessage(msg: p.Message) {
       if (extName === c.resExt) {
         send(inlayHint(msg));
       }
+    } else if (msg.method === p.CodeLensRequest.method) {
+      let params = msg.params as CodeLensParams;
+      let extName = path.extname(params.textDocument.uri);
+      if (extName === c.resExt) {
+        send(codeLens(msg));
+      }
     } else {
       let response: p.ResponseMessage = {
         jsonrpc: c.jsonrpcVersion,