Tracearr/apps/mobile/src/lib/api.ts

/**
 * API client for Tracearr mobile app
 * Uses axios with automatic token refresh
 * Supports multiple servers with active server selection
 */
import axios from 'axios';
import type { AxiosInstance, AxiosError, InternalAxiosRequestConfig } from 'axios';
import { storage } from './storage';
import type {
  ActiveSession,
  DashboardStats,
  ServerUserWithIdentity,
  ServerUserDetail,
  Session,
  SessionWithDetails,
  UserLocation,
  UserDevice,
  Violation,
  ViolationWithDetails,
  Rule,
  Server,
  Settings,
  MobilePairResponse,
  PaginatedResponse,
  NotificationPreferences,
  NotificationPreferencesWithStatus,
  ServerResourceStats,
  TerminationLogWithDetails,
} from '@tracearr/shared';

// Cache of API clients per server
const apiClients = new Map<string, AxiosInstance>();
let activeServerId: string | null = null;

/**
 * Initialize or get the API client for the active server
 */
export async function getApiClient(): Promise<AxiosInstance> {
  const serverId = await storage.getActiveServerId();
  if (!serverId) {
    throw new Error('No server configured');
  }

  // If server changed, update active
  if (activeServerId !== serverId) {
    activeServerId = serverId;
  }

  // Check cache
  const cached = apiClients.get(serverId);
  if (cached) {
    return cached;
  }

  // Get server info
  const server = await storage.getServer(serverId);
  if (!server) {
    throw new Error('Server not found');
  }

  const client = createApiClient(server.url, serverId);
  apiClients.set(serverId, client);
  return client;
}

/**
 * Create a new API client for a given server
 */
export function createApiClient(baseURL: string, serverId: string): AxiosInstance {
  const client = axios.create({
    baseURL: `${baseURL}/api/v1`,
    timeout: 30000,
    headers: {
      'Content-Type': 'application/json',
    },
  });

  // Request interceptor - add auth token for this server
  client.interceptors.request.use(
    async (config: InternalAxiosRequestConfig) => {
      const credentials = await storage.getServerCredentials(serverId);
      if (credentials) {
        config.headers.Authorization = `Bearer ${credentials.accessToken}`;
      }
      return config;
    },
    (error: unknown) => Promise.reject(error instanceof Error ? error : new Error(String(error)))
  );

  // Response interceptor - handle token refresh for this server
  client.interceptors.response.use(
    (response) => response,
    async (error: AxiosError) => {
      const originalRequest = error.config as InternalAxiosRequestConfig & { _retry?: boolean };

      // If 401 and not already retrying, attempt token refresh
      if (error.response?.status === 401 && !originalRequest._retry) {
        originalRequest._retry = true;

        try {
          const credentials = await storage.getServerCredentials(serverId);
          if (!credentials?.refreshToken) {
            throw new Error('No refresh token');
          }

          const response = await client.post<{ accessToken: string; refreshToken: string }>(
            '/mobile/refresh',
            { refreshToken: credentials.refreshToken }
          );

          await storage.updateServerTokens(
            serverId,
            response.data.accessToken,
            response.data.refreshToken
          );

          // Retry original request with new token
          originalRequest.headers.Authorization = `Bearer ${response.data.accessToken}`;
          return await client(originalRequest);
        } catch {
          // Refresh failed - remove this server's client from cache
          apiClients.delete(serverId);
          throw new Error('Session expired');
        }
      }

      return Promise.reject(error);
    }
  );

  return client;
}

/**
 * Reset the API client cache (call when switching servers or logging out)
 */
export function resetApiClient(): void {
  apiClients.clear();
  activeServerId = null;
}

/**
 * Remove a specific server's client from cache
 */
export function removeApiClient(serverId: string): void {
  apiClients.delete(serverId);
}

/**
 * Get the current server URL (for building absolute URLs like images)
 */
export async function getServerUrl(): Promise<string | null> {
  return storage.getServerUrl();
}

/**
 * API methods organized by domain
 * All methods use the active server's client
 */
