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

add statusline option to dedupe or not

Browse source
This commit is contained in:
Nat Williams 2023-02-02 13:16:44 -06:00 • committed by Stephan Seitz
parent 34adfe8e95
commit 3fab7abf6f
2 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
doc/nvim-treesitter.txt
View file

@ -446,7 +446,8 @@ Default options (lua syntax):
indicator_size = 100, indicator_size = 100,
type_patterns = {'class', 'function', 'method'}, type_patterns = {'class', 'function', 'method'},
transform_fn = function(line, _node) return line:gsub('%s*[%[%(%{]*%s*$', '') end, transform_fn = function(line, _node) return line:gsub('%s*[%[%(%{]*%s*$', '') end,
separator = ' -> ' separator = ' -> ',
dedupe = true
} }
< <
- `indicator_size` - How long should the string be. If longer, it is cut from - `indicator_size` - How long should the string be. If longer, it is cut from
@ -456,6 +457,7 @@ Default options (lua syntax):
default removes opening brackets and spaces from end. Takes two arguments: default removes opening brackets and spaces from end. Takes two arguments:
the text of the line in question, and the corresponding treesitter node. the text of the line in question, and the corresponding treesitter node.
- `separator` - Separator between nodes. - `separator` - Separator between nodes.
- `dedupe` - Whether or not to remove duplicate components.
*nvim_treesitter#foldexpr()* *nvim_treesitter#foldexpr()*
nvim_treesitter#foldexpr()~ nvim_treesitter#foldexpr()~

10
lua/nvim-treesitter/statusline.lua
View file

@ -21,6 +21,10 @@ function M.statusline(opts)
local type_patterns = options.type_patterns or { "class", "function", "method" } local type_patterns = options.type_patterns or { "class", "function", "method" }
local transform_fn = options.transform_fn or transform_line local transform_fn = options.transform_fn or transform_line
local separator = options.separator or " -> " local separator = options.separator or " -> "
local dedupe = options.dedupe
if dedupe == nil then
dedupe = true
end
local current_node = ts_utils.get_node_at_cursor() local current_node = ts_utils.get_node_at_cursor()
if not current_node then if not current_node then
@ -32,8 +36,10 @@ function M.statusline(opts)
while expr do while expr do
local line = ts_utils._get_line_for_node(expr, type_patterns, transform_fn, bufnr) local line = ts_utils._get_line_for_node(expr, type_patterns, transform_fn, bufnr)
if line ~= "" and not vim.tbl_contains(lines, line) then if line ~= "" then
table.insert(lines, 1, line) if not dedupe or not vim.tbl_contains(lines, line) then
table.insert(lines, 1, line)
end
end end
expr = expr:parent() expr = expr:parent()
end end