GitNexus/gitnexus/bench/scope-capture/measure.mjs

/**
 * Unified build-free scope-capture measurement harness for every currently
 * benchmarked language (the ones with a `*-pipeline-benchmark.test.ts`):
 * go, csharp, rust, php, ruby, cobol, swift — plus python lives in its own
 * `bench/python-scope/` harness (richer: it also covers import resolution).
 *
 * For each language it:
 *   - times `emit<Lang>ScopeCaptures` on a synthetic DAO-style source at two
 *     sizes (250 / 800 top-level entities), reporting elapsed_ms + a scaling
 *     ratio `(t_large/t_small)/(800/250)`: ~1.0 is linear, ~3.2 is quadratic
 *     (the O(matches × rootChildren) shape #1848 hit in Go);
 *   - computes an order-independent sha256 fingerprint over the whole
 *     `lang-resolution/<lang>-*` fixture corpus + a fixed 20-entity synthetic
 *     source, as the correctness gate.
 *
 * Build-free: imports the `.ts` hotpaths through tsx
 * (`node --import tsx bench/scope-capture/measure.mjs`). Static `.ts` imports
 * work; a top-level `await import()` breaks tsx's lexer.
 *
 * Without args: prints one JSON object per language.
 * With `--check`: asserts each language's fingerprint == its committed baseline
 * (baselines.json) AND scaling_ratio < that language's recorded budget; exits
 * non-zero on any drift/regression.
 */
import fs from 'node:fs';
import path from 'node:path';
import crypto from 'node:crypto';
import { fileURLToPath } from 'node:url';

import { emitGoScopeCaptures } from '../../src/core/ingestion/languages/go/index.ts';
import { emitCsharpScopeCaptures } from '../../src/core/ingestion/languages/csharp/index.ts';
import { emitRustScopeCaptures } from '../../src/core/ingestion/languages/rust/index.ts';
import { emitPhpScopeCaptures } from '../../src/core/ingestion/languages/php/index.ts';
import { emitRubyScopeCaptures } from '../../src/core/ingestion/languages/ruby/index.ts';
import { emitCobolScopeCaptures } from '../../src/core/ingestion/languages/cobol/index.ts';
import { emitSwiftScopeCaptures } from '../../src/core/ingestion/languages/swift/index.ts';
import { emitTsScopeCaptures } from '../../src/core/ingestion/languages/typescript/index.ts';
import { emitJsScopeCaptures } from '../../src/core/ingestion/languages/javascript/index.ts';
import { emitKotlinScopeCaptures } from '../../src/core/ingestion/languages/kotlin/index.ts';
import { emitJavaScopeCaptures } from '../../src/core/ingestion/languages/java/index.ts';
import { emitCScopeCaptures } from '../../src/core/ingestion/languages/c/index.ts';
import { emitCppScopeCaptures } from '../../src/core/ingestion/languages/cpp/index.ts';
import { emitDartScopeCaptures } from '../../src/core/ingestion/languages/dart/index.ts';

const __dirname = path.dirname(fileURLToPath(import.meta.url));
const FIXTURE_ROOT = path.resolve(__dirname, '..', '..', 'test', 'fixtures', 'lang-resolution');
const BASELINE_PATH = path.resolve(__dirname, 'baselines.json');

// ---- correctness fingerprint (order-independent; mirrors python harness) ----

function canonicalizeMatch(match) {
  const parts = [];
  for (const tag of Object.keys(match)) {
    const cap = match[tag];
    if (cap === undefined || cap === null || cap.range === undefined) {
      parts.push(`${tag}|<no-range>`);
      continue;
    }
    const r = cap.range;
    parts.push(`${tag}|${cap.text}|${r.startLine}:${r.startCol}-${r.endLine}:${r.endCol}`);
  }
  parts.sort();
  return parts.join(';');
}

function digestCaptures(matches) {
  return crypto
    .createHash('sha256')
    .update(matches.map(canonicalizeMatch).sort().join('\n'))
    .digest('hex');
}

