fix: Return empty arrays if pane isn't ready yet

bcakmakoglu · Apr 9, 2022 · 6829efc · 6829efc
1 parent c295a83
commit 6829efc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/package/src/composables/useVueFlow.ts b/package/src/composables/useVueFlow.ts
@@ -53,6 +53,7 @@ export class Storage {
       ...getters,
       ...actions,
     })
+
     const flow: UseVueFlow = {
       ...hooksOn,
       ...getters,

diff --git a/package/src/store/getters.ts b/package/src/store/getters.ts
@@ -3,6 +3,10 @@ import { State, GraphEdge, GraphNode, ComputedGetters } from '~/types'
 import { getNodesInside, isEdgeVisible } from '~/utils'
 
 export default (state: State): ComputedGetters => {
+  const paneReady = ref(false)
+
+  state.hooks.paneReady.on(() => (paneReady.value = true))
+
   const getEdgeTypes = computed(() => {
     const edgeTypes: Record<string, any> = {
       ...defaultEdgeTypes,
@@ -24,6 +28,8 @@ export default (state: State): ComputedGetters => {
   })
 
   const getNodes = computed<GraphNode[]>(() => {
+    if (!paneReady.value) return []
+
     const nodes = state.nodes.filter((n) => !n.hidden)
     return state.onlyRenderVisibleElements
       ? nodes &&
@@ -42,6 +48,8 @@ export default (state: State): ComputedGetters => {
   })
 
   const getEdges = computed<GraphEdge[]>(() => {
+    if (!paneReady.value) return []
+
     if (!state.onlyRenderVisibleElements)
       return state.edges.filter((e) => !e.hidden && e.targetNode && !e.targetNode.hidden && e.sourceNode && !e.sourceNode.hidden)
     else