from __future__ import annotations
import abc
import typing

import mirai
from mirai.models.message import MessageComponent

from ...core import app
from ...core import entities as core_entities


preregistered_strategies: list[typing.Type[LongTextStrategy]] = []


def strategy_class(
    name: str
) -> typing.Callable[[typing.Type[LongTextStrategy]], typing.Type[LongTextStrategy]]:
    """长文本处理策略类装饰器

    Args:
        name (str): 策略名称

    Returns:
        typing.Callable[[typing.Type[LongTextStrategy]], typing.Type[LongTextStrategy]]: 装饰器
    """

    def decorator(cls: typing.Type[LongTextStrategy]) -> typing.Type[LongTextStrategy]:
        assert issubclass(cls, LongTextStrategy)

        cls.name = name

        preregistered_strategies.append(cls)

        return cls

    return decorator


class LongTextStrategy(metaclass=abc.ABCMeta):
    """长文本处理策略抽象类
    """

    name: str

    ap: app.Application

    def __init__(self, ap: app.Application):
        self.ap = ap
    
    async def initialize(self):
        pass
    
    @abc.abstractmethod
    async def process(self, message: str, query: core_entities.Query) -> list[MessageComponent]:
        """处理长文本

        在 platform.json 中配置 long-text-process 字段，只要 文本长度超过了 threshold 就会调用此方法

        Args:
            message (str): 消息
            query (core_entities.Query): 此次请求的上下文对象

        Returns:
            list[mirai.models.messages.MessageComponent]: 转换后的 YiriMirai 消息组件列表
        """
        return []