@dreamtree-org/graphify 1.3.0 → 1.5.0

package/dist/cli/index.cjs

@@ -1564,13 +1564,15 @@ var RelationSchema = import_zod.z.enum([
   "references",
   "contains",
   "method",
-  "re_exports"
+  "re_exports",
+  "follows"
 ]);
 var GraphNodeSchema = import_zod.z.object({
   id: import_zod.z.string().min(1, "node id must be non-empty"),
   label: import_zod.z.string(),
   sourceFile: import_zod.z.string(),
-  sourceLocation: import_zod.z.string()
+  sourceLocation: import_zod.z.string(),
+  kind: import_zod.z.string().optional()
 });
 var GraphEdgeSchema = import_zod.z.object({
   source: import_zod.z.string().min(1, "edge source must be non-empty"),
@@ -1618,16 +1620,27 @@ function buildGraph(extractions) {
   const validated = extractions.map(
     (extraction, index) => validateExtraction(extraction, `extraction #${index}`)
   );
+  mergeExtractionsInto(graph, validated);
+  return graph;
+}
+function mergeExtractionsInto(graph, validated) {
   const allNodes = validated.flatMap((extraction) => extraction.nodes);
   const allEdges = validated.flatMap((extraction) => extraction.edges);
   const sortedNodes = [...allNodes].sort((a, b) => a.id.localeCompare(b.id));
   for (const node of sortedNodes) {
-    if (graph.hasNode(node.id)) continue;
-    graph.addNode(node.id, {
+    const attributes = {
       label: node.label,
       sourceFile: node.sourceFile,
-      sourceLocation: node.sourceLocation
-    });
+      sourceLocation: node.sourceLocation,
+      ...node.kind !== void 0 ? { kind: node.kind } : {}
+    };
+    if (graph.hasNode(node.id)) {
+      if (graph.getNodeAttribute(node.id, "sourceFile") === "<unknown>") {
+        graph.mergeNodeAttributes(node.id, attributes);
+      }
+      continue;
+    }
+    graph.addNode(node.id, attributes);
   }
   const sortedEdges = [...allEdges].sort(
     (a, b) => a.source.localeCompare(b.source) || a.target.localeCompare(b.target) || a.relation.localeCompare(b.relation)
@@ -1646,7 +1659,6 @@ function buildGraph(extractions) {
       confidence: edge.confidence
     });
   }
-  return graph;
 }
 
 // src/extractionCache.ts