@dreamtree-org/graphify 1.1.3 → 1.3.0

package/dist/cli/index.cjs

@@ -246,22 +246,73 @@ var init_security = __esm({
   }
 });
 
+// src/store/serialize.ts
+function serializeGraph(graph) {
+  return graph.export();
+}
+function deserializeGraph(data) {
+  return import_graphology2.default.from(data);
+}
+var import_graphology2;
+var init_serialize = __esm({
+  "src/store/serialize.ts"() {
+    "use strict";
+    init_cjs_shims();
+    import_graphology2 = __toESM(require("graphology"), 1);
+  }
+});
+
+// src/store/localFile.ts
+var fs2, LocalFileGraphStore;
+var init_localFile = __esm({
+  "src/store/localFile.ts"() {
+    "use strict";
+    init_cjs_shims();
+    fs2 = __toESM(require("fs/promises"), 1);
+    init_security();
+    init_serialize();
+    LocalFileGraphStore = class {
+      async load(ref) {
+        let jsonPath;
+        try {
+          jsonPath = validateGraphPath("graph.json", ref);
+        } catch (error) {
+          if (error.message.startsWith("Base directory does not exist")) return null;
+          throw error;
+        }
+        let raw;
+        try {
+          raw = await fs2.readFile(jsonPath, "utf-8");
+        } catch (error) {
+          if (error.code === "ENOENT") return null;
+          throw error;
+        }
+        return deserializeGraph(JSON.parse(raw));
+      }
+      async save(ref, graph) {
+        await fs2.mkdir(ref, { recursive: true });
+        const jsonPath = validateGraphPath("graph.json", ref);
+        await fs2.writeFile(jsonPath, JSON.stringify(serializeGraph(graph), null, 2), "utf-8");
+      }
+    };
+  }
+});
+
 // src/graphStore.ts
 async function loadGraph(outDir = path2.join(process.cwd(), "graphify-out")) {
-  const jsonPath = validateGraphPath("graph.json", outDir);
-  const raw = await fs3.readFile(jsonPath, "utf-8");
-  const data = JSON.parse(raw);
-  return import_graphology2.default.from(data);
+  const graph = await new LocalFileGraphStore().load(outDir);
+  if (graph === null) {
+    throw new Error(`No graph found in ${outDir} \u2014 run \`graphify <path>\` to build one first.`);
+  }
+  return graph;
 }
-var fs3, path2, import_graphology2;
+var path2;
 var init_graphStore = __esm({
   "src/graphStore.ts"() {
     "use strict";
     init_cjs_shims();
-    fs3 = __toESM(require("fs/promises"), 1);
     path2 = __toESM(require("path"), 1);
-    import_graphology2 = __toESM(require("graphology"), 1);
-    init_security();
+    init_localFile();
   }
 });
 
