maternal-app/maternal-web/lib/utils/tokenStorage.ts

/**
 * Safe token storage utilities that work with both SSR and client-side rendering
 */

export const tokenStorage = {
  /**
   * Get access token from storage
   */
  getAccessToken: (): string | null => {
    if (typeof window === 'undefined') {
      return null;
    }
    try {
      return localStorage.getItem('accessToken');
    } catch (error) {
      console.error('Error reading accessToken:', error);
      return null;
    }
  },

  /**
   * Get refresh token from storage
   */
  getRefreshToken: (): string | null => {
    if (typeof window === 'undefined') {
      return null;
    }
    try {
      return localStorage.getItem('refreshToken');
    } catch (error) {
      console.error('Error reading refreshToken:', error);
      return null;
    }
  },

  /**
   * Set access token in storage
   */
  setAccessToken: (token: string): void => {
    if (typeof window === 'undefined') {
      return;
    }
    try {
      localStorage.setItem('accessToken', token);
    } catch (error) {
      console.error('Error setting accessToken:', error);
    }
  },

  /**
   * Set refresh token in storage
   */
  setRefreshToken: (token: string): void => {
    if (typeof window === 'undefined') {
      return;
    }
    try {
      localStorage.setItem('refreshToken', token);
    } catch (error) {
      console.error('Error setting refreshToken:', error);
    }
  },

  /**
   * Get device ID from storage
   */
  getDeviceId: (): string | null => {
    if (typeof window === 'undefined') {
      return null;
    }
    try {
      return localStorage.getItem('deviceId');
    } catch (error) {
      console.error('Error reading deviceId:', error);
      return null;
    }
  },

  /**
   * Set device ID in storage
   */
  setDeviceId: (deviceId: string): void => {
    if (typeof window === 'undefined') {
      return;
    }
    try {
      localStorage.setItem('deviceId', deviceId);
    } catch (error) {
      console.error('Error setting deviceId:', error);
    }
  },

  /**
   * Set both tokens at once
   */
  setTokens: (accessToken: string, refreshToken: string, deviceId?: string): void => {
    tokenStorage.setAccessToken(accessToken);
    tokenStorage.setRefreshToken(refreshToken);
    if (deviceId) {
      tokenStorage.setDeviceId(deviceId);
    }
  },

  /**
   * Clear all tokens from storage
   */
  clearTokens: (): void => {
    if (typeof window === 'undefined') {
      return;
    }
    try {
      localStorage.removeItem('accessToken');
      localStorage.removeItem('refreshToken');
      localStorage.removeItem('deviceId');
    } catch (error) {
      console.error('Error clearing tokens:', error);
    }
  },

  /**
   * Check if user has valid tokens
   */
  hasTokens: (): boolean => {
    return !!(tokenStorage.getAccessToken() && tokenStorage.getRefreshToken());
  },
};