import chalk from "chalk";
import { Configuration } from "@nestbox-ai/admin";
import { getAuthToken } from "./auth";

export interface AuthResult {
  authToken: {
    token: string;
    serverUrl: string;
    accessToken?: string;
  };
  configuration: Configuration;
}

/**
 * Common authentication and configuration setup
 * Returns authentication token and configured API client
 * Exits the process if authentication fails
 */
export function setupAuthAndConfig(): AuthResult | null {
  const authToken = getAuthToken();
  
  if (!authToken) {
    console.error(chalk.red('No authentication token found. Please login first.'));
    return null;
  }
  
  const configuration = new Configuration({
    basePath: authToken.serverUrl,
    baseOptions: {
      headers: {
        "Authorization": authToken.token,
      }
    }
  });

  return {
    authToken,
    configuration
  };
}

/**
 * Wrapper function that ensures authentication is set up before executing a callback
 * Automatically handles authentication errors and provides configured API instances
 */
export async function withAuth<T>(
  callback: (authResult: AuthResult) => Promise<T>
): Promise<T | void> {
  const authResult = setupAuthAndConfig();
  
  if (!authResult) {
    return;
  }

  try {
    return await callback(authResult);
  } catch (error: any) {
    if (error.response && error.response.status === 401) {
      console.error(chalk.red('Authentication token has expired. Please login again using "nestbox login <domain>".'));
    } else {
      throw error; // Re-throw non-auth errors
    }
  }
}