@oxyhq/services 5.16.26 → 5.16.27

package/src/ui/context/OxyContext.tsx

@@ -34,6 +34,8 @@ import { translate } from '../../i18n';
 import { updateAvatarVisibility, updateProfileWithAvatar } from '../utils/avatarUtils';
 import { useAccountStore } from '../stores/accountStore';
 import { logger as loggerUtil } from '../../utils/loggerUtils';
+import { useTransferStore, useTransferCodesForPersistence } from '../stores/transferStore';
+import { useCheckPendingTransfers } from '../hooks/useTransferQueries';
 
 export interface OxyContextState {
   user: User | null;
@@ -387,20 +389,22 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
     oxyServices.clearCache();
   }, [queryClient, storage, clearSessionState, logger, oxyServices]);
 
+  // Extract Zustand store functions early (before they're used in callbacks)
+  const getAllPendingTransfersStore = useTransferStore((state) => state.getAllPendingTransfers);
+  const getActiveTransferIdStore = useTransferStore((state) => state.getActiveTransferId);
+  const storeTransferCodeStore = useTransferStore((state) => state.storeTransferCode);
+  const getTransferCodeStore = useTransferStore((state) => state.getTransferCode);
+  const updateTransferStateStore = useTransferStore((state) => state.updateTransferState);
+  const clearTransferCodeStore = useTransferStore((state) => state.clearTransferCode);
+
   // Transfer code management functions (must be defined before deleteIdentityAndClearAccount)
   const getAllPendingTransfers = useCallback(() => {
-    const pending: Array<{ transferId: string; data: TransferCodeData }> = [];
-    transferCodesRef.current.forEach((data, transferId) => {
-      if (data.state === 'pending') {
-        pending.push({ transferId, data });
-      }
-    });
-    return pending;
-  }, []);
+    return getAllPendingTransfersStore();
+  }, [getAllPendingTransfersStore]);
 
   const getActiveTransferId = useCallback(() => {
-    return activeTransferIdRef.current;
-  }, []);
+    return getActiveTransferIdStore();
+  }, [getActiveTransferIdStore]);
 
   // Delete identity and clear all account data
   // In accounts app, deleting identity means losing the account completely
@@ -415,7 +419,7 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
       const pendingTransfers = getAllPendingTransfers();
       if (pendingTransfers.length > 0) {
         const activeTransferId = getActiveTransferId();
-        const hasActiveTransfer = activeTransferId && pendingTransfers.some(t => t.transferId === activeTransferId);
+        const hasActiveTransfer = activeTransferId && pendingTransfers.some((t: { transferId: string; data: any }) => t.transferId === activeTransferId);
         
         if (hasActiveTransfer) {
           throw new Error(
@@ -510,6 +514,10 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
                 loggerUtil.debug('Identity sync timeout (expected when offline)', { component: 'OxyContext', method: 'checkNetworkAndSync' }, syncError as unknown);
               }
             }
+
+            // Check for pending transfers that may have completed while offline
+            // This is handled by useCheckPendingTransfers hook which runs automatically
+            // when authenticated and online
           }
           
           // TanStack Query will automatically retry pending mutations
@@ -551,7 +559,7 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
         clearTimeout(checkTimeout);
       }
     };
