-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