X-Git-Url: https://git.xinqibao.xyz/dotfiles.git/blobdiff_plain/31c7165b0830fb690d133d7d6bced251775ecf94..1351bd7d4ef69b38593e9967de7f6a12c6cc01f1:/.config/nvim/lua/plugin-config/nvim-treesitter.lua?ds=inline

diff --git a/.config/nvim/lua/plugin-config/nvim-treesitter.lua b/.config/nvim/lua/plugin-config/nvim-treesitter.lua
index d9797b2..4d17d51 100644
--- a/.config/nvim/lua/plugin-config/nvim-treesitter.lua
+++ b/.config/nvim/lua/plugin-config/nvim-treesitter.lua
@@ -1,30 +1,40 @@
-require'nvim-treesitter.configs'.setup {
-  -- 安装 language parser
-  -- :TSInstallInfo 命令查看支持的语言
-  ensure_installed = {"cpp", "vim", "lua"},
-  -- 启用代码高亮功能
-  highlight = {
-    enable = true,
-    additional_vim_regex_highlighting = false
-  },
-  -- 启用增量选择
-  incremental_selection = {
-    enable = true,
-    keymaps = {
-      init_selection = '<CR>',
-      node_incremental = '<CR>',
-      node_decremental = '<BS>',
-      scope_incremental = '<TAB>',
-    }
-  },
-  -- 启用基于Treesitter的代码格式化(=) . NOTE: This is an experimental feature.
-  indent = {
-    enable = false
-  }
+local treesitter = require('nvim-treesitter')
+
+treesitter.setup {
+  install_dir = vim.fn.stdpath('data') .. '/site',
 }
--- 开启 Folding
---vim.wo.foldmethod = 'expr'
---vim.wo.foldexpr = 'nvim_treesitter#foldexpr()'
--- 默认不要折叠
--- https://stackoverflow.com/questions/8316139/how-to-set-the-default-to-unfolded-when-you-open-a-file
---vim.wo.foldlevel = 99
+
+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