ALwrity/lib/utils/api_key_manager/validation.py

"""API key validation module."""

from typing import Dict, Any, List, Tuple
from loguru import logger
import os
from dotenv import load_dotenv
from .manager import APIKeyManager

def check_all_api_keys(api_key_manager: APIKeyManager) -> bool:
    """Check if minimum required API keys are present.
    
    Args:
        api_key_manager (APIKeyManager): The API key manager instance
        
    Returns:
        bool: True if minimum required keys are present (at least one AI provider and one research provider)
    """
    try:
        # Load environment variables
        logger.info("Starting API key validation process...")
        
        # Get the current working directory and .env file path
        current_dir = os.getcwd()
        env_path = os.path.join(current_dir, '.env')
        logger.info(f"Looking for .env file at: {env_path}")
        
        # Check if .env file exists
        if not os.path.exists(env_path):
            logger.error(f".env file not found at {env_path}")
            return False
            
        # Load environment variables
        load_dotenv(env_path, override=True)  # Add override=True to ensure variables are reloaded
        logger.debug("Environment variables loaded")
        
        # Log all environment variables (without their values)
        logger.debug("Available environment variables:")
        for key in os.environ.keys():
            if any(provider in key for provider in ['API_KEY', 'SERPAPI', 'TAVILY', 'METAPHOR', 'FIRECRAWL']):
                logger.debug(f"Found environment variable: {key}")
        
        # Step 1: Check for at least one AI provider
        logger.info("Checking AI provider API keys...")
        ai_providers = [
            'OPENAI_API_KEY',
            'GEMINI_API_KEY',
            'ANTHROPIC_API_KEY',
            'MISTRAL_API_KEY'
        ]
        
        # Log which AI providers are found
        for provider in ai_providers:
            value = os.getenv(provider)
            if value:
                logger.info(f"Found {provider} (length: {len(value)})")
            else:
                logger.debug(f"Missing {provider}")
        
        has_ai_provider = any(os.getenv(key) for key in ai_providers)
        if not has_ai_provider:
            logger.warning("No AI provider API key found")
            # Check if keys are in the API key manager
            if hasattr(api_key_manager, 'get_api_key'):
                for provider in ai_providers:
                    key = api_key_manager.get_api_key(provider)
                    if key:
                        logger.info(f"Found {provider} in API key manager")
                        has_ai_provider = True
                        break
            
            if not has_ai_provider:
                return False
        else:
            logger.success("✓ At least one AI provider key found")
        
        # Step 2: Check for at least one research provider
        logger.info("Checking research provider API keys...")
        research_providers = [
            'SERPAPI_KEY',
            'TAVILY_API_KEY',
            'METAPHOR_API_KEY',
            'FIRECRAWL_API_KEY'
        ]
        
        # Log which research providers are found
        for provider in research_providers:
            value = os.getenv(provider)
            if value:
                logger.info(f"Found {provider} (length: {len(value)})")
            else:
                logger.debug(f"Missing {provider}")
        
        has_research_provider = any(os.getenv(key) for key in research_providers)
        if not has_research_provider:
            logger.warning("No research provider API key found")
            # Check if keys are in the API key manager
            if hasattr(api_key_manager, 'get_api_key'):
                for provider in research_providers:
                    key = api_key_manager.get_api_key(provider)
                    if key:
                        logger.info(f"Found {provider} in API key manager")
                        has_research_provider = True
                        break
                        
            if not has_research_provider:
                return False
        else:
            logger.success("✓ At least one research provider key found")
        
        # Step 3: Check for website URL
        logger.info("Checking website URL...")
        website_url = os.getenv('WEBSITE_URL')
        if not website_url:
            logger.warning("No website URL found in environment variables")
            return False
        else:
            logger.success(f"✓ Website URL found: {website_url}")
        
        # Step 4: Check for personalization status
        logger.info("Checking personalization status...")
        if 'PERSONALIZATION_DONE' not in os.environ:
            logger.warning("PERSONALIZATION_DONE environment variable is not defined")
            return False
        else:
            logger.success(f"✓ Personalization status: {os.environ['PERSONALIZATION_DONE']}")
        
        # Step 5: Check for integration status
        logger.info("Checking integration status...")
        if 'INTEGRATION_DONE' not in os.environ:
            logger.warning("INTEGRATION_DONE environment variable is not defined")
            return False
        else:
            logger.success(f"✓ Integration status: {os.environ['INTEGRATION_DONE']}")
        
        # Step 6: Check for final setup status
        logger.info("Checking final setup status...")
        if 'FINAL_SETUP_COMPLETE' not in os.environ:
            logger.warning("FINAL_SETUP_COMPLETE environment variable is not defined")
            return False
        else:
            final_setup_status = os.environ['FINAL_SETUP_COMPLETE']
            if final_setup_status.lower() == 'true':
                logger.success("✓ Final setup completed successfully")
            else:
                logger.warning("Final setup validation failed")
                return False
        
        logger.success("All required API keys and setup steps validated successfully!")
        return True
        
    except Exception as e:
        logger.error(f"Error checking API keys: {str(e)}", exc_info=True)
        return False