mirror of
https://github.com/nvim-treesitter/nvim-treesitter.git
synced 2026-07-01 19:17:02 -04:00
399 lines
11 KiB
Lua
399 lines
11 KiB
Lua
local api = vim.api
|
|
local fn = vim.fn
|
|
local luv = vim.loop
|
|
|
|
local utils = require'nvim-treesitter.utils'
|
|
local parsers = require'nvim-treesitter.parsers'
|
|
local info = require'nvim-treesitter.info'
|
|
local configs = require'nvim-treesitter.configs'
|
|
local shell = require'nvim-treesitter.shell_command_selectors'
|
|
|
|
local M = {}
|
|
local lockfile = {}
|
|
|
|
M.compilers = { vim.fn.getenv('CC'), "cc", "gcc", "clang" }
|
|
|
|
local started_commands = 0
|
|
local finished_commands = 0
|
|
local failed_commands = 0
|
|
|
|
local function reset_progress_counter()
|
|
if started_commands ~= finished_commands then
|
|
return
|
|
end
|
|
started_commands = 0
|
|
finished_commands = 0
|
|
failed_commands = 0
|
|
end
|
|
|
|
local function get_job_status()
|
|
return "[nvim-treesitter] ["..finished_commands.."/"..started_commands
|
|
..(failed_commands > 0 and ", failed: "..failed_commands or "").."]"
|
|
end
|
|
|
|
local function get_revision(lang)
|
|
if #lockfile == 0 then
|
|
lockfile = vim.fn.json_decode(vim.fn.readfile(utils.join_path(utils.get_package_path(), 'lockfile.json')))
|
|
end
|
|
return (lockfile[lang] and lockfile[lang].revision)
|
|
end
|
|
|
|
local function get_installed_revision(lang)
|
|
local lang_file = utils.join_path(utils.get_parser_info_dir(), lang..'.revision')
|
|
if vim.fn.filereadable(lang_file) == 1 then
|
|
return vim.fn.readfile(lang_file)[1]
|
|
end
|
|
end
|
|
|
|
local function needs_update(lang)
|
|
return not get_revision(lang) or get_revision(lang) ~= get_installed_revision(lang)
|
|
end
|
|
|
|
local function outdated_parsers()
|
|
return vim.tbl_filter(function(lang)
|
|
return needs_update(lang)
|
|
end,
|
|
info.installed_parsers())
|
|
end
|
|
|
|
function M.iter_cmd(cmd_list, i, lang, success_message)
|
|
if i == 1 then
|
|
started_commands = started_commands + 1
|
|
end
|
|
if i == #cmd_list + 1 then
|
|
finished_commands = finished_commands + 1
|
|
return print(get_job_status().." "..success_message)
|
|
end
|
|
|
|
local attr = cmd_list[i]
|
|
if attr.info then print(get_job_status().." "..attr.info) end
|
|
|
|
if type(attr.cmd) == 'function' then
|
|
local ok, err = pcall(attr.cmd)
|
|
if ok then
|
|
M.iter_cmd(cmd_list, i + 1, lang, success_message)
|
|
else
|
|
failed_commands = failed_commands + 1
|
|
finished_commands = finished_commands + 1
|
|
return api.nvim_err_writeln((attr.err or ("Failed to execute the following command:\n"..vim.inspect(attr)))
|
|
..'\n'..vim.inspect(err))
|
|
end
|
|
else
|
|
local handle
|
|
handle = luv.spawn(attr.cmd, attr.opts, vim.schedule_wrap(function(code)
|
|
handle:close()
|
|
if code ~= 0 then
|
|
failed_commands = failed_commands + 1
|
|
finished_commands = finished_commands + 1
|
|
return api.nvim_err_writeln(attr.err or ("Failed to execute the following command:\n"..vim.inspect(attr)))
|
|
end
|
|
M.iter_cmd(cmd_list, i + 1, lang, success_message)
|
|
end))
|
|
end
|
|
end
|
|
|
|
local function get_command(cmd)
|
|
local ret = ''
|
|
if cmd.opts and cmd.opts.cwd then
|
|
ret = string.format('cd %s;\n', cmd.opts.cwd)
|
|
end
|
|
|
|
ret = string.format('%s%s ', ret, cmd.cmd)
|
|
|
|
local options = ""
|
|
if cmd.opts and cmd.opts.args then
|
|
for _, opt in ipairs(cmd.opts.args) do
|
|
options = string.format("%s %s", options, opt)
|
|
end
|
|
end
|
|
|
|
return string.format('%s%s', ret, options)
|
|
end
|
|
|
|
local function iter_cmd_sync(cmd_list)
|
|
for _, cmd in ipairs(cmd_list) do
|
|
if cmd.info then
|
|
print(cmd.info)
|
|
end
|
|
|
|
if type(cmd.cmd) == 'function' then
|
|
cmd.cmd()
|
|
else
|
|
local ret = vim.fn.system(get_command(cmd))
|
|
if vim.v.shell_error ~= 0 then
|
|
print(ret)
|
|
api.nvim_err_writeln((cmd.err and cmd.err..'\n' or '')
|
|
.."Failed to execute the following command:\n"
|
|
..vim.inspect(cmd))
|
|
return false
|
|
end
|
|
end
|
|
|
|
end
|
|
|
|
return true
|
|
end
|
|
|
|
local function run_install(cache_folder, install_folder, lang, repo, with_sync, generate_from_grammar)
|
|
parsers.reset_cache()
|
|
|
|
local path_sep = utils.get_path_sep()
|
|
|
|
local project_name = 'tree-sitter-'..lang
|
|
local maybe_local_path = vim.fn.expand(repo.url)
|
|
local from_local_path = vim.fn.isdirectory(maybe_local_path) == 1
|
|
if from_local_path then
|
|
repo.url = maybe_local_path
|
|
end
|
|
|
|
-- compile_location only needed for typescript installs.
|
|
local compile_location
|
|
if from_local_path then
|
|
compile_location = repo.url
|
|
else
|
|
compile_location = cache_folder..path_sep..(repo.location or project_name)
|
|
end
|
|
local parser_lib_name = install_folder..path_sep..lang..".so"
|
|
|
|
if generate_from_grammar and vim.fn.executable('tree-sitter') ~= 1 then
|
|
api.nvim_err_writeln('tree-sitter CLI not found: `tree-sitter` is not executable!')
|
|
return
|
|
end
|
|
local cc = shell.select_executable(M.compilers)
|
|
if not cc then
|
|
api.nvim_err_writeln('No C compiler found! "'
|
|
..table.concat(vim.tbl_filter(function(c) return type(c) == 'string' end, M.compilers), '", "')
|
|
..'" are not executable.')
|
|
return
|
|
end
|
|
local revision = configs.get_update_strategy() == 'lockfile' and get_revision(lang)
|
|
|
|
local command_list = {}
|
|
if not from_local_path then
|
|
vim.list_extend(command_list, { shell.select_install_rm_cmd(cache_folder, project_name) })
|
|
vim.list_extend(command_list, shell.select_download_commands(repo, project_name, cache_folder, revision))
|
|
end
|
|
if generate_from_grammar then
|
|
vim.list_extend(command_list, {
|
|
{
|
|
cmd = 'tree-sitter',
|
|
info = 'Generating source files from grammar.js...',
|
|
err = 'Error during "tree-sitter generate"',
|
|
opts = {
|
|
args = {'generate'},
|
|
cwd = compile_location
|
|
}
|
|
}
|
|
})
|
|
end
|
|
vim.list_extend(command_list, {
|
|
{
|
|
cmd = cc,
|
|
info = 'Compiling...',
|
|
err = 'Error during compilation',
|
|
opts = {
|
|
args = vim.tbl_flatten(shell.select_compiler_args(repo)),
|
|
cwd = compile_location
|
|
}
|
|
},
|
|
shell.select_mv_cmd('parser.so', parser_lib_name, compile_location),
|
|
{
|
|
cmd = function()
|
|
vim.fn.writefile({revision or ''}, utils.join_path(utils.get_parser_info_dir(), lang..'.revision'))
|
|
end
|
|
}
|
|
})
|
|
if not from_local_path then
|
|
vim.list_extend(command_list, {shell.select_install_rm_cmd(cache_folder, project_name)})
|
|
end
|
|
|
|
if with_sync then
|
|
if iter_cmd_sync(command_list) == true then
|
|
print('Treesitter parser for '..lang..' has been installed')
|
|
end
|
|
else
|
|
M.iter_cmd(command_list, 1, lang, 'Treesitter parser for '..lang..' has been installed')
|
|
end
|
|
end
|
|
|
|
local function install_lang(lang, ask_reinstall, cache_folder, install_folder, with_sync, generate_from_grammar)
|
|
if #api.nvim_get_runtime_file('parser/'..lang..'.so', false) > 0 then
|
|
if ask_reinstall ~= 'force' then
|
|
if not ask_reinstall then return end
|
|
|
|
local yesno = fn.input(lang .. ' parser already available: would you like to reinstall ? y/n: ')
|
|
print('\n ') -- mandatory to avoid messing up command line
|
|
if not string.match(yesno, '^y.*') then return end
|
|
end
|
|
end
|
|
|
|
local parser_config = parsers.get_parser_configs()[lang]
|
|
if not parser_config then
|
|
return api.nvim_err_writeln('Parser not available for language '..lang)
|
|
end
|
|
|
|
local install_info = parser_config.install_info
|
|
vim.validate {
|
|
url={ install_info.url, 'string' },
|
|
files={ install_info.files, 'table' }
|
|
}
|
|
|
|
run_install(cache_folder, install_folder, lang, install_info, with_sync, generate_from_grammar)
|
|
end
|
|
|
|
local function install(with_sync, ask_reinstall, generate_from_grammar)
|
|
return function (...)
|
|
if fn.executable('git') == 0 then
|
|
return api.nvim_err_writeln('Git is required on your system to run this command')
|
|
end
|
|
|
|
local cache_folder, err = utils.get_cache_dir()
|
|
if err then return api.nvim_err_writeln(err) end
|
|
|
|
local install_folder, err = utils.get_parser_install_dir()
|
|
if err then return api.nvim_err_writeln(err) end
|
|
|
|
local languages
|
|
local ask
|
|
if ... == 'all' then
|
|
languages = parsers.available_parsers()
|
|
ask = false
|
|
elseif ... == 'maintained' then
|
|
languages = parsers.maintained_parsers()
|
|
ask = false
|
|
else
|
|
languages = vim.tbl_flatten({...})
|
|
ask = ask_reinstall
|
|
end
|
|
|
|
if #languages > 1 then
|
|
reset_progress_counter()
|
|
end
|
|
|
|
for _, lang in ipairs(languages) do
|
|
install_lang(lang, ask, cache_folder, install_folder, with_sync, generate_from_grammar)
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.update(lang)
|
|
M.lockfile = {}
|
|
reset_progress_counter()
|
|
if lang and lang ~= 'all' then
|
|
install(false, 'force')(lang)
|
|
else
|
|
local parsers_to_update = configs.get_update_strategy() == 'lockfile'
|
|
and outdated_parsers()
|
|
or info.installed_parsers()
|
|
if #parsers_to_update == 0 then
|
|
print('All parsers are up-to-date!')
|
|
end
|
|
for _, lang in pairs(parsers_to_update) do
|
|
install(false, 'force')(lang)
|
|
end
|
|
end
|
|
end
|
|
|
|
function M.uninstall(lang)
|
|
local path_sep = '/'
|
|
if fn.has('win32') == 1 then
|
|
path_sep = '\\'
|
|
end
|
|
|
|
if vim.tbl_contains({'all', 'maintained'}, lang) then
|
|
reset_progress_counter()
|
|
local installed = info.installed_parsers()
|
|
if lang == "maintained" then
|
|
local maintained = parsers.maintained_parsers()
|
|
installed = vim.tbl_filter(function(l) return vim.tbl_contains(maintained, l) end, installed)
|
|
end
|
|
for _, lang in pairs(installed) do
|
|
M.uninstall(lang)
|
|
end
|
|
elseif lang then
|
|
local install_dir, err = utils.get_parser_install_dir()
|
|
if err then return api.nvim_err_writeln(err) end
|
|
|
|
local parser_lib = install_dir..path_sep..lang..".so"
|
|
|
|
local command_list = {
|
|
shell.select_rm_file_cmd(parser_lib, "Uninstalling parser for "..lang)
|
|
}
|
|
M.iter_cmd(command_list, 1, lang, 'Treesitter parser for '..lang..' has been uninstalled')
|
|
end
|
|
end
|
|
|
|
function M.write_lockfile(verbose, skip_langs)
|
|
local sorted_parsers = {}
|
|
-- Load previous lockfile
|
|
get_revision()
|
|
skip_langs = skip_langs or {}
|
|
|
|
for k, v in pairs(parsers.get_parser_configs()) do
|
|
table.insert(sorted_parsers, {name = k, parser = v})
|
|
end
|
|
|
|
table.sort(sorted_parsers, function(a, b) return a.name < b.name end)
|
|
|
|
for _, v in ipairs(sorted_parsers) do
|
|
|
|
if not vim.tbl_contains(skip_langs, v.name) then
|
|
-- I'm sure this can be done in aync way with iter_cmd
|
|
local sha = vim.split(vim.fn.systemlist('git ls-remote '..v.parser.install_info.url)[1], '\t')[1]
|
|
lockfile[v.name] = { revision = sha }
|
|
if verbose then
|
|
print(v.name..': '..sha)
|
|
end
|
|
else
|
|
print('Skipping '..v.name)
|
|
end
|
|
end
|
|
|
|
if verbose then
|
|
print(vim.inspect(lockfile))
|
|
end
|
|
vim.fn.writefile(vim.fn.split(vim.fn.json_encode(lockfile), '\n'),
|
|
utils.join_path(utils.get_package_path(), "lockfile.json"))
|
|
end
|
|
|
|
M.ensure_installed = install(false, false)
|
|
|
|
M.commands = {
|
|
TSInstall = {
|
|
run = install(false, true),
|
|
args = {
|
|
"-nargs=+",
|
|
"-complete=custom,nvim_treesitter#installable_parsers",
|
|
},
|
|
},
|
|
TSInstallFromGrammar = {
|
|
run = install(false, true, true),
|
|
args = {
|
|
"-nargs=+",
|
|
"-complete=custom,nvim_treesitter#installable_parsers",
|
|
},
|
|
},
|
|
TSInstallSync = {
|
|
run = install(true, true),
|
|
args = {
|
|
"-nargs=+",
|
|
"-complete=custom,nvim_treesitter#installable_parsers",
|
|
},
|
|
},
|
|
TSUpdate = {
|
|
run = M.update,
|
|
args = {
|
|
"-nargs=*",
|
|
"-complete=custom,nvim_treesitter#installed_parsers",
|
|
},
|
|
},
|
|
TSUninstall = {
|
|
run = M.uninstall,
|
|
args = {
|
|
"-nargs=+",
|
|
"-complete=custom,nvim_treesitter#installed_parsers",
|
|
},
|
|
},
|
|
}
|
|
|
|
return M
|