Implement useCloudSync hook for cloud data sync

2026-08-12 23:33:43 +08:00 · 2026-02-24 14:25:13 +08:00
parent 57c00c9611
commit 224ec37e40
1 changed files with 66 additions and 0 deletions
@@ -0,0 +1,66 @@
+import { useState, useCallback } from 'react';
+import { useHistoryStore, usePremiumHistoryStore } from '@/lib/store/history-store';
+import { useFavoritesStore, usePremiumFavoritesStore } from '@/lib/store/favorites-store';
+import { getProfileId } from '@/lib/store/auth-store';
+
+export function useCloudSync(isPremium = false) {
+  const [isSyncing, setIsSyncing] = useState(false);
+
+  const historyStore = isPremium ? usePremiumHistoryStore : useHistoryStore;
+  const favoritesStore = isPremium ? usePremiumFavoritesStore : useFavoritesStore;
+
+  const pullFromCloud = useCallback(async () => {
+    const profileId = getProfileId();
+    if (!profileId) return;
+
+    setIsSyncing(true);
+    try {
+      const response = await fetch('/api/user/sync', {
+        headers: { 'x-profile-id': profileId }
+      });
+      const result = await response.json();
+
+      if (result.success && result.data) {
+        if (result.data.history?.length > 0) {
+          historyStore.getState().importHistory(result.data.history);
+        }
+        if (result.data.favorites?.length > 0) {
+          favoritesStore.getState().importFavorites(result.data.favorites);
+        }
+      }
+    } catch (error) {
+      console.error('Failed to pull from cloud:', error);
+    } finally {
+      setIsSyncing(false);
+    }
+  }, [historyStore, favoritesStore]);
+
+  const pushToCloud = useCallback(async () => {
+    const profileId = getProfileId();
+    if (!profileId) return;
+
+    setIsSyncing(true);
+    try {
+      const currentHistory = historyStore.getState().viewingHistory;
+      const currentFavorites = favoritesStore.getState().favorites;
+
+      await fetch('/api/user/sync', {
+        method: 'POST',
+        headers: { 
+          'x-profile-id': profileId,
+          'Content-Type': 'application/json'
+        },
+        body: JSON.stringify({
+          history: currentHistory,
+          favorites: currentFavorites
+        })
+      });
+    } catch (error) {
+      console.error('Failed to push to cloud:', error);
+    } finally {
+      setIsSyncing(false);
+    }
+  }, [historyStore, favoritesStore]);
+
+  return { pushToCloud, pullFromCloud, isSyncing };
+}