config/settings.py

"""
ARF Settings Configuration with Pydantic 2.x Compatibility
Fixed for both OSS and Enterprise deployments - FIXED VERSION CONSISTENCY
"""
import os
import sys
from enum import Enum
from typing import Optional, Dict, Any, List, Union
from datetime import datetime

# Try to import pydantic with version detection
try:
    from pydantic import Field, validator, model_validator
    from pydantic_settings import BaseSettings, SettingsConfigDict
    PYDANTIC_V2 = True
except ImportError:
    # Fallback for pydantic v1
    from pydantic import Field, validator
    from pydantic_settings import BaseSettings
    PYDANTIC_V2 = False

import logging
logger = logging.getLogger(__name__)

# Enums for ARF modes
class ARFMode(str, Enum):
    """ARF Operational Modes"""
    DEMO = "demo"
    OSS = "oss"
    ENTERPRISE = "enterprise"
    HYBRID = "hybrid"
    SIMULATION = "simulation"

class SafetyMode(str, Enum):
    """Safety and Compliance Modes"""
    STANDARD = "standard"
    STRICT = "strict"
    PERMISSIVE = "permissive"
    ENTERPRISE = "enterprise"
    COMPLIANCE = "compliance"

class InstallationStatus(str, Enum):
    """Package Installation Status"""
    NOT_INSTALLED = "not_installed"
    OSS_ONLY = "oss_only"
    ENTERPRISE_ONLY = "enterprise_only"
    FULL_INSTALL = "full_install"
    DEMO_MODE = "demo_mode"

# Pydantic 2.x compatible Settings class
if PYDANTIC_V2:
    class Settings(BaseSettings):
        """ARF Settings with Pydantic 2.x compatibility - FIXED: Version consistency"""
        model_config = SettingsConfigDict(
            env_file=".env",
            env_file_encoding="utf-8",
            env_prefix="ARF_",
            case_sensitive=False,
            extra="ignore"
        )
        
        # Core Settings
        arf_mode: ARFMode = Field(default=ARFMode.DEMO, description="ARF operational mode")
        safety_mode: SafetyMode = Field(default=SafetyMode.STANDARD, description="Safety compliance mode")
        debug: bool = Field(default=False, description="Enable debug logging")
        log_level: str = Field(default="INFO", description="Logging level")
        
        # Version Info - FIXED: Updated to v3.3.9 (actual PyPI version)
        arf_version: str = Field(default="3.3.9", description="ARF version")
        demo_version: str = Field(default="3.8.0", description="Demo app version")
        
        # Installation Status
        oss_installed: bool = Field(default=False, description="ARF OSS package installed")
        enterprise_installed: bool = Field(default=False, description="Enterprise package installed")
        installation_status: InstallationStatus = Field(
            default=InstallationStatus.DEMO_MODE,
            description="Overall installation status"
        )
        
        # Feature Flags
        enable_telemetry: bool = Field(default=True, description="Enable telemetry collection")
        enable_anomaly_detection: bool = Field(default=True, description="Enable anomaly detection")
        enable_business_metrics: bool = Field(default=True, description="Enable business metrics")
        enable_audit_trail: bool = Field(default=True, description="Enable audit trail")
        
        # Paths
        data_dir: str = Field(default="./data", description="Data directory")
        log_dir: str = Field(default="./logs", description="Log directory")
        cache_dir: str = Field(default="./cache", description="Cache directory")
        scenario_config_path: str = Field(default="./config/scenarios", description="Path to scenario configuration files")
        
        # API Settings
        api_timeout: int = Field(default=30, description="API timeout in seconds")
        max_retries: int = Field(default=3, description="Maximum API retries")
        rate_limit: int = Field(default=100, description="API rate limit per minute")
        
        # UI Settings
        ui_theme: str = Field(default="dark", description="UI theme")
        refresh_interval: int = Field(default=5, description="UI refresh interval in seconds")
        
        # Business Settings
        default_currency: str = Field(default="USD", description="Default currency")
        cost_per_incident: float = Field(default=10000.0, description="Default cost per incident")
        engineer_hourly_rate: float = Field(default=150.0, description="Engineer hourly rate")
        engineer_annual_cost: float = Field(default=200000.0, description="Engineer annual cost")
        
        @model_validator(mode='before')
        @classmethod
        def validate_enums(cls, data: Any) -> Any:
            """Validate and convert enum fields from strings"""
            if isinstance(data, dict):
                # Convert string values to enums if needed
                if 'arf_mode' in data and isinstance(data['arf_mode'], str):
                    try:
                        data['arf_mode'] = ARFMode(data['arf_mode'].lower())
                    except ValueError:
                        data['arf_mode'] = ARFMode.DEMO
                
                if 'safety_mode' in data and isinstance(data['safety_mode'], str):
                    try:
                        data['safety_mode'] = SafetyMode(data['safety_mode'].lower())
                    except ValueError:
                        data['safety_mode'] = SafetyMode.STANDARD
                
                if 'installation_status' in data and isinstance(data['installation_status'], str):
                    try:
                        data['installation_status'] = InstallationStatus(data['installation_status'].lower())
                    except ValueError:
                        data['installation_status'] = InstallationStatus.DEMO_MODE
            return data
        
        @model_validator(mode='after')
        def check_installation_status(self):
            """Update installation status based on detected packages"""
            # Check if OSS is installed
            try:
                import agentic_reliability_framework
                self.oss_installed = True
            except ImportError:
                self.oss_installed = False
            
            # Check if Enterprise is installed
            try:
                import agentic_reliability_enterprise
                self.enterprise_installed = True
            except ImportError:
                self.enterprise_installed = False
            
            # Update installation status
            if self.oss_installed and self.enterprise_installed:
                self.installation_status = InstallationStatus.FULL_INSTALL
            elif self.oss_installed:
                self.installation_status = InstallationStatus.OSS_ONLY
            elif self.enterprise_installed:
                self.installation_status = InstallationStatus.ENTERPRISE_ONLY
            else:
                self.installation_status = InstallationStatus.DEMO_MODE
            
            return self

