from typing import Dict, Optional, Any
from bot_definitions import BotType, call_transfer, simple_dialin, simple_dialout, voicemail_detection


class BotRegistry:
    def __init__(self):
        self.bots: Dict[str, BotType] = {
            "call_transfer": call_transfer,
            "simple_dialin": simple_dialin,
            "simple_dialout": simple_dialout,
            "voicemail_detection": voicemail_detection,
        }

    def get_bot(self, name: str) -> Optional[BotType]:
        return self.bots.get(name)

    def detect_bot_type(self, body: Dict[str, Any]) -> Optional[str]:
        for bot in self.bots.values():
            if bot.config_key in body:
                return bot.name
        if all(key in body for key in ["From", "To", "callId", "callDomain"]):
            return "call_transfer"
        return None

    def setup_configuration(self, config: Dict[str, Any]) -> Dict[str, Any]:
        body = {}
        bot_name = config.get("bot_type")
        bot = self.get_bot(bot_name)
        if bot:
            body[bot.config_key] = config.get("settings", {})
        return body