"""
消息事件模型模块

定义了消息相关的事件类，包括 MessageEvent, PrivateMessageEvent, GroupMessageEvent。
"""
from dataclasses import dataclass, field
from typing import List, Optional, Union

from core.permission import Permission
from models.message import MessageSegment
from models.sender import Sender
from .base import OneBotEvent, EventType


@dataclass
class Anonymous:
    """
    匿名信息
    """
    id: int = 0
    """匿名用户 ID"""
    
    name: str = ""
    """匿名用户名称"""
    
    flag: str = ""
    """匿名用户 flag"""


# 权限级别常量，用于装饰器参数
# 定义在类外部，避免 dataclass 参数顺序问题
MESSAGE_EVENT_ADMIN = Permission.ADMIN
MESSAGE_EVENT_OP = Permission.OP
MESSAGE_EVENT_USER = Permission.USER


@dataclass(slots=True)
class MessageEvent(OneBotEvent):
    """
    消息事件基类
    """

    message_type: str
    """消息类型: private (私聊), group (群聊)"""

    sub_type: str
    """
    消息子类型
    如果是私聊消息，可能是 friend, group, other, normal, anonymous, notice
    如果是群聊消息，可能是 normal, anonymous, notice
    """

    message_id: int
    """消息 ID"""

    user_id: int
    """发送者 QQ 号"""

    message: List[MessageSegment] = field(default_factory=list)
    """消息内容列表"""

    raw_message: str = ""
    """原始消息内容"""

    font: int = 0
    """字体"""

    sender: Optional[Sender] = None
    """发送者信息"""

    @property
    def post_type(self) -> str:
        return EventType.MESSAGE



    async def reply(self, message: Union[str, "MessageSegment", List["MessageSegment"]], auto_escape: bool = False):
        """
        回复消息（抽象方法，由子类实现）

        :param message: 回复内容
        :param auto_escape: 是否自动转义
        """
        raise NotImplementedError("reply method must be implemented by subclasses")


# 在类定义之后添加权限常量作为类变量
MessageEvent.ADMIN = MESSAGE_EVENT_ADMIN
MessageEvent.OP = MESSAGE_EVENT_OP
MessageEvent.USER = MESSAGE_EVENT_USER


@dataclass(slots=True)
class PrivateMessageEvent(MessageEvent):
    """
    私聊消息事件
    """

    async def reply(self, message: Union[str, "MessageSegment", List["MessageSegment"]], auto_escape: bool = False):
        """
        回复私聊消息

        :param message: 回复内容
        :param auto_escape: 是否自动转义
        """
        await self.bot.send_private_msg(
            user_id=self.user_id, message=message, auto_escape=auto_escape
        )


@dataclass(slots=True)
class GroupMessageEvent(MessageEvent):
    """
    群聊消息事件
    """

    group_id: int = 0
    """群号"""

    anonymous: Optional[Anonymous] = None
    """匿名信息"""

    async def reply(self, message: Union[str, "MessageSegment", List["MessageSegment"]], auto_escape: bool = False):
        """
        回复群聊消息

        :param message: 回复内容
        :param auto_escape: 是否自动转义
        """
        await self.bot.send_group_msg(
            group_id=self.group_id, message=message, auto_escape=auto_escape
        )