feat: move to new vim.lsp.semantic_tokens API

Author: mehalter

lua/astrolsp/init.lua

@@ -94,7 +94,8 @@ function M.on_attach(client, bufnr)
     end
   end
 
-  if client:supports_method("textDocument/semanticTokens/full", bufnr) and vim.lsp.semantic_tokens then
+  -- TODO: remove when dropping support for Neovim v0.11
+  if client:supports_method("textDocument/semanticTokens/full", bufnr) and not vim.lsp.semantic_tokens.enable then
     if M.config.features.semantic_tokens then
       if vim.b[bufnr].semantic_tokens == nil then vim.b[bufnr].semantic_tokens = true end
     else
@@ -283,6 +284,8 @@ function M.setup(opts)
   end
 
   vim.lsp.inlay_hint.enable(M.config.features.inlay_hints ~= false)
+  -- TODO: remove check when dropping support for Neovim v0.11
+  if vim.lsp.semantic_tokens.enable then vim.lsp.semantic_tokens.enable(M.config.features.semantic_tokens ~= false) end
 
   -- Set up tracking of signature help trigger characters
   local augroup = vim.api.nvim_create_augroup("track_signature_help_triggers", { clear = true })

lua/astrolsp/toggles.lua

@@ -62,19 +62,37 @@ end
 ---@param silent? boolean if true then don't sent a notification
 function M.buffer_semantic_tokens(bufnr, silent)
   bufnr = bufnr or 0
-  vim.b[bufnr].semantic_tokens = not vim.b[bufnr].semantic_tokens
-  local toggled = false
-  for _, client in ipairs(vim.lsp.get_clients { bufnr = bufnr }) do
-    if client:supports_method("textDocument/semanticTokens/full", bufnr) then
-      vim.lsp.semantic_tokens[vim.b[bufnr].semantic_tokens and "start" or "stop"](bufnr, client.id)
-      vim.lsp.semantic_tokens.force_refresh(bufnr)
-      toggled = true
+  if vim.lsp.semantic_tokens.enable then
+    vim.lsp.semantic_tokens.enable(not vim.lsp.semantic_tokens.is_enabled { bufnr = bufnr }, { bufnr = bufnr })
+    vim.lsp.semantic_tokens.force_refresh(bufnr)
+    ui_notify(
+      silent,
+      ("Buffer lsp semantic highlighting %s"):format(bool2str(vim.lsp.semantic_tokens.is_enabled { bufnr = bufnr }))
+    )
+  else -- TODO: remove when dropping support for Neovim v0.11
+    vim.b[bufnr].semantic_tokens = not vim.b[bufnr].semantic_tokens
+    local toggled = false
+    for _, client in ipairs(vim.lsp.get_clients { bufnr = bufnr }) do
+      if client:supports_method("textDocument/semanticTokens/full", bufnr) then
+        vim.lsp.semantic_tokens[vim.b[bufnr].semantic_tokens and "start" or "stop"](bufnr, client.id)
+        toggled = true
+      end
     end
+    if toggled then vim.lsp.semantic_tokens.force_refresh(bufnr) end
+    ui_notify(silent, ("Buffer lsp semantic highlighting %s"):format(bool2str(vim.b[bufnr].semantic_tokens)))
   end
-  ui_notify(
-    not toggled or silent,
-    ("Buffer lsp semantic highlighting %s"):format(bool2str(vim.b[bufnr].semantic_tokens))
-  )
+end
+
+--- Toggle global semantic token highlighting for all language servers that support it
+---@param silent? boolean if true then don't sent a notification
+function M.semantic_tokens(silent)
+  if not vim.lsp.semantic_tokens.enable then
+    ui_notify(silent, "Only available in Neovim v0.12+", vim.log.levels.WARN)
+    return
+  end
+  vim.lsp.semantic_tokens.enable(not vim.lsp.semantic_tokens.is_enabled())
+  vim.lsp.semantic_tokens.force_refresh()
+  ui_notify(silent, ("Global lsp semantic highlighting %s"):format(bool2str(vim.lsp.semantic_tokens.is_enabled())))
 end
 
 --- Toggle codelens