export const api = {
  /**
   * Pair with server using mobile token
   * This is called before we have a client, so it uses direct axios
   */
  pair: async (
    serverUrl: string,
    token: string,
    deviceName: string,
    deviceId: string,
    platform: 'ios' | 'android',
    deviceSecret?: string
  ): Promise<MobilePairResponse> => {
    try {
      const response = await axios.post<MobilePairResponse>(
        `${serverUrl}/api/v1/mobile/pair`,
        { token, deviceName, deviceId, platform, deviceSecret },
        { timeout: 15000 }
      );
      return response.data;
    } catch (error) {
      if (axios.isAxiosError(error)) {
        // Extract server's error message if available
        const serverMessage =
          error.response?.data?.message ||
          error.response?.data?.error;

        if (serverMessage) {
          throw new Error(serverMessage);
        }

        // Handle specific HTTP status codes
        if (error.response?.status === 429) {
          throw new Error('Too many pairing attempts. Please wait a few minutes.');
        }
        if (error.response?.status === 401) {
          throw new Error('Invalid or expired pairing token.');
        }
        if (error.response?.status === 400) {
          throw new Error('Invalid pairing request. Check your token.');
        }

        // Handle network errors
        if (error.code === 'ECONNABORTED') {
          throw new Error('Connection timed out. Check your server URL.');
        }
        if (error.code === 'ERR_NETWORK' || !error.response) {
          throw new Error('Cannot reach server. Check URL and network connection.');
        }

        // Fallback to axios message
        throw new Error(error.message);
      }
      throw error;
    }
  },

  /**
   * Register push token for notifications
   */
  registerPushToken: async (
    expoPushToken: string,
    deviceSecret?: string
  ): Promise<{ success: boolean; updatedSessions: number }> => {
    const client = await getApiClient();
    const response = await client.post<{ success: boolean; updatedSessions: number }>(
      '/mobile/push-token',
      { expoPushToken, deviceSecret }
    );
    return response.data;
  },

  /**
   * Dashboard stats
   */
  stats: {
    dashboard: async (serverId?: string): Promise<DashboardStats> => {
      const client = await getApiClient();
      const response = await client.get<DashboardStats>('/stats/dashboard', {
        params: serverId ? { serverId } : undefined,
      });
      return response.data;
    },
    plays: async (params?: {
      period?: string;
      serverId?: string;
    }): Promise<{ data: { date: string; count: number }[] }> => {
      const client = await getApiClient();
      const response = await client.get<{ data: { date: string; count: number }[] }>(
        '/stats/plays',
        { params }
      );
      return response.data;
    },
    playsByDayOfWeek: async (params?: {
      period?: string;
      serverId?: string;
    }): Promise<{ data: { day: number; name: string; count: number }[] }> => {
      const client = await getApiClient();
      const response = await client.get<{ data: { day: number; name: string; count: number }[] }>(
        '/stats/plays-by-dayofweek',
        { params }
      );
      return response.data;
    },
    playsByHourOfDay: async (params?: {
      period?: string;
      serverId?: string;
    }): Promise<{ data: { hour: number; count: number }[] }> => {
      const client = await getApiClient();
      const response = await client.get<{ data: { hour: number; count: number }[] }>(
        '/stats/plays-by-hourofday',
        { params }
      );
      return response.data;
    },
    platforms: async (params?: {
      period?: string;
      serverId?: string;
    }): Promise<{ data: { platform: string; count: number }[] }> => {
      const client = await getApiClient();
      const response = await client.get<{ data: { platform: string; count: number }[] }>(
        '/stats/platforms',
        { params }
      );
      return response.data;
    },
    quality: async (params?: { period?: string; serverId?: string }): Promise<{
      directPlay: number;
      transcode: number;
      total: number;
      directPlayPercent: number;
      transcodePercent: number;
    }> => {
      const client = await getApiClient();
      const response = await client.get<{
        directPlay: number;
        transcode: number;
        total: number;
        directPlayPercent: number;
        transcodePercent: number;
      }>('/stats/quality', { params });
      return response.data;
    },
    concurrent: async (params?: {
      period?: string;
      serverId?: string;
    }): Promise<{ data: { hour: string; maxConcurrent: number }[] }> => {
      const client = await getApiClient();
      const response = await client.get<{ data: { hour: string; maxConcurrent: number }[] }>(
        '/stats/concurrent',
        { params }
      );
      return response.data;
    },
    locations: async (params?: {
      serverId?: string;
      userId?: string;
    }): Promise<{
      data: {
        latitude: number;
        longitude: number;
        city: string;
        country: string;
        playCount: number;
      }[];
    }> => {
      const client = await getApiClient();
      const response = await client.get<{
        data: {
          latitude: number;
          longitude: number;
          city: string;
          country: string;
          playCount: number;
        }[];
      }>('/stats/locations', { params });
      return response.data;
    },
  },

  /**
   * Sessions
   */
  sessions: {
    active: async (serverId?: string): Promise<ActiveSession[]> => {
      const client = await getApiClient();
      const response = await client.get<{ data: ActiveSession[] }>('/sessions/active', {
        params: serverId ? { serverId } : undefined,
      });
      return response.data.data;
    },
    list: async (params?: {
      page?: number;
      pageSize?: number;
      userId?: string;
      serverId?: string;
    }) => {
      const client = await getApiClient();
      const response = await client.get<PaginatedResponse<ActiveSession>>('/sessions', { params });
      return response.data;
    },
    get: async (id: string): Promise<SessionWithDetails> => {
      const client = await getApiClient();
      const response = await client.get<SessionWithDetails>(`/sessions/${id}`);
      return response.data;
    },
    terminate: async (
      id: string,
      reason?: string
    ): Promise<{ success: boolean; terminationLogId: string; message: string }> => {
      const client = await getApiClient();
      const response = await client.post<{
        success: boolean;
        terminationLogId: string;
        message: string;
      }>(`/mobile/streams/${id}/terminate`, { reason });
      return response.data;
    },
  },

  /**
   * Users
   */
  users: {
    list: async (params?: { page?: number; pageSize?: number; serverId?: string }) => {
      const client = await getApiClient();
      const response = await client.get<PaginatedResponse<ServerUserWithIdentity>>('/users', {
        params,
      });
      return response.data;
    },
    get: async (id: string): Promise<ServerUserDetail> => {
      const client = await getApiClient();
      const response = await client.get<ServerUserDetail>(`/users/${id}`);
      return response.data;
    },
    sessions: async (id: string, params?: { page?: number; pageSize?: number }) => {
      const client = await getApiClient();
      const response = await client.get<PaginatedResponse<Session>>(`/users/${id}/sessions`, {
        params,
      });
      return response.data;
    },
    locations: async (id: string): Promise<UserLocation[]> => {
      const client = await getApiClient();
      const response = await client.get<{ data: UserLocation[] }>(`/users/${id}/locations`);
      return response.data.data;
    },
    devices: async (id: string): Promise<UserDevice[]> => {
      const client = await getApiClient();
      const response = await client.get<{ data: UserDevice[] }>(`/users/${id}/devices`);
      return response.data.data;
    },
    terminations: async (
      id: string,
      params?: { page?: number; pageSize?: number }
    ): Promise<PaginatedResponse<TerminationLogWithDetails>> => {
      const client = await getApiClient();
      const response = await client.get<PaginatedResponse<TerminationLogWithDetails>>(
        `/users/${id}/terminations`,
        { params }
      );
      return response.data;
    },
  },

  /**
   * Violations
   */
  violations: {
    list: async (params?: {
      page?: number;
      pageSize?: number;
      userId?: string;
      severity?: string;
      acknowledged?: boolean;
      serverId?: string;
    }) => {
      const client = await getApiClient();
      const response = await client.get<PaginatedResponse<ViolationWithDetails>>('/violations', {
        params,
      });
      return response.data;
    },
    acknowledge: async (id: string): Promise<Violation> => {
      const client = await getApiClient();
      const response = await client.patch<Violation>(`/violations/${id}`);
      return response.data;
    },
    dismiss: async (id: string): Promise<void> => {
      const client = await getApiClient();
      await client.delete(`/violations/${id}`);
    },
  },

  /**
   * Rules
   */
  rules: {
    list: async (serverId?: string): Promise<Rule[]> => {
      const client = await getApiClient();
      const response = await client.get<{ data: Rule[] }>('/rules', {
        params: serverId ? { serverId } : undefined,
      });
      return response.data.data;
    },
    toggle: async (id: string, isActive: boolean): Promise<Rule> => {
      const client = await getApiClient();
      const response = await client.patch<Rule>(`/rules/${id}`, { isActive });
      return response.data;
    },
  },

  /**
   * Servers
   */
  servers: {
    list: async (): Promise<Server[]> => {
      const client = await getApiClient();
      const response = await client.get<{ data: Server[] }>('/servers');
      return response.data.data;
    },
    statistics: async (id: string): Promise<ServerResourceStats> => {
      const client = await getApiClient();
      const response = await client.get<ServerResourceStats>(`/servers/${id}/statistics`);
      return response.data;
    },
  },

  /**
   * Notification preferences (per-device settings)
   */
  notifications: {
    /**
     * Get notification preferences for current device
     * Returns preferences with live rate limit status from Redis
     */
    getPreferences: async (): Promise<NotificationPreferencesWithStatus> => {
      const client = await getApiClient();
      const response = await client.get<NotificationPreferencesWithStatus>(
        '/notifications/preferences'
      );
      return response.data;
    },

    /**
     * Update notification preferences for current device
     * Supports partial updates - only send fields you want to change
     */
    updatePreferences: async (
      data: Partial<
        Omit<NotificationPreferences, 'id' | 'mobileSessionId' | 'createdAt' | 'updatedAt'>
      >
    ): Promise<NotificationPreferences> => {
      const client = await getApiClient();
      const response = await client.patch<NotificationPreferences>(
        '/notifications/preferences',
        data
      );
      return response.data;
    },

    /**
     * Send a test notification to verify push is working
     */
    sendTest: async (): Promise<{ success: boolean; message: string }> => {
      const client = await getApiClient();
      const response = await client.post<{ success: boolean; message: string }>(
        '/notifications/test'
      );
      return response.data;
    },
  },

  /**
   * Global settings (display preferences, etc.)
   */
  settings: {
    get: async (): Promise<Settings> => {
      const client = await getApiClient();
      const response = await client.get<Settings>('/settings');
      return response.data;
    },
  },
};