from astrbot.api.event import filter, AstrMessageEvent, MessageEventResult
from astrbot.api.star import Context, Star, register
from astrbot.api import logger
from astrbot.api.provider import ProviderRequest
from astrbot.core.agent.message import TextPart
@register("helloworld", "YourName", "一个简单的 Hello World 插件", "1.0.0")
class MyPlugin(Star):
    def __init__(self, context: Context):
        super().__init__(context)

    async def initialize(self):
        """可选择实现异步的插件初始化方法，当实例化该插件类之后会自动调用该方法。"""

    # 注册指令的装饰器。指令名为 helloworld。注册成功后，发送 `/helloworld` 就会触发这个指令，并回复 `你好, {user_name}!`
    @filter.on_llm_request()
    async def helloworld(self, event: AstrMessageEvent,req : ProviderRequest):
        """这是一个 hello world 指令""" # 这是 handler 的描述，将会被解析方便用户了解插件内容。建议填写。
        # logger.info("系统提示词："+req.system_prompt)
        req.extra_user_content_parts.append(TextPart(
            text=(
                "在回复内容中加入<wt>n</wt>模板输入想要等待的时间,n=等待秒数\n"
                "最短2s最长30min(1800s),不添加则正常回复(只有对方回复你才会被唤醒)\n"
                "例如'<wt>5</wt>'-->等待5秒后再次唤醒.\n"
                "使用场景:希望对方在多久之后能回你，或者希望这个话题能在短期内继续而不是因为对方的不回复就夭折\n"
                "例如如下场景:\n"
                "assistant:'你觉得咸粽子好吃韩式甜粽子好吃?<wt>40</wt>'\n"
                "(系统时间：四十秒后,系统定时唤醒)\n"
                "user:'<wt>无回应</wt>'\n"
                "assistant:'你怎么不理我<wt>40</wt>'\n"
                "(系统时间：二十秒后,用户回复，定时任务取消)\n"
                "user:'抱歉啊刚看到，我觉得咸的好吃awa'"
            )
        ))
    # @filter.event_message_type(filter.EventMessageType.PRIVATE_MESSAGE)
    # async def awa(self,event:AstrMessageEvent):
    #     yield event.plain_result(event.get_message_str())

    async def terminate(self):
        """可选择实现异步的插件销毁方法，当插件被卸载/停用时会调用。"""