feat: debounce results from code actions

lua/null-ls/handlers.lua

@@ -0,0 +1,29 @@
+local utils = require("null-ls.utils")
+
+local M = {}
+
+function M.setup()
+    M.combine("textDocument/codeAction")
+end
+
+-- this will override a handler, batch results and debounce them
+function M.combine(method, ms)
+    ms = ms or 100
+    local orig = vim.lsp.handlers[method]
+
+    local results = {}
+
+    local handler = utils.debounce(ms, function()
+        if #results > 0 then
+            orig(nil, nil, results)
+            results = {}
+        end
+    end)
+
+    vim.lsp.handlers[method] = function(_, _, actions)
+        vim.list_extend(results, actions or {})
+        handler()
+    end
+end
+
+return M

lua/null-ls/init.lua

@@ -19,6 +19,7 @@ M.config = function(user_config)
     config.setup(user_config or {})
     require("null-ls.rpc").setup()
     require("null-ls.lspconfig").setup()
+    require("null-ls.handlers").setup()
 end
 
 M.setup = function(user_config)

lua/null-ls/utils.lua

@@ -166,4 +166,32 @@ M.table = {
     end,
 }
 
+function M.debounce(ms, fn)
+    local timer = vim.loop.new_timer()
+    return function(...)
+        local argv = { ... }
+        timer:start(ms, 0, function()
+            timer:stop()
+            vim.schedule_wrap(fn)(unpack(argv))
+        end)
+    end
+end
+
+function M.throttle(ms, fn)
+    local timer = vim.loop.new_timer()
+    local running = false
+    return function(...)
+        if not running then
+            local argv = { ... }
+            local argc = select("#", ...)
+
+            timer:start(ms, 0, function()
+                running = false
+                pcall(vim.schedule_wrap(fn), unpack(argv, 1, argc))
+            end)
+            running = true
+        end
+    end
+end
+
 return M