@isac322/pi-codegraph 0.2.0 → 0.3.1

package/dist/lib/types.d.ts

@@ -0,0 +1,69 @@
+export interface CodeGraphSettings {
+    autoSync: boolean;
+    autoGc: boolean;
+    indexStore: string;
+    workerIdleTimeoutMs: number;
+    maxWorkers: number;
+    requestTimeoutMs: number;
+    syncMinIntervalMs: number;
+    maxOutputChars: number;
+    allowedProjectRoots: string[];
+    promptInjection: boolean;
+    codegraphExecutable: string;
+    configFile: string;
+}
+export interface WorkspaceIdentity {
+    sourcePath: string;
+    repoRoot: string;
+    repoIdentity: string;
+    worktreeIdentity: string;
+    gitCommonDir: string;
+}
+export interface WorkspaceStatus {
+    state: "ready" | "uninitialized" | "missing" | string;
+    sourcePath: string;
+    indexPath: string | null;
+    lastSyncAt: string | number | null;
+    identityMatches: boolean;
+    managed?: boolean;
+}
+export interface JsonRpcRequestOptions {
+    signal?: AbortSignal;
+    timeoutMs?: number;
+}
+export interface JsonRpcErrorShape {
+    code?: number | string;
+    message: string;
+}
+export interface JsonRpcMessage {
+    jsonrpc: "2.0";
+    id?: number | string | null;
+    method?: string;
+    params?: Record<string, unknown>;
+    result?: unknown;
+    error?: JsonRpcErrorShape;
+}
+export interface CodeGraphTool {
+    name: string;
+    label: string;
+    description: string;
+    promptSnippet: string;
+    promptGuidelines: string[];
+    inputSchema: {
+        type: "object";
+        properties: Record<string, unknown>;
+        required: string[];
+        additionalProperties: boolean;
+    };
+}
+export interface ToolResult {
+    content?: Array<{
+        type: string;
+        text?: string;
+        [key: string]: unknown;
+    }>;
+    isError?: boolean;
+    _meta?: Record<string, unknown>;
+    [key: string]: unknown;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/lib/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../lib/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,EAAE,OAAO,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,cAAc,EAAE,MAAM,CAAC;IACvB,mBAAmB,EAAE,MAAM,EAAE,CAAC;IAC9B,eAAe,EAAE,OAAO,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,OAAO,GAAG,eAAe,GAAG,SAAS,GAAG,MAAM,CAAC;IACtD,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IACnC,eAAe,EAAE,OAAO,CAAC;IACzB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,KAAK,CAAC;IACf,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAC;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,iBAAiB,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,MAAM,EAAE,CAAC;IAC3B,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ,CAAC;QACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpC,QAAQ,EAAE,MAAM,EAAE,CAAC;QACnB,oBAAoB,EAAE,OAAO,CAAC;KAC/B,CAAC;CACH;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC,CAAC;IACzE,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB"}

package/dist/lib/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/lib/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../lib/types.ts"],"names":[],"mappings":""}

package/dist/lib/worker-pool.d.ts

@@ -0,0 +1,19 @@
+import type { CodeGraphSettings, ToolResult } from "./types.js";
+export declare function codeGraphDaemonRoot(indexStore: string): string;
+export declare class CodeGraphWorkerPool {
+    readonly settings: CodeGraphSettings;
+    readonly projects: Set<string>;
+    private entry?;
+    private creating?;
+    private closed;
+    constructor(settings: CodeGraphSettings);
+    activeProjects(): Set<string>;
+    call(projectPath: string, toolName: string, args: Record<string, unknown>, signal?: AbortSignal): Promise<ToolResult>;
+    closeProject(projectPath: string, _error?: Error): Promise<void>;
+    close(): Promise<void>;
+    private getOrCreateEntry;
+    private createEntry;
+    private closeEntry;
+    private stopEntry;
+}
+//# sourceMappingURL=worker-pool.d.ts.map

package/dist/lib/worker-pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-pool.d.ts","sourceRoot":"","sources":["../../lib/worker-pool.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AA8BhE,wBAAgB,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAE9D;AA4DD,qBAAa,mBAAmB;IAC9B,QAAQ,CAAC,QAAQ,EAAE,iBAAiB,CAAC;IACrC,QAAQ,CAAC,QAAQ,cAAqB;IACtC,OAAO,CAAC,KAAK,CAAC,CAAc;IAC5B,OAAO,CAAC,QAAQ,CAAC,CAAuB;IACxC,OAAO,CAAC,MAAM,CAAS;IAEvB,YAAY,QAAQ,EAAE,iBAAiB,EAEtC;IAED,cAAc,IAAI,GAAG,CAAC,MAAM,CAAC,CAE5B;IAEK,IAAI,CACR,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,UAAU,CAAC,CAmBrB;IAEK,YAAY,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAErE;IAEK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAY3B;YAEa,gBAAgB;YAsBhB,WAAW;YAyEX,UAAU;YAQV,SAAS;CAexB"}

package/dist/lib/worker-pool.js

@@ -0,0 +1,221 @@
+import { spawn } from "node:child_process";
+import { access, mkdir, rm, stat } from "node:fs/promises";
+import { join } from "node:path";
+import { setTimeout as delay } from "node:timers/promises";
+import { pathToFileURL } from "node:url";
+import { resolveCodeGraphLaunch, runCodeGraph, sanitizeDiagnostic, } from "./codegraph.js";
+import { JsonRpcPeer } from "./jsonrpc.js";
+const DAEMON_DIRECTORY = "daemon";
+const INITIALIZE_LOCK = ".initialize.lock";
+const DATABASE_PATH = join(".codegraph", "codegraph.db");
+const LOCK_POLL_MS = 50;
+const MIN_STALE_LOCK_MS = 30_000;
+function isNodeError(error) {
+    return error instanceof Error && "code" in error;
+}
+async function pathExists(path) {
+    try {
+        await access(path);
+        return true;
+    }
+    catch (error) {
+        if (isNodeError(error) && error.code === "ENOENT") {
+            return false;
+        }
+        throw error;
+    }
+}
+export function codeGraphDaemonRoot(indexStore) {
+    return join(indexStore, DAEMON_DIRECTORY);
+}
+async function acquireInitializeLock(lockPath, timeoutMs) {
+    const startedAt = Date.now();
+    const staleAfterMs = Math.max(timeoutMs, MIN_STALE_LOCK_MS);
+    while (true) {
+        try {
+            await mkdir(lockPath);
+            return () => rm(lockPath, { force: true, recursive: true });
+        }
+        catch (error) {
+            if (!isNodeError(error) || error.code !== "EEXIST") {
+                throw error;
+            }
+            const lockStat = await stat(lockPath).catch(() => undefined);
+            if (lockStat && Date.now() - lockStat.mtimeMs > staleAfterMs) {
+                await rm(lockPath, { force: true, recursive: true });
+                continue;
+            }
+            if (Date.now() - startedAt >= timeoutMs) {
+                throw new Error(`Timed out waiting for CodeGraph daemon initialization lock: ${lockPath}`);
+            }
+            await delay(LOCK_POLL_MS);
+        }
+    }
+}
+async function ensureDaemonRoot(settings) {
+    const daemonRoot = codeGraphDaemonRoot(settings.indexStore);
+    const databasePath = join(daemonRoot, DATABASE_PATH);
+    await mkdir(daemonRoot, { recursive: true });
+    if (await pathExists(databasePath)) {
+        return daemonRoot;
+    }
+    const release = await acquireInitializeLock(join(settings.indexStore, INITIALIZE_LOCK), settings.requestTimeoutMs);
+    try {
+        if (!(await pathExists(databasePath))) {
+            await runCodeGraph(settings, daemonRoot, ["init", "-i"], {
+                timeoutMs: settings.requestTimeoutMs,
+            });
+        }
+    }
+    finally {
+        await release();
+    }
+    return daemonRoot;
+}
+export class CodeGraphWorkerPool {
+    settings;
+    projects = new Set();
+    entry;
+    creating;
+    closed = false;
+    constructor(settings) {
+        this.settings = settings;
+    }
+    activeProjects() {
+        return new Set(this.projects);
+    }
+    async call(projectPath, toolName, args, signal) {
+        if (this.closed) {
+            throw new Error("CodeGraph worker pool is closed");
+        }
+        this.projects.add(projectPath);
+        const entry = await this.getOrCreateEntry();
+        const result = await entry.peer.request("tools/call", {
+            name: toolName,
+            arguments: { ...args, projectPath },
+        }, {
+            signal,
+            timeoutMs: this.settings.requestTimeoutMs,
+        });
+        return result;
+    }
+    async closeProject(projectPath, _error) {
+        this.projects.delete(projectPath);
+    }
+    async close() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        this.projects.clear();
+        const pending = this.creating;
+        if (pending) {
+            await pending.catch(() => undefined);
+        }
+        await this.closeEntry();
+    }
+    async getOrCreateEntry() {
+        if (this.entry && !this.entry.exited) {
+            return this.entry;
+        }
+        if (this.creating) {
+            return this.creating;
+        }
+        this.creating = this.createEntry();
+        try {
+            const entry = await this.creating;
+            if (this.closed) {
+                await this.stopEntry(entry);
+                throw new Error("CodeGraph worker pool closed while connecting");
+            }
+            this.entry = entry;
+            return entry;
+        }
+        finally {
+            this.creating = undefined;
+        }
+    }
+    async createEntry() {
+        const daemonRoot = await ensureDaemonRoot(this.settings);
+        const launch = await resolveCodeGraphLaunch(this.settings, [
+            "serve",
+            "--mcp",
+            "--path",
+            daemonRoot,
+        ]);
+        const child = spawn(launch.command, launch.args, {
+            cwd: daemonRoot,
+            env: {
+                ...process.env,
+                CODEGRAPH_DAEMON_IDLE_TIMEOUT_MS: process.env.CODEGRAPH_DAEMON_IDLE_TIMEOUT_MS ??
+                    String(this.settings.workerIdleTimeoutMs),
+                CODEGRAPH_QUERY_POOL_SIZE: process.env.CODEGRAPH_QUERY_POOL_SIZE ??
+                    String(this.settings.maxWorkers),
+            },
+            stdio: ["pipe", "pipe", "pipe"],
+        });
+        const peer = new JsonRpcPeer(child.stdout, child.stdin, {
+            name: "shared-codegraph-daemon",
+        });
+        const entry = { child, peer, exited: false };
+        let stderr = "";
+        child.stderr.setEncoding("utf8");
+        child.stderr.on("data", (chunk) => {
+            stderr = (stderr + chunk).slice(-8_192);
+        });
+        child.once("error", (error) => {
+            entry.exited = true;
+            peer.close(error);
+        });
+        child.once("exit", (code, signal) => {
+            entry.exited = true;
+            if (this.entry === entry) {
+                this.entry = undefined;
+            }
+            const detail = stderr.trim() ? `: ${sanitizeDiagnostic(stderr)}` : "";
+            peer.close(new Error(`CodeGraph proxy exited (code=${String(code)}, signal=${String(signal)})${detail}`));
+        });
+        try {
+            await peer.request("initialize", {
+                protocolVersion: "2024-11-05",
+                capabilities: {},
+                clientInfo: {
+                    name: "pi-codegraph",
+                    version: "0.0.0",
+                },
+                rootUri: pathToFileURL(daemonRoot).href,
+            }, { timeoutMs: this.settings.requestTimeoutMs });
+            peer.notify("notifications/initialized");
+            return entry;
+        }
+        catch (error) {
+            await this.stopEntry(entry);
+            const detail = stderr.trim() ? `: ${sanitizeDiagnostic(stderr)}` : "";
+            throw new Error(`Failed to connect to shared CodeGraph daemon${detail}`, {
+                cause: error,
+            });
+        }
+    }
+    async closeEntry() {
+        const entry = this.entry;
+        this.entry = undefined;
+        if (entry) {
+            await this.stopEntry(entry);
+        }
+    }
+    async stopEntry(entry) {
+        entry.peer.close(new Error("CodeGraph proxy closed"));
+        if (entry.exited) {
+            return;
+        }
+        entry.child.stdin.end();
+        const exited = new Promise((resolve) => {
+            entry.child.once("exit", () => resolve());
+        });
+        await Promise.race([exited, delay(500)]);
+        if (!entry.exited) {
+            entry.child.kill();
+        }
+    }
+}
+//# sourceMappingURL=worker-pool.js.map

