mirror of
https://github.com/neovim/neovim.git
synced 2025-11-27 04:30:41 +00:00
8714a4009c
Problem:
vim.treesitter does not know how to map a specific filetype to a parser.
This creates problems since in a few places (including in vim.treesitter itself), the filetype is incorrectly used in place of lang.
Solution:
Add an API to enable this:
- Add vim.treesitter.language.add() as a replacement for vim.treesitter.language.require_language().
- Optional arguments are now passed via an opts table.
- Also takes a filetype (or list of filetypes) so we can keep track of what filetypes are associated with which langs.
- Deprecated vim.treesitter.language.require_language().
- Add vim.treesitter.language.get_lang() which returns the associated lang for a given filetype.
- Add vim.treesitter.language.register() to associate filetypes to a lang without loading the parser.
118 lines
3.8 KiB
Lua
118 lines
3.8 KiB
Lua
local helpers = require('test.functional.helpers')(after_each)
|
|
|
|
local clear = helpers.clear
|
|
local eq = helpers.eq
|
|
local command = helpers.command
|
|
local exec_lua = helpers.exec_lua
|
|
local pcall_err = helpers.pcall_err
|
|
local matches = helpers.matches
|
|
local insert = helpers.insert
|
|
|
|
before_each(clear)
|
|
|
|
describe('treesitter language API', function()
|
|
-- error tests not requiring a parser library
|
|
it('handles missing language', function()
|
|
eq(".../language.lua:0: no parser for 'borklang' language, see :help treesitter-parsers",
|
|
pcall_err(exec_lua, "parser = vim.treesitter.get_parser(0, 'borklang')"))
|
|
|
|
-- actual message depends on platform
|
|
matches("Failed to load parser for language 'borklang': uv_dlopen: .+",
|
|
pcall_err(exec_lua, "parser = vim.treesitter.add('borklang', { path = 'borkbork.so' })"))
|
|
|
|
-- Should not throw an error when silent
|
|
eq(false, exec_lua("return vim.treesitter.add('borklang', { silent = true })"))
|
|
|
|
eq(false, exec_lua("return vim.treesitter.add('borklang', { path = 'borkbork.so', silent = true })"))
|
|
|
|
eq(".../language.lua:0: no parser for 'borklang' language, see :help treesitter-parsers",
|
|
pcall_err(exec_lua, "parser = vim.treesitter.inspect_language('borklang')"))
|
|
|
|
matches("Failed to load parser: uv_dlsym: .+",
|
|
pcall_err(exec_lua, 'vim.treesitter.add("c", { symbol_name = "borklang" })'))
|
|
end)
|
|
|
|
it('shows error for invalid language name', function()
|
|
eq(".../language.lua:0: '/foo/' is not a valid language name",
|
|
pcall_err(exec_lua, 'vim.treesitter.add("/foo/", nil, false)'))
|
|
end)
|
|
|
|
it('inspects language', function()
|
|
local keys, fields, symbols = unpack(exec_lua([[
|
|
local lang = vim.treesitter.inspect_language('c')
|
|
local keys, symbols = {}, {}
|
|
for k,_ in pairs(lang) do
|
|
keys[k] = true
|
|
end
|
|
|
|
-- symbols array can have "holes" and is thus not a valid msgpack array
|
|
-- but we don't care about the numbers here (checked in the parser test)
|
|
for _, v in pairs(lang.symbols) do
|
|
table.insert(symbols, v)
|
|
end
|
|
return {keys, lang.fields, symbols}
|
|
]]))
|
|
|
|
eq({fields=true, symbols=true, _abi_version=true}, keys)
|
|
|
|
local fset = {}
|
|
for _,f in pairs(fields) do
|
|
eq("string", type(f))
|
|
fset[f] = true
|
|
end
|
|
eq(true, fset["directive"])
|
|
eq(true, fset["initializer"])
|
|
|
|
local has_named, has_anonymous
|
|
for _,s in pairs(symbols) do
|
|
eq("string", type(s[1]))
|
|
eq("boolean", type(s[2]))
|
|
if s[1] == "for_statement" and s[2] == true then
|
|
has_named = true
|
|
elseif s[1] == "|=" and s[2] == false then
|
|
has_anonymous = true
|
|
end
|
|
end
|
|
eq({true,true}, {has_named,has_anonymous})
|
|
end)
|
|
|
|
it('checks if vim.treesitter.get_parser tries to create a new parser on filetype change', function ()
|
|
command("set filetype=c")
|
|
-- Should not throw an error when filetype is c
|
|
eq('c', exec_lua("return vim.treesitter.get_parser(0):lang()"))
|
|
command("set filetype=borklang")
|
|
-- Should throw an error when filetype changes to borklang
|
|
eq(".../language.lua:0: no parser for 'borklang' language, see :help treesitter-parsers",
|
|
pcall_err(exec_lua, "new_parser = vim.treesitter.get_parser(0, 'borklang')"))
|
|
end)
|
|
|
|
it('retrieve the tree given a range', function ()
|
|
insert([[
|
|
int main() {
|
|
int x = 3;
|
|
}]])
|
|
|
|
exec_lua([[
|
|
langtree = vim.treesitter.get_parser(0, "c")
|
|
tree = langtree:tree_for_range({1, 3, 1, 3})
|
|
]])
|
|
|
|
eq('<node translation_unit>', exec_lua('return tostring(tree:root())'))
|
|
end)
|
|
|
|
it('retrieve the node given a range', function ()
|
|
insert([[
|
|
int main() {
|
|
int x = 3;
|
|
}]])
|
|
|
|
exec_lua([[
|
|
langtree = vim.treesitter.get_parser(0, "c")
|
|
node = langtree:named_node_for_range({1, 3, 1, 3})
|
|
]])
|
|
|
|
eq('<node primitive_type>', exec_lua('return tostring(node)'))
|
|
end)
|
|
end)
|
|
|