@kodus/kodus-graph 0.2.8 → 0.2.9

package/src/parser/extractors/python.ts

@@ -1,133 +0,0 @@
-import type { SgNode, SgRoot } from '@ast-grep/napi';
-import type { RawCallSite, RawGraph } from '../../graph/types';
-import { type CallExtractionConfig, extractCalls } from '../../shared/extract-calls';
-import { computeContentHash } from '../../shared/file-hash';
-import { LANG_KINDS } from '../languages';
-
-export function extractPython(root: SgRoot, fp: string, seen: Set<string>, graph: RawGraph): void {
-  const kinds = LANG_KINDS.python;
-  const rootNode = root.root();
-
-  // ── Classes ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.class } })) {
-    const name = node.field('name')?.text();
-    if (!name || seen.has(`c:${fp}:${name}`)) continue;
-    seen.add(`c:${fp}:${name}`);
-
-    const argList = node.field('superclasses') || node.children().find((c: SgNode) => c.kind() === 'argument_list');
-    const extendsName =
-      argList
-        ?.children()
-        .find((c: SgNode) => c.kind() === 'identifier')
-        ?.text() || '';
-
-    graph.classes.push({
-      name,
-      file: fp,
-      line_start: node.range().start.line,
-      line_end: node.range().end.line,
-      extends: extendsName,
-      implements: '',
-      qualified: `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Functions / Methods ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.function } })) {
-    const name = node.field('name')?.text();
-    if (!name) continue;
-    const line = node.range().start.line;
-    if (seen.has(`m:${fp}:${name}:${line}`)) continue;
-    seen.add(`m:${fp}:${name}:${line}`);
-
-    const classAncestor = node.ancestors().find((a: SgNode) => a.kind() === kinds.class);
-    const className = classAncestor?.field('name')?.text() || '';
-    const retType =
-      node
-        .field('return_type')
-        ?.text()
-        ?.replace(/^->\s*/, '') || '';
-
-    const isTest = name.startsWith('test_');
-    if (isTest) {
-      graph.tests.push({
-        name,
-        file: fp,
-        line_start: line,
-        line_end: node.range().end.line,
-        qualified: `${fp}::test:${name}`,
-        content_hash: computeContentHash(node.text()),
-      });
-    }
-
-    graph.functions.push({
-      name,
-      file: fp,
-      line_start: line,
-      line_end: node.range().end.line,
-      params: node.field('parameters')?.text() || '()',
-      returnType: retType,
-      kind: name === '__init__' ? 'Constructor' : className ? 'Method' : 'Function',
-      className,
-      qualified: className ? `${fp}::${className}.${name}` : `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Imports (from X import Y) ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.import } })) {
-    const modNode = node.children().find((c: SgNode) => c.kind() === 'dotted_name' || c.kind() === 'relative_import');
-    const modulePath = modNode?.text() || '';
-    if (!modulePath) continue;
-
-    const names: string[] = [];
-    for (const child of node.children()) {
-      if (child.kind() === 'dotted_name' && child !== modNode) names.push(child.text());
-      if (child.kind() === 'identifier' && child !== modNode) names.push(child.text());
-    }
-    graph.imports.push({
-      module: modulePath,
-      file: fp,
-      line: node.range().start.line,
-      names,
-      lang: 'python',
-    });
-  }
-
-  // ── Regular imports (import X) ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.importRegular } })) {
-    const modNode = node.children().find((c: SgNode) => c.kind() === 'dotted_name');
-    if (modNode) {
-      graph.imports.push({
-        module: modNode.text(),
-        file: fp,
-        line: node.range().start.line,
-        names: [modNode.text()],
-        lang: 'python',
-      });
-    }
-  }
-}
-
-/** Python-specific call extraction config for shared extractCalls(). */
-const PYTHON_CALL_CONFIG: CallExtractionConfig = {
-  selfPrefixes: ['self.'],
-  superPrefixes: ['super().'],
-  findEnclosingClass: (node) =>
-    node.ancestors().find((a: SgNode) => a.kind() === 'class_definition') ?? null,
-  getParentClass: (classNode) => {
-    const argList =
-      classNode.field('superclasses') ||
-      classNode.children().find((c: SgNode) => c.kind() === 'argument_list');
-    return argList?.children().find((c: SgNode) => c.kind() === 'identifier')?.text();
-  },
-};
-
-/**
- * Extract raw call sites from a Python AST.
- * Detects self.X() and super().X() to preserve class resolution context.
- */
-export function extractCallsFromPython(root: SgRoot, fp: string, calls: RawCallSite[]): void {
-  extractCalls(root.root(), fp, PYTHON_CALL_CONFIG, calls);
-}

