diff options
| author | Thomas Vigouroux <tomvig38@gmail.com> | 2020-05-05 15:04:01 +0200 |
|---|---|---|
| committer | Thomas Vigouroux <tomvig38@gmail.com> | 2020-05-05 20:06:45 +0200 |
| commit | 7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f (patch) | |
| tree | e3534fa1b90b4769baff692dfbd1d6239e0b5a1a | |
| parent | Merge pull request #37 from theHamsta/node-movement (diff) | |
| download | nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar.gz nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar.bz2 nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar.lz nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar.xz nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.tar.zst nvim-treesitter-7682a1a49f7c3be58df18dfc18ad3ba65dd3be1f.zip | |
feat: provide a statusline indicator
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
| -rw-r--r-- | autoload/nvim_treesitter.vim | 3 | ||||
| -rw-r--r-- | lua/nvim-treesitter.lua | 31 | ||||
| -rw-r--r-- | lua/nvim-treesitter/parsers.lua | 3 |
3 files changed, 36 insertions, 1 deletions
diff --git a/autoload/nvim_treesitter.vim b/autoload/nvim_treesitter.vim new file mode 100644 index 000000000..715befe0f --- /dev/null +++ b/autoload/nvim_treesitter.vim @@ -0,0 +1,3 @@ +function! nvim_treesitter#statusline(len) + return luaeval("require'nvim-treesitter'.statusline(_A)", a:len) +endfunction diff --git a/lua/nvim-treesitter.lua b/lua/nvim-treesitter.lua index e1c95ed31..90b5468a3 100644 --- a/lua/nvim-treesitter.lua +++ b/lua/nvim-treesitter.lua @@ -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 diff --git a/lua/nvim-treesitter/parsers.lua b/lua/nvim-treesitter/parsers.lua index e046ca45c..07f3e9d34 100644 --- a/lua/nvim-treesitter/parsers.lua +++ b/lua/nvim-treesitter/parsers.lua @@ -5,11 +5,12 @@ local M = {} function M.has_parser(lang) local lang = lang or api.nvim_buf_get_option(0, 'filetype') + if not lang or #lang == 0 then return false end return #api.nvim_get_runtime_file('parser/' .. lang .. '.*', false) > 0 end function M.get_parser(bufnr, lang) - if M.has_parser() then + if M.has_parser(lang) then local buf = bufnr or api.nvim_get_current_buf() local lang = lang or api.nvim_buf_get_option(buf, 'ft') if not M[buf] then |
