feat(lsp): support workspace/codeLens/refresh server request

Author: matoous

helix-lsp/src/lib.rs

@@ -476,6 +476,7 @@ pub enum MethodCall {
     UnregisterCapability(lsp::UnregistrationParams),
     ShowDocument(lsp::ShowDocumentParams),
     WorkspaceDiagnosticRefresh,
+    CodeLensRefresh,
 }
 
 impl MethodCall {
@@ -508,6 +509,7 @@ impl MethodCall {
                 Self::ShowDocument(params)
             }
             lsp::request::WorkspaceDiagnosticRefresh::METHOD => Self::WorkspaceDiagnosticRefresh,
+            lsp::request::CodeLensRefresh::METHOD => Self::CodeLensRefresh,
             _ => {
                 return Err(Error::Unhandled);
             }

helix-term/src/application.rs

@@ -1123,6 +1123,26 @@ impl Application {
                             );
                         }
 
+                        Ok(serde_json::Value::Null)
+                    }
+                    Ok(MethodCall::CodeLensRefresh) => {
+                        let language_server = language_server!().id();
+
+                        let documents: Vec<_> = self
+                            .editor
+                            .documents
+                            .values()
+                            .filter(|x| x.supports_language_server(language_server))
+                            .map(|x| x.id())
+                            .collect();
+
+                        for document in documents {
+                            handlers::code_lenses::request_document_code_lenses(
+                                &mut self.editor,
+                                document,
+                            );
+                        }
+
                         Ok(serde_json::Value::Null)
                     }
                 };

helix-term/src/handlers.rs

@@ -15,7 +15,7 @@ pub use helix_view::handlers::{word_index, Handlers};
 use self::document_colors::DocumentColorsHandler;
 
 mod auto_save;
-mod code_lenses;
+pub mod code_lenses;
 pub mod completion;
 pub mod diagnostics;
 mod document_colors;

helix-term/src/handlers/code_lenses.rs

@@ -44,7 +44,7 @@ impl helix_event::AsyncHook for DocumentCodeLensesHandler {
     }
 }
 
-fn request_document_code_lenses(editor: &mut Editor, doc_id: DocumentId) {
+pub fn request_document_code_lenses(editor: &mut Editor, doc_id: DocumentId) {
     if !editor.config().lsp.display_code_lenses {
         return;
     }