From ff33a006a1356f59122534e31b13e113809200ca Mon Sep 17 00:00:00 2001
From: Patrik Oldsberg <poldsberg@gmail.com>
Date: Wed, 18 Oct 2023 11:01:30 +0200
Subject: [PATCH] frontend-app-api: add node map to app graph

Signed-off-by: Patrik Oldsberg <poldsberg@gmail.com>
---
 .../src/wiring/graph/buildAppGraph.test.ts    | 21 +++++++++++++++++++
 .../src/wiring/graph/buildAppGraph.ts         |  2 +-
 .../src/wiring/graph/types.ts                 |  1 +
 3 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/packages/frontend-app-api/src/wiring/graph/buildAppGraph.test.ts b/packages/frontend-app-api/src/wiring/graph/buildAppGraph.test.ts
index dc489565f4..8cf8279e96 100644
--- a/packages/frontend-app-api/src/wiring/graph/buildAppGraph.test.ts
+++ b/packages/frontend-app-api/src/wiring/graph/buildAppGraph.test.ts
@@ -46,6 +46,7 @@ describe('buildAppGraph', () => {
       edges: { attachments: new Map() },
     });
     expect(Array.from(graph.orphans)).toEqual([]);
+    expect(Array.from(graph.nodes.keys())).toEqual(['core']);
   });
 
   it('should create a graph', () => {
@@ -62,6 +63,16 @@ describe('buildAppGraph', () => {
       'b',
     );
 
+    expect(Array.from(graph.nodes.keys())).toEqual([
+      'a',
+      'b',
+      'c',
+      'bx1',
+      'bx2',
+      'by1',
+      'dx1',
+    ]);
+
     expect(JSON.parse(JSON.stringify(graph.root))).toMatchInlineSnapshot(`
       {
         "attachments": {
@@ -118,6 +129,16 @@ describe('buildAppGraph', () => {
       'b',
     );
 
+    expect(Array.from(graph.nodes.keys())).toEqual([
+      'bx2',
+      'a',
+      'by1',
+      'b',
+      'bx1',
+      'c',
+      'dx1',
+    ]);
+
     expect(String(graph.root)).toMatchInlineSnapshot(`
       "<b>
         x [
diff --git a/packages/frontend-app-api/src/wiring/graph/buildAppGraph.ts b/packages/frontend-app-api/src/wiring/graph/buildAppGraph.ts
index 72d4d90d48..f87d191908 100644
--- a/packages/frontend-app-api/src/wiring/graph/buildAppGraph.ts
+++ b/packages/frontend-app-api/src/wiring/graph/buildAppGraph.ts
@@ -157,5 +157,5 @@ export function buildAppGraph(
     throw new Error(`No root node with id '${rootNodeId}' found in app graph`);
   }
 
-  return { root: rootNode, orphans: orphanNodes };
+  return { root: rootNode, nodes, orphans: orphanNodes };
 }
diff --git a/packages/frontend-app-api/src/wiring/graph/types.ts b/packages/frontend-app-api/src/wiring/graph/types.ts
index 6cf4ade360..21ea55c780 100644
--- a/packages/frontend-app-api/src/wiring/graph/types.ts
+++ b/packages/frontend-app-api/src/wiring/graph/types.ts
@@ -71,5 +71,6 @@ export interface AppNode {
 
 export interface AppGraph {
   root: AppNode;
+  nodes: Map<string /* id */, AppNode>;
   orphans: Iterable<AppNode>;
 }