aboutsummaryrefslogtreecommitdiffstats
path: root/plugin
diff options
context:
space:
mode:
Diffstat (limited to 'plugin')
-rw-r--r--plugin/query_predicates.lua16
1 files changed, 8 insertions, 8 deletions
diff --git a/plugin/query_predicates.lua b/plugin/query_predicates.lua
index 1a500f591..6511d3104 100644
--- a/plugin/query_predicates.lua
+++ b/plugin/query_predicates.lua
@@ -1,8 +1,8 @@
local query = vim.treesitter.query
local predicates = {
- ---@param match TSQueryMatch
- ---@param pred string[]
+ ---@param match table<integer,TSNode[]>
+ ---@param pred any[]
---@param any boolean
---@return boolean
['kind-eq'] = function(match, pred, any)
@@ -26,16 +26,16 @@ local predicates = {
-- register custom predicates (overwrite existing; needed for CI)
----@param match TSQueryMatch
----@param pred string[]
----@return boolean|nil
+---@param match table<integer,TSNode[]>
+---@param pred any[]
+---@return boolean
query.add_predicate('kind-eq?', function(match, _, _, pred)
return predicates['kind-eq'](match, pred, false)
end, { force = true })
----@param match TSQueryMatch
----@param pred string[]
----@return boolean|nil
+---@param match table<integer,TSNode[]>
+---@param pred any[]
+---@return boolean
query.add_predicate('any-kind-eq?', function(match, _, _, pred)
return predicates['kind-eq'](match, pred, true)
end, { force = true })