rho-graph 0.1.0

package/dist/visualize/queries.js

@@ -0,0 +1,84 @@
+// src/visualize/queries.ts
+export const INITIAL_LIMIT = 500;
+export const EXPAND_FILE_LIMIT = 100;
+export const EXPAND_FUNCTION_LIMIT = 50;
+export const SEARCH_LIMIT = 25;
+export function repoOverview(repoName) {
+    return {
+        cypher: `
+      MATCH (r:Repository)
+      WHERE $repoName IS NULL OR r.name = $repoName
+      MATCH (r)-[contains:CONTAINS_FILE]->(f:File)
+      OPTIONAL MATCH (f)-[imp:IMPORTS]->(other:File)
+      WHERE (r)-[:CONTAINS_FILE]->(other)
+      RETURN r, contains, f, imp, other
+      LIMIT toInteger($limit)
+    `,
+        params: { repoName: repoName ?? null, limit: INITIAL_LIMIT },
+    };
+}
+export function filterByFile(relativePath) {
+    return {
+        cypher: `
+      MATCH (f:File {relativePath: $relativePath})
+      OPTIONAL MATCH (f)-[contains:CONTAINS]->(sym)
+      WHERE sym:Function OR sym:Class
+      OPTIONAL MATCH (sym)-[hasMethod:HAS_METHOD]->(method:Function)
+      RETURN f, contains, sym, hasMethod, method
+      LIMIT toInteger($limit)
+    `,
+        params: { relativePath, limit: INITIAL_LIMIT },
+    };
+}
+export function filterByFunction(name) {
+    return {
+        cypher: `
+      MATCH (fn:Function {name: $name})
+      OPTIONAL MATCH (file:File)-[contains:CONTAINS]->(fn)
+      OPTIONAL MATCH (fn)-[outCall:CALLS]->(callee:Function)
+      OPTIONAL MATCH (caller:Function)-[inCall:CALLS]->(fn)
+      RETURN fn, file, contains, outCall, callee, inCall, caller
+      LIMIT toInteger($limit)
+    `,
+        params: { name, limit: INITIAL_LIMIT },
+    };
+}
+export function expandFile(filePath) {
+    return {
+        cypher: `
+      MATCH (f:File {path: $filePath})
+      MATCH (f)-[contains:CONTAINS]->(sym)
+      WHERE sym:Function OR sym:Class
+      OPTIONAL MATCH (sym)-[hasMethod:HAS_METHOD]->(method:Function)
+      RETURN f, contains, sym, hasMethod, method
+      LIMIT toInteger($limit)
+    `,
+        params: { filePath, limit: EXPAND_FILE_LIMIT },
+    };
+}
+export function expandFunction(name, filePath) {
+    return {
+        cypher: `
+      MATCH (fn:Function {name: $name, filePath: $filePath})
+      OPTIONAL MATCH (fn)-[outCall:CALLS]->(callee:Function)
+      OPTIONAL MATCH (caller:Function)-[inCall:CALLS]->(fn)
+      RETURN fn, outCall, callee, inCall, caller
+      LIMIT toInteger($limit)
+    `,
+        params: { name, filePath, limit: EXPAND_FUNCTION_LIMIT },
+    };
+}
+export function searchByName(prefix) {
+    return {
+        cypher: `
+      CALL db.index.fulltext.queryNodes('code_search', $q) YIELD node, score
+      WITH node, score
+      WHERE node:Function OR node:Class
+      RETURN node
+      ORDER BY score DESC
+      LIMIT toInteger($limit)
+    `,
+        params: { q: `${prefix}*`, limit: SEARCH_LIMIT },
+    };
+}
+//# sourceMappingURL=queries.js.map