package/src/parser/extractors/ruby.ts

@@ -1,147 +0,0 @@
-import type { SgNode, SgRoot } from '@ast-grep/napi';
-import type { RawCallSite, RawGraph } from '../../graph/types';
-import { type CallExtractionConfig, extractCalls } from '../../shared/extract-calls';
-import { computeContentHash } from '../../shared/file-hash';
-import { log } from '../../shared/logger';
-import { LANG_KINDS } from '../languages';
-
-export function extractRuby(root: SgRoot, fp: string, seen: Set<string>, graph: RawGraph): void {
-  const kinds = LANG_KINDS.ruby;
-  const rootNode = root.root();
-
-  // ── Classes ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.class } })) {
-    const name = node.field('name')?.text();
-    if (!name || seen.has(`c:${fp}:${name}`)) continue;
-    seen.add(`c:${fp}:${name}`);
-
-    const superclass = node.field('superclass')?.text() || '';
-    graph.classes.push({
-      name,
-      file: fp,
-      line_start: node.range().start.line,
-      line_end: node.range().end.line,
-      extends: superclass,
-      implements: '',
-      qualified: `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Modules ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.module } })) {
-    const name = node.field('name')?.text();
-    if (!name || seen.has(`c:${fp}:${name}`)) continue;
-    seen.add(`c:${fp}:${name}`);
-    graph.classes.push({
-      name,
-      file: fp,
-      line_start: node.range().start.line,
-      line_end: node.range().end.line,
-      extends: '',
-      implements: '',
-      qualified: `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Methods ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.method } })) {
-    const name = node.field('name')?.text();
-    if (!name) continue;
-    const line = node.range().start.line;
-    if (seen.has(`m:${fp}:${name}:${line}`)) continue;
-    seen.add(`m:${fp}:${name}:${line}`);
-
-    const classAncestor = node.ancestors().find((a: SgNode) => a.kind() === kinds.class || a.kind() === kinds.module);
-    const className = classAncestor?.field('name')?.text() || '';
-
-    graph.functions.push({
-      name,
-      file: fp,
-      line_start: line,
-      line_end: node.range().end.line,
-      params: node.field('parameters')?.text() || '()',
-      returnType: '',
-      kind: className ? 'Method' : 'Function',
-      className,
-      qualified: className ? `${fp}::${className}.${name}` : `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Tests (RSpec: describe/it/context) ──
-  for (const p of [
-    "describe '$NAME' do $$$BODY end",
-    'describe "$NAME" do $$$BODY end',
-    "it '$NAME' do $$$BODY end",
-    'it "$NAME" do $$$BODY end',
-    "context '$NAME' do $$$BODY end",
-    'context "$NAME" do $$$BODY end',
-  ]) {
-    try {
-      for (const m of rootNode.findAll(p)) {
-        const name = m.getMatch('NAME')?.text();
-        if (!name) continue;
-        const key = `t:${fp}:${name}:${m.range().start.line}`;
-        if (seen.has(key)) continue;
-        seen.add(key);
-        graph.tests.push({
-          name,
-          file: fp,
-          line_start: m.range().start.line,
-          line_end: m.range().end.line,
-          qualified: `${fp}::test:${name}`,
-          content_hash: computeContentHash(m.text()),
-        });
-      }
-    } catch (err) {
-      log.debug('Ruby pattern mismatch', { file: fp, pattern: p, error: String(err) });
-    }
-  }
-
-  // ── Imports (require/require_relative) ──
-  for (const p of [
-    "require '$MODULE'",
-    'require "$MODULE"',
-    "require_relative '$MODULE'",
-    'require_relative "$MODULE"',
-  ]) {
-    try {
-      for (const m of rootNode.findAll(p)) {
-        const mod = m.getMatch('MODULE')?.text();
-        if (mod) {
-          graph.imports.push({
-            module: mod,
-            file: fp,
-            line: m.range().start.line,
-            names: [],
-            lang: 'ruby',
-          });
-        }
-      }
-    } catch (err) {
-      log.debug('Ruby pattern mismatch', { file: fp, pattern: p, error: String(err) });
-    }
-  }
-}
-
-/** Ruby-specific call extraction config for shared extractCalls(). */
-function createRubyCallConfig(): CallExtractionConfig {
-  const kinds = LANG_KINDS.ruby;
-  return {
-    selfPrefixes: ['self.'],
-    superPrefixes: ['super'],
-    findEnclosingClass: (node) =>
-      node.ancestors().find((a: SgNode) => a.kind() === kinds.class || a.kind() === kinds.module) ?? null,
-    getParentClass: (classNode) => classNode.field('superclass')?.text(),
-  };
-}
-
-/**
- * Extract raw call sites from a Ruby AST.
- * Detects self.X() and super() to preserve class resolution context.
- */
-export function extractCallsFromRuby(root: SgRoot, fp: string, calls: RawCallSite[]): void {
-  extractCalls(root.root(), fp, createRubyCallConfig(), calls);
-}

package/src/parser/extractors/typescript.ts

@@ -1,350 +0,0 @@
-import type { SgNode, SgRoot } from '@ast-grep/napi';
-import { Lang } from '@ast-grep/napi';
-import type { RawCallSite, RawGraph } from '../../graph/types';
-import { type CallExtractionConfig, extractCalls } from '../../shared/extract-calls';
-import { computeContentHash } from '../../shared/file-hash';
-import { NOISE } from '../../shared/filters';
-import { LANG_KINDS } from '../languages';
-
-export function extractTypeScript(
-  root: SgRoot,
-  fp: string,
-  seen: Set<string>,
-  graph: RawGraph,
-  lang: Lang | string = Lang.TypeScript,
-): void {
-  const kinds = LANG_KINDS.typescript;
-  const rootNode = root.root();
-  const isTS = lang === Lang.TypeScript || lang === Lang.Tsx;
-
-  // ── Classes ──
-  const classKinds = isTS ? [kinds.class, kinds.abstractClass] : [kinds.class];
-  for (const kind of classKinds) {
-    for (const node of rootNode.findAll({ rule: { kind } })) {
-      const name = node.field('name')?.text();
-      if (!name || seen.has(`c:${fp}:${name}`)) continue;
-      seen.add(`c:${fp}:${name}`);
-
-      let extendsName = '';
-      let implementsName = '';
-      const heritage = node.children().find((c: SgNode) => c.kind() === 'class_heritage');
-      if (heritage) {
-        const ext = heritage.children().find((c: SgNode) => c.kind() === 'extends_clause');
-        extendsName =
-          ext
-            ?.children()
-            .find(
-              (c: SgNode) =>
-                c.kind() === 'identifier' || c.kind() === 'type_identifier' || c.kind() === 'member_expression',
-            )
-            ?.text() || '';
-        const impl = heritage.children().find((c: SgNode) => c.kind() === 'implements_clause');
-        implementsName =
-          impl
-            ?.children()
-            .find((c: SgNode) => c.kind() === 'type_identifier' || c.kind() === 'identifier')
-            ?.text() || '';
-      }
-
-      graph.classes.push({
-        name,
-        file: fp,
-        line_start: node.range().start.line,
-        line_end: node.range().end.line,
-        extends: extendsName,
-        implements: implementsName,
-        qualified: `${fp}::${name}`,
-        content_hash: computeContentHash(node.text()),
-      });
-    }
-  }
-
-  // ── Methods (kind-based: catches constructor, async, getters/setters) ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.method } })) {
-    const name = node.field('name')?.text();
-    if (!name) continue;
-    const line = node.range().start.line;
-    if (seen.has(`m:${fp}:${name}:${line}`)) continue;
-    seen.add(`m:${fp}:${name}:${line}`);
-
-    const classAncestor = node
-      .ancestors()
-      .find((a: SgNode) => a.kind() === kinds.class || (isTS && a.kind() === kinds.abstractClass));
-    const className = classAncestor?.field('name')?.text() || '';
-    const params = node.field('parameters');
-    const retType = node.field('return_type')?.text()?.replace(/^:\s*/, '') || '';
-
-    if (name === 'constructor' && className) {
-      // Constructor DI extraction
-      const fieldTypeMap = new Map<string, string>();
-      if (params) {
-        for (const p of params.children()) {
-          if (p.kind() !== 'required_parameter') continue;
-          if (!p.children().some((c: SgNode) => c.kind() === 'accessibility_modifier')) continue;
-          const ident = p.children().find((c: SgNode) => c.kind() === 'identifier');
-          const typeAnn = p.children().find((c: SgNode) => c.kind() === 'type_annotation');
-          if (ident && typeAnn) {
-            const typeNode = typeAnn
-              .children()
-              .find(
-                (c: SgNode) =>
-                  c.kind() === 'type_identifier' || c.kind() === 'identifier' || c.kind() === 'generic_type',
-              );
-            if (typeNode) {
-              const typeName =
-                typeNode.kind() === 'generic_type'
-                  ? typeNode
-                      .children()
-                      .find((c: SgNode) => c.kind() === 'type_identifier')
-                      ?.text() || typeNode.text()
-                  : typeNode.text();
-              fieldTypeMap.set(ident.text(), typeName);
-            }
-          }
-        }
-      }
-      if (fieldTypeMap.size > 0) graph.diMaps.set(fp, fieldTypeMap);
-
-      graph.functions.push({
-        name: `${className}.constructor`,
-        file: fp,
-        line_start: line,
-        line_end: node.range().end.line,
-        params: params?.text() || '()',
-        returnType: '',
-        kind: 'Constructor',
-        className,
-        qualified: `${fp}::${className}.constructor`,
-        content_hash: computeContentHash(node.text()),
-      });
-    } else {
-      graph.functions.push({
-        name,
-        file: fp,
-        line_start: line,
-        line_end: node.range().end.line,
-        params: params?.text() || '()',
-        returnType: retType,
-        kind: className ? 'Method' : 'Function',
-        className,
-        qualified: className ? `${fp}::${className}.${name}` : `${fp}::${name}`,
-        content_hash: computeContentHash(node.text()),
-      });
-    }
-  }
-
-  // ── Standalone functions ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.function } })) {
-    const name = node.field('name')?.text();
-    if (!name) continue;
-    const line = node.range().start.line;
-    if (seen.has(`f:${fp}:${name}:${line}`)) continue;
-    if (node.ancestors().some((a: SgNode) => a.kind() === kinds.class || (isTS && a.kind() === kinds.abstractClass)))
-      continue;
-    seen.add(`f:${fp}:${name}:${line}`);
-
-    graph.functions.push({
-      name,
-      file: fp,
-      line_start: line,
-      line_end: node.range().end.line,
-      params: node.field('parameters')?.text() || '()',
-      returnType: node.field('return_type')?.text()?.replace(/^:\s*/, '') || '',
-      kind: 'Function',
-      className: '',
-      qualified: `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Arrow functions ──
-  for (const node of rootNode.findAll({
-    rule: { kind: kinds.arrowContainer, has: { kind: kinds.arrowFunction } },
-  })) {
-    const name = node.field('name')?.text();
-    if (!name) continue;
-    const line = node.range().start.line;
-    if (seen.has(`f:${fp}:${name}:${line}`)) continue;
-    seen.add(`f:${fp}:${name}:${line}`);
-
-    const arrow = node.children().find((c: SgNode) => c.kind() === kinds.arrowFunction);
-    graph.functions.push({
-      name,
-      file: fp,
-      line_start: line,
-      line_end: node.range().end.line,
-      params: arrow?.field('parameters')?.text() || '()',
-      returnType: arrow?.field('return_type')?.text()?.replace(/^:\s*/, '') || '',
-      kind: 'Function',
-      className: '',
-      qualified: `${fp}::${name}`,
-      content_hash: computeContentHash(node.text()),
-    });
-  }
-
-  // ── Interfaces (TS only — JS grammar has no interface_declaration) ──
-  if (isTS)
-    for (const node of rootNode.findAll({ rule: { kind: kinds.interface } })) {
-      const name = node.field('name')?.text();
-      if (!name || seen.has(`i:${fp}:${name}`)) continue;
-      seen.add(`i:${fp}:${name}`);
-
-      const methods: string[] = [];
-      const body = node.field('body');
-      if (body) {
-        for (const child of body.findAll({ rule: { kind: kinds.methodSignature } })) {
-          const mn = child.field('name')?.text();
-          if (mn) methods.push(mn);
-        }
-      }
-
-      graph.interfaces.push({
-        name,
-        file: fp,
-        line_start: node.range().start.line,
-        line_end: node.range().end.line,
-        methods,
-        qualified: `${fp}::${name}`,
-        content_hash: computeContentHash(node.text()),
-      });
-    }
-
-  // ── Enums (TS only — JS grammar has no enum_declaration) ──
-  if (isTS)
-    for (const node of rootNode.findAll({ rule: { kind: kinds.enum } })) {
-      const name = node.field('name')?.text();
-      if (!name || seen.has(`e:${fp}:${name}`)) continue;
-      seen.add(`e:${fp}:${name}`);
-      graph.enums.push({
-        name,
-        file: fp,
-        line_start: node.range().start.line,
-        line_end: node.range().end.line,
-        qualified: `${fp}::${name}`,
-        content_hash: computeContentHash(node.text()),
-      });
-    }
-
-  // ── Imports ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.import } })) {
-    const sourceNode = node.children().find((c: SgNode) => c.kind() === 'string');
-    const frag = sourceNode?.children().find((c: SgNode) => c.kind() === 'string_fragment');
-    const modulePath = frag?.text() || sourceNode?.text()?.replace(/['"]/g, '') || '';
-    if (!modulePath) continue;
-
-    const names: string[] = [];
-    const importClause = node.children().find((c: SgNode) => c.kind() === 'import_clause');
-    if (importClause) {
-      for (const child of importClause.children()) {
-        if (child.kind() === 'identifier') {
-          names.push(child.text());
-        } else if (child.kind() === 'named_imports') {
-          for (const spec of child.findAll({ rule: { kind: 'import_specifier' } })) {
-            const n =
-              spec.field('name')?.text() ||
-              spec
-                .children()
-                .find((c: SgNode) => c.kind() === 'identifier')
-                ?.text();
-            if (n) names.push(n);
-          }
-        } else if (child.kind() === 'namespace_import') {
-          const alias = child.children().find((c: SgNode) => c.kind() === 'identifier');
-          if (alias) names.push(alias.text());
-        }
-      }
-    }
-    graph.imports.push({
-      module: modulePath,
-      file: fp,
-      line: node.range().start.line,
-      names,
-      lang: 'ts',
-    });
-  }
-
-  // ── Re-exports ──
-  for (const node of rootNode.findAll({ rule: { kind: kinds.export } })) {
-    const src = node.children().find((c: SgNode) => c.kind() === 'string');
-    if (src) {
-      const frag = src.children().find((c: SgNode) => c.kind() === 'string_fragment');
-      graph.reExports.push({
-        module: frag?.text() || src.text().replace(/['"]/g, ''),
-        file: fp,
-        line: node.range().start.line,
-      });
-    }
-  }
-
-  // ── Tests (pattern-based) ──
-  for (const p of [
-    'describe("$NAME", $$$BODY)',
-    "describe('$NAME', $$$BODY)",
-    'it("$NAME", $$$BODY)',
-    "it('$NAME', $$$BODY)",
-    'test("$NAME", $$$BODY)',
-    "test('$NAME', $$$BODY)",
-  ]) {
-    for (const m of rootNode.findAll(p)) {
-      const name = m.getMatch('NAME')?.text();
-      if (!name) continue;
-      const key = `t:${fp}:${name}:${m.range().start.line}`;
-      if (seen.has(key)) continue;
-      seen.add(key);
-      graph.tests.push({
-        name,
-        file: fp,
-        line_start: m.range().start.line,
-        line_end: m.range().end.line,
-        qualified: `${fp}::test:${name}`,
-        content_hash: computeContentHash(m.text()),
-      });
-    }
-  }
-}
-
-/** TypeScript-specific call extraction config for shared extractCalls(). */
-const TS_CALL_CONFIG: CallExtractionConfig = {
-  selfPrefixes: ['this.'],
-  superPrefixes: ['super.'],
-  findEnclosingClass: (node) => {
-    const kinds = LANG_KINDS.typescript;
-    return node.ancestors().find(
-      (a: SgNode) => a.kind() === kinds.class || a.kind() === kinds.abstractClass,
-    ) ?? null;
-  },
-  getParentClass: (classNode) => {
-    const heritage = classNode.children().find((c: SgNode) => c.kind() === 'class_heritage');
-    const ext = heritage?.children().find((c: SgNode) => c.kind() === 'extends_clause');
-    return ext?.children().find(
-      (c: SgNode) => c.kind() === 'identifier' || c.kind() === 'type_identifier' || c.kind() === 'member_expression',
-    )?.text();
-  },
-  // Skip this.field.method — already handled by the DI pattern above
-  skipCallee: (callee) => callee.startsWith('this.') && callee.substring(5).includes('.'),
-};
-
-/**
- * Extract raw call sites from a TypeScript/JavaScript AST.
- * Finds DI calls (this.field.method) and direct calls ($CALLEE($$$ARGS)).
- * Filters NOISE. Does NOT resolve — just collects raw sites.
- */
-export function extractCallsFromTypeScript(root: SgRoot, fp: string, calls: RawCallSite[]): void {
-  const rootNode = root.root();
-
-  // DI pattern: this.$FIELD.$METHOD($$$ARGS)
-  for (const m of rootNode.findAll('this.$FIELD.$METHOD($$$ARGS)')) {
-    const field = m.getMatch('FIELD')?.text();
-    const method = m.getMatch('METHOD')?.text();
-    if (!method || NOISE.has(method)) continue;
-    calls.push({
-      source: fp,
-      callName: method,
-      line: m.range().start.line,
-      diField: field,
-    });
-  }
-
-  // Direct calls + self/super detection via shared function
-  extractCalls(rootNode, fp, TS_CALL_CONFIG, calls);
-}