@optave/codegraph 3.2.0 → 3.3.0

package/dist/extractors/ruby.js

@@ -0,0 +1,284 @@
+import { findChild, nodeEndLine } from './helpers.js';
+/**
+ * Extract symbols from Ruby files.
+ */
+export function extractRubySymbols(tree, _filePath) {
+    const ctx = {
+        definitions: [],
+        calls: [],
+        imports: [],
+        classes: [],
+        exports: [],
+    };
+    walkRubyNode(tree.rootNode, ctx);
+    return ctx;
+}
+function walkRubyNode(node, ctx) {
+    switch (node.type) {
+        case 'class':
+            handleRubyClass(node, ctx);
+            break;
+        case 'module':
+            handleRubyModule(node, ctx);
+            break;
+        case 'method':
+            handleRubyMethod(node, ctx);
+            break;
+        case 'singleton_method':
+            handleRubySingletonMethod(node, ctx);
+            break;
+        case 'assignment':
+            handleRubyAssignment(node, ctx);
+            break;
+        case 'call':
+            handleRubyCall(node, ctx);
+            break;
+    }
+    for (let i = 0; i < node.childCount; i++)
+        walkRubyNode(node.child(i), ctx);
+}
+// ── Walk-path per-node-type handlers ────────────────────────────────────────
+function handleRubyClass(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const classChildren = extractRubyClassChildren(node);
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'class',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: classChildren.length > 0 ? classChildren : undefined,
+    });
+    const superclass = node.childForFieldName('superclass');
+    if (superclass) {
+        for (let i = 0; i < superclass.childCount; i++) {
+            const child = superclass.child(i);
+            if (child && (child.type === 'constant' || child.type === 'scope_resolution')) {
+                ctx.classes.push({
+                    name: nameNode.text,
+                    extends: child.text,
+                    line: node.startPosition.row + 1,
+                });
+                break;
+            }
+        }
+        if (superclass.type === 'superclass') {
+            for (let i = 0; i < superclass.childCount; i++) {
+                const child = superclass.child(i);
+                if (child && (child.type === 'constant' || child.type === 'scope_resolution')) {
+                    ctx.classes.push({
+                        name: nameNode.text,
+                        extends: child.text,
+                        line: node.startPosition.row + 1,
+                    });
+                    break;
+                }
+            }
+        }
+    }
+}
+function handleRubyModule(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const moduleChildren = extractRubyBodyConstants(node);
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'module',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: moduleChildren.length > 0 ? moduleChildren : undefined,
+    });
+}
+function handleRubyMethod(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const parentClass = findRubyParentClass(node);
+    const fullName = parentClass ? `${parentClass}.${nameNode.text}` : nameNode.text;
+    const params = extractRubyParameters(node);
+    ctx.definitions.push({
+        name: fullName,
+        kind: 'method',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: params.length > 0 ? params : undefined,
+    });
+}
+function handleRubySingletonMethod(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const parentClass = findRubyParentClass(node);
+    const fullName = parentClass ? `${parentClass}.${nameNode.text}` : nameNode.text;
+    const params = extractRubyParameters(node);
+    ctx.definitions.push({
+        name: fullName,
+        kind: 'function',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: params.length > 0 ? params : undefined,
+    });
+}
+function handleRubyAssignment(node, ctx) {
+    if (node.parent && node.parent.type === 'program') {
+        const left = node.childForFieldName('left');
+        if (left && left.type === 'constant') {
+            ctx.definitions.push({
+                name: left.text,
+                kind: 'constant',
+                line: node.startPosition.row + 1,
+                endLine: nodeEndLine(node),
+            });
+        }
+    }
+}
+function handleRubyCall(node, ctx) {
+    const methodNode = node.childForFieldName('method');
+    if (!methodNode)
+        return;
+    if (methodNode.text === 'require' || methodNode.text === 'require_relative') {
+        handleRubyRequire(node, ctx);
+    }
+    else if (methodNode.text === 'include' ||
+        methodNode.text === 'extend' ||
+        methodNode.text === 'prepend') {
+        handleRubyModuleInclusion(node, methodNode, ctx);
+    }
+    else {
+        const recv = node.childForFieldName('receiver');
+        const call = { name: methodNode.text, line: node.startPosition.row + 1 };
+        if (recv)
+            call.receiver = recv.text;
+        ctx.calls.push(call);
+    }
+}
+function handleRubyRequire(node, ctx) {
+    const args = node.childForFieldName('arguments');
+    if (!args)
+        return;
+    for (let i = 0; i < args.childCount; i++) {
+        const arg = args.child(i);
+        if (arg && (arg.type === 'string' || arg.type === 'string_content')) {
+            const strContent = arg.text.replace(/^['"]|['"]$/g, '');
+            ctx.imports.push({
+                source: strContent,
+                names: [strContent.split('/').pop()],
+                line: node.startPosition.row + 1,
+                rubyRequire: true,
+            });
+            break;
+        }
+        if (arg && arg.type === 'string') {
+            const content = findChild(arg, 'string_content');
+            if (content) {
+                ctx.imports.push({
+                    source: content.text,
+                    names: [content.text.split('/').pop()],
+                    line: node.startPosition.row + 1,
+                    rubyRequire: true,
+                });
+                break;
+            }
+        }
+    }
+}
+function handleRubyModuleInclusion(node, _methodNode, ctx) {
+    const parentClass = findRubyParentClass(node);
+    if (!parentClass)
+        return;
+    const args = node.childForFieldName('arguments');
+    if (!args)
+        return;
+    for (let i = 0; i < args.childCount; i++) {
+        const arg = args.child(i);
+        if (arg && (arg.type === 'constant' || arg.type === 'scope_resolution')) {
+            ctx.classes.push({
+                name: parentClass,
+                implements: arg.text,
+                line: node.startPosition.row + 1,
+            });
+        }
+    }
+}
+function findRubyParentClass(node) {
+    let current = node.parent;
+    while (current) {
+        if (current.type === 'class' || current.type === 'module') {
+            const nameNode = current.childForFieldName('name');
+            return nameNode ? nameNode.text : null;
+        }
+        current = current.parent;
+    }
+    return null;
+}
+// ── Child extraction helpers ────────────────────────────────────────────────
+const RUBY_PARAM_TYPES = new Set([
+    'identifier',
+    'optional_parameter',
+    'splat_parameter',
+    'hash_splat_parameter',
+    'block_parameter',
+    'keyword_parameter',
+]);
+function extractRubyParameters(methodNode) {
+    const params = [];
+    const paramList = methodNode.childForFieldName('parameters') || findChild(methodNode, 'method_parameters');
+    if (!paramList)
+        return params;
+    for (let i = 0; i < paramList.childCount; i++) {
+        const param = paramList.child(i);
+        if (!param || !RUBY_PARAM_TYPES.has(param.type))
+            continue;
+        let name;
+        if (param.type === 'identifier') {
+            name = param.text;
+        }
+        else {
+            // Compound parameter types have an identifier child for the name
+            const id = findChild(param, 'identifier');
+            name = id ? id.text : param.text;
+        }
+        params.push({ name, kind: 'parameter', line: param.startPosition.row + 1 });
+    }
+    return params;
+}
+function extractRubyBodyConstants(containerNode) {
+    const children = [];
+    const body = containerNode.childForFieldName('body') || findChild(containerNode, 'body');
+    if (!body)
+        return children;
+    for (let i = 0; i < body.childCount; i++) {
+        const child = body.child(i);
+        if (!child || child.type !== 'assignment')
+            continue;
+        const left = child.childForFieldName('left');
+        if (left && left.type === 'constant') {
+            children.push({ name: left.text, kind: 'constant', line: child.startPosition.row + 1 });
+        }
+    }
+    return children;
+}
+function extractRubyClassChildren(classNode) {
+    const children = [];
+    const body = classNode.childForFieldName('body') || findChild(classNode, 'body');
+    if (!body)
+        return children;
+    for (let i = 0; i < body.childCount; i++) {
+        const child = body.child(i);
+        if (!child || child.type !== 'assignment')
+            continue;
+        const left = child.childForFieldName('left');
+        if (!left)
+            continue;
+        if (left.type === 'instance_variable') {
+            children.push({ name: left.text, kind: 'property', line: child.startPosition.row + 1 });
+        }
+        else if (left.type === 'constant') {
+            children.push({ name: left.text, kind: 'constant', line: child.startPosition.row + 1 });
+        }
+    }
+    return children;
+}
+//# sourceMappingURL=ruby.js.map

package/dist/extractors/ruby.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ruby.js","sourceRoot":"","sources":["../../src/extractors/ruby.js"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAEtD;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAI,EAAE,SAAS;IAChD,MAAM,GAAG,GAAG;QACV,WAAW,EAAE,EAAE;QACf,KAAK,EAAE,EAAE;QACT,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;KACZ,CAAC;IAEF,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IACjC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,YAAY,CAAC,IAAI,EAAE,GAAG;IAC7B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,KAAK,OAAO;YACV,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM;QACR,KAAK,QAAQ;YACX,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,QAAQ;YACX,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC5B,MAAM;QACR,KAAK,kBAAkB;YACrB,yBAAyB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YACrC,MAAM;QACR,KAAK,YAAY;YACf,oBAAoB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAChC,MAAM;QACR,KAAK,MAAM;YACT,cAAc,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC1B,MAAM;IACV,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE;QAAE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AAC7E,CAAC;AAED,+EAA+E;AAE/E,SAAS,eAAe,CAAC,IAAI,EAAE,GAAG;IAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,MAAM,aAAa,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;IACrD,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,QAAQ,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;KAC/D,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACxD,IAAI,UAAU,EAAE,CAAC;QACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAAE,CAAC;gBAC9E,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,OAAO,EAAE,KAAK,CAAC,IAAI;oBACnB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;iBACjC,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,UAAU,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC/C,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAClC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAAE,CAAC;oBAC9E,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;wBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;wBACnB,OAAO,EAAE,KAAK,CAAC,IAAI;wBACnB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;qBACjC,CAAC,CAAC;oBACH,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAI,EAAE,GAAG;IACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,MAAM,cAAc,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;IACtD,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,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;KACjE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAI,EAAE,GAAG;IACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;IACjF,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAC3C,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;QACnB,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;QAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;QAC1B,QAAQ,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KACjD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAI,EAAE,GAAG;IAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,CAAC,QAAQ;QAAE,OAAO;IACtB,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;IACjF,MAAM,MAAM,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAC3C,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,QAAQ,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KACjD,CAAC,CAAC;AACL,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAI,EAAE,GAAG;IACrC,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QAClD,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC5C,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACrC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;gBACnB,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gBAChC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,IAAI,EAAE,GAAG;IAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACpD,IAAI,CAAC,UAAU;QAAE,OAAO;IACxB,IAAI,UAAU,CAAC,IAAI,KAAK,SAAS,IAAI,UAAU,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;QAC5E,iBAAiB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC;SAAM,IACL,UAAU,CAAC,IAAI,KAAK,SAAS;QAC7B,UAAU,CAAC,IAAI,KAAK,QAAQ;QAC5B,UAAU,CAAC,IAAI,KAAK,SAAS,EAC7B,CAAC;QACD,yBAAyB,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;IACnD,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC;QACzE,IAAI,IAAI;YAAE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;QACpC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAI,EAAE,GAAG;IAClC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACjD,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,IAAI,KAAK,gBAAgB,CAAC,EAAE,CAAC;YACpE,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YACxD,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;gBACf,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;gBACpC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;gBAChC,WAAW,EAAE,IAAI;aAClB,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QACD,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC;YACjD,IAAI,OAAO,EAAE,CAAC;gBACZ,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;oBACf,MAAM,EAAE,OAAO,CAAC,IAAI;oBACpB,KAAK,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;oBACtC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBAChC,WAAW,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,yBAAyB,CAAC,IAAI,EAAE,WAAW,EAAE,GAAG;IACvD,MAAM,WAAW,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,CAAC,WAAW;QAAE,OAAO;IACzB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;IACjD,IAAI,CAAC,IAAI;QAAE,OAAO;IAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,UAAU,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAAE,CAAC;YACxE,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,GAAG,CAAC,IAAI;gBACpB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;aACjC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAI;IAC/B,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;IAC1B,OAAO,OAAO,EAAE,CAAC;QACf,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1D,MAAM,QAAQ,GAAG,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACnD,OAAO,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;QACzC,CAAC;QACD,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAC3B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,+EAA+E;AAE/E,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC;IAC/B,YAAY;IACZ,oBAAoB;IACpB,iBAAiB;IACjB,sBAAsB;IACtB,iBAAiB;IACjB,mBAAmB;CACpB,CAAC,CAAC;AAEH,SAAS,qBAAqB,CAAC,UAAU;IACvC,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,MAAM,SAAS,GACb,UAAU,CAAC,iBAAiB,CAAC,YAAY,CAAC,IAAI,SAAS,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;IAC3F,IAAI,CAAC,SAAS;QAAE,OAAO,MAAM,CAAC;IAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,SAAS;QAC1D,IAAI,IAAI,CAAC;QACT,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAChC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACpB,CAAC;aAAM,CAAC;YACN,iEAAiE;YACjE,MAAM,EAAE,GAAG,SAAS,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YAC1C,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;QACnC,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,wBAAwB,CAAC,aAAa;IAC7C,MAAM,QAAQ,GAAG,EAAE,CAAC;IACpB,MAAM,IAAI,GAAG,aAAa,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IACzF,IAAI,CAAC,IAAI;QAAE,OAAO,QAAQ,CAAC;IAC3B,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,YAAY;YAAE,SAAS;QACpD,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACrC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,wBAAwB,CAAC,SAAS;IACzC,MAAM,QAAQ,GAAG,EAAE,CAAC;IACpB,MAAM,IAAI,GAAG,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACjF,IAAI,CAAC,IAAI;QAAE,OAAO,QAAQ,CAAC;IAC3B,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,YAAY;YAAE,SAAS;QACpD,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,IAAI;YAAE,SAAS;QACpB,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACtC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACpC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/extractors/rust.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Extract symbols from Rust files.
+ */
+export function extractRustSymbols(tree: any, _filePath: any): {
+    definitions: never[];
+    calls: never[];
+    imports: never[];
+    classes: never[];
+    exports: never[];
+    typeMap: Map<any, any>;
+};
+//# sourceMappingURL=rust.d.ts.map

package/dist/extractors/rust.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rust.d.ts","sourceRoot":"","sources":["../../src/extractors/rust.js"],"names":[],"mappings":"AAEA;;GAEG;AACH;;;;;;;EAaC"}

package/dist/extractors/rust.js

@@ -0,0 +1,366 @@
+import { findChild, nodeEndLine, rustVisibility } from './helpers.js';
+/**
+ * Extract symbols from Rust files.
+ */
+export function extractRustSymbols(tree, _filePath) {
+    const ctx = {
+        definitions: [],
+        calls: [],
+        imports: [],
+        classes: [],
+        exports: [],
+        typeMap: new Map(),
+    };
+    walkRustNode(tree.rootNode, ctx);
+    extractRustTypeMap(tree.rootNode, ctx);
+    return ctx;
+}
+function walkRustNode(node, ctx) {
+    switch (node.type) {
+        case 'function_item':
+            handleRustFuncItem(node, ctx);
+            break;
+        case 'struct_item':
+            handleRustStructItem(node, ctx);
+            break;
+        case 'enum_item':
+            handleRustEnumItem(node, ctx);
+            break;
+        case 'const_item':
+            handleRustConstItem(node, ctx);
+            break;
+        case 'trait_item':
+            handleRustTraitItem(node, ctx);
+            break;
+        case 'impl_item':
+            handleRustImplItem(node, ctx);
+            break;
+        case 'use_declaration':
+            handleRustUseDecl(node, ctx);
+            break;
+        case 'call_expression':
+            handleRustCallExpr(node, ctx);
+            break;
+        case 'macro_invocation':
+            handleRustMacroInvocation(node, ctx);
+            break;
+    }
+    for (let i = 0; i < node.childCount; i++)
+        walkRustNode(node.child(i), ctx);
+}
+// ── Walk-path per-node-type handlers ────────────────────────────────────────
+function handleRustFuncItem(node, ctx) {
+    // Skip default-impl functions already emitted by handleRustTraitItem
+    if (node.parent?.parent?.type === 'trait_item')
+        return;
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const implType = findCurrentImpl(node);
+    const fullName = implType ? `${implType}.${nameNode.text}` : nameNode.text;
+    const kind = implType ? 'method' : 'function';
+    const params = extractRustParameters(node.childForFieldName('parameters'));
+    ctx.definitions.push({
+        name: fullName,
+        kind,
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: params.length > 0 ? params : undefined,
+        visibility: rustVisibility(node),
+    });
+}
+function handleRustStructItem(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const fields = extractStructFields(node);
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'struct',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: fields.length > 0 ? fields : undefined,
+        visibility: rustVisibility(node),
+    });
+}
+function handleRustEnumItem(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    const variants = extractEnumVariants(node);
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'enum',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+        children: variants.length > 0 ? variants : undefined,
+    });
+}
+function handleRustConstItem(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'constant',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+    });
+}
+function handleRustTraitItem(node, ctx) {
+    const nameNode = node.childForFieldName('name');
+    if (!nameNode)
+        return;
+    ctx.definitions.push({
+        name: nameNode.text,
+        kind: 'trait',
+        line: node.startPosition.row + 1,
+        endLine: nodeEndLine(node),
+    });
+    const body = node.childForFieldName('body');
+    if (body) {
+        for (let i = 0; i < body.childCount; i++) {
+            const child = body.child(i);
+            if (child && (child.type === 'function_signature_item' || child.type === 'function_item')) {
+                const methName = child.childForFieldName('name');
+                if (methName) {
+                    ctx.definitions.push({
+                        name: `${nameNode.text}.${methName.text}`,
+                        kind: 'method',
+                        line: child.startPosition.row + 1,
+                        endLine: child.endPosition.row + 1,
+                    });
+                }
+            }
+        }
+    }
+}
+function handleRustImplItem(node, ctx) {
+    const typeNode = node.childForFieldName('type');
+    const traitNode = node.childForFieldName('trait');
+    if (typeNode && traitNode) {
+        ctx.classes.push({
+            name: typeNode.text,
+            implements: traitNode.text,
+            line: node.startPosition.row + 1,
+        });
+    }
+}
+function handleRustUseDecl(node, ctx) {
+    const argNode = node.child(1);
+    if (!argNode)
+        return;
+    const usePaths = extractRustUsePath(argNode);
+    for (const imp of usePaths) {
+        ctx.imports.push({
+            source: imp.source,
+            names: imp.names,
+            line: node.startPosition.row + 1,
+            rustUse: true,
+        });
+    }
+}
+function handleRustCallExpr(node, ctx) {
+    const fn = node.childForFieldName('function');
+    if (!fn)
+        return;
+    if (fn.type === 'identifier') {
+        ctx.calls.push({ name: fn.text, line: node.startPosition.row + 1 });
+    }
+    else if (fn.type === 'field_expression') {
+        const field = fn.childForFieldName('field');
+        if (field) {
+            const value = fn.childForFieldName('value');
+            const call = { name: field.text, line: node.startPosition.row + 1 };
+            if (value)
+                call.receiver = value.text;
+            ctx.calls.push(call);
+        }
+    }
+    else if (fn.type === 'scoped_identifier') {
+        const name = fn.childForFieldName('name');
+        if (name) {
+            const path = fn.childForFieldName('path');
+            const call = { name: name.text, line: node.startPosition.row + 1 };
+            if (path)
+                call.receiver = path.text;
+            ctx.calls.push(call);
+        }
+    }
+}
+function handleRustMacroInvocation(node, ctx) {
+    const macroNode = node.child(0);
+    if (macroNode) {
+        ctx.calls.push({ name: `${macroNode.text}!`, line: node.startPosition.row + 1 });
+    }
+}
+function findCurrentImpl(node) {
+    let current = node.parent;
+    while (current) {
+        if (current.type === 'impl_item') {
+            const typeNode = current.childForFieldName('type');
+            return typeNode ? typeNode.text : null;
+        }
+        current = current.parent;
+    }
+    return null;
+}
+// ── Child extraction helpers ────────────────────────────────────────────────
+function extractRustParameters(paramListNode) {
+    const params = [];
+    if (!paramListNode)
+        return params;
+    for (let i = 0; i < paramListNode.childCount; i++) {
+        const param = paramListNode.child(i);
+        if (!param)
+            continue;
+        if (param.type === 'self_parameter') {
+            params.push({ name: 'self', kind: 'parameter', line: param.startPosition.row + 1 });
+        }
+        else if (param.type === 'parameter') {
+            const pattern = param.childForFieldName('pattern');
+            if (pattern) {
+                params.push({ name: pattern.text, kind: 'parameter', line: param.startPosition.row + 1 });
+            }
+        }
+    }
+    return params;
+}
+function extractStructFields(structNode) {
+    const fields = [];
+    const fieldList = structNode.childForFieldName('body') || findChild(structNode, 'field_declaration_list');
+    if (!fieldList)
+        return fields;
+    for (let i = 0; i < fieldList.childCount; i++) {
+        const field = fieldList.child(i);
+        if (!field || field.type !== 'field_declaration')
+            continue;
+        const nameNode = field.childForFieldName('name');
+        if (nameNode) {
+            fields.push({ name: nameNode.text, kind: 'property', line: field.startPosition.row + 1 });
+        }
+    }
+    return fields;
+}
+function extractEnumVariants(enumNode) {
+    const variants = [];
+    const body = enumNode.childForFieldName('body') || findChild(enumNode, 'enum_variant_list');
+    if (!body)
+        return variants;
+    for (let i = 0; i < body.childCount; i++) {
+        const variant = body.child(i);
+        if (!variant || variant.type !== 'enum_variant')
+            continue;
+        const nameNode = variant.childForFieldName('name');
+        if (nameNode) {
+            variants.push({ name: nameNode.text, kind: 'constant', line: variant.startPosition.row + 1 });
+        }
+    }
+    return variants;
+}
+function extractRustTypeMap(node, ctx) {
+    extractRustTypeMapDepth(node, ctx, 0);
+}
+function extractRustTypeMapDepth(node, ctx, depth) {
+    if (depth >= 200)
+        return;
+    // let x: MyType = ...
+    if (node.type === 'let_declaration') {
+        const pattern = node.childForFieldName('pattern');
+        const typeNode = node.childForFieldName('type');
+        if (pattern && pattern.type === 'identifier' && typeNode) {
+            const typeName = extractRustTypeName(typeNode);
+            if (typeName)
+                ctx.typeMap.set(pattern.text, { type: typeName, confidence: 0.9 });
+        }
+    }
+    // fn foo(x: MyType) — parameter node has pattern + type fields
+    if (node.type === 'parameter') {
+        const pattern = node.childForFieldName('pattern');
+        const typeNode = node.childForFieldName('type');
+        if (pattern && typeNode) {
+            const name = pattern.type === 'identifier' ? pattern.text : null;
+            if (name && name !== 'self' && name !== '&self' && name !== '&mut self') {
+                const typeName = extractRustTypeName(typeNode);
+                if (typeName)
+                    ctx.typeMap.set(name, { type: typeName, confidence: 0.9 });
+            }
+        }
+    }
+    for (let i = 0; i < node.childCount; i++) {
+        const child = node.child(i);
+        if (child)
+            extractRustTypeMapDepth(child, ctx, depth + 1);
+    }
+}
+function extractRustTypeName(typeNode) {
+    if (!typeNode)
+        return null;
+    const t = typeNode.type;
+    if (t === 'type_identifier' || t === 'identifier')
+        return typeNode.text;
+    if (t === 'scoped_type_identifier')
+        return typeNode.text;
+    // Reference: &MyType or &mut MyType → MyType
+    if (t === 'reference_type') {
+        for (let i = 0; i < typeNode.childCount; i++) {
+            const child = typeNode.child(i);
+            if (child && (child.type === 'type_identifier' || child.type === 'scoped_type_identifier')) {
+                return child.text;
+            }
+        }
+    }
+    // Generic: Vec<T> → Vec
+    if (t === 'generic_type') {
+        const first = typeNode.child(0);
+        return first ? first.text : null;
+    }
+    return null;
+}
+function extractRustUsePath(node) {
+    if (!node)
+        return [];
+    if (node.type === 'use_list') {
+        const results = [];
+        for (let i = 0; i < node.childCount; i++) {
+            results.push(...extractRustUsePath(node.child(i)));
+        }
+        return results;
+    }
+    if (node.type === 'scoped_use_list') {
+        const pathNode = node.childForFieldName('path');
+        const listNode = node.childForFieldName('list');
+        const prefix = pathNode ? pathNode.text : '';
+        if (listNode) {
+            const names = [];
+            for (let i = 0; i < listNode.childCount; i++) {
+                const child = listNode.child(i);
+                if (child &&
+                    (child.type === 'identifier' || child.type === 'use_as_clause' || child.type === 'self')) {
+                    const name = child.type === 'use_as_clause'
+                        ? (child.childForFieldName('alias') || child.childForFieldName('name'))?.text
+                        : child.text;
+                    if (name)
+                        names.push(name);
+                }
+            }
+            return [{ source: prefix, names }];
+        }
+        return [{ source: prefix, names: [] }];
+    }
+    if (node.type === 'use_as_clause') {
+        const name = node.childForFieldName('alias') || node.childForFieldName('name');
+        return [{ source: node.text, names: name ? [name.text] : [] }];
+    }
+    if (node.type === 'use_wildcard') {
+        const pathNode = node.childForFieldName('path');
+        return [{ source: pathNode ? pathNode.text : '*', names: ['*'] }];
+    }
+    if (node.type === 'scoped_identifier' || node.type === 'identifier') {
+        const text = node.text;
+        const lastName = text.split('::').pop();
+        return [{ source: text, names: [lastName] }];
+    }
+    return [];
+}
+//# sourceMappingURL=rust.js.map