@oxyhq/services 5.12.11 → 5.13.1

package/lib/module/ui/context/OxyContext.js

@@ -1,7 +1,6 @@
 "use strict";
 
 import { createContext, useContext, useEffect, useCallback, useMemo, useRef, useState } from 'react';
-import { View, Text } from 'react-native';
 import { OxyServices } from '../../core';
 import { DeviceManager } from '../../utils/deviceManager';
 import { useSessionSocket } from '../hooks/useSessionSocket';
@@ -12,6 +11,26 @@ import { useAuthStore } from '../stores/authStore';
 // NOTE: We intentionally avoid importing useFollow here to prevent a require cycle.
 // If consumers relied on `const { useFollow } = useOxy()`, we provide a lazy proxy below.
 import { jsx as _jsx } from "react/jsx-runtime";
+// Empty follow hook fallback
+const createEmptyFollowHook = () => {
+  const emptyResult = {
+    isFollowing: false,
+    isLoading: false,
+    error: null,
+    toggleFollow: async () => {},
+    setFollowStatus: () => {},
+    fetchStatus: async () => {},
+    clearError: () => {},
+    followerCount: null,
+    followingCount: null,
+    isLoadingCounts: false,
+    fetchUserCounts: async () => {},
+    setFollowerCount: () => {},
+    setFollowingCount: () => {}
+  };
+  return () => emptyResult;
+};
+
 // Create the context with default values
 const OxyContext = /*#__PURE__*/createContext(null);
 
@@ -106,6 +125,9 @@ export const OxyProvider = ({
   const [storage, setStorage] = useState(null);
   const [currentLanguage, setCurrentLanguage] = useState('en-US');
 
+  // Storage keys (memoized to prevent infinite loops) - declared early for use in helpers
+  const keys = useMemo(() => getStorageKeys(storageKeyPrefix), [storageKeyPrefix]);
+
   // Normalize language codes to BCP-47 (e.g., en-US)
   const normalizeLanguageCode = useCallback(lang => {
     if (!lang) return null;
@@ -125,21 +147,52 @@ export const OxyProvider = ({
     };
     return map[lang] || lang;
   }, []);
-  // Add a new state to track token restoration
-  const [tokenReady, setTokenReady] = useState(false);
 
-  // Storage keys (memoized to prevent infinite loops)
-  const keys = useMemo(() => getStorageKeys(storageKeyPrefix), [storageKeyPrefix]);
+  // Helper to apply language preference from user/server
+  const applyLanguagePreference = useCallback(async user => {
+    const userLanguage = user?.language;
+    if (!userLanguage || !storage) return;
+    try {
+      const serverLang = normalizeLanguageCode(userLanguage) || userLanguage;
+      await storage.setItem(keys.language, serverLang);
+      setCurrentLanguage(serverLang);
+    } catch (e) {
+      if (__DEV__) {
+        console.warn('Failed to apply server language preference', e);
+      }
+    }
+  }, [storage, keys.language, normalizeLanguageCode]);
+
+  // Helper to map server sessions to client sessions
+  const mapServerSessionsToClient = useCallback((serverSessions, fallbackUserId) => {
+    return serverSessions.map(s => ({
+      sessionId: s.sessionId,
+      deviceId: s.deviceId,
+      expiresAt: s.expiresAt || new Date().toISOString(),
+      lastActive: s.lastActive || new Date().toISOString(),
+      userId: s.userId || fallbackUserId
+    }));
+  }, []);
+
+  // Token ready state - start optimistically so children render immediately
+  const [tokenReady, setTokenReady] = useState(true);
 
-  // Clear all storage - defined before initAuth to avoid dependency issues
+  // Clear all storage
   const clearAllStorage = useCallback(async () => {
     if (!storage) return;
     try {
       await storage.removeItem(keys.activeSessionId);
     } catch (err) {
-      console.error('Clear storage error:', err);
+      if (__DEV__) {
+        console.error('Clear storage error:', err);
+      }
+      onError?.({
+        message: 'Failed to clear storage',
+        code: 'STORAGE_ERROR',
+        status: 500
+      });
     }
-  }, [storage, keys]);
+  }, [storage, keys, onError]);
 
   // Initialize storage
   useEffect(() => {
@@ -148,25 +201,28 @@ export const OxyProvider = ({
         const platformStorage = await getStorage();
         setStorage(platformStorage);
       } catch (error) {
-        console.error('Init storage failed', error);
+        const errorMessage = error instanceof Error ? error.message : 'Failed to initialize storage';
         useAuthStore.setState({
-          error: 'Failed to initialize storage'
+          error: errorMessage
+        });
+        onError?.({
+          message: errorMessage,
+          code: 'STORAGE_INIT_ERROR',
+          status: 500
         });
       }
     };
     initStorage();
-  }, []);
+  }, [onError]);
 
   // Initialize authentication state