/** All fixture files for a language, sorted by repo-relative key. */
function collectFixtures(prefix, exts) {
  const out = [];
  for (const entry of fs.readdirSync(FIXTURE_ROOT, { withFileTypes: true })) {
    if (!entry.isDirectory() || !entry.name.startsWith(`${prefix}-`)) continue;
    const stack = [path.join(FIXTURE_ROOT, entry.name)];
    while (stack.length) {
      const dir = stack.pop();
      for (const c of fs.readdirSync(dir, { withFileTypes: true })) {
        const p = path.join(dir, c.name);
        if (c.isDirectory()) stack.push(p);
        else if (exts.some((e) => c.name.endsWith(e))) {
          out.push({ key: path.relative(FIXTURE_ROOT, p).split(path.sep).join('/'), absPath: p });
        }
      }
    }
  }
  out.sort((a, b) => a.key.localeCompare(b.key));
  return out;
}

// ---- per-language config: synthetic DAO generators + fixture globs ----

const LANGS = [
  {
    name: 'go',
    emit: emitGoScopeCaptures,
    fixturePrefix: 'go',
    exts: ['.go'],
    file: 'bench.go',
    // Heritage-bearing: each Entity embeds Base (Go inheritance = struct
    // embedding) so the @reference.inherits synth (#1951) is driven at scale.
    header:
      'package generated\n\ntype Base struct{}\n\nfunc (b *Base) BaseMethod() string { return "base" }\n\n',
    unit: (n) =>
      `type Entity${n} struct {\n\tBase\n\tid int64\n\tname string\n}\n\n` +
      `func (e *Entity${n}) GetID() int64 { return e.id }\n` +
      `func (e *Entity${n}) SetName(v string) { e.name = v }\n\n`,
  },
  {
    name: 'csharp',
    emit: emitCsharpScopeCaptures,
    fixturePrefix: 'csharp',
    exts: ['.cs'],
    file: 'bench.cs',
    // Heritage-bearing: extends Base + implements IEntity (both forms) so the
    // @reference.inherits synth (#1951) is driven at scale, not just the base loop.
    header:
      'namespace Generated;\n\npublic class Base { }\n\npublic interface IEntity {\n  long GetId();\n}\n\n',
    unit: (n) =>
      `public class Entity${n} : Base, IEntity {\n` +
      `  public long Id;\n  public string Name;\n` +
      `  public long GetId() { return Id; }\n` +
      `  public void SetName(string v) { Name = v; }\n}\n\n`,
  },
  {
    name: 'rust',
    emit: emitRustScopeCaptures,
    fixturePrefix: 'rust',
    exts: ['.rs'],
    file: 'bench.rs',
    // Heritage-bearing: `impl Shape for Entity_n` (Rust inheritance lives on
    // impl_item) so the @reference.inherits trait-impl synth (#1951) is driven
    // at scale. The two methods move into the trait impl to keep unit size flat.
    header: 'trait Shape {\n  fn area(&self) -> i64;\n  fn name(&self) -> String;\n}\n\n',
    unit: (n) =>
      `struct Entity${n} {\n  id: i64,\n  name: String,\n}\n\n` +
      `impl Shape for Entity${n} {\n` +
      `  fn area(&self) -> i64 { self.id }\n` +
      `  fn name(&self) -> String { self.name.clone() }\n}\n\n`,
  },
  {
    name: 'php',
    emit: emitPhpScopeCaptures,
    fixturePrefix: 'php',
    exts: ['.php'],
    file: 'bench.php',
    // Heritage-bearing: extends Base + uses a trait (both forms) so the
    // @reference.inherits synth (#1951) is driven at scale.
    header:
      '<?php\n\nclass Base {}\n\ntrait Auditable {\n  public function audit() { return true; }\n}\n\n',
    unit: (n) =>
      `class Entity${n} extends Base {\n` +
      `  use Auditable;\n` +
      `  public $id;\n  public $name;\n` +
      `  function getId() { return $this->id; }\n` +
      `  function setName($v) { $this->name = $v; }\n}\n\n`,
  },
  {
    name: 'ruby',
    emit: emitRubyScopeCaptures,
    fixturePrefix: 'ruby',
    exts: ['.rb'],
    file: 'bench.rb',
    // Heritage-bearing: `< Base` superclass + `include Trackable` mixin (both
    // forms) so the @reference.inherits synth (#1951) is driven at scale.
    header:
      'class Base\n  def base_id\n    @id\n  end\nend\n\nmodule Trackable\n  def track\n    @tracked = true\n  end\nend\n\n',
    unit: (n) =>
      `class Entity${n} < Base\n` +
      `  include Trackable\n` +
      `  def get_id\n    @id\n  end\n` +
      `  def set_name(v)\n    @name = v\n  end\nend\n\n`,
  },
  {
    name: 'cobol',
    emit: emitCobolScopeCaptures,
    fixturePrefix: 'cobol',
    exts: ['.cbl', '.cpy'],
    file: 'bench.cbl',
    header:
      '       IDENTIFICATION DIVISION.\n' +
      '       PROGRAM-ID. BENCH.\n' +
      '       PROCEDURE DIVISION.\n',
    unit: (n) => `       PARA-${String(n).padStart(5, '0')}.\n           DISPLAY "P${n}".\n`,
  },
  {
    name: 'c',
    emit: emitCScopeCaptures,
    fixturePrefix: 'c',
    exts: ['.c', '.h'],
    file: 'bench.c',
    // C has no inheritance construct — flat scale source. Added (was unbenched);
    // adding it exposed + fixed the same O(n²) findNodeAtRange root-walk (#1956).
    header: '#include <stdint.h>\n#include <stddef.h>\n\ntypedef int64_t id_t;\n\n',
    unit: (n) =>
      `typedef struct Entity${n} {\n  id_t id;\n  const char *name;\n} Entity${n};\n\n` +
      `id_t entity_${n}_get_id(Entity${n} *e) { return e->id; }\n` +
      `void entity_${n}_set_name(Entity${n} *e, const char *v) { e->name = v; }\n\n`,
  },
  {
    name: 'cpp',
    emit: emitCppScopeCaptures,
    fixturePrefix: 'cpp',
    exts: ['.cpp', '.cc', '.cxx', '.hpp', '.h'],
    file: 'bench.cpp',
    // Heritage-bearing: `: public Base, public Mixin` (single + multiple
    // inheritance) drives emitCppInheritanceCaptures (#1951) at scale. Added
    // (was unbenched); adding it exposed + fixed the same O(n²) root-walk (#1956).
    //
    // Also GENERIC-MEMBER-bearing (#2833): `Repo<Entity_n> repo;` is a member
    // whose declared type is a bare `template_type`, and
    // `std::vector<Entity_n> items;` is the far commoner spelling where a
    // `qualified_identifier` WRAPS that template_type. Both were absent, and
    // their absence is why two successive rounds of `field_declaration`
    // type-binding rules landed with a byte-identical cpp fingerprint: the gate
    // could not see a member field it had no instance of. With them present,
    // reverting either round of rules drifts the fingerprint, which is the
    // property that makes the gate worth running.
    header:
      '#include <string>\n#include <vector>\n\ntemplate <typename T>\nclass Repo {\n public:\n  void save(T v) {}\n};\n\nclass Base {\n public:\n  long baseId() const { return 0; }\n};\n\nclass Mixin {\n public:\n  void mix() {}\n};\n\n',
    unit: (n) =>
      `class Entity${n} : public Base, public Mixin {\n public:\n  long id;\n  std::string name;\n` +
      `  Repo<Entity${n}> repo;\n` +
      `  std::vector<Entity${n}> items;\n` +
      `  long getId() const { return id; }\n` +
      `  void setName(std::string v) { name = v; }\n};\n\n`,
  },
  {
    name: 'swift',
    emit: emitSwiftScopeCaptures,
    fixturePrefix: 'swift',
    exts: ['.swift'],
    file: 'bench.swift',
    // Heritage-bearing: inherits Base + conforms to Serviceable (both forms) so
    // the @reference.inherits synth (#1951) is driven at scale.
    header:
      'class Base {\n  func ping() -> String { return "base" }\n}\n\nprotocol Serviceable {\n  func serve() -> String\n}\n\n',
    unit: (n) =>
      `class Entity${n}: Base, Serviceable {\n` +
      `  var id: Int64 = 0\n  var name: String = ""\n` +
      `  func getId() -> Int64 { return self.id }\n` +
      `  func serve() -> String { return self.name }\n}\n\n`,
  },
  {
    name: 'dart',
    emit: emitDartScopeCaptures,
    fixturePrefix: 'dart',
    exts: ['.dart'],
    file: 'bench.dart',
    // Heritage-bearing: `extends Base` (generic @reference.inherits pre-pass)
    // + `implements Marker` (Dart `implements <class>` → IMPLEMENTS marker) so
    // both heritage paths and the postfix-chain reference walk run at scale.
    header:
      'class Base {\n  String ping() { return "base"; }\n}\n\nabstract class Marker {\n  String mark();\n}\n\n',
    unit: (n) =>
      `class Entity${n} extends Base implements Marker {\n` +
      `  int id = 0;\n  String name = '';\n` +
      `  int getId() { return this.id; }\n` +
      `  String mark() { return this.name; }\n}\n\n`,
  },
  {
    name: 'java',
    emit: emitJavaScopeCaptures,
    fixturePrefix: 'java',
    exts: ['.java'],
    file: 'bench.java',
    // Java was previously unbenched. Class and record heritage both implement
    // Marker so the @reference.inherits synth (#1951, #2900) is driven at scale.
    header: 'package generated;\n\nclass Base {}\n\ninterface Marker {}\n\n',
    unit: (n) =>
      `class Entity${n} extends Base implements Marker {\n` +
      `  long id = 0L;\n  String name = "";\n` +
      `  public long getId() { return this.id; }\n` +
      `  public void setName(String v) { this.name = v; }\n}\n\n` +
      `record RecordEntity${n}(long id) implements Marker {}\n\n`,
  },
  {
    name: 'java-local-types',
    emit: emitJavaScopeCaptures,
    fixturePrefix: 'java-local',
    exts: ['.java'],
    file: 'bench-local.java',
    header:
      'package generated;\n\nclass Base {}\n\ninterface Marker {}\n\nclass Bench {\n  void run() {\n',
    // Co-scale both independent ordinal sequences under one host; construction
    // and dispatch keep lexical-alias captures hot. The old per-identity
    // host-candidate filter made this combined workload quadratic.
    unit: (n) =>
      `    { class Local extends Base implements Marker { long value() { return ${n}L; } } ` +
      `new Local().value(); }\n` +
      `    Marker marker${n} = new Marker() {};\n`,
    footer: '  }\n}\n',
  },
  {
    name: 'typescript',
    emit: emitTsScopeCaptures,
    fixturePrefix: 'typescript',
    exts: ['.ts', '.tsx'],
    file: 'bench.ts',
    // Inheritance-bearing units so the @reference.inherits synth pass (#1951)
    // is exercised at scale, not just the base capture loop.
    header: 'class Base {}\n\n',
    unit: (n) =>
      `class Entity${n} extends Base {\n` +
      `  id: number = 0;\n  name: string = '';\n` +
      `  getId(): number { return this.id; }\n` +
      `  setName(v: string): void { this.name = v; }\n}\n\n`,
  },
  {
    // One `a || b || …` chain whose operand count is the entity count (#3354
    // review): each operand is one more nesting level, so this guards the
    // callable-flow value-alternative expansion against per-leaf work that
    // grows with chain depth (ratio ~8 before the fix, and 8000 operands
    // overflowed the stack). No fixture corpus — the fingerprint is the synthetic
    // source alone; `typescript` above already covers the TS fixtures.
    name: 'typescript-deep-chain',
    emit: emitTsScopeCaptures,
    exts: ['.ts'],
    file: 'bench-chain.ts',
    header: 'function handler() {}\n\nexport function isKw(w: string) {\n  const f = handler',
    unit: (n) => ` || w === "k${n}"`,
    footer: ';\n  return f;\n}\n',
  },
  {
    name: 'javascript',
    emit: emitJsScopeCaptures,
    fixturePrefix: 'javascript',
    exts: ['.js', '.jsx', '.mjs', '.cjs'],
    file: 'bench.js',
    header: 'class Base {}\n\n',
    unit: (n) =>
      `class Entity${n} extends Base {\n` +
      `  getId() { return this.id; }\n` +
      `  setName(v) { this.name = v; }\n}\n\n`,
  },
  {
    name: 'kotlin',
    emit: emitKotlinScopeCaptures,
    fixturePrefix: 'kotlin',
    exts: ['.kt', '.kts'],
    file: 'bench.kt',
    header: 'open class Base\n\n',
    unit: (n) =>
      `class Entity${n} : Base() {\n` +
      `  var id: Long = 0\n  var name: String = ""\n` +
      `  fun getId(): Long { return id }\n` +
      `  fun setName(v: String) { name = v }\n}\n\n`,
  },
];

