Mirrors/LunarVim
Mirrors/LunarVim
1
0
Fork 0
mirror of https://github.com/LunarVim/LunarVim.git synced 2024-09-20 10:35:08 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refactor(treesitter)!: use auto-install by default (#3677)

Browse source
This commit is contained in:
kylo252 2023-01-10 16:32:41 +01:00 committed by GitHub
parent 38587e2241
commit 5fe3075b34
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 69 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
lua/lvim/core/treesitter.lua
View file

@ -20,7 +20,7 @@ function M.config()
sync_install = false,
-- Automatically install missing parsers when entering buffer
auto_install = false,
auto_install = true,
matchup = {
enable = false, -- mandatory, false will disable the whole extension
@ -95,69 +95,6 @@ function M.config()
}
end
---@class bundledParsersOpts
---@field name_only boolean
---@field filter function
---Retrives a list of bundled parsers paths (any parser not found in default `install_dir`)
---@param opts bundledParsersOpts
---@return string[]
local function get_parsers(opts)
opts = opts or {}
opts.filter = opts.filter or function()
return true
end
local bundled_parsers = vim.tbl_filter(opts.filter, vim.api.nvim_get_runtime_file("parser/*.so", true))
if opts.name_only then
bundled_parsers = vim.tbl_map(function(parser)
return vim.fn.fnamemodify(parser, ":t:r")
end, bundled_parsers)
end
return bundled_parsers
end
---Checks if parser is installed with nvim-treesitter
---@param lang string
---@return boolean
local function is_installed(lang)
local configs = require "nvim-treesitter.configs"
local result = get_parsers {
filter = function(parser)
local install_dir = configs.get_parser_install_dir()
return vim.startswith(parser, install_dir) and (vim.fn.fnamemodify(parser, ":t:r") == lang)
end,
}
local parser_file = result and result[1] or ""
local stat = vim.loop.fs_stat(parser_file)
return stat and stat.type == "file"
end
local function ensure_updated_bundled()
local configs = require "nvim-treesitter.configs"
local bundled_parsers = get_parsers {
name_only = true,
filter = function(parser)
local install_dir = configs.get_parser_install_dir()
return not vim.startswith(parser, install_dir)
end,
}
vim.api.nvim_create_autocmd("VimEnter", {
callback = function()
local missing = vim.tbl_filter(function(parser)
return not is_installed(parser)
end, bundled_parsers)
if #missing > 0 then
vim.cmd { cmd = "TSInstall", args = missing, bang = true }
end
end,
})
end
function M.setup()
-- avoid running in headless mode since it's harder to detect failures
if #vim.api.nvim_list_uis() == 0 then
@ -175,14 +112,9 @@ function M.setup()
treesitter_configs.setup(opts)
ensure_updated_bundled()
if lvim.builtin.treesitter.on_config_done then
lvim.builtin.treesitter.on_config_done(treesitter_configs)
end
end
M.get_parsers = get_parsers
M.is_installed = is_installed
return M