+  // Note: We don't set isLoading during initialization to avoid showing spinners
+  // Children render immediately and can check isTokenReady/isAuthenticated themselves
   useEffect(() => {
     const initAuth = async () => {
       if (!storage) return;
-      useAuthStore.setState({
-        isLoading: true
-      });
+      // Don't set isLoading during initialization - let it happen in background
       try {
-        setTokenReady(false);
-
         // Load saved language preference
         const savedLanguageRaw = await storage.getItem(keys.language);
         const savedLanguage = normalizeLanguageCode(savedLanguageRaw) || savedLanguageRaw;
@@ -191,67 +247,49 @@ export const OxyProvider = ({
                 username: fullUser.username,
                 avatar: fullUser.avatar
               });
-              // Apply server language if present
-              if (fullUser?.language) {
-                try {
-                  const serverLang = normalizeLanguageCode(fullUser.language) || fullUser.language;
-                  await storage.setItem(keys.language, serverLang);
-                  setCurrentLanguage(serverLang);
-                } catch (e) {
-                  console.warn('Failed to apply server language preference', e);
+              await applyLanguagePreference(fullUser);
+
+              // Get all device sessions to support multiple accounts
+              try {
+                const deviceSessions = await oxyServices.getDeviceSessions(storedActiveSessionId);
+                const allDeviceSessions = deviceSessions.map(ds => ({
+                  sessionId: ds.sessionId,
+                  deviceId: ds.deviceId,
+                  expiresAt: ds.expiresAt || new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toISOString(),
+                  lastActive: ds.lastActive || new Date().toISOString(),
+                  userId: ds.user?.id || ds.userId || fullUser.id
+                }));
+                setSessions(allDeviceSessions);
+              } catch (e) {
+                // Fallback to user sessions
+                if (__DEV__) {
+                  console.warn('Failed to get device sessions on init, falling back to user sessions:', e);
                 }
+                const serverSessions = await oxyServices.getSessionsBySessionId(storedActiveSessionId);
+                setSessions(mapServerSessionsToClient(serverSessions, fullUser.id));
               }
-              const serverSessions = await oxyServices.getSessionsBySessionId(storedActiveSessionId);
-              const clientSessions = serverSessions.map(s => ({
-                sessionId: s.sessionId,
-                deviceId: s.deviceId,
-                expiresAt: s.expiresAt || new Date().toISOString(),
-                lastActive: s.lastActive || new Date().toISOString(),
-                userId: s.userId || fullUser.id
-              }));
-              setSessions(clientSessions);
               onAuthStateChange?.(fullUser);
             } else {
               await clearAllStorage();
             }
           } catch (e) {
-            console.error('Session validation error', e);
+            if (__DEV__) {
+              console.error('Session validation error', e);
+            }
             await clearAllStorage();
           }
         }
         setTokenReady(true);
       } catch (e) {
-        console.error('Auth init error', e);
+        if (__DEV__) {
+          console.error('Auth init error', e);
+        }
         await clearAllStorage();
-      } finally {
-        useAuthStore.setState({
-          isLoading: false
-        });
+        setTokenReady(true);
       }
     };
     initAuth();
-  }, [storage, oxyServices, keys, onAuthStateChange, loginSuccess, clearAllStorage]);
-
-  // Remove invalid session - refresh sessions from backend
-  const removeInvalidSession = useCallback(async sessionId => {
-    // Remove from local state
-    const filteredSessions = sessions.filter(s => s.sessionId !== sessionId);
-    setSessions(filteredSessions);
-
-    // If there are other sessions, switch to the first one
-    if (filteredSessions.length > 0) {
-      await switchToSession(filteredSessions[0].sessionId);
-    } else {
-      // No valid sessions left
-      setActiveSessionId(null);
-      logoutStore();
-      setMinimalUser(null);
-      await storage?.removeItem(keys.activeSessionId);
-      if (onAuthStateChange) {
-        onAuthStateChange(null);
-      }
-    }
-  }, [sessions, storage, keys, onAuthStateChange, logoutStore]);
+  }, [storage, oxyServices, keys, onAuthStateChange, loginSuccess, clearAllStorage, applyLanguagePreference, mapServerSessionsToClient]);
 
   // Save active session ID to storage (only session ID, no user data)
   const saveActiveSessionId = useCallback(async sessionId => {
@@ -262,15 +300,23 @@ export const OxyProvider = ({
   // Switch to a different session
   const switchToSession = useCallback(async sessionId => {
     try {
-      useAuthStore.setState({
-        isLoading: true
+      // Don't set isLoading - session switches should happen silently in background
+      // Validate session first before attempting to switch
+      const validation = await oxyServices.validateSession(sessionId, {
+        useHeaderValidation: true
       });
+      if (!validation.valid) {
+        // Session is invalid, remove it from the sessions list
+        setSessions(prevSessions => prevSessions.filter(s => s.sessionId !== sessionId));
+        throw new Error('Session is invalid or expired');
+      }
 
       // Get access token for this session
       await oxyServices.getTokenBySession(sessionId);
+      setTokenReady(true);
 
-      // Load full user data
-      const fullUser = await oxyServices.getUserBySession(sessionId);
+      // Load full user data - use user from validation if available, otherwise fetch
+      const fullUser = validation.user || (await oxyServices.getUserBySession(sessionId));
       setActiveSessionId(sessionId);
       loginSuccess(fullUser);
       setMinimalUser({
@@ -279,30 +325,88 @@ export const OxyProvider = ({
         avatar: fullUser.avatar
       });
       await saveActiveSessionId(sessionId);
-      // Apply server language if present
-      if (fullUser?.language) {
-        try {
-          const serverLang = normalizeLanguageCode(fullUser.language) || fullUser.language;
-          await storage?.setItem(keys.language, serverLang);
-          setCurrentLanguage(serverLang);
-        } catch (e) {
-          console.warn('Failed to apply server language after switch', e);
+      await applyLanguagePreference(fullUser);
+
+      // Refresh all device sessions after switching
+      // Preserve existing sessions from other users to avoid losing accounts
+      try {
+        const deviceSessions = await oxyServices.getDeviceSessions(sessionId);
+        const allDeviceSessions = deviceSessions.map(ds => ({
+          sessionId: ds.sessionId,
+          deviceId: ds.deviceId,
+          expiresAt: ds.expiresAt || new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toISOString(),
+          lastActive: ds.lastActive || new Date().toISOString(),
+          userId: ds.user?.id || ds.userId || fullUser.id
+        }));
+        // Merge with existing sessions to preserve other accounts
+        setSessions(prevSessions => {
+          const existingSessionIds = new Set(prevSessions.map(s => s.sessionId));
+          const newSessions = allDeviceSessions.filter(s => !existingSessionIds.has(s.sessionId));
+          // Combine existing sessions with new ones, prioritizing new data for existing sessions
+          const sessionMap = new Map(prevSessions.map(s => [s.sessionId, s]));
+          allDeviceSessions.forEach(s => sessionMap.set(s.sessionId, s));
+          return Array.from(sessionMap.values());
+        });
+      } catch (error) {
+        // Fallback to user sessions - merge with existing to preserve other accounts
+        if (__DEV__) {
+          console.warn('Failed to get device sessions after switch, falling back to user sessions:', error);
         }
+        const serverSessions = await oxyServices.getSessionsBySessionId(sessionId);
+        const userSessions = mapServerSessionsToClient(serverSessions, fullUser.id);
+        // Merge with existing sessions to preserve other accounts
+        setSessions(prevSessions => {
+          const existingSessionIds = new Set(prevSessions.map(s => s.sessionId));
+          const newSessions = userSessions.filter(s => !existingSessionIds.has(s.sessionId));
+          // Combine existing sessions with new ones, prioritizing new data for existing sessions
+          const sessionMap = new Map(prevSessions.map(s => [s.sessionId, s]));
+          userSessions.forEach(s => sessionMap.set(s.sessionId, s));
+          return Array.from(sessionMap.values());
+        });
       }
-      if (onAuthStateChange) {
-        onAuthStateChange(fullUser);
-      }
+      onAuthStateChange?.(fullUser);
     } catch (error) {
-      console.error('Switch session error:', error);
+      // Check if the error is due to invalid/expired session
+      const isInvalidSession = error?.response?.status === 401 || error?.message?.includes('Invalid or expired session') || error?.message?.includes('Session is invalid');
+      if (isInvalidSession) {
+        // Remove invalid session from the sessions list
+        setSessions(prevSessions => prevSessions.filter(s => s.sessionId !== sessionId));
+
+        // If this was the active session, try to switch to another valid session
+        if (sessionId === activeSessionId && sessions.length > 1) {
+          const otherSessions = sessions.filter(s => s.sessionId !== sessionId);
+          for (const otherSession of otherSessions) {
+            try {
+              const otherValidation = await oxyServices.validateSession(otherSession.sessionId, {
+                useHeaderValidation: true
+              });
+              if (otherValidation.valid) {
+                await switchToSession(otherSession.sessionId);
+                return;
+              }
+            } catch {
+              // Continue to next session
+              continue;
+            }
+          }
+        }
+      }
+      const errorMessage = error instanceof Error ? error.message : 'Failed to switch session';
+      if (__DEV__) {
+        console.error('Switch session error:', error);
+      }
       useAuthStore.setState({
-        error: 'Failed to switch session'
+        error: errorMessage
       });
-    } finally {
-      useAuthStore.setState({
-        isLoading: false
+      onError?.({
+        message: errorMessage,
+        code: isInvalidSession ? 'INVALID_SESSION' : 'SESSION_SWITCH_ERROR',
+        status: isInvalidSession ? 401 : 500
       });
+      setTokenReady(false);
+      throw error; // Re-throw so calling code can handle it
     }
-  }, [oxyServices, onAuthStateChange, loginSuccess, saveActiveSessionId]);
+  }, [oxyServices, onAuthStateChange, loginSuccess, saveActiveSessionId, applyLanguagePreference, mapServerSessionsToClient, onError, activeSessionId, sessions]);
 
   // Login method - only store session ID, retrieve data from backend
   const login = useCallback(async (username, password, deviceName) => {
@@ -312,59 +416,100 @@ export const OxyProvider = ({
       error: null
     });
     try {
-      // Get device fingerprint for enhanced device identification
       const deviceFingerprint = DeviceManager.getDeviceFingerprint();
-
-      // Get or generate persistent device info
       const deviceInfo = await DeviceManager.getDeviceInfo();
-      console.log('Auth - Using device fingerprint:', deviceFingerprint);
-      console.log('Auth - Using device ID:', deviceInfo.deviceId);
       const response = await oxyServices.signIn(username, password, deviceName || deviceInfo.deviceName || DeviceManager.getDefaultDeviceName(), deviceFingerprint);
 
       // Handle MFA requirement
-      if (response && response.mfaRequired) {
-        const err = new Error('Multi-factor authentication required');
-        err.code = 'MFA_REQUIRED';
-        err.mfaToken = response.mfaToken;
-        err.expiresAt = response.expiresAt;
-        throw err;
+      if (response && 'mfaRequired' in response && response.mfaRequired) {
+        const mfaError = new Error('Multi-factor authentication required');
+        mfaError.code = 'MFA_REQUIRED';
+        mfaError.mfaToken = response.mfaToken;
+        mfaError.expiresAt = response.expiresAt;
+        throw mfaError;
       }
+      const sessionResponse = response;
+      await oxyServices.getTokenBySession(sessionResponse.sessionId);
+      const fullUser = await oxyServices.getUserBySession(sessionResponse.sessionId);
 
-      // Set as active session (only store session ID)
-      setActiveSessionId(response.sessionId);
-      await saveActiveSessionId(response.sessionId);
+      // Get all device sessions to check for duplicates BEFORE setting the new session as active
+      // This returns all sessions on the device, not just for the current user
+      let allDeviceSessions = [];
+      try {
+        const deviceSessions = await oxyServices.getDeviceSessions(sessionResponse.sessionId);
+
+        // Map device sessions to client format
+        // Device sessions include user info, so we can map them directly
+        allDeviceSessions = deviceSessions.map(ds => ({
+          sessionId: ds.sessionId,
+          deviceId: ds.deviceId || sessionResponse.deviceId,
+          expiresAt: ds.expiresAt || new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toISOString(),
+          lastActive: ds.lastActive || new Date().toISOString(),
+          userId: ds.user?.id || ds.userId || ds.user?._id?.toString() || fullUser.id
+        }));
+      } catch (error) {
+        // Fallback to user sessions if device sessions fail
+        if (__DEV__) {
+          console.warn('Failed to get device sessions, falling back to user sessions:', error);
+        }
+        const serverSessions = await oxyServices.getSessionsBySessionId(sessionResponse.sessionId);
+        const userSessions = mapServerSessionsToClient(serverSessions, fullUser.id);
 
-      // Get access token for API calls
-      await oxyServices.getTokenBySession(response.sessionId);
+        // Merge with existing sessions to preserve other accounts
+        const existingSessionIds = new Set((sessions || []).map(s => s.sessionId));
+        const newSessions = userSessions.filter(s => !existingSessionIds.has(s.sessionId));
+        allDeviceSessions = [...(sessions || []), ...newSessions];
+      }
 
-      // Load full user data from backend
-      const fullUser = await oxyServices.getUserBySession(response.sessionId);
-      loginSuccess(fullUser);
-      setMinimalUser(response.user);
-
-      // Load sessions from backend
-      const serverSessions = await oxyServices.getSessionsBySessionId(response.sessionId);
-      const clientSessions = serverSessions.map(serverSession => ({
-        sessionId: serverSession.sessionId,
-        deviceId: serverSession.deviceId,
-        expiresAt: serverSession.expiresAt || new Date().toISOString(),
-        lastActive: serverSession.lastActive || new Date().toISOString(),
-        userId: serverSession.userId || fullUser.id
-      }));
-      setSessions(clientSessions);
-      if (onAuthStateChange) {
-        onAuthStateChange(fullUser);
+      // Check if this user is already signed in with another session on this device
+      // Compare userId as string to handle both string and ObjectId formats
+      const userUserId = fullUser.id?.toString();
+      const existingSession = allDeviceSessions.find(s => s.userId?.toString() === userUserId && s.sessionId !== sessionResponse.sessionId);
+      if (existingSession) {
+        // User is already signed in on this device, switch to existing session instead
+        // Logout the newly created session to clean it up
+        try {
+          await oxyServices.logoutSession(sessionResponse.sessionId, sessionResponse.sessionId);
+        } catch (logoutError) {
+          if (__DEV__) {
+            console.warn('Failed to logout duplicate session:', logoutError);
+          }
+        }
+
+        // Switch to the existing session
+        await switchToSession(existingSession.sessionId);
+        loginSuccess(fullUser);
+        setMinimalUser(sessionResponse.user);
+
+        // Update sessions list (excluding the duplicate we just created)
+        setSessions(allDeviceSessions.filter(s => s.sessionId !== sessionResponse.sessionId));
+        onAuthStateChange?.(fullUser);
+        return fullUser;
       }
+
+      // No duplicate found, proceed with the new session
+      setActiveSessionId(sessionResponse.sessionId);
+      await saveActiveSessionId(sessionResponse.sessionId);
+      loginSuccess(fullUser);
+      setMinimalUser(sessionResponse.user);
+      setSessions(allDeviceSessions);
+      onAuthStateChange?.(fullUser);
       return fullUser;
     } catch (error) {
-      loginFailure(error.message || 'Login failed');
+      const errorMessage = error instanceof Error ? error.message : 'Login failed';
+      loginFailure(errorMessage);
+      onError?.({
+        message: errorMessage,
+        code: 'LOGIN_ERROR',
+        status: 401
+      });
       throw error;
     } finally {
       useAuthStore.setState({
         isLoading: false
       });
     }
-  }, [storage, oxyServices, saveActiveSessionId, loginSuccess, setMinimalUser, onAuthStateChange, loginFailure]);
+  }, [storage, oxyServices, saveActiveSessionId, loginSuccess, onAuthStateChange, loginFailure, mapServerSessionsToClient, onError, sessions, switchToSession]);
 
   // Logout method
   const logout = useCallback(async targetSessionId => {
@@ -394,74 +539,59 @@ export const OxyProvider = ({
         }
       }
     } catch (error) {
-      console.error('Logout error:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Logout failed';
+      if (__DEV__) {
+        console.error('Logout error:', error);
+      }
       useAuthStore.setState({
-        error: 'Logout failed'
+        error: errorMessage
+      });
+      onError?.({
+        message: errorMessage,
+        code: 'LOGOUT_ERROR',
+        status: 500
       });
     }
-  }, [activeSessionId, oxyServices, sessions, switchToSession, logoutStore, setMinimalUser, storage, keys.activeSessionId, onAuthStateChange]);
+  }, [activeSessionId, oxyServices, sessions, switchToSession, logoutStore, storage, keys.activeSessionId, onAuthStateChange, onError]);
 
   // Logout all sessions
   const logoutAll = useCallback(async () => {
-    console.log('logoutAll called with activeSessionId:', activeSessionId);
     if (!activeSessionId) {
-      console.error('No active session ID found, cannot logout all');
+      const error = new Error('No active session found');
       useAuthStore.setState({
-        error: 'No active session found'
+        error: error.message
       });
-      throw new Error('No active session found');
-    }
-    if (!oxyServices) {
-      console.error('OxyServices not initialized');
-      useAuthStore.setState({
-        error: 'Service not available'
+      onError?.({
+        message: error.message,
+        code: 'NO_SESSION_ERROR',
+        status: 404
       });
-      throw new Error('Service not available');
+      throw error;
     }
     try {
-      console.log('Calling oxyServices.logoutAllSessions with sessionId:', activeSessionId);
       await oxyServices.logoutAllSessions(activeSessionId);
-      console.log('logoutAllSessions completed successfully');
-
-      // Clear all local data
       setSessions([]);
       setActiveSessionId(null);
       logoutStore();
       setMinimalUser(null);
       await clearAllStorage();
-      console.log('Local storage cleared');
-      if (onAuthStateChange) {
-        onAuthStateChange(null);
-        console.log('Auth state change callback called');
-      }
+      onAuthStateChange?.(null);
     } catch (error) {
-      console.error('Logout all error:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Logout all failed';
       useAuthStore.setState({
-        error: `Logout all failed: ${error instanceof Error ? error.message : 'Unknown error'}`
+        error: errorMessage
+      });
+      onError?.({
+        message: errorMessage,
+        code: 'LOGOUT_ALL_ERROR',
+        status: 500
       });
       throw error;
     }
-  }, [activeSessionId, oxyServices, logoutStore, setMinimalUser, clearAllStorage, onAuthStateChange]);
+  }, [activeSessionId, oxyServices, logoutStore, clearAllStorage, onAuthStateChange, onError]);
 
-  // Effect to restore token on app load or session switch
-  useEffect(() => {
-    const restoreToken = async () => {
-      if (activeSessionId && oxyServices) {
-        try {
-          await oxyServices.getTokenBySession(activeSessionId);
-          setTokenReady(true);
-        } catch (err) {
-          // If token restoration fails, force logout
-          await logout();
-          setTokenReady(false);
-        }
-      } else {
-        setTokenReady(true); // No session, so token is not needed
-      }
-    };
-    restoreToken();
-    // Only run when activeSessionId or oxyServices changes
-  }, [activeSessionId, oxyServices, logout]);
+  // Token restoration is handled in initAuth and switchToSession
+  // No separate effect needed - children render immediately with isTokenReady available
 
   // Sign up method
   const signUp = useCallback(async (username, email, password) => {
@@ -471,23 +601,24 @@ export const OxyProvider = ({
       error: null
     });
     try {
-      // Create new account using the OxyServices signUp method
-      const response = await oxyServices.signUp(username, email, password);
-      console.log('SignUp successful:', response);
-
-      // Now log the user in to create a session
-      // This will handle the session creation and device registration
+      await oxyServices.signUp(username, email, password);
       const user = await login(username, password);
       return user;
     } catch (error) {
-      loginFailure(error.message || 'Sign up failed');
+      const errorMessage = error instanceof Error ? error.message : 'Sign up failed';
+      loginFailure(errorMessage);
+      onError?.({
+        message: errorMessage,
+        code: 'SIGNUP_ERROR',
+        status: 400
+      });
       throw error;
     } finally {
       useAuthStore.setState({
         isLoading: false
       });
     }
-  }, [storage, oxyServices, login, loginFailure]);
+  }, [storage, oxyServices, login, loginFailure, onError]);
 
   // Complete MFA login by verifying TOTP
   const completeMfaLogin = useCallback(async (mfaToken, code) => {
@@ -512,116 +643,127 @@ export const OxyProvider = ({
         username: fullUser.username,
         avatar: fullUser.avatar
       });
-      // Apply server language if present
-      if (fullUser?.language) {
-        try {
-          const serverLang = normalizeLanguageCode(fullUser.language) || fullUser.language;
-          await storage.setItem(keys.language, serverLang);
-          setCurrentLanguage(serverLang);
-        } catch (e) {
-          console.warn('Failed to apply server language on MFA login', e);
-        }
-      }
+      await applyLanguagePreference(fullUser);
 
-      // Load sessions list
-      const serverSessions = await oxyServices.getSessionsBySessionId(response.sessionId);
-      const clientSessions = serverSessions.map(s => ({
-        sessionId: s.sessionId,
-        deviceId: s.deviceId,
-        expiresAt: s.expiresAt || new Date().toISOString(),
-        lastActive: s.lastActive || new Date().toISOString(),
-        userId: s.userId || fullUser.id
-      }));
-      setSessions(clientSessions);
-      // Apply server language if present
-      if (fullUser?.language) {
-        try {
-          await storage.setItem(keys.language, fullUser.language);
-          setCurrentLanguage(fullUser.language);
-        } catch (e) {
-          console.warn('Failed to apply server language on MFA login', e);
+      // Get all device sessions to support multiple accounts
+      try {
+        const deviceSessions = await oxyServices.getDeviceSessions(response.sessionId);
+        const allDeviceSessions = deviceSessions.map(ds => ({
+          sessionId: ds.sessionId,
+          deviceId: ds.deviceId,
+          expiresAt: ds.expiresAt || new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toISOString(),
+          lastActive: ds.lastActive || new Date().toISOString(),
+          userId: ds.user?.id || ds.userId || fullUser.id
+        }));
+        setSessions(allDeviceSessions);
+      } catch (error) {
+        // Fallback to user sessions if device sessions fail
+        if (__DEV__) {
+          console.warn('Failed to get device sessions for MFA, falling back to user sessions:', error);
         }
+        const serverSessions = await oxyServices.getSessionsBySessionId(response.sessionId);
+        const userSessions = mapServerSessionsToClient(serverSessions, fullUser.id);
+
+        // Merge with existing sessions to preserve other accounts
+        setSessions(prevSessions => {
+          const existingSessionIds = new Set(prevSessions.map(s => s.sessionId));
+          const newSessions = userSessions.filter(s => !existingSessionIds.has(s.sessionId));
+          return [...prevSessions, ...newSessions];
+        });
       }
-      if (onAuthStateChange) onAuthStateChange(fullUser);
+      onAuthStateChange?.(fullUser);
       return fullUser;
     } catch (error) {
-      loginFailure(error.message || 'MFA verification failed');
+      const errorMessage = error instanceof Error ? error.message : 'MFA verification failed';
+      loginFailure(errorMessage);
+      onError?.({
+        message: errorMessage,
+        code: 'MFA_ERROR',
+        status: 401
+      });
       throw error;
     } finally {
       useAuthStore.setState({
         isLoading: false
       });
     }
-  }, [storage, oxyServices, loginSuccess, loginFailure, saveActiveSessionId, onAuthStateChange]);
+  }, [storage, oxyServices, loginSuccess, loginFailure, saveActiveSessionId, onAuthStateChange, applyLanguagePreference, onError]);
 
-  // Switch session method
+  // Switch session method (wrapper for consistency)
   const switchSession = useCallback(async sessionId => {
     await switchToSession(sessionId);
   }, [switchToSession]);
 
-  // Remove session method
+  // Remove session method (wrapper for consistency)
   const removeSession = useCallback(async sessionId => {
     await logout(sessionId);
   }, [logout]);
 
   // Refresh sessions method
   const refreshSessions = useCallback(async () => {
-    console.log('refreshSessions called with activeSessionId:', activeSessionId);
-    if (!activeSessionId) {
-      console.log('refreshSessions: No activeSessionId, returning');
-      return;
-    }
+    if (!activeSessionId) return;
     try {
-      console.log('refreshSessions: Calling getSessionsBySessionId...');
-      const serverSessions = await oxyServices.getSessionsBySessionId(activeSessionId);
-      console.log('refreshSessions: Server sessions received:', serverSessions);
-
-      // Update local sessions with server data
-      const updatedSessions = serverSessions.map(serverSession => ({
-        sessionId: serverSession.sessionId,
-        deviceId: serverSession.deviceId,
-        expiresAt: serverSession.expiresAt || new Date().toISOString(),
-        lastActive: serverSession.lastActive || new Date().toISOString(),
-        userId: serverSession.userId || user?.id
+      // Get all device sessions to support multiple accounts
+      const deviceSessions = await oxyServices.getDeviceSessions(activeSessionId);
+      const allDeviceSessions = deviceSessions.map(ds => ({
+        sessionId: ds.sessionId,
+        deviceId: ds.deviceId,
+        expiresAt: ds.expiresAt || new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toISOString(),
+        lastActive: ds.lastActive || new Date().toISOString(),
+        userId: ds.user?.id || ds.userId || user?.id
       }));
-      console.log('refreshSessions: Updated sessions:', updatedSessions);
-      setSessions(updatedSessions);
-      console.log('refreshSessions: Sessions updated in state');
+      setSessions(allDeviceSessions);
     } catch (error) {
-      console.error('Refresh sessions error:', error);
+      // Fallback to user sessions if device sessions fail
+      // Merge with existing sessions to preserve other accounts
+      if (__DEV__) {
+        console.warn('Failed to refresh device sessions, falling back to user sessions:', error);
+      }
+      try {
+        const serverSessions = await oxyServices.getSessionsBySessionId(activeSessionId);
+        const userSessions = mapServerSessionsToClient(serverSessions, user?.id);
+        // Merge with existing sessions to preserve other accounts
+        setSessions(prevSessions => {
+          const existingSessionIds = new Set(prevSessions.map(s => s.sessionId));
+          const newSessions = userSessions.filter(s => !existingSessionIds.has(s.sessionId));
+          // Combine existing sessions with new ones, prioritizing new data for existing sessions
+          const sessionMap = new Map(prevSessions.map(s => [s.sessionId, s]));
+          userSessions.forEach(s => sessionMap.set(s.sessionId, s));
+          return Array.from(sessionMap.values());
+        });
+      } catch (fallbackError) {
+        if (__DEV__) {
+          console.error('Refresh sessions error:', fallbackError);
+        }
 
-      // If the current session is invalid, try to find another valid session
-      if (sessions.length > 1) {
-        console.log('Current session invalid, trying to switch to another session...');
-        const otherSessions = sessions.filter(s => s.sessionId !== activeSessionId);
-        for (const session of otherSessions) {
-          try {
-            // Try to validate this session
-            await oxyServices.validateSession(session.sessionId, {
-              useHeaderValidation: true
-            });
-            console.log('Found valid session, switching to:', session.sessionId);
-            await switchToSession(session.sessionId);
-            return; // Successfully switched to another session
-          } catch (sessionError) {
-            console.log('Session validation failed for:', session.sessionId, sessionError);
-            continue; // Try next session
+        // If the current session is invalid, try to find another valid session
+        if (sessions.length > 1) {
+          const otherSessions = sessions.filter(s => s.sessionId !== activeSessionId);
+          for (const session of otherSessions) {
+            try {
+              const validation = await oxyServices.validateSession(session.sessionId, {
+                useHeaderValidation: true
+              });
+              if (validation.valid) {
+                await switchToSession(session.sessionId);
+                return;
+              }
+            } catch {
+              continue;
+            }
           }
         }
-      }
 
-      // If no valid sessions found, clear all sessions
-      console.log('No valid sessions found, clearing all sessions');
-      setSessions([]);
-      setActiveSessionId(null);
-      logoutStore();
-      setMinimalUser(null);
-      await clearAllStorage();
-      if (onAuthStateChange) {
-        onAuthStateChange(null);
+        // No valid sessions found, clear all
+        setSessions([]);
+        setActiveSessionId(null);
+        logoutStore();
+        setMinimalUser(null);
+        await clearAllStorage();
+        onAuthStateChange?.(null);
       }
     }
-  }, [activeSessionId, oxyServices, user?.id, sessions, switchToSession, logoutStore, setMinimalUser, clearAllStorage, onAuthStateChange]);
+  }, [activeSessionId, oxyServices, user?.id, sessions, switchToSession, logoutStore, clearAllStorage, onAuthStateChange, mapServerSessionsToClient]);
 
   // Device management methods
   const getDeviceSessions = useCallback(async () => {
@@ -629,58 +771,67 @@ export const OxyProvider = ({
     try {
       return await oxyServices.getDeviceSessions(activeSessionId);
     } catch (error) {
-      console.error('Get device sessions error:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Failed to get device sessions';
+      onError?.({
+        message: errorMessage,
+        code: 'GET_DEVICE_SESSIONS_ERROR',
+        status: 500
+      });
       throw error;
     }
-  }, [activeSessionId, oxyServices]);
+  }, [activeSessionId, oxyServices, onError]);
   const logoutAllDeviceSessions = useCallback(async () => {
     if (!activeSessionId) throw new Error('No active session');
     try {
       await oxyServices.logoutAllDeviceSessions(activeSessionId);
-
-      // Clear all local sessions since we logged out from all devices
       setSessions([]);
       setActiveSessionId(null);
       logoutStore();
       setMinimalUser(null);
       await clearAllStorage();
-      if (onAuthStateChange) {
-        onAuthStateChange(null);
-      }
+      onAuthStateChange?.(null);
     } catch (error) {
-      console.error('Logout all device sessions error:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Failed to logout all device sessions';
+      onError?.({
+        message: errorMessage,
+        code: 'LOGOUT_ALL_DEVICES_ERROR',
+        status: 500
+      });
       throw error;
     }
-  }, [activeSessionId, oxyServices, logoutStore, setMinimalUser, clearAllStorage, onAuthStateChange]);
+  }, [activeSessionId, oxyServices, logoutStore, clearAllStorage, onAuthStateChange, onError]);
   const updateDeviceName = useCallback(async deviceName => {
     if (!activeSessionId) throw new Error('No active session');
     try {
       await oxyServices.updateDeviceName(activeSessionId, deviceName);
-
-      // Update local device info
       await DeviceManager.updateDeviceName(deviceName);
     } catch (error) {
-      console.error('Update device name error:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Failed to update device name';
+      onError?.({
+        message: errorMessage,
+        code: 'UPDATE_DEVICE_NAME_ERROR',
+        status: 500
+      });
       throw error;
     }
-  }, [activeSessionId, oxyServices]);
+  }, [activeSessionId, oxyServices, onError]);
 
   // Language management method
   const setLanguage = useCallback(async languageId => {
     if (!storage) throw new Error('Storage not initialized');
     try {
-      // Save language preference
       await storage.setItem(keys.language, languageId);
       setCurrentLanguage(languageId);
-      console.log(`Language changed to ${languageId}`);
-
-      // TODO: Here you can add any additional logic needed for app-wide language updates
-      // such as updating i18n configuration, refreshing translations, etc.
     } catch (error) {
-      console.error('Error saving language preference:', error);
+      const errorMessage = error instanceof Error ? error.message : 'Failed to save language preference';
+      onError?.({
+        message: errorMessage,
+        code: 'LANGUAGE_SAVE_ERROR',
+        status: 500
+      });
       throw error;
     }
-  }, [storage, keys.language]);
+  }, [storage, keys.language, onError]);
 
   // Bottom sheet control methods
   const showBottomSheet = useCallback(screenOrConfig => {
@@ -695,9 +846,8 @@ export const OxyProvider = ({
       } else if (bottomSheetRef.current.present) {
         if (__DEV__) console.log('Presenting bottom sheet');
         bottomSheetRef.current.present();
-      } else {
+      } else if (__DEV__) {
         console.warn('No expand or present method available on bottomSheetRef');
-        if (__DEV__) console.log('Available methods on bottomSheetRef.current:', Object.keys(bottomSheetRef.current));
       }
 
       // Then navigate to the specified screen if provided
@@ -715,10 +865,8 @@ export const OxyProvider = ({
           }
         }, 100);
       }
-    } else {
-      console.warn('bottomSheetRef is not available');
-      console.warn('To fix this, ensure you pass a bottomSheetRef to OxyProvider:');
-      console.warn('<OxyProvider baseURL="..." bottomSheetRef={yourBottomSheetRef}>');
+    } else if (__DEV__) {
+      console.warn('bottomSheetRef is not available. Pass a bottomSheetRef to OxyProvider.');
     }
   }, [bottomSheetRef]);
   const hideBottomSheet = useCallback(() => {
@@ -751,39 +899,15 @@ export const OxyProvider = ({
       if (mod && typeof mod.useFollow === 'function') {
         return mod.useFollow(userId);
       }
-      console.warn('useFollow module did not export a function as expected');
-      return {
-        isFollowing: false,
-        isLoading: false,
-        error: null,
-        toggleFollow: async () => {},
-        setFollowStatus: () => {},
-        fetchStatus: async () => {},
-        clearError: () => {},
-        followerCount: null,
-        followingCount: null,
-        isLoadingCounts: false,
-        fetchUserCounts: async () => {},
-        setFollowerCount: () => {},
-        setFollowingCount: () => {}
-      };
+      if (__DEV__) {
+        console.warn('useFollow module did not export a function as expected');
+      }
+      return createEmptyFollowHook()(userId);
     } catch (e) {
-      console.warn('Failed to dynamically load useFollow hook:', e);
-      return {
-        isFollowing: false,
-        isLoading: false,
-        error: null,
-        toggleFollow: async () => {},
-        setFollowStatus: () => {},
-        fetchStatus: async () => {},
-        clearError: () => {},
-        followerCount: null,
-        followingCount: null,
-        isLoadingCounts: false,
-        fetchUserCounts: async () => {},
-        setFollowerCount: () => {},
-        setFollowingCount: () => {}
-      };
+      if (__DEV__) {
+        console.warn('Failed to dynamically load useFollow hook:', e);
+      }
+      return createEmptyFollowHook()(userId);
     }
   };
   const contextValue = useMemo(() => ({
@@ -793,6 +917,7 @@ export const OxyProvider = ({
     activeSessionId,
     isAuthenticated,
     isLoading,
+    isTokenReady: tokenReady,
     error,
     currentLanguage,
     login,
@@ -816,21 +941,9 @@ export const OxyProvider = ({
   // Only depend on user ID, not the entire user object
   minimalUser?.id, sessions.length,
   // Only depend on sessions count, not the entire array
-  activeSessionId, isAuthenticated, isLoading, error, currentLanguage, login, logout, logoutAll, signUp, completeMfaLogin, switchSession, removeSession, refreshSessions, setLanguage, getDeviceSessions, logoutAllDeviceSessions, updateDeviceName, oxyServices, bottomSheetRef, showBottomSheet, hideBottomSheet]);
-
-  // Wrap children rendering to block until token is ready
-  if (!tokenReady) {
-    return /*#__PURE__*/_jsx(View, {
-      style: {
-        flex: 1,
-        justifyContent: 'center',
-        alignItems: 'center'
-      },
-      children: /*#__PURE__*/_jsx(Text, {
-        children: "Loading authentication..."
-      })
-    });
-  }
+  activeSessionId, isAuthenticated, isLoading, tokenReady, error, currentLanguage, login, logout, logoutAll, signUp, completeMfaLogin, switchSession, removeSession, refreshSessions, setLanguage, getDeviceSessions, logoutAllDeviceSessions, updateDeviceName, oxyServices, bottomSheetRef, showBottomSheet, hideBottomSheet]);
+
+  // Always render children - let the consuming app decide how to handle token loading state
   return /*#__PURE__*/_jsx(OxyContext.Provider, {
     value: contextValue,
     children: children