mcp-codebase-intelligence 0.1.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.
Files changed (84) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +166 -0
  3. package/dist/graph/code-graph.d.ts +119 -0
  4. package/dist/graph/code-graph.js +279 -0
  5. package/dist/graph/code-graph.js.map +1 -0
  6. package/dist/graph/schema.d.ts +2 -0
  7. package/dist/graph/schema.js +62 -0
  8. package/dist/graph/schema.js.map +1 -0
  9. package/dist/index.d.ts +5 -0
  10. package/dist/index.js +171 -0
  11. package/dist/index.js.map +1 -0
  12. package/dist/indexer/file-watcher.d.ts +13 -0
  13. package/dist/indexer/file-watcher.js +106 -0
  14. package/dist/indexer/file-watcher.js.map +1 -0
  15. package/dist/indexer/lang-go.d.ts +2 -0
  16. package/dist/indexer/lang-go.js +246 -0
  17. package/dist/indexer/lang-go.js.map +1 -0
  18. package/dist/indexer/lang-java.d.ts +2 -0
  19. package/dist/indexer/lang-java.js +307 -0
  20. package/dist/indexer/lang-java.js.map +1 -0
  21. package/dist/indexer/lang-python.d.ts +2 -0
  22. package/dist/indexer/lang-python.js +232 -0
  23. package/dist/indexer/lang-python.js.map +1 -0
  24. package/dist/indexer/lang-rust.d.ts +2 -0
  25. package/dist/indexer/lang-rust.js +323 -0
  26. package/dist/indexer/lang-rust.js.map +1 -0
  27. package/dist/indexer/language-plugin.d.ts +22 -0
  28. package/dist/indexer/language-plugin.js +22 -0
  29. package/dist/indexer/language-plugin.js.map +1 -0
  30. package/dist/indexer/tree-sitter-indexer.d.ts +12 -0
  31. package/dist/indexer/tree-sitter-indexer.js +618 -0
  32. package/dist/indexer/tree-sitter-indexer.js.map +1 -0
  33. package/dist/lsp/lsp-client.d.ts +45 -0
  34. package/dist/lsp/lsp-client.js +315 -0
  35. package/dist/lsp/lsp-client.js.map +1 -0
  36. package/dist/lsp/lsp-manager.d.ts +13 -0
  37. package/dist/lsp/lsp-manager.js +96 -0
  38. package/dist/lsp/lsp-manager.js.map +1 -0
  39. package/dist/tools/analyze-impact.d.ts +38 -0
  40. package/dist/tools/analyze-impact.js +126 -0
  41. package/dist/tools/analyze-impact.js.map +1 -0
  42. package/dist/tools/architecture-diagram.d.ts +33 -0
  43. package/dist/tools/architecture-diagram.js +216 -0
  44. package/dist/tools/architecture-diagram.js.map +1 -0
  45. package/dist/tools/find-implementations.d.ts +33 -0
  46. package/dist/tools/find-implementations.js +68 -0
  47. package/dist/tools/find-implementations.js.map +1 -0
  48. package/dist/tools/find-symbol.d.ts +39 -0
  49. package/dist/tools/find-symbol.js +52 -0
  50. package/dist/tools/find-symbol.js.map +1 -0
  51. package/dist/tools/get-call-graph.d.ts +40 -0
  52. package/dist/tools/get-call-graph.js +174 -0
  53. package/dist/tools/get-call-graph.js.map +1 -0
  54. package/dist/tools/get-dependencies.d.ts +28 -0
  55. package/dist/tools/get-dependencies.js +68 -0
  56. package/dist/tools/get-dependencies.js.map +1 -0
  57. package/dist/tools/get-exports.d.ts +23 -0
  58. package/dist/tools/get-exports.js +40 -0
  59. package/dist/tools/get-exports.js.map +1 -0
  60. package/dist/tools/get-references.d.ts +28 -0
  61. package/dist/tools/get-references.js +53 -0
  62. package/dist/tools/get-references.js.map +1 -0
  63. package/dist/tools/get-stats.d.ts +15 -0
  64. package/dist/tools/get-stats.js +23 -0
  65. package/dist/tools/get-stats.js.map +1 -0
  66. package/dist/tools/get-type-info.d.ts +33 -0
  67. package/dist/tools/get-type-info.js +69 -0
  68. package/dist/tools/get-type-info.js.map +1 -0
  69. package/dist/tools/goto-definition.d.ts +33 -0
  70. package/dist/tools/goto-definition.js +77 -0
  71. package/dist/tools/goto-definition.js.map +1 -0
  72. package/dist/tools/natural-language-query.d.ts +23 -0
  73. package/dist/tools/natural-language-query.js +426 -0
  74. package/dist/tools/natural-language-query.js.map +1 -0
  75. package/dist/tools/reindex.d.ts +15 -0
  76. package/dist/tools/reindex.js +20 -0
  77. package/dist/tools/reindex.js.map +1 -0
  78. package/dist/tools/semantic-diff.d.ts +32 -0
  79. package/dist/tools/semantic-diff.js +407 -0
  80. package/dist/tools/semantic-diff.js.map +1 -0
  81. package/dist/utils/logger.d.ts +6 -0
  82. package/dist/utils/logger.js +15 -0
  83. package/dist/utils/logger.js.map +1 -0
  84. package/package.json +68 -0