package/dist/lib/worker-pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-pool.js","sourceRoot":"","sources":["../../lib/worker-pool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAuC,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,IAAI,KAAK,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EACL,sBAAsB,EACtB,YAAY,EACZ,kBAAkB,GACnB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAS3C,MAAM,gBAAgB,GAAG,QAAQ,CAAC;AAClC,MAAM,eAAe,GAAG,kBAAkB,CAAC;AAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;AACzD,MAAM,YAAY,GAAG,EAAE,CAAC;AACxB,MAAM,iBAAiB,GAAG,MAAM,CAAC;AAEjC,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,CAAC;AACnD,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,IAAY;IACpC,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClD,OAAO,KAAK,CAAC;QACf,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,UAAkB;IACpD,OAAO,IAAI,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;AAC5C,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,QAAgB,EAChB,SAAiB;IAEjB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;IAE5D,OAAO,IAAI,EAAE,CAAC;QACZ,IAAI,CAAC;YACH,MAAM,KAAK,CAAC,QAAQ,CAAC,CAAC;YACtB,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACnD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC7D,IAAI,QAAQ,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,OAAO,GAAG,YAAY,EAAE,CAAC;gBAC7D,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrD,SAAS;YACX,CAAC;YAED,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,SAAS,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CACb,+DAA+D,QAAQ,EAAE,CAC1E,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;AACH,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,QAA2B;IACzD,MAAM,UAAU,GAAG,mBAAmB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAC5D,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IACrD,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAE7C,IAAI,MAAM,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,qBAAqB,CACzC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,eAAe,CAAC,EAC1C,QAAQ,CAAC,gBAAgB,CAC1B,CAAC;IACF,IAAI,CAAC;QACH,IAAI,CAAC,CAAC,MAAM,UAAU,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC;YACtC,MAAM,YAAY,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE;gBACvD,SAAS,EAAE,QAAQ,CAAC,gBAAgB;aACrC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,EAAE,CAAC;IAClB,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,OAAO,mBAAmB;IACrB,QAAQ,CAAoB;IAC5B,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,KAAK,CAAe;IACpB,QAAQ,CAAwB;IAChC,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAY,QAA2B;QACrC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,WAAmB,EACnB,QAAgB,EAChB,IAA6B,EAC7B,MAAoB;QAEpB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CACrC,YAAY,EACZ;YACE,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE;SACpC,EACD;YACE,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB;SAC1C,CACF,CAAC;QACF,OAAO,MAAoB,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,WAAmB,EAAE,MAAc;QACpD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEtB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC;QACD,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,KAAK,CAAC;QACpB,CAAC;QACD,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC;YAClC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;YACnE,CAAC;YACD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YACnB,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW;QACvB,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,QAAQ,EAAE;YACzD,OAAO;YACP,OAAO;YACP,QAAQ;YACR,UAAU;SACX,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE;YAC/C,GAAG,EAAE,UAAU;YACf,GAAG,EAAE;gBACH,GAAG,OAAO,CAAC,GAAG;gBACd,gCAAgC,EAC9B,OAAO,CAAC,GAAG,CAAC,gCAAgC;oBAC5C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC;gBAC3C,yBAAyB,EACvB,OAAO,CAAC,GAAG,CAAC,yBAAyB;oBACrC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC;aACnC;YACD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,EAAE;YACtD,IAAI,EAAE,yBAAyB;SAChC,CAAC,CAAC;QACH,MAAM,KAAK,GAAgB,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QAC1D,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACjC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YACxC,MAAM,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAClC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC;YACpB,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;YACzB,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtE,IAAI,CAAC,KAAK,CACR,IAAI,KAAK,CACP,gCAAgC,MAAM,CAAC,IAAI,CAAC,YAAY,MAAM,CAAC,MAAM,CAAC,IAAI,MAAM,EAAE,CACnF,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAChB,YAAY,EACZ;gBACE,eAAe,EAAE,YAAY;gBAC7B,YAAY,EAAE,EAAE;gBAChB,UAAU,EAAE;oBACV,IAAI,EAAE,cAAc;oBACpB,OAAO,EAAE,OAAO;iBACjB;gBACD,OAAO,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC,IAAI;aACxC,EACD,EAAE,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAC9C,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACzC,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC5B,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACtE,MAAM,IAAI,KAAK,CAAC,+CAA+C,MAAM,EAAE,EAAE;gBACvE,KAAK,EAAE,KAAK;aACb,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACvB,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,KAAkB;QACxC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;QACtD,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAC3C,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAClB,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@isac322/pi-codegraph",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "description": "CodeGraph tools for Pi and OMP with worktree-aware index lifecycle management.",
   "license": "MIT",
   "type": "module",