package/dist/visualize/queries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queries.js","sourceRoot":"","sources":["../../src/visualize/queries.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAO3B,MAAM,CAAC,MAAM,aAAa,GAAG,GAAG,CAAC;AACjC,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAG,CAAC;AACrC,MAAM,CAAC,MAAM,qBAAqB,GAAG,EAAE,CAAC;AACxC,MAAM,CAAC,MAAM,YAAY,GAAG,EAAE,CAAC;AAE/B,MAAM,UAAU,YAAY,CAAC,QAAiB;IAC5C,OAAO;QACL,MAAM,EAAE;;;;;;;;KAQP;QACD,MAAM,EAAE,EAAE,QAAQ,EAAE,QAAQ,IAAI,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE;KAC7D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,YAAoB;IAC/C,OAAO;QACL,MAAM,EAAE;;;;;;;KAOP;QACD,MAAM,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,OAAO;QACL,MAAM,EAAE;;;;;;;KAOP;QACD,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE;KACvC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,QAAgB;IACzC,OAAO;QACL,MAAM,EAAE;;;;;;;KAOP;QACD,MAAM,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,iBAAiB,EAAE;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAY,EAAE,QAAgB;IAC3D,OAAO;QACL,MAAM,EAAE;;;;;;KAMP;QACD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,qBAAqB,EAAE;KACzD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc;IACzC,OAAO;QACL,MAAM,EAAE;;;;;;;KAOP;QACD,MAAM,EAAE,EAAE,CAAC,EAAE,GAAG,MAAM,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE;KACjD,CAAC;AACJ,CAAC"}

package/dist/visualize/server.d.ts

@@ -0,0 +1,19 @@
+import type { Neo4jConfig } from "../config.js";
+interface VisualizationOptions {
+    neo4jConfig: Neo4jConfig;
+    port: number;
+    filter: {
+        repo?: string;
+        file?: string;
+        function?: string;
+    };
+}
+export declare function startVisualizationServer(opts: VisualizationOptions): Promise<void>;
+/**
+ * Test-only variant of startVisualizationServer:
+ *   - returns the Server instance so tests can close it
+ *   - does not call openBrowser
+ *   - throws (does not process.exit) on connection failure
+ */
+export declare function startVisualizationServerForTest(opts: VisualizationOptions): Promise<import("node:http").Server>;
+export {};

package/dist/visualize/server.js

