refactor utils

packages/utils/src/dag.ts

@@ -0,0 +1,76 @@
+interface Node {
+  _id: string
+  requirements: string[]
+}
+
+export default class DAG {
+  private nodes = new Map()
+  private edges = new Set()
+  private layers: number[] = []
+
+  constructor(nodes: Node[]) {
+    this.nodes = new Map(nodes.map((node) => [node._id, node]))
+    this.edges = new Set()
+    this.nodes.forEach((_, id) => {
+      this.setLayer(id)
+      this.addEdges(id)
+    })
+  }
+
+  getNode(id: string) {
+    return this.nodes.get(id)
+  }
+
+  getNodes() {
+    return Array.from(this.nodes.values())
+  }
+
+  getEdges() {
+    return this.edges
+  }
+
+  getEdgesArray() {
+    return Array.from(this.edges.values())
+  }
+
+  getLayers() {
+    return this.layers
+  }
+
+  getLayer(id: string) {
+    const node = this.getNode(id)
+    if (!node) {
+      throw new Error(`Node ${id} not found`)
+    }
+    return this.layers[node.layer - 1]
+  }
+
+  private setLayer(id: string) {
+    const node = this.getNode(id)
+    if (!node) {
+      throw new Error(`Node ${id} not found`)
+    }
+    if (node.layer !== undefined) {
+      return node.layer
+    }
+    const previousLayer = node.requirements.map((rId: string) => this.setLayer(rId))
+    node.layer = Math.max(...previousLayer, 0) + 1
+    if (this.layers.length < node.layer) {
+      this.layers.push(1)
+    } else {
+      this.layers[node.layer - 1] += 1
+    }
+    node.index = this.layers[node.layer - 1]
+    return node.layer
+  }
+
+  addEdges(id: string) {
+    const node = this.getNode(id)
+    if (!node) {
+      throw new Error(`Node ${id} not found`)
+    }
+    node.requirements.forEach((rId: string) => {
+      this.edges.add([rId, id])
+    })
+  }
+}