Add workspace command picker

Author: MDeiml

helix-lsp/src/client.rs

@@ -298,6 +298,9 @@ impl Client {
                         dynamic_registration: Some(false),
                         ..Default::default()
                     }),
+                    execute_command: Some(lsp::DynamicRegistrationClientCapabilities {
+                        dynamic_registration: Some(false),
+                    }),
                     ..Default::default()
                 }),
                 text_document: Some(lsp::TextDocumentClientCapabilities {

helix-term/src/commands.rs

@@ -268,6 +268,7 @@ impl MappableCommand {
         file_picker, "Open file picker",
         file_picker_in_current_directory, "Open file picker at current working directory",
         code_action, "Perform code action",
+        workspace_command_picker, "Open workspace command picker",
         buffer_picker, "Open buffer picker",
         jumplist_picker, "Open jumplist picker",
         symbol_picker, "Open symbol picker",

helix-term/src/commands/lsp.rs

@@ -603,6 +603,42 @@ pub fn code_action(cx: &mut Context) {
         },
     )
 }
+
+impl ui::menu::Item for lsp::Command {
+    type Data = ();
+    fn label(&self, _data: &Self::Data) -> Spans {
+        self.title.as_str().into()
+    }
+}
+
+pub fn workspace_command_picker(cx: &mut Context) {
+    let (_, doc) = current!(cx.editor);
+
+    let language_server = language_server!(cx.editor, doc);
+
+    let options = match &language_server.capabilities().execute_command_provider {
+        Some(options) => options,
+        None => return,
+    };
+    let commands = options
+        .commands
+        .iter()
+        .map(|command| lsp::Command {
+            title: command.clone(),
+            command: command.clone(),
+            arguments: None,
+        })
+        .collect::<Vec<_>>();
+    cx.callback = Some(Box::new(
+        |compositor: &mut Compositor, _cx: &mut compositor::Context| {
+            let picker = ui::Picker::new(commands, (), |cx, command, _action| {
+                execute_lsp_command(cx.editor, command.clone());
+            });
+            compositor.push(Box::new(overlayed(picker)))
+        },
+    ));
+}
+
 pub fn execute_lsp_command(editor: &mut Editor, cmd: lsp::Command) {
     let doc = doc!(editor);
     let language_server = language_server!(editor, doc);