@oxyhq/services/src/ui/stores/authStore.ts

Reusable OxyHQ module to handle authentication, user management, karma system, device-based session management and more 🚀

import { create } from 'zustand';
import type { User } from '../../models/interfaces';

interface AuthState {
  user: User | null;
  isAuthenticated: boolean;
  isLoading: boolean;
  error: string | null;
  lastUserFetch: number | null; // Timestamp of last user fetch for caching
  loginSuccess: (user: User) => void;
  loginFailure: (error: string) => void;
  logout: () => void;
  fetchUser: (oxyServices: { getCurrentUser: () => Promise<User> }, forceRefresh?: boolean) => Promise<void>;
  updateUser: (updates: Partial<User>, oxyServices: { updateProfile: (updates: Partial<User>) => Promise<User>; getCurrentUser: () => Promise<User> }) => Promise<void>;
  setUser: (user: User) => void; // Direct user setter for caching
}

export const useAuthStore = create<AuthState>((set: (state: Partial<AuthState>) => void, get: () => AuthState) => ({
  user: null,
  isAuthenticated: false,
  isLoading: false,
  error: null,
  lastUserFetch: null,
  loginSuccess: (user: User) => set({ isLoading: false, isAuthenticated: true, user, lastUserFetch: Date.now() }),
  loginFailure: (error: string) => set({ isLoading: false, error }),
  logout: () => set({ user: null, isAuthenticated: false, lastUserFetch: null }),
  setUser: (user: User) => set({ user, lastUserFetch: Date.now() }),
  fetchUser: async (oxyServices, forceRefresh = false) => {
    const state = get();
    const now = Date.now();
    const cacheAge = state.lastUserFetch ? now - state.lastUserFetch : Number.POSITIVE_INFINITY;
    const cacheValid = cacheAge < 5 * 60 * 1000; // 5 minutes cache

    // Use cached data if available and not forcing refresh
    if (!forceRefresh && state.user && cacheValid) {
      if (__DEV__) {
      console.log('AuthStore: Using cached user data (age:', cacheAge, 'ms)');
      }
      return;
    }

    set({ isLoading: true, error: null });
    try {
      const user = await oxyServices.getCurrentUser();
      set({ user, isLoading: false, isAuthenticated: true, lastUserFetch: now });
    } catch (error) {
      const errorMessage = error instanceof Error ? error.message : 'Failed to fetch user';
      if (__DEV__) {
      console.error('AuthStore: Error fetching user:', error);
      }
      set({ error: errorMessage, isLoading: false });
    }
  },
  updateUser: async (updates, oxyServices) => {
    set({ isLoading: true, error: null });
    try {
      await oxyServices.updateProfile(updates);
      // Immediately fetch the latest user data after update
      // Use arrow function to preserve 'this' context
      await useAuthStore.getState().fetchUser({ getCurrentUser: () => oxyServices.getCurrentUser() }, true);
    } catch (error) {
      const errorMessage = error instanceof Error ? error.message : 'Failed to update user';
      if (__DEV__) {
      console.error('AuthStore: Error updating user:', error);
      }
      set({ error: errorMessage, isLoading: false });
    }
  },
}));