feat: Track internal portals when indexing nodes (#510)

Implemented in theopensystemslab/planx-new#3666

src/models/session/logic.ts

@@ -14,26 +14,43 @@ import { ComponentType } from "../../types";
 
 export function sortFlow(flow: FlowGraph): OrderedFlow {
   let sectionId: string | undefined;
+  let internalPortalId: string | undefined;
+
   const nodes: IndexedNode[] = [];
+  const nodeIds = new Set<string>();
+
   const searchNodeEdges = (id: string, parentId: string) => {
-    // skip already added nodes
-    if (nodes.map((n) => n.id).includes(id)) return;
+    // Skip already added nodes
+    if (nodeIds.has(id)) return;
+    nodeIds.add(id);
+
     const foundNode = flow[id];
     if (!foundNode) {
       throw new Error(`Referenced node edge "${id}" not found`);
     }
     if (!foundNode.type) {
       throw new Error(`Node is missing a type: "${JSON.stringify(foundNode)}"`);
     }
+
     sectionId = foundNode.type == ComponentType.Section ? id : sectionId;
-    nodes.push({
+
+    const nodeToAdd: IndexedNode = {
       id,
       parentId,
-      ...(sectionId && { sectionId }),
-      type: foundNode.type!,
-      ...(foundNode.edges && { edges: foundNode.edges }),
+      type: foundNode.type,
       data: foundNode.data,
-    });
+    };
+
+    if (sectionId) nodeToAdd.sectionId = sectionId;
+    if (internalPortalId) nodeToAdd.internalPortalId = internalPortalId;
+    if (foundNode.edges) nodeToAdd.edges = foundNode.edges;
+
+    nodes.push(nodeToAdd);
+
+    // Subsequent nodes are within this internal portal
+    internalPortalId =
+      foundNode.type == ComponentType.InternalPortal ? id : internalPortalId;
+
     foundNode.edges?.forEach((childEdgeId) => {
       searchNodeEdges(childEdgeId, id);
     });

src/types/flow.ts

@@ -31,6 +31,7 @@ export type IndexedNode = Node & {
   type: ComponentType;
   parentId: string;
   sectionId?: string;
+  internalPortalId?: string;
 };
 
 export type IndexedFlowGraph = {