function generate(lang, entityCount) {
  let src = lang.header;
  for (let i = 0; i < entityCount; i++) src += lang.unit(i);
  return src + (lang.footer ?? '');
}

// ---- timing ----

function median(xs) {
  const s = [...xs].sort((a, b) => a - b);
  const m = Math.floor(s.length / 2);
  return s.length % 2 ? s[m] : (s[m - 1] + s[m]) / 2;
}

function timeEmit(emit, src, file, reps) {
  emit(src, `warmup-${file}`); // warm parser/query JIT (not counted)
  const samples = [];
  let count = 0;
  for (let i = 0; i < reps; i++) {
    const start = process.hrtime.bigint();
    const out = emit(src, file);
    samples.push(Number(process.hrtime.bigint() - start) / 1e6);
    count = out.length;
  }
  return { ms: median(samples), count };
}

const SMALL = 250;
const LARGE = 800;
const REPS = 7;

function measureLang(lang) {
  // Correctness fingerprint over the fixture corpus + a fixed 20-entity source.
  const perFixture = [];
  let groups = 0;
  const fixtures =
    lang.fixturePrefix === undefined ? [] : collectFixtures(lang.fixturePrefix, lang.exts);
  for (const { key, absPath } of fixtures) {
    const matches = lang.emit(fs.readFileSync(absPath, 'utf8'), absPath);
    groups += matches.length;
    perFixture.push(`${key}\t${matches.length}\t${digestCaptures(matches)}`);
  }
  const daoMatches = lang.emit(generate(lang, 20), `synthetic-dao-20${path.extname(lang.file)}`);
  groups += daoMatches.length;
  perFixture.push(`synthetic:dao-20\t${daoMatches.length}\t${digestCaptures(daoMatches)}`);
  const fingerprint = crypto
    .createHash('sha256')
    .update(perFixture.sort().join('\n'))
    .digest('hex');

  // Scaling.
  const small = timeEmit(lang.emit, generate(lang, SMALL), lang.file, REPS);
  const large = timeEmit(lang.emit, generate(lang, LARGE), lang.file, REPS);
  const scalingRatio = small.ms > 0 ? large.ms / small.ms / (LARGE / SMALL) : 0;

  return {
    language: lang.name,
    elapsed_ms_small: Number(small.ms.toFixed(2)),
    elapsed_ms_large: Number(large.ms.toFixed(2)),
    scaling_ratio: Number(scalingRatio.toFixed(3)),
    capture_groups_small: small.count,
    capture_groups_large: large.count,
    fingerprint,
    capture_groups_fp: groups,
    fixture_count: perFixture.length,
  };
}

