mirror of
https://github.com/chenasraf/nvim-treesitter.git
synced 2026-05-18 01:39:00 +00:00
142 lines
3.6 KiB
Lua
142 lines
3.6 KiB
Lua
local query = vim.treesitter.query
|
|
|
|
-- register custom predicates
|
|
|
|
---@param match (TSNode|nil)[]
|
|
---@param pred string[]
|
|
---@return boolean|nil
|
|
query.add_predicate('kind-eq?', function(match, _, _, pred)
|
|
local node = match[pred[2]]
|
|
if not node then
|
|
return true
|
|
end
|
|
|
|
local types = { unpack(pred, 3) }
|
|
return vim.list_contains(types, node:type())
|
|
end)
|
|
|
|
-- register custom directives
|
|
|
|
local mimetype_aliases = {
|
|
['importmap'] = 'json',
|
|
['module'] = 'javascript',
|
|
['application/ecmascript'] = 'javascript',
|
|
['text/ecmascript'] = 'javascript',
|
|
}
|
|
|
|
---@param match (TSNode|nil)[]
|
|
---@param _ string
|
|
---@param bufnr integer
|
|
---@param pred string[]
|
|
---@return boolean|nil
|
|
query.add_directive('set-lang-from-mimetype!', function(match, _, bufnr, pred, metadata)
|
|
local capture_id = pred[2]
|
|
local node = match[capture_id]
|
|
if not node then
|
|
return
|
|
end
|
|
local type_attr_value = vim.treesitter.get_node_text(node, bufnr)
|
|
local configured = mimetype_aliases[type_attr_value]
|
|
if configured then
|
|
metadata['injection.language'] = configured
|
|
else
|
|
local parts = vim.split(type_attr_value, '/', {})
|
|
metadata['injection.language'] = parts[#parts]
|
|
end
|
|
end)
|
|
|
|
local injection_aliases = {
|
|
ex = 'elixir',
|
|
pl = 'perl',
|
|
sh = 'bash',
|
|
uxn = 'uxntal',
|
|
ts = 'typescript',
|
|
}
|
|
|
|
---@param match (TSNode|nil)[]
|
|
---@param _ string
|
|
---@param bufnr integer
|
|
---@param pred string[]
|
|
---@return boolean|nil
|
|
query.add_directive('set-lang-from-info-string!', function(match, _, bufnr, pred, metadata)
|
|
local capture_id = pred[2]
|
|
local node = match[capture_id]
|
|
if not node then
|
|
return
|
|
end
|
|
|
|
local injection_alias = vim.treesitter.get_node_text(node, bufnr)
|
|
local filetype = vim.filetype.match({ filename = 'a.' .. injection_alias })
|
|
metadata['injection.language'] = filetype or injection_aliases[injection_alias] or injection_alias
|
|
end)
|
|
|
|
query.add_directive('downcase!', function(match, _, bufnr, pred, metadata)
|
|
local text, key, value ---@type string|string[], string, string|integer
|
|
|
|
if #pred == 3 then
|
|
-- (#downcase! @capture "key")
|
|
key = pred[3]
|
|
value = metadata[pred[2]][key]
|
|
else
|
|
-- (#downcase! "key")
|
|
key = pred[2]
|
|
value = metadata[key]
|
|
end
|
|
|
|
if type(value) == 'string' then
|
|
text = value
|
|
else
|
|
local node = match[value]
|
|
text = vim.treesitter.get_node_text(node, bufnr) or ''
|
|
end
|
|
|
|
if #pred == 3 then
|
|
metadata[pred[2]][key] = string.lower(text)
|
|
else
|
|
metadata[key] = string.lower(text)
|
|
end
|
|
end)
|
|
|
|
-- Trim blank lines from end of the region
|
|
-- Arguments are the captures to trim.
|
|
---@param match (TSNode|nil)[]
|
|
---@param _ string
|
|
---@param bufnr integer
|
|
---@param pred string[]
|
|
---@param metadata table
|
|
---TODO(clason): upstream
|
|
query.add_directive('trim!', function(match, _, bufnr, pred, metadata)
|
|
for _, id in ipairs({ select(2, unpack(pred)) }) do
|
|
local node = match[id]
|
|
if not node then
|
|
return
|
|
end
|
|
|
|
local start_row, start_col, end_row, end_col = node:range()
|
|
|
|
-- Don't trim if region ends in middle of a line
|
|
if end_col ~= 0 then
|
|
return
|
|
end
|
|
|
|
while true do
|
|
-- As we only care when end_col == 0, always inspect one line above end_row.
|
|
local end_line = vim.api.nvim_buf_get_lines(bufnr, end_row - 1, end_row, true)[1]
|
|
|
|
if end_line ~= '' then
|
|
break
|
|
end
|
|
|
|
end_row = end_row - 1
|
|
end
|
|
|
|
-- If this produces an invalid range, we just skip it.
|
|
if start_row < end_row or (start_row == end_row and start_col <= end_col) then
|
|
if not metadata[id] then
|
|
metadata[id] = {}
|
|
end
|
|
metadata[id].range = { start_row, start_col, end_row, end_col }
|
|
end
|
|
end
|
|
end)
|