feat(nvim): expand Tree-sitter parser baseline

[dotfiles.git] / .config / nvim / lua / plugin-config / nvim-treesitter.lua

diff --git a/.config/nvim/lua/plugin-config/nvim-treesitter.lua b/.config/nvim/lua/plugin-config/nvim-treesitter.lua
index ab75996..4d17d51 100644 (file)
--- a/.config/nvim/lua/plugin-config/nvim-treesitter.lua
+++ b/.config/nvim/lua/plugin-config/nvim-treesitter.lua
@@ -1,22 +1,40 @@
-require'nvim-treesitter'.setup {
-  -- A list of parser names, or "all"
-  ensure_installed = { "c", "lua", "vim", "vimdoc", "query" },
+local treesitter = require('nvim-treesitter')

 
 

-  -- Automatically install missing parsers when entering buffer
-  -- Recommendation: set to false if you don't have `tree-sitter` CLI installed locally
-  auto_install = true,
-
-  highlight = {
-    enable = true,
+treesitter.setup {
+  install_dir = vim.fn.stdpath('data') .. '/site',
+}

 
 

-    -- to disable slow treesitter highlight for large files
-    disable = function(lang, buf)
-        local max_filesize = 100 * 1024 -- 100 KB
-        local ok, stats = pcall(vim.uv.fs_stat, vim.api.nvim_buf_get_name(buf))
-        if ok and stats and stats.size > max_filesize then
-            return true
-        end
-    end,
-    additional_vim_regex_highlighting = false,
-  },
+treesitter.install {
+  'bash',
+  'c',
+  'cmake',
+  'cpp',
+  'json',
+  'lua',
+  'markdown',
+  'markdown_inline',
+  'python',
+  'query',
+  'toml',
+  'vim',
+  'vimdoc',
+  'yaml',

 }
 }

+
+local function enable(buf)
+  if not pcall(vim.treesitter.start, buf) then
+    return
+  end
+
+  vim.wo[0][0].foldexpr = 'v:lua.vim.treesitter.foldexpr()'
+  vim.wo[0][0].foldmethod = 'expr'
+end
+
+vim.api.nvim_create_autocmd('FileType', {
+  callback = function(args) enable(args.buf) end,
+})
+
+-- Startup filetype detection can precede this config being sourced.
+if vim.bo.filetype ~= '' then
+  enable(0)
+end