/**
 * Test configuration and setup utilities
 * This file provides configuration for running tests with different LLM providers
 */

// Environment variable configuration
export interface TestConfig {
  baseUrl: string;
  apiKey: string;
  model: string;
  claudeApiKey?: string;
  openaiApiKey?: string;
  geminiApiKey?: string;
}

// Get test configuration from environment variables
export function getTestConfig(): TestConfig {
  return {
    baseUrl: Bun.env.ZAI_BASE_URL || 'https://api.openai.com/v1',
    apiKey: Bun.env.ZAI_API_KEY || 'test-api-key',
    model: Bun.env.TEST_MODEL || 'glm-4.6',
    claudeApiKey: Bun.env.CLAUDE_API_KEY,
    openaiApiKey: Bun.env.OPENAI_API_KEY,
    geminiApiKey: Bun.env.GEMINI_API_KEY
  };
}

// Check if we have real API credentials for integration testing
export function hasIntegrationCredentials(): {
  claude: boolean;
  openai: boolean;
  gemini: boolean;
  any: boolean;
} {
  const config = getTestConfig();

  return {
    claude: !!config.claudeApiKey,
    openai: !!config.openaiApiKey,
    gemini: !!config.geminiApiKey,
    any: !!(config.claudeApiKey || config.openaiApiKey || config.geminiApiKey)
  };
}

// Mock API responses for testing without real credentials
export const mockResponses = {
  claude: {
    success: 'SUCCESS',
    counting: '1\n2\n3\n4\n5',
    error: 'Mock Claude error'
  },
  openai: {
    success: 'SUCCESS',
    error: 'Mock OpenAI error'
  },
  gemini: {
    success: 'SUCCESS',
    error: 'Mock Gemini error'
  }
};

// Test data generators
export const testData = {
  simpleText: () => 'Hello, this is a simple test message.',
  longText: () => 'This is a longer test message. '.repeat(100),
  unicodeText: () => 'Hello 🌍! 测试中文! Тест на русском! العربية!',
  specialChars: () => '!@#$%^&*()_+-=[]{}|;:,.<>?/~`',
  emptyString: () => '',
  whitespaceOnly: () => '   \n\t   ',

  // Input token generators
  textTokens: (count: number = 3) =>
    Array.from({ length: count }, (_, i) => ({ text: `Token ${i + 1}: ` })),

  imageTokens: (count: number = 1) =>
    Array.from({ length: count }, () => ({
      img: 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg=='
    })),

  mixedTokens: () => [
    { text: 'Describe this image: ' },
    { img: 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==' }
  ]
};

// Performance test utilities
export const performance = {
  measureTokenization: (api: any, text: string, iterations: number = 1000) => {
    const start = performance.now();
    for (let i = 0; i < iterations; i++) {
      api.tokenizer(text + i);
    }
    const end = performance.now();
    return {
      totalTime: end - start,
      averageTime: (end - start) / iterations,
      iterations
    };
  },

  measureApiCreation: (choice: any, iterations: number = 100) => {
    const start = performance.now();
    for (let i = 0; i < iterations; i++) {
      const models = require('../index').default;
      models(choice);
    }
    const end = performance.now();
    return {
      totalTime: end - start,
      averageTime: (end - start) / iterations,
      iterations
    };
  }
};

// Error simulation utilities
export const errorSimulator = {
  // Simulate network errors
  networkError: () => new Error('Network connection failed'),

  // Simulate API key errors
  apiKeyError: () => new Error('Invalid API key'),

  // Simulate timeout errors
  timeoutError: () => new Error('Request timeout'),

  // Simulate rate limit errors
  rateLimitError: () => new Error('Rate limit exceeded'),

  // Simulate invalid response errors
  invalidResponseError: () => new Error('Invalid API response format')
};

// Logging utilities for tests
export const testLogger = {
  info: (message: string, ...args: any[]) => {
    console.log(`[TEST-INFO] ${message}`, ...args);
  },

  warn: (message: string, ...args: any[]) => {
    console.warn(`[TEST-WARN] ${message}`, ...args);
  },

  error: (message: string, ...args: any[]) => {
    console.error(`[TEST-ERROR] ${message}`, ...args);
  },

  debug: (message: string, ...args: any[]) => {
    if (Bun.env.DEBUG_TESTS === 'true') {
      console.debug(`[TEST-DEBUG] ${message}`, ...args);
    }
  }
};

// Cleanup utilities
export const cleanup = {
  // Clear environment variables after tests
  clearEnv: (...vars: string[]) => {
    vars.forEach(varName => {
      delete Bun.env[varName];
    });
  },

  // Reset global state if needed
  resetGlobal: () => {
    // Add any global state cleanup here
  }
};

// Export a default test setup object
export default {
  getTestConfig,
  hasIntegrationCredentials,
  mockResponses,
  testData,
  performance,
  errorSimulator,
  testLogger,
  cleanup
};