rho-graph 0.1.0

package/dist/db/schema.js

@@ -0,0 +1,22 @@
+export async function setupSchema(db) {
+    const session = db.session();
+    try {
+        // Uniqueness constraints
+        await session.run("CREATE CONSTRAINT repo_path IF NOT EXISTS FOR (r:Repository) REQUIRE r.path IS UNIQUE");
+        await session.run("CREATE CONSTRAINT file_path IF NOT EXISTS FOR (f:File) REQUIRE f.path IS UNIQUE");
+        // Indexes for fast lookup
+        await session.run("CREATE INDEX function_name IF NOT EXISTS FOR (f:Function) ON (f.name)");
+        await session.run("CREATE INDEX class_name IF NOT EXISTS FOR (c:Class) ON (c.name)");
+        await session.run("CREATE INDEX file_language IF NOT EXISTS FOR (f:File) ON (f.language)");
+        // Full-text index for search_code
+        await session.run(`
+      CREATE FULLTEXT INDEX code_search IF NOT EXISTS
+      FOR (n:Function|Class)
+      ON EACH [n.name, n.snippet, n.docstring]
+    `);
+    }
+    finally {
+        await session.close();
+    }
+}
+//# sourceMappingURL=schema.js.map

package/dist/db/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/db/schema.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,EAAgB;IAChD,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,IAAI,CAAC;QACH,yBAAyB;QACzB,MAAM,OAAO,CAAC,GAAG,CACf,uFAAuF,CACxF,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CACf,iFAAiF,CAClF,CAAC;QAEF,0BAA0B;QAC1B,MAAM,OAAO,CAAC,GAAG,CACf,uEAAuE,CACxE,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CACf,iEAAiE,CAClE,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CACf,uEAAuE,CACxE,CAAC;QAEF,kCAAkC;QAClC,MAAM,OAAO,CAAC,GAAG,CAAC;;;;KAIjB,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;AACH,CAAC"}

package/dist/docker/neo4j.d.ts

@@ -0,0 +1,5 @@
+export declare function isDockerAvailable(): boolean;
+export declare function getContainerStatus(): "running" | "stopped" | "not_found";
+export declare function startNeo4j(composeFilePath: string): void;
+export declare function stopNeo4j(): void;
+export declare function waitForNeo4j(uri: string, username: string, password: string, maxAttempts?: number): Promise<boolean>;

package/dist/docker/neo4j.js

