mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2025-09-27 13:38:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(treesitter): add more metadata to language.inspect() (#32657)

Browse Source 
Problem: No way to check the version of a treesitter parser.

Solution: Add version metadata (ABI 15 parsers only) as well as parser state count and supertype information (ABI 15) in `vim.treesitter.language.inspect()`. Also graduate the `abi_version` field, as this is now the official upstream name.

---------

Co-authored-by: Christian Clason <c.clason@uni-graz.at>
This commit is contained in:
Lewis Russell
2025-03-01 15:51:09 +00:00
committed by GitHub
parent 48e6147e64
commit ec8922978e
7 changed files with 104 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
src/nvim/lua/treesitter.c
View File

@@ -263,45 +263,92 @@ int tslua_inspect_lang(lua_State *L)
lua_createtable(L, 0, 2); // [retval]
uint32_t nsymbols = ts_language_symbol_count(lang);
assert(nsymbols < INT_MAX);
{ // Symbols
uint32_t nsymbols = ts_language_symbol_count(lang);
assert(nsymbols < INT_MAX);
lua_createtable(L, (int)(nsymbols - 1), 1); // [retval, symbols]
for (uint32_t i = 0; i < nsymbols; i++) {
TSSymbolType t = ts_language_symbol_type(lang, (TSSymbol)i);
if (t == TSSymbolTypeAuxiliary) {
// not used by the API
continue;
}
const char *name = ts_language_symbol_name(lang, (TSSymbol)i);
bool named = t != TSSymbolTypeAnonymous;
lua_pushboolean(L, named); // [retval, symbols, is_named]
if (!named) {
char buf[256];
snprintf(buf, sizeof(buf), "\"%s\"", name);
lua_setfield(L, -2, buf); // [retval, symbols]
} else {
lua_setfield(L, -2, name); // [retval, symbols]
lua_createtable(L, (int)(nsymbols - 1), 1); // [retval, symbols]
for (uint32_t i = 0; i < nsymbols; i++) {
TSSymbolType t = ts_language_symbol_type(lang, (TSSymbol)i);
if (t == TSSymbolTypeAuxiliary) {
// not used by the API
continue;
}
const char *name = ts_language_symbol_name(lang, (TSSymbol)i);
bool named = t != TSSymbolTypeAnonymous;
lua_pushboolean(L, named); // [retval, symbols, is_named]
if (!named) {
char buf[256];
snprintf(buf, sizeof(buf), "\"%s\"", name);
lua_setfield(L, -2, buf); // [retval, symbols]
} else {
lua_setfield(L, -2, name); // [retval, symbols]
}
}
lua_setfield(L, -2, "symbols"); // [retval]
}
lua_setfield(L, -2, "symbols"); // [retval]
{ // Fields
uint32_t nfields = ts_language_field_count(lang);
lua_createtable(L, (int)nfields, 1); // [retval, fields]
// Field IDs go from 1 to nfields inclusive (extra index 0 maps to NULL)
for (uint32_t i = 1; i <= nfields; i++) {
lua_pushstring(L, ts_language_field_name_for_id(lang, (TSFieldId)i));
lua_rawseti(L, -2, (int)i); // [retval, fields]
}
uint32_t nfields = ts_language_field_count(lang);
lua_createtable(L, (int)nfields, 1); // [retval, fields]
// Field IDs go from 1 to nfields inclusive (extra index 0 maps to NULL)
for (uint32_t i = 1; i <= nfields; i++) {
lua_pushstring(L, ts_language_field_name_for_id(lang, (TSFieldId)i));
lua_rawseti(L, -2, (int)i); // [retval, fields]
lua_setfield(L, -2, "fields"); // [retval]
}
lua_setfield(L, -2, "fields"); // [retval]
lua_pushboolean(L, ts_language_is_wasm(lang));
lua_setfield(L, -2, "_wasm");
lua_pushinteger(L, ts_language_abi_version(lang)); // [retval, version]
lua_setfield(L, -2, "_abi_version");
lua_setfield(L, -2, "abi_version");
{ // Metadata
const TSLanguageMetadata *meta = ts_language_metadata(lang);
if (meta != NULL) {
lua_createtable(L, 0, 3);
lua_pushinteger(L, meta->major_version);
lua_setfield(L, -2, "major_version");
lua_pushinteger(L, meta->minor_version);
lua_setfield(L, -2, "minor_version");
lua_pushinteger(L, meta->patch_version);
lua_setfield(L, -2, "patch_version");
lua_setfield(L, -2, "metadata");
}
}
lua_pushinteger(L, ts_language_state_count(lang));
lua_setfield(L, -2, "state_count");
{ // Supertypes
uint32_t nsupertypes;
const TSSymbol *supertypes = ts_language_supertypes(lang, &nsupertypes);
lua_createtable(L, 0, (int)nsupertypes); // [retval, supertypes]
for (uint32_t i = 0; i < nsupertypes; i++) {
const TSSymbol supertype = *(supertypes + i);
uint32_t nsubtypes;
const TSSymbol *subtypes = ts_language_subtypes(lang, supertype, &nsubtypes);
lua_createtable(L, (int)nsubtypes, 0);
for (uint32_t j = 1; j <= nsubtypes; j++) {
lua_pushstring(L, ts_language_symbol_name(lang, *(subtypes + j)));
lua_rawseti(L, -2, (int)j);
}
lua_setfield(L, -2, ts_language_symbol_name(lang, supertype));
}
lua_setfield(L, -2, "supertypes"); // [retval]
}
return 1;
}