-  }, [oxyServices, storage, syncIdentity, logger]);
+  }, [oxyServices, storage, syncIdentity, logger, hasIdentity, isAuthenticated]);
 
   const { getDeviceSessions, logoutAllDeviceSessions, updateDeviceName } = useDeviceManagement({
     oxyServices,
@@ -670,207 +678,110 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
   // The JWT token's userId field contains the MongoDB ObjectId
   const userId = oxyServices.getCurrentUserId() || user?.id;
 
-  // Transfer code storage interface
-  interface TransferCodeData {
-    code: string;
-    sourceDeviceId: string | null;
-    publicKey: string;
-    timestamp: number;
-    state: 'pending' | 'completed' | 'failed';
-  }
-
-  // Store transfer codes in memory for verification (also persisted to storage)
-  // Map: transferId -> TransferCodeData
-  const transferCodesRef = useRef<Map<string, TransferCodeData>>(new Map());
-  const activeTransferIdRef = useRef<string | null>(null);
+  // Use Zustand store for transfer state management
   const TRANSFER_CODES_STORAGE_KEY = `${storageKeyPrefix}_transfer_codes`;
   const ACTIVE_TRANSFER_STORAGE_KEY = `${storageKeyPrefix}_active_transfer_id`;
+  const isRestored = useTransferStore((state) => state.isRestored);
+  const restoreFromStorage = useTransferStore((state) => state.restoreFromStorage);
+  const markRestored = useTransferStore((state) => state.markRestored);
+  const cleanupExpired = useTransferStore((state) => state.cleanupExpired);
 
-  // Load transfer codes from storage on startup
+  // Load transfer codes from storage on startup (only once)
   useEffect(() => {
-    if (!storage || !isStorageReady) return;
+    if (!storage || !isStorageReady || isRestored) return;
 
     const loadTransferCodes = async () => {
       try {
         // Load transfer codes
         const storedCodes = await storage.getItem(TRANSFER_CODES_STORAGE_KEY);
-        if (storedCodes) {
-          const parsed = JSON.parse(storedCodes);
-          const now = Date.now();
-          const fifteenMinutes = 15 * 60 * 1000;
-          
-          // Only restore non-expired pending transfers
-          Object.entries(parsed).forEach(([transferId, data]: [string, any]) => {
-            if (data.state === 'pending' && (now - data.timestamp) < fifteenMinutes) {
-              transferCodesRef.current.set(transferId, data);
-              if (__DEV__) {
-                logger('Restored pending transfer from storage', { transferId });
-              }
-            }
+        const storedActiveTransferId = await storage.getItem(ACTIVE_TRANSFER_STORAGE_KEY);
+        
+        const parsedCodes = storedCodes ? JSON.parse(storedCodes) : {};
+        const activeTransferId = storedActiveTransferId || null;
+        
+        // Restore to Zustand store (store handles validation and expiration)
+        restoreFromStorage(parsedCodes, activeTransferId);
+        markRestored();
+        
+        if (__DEV__ && Object.keys(parsedCodes).length > 0) {
+          logger('Restored transfer codes from storage', { 
+            count: Object.keys(parsedCodes).length,
+            hasActiveTransfer: !!activeTransferId,
           });
         }
-
-        // Load active transfer ID
-        const activeTransferId = await storage.getItem(ACTIVE_TRANSFER_STORAGE_KEY);
-        if (activeTransferId) {
-          // Verify it's still valid
-          const transferData = transferCodesRef.current.get(activeTransferId);
-          if (transferData && transferData.state === 'pending') {
-            activeTransferIdRef.current = activeTransferId;
-            if (__DEV__) {
-              logger('Restored active transfer ID from storage', { transferId: activeTransferId });
-            }
-          } else {
-            // Clear invalid active transfer
-            await storage.removeItem(ACTIVE_TRANSFER_STORAGE_KEY);
-          }
-        }
       } catch (error) {
         if (__DEV__) {
           logger('Failed to load transfer codes from storage', error);
         }
+        // Mark as restored even on error to prevent retries
+        markRestored();
       }
     };
 
     loadTransferCodes();
-  }, [storage, isStorageReady, logger, storageKeyPrefix]);
+  }, [storage, isStorageReady, isRestored, restoreFromStorage, markRestored, logger, storageKeyPrefix]);
 
-  // Persist transfer codes to storage whenever they change
-  const persistTransferCodes = useCallback(async () => {
-    if (!storage) return;
-    
-    try {
-      const codesToStore: Record<string, TransferCodeData> = {};
-      transferCodesRef.current.forEach((value, key) => {
-        codesToStore[key] = value;
-      });
-      await storage.setItem(TRANSFER_CODES_STORAGE_KEY, JSON.stringify(codesToStore));
-    } catch (error) {
-      if (__DEV__) {
-        logger('Failed to persist transfer codes', error);
-      }
-    }
-  }, [storage, logger]);
-
-  // Cleanup old transfer codes (older than 15 minutes)
+  // Persist transfer codes to storage whenever store changes
+  const { transferCodes, activeTransferId } = useTransferCodesForPersistence();
   useEffect(() => {
-    const cleanup = setInterval(async () => {
-      const now = Date.now();
-      const fifteenMinutes = 15 * 60 * 1000;
-      let needsPersist = false;
-
-      transferCodesRef.current.forEach((value, key) => {
-        if (now - value.timestamp > fifteenMinutes) {
-          transferCodesRef.current.delete(key);
-          needsPersist = true;
-          if (__DEV__) {
-            logger('Cleaned up expired transfer code', { transferId: key });
-          }
-        }
-      });
+    if (!storage || !isStorageReady || !isRestored) return;
 
-      // Clear active transfer if it was deleted
-      if (activeTransferIdRef.current && !transferCodesRef.current.has(activeTransferIdRef.current)) {
-        activeTransferIdRef.current = null;
-        if (storage) {
-          try {
-            await storage.removeItem(ACTIVE_TRANSFER_STORAGE_KEY);
-          } catch (error) {
-            // Ignore storage errors
-          }
+    const persistTransferCodes = async () => {
+      try {
+        await storage.setItem(TRANSFER_CODES_STORAGE_KEY, JSON.stringify(transferCodes));
+        
+        if (activeTransferId) {
+          await storage.setItem(ACTIVE_TRANSFER_STORAGE_KEY, activeTransferId);
+        } else {
+          await storage.removeItem(ACTIVE_TRANSFER_STORAGE_KEY);
+        }
+      } catch (error) {
+        if (__DEV__) {
+          logger('Failed to persist transfer codes', error);
         }
       }
+    };
 
-      if (needsPersist) {
-        await persistTransferCodes();
-      }
+    persistTransferCodes();
+  }, [transferCodes, activeTransferId, storage, isStorageReady, isRestored, logger]);
+
+  // Cleanup expired transfer codes (every minute)
+  useEffect(() => {
+    const cleanup = setInterval(() => {
+      cleanupExpired();
     }, 60000); // Check every minute
 
     return () => clearInterval(cleanup);
-  }, [logger, persistTransferCodes, storage]);
+  }, [cleanupExpired]);
 
-  // Transfer code management functions
+  // Transfer code management functions using Zustand store
   const storeTransferCode = useCallback(async (transferId: string, code: string, sourceDeviceId: string | null, publicKey: string) => {
-    const transferData: TransferCodeData = {
-      code,
-      sourceDeviceId,
-      publicKey,
-      timestamp: Date.now(),
-      state: 'pending',
-    };
-    
-    transferCodesRef.current.set(transferId, transferData);
-    activeTransferIdRef.current = transferId;
-    
-    // Persist to storage
-    await persistTransferCodes();
-    if (storage) {
-      try {
-        await storage.setItem(ACTIVE_TRANSFER_STORAGE_KEY, transferId);
-      } catch (error) {
-        if (__DEV__) {
-          logger('Failed to persist active transfer ID', error);
-        }
-      }
-    }
+    storeTransferCodeStore(transferId, code, sourceDeviceId, publicKey);
     
     if (__DEV__) {
       logger('Stored transfer code', { transferId, sourceDeviceId, publicKey: publicKey.substring(0, 16) + '...' });
     }
-  }, [logger, persistTransferCodes, storage]);
+  }, [logger, storeTransferCodeStore]);
 
   const getTransferCode = useCallback((transferId: string) => {
-    return transferCodesRef.current.get(transferId) || null;
-  }, []);
+    return getTransferCodeStore(transferId);
+  }, [getTransferCodeStore]);
 
   const updateTransferState = useCallback(async (transferId: string, state: 'pending' | 'completed' | 'failed') => {
-    const transferData = transferCodesRef.current.get(transferId);
-    if (transferData) {
-      transferData.state = state;
-      transferCodesRef.current.set(transferId, transferData);
-      
-      // Clear active transfer if completed or failed
-      if (state === 'completed' || state === 'failed') {
-        if (activeTransferIdRef.current === transferId) {
-          activeTransferIdRef.current = null;
-          if (storage) {
-            try {
-              await storage.removeItem(ACTIVE_TRANSFER_STORAGE_KEY);
-            } catch (error) {
-              // Ignore storage errors
-            }
-          }
-        }
-      }
-      
-      await persistTransferCodes();
-      
-      if (__DEV__) {
-        logger('Updated transfer state', { transferId, state });
-      }
+    updateTransferStateStore(transferId, state);
+    
+    if (__DEV__) {
+      logger('Updated transfer state', { transferId, state });
     }
-  }, [logger, persistTransferCodes, storage]);
+  }, [logger, updateTransferStateStore]);
 
   const clearTransferCode = useCallback(async (transferId: string) => {
-    transferCodesRef.current.delete(transferId);
-    
-    if (activeTransferIdRef.current === transferId) {
-      activeTransferIdRef.current = null;
-      if (storage) {
-        try {
-          await storage.removeItem(ACTIVE_TRANSFER_STORAGE_KEY);
-        } catch (error) {
-          // Ignore storage errors
-        }
-      }
-    }
-    
-    await persistTransferCodes();
+    clearTransferCodeStore(transferId);
     
     if (__DEV__) {
       logger('Cleared transfer code', { transferId });
     }
-  }, [logger, persistTransferCodes, storage]);
+  }, [logger, clearTransferCodeStore]);
 
   const refreshSessionsWithUser = useCallback(
     () => refreshSessions(userId),
@@ -889,6 +800,9 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
     logout().catch((remoteError) => logger('Failed to process remote sign out', remoteError));
   }, [logger, logout]);
 
