fix(context): in_treesitter_capture (#1211)

* fix(context): `in_treesitter_capture`

`in_treesitter_capture` used to compare a node:type() with what should
be a capture, returning always a falsy value.
The example https://github.com/hrsh7th/nvim-cmp/wiki/Advanced-techniques#disabling-completion-in-certain-contexts-such-as-comments
did not work as expected, being in_treesitter_capture("comment") always
not true.
Now it works as expected.

* feat(context): `in...capture`, `in...group` table

`in_treesitter_capture` and `in_syntax_group` can take a list of string
as argument.

* refactor(context): in...capture use vim.treesitter

Use vim.treesitter.get_captures_at_cursor()

* fix(context): get_captures_at_cursor -> ..._at_pos

get_captures_at_cursor() sometimes fails. Do it explicitly with
get_captures_at_pos(buf, row, col)

* refactor(context): vim.fn -> vim.api

Get row and col using vim.api and not vim.fn in in_syntax_group, as done
in in_treesitter_capture

lua/cmp/config/context.lua

@@ -1,65 +1,55 @@
 local context = {}
 
 ---Check if cursor is in syntax group
----@param group string
+---@param group string | []string
 ---@return boolean
 context.in_syntax_group = function(group)
-  local lnum, col = vim.fn.line('.'), math.min(vim.fn.col('.'), #vim.fn.getline('.'))
+  local row, col = unpack(vim.api.nvim_win_get_cursor(0))
-  for _, syn_id in ipairs(vim.fn.synstack(lnum, col)) do
+  if not vim.api.nvim_get_mode().mode == 'i' then
+    col = col + 1
+  end
+
+  for _, syn_id in ipairs(vim.fn.synstack(row, col)) do
     syn_id = vim.fn.synIDtrans(syn_id) -- Resolve :highlight links
-    if vim.fn.synIDattr(syn_id, 'name') == group then
+    local g = vim.fn.synIDattr(syn_id, 'name')
+    if type(group) == 'string' and g == group then
+      return true
+    elseif type(group) == 'table' and vim.tbl_contains(group, g) then
       return true
     end
   end
+
   return false
 end
 
 ---Check if cursor is in treesitter capture
----@param capture string
+---@param capture string | []string
 ---@return boolean
 context.in_treesitter_capture = function(capture)
-  local highlighter = require('vim.treesitter.highlighter')
-  local ts_utils = require('nvim-treesitter.ts_utils')
   local buf = vim.api.nvim_get_current_buf()
-
   local row, col = unpack(vim.api.nvim_win_get_cursor(0))
   row = row - 1
   if vim.api.nvim_get_mode().mode == 'i' then
     col = col - 1
   end
 
-  local self = highlighter.active[buf]
+  local captures_at_cursor = vim.tbl_map(function(x)
-  if not self then
+    return x.capture
+  end, require('vim.treesitter').get_captures_at_pos(buf, row, col))
+
+  if vim.tbl_isempty(captures_at_cursor) then
     return false
-  end
+  elseif type(capture) == 'string' and vim.tbl_contains(captures_at_cursor, capture) then
-
+    return true
-  local node_types = {}
+  elseif type(capture) == 'table' then
-
+    for _, v in ipairs(capture) do
-  self.tree:for_each_tree(function(tstree, tree)
+      if vim.tbl_contains(captures_at_cursor, v) then
-    if not tstree then
+        return true
-      return
-    end
-
-    local root = tstree:root()
-    local root_start_row, _, root_end_row, _ = root:range()
-    if root_start_row > row or root_end_row < row then
-      return
-    end
-
-    local query = self:get_query(tree:lang())
-    if not query:query() then
-      return
-    end
-
-    local iter = query:query():iter_captures(root, self.bufnr, row, row + 1)
-    for _, node, _ in iter do
-      if ts_utils.is_in_node_range(node, row, col) then
-        table.insert(node_types, node:type())
       end
     end
-  end, true)
+  end
 
-  return vim.tbl_contains(node_types, capture)
+  return false
 end
 
 return context