lua/nvim-treesitter.lua


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

local api = vim.api

local install = require'nvim-treesitter.install'
local utils = require'nvim-treesitter.utils'
local info = require'nvim-treesitter.info'
local configs = require'nvim-treesitter.configs'
local state = require'nvim-treesitter.state'

local M = {}

function M.setup()
  utils.setup_commands('install', install.commands)
  utils.setup_commands('info', info.commands)
  utils.setup_commands('configs', configs.commands)

  for _, ft in pairs(configs.available_parsers()) do
    for _, mod in pairs(configs.available_modules()) do
      if configs.is_enabled(mod, ft) then
        local cmd = string.format("lua require'nvim-treesitter.%s'.attach()", mod)
        api.nvim_command(string.format("autocmd NvimTreesitter FileType %s %s", ft, cmd))
      end
    end
    local cmd = string.format("lua require'nvim-treesitter.state'.attach_to_buffer(%s)", ft)
    api.nvim_command(string.format('autocmd NvimTreesitter FileType %s %s', ft, cmd))
  end

  state.run_update()
end

function M.statusline(indicator_size)
  local indicator_size = indicator_size or 100
  local bufnr = api.nvim_get_current_buf()
  local buf_state = state.get_buf_state(bufnr)
  if not buf_state then return "" end

  local current_node = buf_state.current_node
  if not current_node then return "" end

  local expr = current_node:parent()
  local prefix = ""
  if expr then
    prefix = "->"
  end

  local indicator = current_node:type()
  while expr and (#indicator + #(expr:type()) + 5) < indicator_size do
    indicator = expr:type() .. prefix .. indicator
    expr = expr:parent()
  end

  if expr then
    return "..." .. indicator
  else
    return indicator
  end
end

return M