const colors = {
  reset: "\x1b[0m",
  bright: "\x1b[1m",
  dim: "\x1b[2m",
  green: "\x1b[32m",
  yellow: "\x1b[33m",
  red: "\x1b[31m",
  blue: "\x1b[34m",
};

export class MCPClientLogger {
  private readonly prefix: string;
  private readonly debugMode: boolean;

  constructor(
    prefix: string = "NextChat MCP Client",
    debugMode: boolean = false,
  ) {
    this.prefix = prefix;
    this.debugMode = debugMode;
  }

  info(message: any) {
    this.log(colors.blue, message);
  }

  success(message: any) {
    this.log(colors.green, message);
  }

  error(message: any) {
    const formattedMessage = this.formatMessage(message);
    console.error(
      `${colors.red}${colors.bright}[${this.prefix}]${colors.reset} ${formattedMessage}`,
    );
  }

  warn(message: any) {
    this.log(colors.yellow, message);
  }

  debug(message: any) {
    if (this.debugMode) {
      this.log(colors.dim, message);
    }
  }

  private formatMessage(message: any): string {
    return typeof message === "object"
      ? JSON.stringify(message, null, 2)
      : message;
  }

  private log(color: string, message: any) {
    const formattedMessage = this.formatMessage(message);
    console.log(
      `${color}${colors.bright}[${this.prefix}]${colors.reset} ${formattedMessage}`,
    );
  }
}