@@ -1355,9 +1406,10 @@ function cluster(graph, options = {}) {
 // src/export.ts
 init_cjs_shims();
 var import_node_module = require("module");
-var fs2 = __toESM(require("fs/promises"), 1);
+var fs3 = __toESM(require("fs/promises"), 1);
 var path = __toESM(require("path"), 1);
 init_security();
+init_localFile();
 var require2 = (0, import_node_module.createRequire)(importMetaUrl);
 function resolveVisNetworkAssets() {
   const packageJsonPath = require2.resolve("vis-network/package.json");
@@ -1423,8 +1475,8 @@ community: ${communityLabel}` : ""}`,
 async function renderHtml(graph) {
   const { jsPath, cssPath } = resolveVisNetworkAssets();
   const [visJs, visCss] = await Promise.all([
-    fs2.readFile(jsPath, "utf-8"),
-    fs2.readFile(cssPath, "utf-8")
+    fs3.readFile(jsPath, "utf-8"),
+    fs3.readFile(cssPath, "utf-8")
   ]);
   const { nodes, edges } = buildVisNodesAndEdges(graph);
   const dataJson = safeInlineJson({ nodes, edges });
@@ -1465,16 +1517,15 @@ ${visJs}
 }
 async function exportGraph(graph, options, report) {
   const outDir = path.resolve(options.outDir);
-  await fs2.mkdir(outDir, { recursive: true });
-  const jsonPath = validateGraphPath("graph.json", outDir);
-  await fs2.writeFile(jsonPath, JSON.stringify(graph.toJSON(), null, 2), "utf-8");
+  await fs3.mkdir(outDir, { recursive: true });
+  await new LocalFileGraphStore().save(outDir, graph);
   if (options.html !== false) {
     const htmlPath = validateGraphPath("graph.html", outDir);
-    await fs2.writeFile(htmlPath, await renderHtml(graph), "utf-8");
+    await fs3.writeFile(htmlPath, await renderHtml(graph), "utf-8");
   }
   if (report !== void 0) {
     const reportPath = validateGraphPath("GRAPH_REPORT.md", outDir);
-    await fs2.writeFile(reportPath, report, "utf-8");
+    await fs3.writeFile(reportPath, report, "utf-8");
   }
   const notImplemented = [];
   if (options.svg) notImplemented.push("--svg");
@@ -3912,6 +3963,11 @@ async function runPipeline(root, options = {}) {
     },
     report
   );
+  if (options.store) {
+    const storeRef = options.storeRef ?? outDir;
+    progress(`Saving graph to store (${storeRef}) ...`);
+    await options.store.save(storeRef, graph);
+  }
   progress("Done.");
   return { manifest, graph, analysis, report };
 }
@@ -4778,6 +4834,29 @@ function packageRoot() {
   const packageJsonPath = require4.resolve("@dreamtree-org/graphify/package.json");
   return path15.dirname(packageJsonPath);
 }
+var MCP_SERVER_ENTRY = {
+  command: "npx",
+  args: ["-y", "@dreamtree-org/graphify", "--mcp"]
+};
+async function upsertMcpServer(filePath, serversKey = "mcpServers") {
+  let config = {};
+  try {
+    config = JSON.parse(await fs22.readFile(filePath, "utf-8"));
+  } catch (error) {
+    if (error.code !== "ENOENT") {
+      throw new Error(`cannot update ${filePath} \u2014 it exists but is not valid JSON; fix or remove it and re-run.`, {
+        cause: error
+      });
+    }
+  }
+  const servers = config[serversKey] ?? {};
+  servers["graphify"] = MCP_SERVER_ENTRY;
+  config[serversKey] = servers;
+  await fs22.mkdir(path15.dirname(filePath), { recursive: true });
+  await fs22.writeFile(filePath, `${JSON.stringify(config, null, 2)}
+`, "utf-8");
+  return filePath;
+}
 var MD_MARKER_BEGIN = "<!-- >>> graphify >>> -->";
 var MD_MARKER_END = "<!-- <<< graphify <<< -->";
 async function writeOwnedFile(filePath, content) {
@@ -4812,7 +4891,8 @@ var PLATFORMS = {
   claude: async (cwd, content) => {
     const target = path15.join(cwd, ".claude", "skills", "graphify", "SKILL.md");
     await writeOwnedFile(target, content);
-    return { host: "Claude Code", path: target };
+    const mcpPath = await upsertMcpServer(path15.join(cwd, ".mcp.json"));
+    return { host: "Claude Code", path: target, mcpPath };
   },
   cursor: async (cwd, content) => {
     const target = path15.join(cwd, ".cursor", "rules", "graphify.mdc");
@@ -4825,7 +4905,8 @@ alwaysApply: false
 ---
 ${body}`
     );
-    return { host: "Cursor", path: target };
+    const mcpPath = await upsertMcpServer(path15.join(cwd, ".cursor", "mcp.json"));
+    return { host: "Cursor", path: target, mcpPath };
   },
   windsurf: async (cwd, content) => {
     const target = path15.join(cwd, ".windsurf", "rules", "graphify.md");
@@ -4845,10 +4926,32 @@ ${body}`
   gemini: async (cwd, content) => {
     const target = path15.join(cwd, "GEMINI.md");
     await upsertMarkedBlock(target, content.replace(/^---[\s\S]*?---\n/, ""));
-    return { host: "Gemini CLI", path: target };
+    const mcpPath = await upsertMcpServer(path15.join(cwd, ".gemini", "settings.json"));
+    return { host: "Gemini CLI", path: target, mcpPath };
   }
 };
 var SUPPORTED_PLATFORMS = Object.keys(PLATFORMS).sort((a, b) => a.localeCompare(b));
+var PLATFORM_MARKERS = {
+  cursor: [".cursor"],
+  windsurf: [".windsurf"],
+  cline: [".clinerules"],
+  agents: ["AGENTS.md"],
+  gemini: ["GEMINI.md", ".gemini"]
+};
+async function detectPlatforms(cwd = process.cwd()) {
+  const platforms = ["claude"];
+  for (const [platform, markers] of Object.entries(PLATFORM_MARKERS)) {
+    for (const marker of markers) {
+      try {
+        await fs22.access(path15.join(cwd, marker));
+        platforms.push(platform);
+        break;
+      } catch {
+      }
+    }
+  }
+  return platforms;
+}
 async function runInstall(platforms = ["claude"], cwd = process.cwd()) {
   const sourcePath = path15.join(packageRoot(), "src", "skill", "SKILL.md");
   const content = await fs22.readFile(sourcePath, "utf-8");
@@ -4864,7 +4967,8 @@ async function runInstall(platforms = ["claude"], cwd = process.cwd()) {
     results.push(await installer(cwd, content));
   }
   for (const result of results) {
-    console.log(`Installed graphify for ${result.host} -> ${result.path}`);
+    const mcpNote = result.mcpPath ? ` (MCP server registered in ${result.mcpPath})` : "";
+    console.log(`Installed graphify for ${result.host} -> ${result.path}${mcpNote}`);
   }
   return results;
 }
@@ -4980,7 +5084,7 @@ async function watchAndRebuild(root, runOnce) {
   });
 }
 var program = new import_commander.Command();
-program.name("graphify").description("Turn a folder of code/docs/papers into a queryable knowledge graph.").argument("[path]", "path to scan, or a GitHub URL", ".").option("--mode <mode>", "extraction mode (deep for richer INFERRED edges \u2014 reserved, no-op in v1)").option("--update", "incremental rebuild \u2014 reuse cached extractions for unchanged files").option("--watch", "rebuild on file change").option("--cluster-only", "rerun clustering on an existing graph").option("--leiden", "use Leiden instead of Louvain for community detection (v2 \u2014 better community quality)").option("--max-iterations <n>", "Leiden only: cap on outer iterations for very large graphs", Number).option("--no-viz", "skip graph.html").option("--svg", "also export graph.svg (not implemented in v1)").option("--graphml", "also export graph.graphml (not implemented in v1)").option("--neo4j", "generate graphify-out/cypher.txt for Neo4j (not implemented in v1)").option("--mcp", "start MCP stdio server instead of running the pipeline").option("--mysql <dsn>", "also extract a MySQL schema (mysql://user:pass@host:port/db) into the graph").action(async (targetArg, options) => {
+program.name("graphify").description("Turn a folder of code/docs/papers into a queryable knowledge graph.").argument("[path]", "path to scan, or a GitHub URL", ".").option("--mode <mode>", "extraction mode (deep for richer INFERRED edges \u2014 reserved, no-op in v1)").option("--update", "incremental rebuild \u2014 reuse cached extractions for unchanged files").option("--watch", "rebuild on file change").option("--cluster-only", "rerun clustering on an existing graph").option("--leiden", "use Leiden instead of Louvain for community detection (v2 \u2014 better community quality)").option("--max-iterations <n>", "Leiden only: cap on outer iterations for very large graphs", Number).option("--no-viz", "skip graph.html").option("--svg", "also export graph.svg (not implemented in v1)").option("--graphml", "also export graph.graphml (not implemented in v1)").option("--neo4j", "generate graphify-out/cypher.txt for Neo4j (not implemented in v1)").option("--mcp", "start MCP stdio server instead of running the pipeline").option("--no-install", "skip auto-installing the agent skill + MCP config after the build").option("--mysql <dsn>", "also extract a MySQL schema (mysql://user:pass@host:port/db) into the graph").action(async (targetArg, options) => {
   try {
     if (options.mcp) {
       const outDir = path17.resolve(targetArg === "." ? process.cwd() : targetArg, "graphify-out");
@@ -4989,7 +5093,8 @@ program.name("graphify").description("Turn a folder of code/docs/papers into a q
       return;
     }
     let root = targetArg;
-    if (looksLikeGitUrl(targetArg)) {
+    const cloned = looksLikeGitUrl(targetArg);
+    if (cloned) {
       console.error(`Cloning ${targetArg} ...`);
       root = await cloneRepo(targetArg);
     }
@@ -5015,6 +5120,13 @@ program.name("graphify").description("Turn a folder of code/docs/papers into a q
       onProgress: (m) => console.error(m)
     });
     await runOnce();
+    if (options.install && !cloned) {
+      try {
+        await runInstall(await detectPlatforms(root), root);
+      } catch (error) {
+        console.error(`graphify: agent skill/MCP install skipped: ${error.message}`);
+      }
+    }
     if (options.watch) {
       console.error(`Watching ${root} for changes (Ctrl+C to stop) ...`);
       await watchAndRebuild(root, runOnce);
@@ -5096,7 +5208,7 @@ program.command("benchmark [questions...]").description("measure token savings o
     process.exitCode = 1;
   }
 });
-program.command("install").description("install the skill/rules files into local agents (claude, cursor, windsurf, cline, agents, gemini, all)").option("--platform <names...>", "platform(s) to install for", ["claude"]).action(async (options) => {
+program.command("install").description("install the skill/rules files + project MCP config into local agents (claude, cursor, windsurf, cline, agents, gemini, all)").option("--platform <names...>", "platform(s) to install for", ["claude"]).action(async (options) => {
   try {
     await runInstall(options.platform);
   } catch (error) {