feat: add codegraph tool for querying pre-computed codebase dependency graph

- Implemented `tools-codegraph.ts` to provide various queries on the codebase dependency graph, including impact analysis, dead code detection, module boundaries, and more.
- Created a schema for validating query inputs using Zod.
- Developed helper functions for resolving file paths, formatting output, and handling different query types.
- Added tests for the codegraph queries in `codegraph-queries.test.ts` to ensure functionality and correctness.
- Introduced tests for the codegraph scanning and loading functionalities in `codegraph.test.ts`, verifying the detection of source files, imports, exports, circular dependencies, and entry points.
This commit is contained in:
asepharyana
2026-09-04 22:37:38 +07:00
parent 4dac34b13f
commit bb36af62e2
12 changed files with 1458 additions and 3 deletions
+106
View File
@@ -0,0 +1,106 @@
import { expect, test } from 'bun:test';
import { join } from 'node:path';
import { mkdtempSync, writeFileSync, mkdirSync } from 'node:fs';
import { tmpdir } from 'node:os';
import { scanCodebase, loadGraph, saveGraph, isGraphFresh } from '../src/codegraph';
const SHIRO_ROOT = join(import.meta.dir, '..');
test('scanCodebase detects source files and generates a summary', () => {
const graph = scanCodebase(SHIRO_ROOT);
expect(graph.version).toBe(1);
expect(Object.keys(graph.files).length).toBeGreaterThan(10);
// shiro-neko has src/cli.tsx as a core file
const sourceFiles = Object.keys(graph.files).filter((f) => graph.files[f]?.kind === 'source');
expect(sourceFiles.length).toBeGreaterThan(20);
// Summary is populated
expect(graph.summary).toContain('Codebase Architecture');
expect(graph.summary).toContain('Modules');
expect(graph.summary).toContain('Circular deps');
});
test('scanCodebase extracts imports from source files', () => {
const graph = scanCodebase(SHIRO_ROOT);
// session.ts should import from prompt.ts
const sessionNode = graph.files['src/session.ts'];
expect(sessionNode).toBeDefined();
expect(sessionNode!.imports.length).toBeGreaterThan(0);
expect(sessionNode!.imports).toContain('src/prompt.ts');
});
test('scanCodebase detects exported types and functions', () => {
const graph = scanCodebase(SHIRO_ROOT);
// config.ts should export a Config type
const configNode = graph.files['src/config.ts'];
expect(configNode).toBeDefined();
expect(configNode!.types.length).toBeGreaterThan(0);
expect(configNode!.types).toContain('Config');
});
test('scanCodebase detects circular dependencies', () => {
const graph = scanCodebase(SHIRO_ROOT);
// shiro-neko has a known circular: providers.ts <-> config.ts
expect(graph.circularDeps.length).toBeGreaterThanOrEqual(1);
const hasCircular = graph.circularDeps.some((cycle) =>
cycle.some((f) => f.includes('providers') || f.includes('config')),
);
expect(hasCircular).toBe(true);
});
test('scanCodebase detects entry points', () => {
const graph = scanCodebase(SHIRO_ROOT);
// Should detect cli.tsx as an entry point or at least some
expect(graph.entryPoints.length).toBeGreaterThan(0);
});
test('saveGraph and loadGraph round-trip', () => {
const graph = scanCodebase(SHIRO_ROOT);
const dir = mkdtempSync(join(tmpdir(), 'codegraph-test-'));
saveGraph(dir, graph);
const loaded = loadGraph(dir);
expect(loaded).not.toBeNull();
expect(loaded!.version).toBe(1);
expect(Object.keys(loaded!.files).length).toBe(Object.keys(graph.files).length);
});
test('isGraphFresh returns true for unchanged files', () => {
const graph = scanCodebase(SHIRO_ROOT);
// Graph was just generated from these files, so it should be fresh
expect(isGraphFresh(SHIRO_ROOT, graph)).toBe(true);
});
test('scanCodebase works on a minimal TypeScript project', () => {
const dir = mkdtempSync(join(tmpdir(), 'codegraph-min-'));
// Create a minimal project
writeFileSync(join(dir, 'tsconfig.json'), JSON.stringify({
compilerOptions: { target: 'esnext', module: 'esnext', moduleResolution: 'bundler' },
include: ['src/**/*.ts'],
}));
mkdirSync(join(dir, 'src'), { recursive: true });
writeFileSync(join(dir, 'src/index.ts'), `
import { greet } from './greeting';
export type Config = { name: string };
export function main(): void { greet('world'); }
`);
writeFileSync(join(dir, 'src/greeting.ts'), `
export function greet(name: string): string { return \`Hello \${name}\`; }
`);
const graph = scanCodebase(dir);
expect(Object.keys(graph.files).length).toBe(2);
expect(graph.files['src/index.ts']).toBeDefined();
expect(graph.files['src/greeting.ts']).toBeDefined();
expect(graph.files['src/index.ts']!.imports).toContain('src/greeting.ts');
expect(graph.files['src/index.ts']!.types).toContain('Config');
expect(graph.files['src/index.ts']!.functions).toContain('main');
expect(graph.files['src/greeting.ts']!.functions).toContain('greet');
expect(graph.summary).toContain('Modules');
expect(graph.summary).toContain('src → 2 files');
});