-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