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

feat: provide a statusline indicator

Browse source 
It will show the current branch at the cursor going the tree as such.
root->node->subnode->leaf

If an argument is provided to `statusline`, then the tree will be
truncated as follows :
..->subnode->subnode
This commit is contained in:
Thomas Vigouroux 2020-05-05 15:04:01 +02:00
parent 5cc7407c7f
commit 7682a1a49f
3 changed files with 36 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

31
lua/nvim-treesitter.lua
View file

@ -24,4 +24,35 @@ function M.setup(lang)
end
end
function M.statusline(indicator_size)
local indicator_size = indicator_size or 1000
local expr = require"nvim-treesitter.utils".expression_at_point()
local current_node =
require'nvim-treesitter.node_movement'.current_node[api.nvim_get_current_buf()]
local indicator = ""
while expr and (#indicator + #(expr:type()) + 5) < indicator_size do
local prefix = ""
if expr:parent() then
prefix = "->"
end
if expr == current_node then
indicator = string.format("%s[%s]%s", prefix, expr:type(), indicator)
else
indicator = prefix .. expr:type() .. indicator
end
expr = expr:parent()
end
if expr then
return "..." .. indicator
else
return indicator
end
end
return M