-require'nvim-treesitter.configs'.setup {
- --[[
- ]]
- -- A list of parser names, or "all" (the five listed parsers should always be installed)
- ensure_installed = { "c", "lua", "vim", "vimdoc", "query" },
-
- -- Install parsers synchronously (only applied to `ensure_installed`)
- sync_install = false,
-
- -- Automatically install missing parsers when entering buffer
- -- Recommendation: set to false if you don't have `tree-sitter` CLI installed locally
- auto_install = true,
-
- -- List of parsers to ignore installing (or "all")
- ignore_install = { "javascript" },
-
- ---- If you need to change the installation directory of the parsers (see -> Advanced Setup)
- -- parser_install_dir = "/some/path/to/store/parsers", -- Remember to run vim.opt.runtimepath:append("/some/path/to/store/parsers")!
-
- highlight = {
- enable = true,
-
- -- NOTE: these are the names of the parsers and not the filetype. (for example if you want to
- -- disable highlighting for the `tex` filetype, you need to include `latex` in this list as this is
- -- the name of the parser)
- -- list of language that will be disabled
- disable = { "c", "rust" },
- -- Or use a function for more flexibility, e.g. to disable slow treesitter highlight for large files
- disable = function(lang, buf)
- local max_filesize = 100 * 1024 -- 100 KB
- local ok, stats = pcall(vim.loop.fs_stat, vim.api.nvim_buf_get_name(buf))
- if ok and stats and stats.size > max_filesize then
- return true
- end
- end,
-
- -- Setting this to true will run `:h syntax` and tree-sitter at the same time.
- -- Set this to `true` if you depend on 'syntax' being enabled (like for indentation).
- -- Using this option may slow down your editor, and you may see some duplicate highlights.
- -- Instead of true it can also be a list of languages
- additional_vim_regex_highlighting = false,
- },
-
- -- Incremental selection based on the named nodes from the grammar.
- incremental_selection = {
- enable = true,
- keymaps = {
- init_selection = '<CR>',
- node_incremental = '<CR>',
- node_decremental = '<BS>',
- scope_incremental = '<TAB>',
- }
- },
-
- -- Indentation based on treesitter for the = operator. NOTE: This is an experimental feature.
- indent = {
- enable = false
- }
+local treesitter = require('nvim-treesitter')
+
+treesitter.setup {
+ install_dir = vim.fn.stdpath('data') .. '/site',
+}
+
+treesitter.install {
+ 'bash',
+ 'c',
+ 'cmake',
+ 'cpp',
+ 'json',
+ 'lua',
+ 'markdown',
+ 'markdown_inline',
+ 'python',
+ 'query',
+ 'toml',
+ 'vim',
+ 'vimdoc',
+ 'yaml',
}
+
+local function foldlevel(lnum, exclude_start)
+ if lnum < 1 or lnum > vim.api.nvim_buf_line_count(0) then
+ return 0
+ end
+ local value = vim.treesitter.foldexpr(lnum)
+ local level = tonumber(value:match('%d+')) or 0
+ -- A '>N' starts a new fold even when adjacent folds have the same level.
+ return math.max(0, level - (exclude_start and value:sub(1, 1) == '>' and 1 or 0))
+end
+
+function _G.DotfilesFoldExpr()
+ local line = vim.v.lnum
+ -- Keep only levels spanning the previous, current, and next lines.
+ -- Default foldminlines keeps interiors of fewer than two lines visible.
+ return math.min(foldlevel(line - 1), foldlevel(line, true), foldlevel(line + 1, true))
+end
+
+local function enable(buf)
+ vim.wo[0][0].foldexpr = '0'
+ if not pcall(vim.treesitter.start, buf) then
+ return
+ end
+
+ vim.wo[0][0].foldexpr = 'v:lua.DotfilesFoldExpr()'
+ 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