#!/usr/bin/env python3
"""
Environment Setup Checker for TradingAgents
Checks if the necessary API keys are configured for your provider combination.
"""

import os
import sys


def check_api_keys():
    """Check which API keys are configured."""
    print("\n" + "=" * 70)
    print("TradingAgents - Environment Setup Checker")
    print("=" * 70)

    # Check for common API keys
    api_keys = {
        "OPENAI_API_KEY": os.getenv("OPENAI_API_KEY"),
        "OPENROUTER_API_KEY": os.getenv("OPENROUTER_API_KEY"),
        "ANTHROPIC_API_KEY": os.getenv("ANTHROPIC_API_KEY"),
        "GOOGLE_API_KEY": os.getenv("GOOGLE_API_KEY"),
    }

    print("\n1. API Keys Status:")
    print("-" * 70)

    for key_name, key_value in api_keys.items():
        if key_value:
            # Mask the key for security
            masked = (
                key_value[:8] + "..." + key_value[-4:] if len(key_value) > 12 else "***"
            )
            print(f"   ✅ {key_name}: {masked}")
        else:
            print(f"   ❌ {key_name}: Not set")

    print("\n2. Common Configuration Scenarios:")
    print("-" * 70)

    # Scenario 1: OpenRouter + OpenAI embeddings
    if api_keys["OPENROUTER_API_KEY"] and api_keys["OPENAI_API_KEY"]:
        print("   ✅ OpenRouter (chat) + OpenAI (embeddings) - READY")
    elif api_keys["OPENROUTER_API_KEY"] and not api_keys["OPENAI_API_KEY"]:
        print("   ⚠️  OpenRouter (chat) + OpenAI (embeddings) - MISSING OPENAI_API_KEY")
        print("      You need to set OPENAI_API_KEY for embeddings!")

    # Scenario 2: OpenAI everything
    if api_keys["OPENAI_API_KEY"]:
        print("   ✅ OpenAI (chat + embeddings) - READY")

    # Scenario 3: Anthropic + OpenAI embeddings
    if api_keys["ANTHROPIC_API_KEY"] and api_keys["OPENAI_API_KEY"]:
        print("   ✅ Anthropic (chat) + OpenAI (embeddings) - READY")
    elif api_keys["ANTHROPIC_API_KEY"] and not api_keys["OPENAI_API_KEY"]:
        print("   ⚠️  Anthropic (chat) + OpenAI (embeddings) - MISSING OPENAI_API_KEY")

    # Scenario 4: Google + OpenAI embeddings
    if api_keys["GOOGLE_API_KEY"] and api_keys["OPENAI_API_KEY"]:
        print("   ✅ Google (chat) + OpenAI (embeddings) - READY")
    elif api_keys["GOOGLE_API_KEY"] and not api_keys["OPENAI_API_KEY"]:
        print("   ⚠️  Google (chat) + OpenAI (embeddings) - MISSING OPENAI_API_KEY")

    print("\n3. How to Fix:")
    print("-" * 70)

    if not api_keys["OPENAI_API_KEY"]:
        print("\n   For OpenAI embeddings (recommended), set:")
        print("   export OPENAI_API_KEY='sk-...'")
        print("\n   Or add to your .env file:")
        print("   OPENAI_API_KEY=sk-...")

    if not api_keys["OPENROUTER_API_KEY"]:
        print("\n   For OpenRouter chat models, set:")
        print("   export OPENROUTER_API_KEY='sk-or-...'")
        print("\n   Or add to your .env file:")
        print("   OPENROUTER_API_KEY=sk-or-...")

    print("\n4. Configuration Example:")
    print("-" * 70)
    print("""
   For OpenRouter (chat) + OpenAI (embeddings):

   In your terminal or .env file:

   export OPENROUTER_API_KEY="sk-or-v1-..."
   export OPENAI_API_KEY="sk-proj-..."

   In your config:

   config = {
       # Chat with OpenRouter
       "llm_provider": "openrouter",
       "backend_url": "https://openrouter.ai/api/v1",
       "deep_think_llm": "deepseek/deepseek-chat-v3-0324:free",
       "quick_think_llm": "meta-llama/llama-3.3-8b-instruct:free",

       # Embeddings with OpenAI (separate!)
       "embedding_provider": "openai",
       "embedding_backend_url": "https://api.openai.com/v1",
       "embedding_model": "text-embedding-3-small",
       "enable_memory": True,
   }
    """)

    print("\n5. Alternative: Disable Memory")
    print("-" * 70)
    print("""
   If you don't want to use embeddings/memory, you can disable it:

   config = {
       "llm_provider": "openrouter",
       "backend_url": "https://openrouter.ai/api/v1",
       "enable_memory": False,  # Disable embeddings
   }

   This will run without memory but won't require an OpenAI API key.
    """)

    print("=" * 70)

    # Return status
    issues = []
    if not any(api_keys.values()):
        issues.append("No API keys found")

    return len(issues) == 0


if __name__ == "__main__":
    try:
        success = check_api_keys()

        if success:
            print("\n✅ Environment looks good! You have API keys configured.")
        else:
            print("\n⚠️  Please set the required API keys for your configuration.")

        print("\nFor more help, see: docs/EMBEDDING_CONFIGURATION.md\n")

    except Exception as e:
        print(f"\n❌ Error checking environment: {e}")
        import traceback

        traceback.print_exc()
        sys.exit(1)