X-Git-Url: https://git.xinqibao.xyz/dotfiles.git/blobdiff_plain/37c72c84b198107618b6642e73a3b2574d5dcd07..d3e8cc5876a8246c8fe356311e6b2b9bd22363ad:/.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 e61710a..bbe2824 100644
--- a/.config/nvim/lua/plugin-config/nvim-treesitter.lua
+++ b/.config/nvim/lua/plugin-config/nvim-treesitter.lua
@@ -1,65 +1,58 @@
-require'nvim-treesitter.configs'.setup {
-  --[[
-  ]]
-  -- A list of parser names, or "all" (the five listed parsers should always be installed)
-  ensure_installed = {
-        "c", "cpp", "python", "lua", "bash",
-        "cmake", "make",
-        "json", "yaml",
-        "vim", "vimdoc",
-        "query", "regex", "comment", "markdown",
-  },
+local treesitter = require('nvim-treesitter')
 
-  -- 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 = false,
-
-  -- 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>',
-    }
-  },
+treesitter.setup {
+  install_dir = vim.fn.stdpath('data') .. '/site',
+}
 
-  -- Indentation based on treesitter for the = operator. NOTE: This is an experimental feature.
-  indent = {
-    enable = false
-  }
+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