mirror of
https://github.com/chenasraf/nvim-treesitter.git
synced 2026-05-17 17:38:02 +00:00
38 lines
936 B
Lua
Executable File
38 lines
936 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')
|
|
|
|
---@type async.Task
|
|
local task = update and require('nvim-treesitter.install').update('all')
|
|
or require('nvim-treesitter.install').install(
|
|
#parsers > 0 and parsers or 'all',
|
|
{ force = true, generate = generate, max_jobs = max_jobs }
|
|
)
|
|
|
|
local ok, err_or_ok = task:pwait(1800000) -- wait max. 30 minutes
|
|
if not ok then
|
|
print('ERROR: ', err_or_ok)
|
|
vim.cmd.cq()
|
|
elseif not err_or_ok then
|
|
vim.cmd.cq()
|
|
end
|