mirror of
https://github.com/nvim-treesitter/nvim-treesitter.git
synced 2026-07-22 13:20:09 -04:00
42 lines
968 B
Lua
Executable file
42 lines
968 B
Lua
Executable file
#!/usr/bin/env -S nvim -l
|
|
|
|
local generate = false
|
|
local update = false
|
|
local max_jobs = nil ---@type integer?
|
|
local parsers = {}
|
|
for i = 1, #_G.arg do
|
|
if _G.arg[i] == '--generate' then
|
|
generate = true
|
|
elseif _G.arg[i] == '--update' then
|
|
update = true
|
|
elseif _G.arg[i]:find('^%-%-max%-jobs') then
|
|
max_jobs = _G.arg[i]:match('=(%d+)')
|
|
else
|
|
parsers[#parsers + 1] = _G.arg[i] ---@type string
|
|
end
|
|
end
|
|
|
|
vim.opt.runtimepath:append('.')
|
|
|
|
-- needed on CI
|
|
vim.fn.mkdir(vim.fn.stdpath('cache'), 'p')
|
|
|
|
local task = nil
|
|
if update then
|
|
--- @type async.Task
|
|
task = require('nvim-treesitter.install').update('all')
|
|
else
|
|
--- @type async.Task
|
|
task = require('nvim-treesitter.install').install(
|
|
#parsers > 0 and parsers or 'all',
|
|
{ force = true, generate = generate, max_jobs = max_jobs }
|
|
)
|
|
end
|
|
|
|
local ok, err_or_ok = task:pwait(6000000)
|
|
if not ok then
|
|
print('ERROR: ', err_or_ok)
|
|
vim.cmd.cq()
|
|
elseif not err_or_ok then
|
|
vim.cmd.cq()
|
|
end
|