From 7654bbc04bf6b51e583d5241dab62f48ee564b93 Mon Sep 17 00:00:00 2001 From: Stephan Seitz Date: Fri, 21 Jan 2022 21:40:31 +0100 Subject: feat(indents): add support for language injection --- lua/nvim-treesitter/indent.lua | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'lua') 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, { -- cgit v1.2.3-70-g09d2