@@ -24,27 +24,40 @@
   ],
   "files": [
     ".mcp.json",
-    "bin",
-    "extensions",
-    "lib",
+    "dist",
     "README.md",
+    "CONTRIBUTING.md",
     "LICENSE"
   ],
   "bin": {
-    "pi-codegraph-mcp": "./bin/codegraph-mcp.mjs"
+    "pi-codegraph-mcp": "./dist/bin/codegraph-mcp.js"
   },
   "pi": {
     "extensions": [
-      "./extensions/pi.mjs"
+      "./dist/extensions/pi.js"
     ]
   },
   "omp": {
     "extensions": [
-      "./extensions/omp.mjs"
+      "./dist/extensions/omp.js"
     ]
   },
   "engines": {
-    "node": ">=22.19.0"
+    "node": ">=22.19.0 <25.0.0"
+  },
+  "scripts": {
+    "clean": "node -e \"require('node:fs').rmSync('dist',{recursive:true,force:true})\"",
+    "build": "npm run clean && tsc -p tsconfig.build.json && node scripts/ensure-bin-executable.ts",
+    "check": "biome check .",
+    "check:fix": "biome check --write .",
+    "format": "biome format --write .",
+    "lint": "biome lint .",
+    "test:omp:docker": "docker build --file tests/docker/Dockerfile.omp --tag pi-codegraph-omp-integration . && docker run --rm pi-codegraph-omp-integration",
+    "test": "npm run build && node --test tests/codegraph.test.mjs",
+    "typecheck": "tsc -p tsconfig.json",
+    "prepack": "npm run build",
+    "test:integration:daemon": "docker build -f tests/docker/Dockerfile.omp -t pi-codegraph-omp-integration . && docker run --rm --init pi-codegraph-omp-integration bun tests/shared-daemon-docker.integration.ts",
+    "test:integration:pi": "docker build -f tests/docker/Dockerfile.pi -t pi-codegraph-pi-integration . && docker run --rm --init pi-codegraph-pi-integration"
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",
@@ -59,7 +72,15 @@
     }
   },
   "optionalDependencies": {
-    "@colbymchenry/codegraph": "1.0.1"
+    "@colbymchenry/codegraph": "1.4.1"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "^2.5.4",
+    "@earendil-works/pi-coding-agent": "^0.80.10",
+    "@earendil-works/pi-tui": "^0.80.10",
+    "@types/node": "^22.20.1",
+    "tsx": "^4.23.1",
+    "typescript": "^7.0.2"
   },
   "publishConfig": {
     "access": "public"

package/bin/codegraph-mcp.mjs

@@ -1,146 +0,0 @@
-#!/usr/bin/env node
-import { loadSettings } from "../lib/config.mjs";
-import { ProjectGuard, WorkspaceManager, annotateFilesResult, normalizeFilesPath, publicSettings, sanitizeDiagnostic, truncateText } from "../lib/codegraph.mjs";
-import { CodeGraphWorkerPool } from "../lib/worker-pool.mjs";
-import { codegraphTools, codegraphToolNames } from "../lib/tool-metadata.mjs";
-
-const settings = await loadSettings();
-const baseRoot = process.env.PI_CODEGRAPH_BASE_ROOT || process.cwd();
-const guard = await ProjectGuard.create(baseRoot, settings);
-const manager = new WorkspaceManager(settings);
-const pool = new CodeGraphWorkerPool(settings);
-const controllers = new Map();
-let buffer = "";
-let gcTimer;
-
-function write(message) {
-  process.stdout.write(`${JSON.stringify(message)}\n`);
-}
-
-function response(id, result) {
-  write({ jsonrpc: "2.0", id, result });
-}
-
-function errorResponse(id, error, code = -32603) {
-  write({ jsonrpc: "2.0", id, error: { code, message: sanitizeDiagnostic(error?.message || error) } });
-}
-
-async function resolveIdentity(params = {}) {
-  return guard.resolve(params.projectPath || baseRoot);
-}
-
-function extractText(result) {
-  return (result?.content || []).filter((part) => part?.type === "text").map((part) => part.text).join("\n") || JSON.stringify(result);
-}
-
-async function callTool(params, signal) {
-  const name = params?.name;
-  if (!codegraphToolNames.includes(name)) throw new Error(`Unknown CodeGraph tool: ${name}`);
-  const supplied = { ...(params?.arguments || {}) };
-  const identity = await resolveIdentity(supplied);
-  await manager.prepare(identity, { signal });
-  delete supplied.projectPath;
-  const originalFilesPath = name === "codegraph_files" && typeof supplied.path === "string" ? supplied.path : undefined;
-  if (name === "codegraph_files") {
-    const normalized = normalizeFilesPath(originalFilesPath, identity.sourcePath);
-    if (normalized === undefined) delete supplied.path;
-    else supplied.path = normalized;
-  }
-  const result = await pool.call(identity.sourcePath, name, supplied, signal);
-  if (result?.isError) return result;
-  let text = extractText(result);
-  if (name === "codegraph_files") text = annotateFilesResult(text, originalFilesPath);
-  const limited = truncateText(text, settings.maxOutputChars);
-  return {
-    ...result,
-    content: [{ type: "text", text: limited.text }],
-    _meta: { projectPath: identity.sourcePath, truncated: limited.truncated },
-  };
-}
-
-async function handleRequest(message, signal) {
-  switch (message.method) {
-    case "initialize":
-      return {
-        protocolVersion: "2024-11-05",
-        capabilities: { tools: { listChanged: false } },
-        serverInfo: { name: "pi-codegraph", version: "0.2.0" },
-      };
-    case "tools/list":
-      return { tools: codegraphTools.map((tool) => ({ name: tool.name, description: tool.description, inputSchema: tool.inputSchema })) };
-    case "tools/call":
-      return callTool(message.params, signal);
-    case "codegraph/workspace/prepare": {
-      const identity = await resolveIdentity(message.params);
-      const result = await manager.prepare(identity, { signal });
-      await manager.gc(pool.activeProjects());
-      return result;
-    }
-    case "codegraph/workspace/status": {
-      const identity = await resolveIdentity(message.params);
-      return manager.status(identity);
-    }
-    case "codegraph/workspace/sync": {
-      const identity = await resolveIdentity(message.params);
-      return manager.prepare(identity, { signal, forceSync: true });
-    }
-    case "codegraph/workspace/doctor": {
-      const identity = await resolveIdentity(message.params);
-      return manager.doctor(identity);
-    }
-    case "codegraph/workspace/gc":
-      return manager.gc(pool.activeProjects(), Boolean(message.params?.force));
-    case "codegraph/settings":
-      return publicSettings(settings);
-    case "shutdown":
-      return null;
-    default:
-      throw Object.assign(new Error(`Method not found: ${message.method}`), { rpcCode: -32601 });
-  }
-}
-
-async function receive(message) {
-  if (message.method === "notifications/cancelled") {
-    controllers.get(message.params?.requestId)?.abort();
-    return;
-  }
-  if (message.id === undefined) return;
-  const controller = new AbortController();
-  controllers.set(message.id, controller);
-  try {
-    response(message.id, await handleRequest(message, controller.signal));
-  } catch (error) {
-    if (message.method === "tools/call") {
-      response(message.id, { content: [{ type: "text", text: sanitizeDiagnostic(error.message) }], isError: true });
-    } else {
-      errorResponse(message.id, error, error.rpcCode);
-    }
-  } finally {
-    controllers.delete(message.id);
-  }
-}
-
-process.stdin.on("data", (chunk) => {
-  buffer += chunk.toString("utf8");
-  let newline;
-  while ((newline = buffer.indexOf("\n")) !== -1) {
-    const line = buffer.slice(0, newline).trim();
-    buffer = buffer.slice(newline + 1);
-    if (!line) continue;
-    try { void receive(JSON.parse(line)); } catch { errorResponse(null, new Error("Invalid JSON"), -32700); }
-  }
-});
-
-async function shutdown() {
-  clearInterval(gcTimer);
-  for (const controller of controllers.values()) controller.abort();
-  await pool.close();
-}
-
-if (settings.autoGc) {
-  gcTimer = setInterval(() => { void manager.gc(pool.activeProjects()); }, 60 * 60_000);
-  gcTimer.unref?.();
-}
-process.on("SIGTERM", () => { void shutdown().finally(() => process.exit(0)); });
-process.on("SIGINT", () => { void shutdown().finally(() => process.exit(0)); });
-process.stdin.on("end", () => { void shutdown().finally(() => process.exit(0)); });

package/extensions/omp.mjs

@@ -1,14 +0,0 @@
-import { loadSettings } from "../lib/config.mjs";
-import { workspaceSummary } from "../lib/codegraph.mjs";
-import { buildCodeGraphPrompt } from "../lib/prompt.mjs";
-
-export default async function ompCodeGraphExtension(omp) {
-  const settings = await loadSettings();
-  if (!settings.promptInjection || typeof omp?.on !== "function") return;
-  omp.on("before_agent_start", async (event, ctx) => {
-    const cwd = ctx?.cwd || process.cwd();
-    const status = await workspaceSummary(cwd);
-    const guidance = buildCodeGraphPrompt({ runtime: "omp", cwd, status });
-    return { systemPrompt: event.systemPrompt ? `${event.systemPrompt}\n\n${guidance}` : guidance };
-  });
-}