From c62c18a20d72db1591b69ba8e2cd3fdd7872af9b Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Fri, 10 Feb 2023 15:30:22 +0100
Subject: [PATCH] refactor(core): use computed to group edges

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 .../container/EdgeRenderer/EdgeRenderer.vue   | 27 ++++++-------------
 1 file changed, 8 insertions(+), 19 deletions(-)

diff --git a/packages/core/src/container/EdgeRenderer/EdgeRenderer.vue b/packages/core/src/container/EdgeRenderer/EdgeRenderer.vue
index 3bbc0f4b..da8c582d 100644
--- a/packages/core/src/container/EdgeRenderer/EdgeRenderer.vue
+++ b/packages/core/src/container/EdgeRenderer/EdgeRenderer.vue
@@ -1,5 +1,4 @@
 <script lang="ts" setup>
-import type { WatchStopHandle } from 'vue'
 import EdgeWrapper from '../../components/Edges/EdgeWrapper'
 import ConnectionLine from '../../components/ConnectionLine/ConnectionLine.vue'
 import type { EdgeComponent, EdgeUpdatable, GraphEdge } from '../../types'
@@ -8,7 +7,6 @@ import MarkerDefinitions from './MarkerDefinitions.vue'
 const slots = inject(Slots)
 
 const {
-  onPaneReady,
   connectionStartHandle,
   nodesConnectable,
   edgesUpdatable,
@@ -46,23 +44,14 @@ const connectionLineVisible = $(
   ),
 )
 
-let groups = $ref<ReturnType<typeof groupEdgesByZLevel>>([])
-
-let stop: WatchStopHandle
-onPaneReady(() => {
-  stop = watch(
-    [
-      () => getEdges.length,
-      () => getEdges.map((e) => e.zIndex),
-      () => (elevateEdgesOnSelect ? getSelectedNodes.length : 0),
-      () => (elevateEdgesOnSelect ? getNodesInitialized.map((n) => n.computedPosition.z) : []),
-    ],
-    () => {
-      groups = groupEdgesByZLevel(getEdges, findNode, elevateEdgesOnSelect)
-    },
-    { immediate: true },
-  )
-})
+const groups = controlledComputed(
+  [
+    () => getEdges.map((e) => e.zIndex),
+    () => (elevateEdgesOnSelect ? [getSelectedNodes.length] : [0]),
+    () => (elevateEdgesOnSelect ? getNodesInitialized.map((n) => n.computedPosition.z) : []),
+  ],
+  () => groupEdgesByZLevel(getEdges, findNode, elevateEdgesOnSelect),
+)
 
 onBeforeUnmount(() => {
   stop?.()