1 files changed, 65 insertions, 0 deletions
diff --git a/start/cmp/lua/cmp/config/context.lua b/start/cmp/lua/cmp/config/context.lua
new file mode 100644
index 0000000..584f38a
--- /dev/null
+++ b/start/cmp/lua/cmp/config/context.lua
@@ -0,0 +1,65 @@
+local context = {}
+
+---Check if cursor is in syntax group
+---@param group string
+---@return boolean
+context.in_syntax_group = function(group)
+  local lnum, col = vim.fn.line('.'), math.min(vim.fn.col('.'), #vim.fn.getline('.'))
+  for _, syn_id in ipairs(vim.fn.synstack(lnum, col)) do
+    syn_id = vim.fn.synIDtrans(syn_id) -- Resolve :highlight links
+    if vim.fn.synIDattr(syn_id, 'name') == group then
+      return true
+    end
+  end
+  return false
+end
+
+---Check if cursor is in treesitter capture
+---@param capture 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]
+  if not self then
+    return false
+  end
+
+  local node_types = {}
+
+  self.tree:for_each_tree(function(tstree, tree)
+    if not tstree then
+      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)
+
+  return vim.tbl_contains(node_types, capture)
+end
+
+return context