docker-swarm-termina/frontend/lib/api.ts

import { triggerAuthError } from './store/authErrorHandler';

// Type definition for window.__ENV__
declare global {
  interface Window {
    __ENV__?: {
      NEXT_PUBLIC_API_URL?: string;
    };
  }
}

export const API_BASE_URL =
  typeof window !== 'undefined' && window.__ENV__?.NEXT_PUBLIC_API_URL
    ? window.__ENV__.NEXT_PUBLIC_API_URL
    : process.env.NEXT_PUBLIC_API_URL || 'http://localhost:5000';

export interface Container {
  id: string;
  name: string;
  image: string;
  status: string;
  uptime: string;
}

export interface AuthResponse {
  success: boolean;
  token?: string;
  username?: string;
  message?: string;
}

export interface ContainersResponse {
  containers: Container[];
}

export interface CommandResponse {
  success: boolean;
  output?: string;
  error?: string;
  workdir?: string;
  exit_code?: number;
}

export interface ContainerActionResponse {
  success: boolean;
  message?: string;
  error?: string;
}

class ApiClient {
  private token: string | null = null;

  setToken(token: string | null) {
    this.token = token;
    if (token) {
      localStorage.setItem('auth_token', token);
    } else {
      localStorage.removeItem('auth_token');
      localStorage.removeItem('auth_username');
    }
  }

  getToken(): string | null {
    if (!this.token && typeof window !== 'undefined') {
      this.token = localStorage.getItem('auth_token');
    }
    return this.token;
  }

  getUsername(): string | null {
    if (typeof window !== 'undefined') {
      return localStorage.getItem('auth_username');
    }
    return null;
  }

  setUsername(username: string | null) {
    if (typeof window !== 'undefined') {
      if (username) {
        localStorage.setItem('auth_username', username);
      } else {
        localStorage.removeItem('auth_username');
      }
    }
  }

  async login(username: string, password: string): Promise<AuthResponse> {
    const response = await fetch(`${API_BASE_URL}/api/auth/login`, {
      method: 'POST',
      headers: {
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({ username, password }),
    });

    const data = await response.json();
    if (data.success && data.token) {
      this.setToken(data.token);
      this.setUsername(data.username || username);
    }
    return data;
  }

  async logout(): Promise<void> {
    const token = this.getToken();
    if (token) {
      await fetch(`${API_BASE_URL}/api/auth/logout`, {
        method: 'POST',
        headers: {
          'Authorization': `Bearer ${token}`,
        },
      });
    }
    this.setToken(null);
  }

  async getContainers(): Promise<Container[]> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers`, {
      headers: {
        'Authorization': `Bearer ${token}`,
      },
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      throw new Error('Failed to fetch containers');
    }

    const data: ContainersResponse = await response.json();
    return data.containers;
  }

  async executeCommand(containerId: string, command: string): Promise<CommandResponse> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/exec`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${token}`,
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({ command }),
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      throw new Error('Failed to execute command');
    }

    return response.json();
  }

  async startContainer(containerId: string): Promise<ContainerActionResponse> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/start`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${token}`,
      },
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      const error = await response.json();
      throw new Error(error.error || 'Failed to start container');
    }

    return response.json();
  }

  async stopContainer(containerId: string): Promise<ContainerActionResponse> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/stop`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${token}`,
      },
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      const error = await response.json();
      throw new Error(error.error || 'Failed to stop container');
    }

    return response.json();
  }

  async restartContainer(containerId: string): Promise<ContainerActionResponse> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/restart`, {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${token}`,
      },
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      const error = await response.json();
      throw new Error(error.error || 'Failed to restart container');
    }

    return response.json();
  }

  async removeContainer(containerId: string): Promise<ContainerActionResponse> {
    const token = this.getToken();
    if (!token) {
      triggerAuthError();
      throw new Error('Not authenticated');
    }

    const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}`, {
      method: 'DELETE',
      headers: {
        'Authorization': `Bearer ${token}`,
      },
    });

    if (!response.ok) {
      if (response.status === 401) {
        this.setToken(null);
        triggerAuthError();
        throw new Error('Session expired');
      }
      const error = await response.json();
      throw new Error(error.error || 'Failed to remove container');
    }

    return response.json();
  }
}

export const apiClient = new ApiClient();