feat(lsp): simpler API to deal with code actions

lua/lazyvim/plugins/extras/lang/python.lua

@@ -53,15 +53,7 @@ return {
           keys = {
             {
               "<leader>co",
-              function()
+              LazyVim.lsp.action["source.organizeImports"],
-                vim.lsp.buf.code_action({
-                  apply = true,
-                  context = {
-                    only = { "source.organizeImports" },
-                    diagnostics = {},
-                  },
-                })
-              end,
               desc = "Organize Imports",
             },
           },

lua/lazyvim/plugins/extras/lang/svelte.lua

@@ -31,15 +31,7 @@ return {
           keys = {
             {
               "<leader>co",
-              function()
+              LazyVim.lsp.action["source.organizeImports"],
-                vim.lsp.buf.code_action({
-                  apply = true,
-                  context = {
-                    only = { "source.organizeImports" },
-                    diagnostics = {},
-                  },
-                })
-              end,
               desc = "Organize Imports",
             },
           },

lua/lazyvim/plugins/extras/lang/typescript.lua

@@ -85,30 +85,22 @@ return {
             },
             {
               "<leader>co",
-              function()
+              LazyVim.lsp.action["source.organizeImports"],
-                require("vtsls").commands.organize_imports(0)
-              end,
               desc = "Organize Imports",
             },
             {
               "<leader>cM",
-              function()
+              LazyVim.lsp.action["source.addMissingImports.ts"],
-                require("vtsls").commands.add_missing_imports(0)
-              end,
               desc = "Add missing imports",
             },
             {
               "<leader>cu",
-              function()
+              LazyVim.lsp.action["source.removeUnused.ts"],
-                require("vtsls").commands.remove_unused_imports(0)
-              end,
               desc = "Remove unused imports",
             },
             {
               "<leader>cD",
-              function()
+              LazyVim.lsp.action["source.fixAll.ts"],
-                require("vtsls").commands.fix_all(0)
-              end,
               desc = "Fix all diagnostics",
             },
             {

lua/lazyvim/plugins/lsp/keymaps.lua

@@ -27,21 +27,7 @@ function M.get()
       { "<leader>cC", vim.lsp.codelens.refresh, desc = "Refresh & Display Codelens", mode = { "n" }, has = "codeLens" },
       { "<leader>cR", LazyVim.lsp.rename_file, desc = "Rename File", mode ={"n"}, has = { "workspace/didRenameFiles", "workspace/willRenameFiles" } },
       { "<leader>cr", vim.lsp.buf.rename, desc = "Rename", has = "rename" },
-      {
+      { "<leader>cA", LazyVim.lsp.action.source, desc = "Source Action", has = "codeAction" },
-        "<leader>cA",
-        function()
-          vim.lsp.buf.code_action({
-            context = {
-              only = {
-                "source",
-              },
-              diagnostics = {},
-            },
-          })
-        end,
-        desc = "Source Action",
-        has = "codeAction",
-      },
       { "]]", function() LazyVim.lsp.words.jump(vim.v.count1) end, has = "documentHighlight",
         desc = "Next Reference", cond = function() return LazyVim.lsp.words.enabled end },
       { "[[", function() LazyVim.lsp.words.jump(-vim.v.count1) end, has = "documentHighlight",

lua/lazyvim/util/lsp.lua

@@ -324,4 +324,17 @@ function M.words.jump(count, cycle)
   end
 end
 
+M.action = setmetatable({}, {
+  __index = function(_, action)
+    return function()
+      vim.lsp.buf.code_action({
+        apply = true,
+        context = {
+          only = { action },
+          diagnostics = {},
+        },
+      })
+    end
+  end,
+})
 return M