+  // Check pending transfers when authenticated and online using TanStack Query
+  const { data: pendingTransferResults } = useCheckPendingTransfers();
+
   const handleIdentityTransferComplete = useCallback(
     async (data: { transferId: string; sourceDeviceId: string; publicKey: string; transferCode?: string; completedAt: string }) => {
       try {
@@ -981,43 +895,10 @@ export const OxyProvider: React.FC<OxyContextProviderProps> = ({
           return;
         }
 
-        // CRITICAL: Verify target device has identity before deleting on source device
-        // This ensures we never lose the identity
-        try {
-          logger('Verifying target device has identity before deletion', {
-            transferId: data.transferId,
-            publicKey: data.publicKey.substring(0, 16) + '...',
-          });
-
-          // Verify target device has active session with matching public key
-          const verifyResponse = await oxyServices.makeRequest<{ verified: boolean; hasActiveSession: boolean }>(
-            'GET',
-            `/api/identity/verify-transfer?publicKey=${encodeURIComponent(data.publicKey)}`,
-            undefined,
-            { cache: false }
-          );
-
-          if (!verifyResponse.verified || !verifyResponse.hasActiveSession) {
-            logger('Target device verification failed - identity will not be deleted', {
-              transferId: data.transferId,
-              verified: verifyResponse.verified,
-              hasActiveSession: verifyResponse.hasActiveSession,
-            });
-            await updateTransferState(data.transferId, 'failed');
-            toast.error('Transfer verification failed: Target device does not have active session. Identity will not be deleted.');
-            return;
-          }
-
-          logger('Target device verification passed', {
-            transferId: data.transferId,
-          });
-        } catch (verifyError: any) {
-          // If verification fails, don't delete identity - it's safer to keep it
-          logger('Failed to verify target device - identity will not be deleted', verifyError);
-          await updateTransferState(data.transferId, 'failed');
-          toast.error('Transfer verification failed: Could not verify target device. Identity will not be deleted.');
-          return;
-        }
+        // NOTE: Target device verification already happened server-side when notifyTransferComplete was called
+        // The server verified that the target device is authenticated and has the matching public key
+        // Additional client-side verification is not necessary and would require source device authentication
+        // which may not be available. The existing checks (public key match, transfer code, device ID) are sufficient.
 
         logger('All transfer verifications passed, deleting identity from source device', {
           transferId: data.transferId,

package/src/ui/hooks/useTransferQueries.ts

@@ -0,0 +1,154 @@
+import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
+import { useOxy } from '../context/OxyContext';
+import { useTransferStore } from '../stores/transferStore';
+
+/**
+ * Query keys for transfer-related queries
+ */
+export const transferQueryKeys = {
+  all: ['transfers'] as const,
+  completion: (transferId: string) => ['transfers', 'completion', transferId] as const,
+  pending: () => ['transfers', 'pending'] as const,
+};
+
+/**
+ * Hook to check if a transfer was completed
+ * Only runs when authenticated and transferId is provided
+ */
+export const useCheckTransferCompletion = (transferId: string | null, enabled: boolean = true) => {
+  const { oxyServices, isAuthenticated } = useOxy();
+  
+  return useQuery({
+    queryKey: transferId ? transferQueryKeys.completion(transferId) : ['transfers', 'completion', 'null'],
+    queryFn: async () => {
+      if (!transferId || !oxyServices) {
+        return null;
+      }
+
+      try {
+        const response = await oxyServices.makeRequest<{
+          completed: boolean;
+          transferId?: string;
+          sourceDeviceId?: string;
+          publicKey?: string;
+          transferCode?: string;
+          completedAt?: string;
+        }>(
+          'GET',
+          `/api/identity/check-transfer/${transferId}`,
+          undefined,
+          { cache: false }
+        );
+
+        return response;
+      } catch (error: any) {
+        // Handle 401 errors gracefully - don't throw, just return null
+        if (error?.status === 401 || error?.message?.includes('401') || error?.message?.includes('authentication')) {
+          if (__DEV__) {
+            console.warn('[useCheckTransferCompletion] Authentication required, skipping check');
+          }
+          return null;
+        }
+        throw error;
+      }
+    },
+    enabled: enabled && !!transferId && isAuthenticated && !!oxyServices,
+    staleTime: 30 * 1000, // 30 seconds - completion status doesn't change frequently
+    gcTime: 5 * 60 * 1000, // 5 minutes
+    retry: (failureCount, error: any) => {
+      // Don't retry on 401 errors
+      if (error?.status === 401 || error?.message?.includes('401')) {
+        return false;
+      }
+      return failureCount < 2;
+    },
+    retryDelay: (attemptIndex) => Math.min(1000 * 2 ** attemptIndex, 5000),
+  });
+};
+
+/**
+ * Hook to check all pending transfers for completion
+ * Used when app comes back online
+ */
+export const useCheckPendingTransfers = () => {
+  const { oxyServices, isAuthenticated } = useOxy();
+  const getAllPendingTransfers = useTransferStore((state) => state.getAllPendingTransfers);
+  const pendingTransfers = getAllPendingTransfers();
+  
+  return useQuery({
+    queryKey: [...transferQueryKeys.pending(), pendingTransfers.map(t => t.transferId).join(',')],
+    queryFn: async () => {
+      if (!oxyServices || pendingTransfers.length === 0) {
+        return [];
+      }
+
+      const results: Array<{
+        transferId: string;
+        completed: boolean;
+        data?: {
+          transferId?: string;
+          sourceDeviceId?: string;
+          publicKey?: string;
+          transferCode?: string;
+          completedAt?: string;
+        };
+      }> = [];
+
+      // Check each pending transfer
+      for (const { transferId, data } of pendingTransfers) {
+        try {
+          const response = await oxyServices.makeRequest<{
+            completed: boolean;
+            transferId?: string;
+            sourceDeviceId?: string;
+            publicKey?: string;
+            transferCode?: string;
+            completedAt?: string;
+          }>(
+            'GET',
+            `/api/identity/check-transfer/${transferId}`,
+            undefined,
+            { cache: false }
+          );
+
+          if (response.completed && response.publicKey === data.publicKey) {
+            results.push({
+              transferId,
+              completed: true,
+              data: response,
+            });
+          } else {
+            results.push({
+              transferId,
+              completed: false,
+            });
+          }
+        } catch (error: any) {
+          // Handle 401 errors gracefully - skip this transfer
+          if (error?.status === 401 || error?.message?.includes('401') || error?.message?.includes('authentication')) {
+            if (__DEV__) {
+              console.warn(`[useCheckPendingTransfers] Authentication required for transfer ${transferId}, skipping`);
+            }
+            results.push({
+              transferId,
+              completed: false,
+            });
+            continue;
+          }
+          // For other errors, mark as not completed
+          results.push({
+            transferId,
+            completed: false,
+          });
+        }
+      }
+
+      return results;
+    },
+    enabled: isAuthenticated && !!oxyServices && pendingTransfers.length > 0,
+    staleTime: 30 * 1000, // 30 seconds
+    gcTime: 5 * 60 * 1000, // 5 minutes
+    retry: false, // Don't retry - we'll check again on next reconnect
+  });
+};
+