@@ -0,0 +1,85 @@
+// src/docker/neo4j.ts
+import { execFileSync } from "node:child_process";
+import { resolve } from "node:path";
+const CONTAINER_NAME = "code-graph-rag-neo4j";
+function runDocker(...args) {
+    return execFileSync("docker", args, {
+        encoding: "utf-8",
+        stdio: "pipe",
+    });
+}
+export function isDockerAvailable() {
+    try {
+        runDocker("info");
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+export function getContainerStatus() {
+    try {
+        const result = runDocker("inspect", "--format", "{{.State.Running}}", CONTAINER_NAME).trim();
+        return result === "true" ? "running" : "stopped";
+    }
+    catch {
+        return "not_found";
+    }
+}
+export function startNeo4j(composeFilePath) {
+    const status = getContainerStatus();
+    if (status === "running")
+        return;
+    if (status === "stopped") {
+        runDocker("start", CONTAINER_NAME);
+        return;
+    }
+    const composePath = resolve(composeFilePath);
+    execFileSync("docker", ["compose", "-f", composePath, "up", "-d"], {
+        encoding: "utf-8",
+        stdio: "pipe",
+    });
+}
+export function stopNeo4j() {
+    const status = getContainerStatus();
+    if (status === "running") {
+        runDocker("stop", CONTAINER_NAME);
+    }
+}
+export async function waitForNeo4j(uri, username, password, maxAttempts = 60) {
+    // First wait for the process to start via admin check
+    let processReady = false;
+    for (let i = 0; i < maxAttempts; i++) {
+        try {
+            execFileSync("docker", ["exec", CONTAINER_NAME, "neo4j-admin", "server", "status"], { encoding: "utf-8", stdio: "pipe" });
+            processReady = true;
+            break;
+        }
+        catch {
+            await new Promise((r) => setTimeout(r, 1000));
+        }
+    }
+    if (!processReady)
+        return false;
+    // Then wait for the Bolt port to accept connections
+    const neo4j = await import("neo4j-driver");
+    const driver = neo4j.default.driver(uri, neo4j.default.auth.basic(username, password));
+    try {
+        for (let i = 0; i < maxAttempts; i++) {
+            try {
+                const session = driver.session();
+                await session.run("RETURN 1");
+                await session.close();
+                return true;
+            }
+            catch {
+                await new Promise((r) => setTimeout(r, 1000));
+            }
+        }
+        return false;
+    }
+    finally {
+        await driver.close();
+    }
+}
+//# sourceMappingURL=neo4j.js.map

package/dist/docker/neo4j.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j.js","sourceRoot":"","sources":["../../src/docker/neo4j.ts"],"names":[],"mappings":"AAAA,sBAAsB;AACtB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,cAAc,GAAG,sBAAsB,CAAC;AAE9C,SAAS,SAAS,CAAC,GAAG,IAAc;IAClC,OAAO,YAAY,CAAC,QAAQ,EAAE,IAAI,EAAE;QAClC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,MAAM;KACd,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC/B,IAAI,CAAC;QACH,SAAS,CAAC,MAAM,CAAC,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,SAAS,CACtB,SAAS,EACT,UAAU,EACV,oBAAoB,EACpB,cAAc,CACf,CAAC,IAAI,EAAE,CAAC;QACT,OAAO,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,WAAW,CAAC;IACrB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,eAAuB;IAChD,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;IACpC,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO;IACjC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QACnC,OAAO;IACT,CAAC;IACD,MAAM,WAAW,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;IAC7C,YAAY,CAAC,QAAQ,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE;QACjE,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,MAAM;KACd,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;IACpC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;IACpC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,GAAW,EACX,QAAgB,EAChB,QAAgB,EAChB,WAAW,GAAG,EAAE;IAEhB,sDAAsD;IACtD,IAAI,YAAY,GAAG,KAAK,CAAC;IACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,IAAI,CAAC;YACH,YAAY,CACV,QAAQ,EACR,CAAC,MAAM,EAAE,cAAc,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,CAAC,EAC3D,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CACrC,CAAC;YACF,YAAY,GAAG,IAAI,CAAC;YACpB,MAAM;QACR,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IACD,IAAI,CAAC,YAAY;QAAE,OAAO,KAAK,CAAC;IAEhC,oDAAoD;IACpD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CACjC,GAAG,EACH,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAC7C,CAAC;IACF,IAAI,CAAC;QACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;gBACjC,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC9B,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAChD,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;YAAS,CAAC;QACT,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/indexer/batch-writer.d.ts

@@ -0,0 +1,35 @@
+import type { DbConnection } from "../db/connection.js";
+import type { GraphEntities } from "./extractor.js";
+import type { FileMetadata } from "./graph-writer.js";
+export declare class BatchGraphWriter {
+    private readonly db;
+    private fileMetas;
+    private allFunctions;
+    private allClasses;
+    private allCalls;
+    private allImports;
+    private _estimatedMemoryBytes;
+    private _flushSem;
+    private _inFlight;
+    private _flushError;
+    private readonly batchSize;
+    private _flushesScheduled;
+    private _flushesCompleted;
+    onFlushProgress?: (completed: number, total: number) => void;
+    constructor(db: DbConnection, options?: {
+        batchSize?: number;
+        filePathSet?: Set<string>;
+    });
+    private readonly filePathSet;
+    add(entities: GraphEntities, meta: FileMetadata): void;
+    private _snapshot;
+    private _scheduleFlush;
+    get flushesScheduled(): number;
+    get flushesCompleted(): number;
+    private _runFlush;
+    private _doFlush;
+    flush(): Promise<void>;
+    waitForPendingFlush(): Promise<void>;
+    get pendingFileCount(): number;
+    get estimatedMemoryBytes(): number;
+}

package/dist/indexer/batch-writer.js

@@ -0,0 +1,202 @@
+import { Semaphore } from "./parallel-pipeline.js";
+import { batchDeleteFileChildren, batchUpsertFiles, batchUpsertClasses, batchUpsertFunctions, batchUpsertMethods, batchUpsertCallRelationships, batchUpsertImportRelationships, } from "../db/queries.js";
+import { resolveImport } from "./import-resolver.js";
+export class BatchGraphWriter {
+    db;
+    fileMetas = [];
+    allFunctions = [];
+    allClasses = [];
+    allCalls = [];
+    allImports = [];
+    _estimatedMemoryBytes = 0;
+    _flushSem = new Semaphore(1);
+    _inFlight = new Set();
+    _flushError = null;
+    batchSize;
+    _flushesScheduled = 0;
+    _flushesCompleted = 0;
+    onFlushProgress;
+    constructor(db, options = {}) {
+        this.db = db;
+        this.batchSize = options.batchSize ?? 200;
+        this.filePathSet = options.filePathSet ?? new Set();
+    }
+    filePathSet;
+    add(entities, meta) {
+        if (this._flushError) {
+            const err = this._flushError;
+            this._flushError = null;
+            throw err;
+        }
+        this.fileMetas.push(meta);
+        const validFunctions = entities.functions.filter((fn) => fn.name !== "");
+        const validClasses = entities.classes.filter((cls) => cls.name !== "");
+        this.allFunctions.push(...validFunctions);
+        this.allClasses.push(...validClasses);
+        this.allCalls.push(...entities.calls);
+        for (const imp of entities.imports) {
+            const target = resolveImport(imp.source, meta.filePath, meta.language, this.filePathSet);
+            if (target)
+                this.allImports.push({ sourceFilePath: meta.filePath, targetFilePath: target });
+        }
+        for (const fn of validFunctions) {
+            this._estimatedMemoryBytes += fn.snippet.length;
+        }
+        if (this.fileMetas.length >= this.batchSize) {
+            this._scheduleFlush(this._snapshot());
+        }
+    }
+    _snapshot() {
+        const snapshot = {
+            fileMetas: this.fileMetas,
+            functions: this.allFunctions,
+            classes: this.allClasses,
+            calls: this.allCalls,
+            imports: this.allImports,
+        };
+        this.fileMetas = [];
+        this.allFunctions = [];
+        this.allClasses = [];
+        this.allCalls = [];
+        this.allImports = [];
+        this._estimatedMemoryBytes = 0;
+        return snapshot;
+    }
+    _scheduleFlush(snapshot) {
+        if (snapshot.fileMetas.length === 0)
+            return;
+        this._flushesScheduled++;
+        const p = this._runFlush(snapshot)
+            .then(() => {
+            this._flushesCompleted++;
+            this.onFlushProgress?.(this._flushesCompleted, this._flushesScheduled);
+        })
+            .catch((err) => {
+            this._flushesCompleted++;
+            this._flushError = err instanceof Error ? err : new Error(String(err));
+        });
+        this._inFlight.add(p);
+        p.finally(() => this._inFlight.delete(p));
+    }
+    get flushesScheduled() { return this._flushesScheduled; }
+    get flushesCompleted() { return this._flushesCompleted; }
+    async _runFlush(snapshot) {
+        const release = await this._flushSem.acquire();
+        try {
+            await this._doFlush(snapshot);
+        }
+        finally {
+            release();
+        }
+    }
+    async _doFlush(snapshot) {
+        const { fileMetas, functions: allFunctions, classes: allClasses, calls: allCalls, imports: allImports } = snapshot;
+        const session = this.db.session();
+        const tx = session.beginTransaction();
+        try {
+            const filePaths = fileMetas.map((m) => m.filePath);
+            const deleteQ = batchDeleteFileChildren(filePaths);
+            await tx.run(deleteQ.cypher, deleteQ.params);
+            const fileItems = fileMetas.map((m) => ({
+                path: m.filePath,
+                relativePath: m.relativePath,
+                repoPath: m.repoPath,
+                language: m.language,
+                hash: m.hash,
+                lastModified: m.lastModified,
+            }));
+            const filesQ = batchUpsertFiles(fileItems);
+            await tx.run(filesQ.cypher, filesQ.params);
+            if (allClasses.length > 0) {
+                const classItems = allClasses.map((c) => ({
+                    name: c.name,
+                    filePath: c.filePath,
+                    startLine: c.startLine,
+                    endLine: c.endLine,
+                    docstring: c.docstring,
+                }));
+                const classesQ = batchUpsertClasses(classItems);
+                await tx.run(classesQ.cypher, classesQ.params);
+            }
+            const topLevelFns = allFunctions.filter((fn) => fn.className === null);
+            if (topLevelFns.length > 0) {
+                const fnItems = topLevelFns.map((fn) => ({
+                    name: fn.name,
+                    filePath: fn.filePath,
+                    startLine: fn.startLine,
+                    endLine: fn.endLine,
+                    signature: fn.signature,
+                    docstring: fn.docstring,
+                    snippet: fn.snippet,
+                    className: null,
+                }));
+                const fnsQ = batchUpsertFunctions(fnItems);
+                await tx.run(fnsQ.cypher, fnsQ.params);
+            }
+            const methods = allFunctions.filter((fn) => fn.className !== null);
+            if (methods.length > 0) {
+                const methodItems = methods.map((fn) => ({
+                    name: fn.name,
+                    filePath: fn.filePath,
+                    startLine: fn.startLine,
+                    endLine: fn.endLine,
+                    signature: fn.signature,
+                    docstring: fn.docstring,
+                    snippet: fn.snippet,
+                    className: fn.className,
+                }));
+                const methodsQ = batchUpsertMethods(methodItems);
+                await tx.run(methodsQ.cypher, methodsQ.params);
+            }
+            if (allCalls.length > 0) {
+                const callsQ = batchUpsertCallRelationships(allCalls);
+                await tx.run(callsQ.cypher, callsQ.params);
+            }
+            if (allImports.length > 0) {
+                const importsQ = batchUpsertImportRelationships(allImports);
+                await tx.run(importsQ.cypher, importsQ.params);
+            }
+            await tx.commit();
+        }
+        catch (err) {
+            await tx.rollback();
+            throw err;
+        }
+        finally {
+            await session.close();
+        }
+    }
+    async flush() {
+        if (this._flushError) {
+            const err = this._flushError;
+            this._flushError = null;
+            throw err;
+        }
+        // Flush any remaining buffered files
+        if (this.fileMetas.length > 0) {
+            this._scheduleFlush(this._snapshot());
+        }
+        // Drain all in-flight flushes
+        while (this._inFlight.size > 0) {
+            await Promise.all([...this._inFlight]);
+        }
+        if (this._flushError) {
+            const err = this._flushError;
+            this._flushError = null;
+            throw err;
+        }
+    }
+    // kept for API compatibility — flush() now drains everything
+    async waitForPendingFlush() {
+        while (this._inFlight.size > 0) {
+            await Promise.all([...this._inFlight]);
+        }
+    }
+    get pendingFileCount() {
+        return this.fileMetas.length;
+    }
+    get estimatedMemoryBytes() {
+        return this._estimatedMemoryBytes;
+    }
+}
+//# sourceMappingURL=batch-writer.js.map

package/dist/indexer/batch-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-writer.js","sourceRoot":"","sources":["../../src/indexer/batch-writer.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EACL,uBAAuB,EACvB,gBAAgB,EAChB,kBAAkB,EAClB,oBAAoB,EACpB,kBAAkB,EAClB,4BAA4B,EAC5B,8BAA8B,GAC/B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAUrD,MAAM,OAAO,gBAAgB;IAgBR;IAfX,SAAS,GAAmB,EAAE,CAAC;IAC/B,YAAY,GAAwB,EAAE,CAAC;IACvC,UAAU,GAAqB,EAAE,CAAC;IAClC,QAAQ,GAAoB,EAAE,CAAC;IAC/B,UAAU,GAA8D,EAAE,CAAC;IAC3E,qBAAqB,GAAG,CAAC,CAAC;IAC1B,SAAS,GAAG,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC;IAC7B,SAAS,GAAG,IAAI,GAAG,EAAiB,CAAC;IACrC,WAAW,GAAiB,IAAI,CAAC;IACxB,SAAS,CAAS;IAC3B,iBAAiB,GAAG,CAAC,CAAC;IACtB,iBAAiB,GAAG,CAAC,CAAC;IAC9B,eAAe,CAA8C;IAE7D,YACmB,EAAgB,EACjC,UAA6D,EAAE;QAD9C,OAAE,GAAF,EAAE,CAAc;QAGjC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,GAAG,CAAC;QAC1C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,GAAG,EAAE,CAAC;IACtD,CAAC;IAEgB,WAAW,CAAc;IAE1C,GAAG,CAAC,QAAuB,EAAE,IAAkB;QAC7C,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC;YAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1B,MAAM,cAAc,GAAG,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC;QAEvE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QAC1C,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAEtC,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACzF,IAAI,MAAM;gBAAE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9F,CAAC;QAED,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;YAChC,IAAI,CAAC,qBAAqB,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;QAClD,CAAC;QAED,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC5C,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEO,SAAS;QACf,MAAM,QAAQ,GAAkB;YAC9B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,CAAC,YAAY;YAC5B,OAAO,EAAE,IAAI,CAAC,UAAU;YACxB,KAAK,EAAE,IAAI,CAAC,QAAQ;YACpB,OAAO,EAAE,IAAI,CAAC,UAAU;SACzB,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,EAAE,CAAC;QACpB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC;QACrB,IAAI,CAAC,qBAAqB,GAAG,CAAC,CAAC;QAC/B,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,cAAc,CAAC,QAAuB;QAC5C,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAE5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,MAAM,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;aAC/B,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,IAAI,CAAC,eAAe,EAAE,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACzE,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,IAAI,CAAC,WAAW,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QACL,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACtB,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,gBAAgB,KAAa,OAAO,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IACjE,IAAI,gBAAgB,KAAa,OAAO,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IAEzD,KAAK,CAAC,SAAS,CAAC,QAAuB;QAC7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAC/C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;gBAAS,CAAC;YACT,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,QAAuB;QAC5C,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC;QAEnH,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC;QAClC,MAAM,EAAE,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;QACtC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,OAAO,GAAG,uBAAuB,CAAC,SAAS,CAAC,CAAC;YACnD,MAAM,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YAE7C,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACtC,IAAI,EAAE,CAAC,CAAC,QAAQ;gBAChB,YAAY,EAAE,CAAC,CAAC,YAAY;gBAC5B,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,YAAY,EAAE,CAAC,CAAC,YAAY;aAC7B,CAAC,CAAC,CAAC;YACJ,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAE3C,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACxC,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,QAAQ,EAAE,CAAC,CAAC,QAAQ;oBACpB,SAAS,EAAE,CAAC,CAAC,SAAS;oBACtB,OAAO,EAAE,CAAC,CAAC,OAAO;oBAClB,SAAS,EAAE,CAAC,CAAC,SAAS;iBACvB,CAAC,CAAC,CAAC;gBACJ,MAAM,QAAQ,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;gBAChD,MAAM,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC;YACvE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvC,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,QAAQ,EAAE,EAAE,CAAC,QAAQ;oBACrB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;oBACnB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;oBACnB,SAAS,EAAE,IAAqB;iBACjC,CAAC,CAAC,CAAC;gBACJ,MAAM,IAAI,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;gBAC3C,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC;YACnE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;oBACvC,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,QAAQ,EAAE,EAAE,CAAC,QAAQ;oBACrB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;oBACnB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,SAAS,EAAE,EAAE,CAAC,SAAS;oBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;oBACnB,SAAS,EAAE,EAAE,CAAC,SAAmB;iBAClC,CAAC,CAAC,CAAC;gBACJ,MAAM,QAAQ,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;gBACjD,MAAM,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YACjD,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,MAAM,MAAM,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC;gBACtD,MAAM,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YAC7C,CAAC;YAED,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,8BAA8B,CAAC,UAAU,CAAC,CAAC;gBAC5D,MAAM,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,EAAE,CAAC,MAAM,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC;YAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,qCAAqC;QACrC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QACxC,CAAC;QAED,8BAA8B;QAC9B,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC;YAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,6DAA6D;IAC7D,KAAK,CAAC,mBAAmB;QACvB,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,IAAI,gBAAgB;QAClB,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,IAAI,oBAAoB;QACtB,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;CACF"}

package/dist/indexer/extractor.d.ts

@@ -0,0 +1,35 @@
+import type Parser from "web-tree-sitter";
+export interface ExtractedFunction {
+    name: string;
+    filePath: string;
+    startLine: number;
+    endLine: number;
+    signature: string;
+    docstring: string | null;
+    snippet: string;
+    className: string | null;
+}
+export interface ExtractedClass {
+    name: string;
+    filePath: string;
+    startLine: number;
+    endLine: number;
+    docstring: string | null;
+}
+export interface ExtractedImport {
+    source: string;
+    specifiers: string[];
+    isDefault: boolean;
+}
+export interface ExtractedCall {
+    callerName: string;
+    callerFilePath: string;
+    calleeName: string;
+}
+export interface GraphEntities {
+    functions: ExtractedFunction[];
+    classes: ExtractedClass[];
+    imports: ExtractedImport[];
+    calls: ExtractedCall[];
+}
+export declare function extractGraphEntities(tree: Parser.Tree, language: string, source: string, filePath: string): GraphEntities;

package/dist/indexer/extractor.js

@@ -0,0 +1,141 @@
+// src/indexer/extractor.ts
+import { readFileSync } from "node:fs";
+import { resolve, join, dirname } from "node:path";
+import { fileURLToPath } from "node:url";
+const __dirname = dirname(fileURLToPath(import.meta.url));
+const languageMap = JSON.parse(readFileSync(join(__dirname, "language-map.json"), "utf-8"));
+function getNodeText(node, source) {
+    return source.slice(node.startIndex, node.endIndex);
+}
+function getDocstring(node, source) {
+    const prev = node.previousNamedSibling;
+    if (prev && prev.type === "comment") {
+        return getNodeText(prev, source).replace(/^\/\*\*?|\*\/$/g, "").trim();
+    }
+    // Python docstrings: first child of body is expression_statement > string
+    const body = node.childForFieldName("body");
+    if (body) {
+        const first = body.firstNamedChild;
+        if (first?.type === "expression_statement") {
+            const str = first.firstNamedChild;
+            if (str?.type === "string") {
+                return getNodeText(str, source)
+                    .replace(/^['"]|['"]$/g, "")
+                    .replace(/^"""|"""$/g, "")
+                    .trim();
+            }
+        }
+    }
+    return null;
+}
+function extractSignature(node, source) {
+    const lines = getNodeText(node, source).split("\n");
+    return lines[0].trim();
+}
+function walkForCalls(node, source, callerName, callerFilePath, callTypes) {
+    const calls = [];
+    function walk(n) {
+        if (callTypes.includes(n.type)) {
+            const fnNode = n.childForFieldName("function") || n.firstNamedChild;
+            if (fnNode) {
+                let calleeName;
+                if (fnNode.type === "member_expression" || fnNode.type === "attribute") {
+                    const prop = fnNode.childForFieldName("property") ||
+                        fnNode.childForFieldName("attribute");
+                    calleeName = prop
+                        ? getNodeText(prop, source)
+                        : getNodeText(fnNode, source);
+                }
+                else if (fnNode.type === "identifier" ||
+                    fnNode.type === "property_identifier") {
+                    calleeName = getNodeText(fnNode, source);
+                }
+                else {
+                    calleeName = getNodeText(fnNode, source);
+                }
+                if (calleeName && calleeName !== callerName) {
+                    calls.push({ callerName, callerFilePath, calleeName });
+                }
+            }
+        }
+        for (let i = 0; i < n.childCount; i++) {
+            walk(n.child(i));
+        }
+    }
+    walk(node);
+    return calls;
+}
+export function extractGraphEntities(tree, language, source, filePath) {
+    const absPath = resolve(filePath);
+    const mapping = languageMap[language];
+    const functions = [];
+    const classes = [];
+    const imports = [];
+    const calls = [];
+    if (!mapping) {
+        return { functions, classes, imports, calls };
+    }
+    const functionTypes = new Set(mapping.function);
+    const classTypes = new Set(mapping.class);
+    const importTypes = new Set(mapping.import);
+    function walkNode(node, currentClassName) {
+        if (classTypes.has(node.type)) {
+            const nameNode = node.childForFieldName(mapping.name_field);
+            if (nameNode) {
+                const className = getNodeText(nameNode, source);
+                classes.push({
+                    name: className,
+                    filePath: absPath,
+                    startLine: node.startPosition.row + 1,
+                    endLine: node.endPosition.row + 1,
+                    docstring: getDocstring(node, source),
+                });
+                for (let i = 0; i < node.childCount; i++) {
+                    walkNode(node.child(i), className);
+                }
+                return;
+            }
+        }
+        if (functionTypes.has(node.type)) {
+            const nameNode = node.childForFieldName(mapping.name_field);
+            if (nameNode) {
+                const funcName = getNodeText(nameNode, source);
+                const snippet = getNodeText(node, source);
+                const signature = extractSignature(node, source);
+                functions.push({
+                    name: funcName,
+                    filePath: absPath,
+                    startLine: node.startPosition.row + 1,
+                    endLine: node.endPosition.row + 1,
+                    signature,
+                    docstring: getDocstring(node, source),
+                    snippet,
+                    className: currentClassName,
+                });
+                const funcCalls = walkForCalls(node, source, funcName, absPath, mapping.call);
+                calls.push(...funcCalls);
+                return;
+            }
+        }
+        if (importTypes.has(node.type)) {
+            const text = getNodeText(node, source);
+            const sourceNode = node.childForFieldName("source") ||
+                node.childForFieldName("module_name") ||
+                node.childForFieldName("path");
+            const importSource = sourceNode
+                ? getNodeText(sourceNode, source).replace(/['"]/g, "")
+                : text;
+            imports.push({
+                source: importSource,
+                specifiers: [],
+                isDefault: text.includes("import default") || !text.includes("{"),
+            });
+        }
+        for (let i = 0; i < node.childCount; i++) {
+            walkNode(node.child(i), currentClassName);
+        }
+    }
+    walkNode(tree.rootNode, null);
+    return { functions, classes, imports, calls };
+}
+//# sourceMappingURL=extractor.js.map

package/dist/indexer/extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractor.js","sourceRoot":"","sources":["../../src/indexer/extractor.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAGzC,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAc1D,MAAM,WAAW,GAAgB,IAAI,CAAC,KAAK,CACzC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,EAAE,OAAO,CAAC,CAC5D,CAAC;AAwCF,SAAS,WAAW,CAAC,IAAuB,EAAE,MAAc;IAC1D,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;AACtD,CAAC;AAED,SAAS,YAAY,CACnB,IAAuB,EACvB,MAAc;IAEd,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC;IACvC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;QACpC,OAAO,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IACzE,CAAC;IACD,0EAA0E;IAC1E,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAC5C,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC;QACnC,IAAI,KAAK,EAAE,IAAI,KAAK,sBAAsB,EAAE,CAAC;YAC3C,MAAM,GAAG,GAAG,KAAK,CAAC,eAAe,CAAC;YAClC,IAAI,GAAG,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC3B,OAAO,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC;qBAC5B,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;qBAC3B,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;qBACzB,IAAI,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CACvB,IAAuB,EACvB,MAAc;IAEd,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACpD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;AACzB,CAAC;AAED,SAAS,YAAY,CACnB,IAAuB,EACvB,MAAc,EACd,UAAkB,EAClB,cAAsB,EACtB,SAAmB;IAEnB,MAAM,KAAK,GAAoB,EAAE,CAAC;IAElC,SAAS,IAAI,CAAC,CAAoB;QAChC,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC;YACpE,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,UAAkB,CAAC;gBACvB,IAAI,MAAM,CAAC,IAAI,KAAK,mBAAmB,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACvE,MAAM,IAAI,GACR,MAAM,CAAC,iBAAiB,CAAC,UAAU,CAAC;wBACpC,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;oBACxC,UAAU,GAAG,IAAI;wBACf,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC;wBAC3B,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAClC,CAAC;qBAAM,IACL,MAAM,CAAC,IAAI,KAAK,YAAY;oBAC5B,MAAM,CAAC,IAAI,KAAK,qBAAqB,EACrC,CAAC;oBACD,UAAU,GAAG,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC3C,CAAC;qBAAM,CAAC;oBACN,UAAU,GAAG,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;gBAC3C,CAAC;gBACD,IAAI,UAAU,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;oBAC5C,KAAK,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;QACH,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAED,IAAI,CAAC,IAAI,CAAC,CAAC;IACX,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,IAAiB,EACjB,QAAgB,EAChB,MAAc,EACd,QAAgB;IAEhB,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IAEtC,MAAM,SAAS,GAAwB,EAAE,CAAC;IAC1C,MAAM,OAAO,GAAqB,EAAE,CAAC;IACrC,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,MAAM,KAAK,GAAoB,EAAE,CAAC;IAElC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAChD,CAAC;IAED,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IAChD,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAE5C,SAAS,QAAQ,CACf,IAAuB,EACvB,gBAA+B;QAE/B,IAAI,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC5D,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;gBAChD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,OAAO;oBACjB,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,SAAS,EAAE,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC;iBACtC,CAAC,CAAC;gBACH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;oBACzC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,SAAS,CAAC,CAAC;gBACtC,CAAC;gBACD,OAAO;YACT,CAAC;QACH,CAAC;QAED,IAAI,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC5D,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,QAAQ,GAAG,WAAW,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;gBAC/C,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC1C,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAEjD,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,QAAQ;oBACd,QAAQ,EAAE,OAAO;oBACjB,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,GAAG,CAAC;oBACrC,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC;oBACjC,SAAS;oBACT,SAAS,EAAE,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC;oBACrC,OAAO;oBACP,SAAS,EAAE,gBAAgB;iBAC5B,CAAC,CAAC;gBAEH,MAAM,SAAS,GAAG,YAAY,CAC5B,IAAI,EACJ,MAAM,EACN,QAAQ,EACR,OAAO,EACP,OAAO,CAAC,IAAI,CACb,CAAC;gBACF,KAAK,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBACzB,OAAO;YACT,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACvC,MAAM,UAAU,GACd,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAChC,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC;gBACrC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,YAAY,GAAG,UAAU;gBAC7B,CAAC,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;gBACtD,CAAC,CAAC,IAAI,CAAC;YAET,OAAO,CAAC,IAAI,CAAC;gBACX,MAAM,EAAE,YAAY;gBACpB,UAAU,EAAE,EAAE;gBACd,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;aAClE,CAAC,CAAC;QACL,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,gBAAgB,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAE9B,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;AAChD,CAAC"}

package/dist/indexer/graph-writer.d.ts

@@ -0,0 +1,12 @@
+import type { DbConnection } from "../db/connection.js";
+import type { GraphEntities } from "./extractor.js";
+export interface FileMetadata {
+    filePath: string;
+    relativePath: string;
+    repoPath: string;
+    language: string;
+    hash: string;
+    lastModified: number;
+}
+export declare function writeRepoOnce(db: DbConnection, repoPath: string): Promise<void>;
+export declare function writeGraphEntities(db: DbConnection, entities: GraphEntities, meta: FileMetadata, filePathSet?: Set<string>): Promise<void>;

package/dist/indexer/graph-writer.js

@@ -0,0 +1,75 @@
+import { upsertRepository, upsertFile, upsertFunction, upsertClass, upsertCallRelationship, upsertImportRelationship, } from "../db/queries.js";
+import { resolveImport } from "./import-resolver.js";
+export async function writeRepoOnce(db, repoPath) {
+    const session = db.session();
+    try {
+        const repoQ = upsertRepository({ path: repoPath, name: repoPath.split("/").pop() || repoPath });
+        await session.run(repoQ.cypher, repoQ.params);
+    }
+    finally {
+        await session.close();
+    }
+}
+export async function writeGraphEntities(db, entities, meta, filePathSet = new Set()) {
+    const session = db.session();
+    try {
+        // Upsert file
+        const fileQ = upsertFile({
+            path: meta.filePath,
+            relativePath: meta.relativePath,
+            repoPath: meta.repoPath,
+            language: meta.language,
+            hash: meta.hash,
+            lastModified: meta.lastModified,
+        });
+        await session.run(fileQ.cypher, fileQ.params);
+        // Delete old children for this file (clean re-index)
+        await session.run(`
+      MATCH (f:File {path: $filePath})-[:CONTAINS]->(child)
+      OPTIONAL MATCH (child)-[:HAS_METHOD]->(method)
+      DETACH DELETE method, child
+      `, { filePath: meta.filePath });
+        // Upsert classes
+        for (const cls of entities.classes) {
+            const q = upsertClass({
+                name: cls.name,
+                filePath: meta.filePath,
+                startLine: cls.startLine,
+                endLine: cls.endLine,
+                docstring: cls.docstring,
+            });
+            await session.run(q.cypher, q.params);
+        }
+        // Upsert functions
+        for (const fn of entities.functions) {
+            const q = upsertFunction({
+                name: fn.name,
+                filePath: meta.filePath,
+                startLine: fn.startLine,
+                endLine: fn.endLine,
+                signature: fn.signature,
+                docstring: fn.docstring,
+                snippet: fn.snippet,
+                className: fn.className ?? undefined,
+            });
+            await session.run(q.cypher, q.params);
+        }
+        // Upsert call relationships
+        for (const call of entities.calls) {
+            const q = upsertCallRelationship(call);
+            await session.run(q.cypher, q.params);
+        }
+        // Upsert import relationships
+        for (const imp of entities.imports) {
+            const target = resolveImport(imp.source, meta.filePath, meta.language, filePathSet);
+            if (target) {
+                const q = upsertImportRelationship({ sourceFilePath: meta.filePath, targetFilePath: target });
+                await session.run(q.cypher, q.params);
+            }
+        }
+    }
+    finally {
+        await session.close();
+    }
+}
+//# sourceMappingURL=graph-writer.js.map

package/dist/indexer/graph-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph-writer.js","sourceRoot":"","sources":["../../src/indexer/graph-writer.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,gBAAgB,EAChB,UAAU,EACV,cAAc,EACd,WAAW,EACX,sBAAsB,EACtB,wBAAwB,GACzB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAWrD,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,EAAgB,EAAE,QAAgB;IACpE,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,EAAE,CAAC,CAAC;QAChG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAChD,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,EAAgB,EAChB,QAAuB,EACvB,IAAkB,EAClB,cAA2B,IAAI,GAAG,EAAE;IAEpC,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,IAAI,CAAC;QACH,cAAc;QACd,MAAM,KAAK,GAAG,UAAU,CAAC;YACvB,IAAI,EAAE,IAAI,CAAC,QAAQ;YACnB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,CAAC,CAAC;QACH,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAE9C,qDAAqD;QACrD,MAAM,OAAO,CAAC,GAAG,CACf;;;;OAIC,EACD,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAC5B,CAAC;QAEF,iBAAiB;QACjB,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,MAAM,CAAC,GAAG,WAAW,CAAC;gBACpB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,SAAS,EAAE,GAAG,CAAC,SAAS;aACzB,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,mBAAmB;QACnB,KAAK,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;YACpC,MAAM,CAAC,GAAG,cAAc,CAAC;gBACvB,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,SAAS,EAAE,EAAE,CAAC,SAAS;gBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;gBACnB,SAAS,EAAE,EAAE,CAAC,SAAS;gBACvB,SAAS,EAAE,EAAE,CAAC,SAAS;gBACvB,OAAO,EAAE,EAAE,CAAC,OAAO;gBACnB,SAAS,EAAE,EAAE,CAAC,SAAS,IAAI,SAAS;aACrC,CAAC,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,4BAA4B;QAC5B,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YAClC,MAAM,CAAC,GAAG,sBAAsB,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC;QAED,8BAA8B;QAC9B,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YACpF,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,GAAG,wBAAwB,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC9F,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;AACH,CAAC"}