1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

{
    -- 其余部分不变
    branch = "main",
    config = function()
        local nvim_treesitter = require "nvim-treesitter"
        nvim_treesitter.setup()

        local ensure_installed = { "lua", "toml" }
        local pattern = {}
        for _, parser in ipairs(ensure_installed) do
            -- neovim 自己的 api，找不到这个 parser 会报错
            local has_parser, _ = pcall(vim.treesitter.language.inspect, parser)

            if not has_parser then
                -- install 是 nvim-treesitter 的新 api，默认情况下无论是否安装 parser 都会执行，所以这里我们做一个判断
                nvim_treesitter.install(parser)
            else
                -- 新版本需要手动启动高亮，但没有安装相应 parser会导致报错
                pattern = vim.tbl_extend("keep", pattern, vim.treesitter.language.get_filetypes(parser))
            end
        end
        vim.api.nvim_create_autocmd("FileType", {
            pattern = pattern,
            callback = function()
                vim.treesitter.start()
            end,
        })
        -- VeryLazy 晚于 FileType，所以需要手动触发一下
        vim.api.nvim_exec_autocmds("FileType", {})
    end,
}