mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2026-04-25 08:44:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

treesitter: add parser on_lines callbacks

Browse Source
This commit is contained in:
Thomas Vigouroux
2020-07-08 22:47:57 +02:00
parent 0f7eaa3555
commit 341e139992
2 changed files with 33 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
runtime/lua/vim/tshighlighter.lua
View File

@@ -25,7 +25,15 @@ TSHighlighter.hl_map = {
function TSHighlighter.new(query, bufnr, ft)
local self = setmetatable({}, TSHighlighter)
self.parser = vim.treesitter.get_parser(bufnr, ft, function(...) self:on_change(...) end)
self.parser = vim.treesitter.get_parser(
bufnr,
ft,
{
on_changedtree = function(...) self:on_changedtree(...) end,
on_lines = function() self.root = self.parser:parse():root() end
}
)
self.buf = self.parser.bufnr
local tree = self.parser:parse()
@@ -35,9 +43,6 @@ function TSHighlighter.new(query, bufnr, ft)
self.redraw_count = 0
self.line_count = {}
a.nvim_buf_set_option(self.buf, "syntax", "")
a.nvim_buf_attach(self.buf, false, {
on_lines=function(_) self.root = self.parser:parse():root() end
})
-- Tricky: if syntax hasn't been enabled, we need to reload color scheme
-- but use synload.vim rather than syntax.vim to not enable
@@ -82,10 +87,10 @@ function TSHighlighter:set_query(query)
end
})
self:on_change({{self.root:range()}})
self:on_changedtree({{self.root:range()}})
end
function TSHighlighter:on_change(changes)
function TSHighlighter:on_changedtree(changes)
-- Get a fresh root
self.root = self.parser.tree:root()