diff options
| author | Stephan Seitz <stephan.seitz@fau.de> | 2022-01-21 21:40:31 +0100 |
|---|---|---|
| committer | Stephan Seitz <stephan.seitz@fau.de> | 2022-02-05 18:54:55 +0100 |
| commit | 7654bbc04bf6b51e583d5241dab62f48ee564b93 (patch) | |
| tree | aea3fc99a952b3dd4445c737db2c6510ca5469df /lua | |
| parent | indents(c): fix ternary (diff) | |
| download | nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar.gz nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar.bz2 nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar.lz nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar.xz nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.tar.zst nvim-treesitter-7654bbc04bf6b51e583d5241dab62f48ee564b93.zip | |
feat(indents): add support for language injection
Diffstat (limited to 'lua')
| -rw-r--r-- | lua/nvim-treesitter/indent.lua | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/lua/nvim-treesitter/indent.lua b/lua/nvim-treesitter/indent.lua index eca649b23..d63f43207 100644 --- a/lua/nvim-treesitter/indent.lua +++ b/lua/nvim-treesitter/indent.lua @@ -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, { |