@@ -0,0 +1,307 @@
1
+ import Parser from "tree-sitter";
2
+ import Java from "tree-sitter-java";
3
+ import { registerLanguage } from "./language-plugin.js";
4
+ import { bareName } from "./tree-sitter-indexer.js";
5
+ const javaParser = new Parser();
6
+ javaParser.setLanguage(Java);
7
+ function hasModifier(node, modifier) {
8
+ const mods = node.childForFieldName("modifiers") ?? findChild(node, "modifiers");
9
+ if (!mods)
10
+ return false;
11
+ for (let i = 0; i < mods.childCount; i++) {
12
+ if (mods.child(i)?.text === modifier)
13
+ return true;
14
+ }
15
+ return false;
16
+ }
17
+ function isPublicJava(node) {
18
+ return hasModifier(node, "public");
19
+ }
20
+ function findChild(node, type) {
21
+ for (let i = 0; i < node.childCount; i++) {
22
+ if (node.child(i)?.type === type)
23
+ return node.child(i);
24
+ }
25
+ return null;
26
+ }
27
+ function extractJavaMethodSig(node) {
28
+ const name = node.childForFieldName("name")?.text ?? "";
29
+ const params = node.childForFieldName("parameters")?.text ?? "()";
30
+ const retType = node.childForFieldName("type")?.text;
31
+ return retType ? `${retType} ${name}${params}` : `${name}${params}`;
32
+ }
33
+ function walkJava(node, filePath, symbols, references, imports, parentIndex) {
34
+ switch (node.type) {
35
+ case "class_declaration": {
36
+ const nameNode = node.childForFieldName("name");
37
+ if (nameNode) {
38
+ const sym = {
39
+ name: nameNode.text,
40
+ kind: "class",
41
+ filePath,
42
+ lineStart: node.startPosition.row + 1,
43
+ lineEnd: node.endPosition.row + 1,
44
+ colStart: node.startPosition.column,
45
+ colEnd: node.endPosition.column,
46
+ parentSymbolId: parentIndex,
47
+ isExported: isPublicJava(node),
48
+ signature: buildClassSignature(node, nameNode.text),
49
+ };
50
+ const idx = symbols.length;
51
+ symbols.push(sym);
52
+ // Track superclass — superclass node wraps a type_identifier
53
+ const superclass = node.childForFieldName("superclass");
54
+ if (superclass) {
55
+ const superType = superclass.namedChild(0);
56
+ const superName = superType?.text ?? superclass.text;
57
+ references.push({
58
+ fromSymbolId: idx,
59
+ toSymbolName: superName,
60
+ toSymbolBareName: bareName(superName),
61
+ kind: "extends",
62
+ line: superclass.startPosition.row + 1,
63
+ col: superclass.startPosition.column,
64
+ });
65
+ }
66
+ // Track interfaces — super_interfaces > type_list > type_identifier*
67
+ const interfaces = node.childForFieldName("interfaces");
68
+ if (interfaces) {
69
+ const typeList = interfaces.namedChild(0); // type_list node
70
+ const source = typeList ?? interfaces;
71
+ for (let i = 0; i < source.namedChildCount; i++) {
72
+ const iface = source.namedChild(i);
73
+ if (iface) {
74
+ references.push({
75
+ fromSymbolId: idx,
76
+ toSymbolName: iface.text,
77
+ toSymbolBareName: bareName(iface.text),
78
+ kind: "implements",
79
+ line: iface.startPosition.row + 1,
80
+ col: iface.startPosition.column,
81
+ });
82
+ }
83
+ }
84
+ }
85
+ // Walk body
86
+ const body = node.childForFieldName("body");
87
+ if (body) {
88
+ walkJavaChildren(body, filePath, symbols, references, imports, idx);
89
+ }
90
+ return;
91
+ }
92
+ break;
93
+ }
94
+ case "interface_declaration": {
95
+ const nameNode = node.childForFieldName("name");
96
+ if (nameNode) {
97
+ const sym = {
98
+ name: nameNode.text,
99
+ kind: "interface",
100
+ filePath,
101
+ lineStart: node.startPosition.row + 1,
102
+ lineEnd: node.endPosition.row + 1,
103
+ colStart: node.startPosition.column,
104
+ colEnd: node.endPosition.column,
105
+ parentSymbolId: parentIndex,
106
+ isExported: isPublicJava(node),
107
+ signature: `interface ${nameNode.text}`,
108
+ };
109
+ const idx = symbols.length;
110
+ symbols.push(sym);
111
+ const body = node.childForFieldName("body");
112
+ if (body) {
113
+ walkJavaChildren(body, filePath, symbols, references, imports, idx);
114
+ }
115
+ return;
116
+ }
117
+ break;
118
+ }
119
+ case "enum_declaration": {
120
+ const nameNode = node.childForFieldName("name");
121
+ if (nameNode) {
122
+ const sym = {
123
+ name: nameNode.text,
124
+ kind: "enum",
125
+ filePath,
126
+ lineStart: node.startPosition.row + 1,
127
+ lineEnd: node.endPosition.row + 1,
128
+ colStart: node.startPosition.column,
129
+ colEnd: node.endPosition.column,
130
+ parentSymbolId: parentIndex,
131
+ isExported: isPublicJava(node),
132
+ signature: `enum ${nameNode.text}`,
133
+ };
134
+ const idx = symbols.length;
135
+ symbols.push(sym);
136
+ const body = node.childForFieldName("body");
137
+ if (body) {
138
+ walkJavaChildren(body, filePath, symbols, references, imports, idx);
139
+ }
140
+ return;
141
+ }
142
+ break;
143
+ }
144
+ case "method_declaration":
145
+ case "constructor_declaration": {
146
+ const nameNode = node.childForFieldName("name");
147
+ if (nameNode) {
148
+ const sym = {
149
+ name: nameNode.text,
150
+ kind: "method",
151
+ filePath,
152
+ lineStart: node.startPosition.row + 1,
153
+ lineEnd: node.endPosition.row + 1,
154
+ colStart: node.startPosition.column,
155
+ colEnd: node.endPosition.column,
156
+ parentSymbolId: parentIndex,
157
+ isExported: isPublicJava(node),
158
+ signature: extractJavaMethodSig(node),
159
+ };
160
+ const idx = symbols.length;
161
+ symbols.push(sym);
162
+ walkJavaChildren(node, filePath, symbols, references, imports, idx);
163
+ return;
164
+ }
165
+ break;
166
+ }
167
+ case "field_declaration": {
168
+ // int x, y; or private String name;
169
+ for (let i = 0; i < node.namedChildCount; i++) {
170
+ const child = node.namedChild(i);
171
+ if (child?.type === "variable_declarator") {
172
+ const nameNode = child.childForFieldName("name");
173
+ if (nameNode) {
174
+ symbols.push({
175
+ name: nameNode.text,
176
+ kind: "property",
177
+ filePath,
178
+ lineStart: node.startPosition.row + 1,
179
+ lineEnd: node.endPosition.row + 1,
180
+ colStart: node.startPosition.column,
181
+ colEnd: node.endPosition.column,
182
+ parentSymbolId: parentIndex,
183
+ isExported: isPublicJava(node),
184
+ });
185
+ }
186
+ }
187
+ }
188
+ break;
189
+ }
190
+ case "method_invocation": {
191
+ const nameNode = node.childForFieldName("name");
192
+ const object = node.childForFieldName("object");
193
+ if (nameNode && parentIndex !== undefined) {
194
+ const fullName = object ? `${object.text}.${nameNode.text}` : nameNode.text;
195
+ references.push({
196
+ fromSymbolId: parentIndex,
197
+ toSymbolName: fullName,
198
+ toSymbolBareName: bareName(fullName),
199
+ kind: "call",
200
+ line: node.startPosition.row + 1,
201
+ col: node.startPosition.column,
202
+ });
203
+ }
204
+ break;
205
+ }
206
+ case "object_creation_expression": {
207
+ const typeNode = node.childForFieldName("type");
208
+ if (typeNode && parentIndex !== undefined) {
209
+ references.push({
210
+ fromSymbolId: parentIndex,
211
+ toSymbolName: typeNode.text,
212
+ toSymbolBareName: bareName(typeNode.text),
213
+ kind: "instantiation",
214
+ line: node.startPosition.row + 1,
215
+ col: node.startPosition.column,
216
+ });
217
+ }
218
+ break;
219
+ }
220
+ case "import_declaration": {
221
+ // import java.util.HashMap; or import java.util.*;
222
+ const lastChild = node.namedChild(node.namedChildCount - 1);
223
+ if (lastChild) {
224
+ const fullPath = lastChild.text;
225
+ const isWildcard = fullPath.endsWith("*");
226
+ const name = isWildcard ? "*" : bareName(fullPath.replace(/\./g, "."));
227
+ imports.push({
228
+ sourcePath: fullPath,
229
+ importedName: name,
230
+ localName: name,
231
+ isDefault: false,
232
+ isNamespace: isWildcard,
233
+ line: node.startPosition.row + 1,
234
+ });
235
+ }
236
+ return;
237
+ }
238
+ case "package_declaration": {
239
+ // package com.example.foo;
240
+ const nameNode = node.namedChild(node.namedChildCount - 1);
241
+ if (nameNode) {
242
+ symbols.push({
243
+ name: nameNode.text,
244
+ kind: "variable",
245
+ filePath,
246
+ lineStart: node.startPosition.row + 1,
247
+ lineEnd: node.endPosition.row + 1,
248
+ colStart: node.startPosition.column,
249
+ colEnd: node.endPosition.column,
250
+ isExported: true,
251
+ signature: `package ${nameNode.text}`,
252
+ });
253
+ }
254
+ return;
255
+ }
256
+ case "annotation":
257
+ case "marker_annotation": {
258
+ // @Override, @Deprecated, @RequestMapping(...), etc.
259
+ const nameNode = node.childForFieldName("name") ?? node.namedChild(0);
260
+ if (nameNode && parentIndex !== undefined) {
261
+ references.push({
262
+ fromSymbolId: parentIndex,
263
+ toSymbolName: nameNode.text,
264
+ toSymbolBareName: bareName(nameNode.text),
265
+ kind: "call",
266
+ line: node.startPosition.row + 1,
267
+ col: node.startPosition.column,
268
+ });
269
+ }
270
+ break;
271
+ }
272
+ }
273
+ walkJavaChildren(node, filePath, symbols, references, imports, parentIndex);
274
+ }
275
+ function buildClassSignature(node, name) {
276
+ const parts = [];
277
+ if (hasModifier(node, "public"))
278
+ parts.push("public");
279
+ if (hasModifier(node, "abstract"))
280
+ parts.push("abstract");
281
+ if (hasModifier(node, "final"))
282
+ parts.push("final");
283
+ parts.push("class", name);
284
+ const superclass = node.childForFieldName("superclass");
285
+ if (superclass)
286
+ parts.push("extends", superclass.text);
287
+ const interfaces = node.childForFieldName("interfaces");
288
+ if (interfaces)
289
+ parts.push("implements", interfaces.text);
290
+ return parts.join(" ");
291
+ }
292
+ function walkJavaChildren(node, filePath, symbols, references, imports, parentIndex) {
293
+ for (let i = 0; i < node.namedChildCount; i++) {
294
+ const child = node.namedChild(i);
295
+ if (child) {
296
+ walkJava(child, filePath, symbols, references, imports, parentIndex);
297
+ }
298
+ }
299
+ }
300
+ export const javaPlugin = {
301
+ id: "java",
302
+ extensions: [".java"],
303
+ getParser: () => javaParser,
304
+ walk: walkJava,
305
+ };
306
+ registerLanguage(javaPlugin);
307
+ //# sourceMappingURL=lang-java.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"lang-java.js","sourceRoot":"","sources":["../../src/indexer/lang-java.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,IAAI,MAAM,kBAAkB,CAAC;AAEpC,OAAO,EAAuB,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAEpD,MAAM,UAAU,GAAG,IAAI,MAAM,EAAE,CAAC;AAChC,UAAU,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;AAE7B,SAAS,WAAW,CAAC,IAAuB,EAAE,QAAgB;IAC5D,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,IAAI,SAAS,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACjF,IAAI,CAAC,IAAI;QAAE,OAAO,KAAK,CAAC;IACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;IACpD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,YAAY,CAAC,IAAuB;IAC3C,OAAO,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AACrC,CAAC;AAED,SAAS,SAAS,CAAC,IAAuB,EAAE,IAAY;IACtD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAuB;IACnD,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC;IAClE,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC;IACrD,OAAO,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,IAAI,IAAI,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,MAAM,EAAE,CAAC;AACtE,CAAC;AAED,SAAS,QAAQ,CACf,IAAuB,EACvB,QAAgB,EAChB,OAAqB,EACrB,UAA2B,EAC3B,OAAqB,EACrB,WAAoB;IAEpB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,OAAO;oBACb,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,YAAY,CAAC,IAAI,CAAC;oBAC9B,SAAS,EAAE,mBAAmB,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC;iBACpD,CAAC;gBACF,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAElB,6DAA6D;gBAC7D,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;gBACxD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,SAAS,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;oBAC3C,MAAM,SAAS,GAAG,SAAS,EAAE,IAAI,IAAI,UAAU,CAAC,IAAI,CAAC;oBACrD,UAAU,CAAC,IAAI,CAAC;wBACd,YAAY,EAAE,GAAG;wBACjB,YAAY,EAAE,SAAS;wBACvB,gBAAgB,EAAE,QAAQ,CAAC,SAAS,CAAC;wBACrC,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,UAAU,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;wBACtC,GAAG,EAAE,UAAU,CAAC,aAAa,CAAC,MAAM;qBACrC,CAAC,CAAC;gBACL,CAAC;gBAED,qEAAqE;gBACrE,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;gBACxD,IAAI,UAAU,EAAE,CAAC;oBACf,MAAM,QAAQ,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB;oBAC5D,MAAM,MAAM,GAAG,QAAQ,IAAI,UAAU,CAAC;oBACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;wBAChD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;wBACnC,IAAI,KAAK,EAAE,CAAC;4BACV,UAAU,CAAC,IAAI,CAAC;gCACd,YAAY,EAAE,GAAG;gCACjB,YAAY,EAAE,KAAK,CAAC,IAAI;gCACxB,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;gCACtC,IAAI,EAAE,YAAY;gCAClB,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCACjC,GAAG,EAAE,KAAK,CAAC,aAAa,CAAC,MAAM;6BAChC,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,YAAY;gBACZ,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBACtE,CAAC;gBACD,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,uBAAuB,CAAC,CAAC,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,WAAW;oBACjB,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,YAAY,CAAC,IAAI,CAAC;oBAC9B,SAAS,EAAE,aAAa,QAAQ,CAAC,IAAI,EAAE;iBACxC,CAAC;gBACF,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBACtE,CAAC;gBACD,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,MAAM;oBACZ,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,YAAY,CAAC,IAAI,CAAC;oBAC9B,SAAS,EAAE,QAAQ,QAAQ,CAAC,IAAI,EAAE;iBACnC,CAAC;gBACF,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBACtE,CAAC;gBACD,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,oBAAoB,CAAC;QAC1B,KAAK,yBAAyB,CAAC,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,QAAQ;oBACd,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,YAAY,CAAC,IAAI,CAAC;oBAC9B,SAAS,EAAE,oBAAoB,CAAC,IAAI,CAAC;iBACtC,CAAC;gBACF,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBACpE,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,oCAAoC;YACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACjC,IAAI,KAAK,EAAE,IAAI,KAAK,qBAAqB,EAAE,CAAC;oBAC1C,MAAM,QAAQ,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;oBACjD,IAAI,QAAQ,EAAE,CAAC;wBACb,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,QAAQ,CAAC,IAAI;4BACnB,IAAI,EAAE,UAAU;4BAChB,QAAQ;4BACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;4BACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;4BACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;4BACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;4BAC/B,cAAc,EAAE,WAAW;4BAC3B,UAAU,EAAE,YAAY,CAAC,IAAI,CAAC;yBAC/B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAChD,IAAI,QAAQ,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC1C,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAC5E,UAAU,CAAC,IAAI,CAAC;oBACd,YAAY,EAAE,WAAW;oBACzB,YAAY,EAAE,QAAQ;oBACtB,gBAAgB,EAAE,QAAQ,CAAC,QAAQ,CAAC;oBACpC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBAChC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;iBAC/B,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,4BAA4B,CAAC,CAAC,CAAC;YAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC1C,UAAU,CAAC,IAAI,CAAC;oBACd,YAAY,EAAE,WAAW;oBACzB,YAAY,EAAE,QAAQ,CAAC,IAAI;oBAC3B,gBAAgB,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACzC,IAAI,EAAE,eAAe;oBACrB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBAChC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;iBAC/B,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,oBAAoB,CAAC,CAAC,CAAC;YAC1B,mDAAmD;YACnD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC;YAC5D,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC;gBAChC,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;gBAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvE,OAAO,CAAC,IAAI,CAAC;oBACX,UAAU,EAAE,QAAQ;oBACpB,YAAY,EAAE,IAAI;oBAClB,SAAS,EAAE,IAAI;oBACf,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,UAAU;oBACvB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;iBACjC,CAAC,CAAC;YACL,CAAC;YACD,OAAO;QACT,CAAC;QAED,KAAK,qBAAqB,CAAC,CAAC,CAAC;YAC3B,2BAA2B;YAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC;YAC3D,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,UAAU;oBAChB,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE,WAAW,QAAQ,CAAC,IAAI,EAAE;iBACtC,CAAC,CAAC;YACL,CAAC;YACD,OAAO;QACT,CAAC;QAED,KAAK,YAAY,CAAC;QAClB,KAAK,mBAAmB,CAAC,CAAC,CAAC;YACzB,qDAAqD;YACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,QAAQ,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC1C,UAAU,CAAC,IAAI,CAAC;oBACd,YAAY,EAAE,WAAW;oBACzB,YAAY,EAAE,QAAQ,CAAC,IAAI;oBAC3B,gBAAgB,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACzC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBAChC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;iBAC/B,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAuB,EAAE,IAAY;IAChE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC;QAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtD,IAAI,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC;QAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC1D,IAAI,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC;QAAE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAE1B,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACxD,IAAI,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;IAEvD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACxD,IAAI,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;IAE1D,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,gBAAgB,CACvB,IAAuB,EACvB,QAAgB,EAChB,OAAqB,EACrB,UAA2B,EAC3B,OAAqB,EACrB,WAAoB;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,KAAK,EAAE,CAAC;YACV,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAAmB;IACxC,EAAE,EAAE,MAAM;IACV,UAAU,EAAE,CAAC,OAAO,CAAC;IACrB,SAAS,EAAE,GAAG,EAAE,CAAC,UAAU;IAC3B,IAAI,EAAE,QAAQ;CACf,CAAC;AAEF,gBAAgB,CAAC,UAAU,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ import { type LanguagePlugin } from "./language-plugin.js";
2
+ export declare const pythonPlugin: LanguagePlugin;
@@ -0,0 +1,232 @@
1
+ import Parser from "tree-sitter";
2
+ import Python from "tree-sitter-python";
3
+ import { registerLanguage } from "./language-plugin.js";
4
+ import { bareName } from "./tree-sitter-indexer.js";
5
+ const pyParser = new Parser();
6
+ pyParser.setLanguage(Python);
7
+ function extractPySignature(node) {
8
+ const name = node.childForFieldName("name")?.text ?? "";
9
+ const params = node.childForFieldName("parameters");
10
+ const returnType = node.childForFieldName("return_type");
11
+ const paramText = params?.text ?? "()";
12
+ const retText = returnType ? ` -> ${returnType.text}` : "";
13
+ return `${name}${paramText}${retText}`;
14
+ }
15
+ function isDecorated(node, decoratorName) {
16
+ if (node.parent?.type === "decorated_definition") {
17
+ for (let i = 0; i < node.parent.namedChildCount; i++) {
18
+ const child = node.parent.namedChild(i);
19
+ if (child?.type === "decorator" && child.text.includes(decoratorName))
20
+ return true;
21
+ }
22
+ }
23
+ return false;
24
+ }
25
+ function walkPython(node, filePath, symbols, references, imports, parentIndex) {
26
+ switch (node.type) {
27
+ case "function_definition": {
28
+ const nameNode = node.childForFieldName("name");
29
+ if (nameNode) {
30
+ const isMethod = parentIndex !== undefined && symbols[parentIndex]?.kind === "class";
31
+ const isPrivate = nameNode.text.startsWith("_") && !nameNode.text.startsWith("__");
32
+ const isDunder = nameNode.text.startsWith("__") && nameNode.text.endsWith("__");
33
+ const sym = {
34
+ name: nameNode.text,
35
+ kind: isMethod ? "method" : "function",
36
+ filePath,
37
+ lineStart: node.startPosition.row + 1,
38
+ lineEnd: node.endPosition.row + 1,
39
+ colStart: node.startPosition.column,
40
+ colEnd: node.endPosition.column,
41
+ parentSymbolId: parentIndex,
42
+ isExported: !isPrivate || isDunder,
43
+ signature: extractPySignature(node),
44
+ };
45
+ const idx = symbols.length;
46
+ symbols.push(sym);
47
+ walkPyChildren(node, filePath, symbols, references, imports, idx);
48
+ return;
49
+ }
50
+ break;
51
+ }
52
+ case "class_definition": {
53
+ const nameNode = node.childForFieldName("name");
54
+ if (nameNode) {
55
+ const isPrivate = nameNode.text.startsWith("_");
56
+ const sym = {
57
+ name: nameNode.text,
58
+ kind: "class",
59
+ filePath,
60
+ lineStart: node.startPosition.row + 1,
61
+ lineEnd: node.endPosition.row + 1,
62
+ colStart: node.startPosition.column,
63
+ colEnd: node.endPosition.column,
64
+ parentSymbolId: parentIndex,
65
+ isExported: !isPrivate,
66
+ signature: `class ${nameNode.text}`,
67
+ };
68
+ // Track superclasses
69
+ const superclasses = node.childForFieldName("superclasses");
70
+ if (superclasses) {
71
+ for (let i = 0; i < superclasses.namedChildCount; i++) {
72
+ const arg = superclasses.namedChild(i);
73
+ if (arg && arg.type !== "keyword_argument") {
74
+ references.push({
75
+ fromSymbolId: symbols.length,
76
+ toSymbolName: arg.text,
77
+ toSymbolBareName: bareName(arg.text),
78
+ kind: "extends",
79
+ line: arg.startPosition.row + 1,
80
+ col: arg.startPosition.column,
81
+ });
82
+ sym.signature += superclasses.text ? `(${superclasses.text})` : "";
83
+ }
84
+ }
85
+ }
86
+ const idx = symbols.length;
87
+ symbols.push(sym);
88
+ walkPyChildren(node, filePath, symbols, references, imports, idx);
89
+ return;
90
+ }
91
+ break;
92
+ }
93
+ case "assignment": {
94
+ // Module-level assignments: MY_CONST = ... or __all__ = [...]
95
+ const left = node.childForFieldName("left");
96
+ if (left?.type === "identifier" && parentIndex === undefined) {
97
+ const name = left.text;
98
+ // Track __all__ for export info, or top-level constants (UPPER_CASE)
99
+ if (name === "__all__") {
100
+ // Don't create a symbol, but we could parse the list
101
+ break;
102
+ }
103
+ const isUpper = name === name.toUpperCase() && name.length > 1;
104
+ symbols.push({
105
+ name,
106
+ kind: "variable",
107
+ filePath,
108
+ lineStart: node.startPosition.row + 1,
109
+ lineEnd: node.endPosition.row + 1,
110
+ colStart: node.startPosition.column,
111
+ colEnd: node.endPosition.column,
112
+ parentSymbolId: parentIndex,
113
+ isExported: !name.startsWith("_"),
114
+ });
115
+ }
116
+ break;
117
+ }
118
+ case "call": {
119
+ const func = node.childForFieldName("function");
120
+ if (func && parentIndex !== undefined) {
121
+ const name = func.text;
122
+ references.push({
123
+ fromSymbolId: parentIndex,
124
+ toSymbolName: name,
125
+ toSymbolBareName: bareName(name),
126
+ kind: "call",
127
+ line: node.startPosition.row + 1,
128
+ col: node.startPosition.column,
129
+ });
130
+ }
131
+ break;
132
+ }
133
+ case "import_statement": {
134
+ // import foo, import foo.bar
135
+ for (let i = 0; i < node.namedChildCount; i++) {
136
+ const child = node.namedChild(i);
137
+ if (child?.type === "dotted_name") {
138
+ imports.push({
139
+ sourcePath: child.text,
140
+ importedName: child.text,
141
+ localName: bareName(child.text),
142
+ isDefault: false,
143
+ isNamespace: true,
144
+ line: node.startPosition.row + 1,
145
+ });
146
+ }
147
+ else if (child?.type === "aliased_import") {
148
+ const name = child.childForFieldName("name")?.text ?? "";
149
+ const alias = child.childForFieldName("alias")?.text ?? name;
150
+ imports.push({
151
+ sourcePath: name,
152
+ importedName: name,
153
+ localName: alias,
154
+ isDefault: false,
155
+ isNamespace: true,
156
+ line: node.startPosition.row + 1,
157
+ });
158
+ }
159
+ }
160
+ return;
161
+ }
162
+ case "import_from_statement": {
163
+ // from foo import bar, baz
164
+ const moduleNode = node.childForFieldName("module_name");
165
+ const moduleName = moduleNode?.text ?? "";
166
+ for (let i = 0; i < node.namedChildCount; i++) {
167
+ const child = node.namedChild(i);
168
+ if (child?.type === "dotted_name" && child !== moduleNode) {
169
+ imports.push({
170
+ sourcePath: moduleName,
171
+ importedName: child.text,
172
+ localName: child.text,
173
+ isDefault: false,
174
+ isNamespace: false,
175
+ line: node.startPosition.row + 1,
176
+ });
177
+ }
178
+ else if (child?.type === "aliased_import") {
179
+ const name = child.childForFieldName("name")?.text ?? "";
180
+ const alias = child.childForFieldName("alias")?.text ?? name;
181
+ imports.push({
182
+ sourcePath: moduleName,
183
+ importedName: name,
184
+ localName: alias,
185
+ isDefault: false,
186
+ isNamespace: false,
187
+ line: node.startPosition.row + 1,
188
+ });
189
+ }
190
+ else if (child?.type === "wildcard_import") {
191
+ imports.push({
192
+ sourcePath: moduleName,
193
+ importedName: "*",
194
+ localName: "*",
195
+ isDefault: false,
196
+ isNamespace: true,
197
+ line: node.startPosition.row + 1,
198
+ });
199
+ }
200
+ }
201
+ return;
202
+ }
203
+ case "decorated_definition": {
204
+ // Walk into the decorated function/class
205
+ for (let i = 0; i < node.namedChildCount; i++) {
206
+ const child = node.namedChild(i);
207
+ if (child && child.type !== "decorator") {
208
+ walkPython(child, filePath, symbols, references, imports, parentIndex);
209
+ }
210
+ }
211
+ return;
212
+ }
213
+ }
214
+ // Default: walk children
215
+ walkPyChildren(node, filePath, symbols, references, imports, parentIndex);
216
+ }
217
+ function walkPyChildren(node, filePath, symbols, references, imports, parentIndex) {
218
+ for (let i = 0; i < node.namedChildCount; i++) {
219
+ const child = node.namedChild(i);
220
+ if (child) {
221
+ walkPython(child, filePath, symbols, references, imports, parentIndex);
222
+ }
223
+ }
224
+ }
225
+ export const pythonPlugin = {
226
+ id: "python",
227
+ extensions: [".py", ".pyi"],
228
+ getParser: () => pyParser,
229
+ walk: walkPython,
230
+ };
231
+ registerLanguage(pythonPlugin);
232
+ //# sourceMappingURL=lang-python.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"lang-python.js","sourceRoot":"","sources":["../../src/indexer/lang-python.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AAExC,OAAO,EAAuB,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AAEpD,MAAM,QAAQ,GAAG,IAAI,MAAM,EAAE,CAAC;AAC9B,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;AAE7B,SAAS,kBAAkB,CAAC,IAAuB;IACjD,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;IACxD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACpD,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;IACzD,MAAM,SAAS,GAAG,MAAM,EAAE,IAAI,IAAI,IAAI,CAAC;IACvC,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,OAAO,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3D,OAAO,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,EAAE,CAAC;AACzC,CAAC;AAED,SAAS,WAAW,CAAC,IAAuB,EAAE,aAAqB;IACjE,IAAI,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,sBAAsB,EAAE,CAAC;QACjD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;YACrD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACxC,IAAI,KAAK,EAAE,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC;gBAAE,OAAO,IAAI,CAAC;QACrF,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,UAAU,CACjB,IAAuB,EACvB,QAAgB,EAChB,OAAqB,EACrB,UAA2B,EAC3B,OAAqB,EACrB,WAAoB;IAEpB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,qBAAqB,CAAC,CAAC,CAAC;YAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,QAAQ,GAAG,WAAW,KAAK,SAAS,IAAI,OAAO,CAAC,WAAW,CAAC,EAAE,IAAI,KAAK,OAAO,CAAC;gBACrF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACnF,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAChF,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU;oBACtC,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,CAAC,SAAS,IAAI,QAAQ;oBAClC,SAAS,EAAE,kBAAkB,CAAC,IAAI,CAAC;iBACpC,CAAC;gBACF,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBAClE,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAChD,MAAM,GAAG,GAAe;oBACtB,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,IAAI,EAAE,OAAO;oBACb,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,CAAC,SAAS;oBACtB,SAAS,EAAE,SAAS,QAAQ,CAAC,IAAI,EAAE;iBACpC,CAAC;gBAEF,qBAAqB;gBACrB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;gBAC5D,IAAI,YAAY,EAAE,CAAC;oBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;wBACtD,MAAM,GAAG,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;wBACvC,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;4BAC3C,UAAU,CAAC,IAAI,CAAC;gCACd,YAAY,EAAE,OAAO,CAAC,MAAM;gCAC5B,YAAY,EAAE,GAAG,CAAC,IAAI;gCACtB,gBAAgB,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC;gCACpC,IAAI,EAAE,SAAS;gCACf,IAAI,EAAE,GAAG,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gCAC/B,GAAG,EAAE,GAAG,CAAC,aAAa,CAAC,MAAM;6BAC9B,CAAC,CAAC;4BACH,GAAG,CAAC,SAAS,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;wBACrE,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;gBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAClB,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;gBAClE,OAAO;YACT,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,8DAA8D;YAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,IAAI,EAAE,IAAI,KAAK,YAAY,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvB,qEAAqE;gBACrE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;oBACvB,qDAAqD;oBACrD,MAAM;gBACR,CAAC;gBACD,MAAM,OAAO,GAAG,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAC/D,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI;oBACJ,IAAI,EAAE,UAAU;oBAChB,QAAQ;oBACR,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;oBACnC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;oBAC/B,cAAc,EAAE,WAAW;oBAC3B,UAAU,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;iBAClC,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAChD,IAAI,IAAI,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvB,UAAU,CAAC,IAAI,CAAC;oBACd,YAAY,EAAE,WAAW;oBACzB,YAAY,EAAE,IAAI;oBAClB,gBAAgB,EAAE,QAAQ,CAAC,IAAI,CAAC;oBAChC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBAChC,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,MAAM;iBAC/B,CAAC,CAAC;YACL,CAAC;YACD,MAAM;QACR,CAAC;QAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;YACxB,6BAA6B;YAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACjC,IAAI,KAAK,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;oBAClC,OAAO,CAAC,IAAI,CAAC;wBACX,UAAU,EAAE,KAAK,CAAC,IAAI;wBACtB,YAAY,EAAE,KAAK,CAAC,IAAI;wBACxB,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC;wBAC/B,SAAS,EAAE,KAAK;wBAChB,WAAW,EAAE,IAAI;wBACjB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,gBAAgB,EAAE,CAAC;oBAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;oBACzD,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC;oBAC7D,OAAO,CAAC,IAAI,CAAC;wBACX,UAAU,EAAE,IAAI;wBAChB,YAAY,EAAE,IAAI;wBAClB,SAAS,EAAE,KAAK;wBAChB,SAAS,EAAE,KAAK;wBAChB,WAAW,EAAE,IAAI;wBACjB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,OAAO;QACT,CAAC;QAED,KAAK,uBAAuB,CAAC,CAAC,CAAC;YAC7B,2BAA2B;YAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YACzD,MAAM,UAAU,GAAG,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC;YAE1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACjC,IAAI,KAAK,EAAE,IAAI,KAAK,aAAa,IAAI,KAAK,KAAK,UAAU,EAAE,CAAC;oBAC1D,OAAO,CAAC,IAAI,CAAC;wBACX,UAAU,EAAE,UAAU;wBACtB,YAAY,EAAE,KAAK,CAAC,IAAI;wBACxB,SAAS,EAAE,KAAK,CAAC,IAAI;wBACrB,SAAS,EAAE,KAAK;wBAChB,WAAW,EAAE,KAAK;wBAClB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,gBAAgB,EAAE,CAAC;oBAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;oBACzD,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC;oBAC7D,OAAO,CAAC,IAAI,CAAC;wBACX,UAAU,EAAE,UAAU;wBACtB,YAAY,EAAE,IAAI;wBAClB,SAAS,EAAE,KAAK;wBAChB,SAAS,EAAE,KAAK;wBAChB,WAAW,EAAE,KAAK;wBAClB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC7C,OAAO,CAAC,IAAI,CAAC;wBACX,UAAU,EAAE,UAAU;wBACtB,YAAY,EAAE,GAAG;wBACjB,SAAS,EAAE,GAAG;wBACd,SAAS,EAAE,KAAK;wBAChB,WAAW,EAAE,IAAI;wBACjB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,OAAO;QACT,CAAC;QAED,KAAK,sBAAsB,CAAC,CAAC,CAAC;YAC5B,yCAAyC;YACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;gBACjC,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACxC,UAAU,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;gBACzE,CAAC;YACH,CAAC;YACD,OAAO;QACT,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,cAAc,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,cAAc,CACrB,IAAuB,EACvB,QAAgB,EAChB,OAAqB,EACrB,UAA2B,EAC3B,OAAqB,EACrB,WAAoB;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,KAAK,EAAE,CAAC;YACV,UAAU,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAmB;IAC1C,EAAE,EAAE,QAAQ;IACZ,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,GAAG,EAAE,CAAC,QAAQ;IACzB,IAAI,EAAE,UAAU;CACjB,CAAC;AAEF,gBAAgB,CAAC,YAAY,CAAC,CAAC"}
@@ -0,0 +1,2 @@
1
+ import { type LanguagePlugin } from "./language-plugin.js";
2
+ export declare const rustPlugin: LanguagePlugin;