@@ -0,0 +1,293 @@
+// src/visualize/server.ts
+import { createServer } from "node:http";
+import { readFileSync } from "node:fs";
+import { join, dirname, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+import { spawn } from "node:child_process";
+import neo4j from "neo4j-driver";
+import { repoOverview, filterByFile, filterByFunction, expandFile, expandFunction, searchByName, } from "./queries.js";
+const __dirname = dirname(fileURLToPath(import.meta.url));
+async function runCypher(driver, q) {
+    const session = driver.session();
+    try {
+        const result = await session.run(q.cypher, q.params, { timeout: 10000 });
+        const nodes = new Map();
+        const edges = [];
+        for (const record of result.records) {
+            for (const key of record.keys) {
+                const val = record.get(key);
+                if (val && typeof val === "object" && "identity" in val && "labels" in val) {
+                    const id = val.identity.toString();
+                    if (!nodes.has(id)) {
+                        nodes.set(id, {
+                            id,
+                            label: val.properties.name || val.properties.relativePath || id,
+                            group: val.labels?.[0] ?? "Unknown",
+                            properties: val.properties,
+                        });
+                    }
+                }
+                if (val && typeof val === "object" && "type" in val && "start" in val && "end" in val) {
+                    edges.push({
+                        from: val.start.toString(),
+                        to: val.end.toString(),
+                        label: val.type,
+                    });
+                }
+            }
+        }
+        return { nodes: Array.from(nodes.values()), edges };
+    }
+    finally {
+        await session.close();
+    }
+}
+function pickInitialQuery(opts, urlParams) {
+    // URL params take precedence over CLI flags so the browser can re-query without restart
+    const file = urlParams.get("file") ?? opts.filter.file;
+    const fn = urlParams.get("function") ?? opts.filter.function;
+    const repo = urlParams.get("repo") ?? opts.filter.repo;
+    if (file)
+        return filterByFile(file);
+    if (fn)
+        return filterByFunction(fn);
+    return repoOverview(repo);
+}
+async function handleRequest(res, url, driver, opts) {
+    const parsedUrl = new URL(url, "http://localhost");
+    const pathname = parsedUrl.pathname;
+    const params = parsedUrl.searchParams;
+    if (pathname === "/api/graph") {
+        try {
+            const data = await runCypher(driver, pickInitialQuery(opts, params));
+            res.writeHead(200, { "Content-Type": "application/json" });
+            res.end(JSON.stringify(data));
+        }
+        catch (err) {
+            console.error("[viz] /api/graph error:", err);
+            res.writeHead(500, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: String(err) }));
+        }
+        return;
+    }
+    if (pathname === "/api/expand") {
+        const type = params.get("type");
+        try {
+            let q;
+            if (type === "file") {
+                const filePath = params.get("filePath");
+                if (!filePath) {
+                    res.writeHead(400, { "Content-Type": "application/json" });
+                    res.end(JSON.stringify({ error: "missing filePath" }));
+                    return;
+                }
+                q = expandFile(filePath);
+            }
+            else if (type === "function") {
+                const name = params.get("name");
+                const filePath = params.get("filePath");
+                if (!name || !filePath) {
+                    res.writeHead(400, { "Content-Type": "application/json" });
+                    res.end(JSON.stringify({ error: "missing name or filePath" }));
+                    return;
+                }
+                q = expandFunction(name, filePath);
+            }
+            else {
+                res.writeHead(400, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ error: `unknown expand type: ${type}` }));
+                return;
+            }
+            const data = await runCypher(driver, q);
+            res.writeHead(200, { "Content-Type": "application/json" });
+            res.end(JSON.stringify(data));
+        }
+        catch (err) {
+            console.error("[viz] /api/expand error:", err);
+            res.writeHead(500, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: String(err) }));
+        }
+        return;
+    }
+    if (pathname === "/api/search") {
+        const q = params.get("q");
+        if (!q) {
+            res.writeHead(400, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: "missing q" }));
+            return;
+        }
+        try {
+            const data = await runCypher(driver, searchByName(q));
+            res.writeHead(200, { "Content-Type": "application/json" });
+            res.end(JSON.stringify(data));
+        }
+        catch (err) {
+            console.error("[viz] /api/search error:", err);
+            res.writeHead(500, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ error: String(err) }));
+        }
+        return;
+    }
+    // Serve static files
+    const publicDir = join(__dirname, "public");
+    const filePath = join(publicDir, pathname === "/" ? "index.html" : pathname);
+    const resolved = resolve(filePath);
+    const resolvedPublicDir = resolve(publicDir);
+    if (!resolved.startsWith(resolvedPublicDir + "/")) {
+        res.writeHead(403, { "Content-Type": "application/json" });
+        res.end(JSON.stringify({ error: "Forbidden" }));
+        return;
+    }
+    console.log(`[viz] serving file: ${filePath}`);
+    try {
+        const content = readFileSync(filePath);
+        const ext = filePath.endsWith(".js") ? "application/javascript" : "text/html";
+        console.log(`[viz] -> 200 ${ext} (${content.length} bytes)`);
+        res.writeHead(200, { "Content-Type": ext });
+        res.end(content);
+    }
+    catch (err) {
+        console.error(`[viz] -> 404: ${filePath}`, err instanceof Error ? err.message : err);
+        res.writeHead(404);
+        res.end("Not found");
+    }
+}
+export async function startVisualizationServer(opts) {
+    console.log(`[viz] connecting to Neo4j at ${opts.neo4jConfig.uri}...`);
+    const driver = neo4j.driver(opts.neo4jConfig.uri, neo4j.auth.basic(opts.neo4jConfig.username, opts.neo4jConfig.password), { connectionTimeout: 5000 });
+    // Verify Neo4j is reachable before starting the HTTP server.
+    try {
+        const pingSession = driver.session();
+        await pingSession.run("RETURN 1", {}, { timeout: 5000 });
+        await pingSession.close();
+        console.log("[viz] Neo4j connection OK");
+    }
+    catch (err) {
+        console.error("[viz] Neo4j connection FAILED:", err);
+        console.error(`[viz] Is Neo4j running at ${opts.neo4jConfig.uri}?`);
+        await driver.close();
+        process.exit(1);
+    }
+    const server = createServer((req, res) => {
+        const url = req.url ?? "/";
+        console.log(`[viz] ${req.method} ${url}`);
+        if (url === "/health") {
+            res.writeHead(200, { "Content-Type": "text/plain" });
+            res.end("ok");
+            return;
+        }
+        handleRequest(res, url, driver, opts).catch((err) => {
+            console.error("[viz] unhandled error:", err);
+            if (!res.headersSent) {
+                res.writeHead(500);
+                res.end("Internal server error");
+            }
+        });
+    });
+    server.on("connection", (socket) => {
+        console.log(`[viz] TCP connection from ${socket.remoteAddress}:${socket.remotePort}`);
+    });
+    server.on("error", (err) => {
+        console.error("[viz] server error:", err);
+    });
+    server.listen(opts.port, "0.0.0.0", () => {
+        const url = `http://localhost:${opts.port}`;
+        console.log(`Graph visualization running at ${url}`);
+        console.log(`[viz] listening on 0.0.0.0:${opts.port}`);
+        console.log(`[viz] serving static files from: ${join(__dirname, "public")}`);
+        openBrowser(url);
+    });
+}
+/**
+ * Test-only variant of startVisualizationServer:
+ *   - returns the Server instance so tests can close it
+ *   - does not call openBrowser
+ *   - throws (does not process.exit) on connection failure
+ */
+export async function startVisualizationServerForTest(opts) {
+    const driver = neo4j.driver(opts.neo4jConfig.uri, neo4j.auth.basic(opts.neo4jConfig.username, opts.neo4jConfig.password), { connectionTimeout: 5000 });
+    const pingSession = driver.session();
+    try {
+        await pingSession.run("RETURN 1", {}, { timeout: 5000 });
+    }
+    finally {
+        await pingSession.close();
+    }
+    const server = createServer((req, res) => {
+        const url = req.url ?? "/";
+        if (url === "/health") {
+            res.writeHead(200, { "Content-Type": "text/plain" });
+            res.end("ok");
+            return;
+        }
+        handleRequest(res, url, driver, opts).catch((err) => {
+            console.error("[viz-test] unhandled error:", err);
+            if (!res.headersSent) {
+                res.writeHead(500);
+                res.end("Internal server error");
+            }
+        });
+    });
+    return new Promise((resolve, reject) => {
+        server.once("error", reject);
+        server.listen(opts.port, "127.0.0.1", () => resolve(server));
+    });
+}
+/**
+ * Launch the user's browser to `url` without ever blocking the Node event loop.
+ *
+ * Why this exists: the previous implementation used `execFileSync`, which holds
+ * the main thread until the child exits. In WSL2 (and other headless Linux
+ * environments) `xdg-open` can hang indefinitely searching for a browser, which
+ * pins the event loop and starves the HTTP server of request callbacks even
+ * though the TCP listener is bound. We MUST spawn detached + ignore stdio +
+ * unref so this function returns immediately regardless of what the child does.
+ */
+function openBrowser(url) {
+    let command;
+    let args;
+    if (process.platform === "win32") {
+        // The empty "" is a placeholder window title — `start` interprets the
+        // first quoted arg as a title, and omitting it breaks URLs with spaces.
+        command = "cmd";
+        args = ["/c", "start", "", url];
+    }
+    else if (process.platform === "darwin") {
+        command = "open";
+        args = [url];
+    }
+    else if (isWSL()) {
+        // WSL2: hand off to the Windows default browser via cmd.exe.
+        // The empty "" is `start`'s window-title placeholder (see win32 branch).
+        // Falls back to a printed URL via the spawn `error` handler if cmd.exe
+        // isn't on PATH (e.g. interop.appendWindowsPath=false in /etc/wsl.conf).
+        command = "cmd.exe";
+        args = ["/c", "start", "", url];
+    }
+    else {
+        command = "xdg-open";
+        args = [url];
+    }
+    try {
+        const child = spawn(command, args, { detached: true, stdio: "ignore" });
+        child.on("error", () => {
+            console.log(`Open your browser to: ${url}`);
+        });
+        child.unref();
+    }
+    catch {
+        console.log(`Open your browser to: ${url}`);
+    }
+}
+function isWSL() {
+    if (process.platform !== "linux")
+        return false;
+    try {
+        const version = readFileSync("/proc/version", "utf8");
+        return /microsoft/i.test(version);
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=server.js.map

package/dist/visualize/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/visualize/server.ts"],"names":[],"mappings":"AAAA,0BAA0B;AAC1B,OAAO,EAAE,YAAY,EAA6C,MAAM,WAAW,CAAC;AACpF,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,KAAsB,MAAM,cAAc,CAAC;AAElD,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,UAAU,EACV,cAAc,EACd,YAAY,GAEb,MAAM,cAAc,CAAC;AAEtB,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAY1D,KAAK,UAAU,SAAS,CAAC,MAAc,EAAE,CAAc;IACrD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;IACjC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACzE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;QACxC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACpC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC5B,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,UAAU,IAAI,GAAG,IAAI,QAAQ,IAAI,GAAG,EAAE,CAAC;oBAC3E,MAAM,EAAE,GAAG,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;oBACnC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;wBACnB,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE;4BACZ,EAAE;4BACF,KAAK,EAAE,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,GAAG,CAAC,UAAU,CAAC,YAAY,IAAI,EAAE;4BAC/D,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,SAAS;4BACnC,UAAU,EAAE,GAAG,CAAC,UAAU;yBAC3B,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,IAAI,GAAG,IAAI,OAAO,IAAI,GAAG,IAAI,KAAK,IAAI,GAAG,EAAE,CAAC;oBACtF,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE;wBAC1B,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;wBACtB,KAAK,EAAE,GAAG,CAAC,IAAI;qBAChB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC;IACtD,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,IAA0B,EAAE,SAA0B;IAC9E,wFAAwF;IACxF,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IACvD,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;IAC7D,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAEvD,IAAI,IAAI;QAAE,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC;IACpC,IAAI,EAAE;QAAE,OAAO,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACpC,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,GAAmB,EACnB,GAAW,EACX,MAAc,EACd,IAA0B;IAE1B,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;IACpC,MAAM,MAAM,GAAG,SAAS,CAAC,YAAY,CAAC;IAEtC,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;YACrE,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAC9C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC;YACH,IAAI,CAAc,CAAC;YACnB,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;oBACvD,OAAO;gBACT,CAAC;gBACD,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC3B,CAAC;iBAAM,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAChC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACxC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACvB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,CAAC,CAAC,CAAC;oBAC/D,OAAO;gBACT,CAAC;gBACD,CAAC,GAAG,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,wBAAwB,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;gBACnE,OAAO;YACT,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACxC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YAC/C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QAC/B,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YAC/C,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;QAClD,CAAC;QACD,OAAO;IACT,CAAC;IAED,qBAAqB;IACrB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,EAAE,QAAQ,KAAK,GAAG,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC7E,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,MAAM,iBAAiB,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC7C,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,iBAAiB,GAAG,GAAG,CAAC,EAAE,CAAC;QAClD,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;QAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;QAChD,OAAO;IACT,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,uBAAuB,QAAQ,EAAE,CAAC,CAAC;IAE/C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,WAAW,CAAC;QAC9E,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,KAAK,OAAO,CAAC,MAAM,SAAS,CAAC,CAAC;QAC7D,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5C,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACnB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,iBAAiB,QAAQ,EAAE,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACrF,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACnB,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,IAA0B;IAE1B,OAAO,CAAC,GAAG,CAAC,gCAAgC,IAAI,CAAC,WAAW,CAAC,GAAG,KAAK,CAAC,CAAC;IACvE,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CACzB,IAAI,CAAC,WAAW,CAAC,GAAG,EACpB,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EACtE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAC5B,CAAC;IAEF,6DAA6D;IAC7D,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;QACrC,MAAM,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;QAC1B,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IAC3C,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC;QACrD,OAAO,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,CAAC;QACpE,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAoB,EAAE,GAAmB,EAAE,EAAE;QACxE,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC,CAAC;QAE1C,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YACrD,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,OAAO;QACT,CAAC;QAED,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YAClD,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC;YAC7C,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;gBACrB,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACnB,GAAG,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YACnC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;QACjC,OAAO,CAAC,GAAG,CAAC,6BAA6B,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;QACzB,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE;QACvC,MAAM,GAAG,GAAG,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,kCAAkC,GAAG,EAAE,CAAC,CAAC;QACrD,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,oCAAoC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC7E,WAAW,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,+BAA+B,CACnD,IAA0B;IAE1B,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CACzB,IAAI,CAAC,WAAW,CAAC,GAAG,EACpB,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EACtE,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAC5B,CAAC;IAEF,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;IACrC,IAAI,CAAC;QACH,MAAM,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3D,CAAC;YAAS,CAAC;QACT,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAoB,EAAE,GAAmB,EAAE,EAAE;QACxE,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC;QAC3B,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YACrD,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACd,OAAO;QACT,CAAC;QACD,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YAClD,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAC;YAClD,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;gBACrB,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACnB,GAAG,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YACnC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,WAAW,CAAC,GAAW;IAC9B,IAAI,OAAe,CAAC;IACpB,IAAI,IAAc,CAAC;IAEnB,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,sEAAsE;QACtE,wEAAwE;QACxE,OAAO,GAAG,KAAK,CAAC;QAChB,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;IAClC,CAAC;SAAM,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACzC,OAAO,GAAG,MAAM,CAAC;QACjB,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;SAAM,IAAI,KAAK,EAAE,EAAE,CAAC;QACnB,6DAA6D;QAC7D,yEAAyE;QACzE,uEAAuE;QACvE,yEAAyE;QACzE,OAAO,GAAG,SAAS,CAAC;QACpB,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;IAClC,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,UAAU,CAAC;QACrB,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxE,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACrB,OAAO,CAAC,GAAG,CAAC,yBAAyB,GAAG,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,GAAG,CAAC,yBAAyB,GAAG,EAAE,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,SAAS,KAAK;IACZ,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC;IAC/C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,YAAY,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QACtD,OAAO,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/docker-compose.yml

@@ -0,0 +1,16 @@
+services:
+  neo4j:
+    image: neo4j:5-community
+    container_name: code-graph-rag-neo4j
+    ports:
+      - "7474:7474"
+      - "7687:7687"
+    environment:
+      NEO4J_AUTH: neo4j/code-graph-rag
+      NEO4J_PLUGINS: '[]'
+    volumes:
+      - code-graph-rag-data:/data
+    restart: unless-stopped
+
+volumes:
+  code-graph-rag-data:

package/package.json

@@ -0,0 +1,69 @@
+{
+  "name": "rho-graph",
+  "version": "0.1.0",
+  "description": "Graph-RAG code indexer for AI agents — token-efficient code search via MCP tools backed by Neo4j",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "rho-graph": "bin/rho-graph.js"
+  },
+  "scripts": {
+    "build": "tsc && cp src/indexer/language-map.json dist/indexer/language-map.json && cp -r src/visualize/public dist/visualize/",
+    "dev": "tsc --watch",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "lint": "tsc --noEmit"
+  },
+  "keywords": [
+    "mcp",
+    "graph-rag",
+    "code-search",
+    "neo4j",
+    "tree-sitter",
+    "ai",
+    "claude",
+    "cursor",
+    "llm",
+    "code-intelligence"
+  ],
+  "author": "rhofield",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/rhofield/rho-graph.git"
+  },
+  "homepage": "https://github.com/rhofield/rho-graph#readme",
+  "bugs": {
+    "url": "https://github.com/rhofield/rho-graph/issues"
+  },
+  "dependencies": {
+    "@anthropic-ai/sdk": "^0.39.0",
+    "@modelcontextprotocol/sdk": "^1.12.1",
+    "commander": "^12.1.0",
+    "glob": "^11.0.1",
+    "neo4j-driver": "^5.27.0",
+    "ora": "^8.1.1",
+    "tree-sitter-wasms": "^0.1.13",
+    "web-tree-sitter": "^0.24.3",
+    "zod": "^3.23.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.0",
+    "typescript": "^5.7.0",
+    "vitest": "^2.1.0"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "files": [
+    "dist",
+    "bin",
+    "src/indexer/language-map.json",
+    "src/visualize/public",
+    "docker-compose.yml"
+  ],
+  "directories": {
+    "doc": "docs",
+    "test": "tests"
+  }
+}

package/src/indexer/language-map.json

@@ -0,0 +1,128 @@
+{
+  "typescript": {
+    "function": ["function_declaration", "arrow_function", "method_definition"],
+    "class": ["class_declaration"],
+    "import": ["import_statement"],
+    "call": ["call_expression"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "tsx": {
+    "function": ["function_declaration", "arrow_function", "method_definition"],
+    "class": ["class_declaration"],
+    "import": ["import_statement"],
+    "call": ["call_expression"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "javascript": {
+    "function": ["function_declaration", "arrow_function", "method_definition"],
+    "class": ["class_declaration"],
+    "import": ["import_statement"],
+    "call": ["call_expression"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "python": {
+    "function": ["function_definition"],
+    "class": ["class_definition"],
+    "import": ["import_from_statement", "import_statement"],
+    "call": ["call"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "go": {
+    "function": ["function_declaration", "method_declaration"],
+    "class": ["type_declaration"],
+    "import": ["import_declaration", "import_spec"],
+    "call": ["call_expression"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "java": {
+    "function": ["method_declaration", "constructor_declaration"],
+    "class": ["class_declaration", "interface_declaration", "enum_declaration"],
+    "import": ["import_declaration"],
+    "call": ["method_invocation"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "rust": {
+    "function": ["function_item"],
+    "class": ["struct_item", "enum_item", "trait_item", "impl_item"],
+    "import": ["use_declaration"],
+    "call": ["call_expression"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "c": {
+    "function": ["function_definition"],
+    "class": ["struct_specifier"],
+    "import": ["preproc_include"],
+    "call": ["call_expression"],
+    "name_field": "declarator",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "cpp": {
+    "function": ["function_definition"],
+    "class": ["class_specifier", "struct_specifier"],
+    "import": ["preproc_include"],
+    "call": ["call_expression"],
+    "name_field": "declarator",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "ruby": {
+    "function": ["method", "singleton_method"],
+    "class": ["class", "module"],
+    "import": ["call"],
+    "call": ["call", "method_call"],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "graphql": {
+    "function": ["operation_definition", "fragment_definition"],
+    "class": ["object_type_definition", "input_object_type_definition", "interface_type_definition", "enum_type_definition"],
+    "import": [],
+    "call": [],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "arguments_definition"
+  },
+  "yaml": {
+    "function": [],
+    "class": [],
+    "import": [],
+    "call": [],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "json": {
+    "function": [],
+    "class": [],
+    "import": [],
+    "call": [],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  },
+  "markdown": {
+    "function": [],
+    "class": [],
+    "import": [],
+    "call": [],
+    "name_field": "name",
+    "body_field": "body",
+    "parameters_field": "parameters"
+  }
+}