mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
# 转发消息组件
|
|
from __future__ import annotations
|
|
import typing
|
|
|
|
from mirai.models import MessageChain
|
|
from mirai.models.message import MessageComponent, ForwardMessageNode
|
|
from mirai.models.base import MiraiBaseModel
|
|
|
|
from .. import strategy as strategy_model
|
|
|
|
|
|
class ForwardMessageDiaplay(MiraiBaseModel):
|
|
title: str = "群聊的聊天记录"
|
|
brief: str = "[聊天记录]"
|
|
source: str = "聊天记录"
|
|
preview: typing.List[str] = []
|
|
summary: str = "查看x条转发消息"
|
|
|
|
|
|
class Forward(MessageComponent):
|
|
"""合并转发。"""
|
|
type: str = "Forward"
|
|
"""消息组件类型。"""
|
|
display: ForwardMessageDiaplay
|
|
"""显示信息"""
|
|
node_list: typing.List[ForwardMessageNode]
|
|
"""转发消息节点列表。"""
|
|
def __init__(self, *args, **kwargs):
|
|
if len(args) == 1:
|
|
self.node_list = args[0]
|
|
super().__init__(**kwargs)
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def __str__(self):
|
|
return '[聊天记录]'
|
|
|
|
|
|
class ForwardComponentStrategy(strategy_model.LongTextStrategy):
|
|
|
|
async def process(self, message: str) -> list[MessageComponent]:
|
|
display = ForwardMessageDiaplay(
|
|
title="群聊的聊天记录",
|
|
brief="[聊天记录]",
|
|
source="聊天记录",
|
|
preview=["QQ用户: "+message],
|
|
summary="查看1条转发消息"
|
|
)
|
|
|
|
node_list = [
|
|
ForwardMessageNode(
|
|
sender_id=self.ap.im_mgr.bot_account_id,
|
|
sender_name='QQ用户',
|
|
message_chain=MessageChain([message])
|
|
)
|
|
]
|
|
|
|
forward = Forward(
|
|
display=display,
|
|
node_list=node_list
|
|
)
|
|
|
|
return [forward]
|