fusion/packages/core/src/knowledge-graph/derive-modules.ts

import { dirname, basename } from "node:path";
import { edgeId, fileNodeId, moduleCanonicalPath, moduleNodeId, type GraphEdge, type GraphNode } from "./graph-types.js";
/** Modules are recomputed from files: assigning a multi-file module to one owner would make pruning unsound. */
export function deriveModules(files:GraphNode[]):{nodes:GraphNode[];edges:GraphEdge[]}{const paths=files.filter(n=>n.kind==="file"&&/\.tsx?$/.test(n.ownerPath)).map(n=>n.ownerPath),dirs=[...new Set(paths.map(dirname).filter(dir=>dir!=="."))].sort(),nodes:GraphNode[]=[],edges:GraphEdge[]=[];for(const dir of dirs){const direct=paths.filter(p=>dirname(p)===dir),canonical=moduleCanonicalPath(dir,paths),id=moduleNodeId(dir),attributes={directFileCount:String(direct.length),barrel:String(paths.includes(`${dir}/index.ts`)||paths.includes(`${dir}/index.tsx`)),syntheticSource:"true"};nodes.push({id,kind:"module",name:basename(dir),owner:"derived",ownerPath:dir,source:{path:canonical,line:1,column:1},attributes});for(const file of direct)edges.push({id:edgeId("contains",id,fileNodeId(file)),kind:"contains",from:id,to:fileNodeId(file),provenance:"extracted",owner:"derived",ownerPath:dir,source:{path:canonical,line:1,column:1},attributes:{syntheticSource:"true"}});const parent=dirs.filter(d=>dir.startsWith(`${d}/`)).sort((a,b)=>b.length-a.length)[0];if(parent){const parentNode=moduleNodeId(parent),parentCanonical=moduleCanonicalPath(parent,paths);edges.push({id:edgeId("contains",parentNode,id),kind:"contains",from:parentNode,to:id,provenance:"extracted",owner:"derived",ownerPath:parent,source:{path:parentCanonical,line:1,column:1},attributes:{syntheticSource:"true"}});}}return{nodes:nodes.sort((a,b)=>a.id.localeCompare(b.id)),edges:edges.sort((a,b)=>a.id.localeCompare(b.id))};}