config.py

#!/usr/bin/env python3
"""
Configuration manager for Healthcare Provider Validation System.
Allows switching between API mode and mock mode.
"""

import os
import sys
from pathlib import Path
from typing import Dict, Any

class ConfigManager:
    """Manages system configuration and mode switching."""
    
    def __init__(self):
        self.config_file = Path("system_config.json")
        self.default_config = {
            "mode": "mock",  # "api" or "mock"
            "apis": {
                "npi_registry": {"enabled": False, "url": "https://npiregistry.cms.hhs.gov/api/"},
                "google_maps": {"enabled": False, "url": "https://maps.googleapis.com/maps/api/"},
                "twilio": {"enabled": False, "url": "https://lookups.twilio.com/v1/"},
                "openai": {"enabled": False, "url": "https://api.openai.com/v1/"}
            },
            "mock_settings": {
                "success_rate": 0.85,
                "delay_simulation": True,
                "realistic_responses": True
            }
        }
        self.load_config()
    
    def load_config(self):
        """Load configuration from file."""
        if self.config_file.exists():
            import json
            with open(self.config_file, 'r') as f:
                self.config = json.load(f)
        else:
            self.config = self.default_config.copy()
            self.save_config()
    
    def save_config(self):
        """Save configuration to file."""
        import json
        with open(self.config_file, 'w') as f:
            json.dump(self.config, f, indent=2)
    
    def set_mode(self, mode: str):
        """Set system mode (api or mock)."""
        if mode not in ["api", "mock"]:
            raise ValueError("Mode must be 'api' or 'mock'")
        
        self.config["mode"] = mode
        
        if mode == "api":
            # Enable all APIs
            for api in self.config["apis"]:
                self.config["apis"][api]["enabled"] = True
        else:
            # Disable all APIs
            for api in self.config["apis"]:
                self.config["apis"][api]["enabled"] = False
        
        self.save_config()
        print(f"✅ System mode set to: {mode.upper()}")
    
    def toggle_api(self, api_name: str, enabled: bool = None):
        """Toggle specific API on/off."""
        if api_name not in self.config["apis"]:
            raise ValueError(f"Unknown API: {api_name}")
        
        if enabled is None:
            # Toggle current state
            self.config["apis"][api_name]["enabled"] = not self.config["apis"][api_name]["enabled"]
        else:
            self.config["apis"][api_name]["enabled"] = enabled
        
        self.save_config()
        status = "enabled" if self.config["apis"][api_name]["enabled"] else "disabled"
        print(f"✅ {api_name.upper()} API {status}")
    
    def get_status(self):
        """Get current system status."""
        mode = self.config["mode"]
        enabled_apis = [api for api, config in self.config["apis"].items() if config["enabled"]]
        disabled_apis = [api for api, config in self.config["apis"].items() if not config["enabled"]]
        
        return {
            "mode": mode,
            "enabled_apis": enabled_apis,
            "disabled_apis": disabled_apis,
            "total_apis": len(self.config["apis"])
        }
    
    def print_status(self):
        """Print current system status."""
        status = self.get_status()
        
        print("\n🏥 Healthcare Provider Validation System - Configuration")
        print("=" * 60)
        print(f"📊 Current Mode: {status['mode'].upper()}")
        print(f"🔌 Enabled APIs: {', '.join(status['enabled_apis']) if status['enabled_apis'] else 'None'}")
        print(f"🚫 Disabled APIs: {', '.join(status['disabled_apis']) if status['disabled_apis'] else 'None'}")
        print(f"📈 Total APIs: {status['total_apis']}")
        
        if status['mode'] == 'mock':
            print("\n🎭 Mock Mode Features:")
            print("  • Simulated API responses")
            print("  • Realistic delays and success rates")
            print("  • No external API calls")
            print("  • Perfect for development and testing")
        else:
            print("\n🌐 API Mode Features:")
            print("  • Real external API calls")
            print("  • Live data validation")
            print("  • Production-ready functionality")
            print("  • Requires valid API keys")
    
    def setup_api_keys(self):
        """Interactive API key setup."""
        print("\n🔑 API Key Setup")
        print("=" * 30)
        
        api_keys = {}
        
        # OpenAI
        if self.config["apis"]["openai"]["enabled"]:
            key = input("Enter OpenAI API Key (or press Enter to skip): ").strip()
            if key:
                api_keys["OPENAI_API_KEY"] = key
        
        # Google Maps
        if self.config["apis"]["google_maps"]["enabled"]:
            key = input("Enter Google Maps API Key (or press Enter to skip): ").strip()
            if key:
                api_keys["GOOGLE_MAPS_API_KEY"] = key
        
        # Twilio
        if self.config["apis"]["twilio"]["enabled"]:
            sid = input("Enter Twilio Account SID (or press Enter to skip): ").strip()
            token = input("Enter Twilio Auth Token (or press Enter to skip): ").strip()
            if sid and token:
                api_keys["TWILIO_ACCOUNT_SID"] = sid
                api_keys["TWILIO_AUTH_TOKEN"] = token
        
        # Update .env file
        if api_keys:
            env_file = Path(".env")
            env_content = env_file.read_text() if env_file.exists() else ""
            
            for key, value in api_keys.items():
                if f"{key}=" in env_content:
                    # Update existing key
                    import re
                    env_content = re.sub(f"{key}=.*", f"{key}={value}", env_content)
                else:
                    # Add new key
                    env_content += f"\n{key}={value}\n"
            
            env_file.write_text(env_content)
            print(f"✅ Updated .env file with {len(api_keys)} API keys")
        else:
            print("ℹ️  No API keys provided")


def main():
    """Main CLI interface."""
    if len(sys.argv) < 2:
        print("🏥 Healthcare Provider Validation System - Configuration Manager")
        print("=" * 70)
        print("\nUsage:")
        print("  python config.py status                    - Show current status")
        print("  python config.py mode <api|mock>           - Set system mode")
        print("  python config.py toggle <api_name>         - Toggle specific API")
        print("  python config.py enable <api_name>         - Enable specific API")
        print("  python config.py disable <api_name>        - Disable specific API")
        print("  python config.py setup                     - Interactive API key setup")
        print("  python config.py reset                    - Reset to default config")
        print("\nAvailable APIs: npi_registry, google_maps, twilio, openai")
        return
    
    config = ConfigManager()
    command = sys.argv[1].lower()
    
    try:
        if command == "status":
            config.print_status()
        
        elif command == "mode":
            if len(sys.argv) < 3:
                print("❌ Please specify mode: 'api' or 'mock'")
                return
            mode = sys.argv[2].lower()
            config.set_mode(mode)
        
        elif command == "toggle":
            if len(sys.argv) < 3:
                print("❌ Please specify API name")
                return
            api_name = sys.argv[2].lower()
            config.toggle_api(api_name)
        
        elif command == "enable":
            if len(sys.argv) < 3:
                print("❌ Please specify API name")
                return
            api_name = sys.argv[2].lower()
            config.toggle_api(api_name, True)
        
        elif command == "disable":
            if len(sys.argv) < 3:
                print("❌ Please specify API name")
                return
            api_name = sys.argv[2].lower()
            config.toggle_api(api_name, False)
        
        elif command == "setup":
            config.setup_api_keys()
        
        elif command == "reset":
            config.config = config.default_config.copy()
            config.save_config()
            print("✅ Configuration reset to defaults")
        
        else:
            print(f"❌ Unknown command: {command}")
            print("Run 'python config.py' for help")
    
    except Exception as e:
        print(f"❌ Error: {e}")


if __name__ == "__main__":
    main()