Function main

scripts/agent-eval/probe-allocation.mjs:263–330

()

}

async function main() {
  const spec = JSON.parse(readFileSync(SPEC_PATH, 'utf-8'));
  const fixtures = spec.fixtures.filter((f) => wanted.length === 0 || wanted.includes(f.id));
  if (fixtures.length === 0) {
    console.error(`no fixture matched ${JSON.stringify(wanted)}; known: ${spec.fixtures.map((f) => f.id).join(', ')}`);
    process.exit(2);
  }

  const dist = await loadDist();
  const sidecarDir = mkdtempSync(join(tmpdir(), 'cg-alloc-diag-'));
  const results = [];
  const temps = [];

  for (const fixture of fixtures) {
    let repoPath;
    if (fixture.kind === 'fixture') {
      repoPath = await materializeFixture(dist, fixture.path);
      temps.push(repoPath);
    } else {
      repoPath = resolve(REPO_ROOT, fixture.path);
      if (!existsSync(join(repoPath, '.codegraph'))) {
        console.error(`${fixture.id}: ${repoPath} has no .codegraph index — run \`codegraph init\` there first.`);
        process.exit(2);
      }
    }

    const sidecar = join(sidecarDir, `${fixture.id}.jsonl`);
    mkdirSync(dirname(sidecar), { recursive: true });
    const { report, text } = await runExplore(dist, repoPath, fixture.query, sidecar);
    const evaluated = evaluate(fixture, report, text);
    printReport(fixture, report, evaluated);
    results.push({
      id: fixture.id,
      kind: fixture.kind,
      query: fixture.query,
      passed: evaluated.checks.every((c) => c.pass),
      checks: evaluated.checks,
      shares: {
        delivered: Object.fromEntries(evaluated.delivered),
        allocated: Object.fromEntries(evaluated.allocated),
      },
      envelope: report.envelope,
      files: report.files.filter((f) => f.emittedChars > 0 || f.finalChars > 0),
    });
  }

  if (!keepTemp) {
    for (const dir of temps) rmSync(dir, { recursive: true, force: true });
    rmSync(sidecarDir, { recursive: true, force: true });
  } else {
    say('');
    say(`   kept: ${[...temps, sidecarDir].join(' ')}`);
  }

  const allPassed = results.every((r) => r.passed);
  if (asJson) {
    console.log(JSON.stringify({ passed: allPassed, fixtures: results }, null, 2));
  } else {