From 3db94e44a0881a4f911afbe1c8d82931a4b073f7 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Mon, 20 May 2024 20:25:55 +0200
Subject: [PATCH] refactor(lsp): refactored again :)

---
 lua/lazyvim/plugins/lsp/init.lua |  2 +-
 lua/lazyvim/util/lsp.lua         | 68 ++++++++++++++++----------------
 2 files changed, 36 insertions(+), 34 deletions(-)

diff --git a/lua/lazyvim/plugins/lsp/init.lua b/lua/lazyvim/plugins/lsp/init.lua
index f99c997c..5cc32494 100644
--- a/lua/lazyvim/plugins/lsp/init.lua
+++ b/lua/lazyvim/plugins/lsp/init.lua
@@ -121,7 +121,7 @@ return {
         require("lazyvim.plugins.lsp.keymaps").on_attach(client, buffer)
       end)
 
-      LazyVim.lsp.setup_dynamic_capability()
+      LazyVim.lsp.setup()
       LazyVim.lsp.on_dynamic_capability(require("lazyvim.plugins.lsp.keymaps").on_attach)
 
       LazyVim.lsp.words.setup(opts.document_highlight)
diff --git a/lua/lazyvim/util/lsp.lua b/lua/lazyvim/util/lsp.lua
index af04b1c4..6479f1df 100644
--- a/lua/lazyvim/util/lsp.lua
+++ b/lua/lazyvim/util/lsp.lua
@@ -34,7 +34,10 @@ function M.on_attach(on_attach)
   })
 end
 
-function M.setup_dynamic_capability()
+---@type table<string, table<vim.lsp.Client, table<number, boolean>>>
+M._supports_method = {}
+
+function M.setup()
   local register_capability = vim.lsp.handlers["client/registerCapability"]
   vim.lsp.handlers["client/registerCapability"] = function(err, res, ctx)
     ---@diagnostic disable-next-line: no-unknown
@@ -49,6 +52,24 @@ function M.setup_dynamic_capability()
     end
     return ret
   end
+  M.on_attach(M._check_methods)
+  M.on_dynamic_capability(M._check_methods)
+end
+
+---@param client vim.lsp.Client
+function M._check_methods(client, buffer)
+  for method, clients in pairs(M._supports_method) do
+    clients[client] = clients[client] or {}
+    if not clients[client][buffer] then
+      if client.supports_method(method, { bufnr = buffer }) then
+        clients[client][buffer] = true
+        vim.api.nvim_exec_autocmds("User", {
+          pattern = "LspSupportsMethod",
+          data = { client_id = client.id, buffer = buffer, method = method },
+        })
+      end
+    end
+  end
 end
 
 ---@param fn fun(client:vim.lsp.Client, buffer):boolean?
@@ -67,33 +88,20 @@ function M.on_dynamic_capability(fn, opts)
   })
 end
 
-M._on_supports_method_id = 0
-
 ---@param method string
 ---@param fn fun(client:vim.lsp.Client, buffer)
 function M.on_supports_method(method, fn)
-  M.on_attach(function(client, buffer)
-    if client.supports_method(method, { bufnr = buffer }) then
-      fn(client, buffer)
-    else
-      M._on_supports_method_id = M._on_supports_method_id + 1
-      local id = M._on_supports_method_id
-      local group = vim.api.nvim_create_augroup("on_supports_method_" .. id, { clear = true })
-      M.on_dynamic_capability(function(c, b)
-        if c == client and b == buffer and client.supports_method(method, { bufnr = buffer }) then
-          fn(client, buffer)
-          pcall(vim.api.nvim_del_augroup_by_id, group)
-        end
-      end, { group = group })
-      vim.api.nvim_create_autocmd({ "LspDetach", "BufDelete" }, {
-        group = group,
-        buffer = buffer,
-        callback = function()
-          pcall(vim.api.nvim_del_augroup_by_id, group)
-        end,
-      })
-    end
-  end)
+  M._supports_method[method] = M._supports_method[method] or setmetatable({}, { __mode = "k" })
+  return vim.api.nvim_create_autocmd("User", {
+    pattern = "LspSupportsMethod",
+    callback = function(args)
+      local client = vim.lsp.get_client_by_id(args.data.client_id)
+      local buffer = args.data.buffer ---@type number
+      if client and method == args.data.method then
+        return fn(client, buffer)
+      end
+    end,
+  })
 end
 
 ---@param from string
@@ -207,10 +215,9 @@ function M.words.setup(opts)
     return handler(err, result, ctx, config)
   end
 
-  M.on_supports_method("textDocument/documentHighlight", function(client, buf)
-    local group = vim.api.nvim_create_augroup("lsp_word_" .. buf, { clear = true })
+  M.on_supports_method("textDocument/documentHighlight", function(_, buf)
     vim.api.nvim_create_autocmd({ "CursorHold", "CursorHoldI", "CursorMoved", "CursorMovedI" }, {
-      group = group,
+      group = vim.api.nvim_create_augroup("lsp_word_" .. buf, { clear = true }),
       buffer = buf,
       callback = function(ev)
         if not M.words.at() then
@@ -222,11 +229,6 @@ function M.words.setup(opts)
         end
       end,
     })
-    vim.api.nvim_create_autocmd("LspDetach", {
-      callback = function()
-        vim.api.nvim_create_augroup("lsp_word_" .. buf, { clear = true })
-      end,
-    })
   end)
 end