@optave/codegraph 3.6.0 → 3.8.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +32 -16
- package/dist/ast-analysis/engine.d.ts.map +1 -1
- package/dist/ast-analysis/engine.js +158 -1
- package/dist/ast-analysis/engine.js.map +1 -1
- package/dist/ast-analysis/rules/javascript.d.ts.map +1 -1
- package/dist/ast-analysis/rules/javascript.js +0 -1
- package/dist/ast-analysis/rules/javascript.js.map +1 -1
- package/dist/ast-analysis/visitors/ast-store-visitor.d.ts.map +1 -1
- package/dist/ast-analysis/visitors/ast-store-visitor.js +2 -75
- package/dist/ast-analysis/visitors/ast-store-visitor.js.map +1 -1
- package/dist/cli/commands/ast.js +2 -2
- package/dist/cli/commands/ast.js.map +1 -1
- package/dist/domain/graph/builder/pipeline.d.ts.map +1 -1
- package/dist/domain/graph/builder/pipeline.js +128 -6
- package/dist/domain/graph/builder/pipeline.js.map +1 -1
- package/dist/domain/graph/builder/stages/build-edges.d.ts.map +1 -1
- package/dist/domain/graph/builder/stages/build-edges.js +101 -1
- package/dist/domain/graph/builder/stages/build-edges.js.map +1 -1
- package/dist/domain/graph/builder/stages/collect-files.d.ts.map +1 -1
- package/dist/domain/graph/builder/stages/collect-files.js +17 -5
- package/dist/domain/graph/builder/stages/collect-files.js.map +1 -1
- package/dist/domain/graph/builder/stages/detect-changes.d.ts.map +1 -1
- package/dist/domain/graph/builder/stages/detect-changes.js +98 -50
- package/dist/domain/graph/builder/stages/detect-changes.js.map +1 -1
- package/dist/domain/graph/builder/stages/finalize.d.ts.map +1 -1
- package/dist/domain/graph/builder/stages/finalize.js +32 -5
- package/dist/domain/graph/builder/stages/finalize.js.map +1 -1
- package/dist/domain/graph/builder/stages/insert-nodes.d.ts.map +1 -1
- package/dist/domain/graph/builder/stages/insert-nodes.js +20 -7
- package/dist/domain/graph/builder/stages/insert-nodes.js.map +1 -1
- package/dist/domain/parser.d.ts +1 -1
- package/dist/domain/parser.d.ts.map +1 -1
- package/dist/domain/parser.js +129 -3
- package/dist/domain/parser.js.map +1 -1
- package/dist/extractors/clojure.d.ts +12 -0
- package/dist/extractors/clojure.d.ts.map +1 -0
- package/dist/extractors/clojure.js +245 -0
- package/dist/extractors/clojure.js.map +1 -0
- package/dist/extractors/cuda.d.ts +11 -0
- package/dist/extractors/cuda.d.ts.map +1 -0
- package/dist/extractors/cuda.js +302 -0
- package/dist/extractors/cuda.js.map +1 -0
- package/dist/extractors/dart.d.ts +6 -0
- package/dist/extractors/dart.d.ts.map +1 -0
- package/dist/extractors/dart.js +277 -0
- package/dist/extractors/dart.js.map +1 -0
- package/dist/extractors/elixir.d.ts +9 -0
- package/dist/extractors/elixir.d.ts.map +1 -0
- package/dist/extractors/elixir.js +223 -0
- package/dist/extractors/elixir.js.map +1 -0
- package/dist/extractors/erlang.d.ts +14 -0
- package/dist/extractors/erlang.d.ts.map +1 -0
- package/dist/extractors/erlang.js +239 -0
- package/dist/extractors/erlang.js.map +1 -0
- package/dist/extractors/fsharp.d.ts +13 -0
- package/dist/extractors/fsharp.d.ts.map +1 -0
- package/dist/extractors/fsharp.js +218 -0
- package/dist/extractors/fsharp.js.map +1 -0
- package/dist/extractors/gleam.d.ts +14 -0
- package/dist/extractors/gleam.d.ts.map +1 -0
- package/dist/extractors/gleam.js +229 -0
- package/dist/extractors/gleam.js.map +1 -0
- package/dist/extractors/groovy.d.ts +10 -0
- package/dist/extractors/groovy.d.ts.map +1 -0
- package/dist/extractors/groovy.js +304 -0
- package/dist/extractors/groovy.js.map +1 -0
- package/dist/extractors/haskell.d.ts +8 -0
- package/dist/extractors/haskell.d.ts.map +1 -0
- package/dist/extractors/haskell.js +217 -0
- package/dist/extractors/haskell.js.map +1 -0
- package/dist/extractors/index.d.ts +17 -0
- package/dist/extractors/index.d.ts.map +1 -1
- package/dist/extractors/index.js +17 -0
- package/dist/extractors/index.js.map +1 -1
- package/dist/extractors/julia.d.ts +16 -0
- package/dist/extractors/julia.d.ts.map +1 -0
- package/dist/extractors/julia.js +287 -0
- package/dist/extractors/julia.js.map +1 -0
- package/dist/extractors/lua.d.ts +6 -0
- package/dist/extractors/lua.d.ts.map +1 -0
- package/dist/extractors/lua.js +162 -0
- package/dist/extractors/lua.js.map +1 -0
- package/dist/extractors/objc.d.ts +9 -0
- package/dist/extractors/objc.d.ts.map +1 -0
- package/dist/extractors/objc.js +406 -0
- package/dist/extractors/objc.js.map +1 -0
- package/dist/extractors/ocaml.d.ts +6 -0
- package/dist/extractors/ocaml.d.ts.map +1 -0
- package/dist/extractors/ocaml.js +310 -0
- package/dist/extractors/ocaml.js.map +1 -0
- package/dist/extractors/r.d.ts +13 -0
- package/dist/extractors/r.d.ts.map +1 -0
- package/dist/extractors/r.js +251 -0
- package/dist/extractors/r.js.map +1 -0
- package/dist/extractors/solidity.d.ts +9 -0
- package/dist/extractors/solidity.d.ts.map +1 -0
- package/dist/extractors/solidity.js +374 -0
- package/dist/extractors/solidity.js.map +1 -0
- package/dist/extractors/verilog.d.ts +9 -0
- package/dist/extractors/verilog.d.ts.map +1 -0
- package/dist/extractors/verilog.js +286 -0
- package/dist/extractors/verilog.js.map +1 -0
- package/dist/extractors/zig.d.ts +9 -0
- package/dist/extractors/zig.d.ts.map +1 -0
- package/dist/extractors/zig.js +276 -0
- package/dist/extractors/zig.js.map +1 -0
- package/dist/features/ast.d.ts.map +1 -1
- package/dist/features/ast.js +1 -2
- package/dist/features/ast.js.map +1 -1
- package/dist/features/cfg.d.ts +1 -1
- package/dist/features/cfg.d.ts.map +1 -1
- package/dist/features/cfg.js +6 -51
- package/dist/features/cfg.js.map +1 -1
- package/dist/graph/algorithms/bfs.d.ts +2 -0
- package/dist/graph/algorithms/bfs.d.ts.map +1 -1
- package/dist/graph/algorithms/bfs.js +27 -0
- package/dist/graph/algorithms/bfs.js.map +1 -1
- package/dist/graph/algorithms/centrality.d.ts +2 -0
- package/dist/graph/algorithms/centrality.d.ts.map +1 -1
- package/dist/graph/algorithms/centrality.js +28 -0
- package/dist/graph/algorithms/centrality.js.map +1 -1
- package/dist/graph/algorithms/louvain.d.ts +3 -4
- package/dist/graph/algorithms/louvain.d.ts.map +1 -1
- package/dist/graph/algorithms/louvain.js +29 -0
- package/dist/graph/algorithms/louvain.js.map +1 -1
- package/dist/graph/algorithms/shortest-path.d.ts +2 -0
- package/dist/graph/algorithms/shortest-path.d.ts.map +1 -1
- package/dist/graph/algorithms/shortest-path.js +18 -1
- package/dist/graph/algorithms/shortest-path.js.map +1 -1
- package/dist/types.d.ts +122 -2
- package/dist/types.d.ts.map +1 -1
- package/grammars/tree-sitter-clojure.wasm +0 -0
- package/grammars/tree-sitter-cuda.wasm +0 -0
- package/grammars/tree-sitter-dart.wasm +0 -0
- package/grammars/tree-sitter-elixir.wasm +0 -0
- package/grammars/tree-sitter-erlang.wasm +0 -0
- package/grammars/tree-sitter-fsharp.wasm +0 -0
- package/grammars/tree-sitter-gleam.wasm +0 -0
- package/grammars/tree-sitter-groovy.wasm +0 -0
- package/grammars/tree-sitter-haskell.wasm +0 -0
- package/grammars/tree-sitter-julia.wasm +0 -0
- package/grammars/tree-sitter-lua.wasm +0 -0
- package/grammars/tree-sitter-objc.wasm +0 -0
- package/grammars/tree-sitter-ocaml.wasm +0 -0
- package/grammars/tree-sitter-ocaml_interface.wasm +0 -0
- package/grammars/tree-sitter-r.wasm +0 -0
- package/grammars/tree-sitter-solidity.wasm +0 -0
- package/grammars/tree-sitter-verilog.wasm +0 -0
- package/grammars/tree-sitter-zig.wasm +0 -0
- package/package.json +24 -7
- package/src/ast-analysis/engine.ts +183 -1
- package/src/ast-analysis/rules/javascript.ts +0 -1
- package/src/ast-analysis/visitors/ast-store-visitor.ts +2 -75
- package/src/cli/commands/ast.ts +2 -2
- package/src/domain/graph/builder/pipeline.ts +142 -6
- package/src/domain/graph/builder/stages/build-edges.ts +158 -1
- package/src/domain/graph/builder/stages/collect-files.ts +18 -7
- package/src/domain/graph/builder/stages/detect-changes.ts +109 -55
- package/src/domain/graph/builder/stages/finalize.ts +39 -9
- package/src/domain/graph/builder/stages/insert-nodes.ts +18 -7
- package/src/domain/parser.ts +161 -1
- package/src/extractors/clojure.ts +273 -0
- package/src/extractors/cuda.ts +316 -0
- package/src/extractors/dart.ts +304 -0
- package/src/extractors/elixir.ts +251 -0
- package/src/extractors/erlang.ts +252 -0
- package/src/extractors/fsharp.ts +253 -0
- package/src/extractors/gleam.ts +246 -0
- package/src/extractors/groovy.ts +332 -0
- package/src/extractors/haskell.ts +235 -0
- package/src/extractors/index.ts +17 -0
- package/src/extractors/julia.ts +318 -0
- package/src/extractors/lua.ts +169 -0
- package/src/extractors/objc.ts +431 -0
- package/src/extractors/ocaml.ts +337 -0
- package/src/extractors/r.ts +253 -0
- package/src/extractors/solidity.ts +398 -0
- package/src/extractors/verilog.ts +315 -0
- package/src/extractors/zig.ts +294 -0
- package/src/features/ast.ts +1 -2
- package/src/features/cfg.ts +6 -51
- package/src/graph/algorithms/bfs.ts +34 -0
- package/src/graph/algorithms/centrality.ts +30 -0
- package/src/graph/algorithms/louvain.ts +31 -4
- package/src/graph/algorithms/shortest-path.ts +20 -1
- package/src/types.ts +123 -2
|
@@ -0,0 +1,223 @@
|
|
|
1
|
+
import { findChild, nodeEndLine } from './helpers.js';
|
|
2
|
+
/**
|
|
3
|
+
* Extract symbols from Elixir files.
|
|
4
|
+
*
|
|
5
|
+
* Elixir's tree-sitter grammar represents most constructs as generic `call` nodes.
|
|
6
|
+
* We distinguish modules, functions, imports etc. by the call target's identifier text.
|
|
7
|
+
*/
|
|
8
|
+
export function extractElixirSymbols(tree, _filePath) {
|
|
9
|
+
const ctx = {
|
|
10
|
+
definitions: [],
|
|
11
|
+
calls: [],
|
|
12
|
+
imports: [],
|
|
13
|
+
classes: [],
|
|
14
|
+
exports: [],
|
|
15
|
+
typeMap: new Map(),
|
|
16
|
+
};
|
|
17
|
+
walkElixirNode(tree.rootNode, ctx, null);
|
|
18
|
+
return ctx;
|
|
19
|
+
}
|
|
20
|
+
function walkElixirNode(node, ctx, currentModule) {
|
|
21
|
+
let nextModule = currentModule;
|
|
22
|
+
if (node.type === 'call') {
|
|
23
|
+
const target = node.childForFieldName('target');
|
|
24
|
+
if (target?.type === 'identifier' && target.text === 'defmodule') {
|
|
25
|
+
const args = findChild(node, 'arguments');
|
|
26
|
+
const aliasNode = args && findChild(args, 'alias');
|
|
27
|
+
if (aliasNode)
|
|
28
|
+
nextModule = aliasNode.text;
|
|
29
|
+
}
|
|
30
|
+
handleElixirCall(node, ctx, nextModule);
|
|
31
|
+
}
|
|
32
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
33
|
+
const child = node.child(i);
|
|
34
|
+
if (child)
|
|
35
|
+
walkElixirNode(child, ctx, nextModule);
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
function handleElixirCall(node, ctx, currentModule) {
|
|
39
|
+
const target = node.childForFieldName('target');
|
|
40
|
+
if (!target)
|
|
41
|
+
return;
|
|
42
|
+
if (target.type === 'identifier') {
|
|
43
|
+
const keyword = target.text;
|
|
44
|
+
switch (keyword) {
|
|
45
|
+
case 'defmodule':
|
|
46
|
+
handleDefmodule(node, ctx);
|
|
47
|
+
return;
|
|
48
|
+
case 'def':
|
|
49
|
+
case 'defp':
|
|
50
|
+
handleDefFunction(node, ctx, currentModule, keyword === 'defp' ? 'private' : 'public');
|
|
51
|
+
return;
|
|
52
|
+
case 'defprotocol':
|
|
53
|
+
handleDefprotocol(node, ctx);
|
|
54
|
+
return;
|
|
55
|
+
case 'defimpl':
|
|
56
|
+
handleDefimpl(node, ctx);
|
|
57
|
+
return;
|
|
58
|
+
case 'import':
|
|
59
|
+
case 'use':
|
|
60
|
+
case 'require':
|
|
61
|
+
case 'alias':
|
|
62
|
+
handleElixirImport(node, ctx, keyword);
|
|
63
|
+
return;
|
|
64
|
+
default:
|
|
65
|
+
// Regular function call
|
|
66
|
+
ctx.calls.push({ name: keyword, line: node.startPosition.row + 1 });
|
|
67
|
+
return;
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
if (target.type === 'dot') {
|
|
71
|
+
handleDotCall(node, target, ctx);
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
function handleDefmodule(node, ctx) {
|
|
75
|
+
const args = findChild(node, 'arguments');
|
|
76
|
+
if (!args)
|
|
77
|
+
return;
|
|
78
|
+
const aliasNode = findChild(args, 'alias');
|
|
79
|
+
if (!aliasNode)
|
|
80
|
+
return;
|
|
81
|
+
const name = aliasNode.text;
|
|
82
|
+
const children = [];
|
|
83
|
+
const doBlock = findChild(node, 'do_block');
|
|
84
|
+
if (doBlock) {
|
|
85
|
+
collectModuleMembers(doBlock, ctx, name, children);
|
|
86
|
+
}
|
|
87
|
+
ctx.definitions.push({
|
|
88
|
+
name,
|
|
89
|
+
kind: 'module',
|
|
90
|
+
line: node.startPosition.row + 1,
|
|
91
|
+
endLine: nodeEndLine(node),
|
|
92
|
+
children: children.length > 0 ? children : undefined,
|
|
93
|
+
});
|
|
94
|
+
}
|
|
95
|
+
function collectModuleMembers(doBlock, _ctx, _moduleName, children) {
|
|
96
|
+
for (let i = 0; i < doBlock.childCount; i++) {
|
|
97
|
+
const child = doBlock.child(i);
|
|
98
|
+
if (!child || child.type !== 'call')
|
|
99
|
+
continue;
|
|
100
|
+
const target = child.childForFieldName('target');
|
|
101
|
+
if (!target || target.type !== 'identifier')
|
|
102
|
+
continue;
|
|
103
|
+
if (target.text === 'def' || target.text === 'defp') {
|
|
104
|
+
const fnName = extractFunctionName(child);
|
|
105
|
+
if (fnName) {
|
|
106
|
+
children.push({
|
|
107
|
+
name: fnName,
|
|
108
|
+
kind: 'property',
|
|
109
|
+
line: child.startPosition.row + 1,
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
function handleDefFunction(node, ctx, currentModule, visibility) {
|
|
116
|
+
const fnName = extractFunctionName(node);
|
|
117
|
+
if (!fnName)
|
|
118
|
+
return;
|
|
119
|
+
const fullName = currentModule ? `${currentModule}.${fnName}` : fnName;
|
|
120
|
+
const params = extractElixirParams(node);
|
|
121
|
+
ctx.definitions.push({
|
|
122
|
+
name: fullName,
|
|
123
|
+
kind: 'function',
|
|
124
|
+
line: node.startPosition.row + 1,
|
|
125
|
+
endLine: nodeEndLine(node),
|
|
126
|
+
visibility,
|
|
127
|
+
children: params.length > 0 ? params : undefined,
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
function extractFunctionName(defCallNode) {
|
|
131
|
+
const args = findChild(defCallNode, 'arguments');
|
|
132
|
+
if (!args)
|
|
133
|
+
return null;
|
|
134
|
+
for (let i = 0; i < args.childCount; i++) {
|
|
135
|
+
const child = args.child(i);
|
|
136
|
+
if (!child)
|
|
137
|
+
continue;
|
|
138
|
+
if (child.type === 'call') {
|
|
139
|
+
const target = child.childForFieldName('target');
|
|
140
|
+
if (target?.type === 'identifier')
|
|
141
|
+
return target.text;
|
|
142
|
+
}
|
|
143
|
+
if (child.type === 'identifier')
|
|
144
|
+
return child.text;
|
|
145
|
+
}
|
|
146
|
+
return null;
|
|
147
|
+
}
|
|
148
|
+
function extractElixirParams(defCallNode) {
|
|
149
|
+
const params = [];
|
|
150
|
+
const args = findChild(defCallNode, 'arguments');
|
|
151
|
+
if (!args)
|
|
152
|
+
return params;
|
|
153
|
+
for (let i = 0; i < args.childCount; i++) {
|
|
154
|
+
const child = args.child(i);
|
|
155
|
+
if (!child || child.type !== 'call')
|
|
156
|
+
continue;
|
|
157
|
+
const innerArgs = findChild(child, 'arguments');
|
|
158
|
+
if (!innerArgs)
|
|
159
|
+
continue;
|
|
160
|
+
for (let j = 0; j < innerArgs.childCount; j++) {
|
|
161
|
+
const param = innerArgs.child(j);
|
|
162
|
+
if (!param)
|
|
163
|
+
continue;
|
|
164
|
+
if (param.type === 'identifier') {
|
|
165
|
+
params.push({ name: param.text, kind: 'parameter', line: param.startPosition.row + 1 });
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
return params;
|
|
170
|
+
}
|
|
171
|
+
function handleDefprotocol(node, ctx) {
|
|
172
|
+
const args = findChild(node, 'arguments');
|
|
173
|
+
if (!args)
|
|
174
|
+
return;
|
|
175
|
+
const aliasNode = findChild(args, 'alias');
|
|
176
|
+
if (!aliasNode)
|
|
177
|
+
return;
|
|
178
|
+
ctx.definitions.push({
|
|
179
|
+
name: aliasNode.text,
|
|
180
|
+
kind: 'interface',
|
|
181
|
+
line: node.startPosition.row + 1,
|
|
182
|
+
endLine: nodeEndLine(node),
|
|
183
|
+
});
|
|
184
|
+
}
|
|
185
|
+
function handleDefimpl(node, ctx) {
|
|
186
|
+
const args = findChild(node, 'arguments');
|
|
187
|
+
if (!args)
|
|
188
|
+
return;
|
|
189
|
+
const aliasNode = findChild(args, 'alias');
|
|
190
|
+
if (!aliasNode)
|
|
191
|
+
return;
|
|
192
|
+
ctx.definitions.push({
|
|
193
|
+
name: aliasNode.text,
|
|
194
|
+
kind: 'class',
|
|
195
|
+
line: node.startPosition.row + 1,
|
|
196
|
+
endLine: nodeEndLine(node),
|
|
197
|
+
});
|
|
198
|
+
}
|
|
199
|
+
function handleElixirImport(node, ctx, keyword) {
|
|
200
|
+
const args = findChild(node, 'arguments');
|
|
201
|
+
if (!args)
|
|
202
|
+
return;
|
|
203
|
+
const aliasNode = findChild(args, 'alias');
|
|
204
|
+
if (!aliasNode)
|
|
205
|
+
return;
|
|
206
|
+
ctx.imports.push({
|
|
207
|
+
source: aliasNode.text,
|
|
208
|
+
names: [keyword],
|
|
209
|
+
line: node.startPosition.row + 1,
|
|
210
|
+
});
|
|
211
|
+
}
|
|
212
|
+
function handleDotCall(node, dotNode, ctx) {
|
|
213
|
+
const call = { name: '', line: node.startPosition.row + 1 };
|
|
214
|
+
const right = findChild(dotNode, 'identifier');
|
|
215
|
+
const left = findChild(dotNode, 'alias');
|
|
216
|
+
if (right)
|
|
217
|
+
call.name = right.text;
|
|
218
|
+
if (left)
|
|
219
|
+
call.receiver = left.text;
|
|
220
|
+
if (call.name)
|
|
221
|
+
ctx.calls.push(call);
|
|
222
|
+
}
|
|
223
|
+
//# sourceMappingURL=elixir.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"elixir.js","sourceRoot":"","sources":["../../src/extractors/elixir.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEtD;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAoB,EAAE,SAAiB;IAC1E,MAAM,GAAG,GAAoB;QAC3B,WAAW,EAAE,EAAE;QACf,KAAK,EAAE,EAAE;QACT,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,IAAI,GAAG,EAAE;KACnB,CAAC;IAEF,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;IACzC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,cAAc,CACrB,IAAoB,EACpB,GAAoB,EACpB,aAA4B;IAE5B,IAAI,UAAU,GAAG,aAAa,CAAC;IAE/B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjE,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,IAAI,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACnD,IAAI,SAAS;gBAAE,UAAU,GAAG,SAAS,CAAC,IAAI,CAAC;QAC7C,CAAC;QACD,gBAAgB,CAAC,IAAI,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,KAAK;YAAE,cAAc,CAAC,KAAK,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;IACpD,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CACvB,IAAoB,EACpB,GAAoB,EACpB,aAA4B;IAE5B,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAChD,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;QAC5B,QAAQ,OAAO,EAAE,CAAC;YAChB,KAAK,WAAW;gBACd,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBAC3B,OAAO;YACT,KAAK,KAAK,CAAC;YACX,KAAK,MAAM;gBACT,iBAAiB,CAAC,IAAI,EAAE,GAAG,EAAE,aAAa,EAAE,OAAO,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;gBACvF,OAAO;YACT,KAAK,aAAa;gBAChB,iBAAiB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBAC7B,OAAO;YACT,KAAK,SAAS;gBACZ,aAAa,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBACzB,OAAO;YACT,KAAK,QAAQ,CAAC;YACd,KAAK,KAAK,CAAC;YACX,KAAK,SAAS,CAAC;YACf,KAAK,OAAO;gBACV,kBAAkB,CAAC,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;gBACvC,OAAO;YACT;gBACE,wBAAwB;gBACxB,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpE,OAAO;QACX,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC1B,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;IACnC,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,IAAoB,EAAE,GAAoB;IACjE,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS;QAAE,OAAO;IACvB,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC;IAE5B,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC5C,IAAI,OAAO,EAAE,CAAC;QACZ,oBAAoB,CAAC,OAAO,EAAE,GAAG,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI;QACJ,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACrD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CAC3B,OAAuB,EACvB,IAAqB,EACrB,WAAmB,EACnB,QAA0B;IAE1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,SAAS;QAC9C,MAAM,MAAM,GAAG,KAAK,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,YAAY;YAAE,SAAS;QAEtD,IAAI,MAAM,CAAC,IAAI,KAAK,KAAK,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACX,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;iBAClC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CACxB,IAAoB,EACpB,GAAoB,EACpB,aAA4B,EAC5B,UAAgC;IAEhC,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACzC,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,GAAG,aAAa,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;IACvE,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAEzC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,UAAU;QACV,QAAQ,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KACjD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,WAA2B;IACtD,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;IACjD,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,KAAK,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY;gBAAE,OAAO,MAAM,CAAC,IAAI,CAAC;QACxD,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IACrD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,mBAAmB,CAAC,WAA2B;IACtD,MAAM,MAAM,GAAqB,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;IACjD,IAAI,CAAC,IAAI;QAAE,OAAO,MAAM,CAAC;IAEzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,SAAS;QAC9C,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;QAChD,IAAI,CAAC,SAAS;YAAE,SAAS;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACjC,IAAI,CAAC,KAAK;gBAAE,SAAS;YACrB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;YAC1F,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAoB,EAAE,GAAoB;IACnE,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,IAAoB,EAAE,GAAoB;IAC/D,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAoB,EAAE,GAAoB,EAAE,OAAe;IACrF,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAC1C,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC3C,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;QACf,MAAM,EAAE,SAAS,CAAC,IAAI;QACtB,KAAK,EAAE,CAAC,OAAO,CAAC;QAChB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;KACjC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,IAAoB,EAAE,OAAuB,EAAE,GAAoB;IACxF,MAAM,IAAI,GAAS,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;IAClE,MAAM,KAAK,GAAG,SAAS,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;IAC/C,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAEzC,IAAI,KAAK;QAAE,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IAClC,IAAI,IAAI;QAAE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAEpC,IAAI,IAAI,CAAC,IAAI;QAAE,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACtC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { ExtractorOutput, TreeSitterTree } from '../types.js';
|
|
2
|
+
/**
|
|
3
|
+
* Extract symbols from Erlang files.
|
|
4
|
+
*
|
|
5
|
+
* tree-sitter-erlang (WhatsApp) grammar notes:
|
|
6
|
+
* - module_attribute: -module(name).
|
|
7
|
+
* - record_decl: -record(name, {fields}).
|
|
8
|
+
* - fun_decl: contains function_clause children
|
|
9
|
+
* - function_clause: atom expr_args clause_body
|
|
10
|
+
* - call: function calls, with remote child for module:func
|
|
11
|
+
* - expr_args: parenthesized argument lists
|
|
12
|
+
*/
|
|
13
|
+
export declare function extractErlangSymbols(tree: TreeSitterTree, _filePath: string): ExtractorOutput;
|
|
14
|
+
//# sourceMappingURL=erlang.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"erlang.d.ts","sourceRoot":"","sources":["../../src/extractors/erlang.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAkC,cAAc,EAAE,MAAM,aAAa,CAAC;AAGnG;;;;;;;;;;GAUG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,eAAe,CAY7F"}
|
|
@@ -0,0 +1,239 @@
|
|
|
1
|
+
import { findChild, nodeEndLine } from './helpers.js';
|
|
2
|
+
/**
|
|
3
|
+
* Extract symbols from Erlang files.
|
|
4
|
+
*
|
|
5
|
+
* tree-sitter-erlang (WhatsApp) grammar notes:
|
|
6
|
+
* - module_attribute: -module(name).
|
|
7
|
+
* - record_decl: -record(name, {fields}).
|
|
8
|
+
* - fun_decl: contains function_clause children
|
|
9
|
+
* - function_clause: atom expr_args clause_body
|
|
10
|
+
* - call: function calls, with remote child for module:func
|
|
11
|
+
* - expr_args: parenthesized argument lists
|
|
12
|
+
*/
|
|
13
|
+
export function extractErlangSymbols(tree, _filePath) {
|
|
14
|
+
const ctx = {
|
|
15
|
+
definitions: [],
|
|
16
|
+
calls: [],
|
|
17
|
+
imports: [],
|
|
18
|
+
classes: [],
|
|
19
|
+
exports: [],
|
|
20
|
+
typeMap: new Map(),
|
|
21
|
+
};
|
|
22
|
+
walkErlangNode(tree.rootNode, ctx);
|
|
23
|
+
return ctx;
|
|
24
|
+
}
|
|
25
|
+
function walkErlangNode(node, ctx) {
|
|
26
|
+
switch (node.type) {
|
|
27
|
+
case 'module_attribute':
|
|
28
|
+
handleModuleAttr(node, ctx);
|
|
29
|
+
break;
|
|
30
|
+
case 'record_decl':
|
|
31
|
+
handleRecordDecl(node, ctx);
|
|
32
|
+
break;
|
|
33
|
+
case 'type_alias':
|
|
34
|
+
case 'opaque':
|
|
35
|
+
handleTypeAlias(node, ctx);
|
|
36
|
+
break;
|
|
37
|
+
case 'fun_decl':
|
|
38
|
+
handleFunDecl(node, ctx);
|
|
39
|
+
break;
|
|
40
|
+
case 'function_clause':
|
|
41
|
+
// Only handle if not inside fun_decl (fun_decl handles its own clauses)
|
|
42
|
+
if (node.parent?.type !== 'fun_decl') {
|
|
43
|
+
handleFunctionClause(node, ctx);
|
|
44
|
+
}
|
|
45
|
+
break;
|
|
46
|
+
case 'pp_define':
|
|
47
|
+
handleDefine(node, ctx);
|
|
48
|
+
break;
|
|
49
|
+
case 'pp_include':
|
|
50
|
+
case 'pp_include_lib':
|
|
51
|
+
handleInclude(node, ctx);
|
|
52
|
+
break;
|
|
53
|
+
case 'import_attribute':
|
|
54
|
+
handleImportAttr(node, ctx);
|
|
55
|
+
break;
|
|
56
|
+
case 'call':
|
|
57
|
+
handleCall(node, ctx);
|
|
58
|
+
break;
|
|
59
|
+
}
|
|
60
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
61
|
+
const child = node.child(i);
|
|
62
|
+
if (child)
|
|
63
|
+
walkErlangNode(child, ctx);
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
function handleModuleAttr(node, ctx) {
|
|
67
|
+
// module_attribute: - module ( atom ) .
|
|
68
|
+
const nameNode = findChild(node, 'atom');
|
|
69
|
+
if (!nameNode)
|
|
70
|
+
return;
|
|
71
|
+
ctx.definitions.push({
|
|
72
|
+
name: nameNode.text,
|
|
73
|
+
kind: 'module',
|
|
74
|
+
line: node.startPosition.row + 1,
|
|
75
|
+
endLine: nodeEndLine(node),
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
function handleRecordDecl(node, ctx) {
|
|
79
|
+
// record_decl: - record ( atom , { record_field, ... } ) .
|
|
80
|
+
const nameNode = findChild(node, 'atom');
|
|
81
|
+
if (!nameNode)
|
|
82
|
+
return;
|
|
83
|
+
const children = [];
|
|
84
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
85
|
+
const child = node.child(i);
|
|
86
|
+
if (!child)
|
|
87
|
+
continue;
|
|
88
|
+
if (child.type === 'record_field' || child.type === 'typed_record_field') {
|
|
89
|
+
const fieldName = findChild(child, 'atom');
|
|
90
|
+
if (fieldName) {
|
|
91
|
+
children.push({
|
|
92
|
+
name: fieldName.text,
|
|
93
|
+
kind: 'property',
|
|
94
|
+
line: child.startPosition.row + 1,
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
ctx.definitions.push({
|
|
100
|
+
name: nameNode.text,
|
|
101
|
+
kind: 'record',
|
|
102
|
+
line: node.startPosition.row + 1,
|
|
103
|
+
endLine: nodeEndLine(node),
|
|
104
|
+
children: children.length > 0 ? children : undefined,
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
function handleTypeAlias(node, ctx) {
|
|
108
|
+
const nameNode = findChild(node, 'atom');
|
|
109
|
+
if (!nameNode)
|
|
110
|
+
return;
|
|
111
|
+
ctx.definitions.push({
|
|
112
|
+
name: nameNode.text,
|
|
113
|
+
kind: 'type',
|
|
114
|
+
line: node.startPosition.row + 1,
|
|
115
|
+
endLine: nodeEndLine(node),
|
|
116
|
+
});
|
|
117
|
+
}
|
|
118
|
+
function handleFunDecl(node, ctx) {
|
|
119
|
+
// fun_decl contains one or more function_clause children + dots
|
|
120
|
+
// Extract from the first function_clause
|
|
121
|
+
const clause = findChild(node, 'function_clause');
|
|
122
|
+
if (!clause)
|
|
123
|
+
return;
|
|
124
|
+
handleFunctionClause(clause, ctx);
|
|
125
|
+
}
|
|
126
|
+
function handleFunctionClause(node, ctx) {
|
|
127
|
+
// function_clause: atom expr_args clause_body
|
|
128
|
+
const nameNode = findChild(node, 'atom');
|
|
129
|
+
if (!nameNode)
|
|
130
|
+
return;
|
|
131
|
+
// Don't duplicate if we already have this function
|
|
132
|
+
if (ctx.definitions.some((d) => d.name === nameNode.text && d.kind === 'function'))
|
|
133
|
+
return;
|
|
134
|
+
const params = extractErlangParams(node);
|
|
135
|
+
ctx.definitions.push({
|
|
136
|
+
name: nameNode.text,
|
|
137
|
+
kind: 'function',
|
|
138
|
+
line: node.startPosition.row + 1,
|
|
139
|
+
endLine: nodeEndLine(node.parent?.type === 'fun_decl' ? node.parent : node),
|
|
140
|
+
children: params.length > 0 ? params : undefined,
|
|
141
|
+
visibility: 'public',
|
|
142
|
+
});
|
|
143
|
+
}
|
|
144
|
+
function extractErlangParams(clauseNode) {
|
|
145
|
+
const params = [];
|
|
146
|
+
const argsNode = findChild(clauseNode, 'expr_args');
|
|
147
|
+
if (!argsNode)
|
|
148
|
+
return params;
|
|
149
|
+
for (let i = 0; i < argsNode.childCount; i++) {
|
|
150
|
+
const child = argsNode.child(i);
|
|
151
|
+
if (!child)
|
|
152
|
+
continue;
|
|
153
|
+
if (child.type === 'var') {
|
|
154
|
+
params.push({ name: child.text, kind: 'parameter', line: child.startPosition.row + 1 });
|
|
155
|
+
}
|
|
156
|
+
if (child.type === 'atom') {
|
|
157
|
+
params.push({ name: child.text, kind: 'parameter', line: child.startPosition.row + 1 });
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
return params;
|
|
161
|
+
}
|
|
162
|
+
function handleDefine(node, ctx) {
|
|
163
|
+
// pp_define: -define(NAME, value).
|
|
164
|
+
const nameNode = findChild(node, 'var') || findChild(node, 'atom') || findChild(node, 'macro_lhs');
|
|
165
|
+
if (!nameNode)
|
|
166
|
+
return;
|
|
167
|
+
const name = nameNode.type === 'macro_lhs'
|
|
168
|
+
? (findChild(nameNode, 'var')?.text ?? nameNode.text)
|
|
169
|
+
: nameNode.text;
|
|
170
|
+
ctx.definitions.push({
|
|
171
|
+
name,
|
|
172
|
+
kind: 'variable',
|
|
173
|
+
line: node.startPosition.row + 1,
|
|
174
|
+
endLine: nodeEndLine(node),
|
|
175
|
+
});
|
|
176
|
+
}
|
|
177
|
+
function handleInclude(node, ctx) {
|
|
178
|
+
const strNode = findChild(node, 'string');
|
|
179
|
+
if (!strNode)
|
|
180
|
+
return;
|
|
181
|
+
const source = strNode.text.replace(/^"|"$/g, '');
|
|
182
|
+
ctx.imports.push({
|
|
183
|
+
source,
|
|
184
|
+
names: ['include'],
|
|
185
|
+
line: node.startPosition.row + 1,
|
|
186
|
+
});
|
|
187
|
+
}
|
|
188
|
+
function handleImportAttr(node, ctx) {
|
|
189
|
+
const moduleNode = findChild(node, 'atom');
|
|
190
|
+
if (!moduleNode)
|
|
191
|
+
return;
|
|
192
|
+
const names = [];
|
|
193
|
+
// Find exported function names
|
|
194
|
+
for (let i = 0; i < node.childCount; i++) {
|
|
195
|
+
const child = node.child(i);
|
|
196
|
+
if (!child)
|
|
197
|
+
continue;
|
|
198
|
+
if (child.type === 'fa') {
|
|
199
|
+
const fnName = findChild(child, 'atom');
|
|
200
|
+
if (fnName)
|
|
201
|
+
names.push(fnName.text);
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
ctx.imports.push({
|
|
205
|
+
source: moduleNode.text,
|
|
206
|
+
names: names.length > 0 ? names : [moduleNode.text],
|
|
207
|
+
line: node.startPosition.row + 1,
|
|
208
|
+
});
|
|
209
|
+
}
|
|
210
|
+
function handleCall(node, ctx) {
|
|
211
|
+
// call: first child is function ref (atom or remote), then expr_args
|
|
212
|
+
const funcNode = node.child(0);
|
|
213
|
+
if (!funcNode)
|
|
214
|
+
return;
|
|
215
|
+
if (funcNode.type === 'atom' || funcNode.type === 'identifier') {
|
|
216
|
+
ctx.calls.push({ name: funcNode.text, line: node.startPosition.row + 1 });
|
|
217
|
+
}
|
|
218
|
+
else if (funcNode.type === 'remote') {
|
|
219
|
+
// module:function — remote has atom : atom children
|
|
220
|
+
const atoms = [];
|
|
221
|
+
for (let i = 0; i < funcNode.childCount; i++) {
|
|
222
|
+
const child = funcNode.child(i);
|
|
223
|
+
if (child && (child.type === 'atom' || child.type === 'var')) {
|
|
224
|
+
atoms.push(child.text);
|
|
225
|
+
}
|
|
226
|
+
}
|
|
227
|
+
if (atoms.length >= 2) {
|
|
228
|
+
ctx.calls.push({
|
|
229
|
+
name: atoms[atoms.length - 1],
|
|
230
|
+
receiver: atoms.slice(0, -1).join(':'),
|
|
231
|
+
line: node.startPosition.row + 1,
|
|
232
|
+
});
|
|
233
|
+
}
|
|
234
|
+
else if (atoms.length === 1) {
|
|
235
|
+
ctx.calls.push({ name: atoms[0], line: node.startPosition.row + 1 });
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
//# sourceMappingURL=erlang.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"erlang.js","sourceRoot":"","sources":["../../src/extractors/erlang.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEtD;;;;;;;;;;GAUG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAoB,EAAE,SAAiB;IAC1E,MAAM,GAAG,GAAoB;QAC3B,WAAW,EAAE,EAAE;QACf,KAAK,EAAE,EAAE;QACT,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,IAAI,GAAG,EAAE;KACnB,CAAC;IAEF,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IACnC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,cAAc,CAAC,IAAoB,EAAE,GAAoB;IAChE,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,kBAAkB;YACrB,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,aAAa;YAChB,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,YAAY,CAAC;QAClB,KAAK,QAAQ;YACX,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM;QACR,KAAK,UAAU;YACb,aAAa,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACzB,MAAM;QACR,KAAK,iBAAiB;YACpB,wEAAwE;YACxE,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,oBAAoB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAClC,CAAC;YACD,MAAM;QACR,KAAK,WAAW;YACd,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACxB,MAAM;QACR,KAAK,YAAY,CAAC;QAClB,KAAK,gBAAgB;YACnB,aAAa,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACzB,MAAM;QACR,KAAK,kBAAkB;YACrB,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,MAAM;YACT,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACtB,MAAM;IACV,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,KAAK;YAAE,cAAc,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IACxC,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAoB,EAAE,GAAoB;IAClE,wCAAwC;IACxC,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAoB,EAAE,GAAoB;IAClE,2DAA2D;IAC3D,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;YACzE,MAAM,SAAS,GAAG,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,SAAS,EAAE,CAAC;gBACd,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,SAAS,CAAC,IAAI;oBACpB,IAAI,EAAE,UAAU;oBAChB,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;iBAClC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;KACrD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,IAAoB,EAAE,GAAoB;IACjE,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,IAAoB,EAAE,GAAoB;IAC/D,gEAAgE;IAChE,yCAAyC;IACzC,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,EAAE,iBAAiB,CAAC,CAAC;IAClD,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,oBAAoB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAoB,EAAE,GAAoB;IACtE,8CAA8C;IAC9C,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,mDAAmD;IACnD,IAAI,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;QAAE,OAAO;IAE3F,MAAM,MAAM,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAEzC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;QAC3E,QAAQ,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;QAChD,UAAU,EAAE,QAAQ;KACrB,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,UAA0B;IACrD,MAAM,MAAM,GAAqB,EAAE,CAAC;IACpC,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IACpD,IAAI,CAAC,QAAQ;QAAE,OAAO,MAAM,CAAC;IAE7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,IAAoB,EAAE,GAAoB;IAC9D,mCAAmC;IACnC,MAAM,QAAQ,GACZ,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACpF,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,MAAM,IAAI,GACR,QAAQ,CAAC,IAAI,KAAK,WAAW;QAC3B,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC;QACrD,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;IAEpB,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI;QACJ,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,IAAoB,EAAE,GAAoB;IAC/D,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC1C,IAAI,CAAC,OAAO;QAAE,OAAO;IAErB,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAClD,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;QACf,MAAM;QACN,KAAK,EAAE,CAAC,SAAS,CAAC;QAClB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;KACjC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAoB,EAAE,GAAoB;IAClE,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC3C,IAAI,CAAC,UAAU;QAAE,OAAO;IAExB,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,+BAA+B;IAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK;YAAE,SAAS;QACrB,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACxC,IAAI,MAAM;gBAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;QACf,MAAM,EAAE,UAAU,CAAC,IAAI;QACvB,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;QACnD,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;KACjC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,UAAU,CAAC,IAAoB,EAAE,GAAoB;IAC5D,qEAAqE;IACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC/B,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtB,IAAI,QAAQ,CAAC,IAAI,KAAK,MAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QAC/D,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;IAC5E,CAAC;SAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,oDAAoD;QACpD,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAChC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBAC7D,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACtB,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAE;gBAC9B,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACtC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;aACjC,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAE,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { ExtractorOutput, TreeSitterTree } from '../types.js';
|
|
2
|
+
/**
|
|
3
|
+
* Extract symbols from F# files.
|
|
4
|
+
*
|
|
5
|
+
* tree-sitter-fsharp grammar notes:
|
|
6
|
+
* - named_module: top-level module declaration
|
|
7
|
+
* - function_declaration_left: LHS of `let name params = ...`
|
|
8
|
+
* - import_decl: `open Namespace`
|
|
9
|
+
* - type_definition > union_type_defn / record_type_defn
|
|
10
|
+
* - application_expression: function calls
|
|
11
|
+
*/
|
|
12
|
+
export declare function extractFSharpSymbols(tree: TreeSitterTree, _filePath: string): ExtractorOutput;
|
|
13
|
+
//# sourceMappingURL=fsharp.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fsharp.d.ts","sourceRoot":"","sources":["../../src/extractors/fsharp.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,eAAe,EAGf,cAAc,EACf,MAAM,aAAa,CAAC;AAGrB;;;;;;;;;GASG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,eAAe,CAY7F"}
|