import traceback
from typing import Dict, Any

from src.common.logger_manager import get_logger
from src.manager.mood_manager import mood_manager  # 导入情绪管理器
from .message import MessageRecv
from ..heartFC_chat.heartflow_processor import HeartFCProcessor
from ..PFC.pfc_processor import PFCProcessor
from ..utils.prompt_builder import Prompt, global_prompt_manager
from ...config.config import global_config

# 定义日志配置


# 配置主程序日志格式
logger = get_logger("chat")


class ChatBot:
    def __init__(self):
        self.bot = None  # bot 实例引用
        self._started = False
        self.mood_manager = mood_manager  # 获取情绪管理器单例
        self.heartflow_processor = HeartFCProcessor()  # 新增
        self.pfc_processor = PFCProcessor()

    async def _ensure_started(self):
        """确保所有任务已启动"""
        if not self._started:
            logger.trace("确保ChatBot所有任务已启动")

            self._started = True

    async def message_process(self, message_data: Dict[str, Any]) -> None:
        """处理转化后的统一格式消息
        这个函数本质是预处理一些数据，根据配置信息和消息内容，预处理消息，并分发到合适的消息处理器中
        heart_flow模式：使用思维流系统进行回复
        - 包含思维流状态管理
        - 在回复前进行观察和状态更新
        - 回复后更新思维流状态
        - 消息过滤
        - 记忆激活
        - 意愿计算
        - 消息生成和发送
        - 表情包处理
        - 性能计时
        """
        try:
            # 确保所有任务已启动
            await self._ensure_started()

            if message_data["message_info"].get("group_info") is not None:
                message_data["message_info"]["group_info"]["group_id"] = str(
                    message_data["message_info"]["group_info"]["group_id"]
                )
            message_data["message_info"]["user_info"]["user_id"] = str(
                message_data["message_info"]["user_info"]["user_id"]
            )
            logger.trace(f"处理消息:{str(message_data)[:120]}...")
            message = MessageRecv(message_data)
            groupinfo = message.message_info.group_info
            userinfo = message.message_info.user_info

            # 用户黑名单拦截
            if userinfo.user_id in global_config.ban_user_id:
                logger.debug(f"用户{userinfo.user_id}被禁止回复")
                return

            if groupinfo is None and global_config.enable_friend_whitelist:
                logger.trace("检测到私聊消息，检查")
                # 好友黑名单拦截
                if userinfo.user_id not in global_config.talk_allowed_private:
                    logger.debug(f"用户{userinfo.user_id}没有私聊权限")
                    return
            elif not global_config.enable_friend_whitelist:
                logger.debug("私聊白名单模式未启用，跳过私聊权限检查。")

            # 群聊黑名单拦截
            if groupinfo is not None and groupinfo.group_id not in global_config.talk_allowed_groups:
                logger.trace(f"群{groupinfo.group_id}被禁止回复")
                return

            # 确认从接口发来的message是否有自定义的prompt模板信息
            if message.message_info.template_info and not message.message_info.template_info.template_default:
                template_group_name = message.message_info.template_info.template_name
                template_items = message.message_info.template_info.template_items
                async with global_prompt_manager.async_message_scope(template_group_name):
                    if isinstance(template_items, dict):
                        for k in template_items.keys():
                            await Prompt.create_async(template_items[k], k)
                            print(f"注册{template_items[k]},{k}")
            else:
                template_group_name = None

            async def preprocess():
                logger.trace("开始预处理消息...")
                # 如果在私聊中
                if groupinfo is None:
                    logger.trace("检测到私聊消息")
                    # 是否在配置信息中开启私聊模式
                    if global_config.enable_friend_chat:
                        logger.trace("私聊模式已启用")
                        # 是否进入PFC
                        if global_config.enable_pfc_chatting:
                            logger.trace("进入PFC私聊处理流程")
                            await self.pfc_processor.process_message(message_data)
                        # 禁止PFC，进入普通的心流消息处理逻辑
                        else:
                            logger.trace("进入普通心流私聊处理")
                            await self.heartflow_processor.process_message(message_data)
                # 群聊默认进入心流消息处理逻辑
                else:
                    logger.trace(f"检测到群聊消息，群ID: {groupinfo.group_id}")
                    await self.heartflow_processor.process_message(message_data)

            if template_group_name:
                async with global_prompt_manager.async_message_scope(template_group_name):
                    await preprocess()
            else:
                await preprocess()

        except Exception as e:
            logger.error(f"预处理消息失败: {e}")
            traceback.print_exc()


# 创建全局ChatBot实例
chat_bot = ChatBot()