import os
from typing import Optional
from composio_client import Composio
from utils.logger import logger


class ComposioClient:
    _instance: Optional[Composio] = None
    
    @classmethod
    def get_client(cls, api_key: Optional[str] = None) -> Composio:
        if cls._instance is None:
            if not api_key:
                api_key = os.getenv("COMPOSIO_API_KEY")
                if not api_key:
                    raise ValueError("COMPOSIO_API_KEY is required")
            
            logger.info("Initializing Composio client")
            cls._instance = Composio(api_key=api_key)
        
        return cls._instance
    
    @classmethod
    def reset_client(cls) -> None:
        cls._instance = None


def get_composio_client(api_key: Optional[str] = None) -> Composio:
    return ComposioClient.get_client(api_key)