@dreamtree-org/graphify 1.1.3 → 1.3.0

package/dist/index.cjs

@@ -34,6 +34,7 @@ __export(index_exports, {
   ExtractionCache: () => ExtractionCache,
   ExtractionResultSchema: () => ExtractionResultSchema,
   ExtractionValidationError: () => ExtractionValidationError,
+  LocalFileGraphStore: () => LocalFileGraphStore,
   affectedBy: () => affectedBy,
   analyze: () => analyze,
   buildContextPack: () => buildContextPack,
@@ -41,6 +42,7 @@ __export(index_exports, {
   checkRules: () => checkRules,
   cluster: () => cluster,
   collectFiles: () => collectFiles,
+  deserializeGraph: () => deserializeGraph,
   diffGraphs: () => diffGraphs,
   explainNode: () => explainNode,
   exportGraph: () => exportGraph,
@@ -64,6 +66,7 @@ __export(index_exports, {
   runPipeline: () => runPipeline,
   saveResult: () => saveResult,
   scoreNodes: () => scoreNodes,
+  serializeGraph: () => serializeGraph,
   shortestPath: () => shortestPath,
   testsForChangedFiles: () => testsForChangedFiles,
   testsForNode: () => testsForNode,
@@ -2570,7 +2573,7 @@ function renderReport(graph, analysis, now = /* @__PURE__ */ new Date()) {
 
 // src/export.ts
 var import_node_module2 = require("module");
-var fs11 = __toESM(require("fs/promises"), 1);
+var fs12 = __toESM(require("fs/promises"), 1);
 var path6 = __toESM(require("path"), 1);
 
 // src/security.ts
@@ -2645,6 +2648,44 @@ function escapeHtml(text) {
   return text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&#39;");
 }
 
+// src/store/localFile.ts
+var fs11 = __toESM(require("fs/promises"), 1);
+
+// src/store/serialize.ts
+var import_graphology3 = __toESM(require("graphology"), 1);
+function serializeGraph(graph) {
+  return graph.export();
+}
+function deserializeGraph(data) {
+  return import_graphology3.default.from(data);
+}
+
+// src/store/localFile.ts
+var 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 fs11.readFile(jsonPath, "utf-8");
+    } catch (error) {
+      if (error.code === "ENOENT") return null;
+      throw error;
+    }
+    return deserializeGraph(JSON.parse(raw));
+  }
+  async save(ref, graph) {
+    await fs11.mkdir(ref, { recursive: true });
+    const jsonPath = validateGraphPath("graph.json", ref);
+    await fs11.writeFile(jsonPath, JSON.stringify(serializeGraph(graph), null, 2), "utf-8");
+  }
+};
+
 // src/export.ts
 var require3 = (0, import_node_module2.createRequire)(importMetaUrl);
 function resolveVisNetworkAssets() {
@@ -2711,8 +2752,8 @@ community: ${communityLabel}` : ""}`,
 async function renderHtml(graph) {
   const { jsPath, cssPath } = resolveVisNetworkAssets();
   const [visJs, visCss] = await Promise.all([
-    fs11.readFile(jsPath, "utf-8"),
-    fs11.readFile(cssPath, "utf-8")
+    fs12.readFile(jsPath, "utf-8"),
+    fs12.readFile(cssPath, "utf-8")
   ]);
   const { nodes, edges } = buildVisNodesAndEdges(graph);
   const dataJson = safeInlineJson({ nodes, edges });
@@ -2753,16 +2794,15 @@ ${visJs}
 }
 async function exportGraph(graph, options, report) {
   const outDir = path6.resolve(options.outDir);
-  await fs11.mkdir(outDir, { recursive: true });
-  const jsonPath = validateGraphPath("graph.json", outDir);
-  await fs11.writeFile(jsonPath, JSON.stringify(graph.toJSON(), null, 2), "utf-8");
+  await fs12.mkdir(outDir, { recursive: true });
+  await new LocalFileGraphStore().save(outDir, graph);
   if (options.html !== false) {
     const htmlPath = validateGraphPath("graph.html", outDir);
-    await fs11.writeFile(htmlPath, await renderHtml(graph), "utf-8");
+    await fs12.writeFile(htmlPath, await renderHtml(graph), "utf-8");
   }
   if (report !== void 0) {
     const reportPath = validateGraphPath("GRAPH_REPORT.md", outDir);
-    await fs11.writeFile(reportPath, report, "utf-8");
+    await fs12.writeFile(reportPath, report, "utf-8");
   }
   const notImplemented = [];
   if (options.svg) notImplemented.push("--svg");
@@ -2775,12 +2815,12 @@ async function exportGraph(graph, options, report) {
 }
 
 // src/pipeline.ts
-var fs13 = __toESM(require("fs/promises"), 1);
+var fs14 = __toESM(require("fs/promises"), 1);
 var path8 = __toESM(require("path"), 1);
 
 // src/extractionCache.ts
 var import_node_crypto3 = require("crypto");
-var fs12 = __toESM(require("fs/promises"), 1);
+var fs13 = __toESM(require("fs/promises"), 1);
 var path7 = __toESM(require("path"), 1);
 var ExtractionCache = class {
   dir;
@@ -2792,7 +2832,7 @@ var ExtractionCache = class {
   }
   async get(key) {
     try {
-      const raw = await fs12.readFile(path7.join(this.dir, `${key}.json`), "utf-8");
+      const raw = await fs13.readFile(path7.join(this.dir, `${key}.json`), "utf-8");
       const parsed = JSON.parse(raw);
       if (!Array.isArray(parsed.nodes) || !Array.isArray(parsed.edges)) return null;
       return parsed;
@@ -2801,8 +2841,8 @@ var ExtractionCache = class {
     }
   }
   async put(key, extraction) {
-    await fs12.mkdir(this.dir, { recursive: true });
-    await fs12.writeFile(path7.join(this.dir, `${key}.json`), JSON.stringify(extraction), "utf-8");
+    await fs13.mkdir(this.dir, { recursive: true });
+    await fs13.writeFile(path7.join(this.dir, `${key}.json`), JSON.stringify(extraction), "utf-8");
   }
 };
 
@@ -2824,7 +2864,7 @@ async function runPipeline(root, options = {}) {
   for (const relFile of manifest.files.code.sort((a, b) => a.localeCompare(b))) {
     const filePath = path8.relative(process.cwd(), path8.join(resolvedRoot, relFile)) || relFile;
     try {
-      const key = cache.key(relFile, await fs13.readFile(path8.join(resolvedRoot, relFile)));
+      const key = cache.key(relFile, await fs14.readFile(path8.join(resolvedRoot, relFile)));
       let extraction = options.update ? await cache.get(key) : null;
       if (extraction) {
         cacheHits++;
@@ -2874,19 +2914,23 @@ 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 };
 }
 
 // src/graphStore.ts
-var fs14 = __toESM(require("fs/promises"), 1);
 var path9 = __toESM(require("path"), 1);
-var import_graphology3 = __toESM(require("graphology"), 1);
 async function loadGraph(outDir = path9.join(process.cwd(), "graphify-out")) {
-  const jsonPath = validateGraphPath("graph.json", outDir);
-  const raw = await fs14.readFile(jsonPath, "utf-8");
-  const data = JSON.parse(raw);
-  return import_graphology3.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;
 }
 
 // src/query.ts
@@ -3951,6 +3995,7 @@ function checkRules(graph, config) {
   ExtractionCache,
   ExtractionResultSchema,
   ExtractionValidationError,
+  LocalFileGraphStore,
   affectedBy,
   analyze,
   buildContextPack,
@@ -3958,6 +4003,7 @@ function checkRules(graph, config) {
   checkRules,
   cluster,
   collectFiles,
+  deserializeGraph,
   diffGraphs,
   explainNode,
   exportGraph,
@@ -3981,6 +4027,7 @@ function checkRules(graph, config) {
   runPipeline,
   saveResult,
   scoreNodes,
+  serializeGraph,
   shortestPath,
   testsForChangedFiles,
   testsForNode,