From 358f00576c0fadc0610d3880e232c820876d04d1 Mon Sep 17 00:00:00 2001
From: John Obelenus <jobelenus@systeminit.com>
Date: Fri, 20 Dec 2024 11:51:10 -0500
Subject: [PATCH] Fix: missing new edges on apply

---
 app/web/src/store/views.store.ts | 40 +++++++++++++++-----------------
 1 file changed, 19 insertions(+), 21 deletions(-)

diff --git a/app/web/src/store/views.store.ts b/app/web/src/store/views.store.ts
index cfec27e701..2b2aba9264 100644
--- a/app/web/src/store/views.store.ts
+++ b/app/web/src/store/views.store.ts
@@ -599,6 +599,23 @@ export const useViewsStore = (forceChangeSetId?: ChangeSetId) => {
           this.selectedComponentIds = valid;
           this.syncSelectionIntoUrl();
         },
+        viewAssignment(view: View) {
+          /* *
+           * i think i want to set these as in-memory references
+           * that way i don't have to do two writes for incoming WsEvents
+           * or two writes for user actions
+           *
+           * this does mean that `draggedElementsPositionsPreDrag` and
+           * `resizedElementSizesPreResize` need to be populated
+           * but those could just be a `structuredClone` of this data
+           * */
+          this.components = view.components;
+          this.groups = view.groups;
+          this.viewNodes = view.viewNodes;
+          // derive the socket position from the component position
+          // to begin, and then adjust it via delta when things move
+          this.sockets = view.sockets;
+        },
         async selectView(id: ViewId) {
           const view = this.viewsById[id];
           if (view) {
@@ -620,22 +637,8 @@ export const useViewsStore = (forceChangeSetId?: ChangeSetId) => {
             this.selectedViewId = id;
             this.outlinerViewId = id;
             this._enforceSelectedComponents();
-            /* *
-             * i think i want to set these as in-memory references
-             * that way i don't have to do two writes for incoming WsEvents
-             * or two writes for user actions
-             *
-             * this does mean that `draggedElementsPositionsPreDrag` and
-             * `resizedElementSizesPreResize` need to be populated
-             * but those could just be a `structuredClone` of this data
-             * */
-            this.components = view.components;
-            this.groups = view.groups;
-            this.viewNodes = view.viewNodes;
             this.setGroupZIndex();
-            // derive the socket position from the component position
-            // to begin, and then adjust it via delta when things move
-            this.sockets = view.sockets;
+            this.viewAssignment(view);
           } else {
             const res = await this.FETCH_VIEW(id);
             if (!res.result.success) {
@@ -734,12 +737,7 @@ export const useViewsStore = (forceChangeSetId?: ChangeSetId) => {
                 },
               );
               this.setGroupZIndex();
-
-              if (this.selectedViewId === view.id) {
-                this.components = view.components;
-                this.groups = view.groups;
-                this.viewNodes = view.viewNodes;
-              }
+              if (this.selectedViewId === view.id) this.viewAssignment(view);
             },
           });
         },