// ---- run ----

const CHECK = process.argv.includes('--check');
const results = LANGS.map(measureLang);

if (!CHECK) {
  for (const r of results) process.stdout.write(JSON.stringify(r) + '\n');
} else {
  const baselines = JSON.parse(fs.readFileSync(BASELINE_PATH, 'utf8'));
  const failures = [];
  for (const r of results) {
    const base = baselines[r.language];
    if (base === undefined) {
      failures.push(`${r.language}: no baseline recorded`);
      continue;
    }
    if (r.fingerprint !== base.fingerprint) {
      failures.push(
        `${r.language}: capture fingerprint drift (got ${r.fingerprint}, expected ${base.fingerprint})`,
      );
    }
    if (r.scaling_ratio >= base.scaling_budget) {
      failures.push(
        `${r.language}: scaling ratio ${r.scaling_ratio} >= budget ${base.scaling_budget} ` +
          `(${SMALL}->${LARGE} ms ${r.elapsed_ms_small}->${r.elapsed_ms_large})`,
      );
    }
    process.stdout.write(JSON.stringify(r) + '\n');
  }
  if (failures.length > 0) {
    for (const f of failures) process.stderr.write(`[scope-capture --check] FAIL: ${f}\n`);
    process.exit(1);
  }
  process.stderr.write(`[scope-capture --check] PASS (${results.length} languages)\n`);
}