import os
import platform
from pathlib import Path
from tkinter import messagebox, simpledialog
from dotenv import load_dotenv

class APIKeyManager:
    """Class to handle API key management operations."""
    
    @staticmethod
    def get_app_support_path_mac():
        """Get the application support path for macOS."""
        home = Path.home()
        app_support_path = home / 'Library' / 'Application Support' / 'scorchsoft-text-to-mic'
        app_support_path.mkdir(parents=True, exist_ok=True)  # Ensure directory exists
        return app_support_path
    
    @staticmethod
    def save_api_key_mac(api_key):
        """Save the API key on macOS."""
        env_path = APIKeyManager.get_app_support_path_mac() / 'config' / '.env'
        env_path.parent.mkdir(parents=True, exist_ok=True)  # Ensure config directory exists
        with open(env_path, 'w') as f:
            f.write(f"OPENAI_API_KEY={api_key}\n")
    
    @staticmethod
    def save_api_key(api_key):
        """Save the API key to the config/.env file."""
        try:
            config_dir = Path("config")
            config_dir.mkdir(parents=True, exist_ok=True)  # Ensure directory exists

            env_path = config_dir / ".env"
            with open(env_path, 'w') as f:
                f.write(f"OPENAI_API_KEY={api_key}\n")

            # Reload environment to include the new API key
            load_dotenv(dotenv_path=env_path)  
            return True
        except Exception as e:
            messagebox.showerror("Error", f"Failed to save API key: {str(e)}")
            return False
    
    @staticmethod
    def load_api_key_mac():
        """Load the API key on macOS."""
        env_path = APIKeyManager.get_app_support_path_mac() / 'config' / '.env'
        if env_path.exists():
            with open(env_path, 'r') as f:
                for line in f:
                    if line.startswith('OPENAI_API_KEY'):
                        return line.strip().split('=')[1]
        return None
    
    @staticmethod
    def get_api_key(parent=None):
        """Get the API key from environment variables or local file, or prompt the user."""
        # First, try to load the API key from environment variables or local file
        api_key = os.getenv("OPENAI_API_KEY")
        if not api_key:  # Check for macOS and use the macOS-specific method
            if platform.system() == 'Darwin':  # Darwin is the system name for macOS
                api_key = APIKeyManager.load_api_key_mac()
            
            # If no API key is found, prompt the user
            if not api_key and parent:
                parent.show_instructions()  # Show the "How to Use" modal after setting the key
                api_key = simpledialog.askstring("API Key", "Enter your OpenAI API Key:", parent=parent)
                if api_key:
                    try:
                        if platform.system() == 'Darwin':
                            APIKeyManager.save_api_key_mac(api_key)
                        else:
                            APIKeyManager.save_api_key(api_key)
                        messagebox.showinfo("API Key Set", "The OpenAI API Key has been updated successfully.")
                    except Exception as e:
                        messagebox.showerror("Error", f"Failed to save API key: {str(e)}")
        
        return api_key
    
    @staticmethod
    def change_api_key(parent):
        """Change the API key."""
        new_key = simpledialog.askstring("API Key", "Enter new OpenAI API Key:", parent=parent)
        if new_key:
            success = APIKeyManager.save_api_key(new_key)
            if success:
                messagebox.showinfo("API Key Updated", "The OpenAI API Key has been updated successfully.")
                return new_key
        return None