From 0c2439fc9d01acba70895dbfb04bad1cb39e399a Mon Sep 17 00:00:00 2001
From: Emir Karabeg <emirkarabeg@berkeley.edu>
Date: Tue, 18 Feb 2025 19:56:52 -0800
Subject: [PATCH] improve(sync): added periodic sync

---
 stores/sync-manager.ts | 96 ++++++++++++++++++++++++------------------
 1 file changed, 54 insertions(+), 42 deletions(-)

diff --git a/stores/sync-manager.ts b/stores/sync-manager.ts
index 59bf5f865..cb14e1d5e 100644
--- a/stores/sync-manager.ts
+++ b/stores/sync-manager.ts
@@ -1,5 +1,4 @@
 import { useWorkflowRegistry } from './workflow/registry/store'
-import { useWorkflowStore } from './workflow/store'
 import { mergeSubblockState } from './workflow/utils'
 
 interface WorkflowSyncPayload {
@@ -34,52 +33,65 @@ async function syncWorkflowsToServer(payloads: WorkflowSyncPayload[]): Promise<b
   }
 }
 
+let syncInterval: NodeJS.Timeout | null = null
+
+async function performSync() {
+  const { workflows } = useWorkflowRegistry.getState()
+
+  // Prepare sync payloads for all workflows
+  const syncPayloads: (WorkflowSyncPayload | null)[] = await Promise.all(
+    Object.entries(workflows).map(async ([id, metadata]) => {
+      const savedState = localStorage.getItem(`workflow-${id}`)
+      if (!savedState) return null
+
+      const state = JSON.parse(savedState)
+      const mergedBlocks = mergeSubblockState(state.blocks)
+
+      return {
+        id,
+        name: metadata.name,
+        description: metadata.description,
+        state: JSON.stringify({
+          blocks: mergedBlocks,
+          edges: state.edges,
+          loops: state.loops,
+          lastSaved: state.lastSaved,
+        }),
+      }
+    })
+  )
+
+  // Filter out null values and sync if there are workflows to sync
+  const validPayloads = syncPayloads.filter(
+    (payload): payload is WorkflowSyncPayload => payload !== null
+  )
+
+  if (validPayloads.length > 0) {
+    await syncWorkflowsToServer(validPayloads)
+  }
+}
+
 export function initializeSyncManager() {
   if (typeof window === 'undefined') return
 
+  // Start periodic sync
+  syncInterval = setInterval(performSync, 30000) // Sync every 30 seconds
+
   const handleBeforeUnload = async (event: BeforeUnloadEvent) => {
-    const { workflows } = useWorkflowRegistry.getState()
-
-    // Prepare sync payloads for all workflows
-    const syncPayloads: (WorkflowSyncPayload | null)[] = await Promise.all(
-      Object.entries(workflows).map(async ([id, metadata]) => {
-        // Get workflow state from localStorage
-        const savedState = localStorage.getItem(`workflow-${id}`)
-        if (!savedState) return null
-
-        const state = JSON.parse(savedState)
-        // Merge subblock states for all blocks in the workflow
-        const mergedBlocks = mergeSubblockState(state.blocks)
-
-        return {
-          id,
-          name: metadata.name,
-          description: metadata.description,
-          state: JSON.stringify({
-            blocks: mergedBlocks,
-            edges: state.edges,
-            loops: state.loops,
-            lastSaved: state.lastSaved,
-          }),
-        }
-      })
-    )
-
-    // Filter out null values and sync if there are workflows to sync
-    const validPayloads = syncPayloads.filter(
-      (payload): payload is WorkflowSyncPayload => payload !== null
-    )
-
-    if (validPayloads.length > 0) {
-      // Show confirmation dialog
-      event.preventDefault()
-      event.returnValue = ''
-
-      // Attempt to sync
-      await syncWorkflowsToServer(validPayloads)
-    }
+    // Perform one final sync before unloading
+    event.preventDefault()
+    event.returnValue = ''
+    await performSync()
   }
 
   window.addEventListener('beforeunload', handleBeforeUnload)
-  return () => window.removeEventListener('beforeunload', handleBeforeUnload)
+
+  // Return cleanup function
+  return () => {
+    window.removeEventListener('beforeunload', handleBeforeUnload)
+    if (syncInterval) {
+      clearInterval(syncInterval)
+      syncInterval = null
+    }
+  }
 }