else:
    # Pydantic v1 compatible Settings class
    class Settings(BaseSettings):
        """ARF Settings with Pydantic v1 compatibility - FIXED: Version consistency"""
        class Config:
            env_file = ".env"
            env_file_encoding = "utf-8"
            env_prefix = "ARF_"
            case_sensitive = False
        
        # Core Settings
        arf_mode: ARFMode = Field(default=ARFMode.DEMO, description="ARF operational mode")
        safety_mode: SafetyMode = Field(default=SafetyMode.STANDARD, description="Safety compliance mode")
        debug: bool = Field(default=False, description="Enable debug logging")
        log_level: str = Field(default="INFO", description="Logging level")
        
        # Version Info - FIXED: Updated to v3.3.9 (actual PyPI version)
        arf_version: str = Field(default="3.3.9", description="ARF version")
        demo_version: str = Field(default="3.8.0", description="Demo app version")
        
        # Installation Status
        oss_installed: bool = Field(default=False, description="ARF OSS package installed")
        enterprise_installed: bool = Field(default=False, description="Enterprise package installed")
        installation_status: InstallationStatus = Field(
            default=InstallationStatus.DEMO_MODE,
            description="Overall installation status"
        )
        
        # Feature Flags
        enable_telemetry: bool = Field(default=True, description="Enable telemetry collection")
        enable_anomaly_detection: bool = Field(default=True, description="Enable anomaly detection")
        enable_business_metrics: bool = Field(default=True, description="Enable business metrics")
        enable_audit_trail: bool = Field(default=True, description="Enable audit trail")
        
        # Paths
        data_dir: str = Field(default="./data", description="Data directory")
        log_dir: str = Field(default="./logs", description="Log directory")
        cache_dir: str = Field(default="./cache", description="Cache directory")
        scenario_config_path: str = Field(default="./config/scenarios", description="Path to scenario configuration files")
        
        # API Settings
        api_timeout: int = Field(default=30, description="API timeout in seconds")
        max_retries: int = Field(default=3, description="Maximum API retries")
        rate_limit: int = Field(default=100, description="API rate limit per minute")
        
        # UI Settings
        ui_theme: str = Field(default="dark", description="UI theme")
        refresh_interval: int = Field(default=5, description="UI refresh interval in seconds")
        
        # Business Settings
        default_currency: str = Field(default="USD", description="Default currency")
        cost_per_incident: float = Field(default=10000.0, description="Default cost per incident")
        engineer_hourly_rate: float = Field(default=150.0, description="Engineer hourly rate")
        engineer_annual_cost: float = Field(default=200000.0, description="Engineer annual cost")
        
        @validator('arf_mode', 'safety_mode', 'installation_status', pre=True)
        def validate_enum_strings(cls, v, field):
            """Convert string values to enums"""
            if isinstance(v, str):
                try:
                    if field.name == 'arf_mode':
                        return ARFMode(v.lower())
                    elif field.name == 'safety_mode':
                        return SafetyMode(v.lower())
                    elif field.name == 'installation_status':
                        return InstallationStatus(v.lower())
                except ValueError:
                    # Return default value
                    if field.name == 'arf_mode':
                        return ARFMode.DEMO
                    elif field.name == 'safety_mode':
                        return SafetyMode.STANDARD
                    elif field.name == 'installation_status':
                        return InstallationStatus.DEMO_MODE
            return v
        
        @validator('installation_status', always=True)
        def update_installation_status(cls, v, values):
            """Update installation status based on detected packages"""
            # Check if OSS is installed
            try:
                import agentic_reliability_framework
                values['oss_installed'] = True
            except ImportError:
                values['oss_installed'] = False
            
            # Check if Enterprise is installed
            try:
                import agentic_reliability_enterprise
                values['enterprise_installed'] = True
            except ImportError:
                values['enterprise_installed'] = False
            
            # Determine installation status
            if values.get('oss_installed') and values.get('enterprise_installed'):
                return InstallationStatus.FULL_INSTALL
            elif values.get('oss_installed'):
                return InstallationStatus.OSS_ONLY
            elif values.get('enterprise_installed'):
                return InstallationStatus.ENTERPRISE_ONLY
            else:
                return InstallationStatus.DEMO_MODE

