from typing import Any, Dict, Optional

from src.chat.message_receive.chat_manager import BotChatSession, chat_manager as _chat_manager
from src.config.config import global_config
from src.common.logger import get_logger

from .maisaka_generator import MaisakaReplyGenerator

logger = get_logger("ReplyerManager")


class ReplyerManager:
    """统一管理不同类型的回复生成器。"""

    def __init__(self) -> None:
        self._repliers: Dict[str, Any] = {}

    @staticmethod
    def _get_maisaka_generator_type() -> str:
        """返回当前配置下 Maisaka replyer 的消息模式。"""
        return global_config.visual.replyer_mode

    def get_replyer(
        self,
        chat_stream: Optional[BotChatSession] = None,
        chat_id: Optional[str] = None,
        request_type: str = "replyer",
        replyer_type: str = "default",
    ) -> Optional[MaisakaReplyGenerator]:
        """按会话和 replyer 类型获取实例。"""
        stream_id = chat_stream.session_id if chat_stream else chat_id
        if not stream_id:
            logger.warning("[ReplyerManager] 缺少 stream_id，无法获取 replyer")
            return None

        generator_type = self._get_maisaka_generator_type() if replyer_type == "maisaka" else ""
        cache_key = f"{replyer_type}:{generator_type}:{stream_id}"
        if cache_key in self._repliers:
            return self._repliers[cache_key]

        target_stream = chat_stream or _chat_manager.get_session_by_session_id(stream_id)
        if not target_stream:
            logger.warning(f"[ReplyerManager] 未找到会话，stream_id={stream_id}")
            return None

        logger.info(
            f"[ReplyerManager] 开始创建 replyer: cache_key={cache_key}, "
            f"replyer_type={replyer_type}, is_group_session={target_stream.is_group_session}"
        )

        try:
            if replyer_type == "maisaka":
                replyer = MaisakaReplyGenerator(
                    chat_stream=target_stream,
                    request_type=request_type,
                )
            else:
                logger.warning(f"[ReplyerManager] 不支持的 replyer_type={replyer_type}")
                return None
        except Exception:
            logger.exception(f"[ReplyerManager] 创建 replyer 失败: cache_key={cache_key}")
            raise

        self._repliers[cache_key] = replyer
        logger.info(f"[ReplyerManager] replyer 创建完成: cache_key={cache_key}")
        return replyer


replyer_manager = ReplyerManager()