benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-02 03:26:52 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: use vim.uv

Browse source
This commit is contained in:
Christian Clason 2023-06-03 12:08:14 +02:00
parent ea192c36ad
commit 161310566a
9 changed files with 44 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

80
scripts/check-queries.lua
View file

@ -48,6 +48,7 @@ local function do_check()
local timings = {}
local parsers = require('nvim-treesitter.config').installed_parsers()
local query_types = require('nvim-treesitter.health').bundled_queries
local configs = require('nvim-treesitter.parsers').configs
local captures = extract_captures()
local last_error
@ -55,34 +56,36 @@ local function do_check()
io_print('::group::Check parsers')
for _, lang in pairs(parsers) do
timings[lang] = {}
for _, query_type in pairs(query_types) do
local before = vim.loop.hrtime()
local ok, query = pcall(vim.treesitter.query.get, lang, query_type)
local after = vim.loop.hrtime()
local duration = after - before
table.insert(timings, { duration = duration, lang = lang, query_type = query_type })
io_print(
'Checking ' .. lang .. ' ' .. query_type .. string.format(' (%.02fms)', duration * 1e-6)
)
if not ok then
local err_msg = lang .. ' (' .. query_type .. '): ' .. query
io_print(err_msg)
last_error = err_msg
else
if query then
for _, capture in ipairs(query.captures) do
local is_valid = (
vim.startswith(capture, '_') -- Helpers.
or list_any(captures[query_type], function(documented_capture)
return vim.startswith(capture, documented_capture)
end)
)
if not is_valid then
local error =
string.format('(x) Invalid capture @%s in %s for %s.', capture, query_type, lang)
io_print(error)
last_error = error
if configs[lang].install_info then
timings[lang] = {}
for _, query_type in pairs(query_types) do
local before = vim.uv.hrtime()
local ok, query = pcall(vim.treesitter.query.get, lang, query_type)
local after = vim.uv.hrtime()
local duration = after - before
table.insert(timings, { duration = duration, lang = lang, query_type = query_type })
io_print(
'Checking ' .. lang .. ' ' .. query_type .. string.format(' (%.02fms)', duration * 1e-6)
)
if not ok then
local err_msg = lang .. ' (' .. query_type .. '): ' .. query
io_print(err_msg)
last_error = err_msg
else
if query then
for _, capture in ipairs(query.captures) do
local is_valid = (
vim.startswith(capture, '_') -- Helpers.
or list_any(captures[query_type], function(documented_capture)
return vim.startswith(capture, documented_capture)
end)
)
if not is_valid then
local error =
string.format('(x) Invalid capture @%s in %s for %s.', capture, query_type, lang)
io_print(error)
last_error = error
end
end
end
end
@ -101,27 +104,6 @@ local function do_check()
end
local ok, result = pcall(do_check)
local allowed_to_fail = vim.split(vim.env.ALLOWED_INSTALLATION_FAILURES or '', ',', true)
for k, v in pairs(require('nvim-treesitter.parsers').configs) do
if v.install_info then
-- skip "query only" languages
if #vim.api.nvim_get_runtime_file('parser/' .. k .. '.*', false) == 0 then
-- On CI all parsers that can be installed from C files should be installed
if
vim.env.CI
and not v.install_info.requires_generate_from_grammar
and not vim.list_contains(allowed_to_fail, k)
then
io_print('Error: parser for ' .. k .. ' is not installed')
vim.cmd('cq')
else
io_print('Warning: parser for ' .. k .. ' is not installed')
end
end
end
end
if ok then
io_print('::group::Timings')
table.sort(result, function(a, b)