# Singleton settings instance
try:
    settings = Settings()
    logger.info("✅ Settings initialized successfully")
    logger.info(f"📊 ARF Mode: {settings.arf_mode}")
    logger.info(f"🛡️ Safety Mode: {settings.safety_mode}")
    logger.info(f"📦 Installation Status: {settings.installation_status}")
    logger.info(f"🔖 ARF Version: {settings.arf_version}")  # Added version logging
except Exception as e:
    logger.error(f"❌ Failed to initialize settings: {e}")
    # Create fallback settings - FIXED: Updated version to 3.3.9
    class FallbackSettings:
        arf_mode = ARFMode.DEMO
        safety_mode = SafetyMode.STANDARD
        debug = False
        log_level = "INFO"
        arf_version = "3.3.9"  # FIXED: Updated from 3.3.7 to 3.3.9
        demo_version = "3.8.0"
        oss_installed = False
        enterprise_installed = False
        installation_status = InstallationStatus.DEMO_MODE
        enable_telemetry = True
        enable_anomaly_detection = True
        enable_business_metrics = True
        enable_audit_trail = True
        data_dir = "./data"
        log_dir = "./logs"
        cache_dir = "./cache"
        scenario_config_path = "./config/scenarios"
        api_timeout = 30
        max_retries = 3
        rate_limit = 100
        ui_theme = "dark"
        refresh_interval = 5
        default_currency = "USD"
        cost_per_incident = 10000.0
        engineer_hourly_rate = 150.0
        engineer_annual_cost = 200000.0
    
    settings = FallbackSettings()
    logger.warning("⚠️ Using fallback settings due to initialization error")

def get_settings() -> Settings:
    """Get the settings singleton instance"""
    return settings

# Export everything
__all__ = [
    "Settings",
    "get_settings",
    "ARFMode",
    "SafetyMode",
    "InstallationStatus",
    "settings"
]