from typing import List, Union, Optional
from dataclasses import dataclass
from type.register import RegisteredPlatform
from type.types import Context
from type.astrbot_message import AstrBotMessage, MessageType

@dataclass
class MessageResult():
    result_message: Union[str, list]
    is_command_call: Optional[bool] = False
    use_t2i: Optional[bool] = None # None 为跟随用户设置
    callback: Optional[callable] = None

class AstrMessageEvent():

    def __init__(self,
                 message_str: str,
                 message_obj: AstrBotMessage,
                 platform: RegisteredPlatform,
                 role: str,
                 context: Context,
                 session_id: str = None,
                 unified_msg_origin: str = None,
                 only_command: bool = False):
        '''
        AstrBot 消息事件。
        
        `message_str`: 纯消息字符串
        `message_obj`: AstrBotMessage 对象
        `platform`: 平台对象
        `role`: 角色，`admin` or `member`
        `context`: 全局对象
        `session_id`: 会话id
        `unified_msg_origin`: 统一消息来源
        `only_command`: 是否只处理指令，而不使用 LLM 回复
        '''
        self.context = context
        self.message_str = message_str
        self.message_obj = message_obj
        self.platform = platform
        self.role = role
        self.session_id = session_id
        self.unified_msg_origin = unified_msg_origin
        self.only_command = only_command
        
    def from_astrbot_message(message: AstrBotMessage, 
                             context: Context,
                             platform_name: str, 
                             session_id: str,
                             role: str = "member",
                             unified_msg_origin: str = None,
                             only_command: bool = False):
        
        ame = AstrMessageEvent(message.message_str, 
                               message, 
                               context.find_platform(platform_name), 
                               role, 
                               context, 
                               session_id,
                               unified_msg_origin,
                               only_command=only_command)
        return ame