From da29a613f0caec30d59b82f19680cafb5431e044 Mon Sep 17 00:00:00 2001
From: Braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Sat, 20 Nov 2021 22:42:56 +0100
Subject: [PATCH] feat(slots): Add node and edge renderer slots

* more flexibility by having the ability to use the node/edge-renderer and wrapper
* export renderer and wrapper from lib entry

Signed-off-by: Braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 src/components/Edges/index.ts       |   1 +
 src/components/Nodes/index.ts       |   1 +
 src/container/VueFlow/VueFlow.vue   | 112 +++++++++++++++-------------
 src/container/ZoomPane/ZoomPane.vue |   3 +-
 src/index.ts                        |   2 +
 5 files changed, 65 insertions(+), 54 deletions(-)

diff --git a/src/components/Edges/index.ts b/src/components/Edges/index.ts
index 20a92f54..f5dc1625 100644
--- a/src/components/Edges/index.ts
+++ b/src/components/Edges/index.ts
@@ -4,3 +4,4 @@ export { default as SmoothStepEdge } from './SmoothStepEdge.vue'
 export { default as StraightEdge } from './StraightEdge.vue'
 export { default as EdgeAnchor } from './EdgeAnchor.vue'
 export { default as EdgeText } from './EdgeText.vue'
+export { default as EdgeWrapper } from './Edge.vue'
diff --git a/src/components/Nodes/index.ts b/src/components/Nodes/index.ts
index 53a63969..bfc2fc83 100644
--- a/src/components/Nodes/index.ts
+++ b/src/components/Nodes/index.ts
@@ -1,3 +1,4 @@
 export { default as DefaultNode } from './DefaultNode.vue'
 export { default as InputNode } from './InputNode.vue'
 export { default as OutputNode } from './OutputNode.vue'
+export { default as NodeWrapper } from './Node.vue'
diff --git a/src/container/VueFlow/VueFlow.vue b/src/container/VueFlow/VueFlow.vue
index c32a84d7..6546eda1 100644
--- a/src/container/VueFlow/VueFlow.vue
+++ b/src/container/VueFlow/VueFlow.vue
@@ -1,6 +1,5 @@
 <script lang="ts" setup>
 import { CSSProperties, onBeforeUnmount } from 'vue'
