benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-02 11:36:54 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(indents): add support for language injection

Browse source
This commit is contained in:
Stephan Seitz 2022-01-21 21:40:31 +01:00
parent 3ff725e654
commit 7654bbc04b
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lua/nvim-treesitter/indent.lua
View file

@ -1,6 +1,7 @@
local parsers = require "nvim-treesitter.parsers"
local queries = require "nvim-treesitter.query"
local tsutils = require "nvim-treesitter.ts_utils"
local highlighter = require "vim.treesitter.highlighter"
local function get_first_node_at_line(root, lnum)
local col = vim.fn.indent(lnum)
@ -38,9 +39,11 @@ local get_indents = tsutils.memoize_by_buf_tick(function(bufnr, root, lang)
aligned_indent = {},
}
for name, node, metadata in queries.iter_captures(bufnr, "indents", root, lang) do
map[name][node:id()] = metadata or {}
end
highlighter.active[bufnr].tree:for_each_tree(function(tstree, tree)
for name, node, metadata in queries.iter_captures(bufnr, "indents", tstree:root(), tree:lang()) do
map[name][node:id()] = metadata or {}
end
end)
return map
end, {