1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
-- Execute as `nvim --headless -c "luafile ./scripts/check-queries.lua"`
local function extract_captures()
local lines = vim.fn.readfile("CONTRIBUTING.md")
local captures = {}
local current_query
for _, line in ipairs(lines) do
if vim.startswith(line, "### ") then
current_query = vim.fn.tolower(line:sub(5))
elseif vim.startswith(line, "@") and current_query then
if not captures[current_query] then
captures[current_query] = {}
end
table.insert(captures[current_query], vim.split(line:sub(2), " ", true)[1])
end
end
return captures
end
local function do_check()
local parsers = require 'nvim-treesitter.parsers'.available_parsers()
local queries = require 'nvim-treesitter.query'
local query_types = queries.built_in_query_groups
local captures = extract_captures()
local last_error
for _, lang in pairs(parsers) do
if parsers.has_parser(lang) then
for _, query_type in pairs(query_types) do
print('Checking '..lang..' '..query_type)
local ok, query = pcall(queries.get_query,lang, query_type)
if not ok then
vim.api.nvim_err_writeln(query)
last_error = query
else
if query then
for _, capture in ipairs(query.captures) do
if not vim.startswith(capture, "_") -- We ignore things like _helper
and captures[query_type]
and not capture:find("^[A-Z]") -- Highlight groups
and not vim.tbl_contains(captures[query_type], capture) then
error(string.format("Invalid capture @%s in %s for %s.", capture, query_type, lang))
end
end
end
end
end
else
print("No parser for "..lang.." installed! Skipping...")
end
end
if last_error then
print()
print("Last error: ")
error(last_error)
end
end
local ok, err = pcall(do_check)
if ok then
print('Check successful!\n')
vim.cmd('q')
else
print('Check failed:')
print(err)
print('\n')
vim.cmd('cq')
end
|
