from abc import ABC, abstractmethod
from typing import List, Optional, TYPE_CHECKING
from src.common.data_models.message_manager_data_model import StreamContext
from .component_types import ChatType
from src.plugin_system.base.component_types import ChatterInfo, ComponentType

if TYPE_CHECKING:
    from src.chat.planner_actions.action_manager import ChatterActionManager
    from src.plugins.built_in.affinity_flow_chatter.planner import ChatterActionPlanner as ActionPlanner

class BaseChatter(ABC):
    chatter_name: str = ""
    """Chatter组件的名称"""
    chatter_description: str = ""
    """Chatter组件的描述"""
    chat_types: List[ChatType] = [ChatType.PRIVATE, ChatType.GROUP]

    def __init__(self, stream_id: str, action_manager: 'ChatterActionManager'):
        """
        初始化聊天处理器

        Args:
            stream_id: 聊天流ID
            action_manager: 动作管理器
        """
        self.stream_id = stream_id
        self.action_manager = action_manager

    @abstractmethod
    async def execute(self, context: StreamContext) -> dict:
        """
        执行聊天处理流程

        Args:
            context: StreamContext对象，包含聊天流的所有消息信息

        Returns:
            处理结果字典
        """
        pass

    @classmethod
    def get_chatter_info(cls) -> "ChatterInfo":
        """从类属性生成ChatterInfo
        Returns:
            ChatterInfo对象
        """
        
        return ChatterInfo(
            name=cls.chatter_name,
            description=cls.chatter_description or "No description provided.",
            chat_type_allow=cls.chat_types[0],
            component_type=ComponentType.CHATTER,
        )