Colby MchenryandGitHub
4329a52bec
Adds Lua (.lua) and Luau (.luau) extraction — functions, methods with receivers, type aliases (Luau), require imports (incl. Roblox instance-path), and call edges. Vendors the ABI-15 Lua and ABI-14 Luau tree-sitter grammars. Addresses #232.
37 lines
1.7 KiB
TypeScript
37 lines
1.7 KiB
TypeScript
import { getNodeText, getChildByField } from '../tree-sitter-helpers';
|
|
import type { LanguageExtractor } from '../tree-sitter-types';
|
|
import { luaExtractor } from './lua';
|
|
|
|
// Luau (https://luau.org) is a gradually-typed superset of Lua. The
|
|
// tree-sitter-luau grammar reuses the same node names as the vendored Lua
|
|
// grammar (function_declaration, variable_declaration, function_call,
|
|
// dot/method_index_expression, …), so the Luau extractor extends the Lua one
|
|
// and adds the type-system pieces Luau introduces:
|
|
// - `type X = ...` / `export type X = ...` → type_definition (type_alias)
|
|
// - typed parameters and return types → richer signatures
|
|
//
|
|
// require detection, receiver-splitting (t.f / t:m → methods), and local
|
|
// variable extraction are inherited unchanged from luaExtractor. The shared
|
|
// `extractVariable` core branch is gated on `lua` || `luau`.
|
|
export const luauExtractor: LanguageExtractor = {
|
|
...luaExtractor,
|
|
|
|
// `type X = ...` and `export type X = ...`
|
|
typeAliasTypes: ['type_definition'],
|
|
|
|
// Only Luau `export type` is exported; the keyword leads the node.
|
|
isExported: (node, source) => source.slice(node.startIndex, node.startIndex + 7) === 'export ',
|
|
|
|
// Params + Luau return type (the named child after `parameters`, before the body).
|
|
getSignature: (node, source) => {
|
|
const params = getChildByField(node, 'parameters');
|
|
if (!params) return undefined;
|
|
let sig = getNodeText(params, source);
|
|
const kids = node.namedChildren;
|
|
const idx = kids.findIndex((c) => c.startIndex === params.startIndex);
|
|
const ret = idx >= 0 ? kids[idx + 1] : null;
|
|
if (ret && ret.type !== 'block') sig += `: ${getNodeText(ret, source)}`;
|
|
return sig;
|
|
},
|
|
};
|