-import diff from 'microdiff'
 import { invoke } from '@vueuse/core'
 import {
   ConnectionLineType,
@@ -155,7 +154,7 @@ invoke(async () => {
   store.instance = instance
 })
 
-throttledWatch(elements, (val) => store.setElements(val), { flush: 'post', deep: true, throttle: 10 })
+throttledWatch(elements, store.setElements, { flush: 'post', deep: true, throttle: 10 })
 throttledWatch(store.elements, (val) => (elements.value = val), { flush: 'post', deep: true, throttle: 10 })
 
 onMounted(() => {
@@ -179,58 +178,65 @@ const transitionName = computed(() => {
   <div class="vue-flow">
     <Transition key="vue-flow-transition" :name="transitionName">
       <Suspense>
-        <ZoomPane
-          key="zoom-pane"
-          :selection-key-code="store.selectionKeyCode"
-          :zoom-activation-key-code="store.zoomActivationKeyCode"
-          :default-zoom="store.defaultZoom"
-          :default-position="store.defaultPosition"
-          :zoom-on-scroll="store.zoomOnScroll"
-          :zoom-on-pinch="store.zoomOnPinch"
-          :zoom-on-double-click="store.zoomOnDoubleClick"
-          :pan-on-scroll="store.panOnScroll"
-          :pan-on-scroll-speed="store.panOnScrollSpeed"
-          :pan-on-scroll-mode="store.panOnScrollMode"
-          :pane-moveable="store.paneMoveable"
-        >
-          <template #default="zoomPaneProps">
-            <SelectionPane
-              key="selection-pane"
-              :delete-key-code="store.deleteKeyCode"
-              :multi-selection-key-code="store.multiSelectionKeyCode"
-              :selection-key-code="store.selectionKeyCode"
-            >
-              <NodeRenderer key="node-renderer" :select-nodes-on-drag="store.selectNodesOnDrag">
-                <template
-                  v-for="nodeName of Object.keys(store.getNodeTypes)"
-                  #[`node-${nodeName}`]="nodeProps"
-                  :key="`node-${nodeName}`"
-                >
-                  <slot :name="`node-${nodeName}`" v-bind="nodeProps"></slot>
-                </template>
-              </NodeRenderer>
-              <EdgeRenderer
-                key="edge-renderer"
-                :connection-line-type="store.connectionLineType"
-                :connection-line-style="store.connectionLineStyle"
-                :arrow-head-color="store.arrowHeadColor"
-                :marker-end-id="store.markerEndId"
+        <template #default>
+          <ZoomPane
+            v-show="store.isReady"
+            key="zoom-pane"
+            :selection-key-code="store.selectionKeyCode"
+            :zoom-activation-key-code="store.zoomActivationKeyCode"
+            :default-zoom="store.defaultZoom"
+            :default-position="store.defaultPosition"
+            :zoom-on-scroll="store.zoomOnScroll"
+            :zoom-on-pinch="store.zoomOnPinch"
+            :zoom-on-double-click="store.zoomOnDoubleClick"
+            :pan-on-scroll="store.panOnScroll"
+            :pan-on-scroll-speed="store.panOnScrollSpeed"
+            :pan-on-scroll-mode="store.panOnScrollMode"
+            :pane-moveable="store.paneMoveable"
+          >
+            <template #default="zoomPaneProps">
+              <SelectionPane
+                key="selection-pane"
+                :delete-key-code="store.deleteKeyCode"
+                :multi-selection-key-code="store.multiSelectionKeyCode"
+                :selection-key-code="store.selectionKeyCode"
               >
-                <template
-                  v-for="edgeName of Object.keys(store.getEdgeTypes)"
-                  #[`edge-${edgeName}`]="edgeProps"
-                  :key="`edge-${edgeName}`"
-                >
-                  <slot :name="`edge-${edgeName}`" v-bind="edgeProps"></slot>
-                </template>
-                <template #custom-connection-line="customConnectionLineProps">
-                  <slot key="connection-line" name="custom-connection-line" v-bind="customConnectionLineProps"></slot>
-                </template>
-              </EdgeRenderer>
-            </SelectionPane>
-            <slot name="zoom-pane" v-bind="zoomPaneProps"></slot>
-          </template>
-        </ZoomPane>
+                <slot name="node-renderer" v-bind="{ nodes: store.getNodes, nodeTypes: store.getNodeTypes }">
+                  <NodeRenderer key="node-renderer" :select-nodes-on-drag="store.selectNodesOnDrag">
+                    <template
+                      v-for="nodeName of Object.keys(store.getNodeTypes)"
+                      #[`node-${nodeName}`]="nodeProps"
+                      :key="`node-${nodeName}`"
+                    >
+                      <slot :name="`node-${nodeName}`" v-bind="nodeProps"></slot>
+                    </template>
+                  </NodeRenderer>
+                </slot>
+                <slot name="edge-renderer" v-bind="{ edges: store.getEdges, edgeTypes: store.getEdgeTypes }">
+                  <EdgeRenderer
+                    key="edge-renderer"
+                    :connection-line-type="store.connectionLineType"
+                    :connection-line-style="store.connectionLineStyle"
+                    :arrow-head-color="store.arrowHeadColor"
+                    :marker-end-id="store.markerEndId"
+                  >
+                    <template
+                      v-for="edgeName of Object.keys(store.getEdgeTypes)"
+                      #[`edge-${edgeName}`]="edgeProps"
+                      :key="`edge-${edgeName}`"
+                    >
+                      <slot :name="`edge-${edgeName}`" v-bind="edgeProps"></slot>
+                    </template>
+                    <template #custom-connection-line="customConnectionLineProps">
+                      <slot key="connection-line" name="custom-connection-line" v-bind="customConnectionLineProps"></slot>
+                    </template>
+                  </EdgeRenderer>
+                </slot>
+              </SelectionPane>
+              <slot name="zoom-pane" v-bind="zoomPaneProps"></slot>
+            </template>
+          </ZoomPane>
+        </template>
         <template v-if="store.loading" #fallback>
           <slot key="loading-indicator" name="loading-indicator">
             <LoadingIndicator key="default-loading-indicator" v-bind="store.loading">
diff --git a/src/container/ZoomPane/ZoomPane.vue b/src/container/ZoomPane/ZoomPane.vue
index d8990ec5..9875d19e 100644
--- a/src/container/ZoomPane/ZoomPane.vue
+++ b/src/container/ZoomPane/ZoomPane.vue
@@ -181,7 +181,8 @@ store.dimensions = {
   height: height.value,
 }
 
-await until(() => store.isReady).toBe(true)
+// skip waiting for ssr
+if (import.meta.env.SSR || typeof window === 'undefined') await until(() => store.isReady)
 
 watch(
   [width, height],
diff --git a/src/index.ts b/src/index.ts
index a135ed25..3836f08c 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,4 +1,6 @@
 export { default as VueFlow } from './container/VueFlow/VueFlow.vue'
+export { default as NodeRenderer } from './container/NodeRenderer/NodeRenderer.vue'
+export { default as EdgeRenderer } from './container/EdgeRenderer/EdgeRenderer.vue'
 export { default as Handle } from './components/Handle/Handle.vue'
 export * from './components/Edges'
 export * from './components/Nodes'