From d7084859188121d2b58c75d2703e8ff2f7738e3d Mon Sep 17 00:00:00 2001 From: crazywhalecc Date: Wed, 4 Jan 2023 21:23:25 +0800 Subject: [PATCH] add match reply mode --- src/Globals/global_defines_app.php | 5 ++++ src/ZM/Context/BotContext.php | 46 +++++++++++++++++++++++------- 2 files changed, 40 insertions(+), 11 deletions(-) diff --git a/src/Globals/global_defines_app.php b/src/Globals/global_defines_app.php index 719229c7..6f70fe2f 100644 --- a/src/Globals/global_defines_app.php +++ b/src/Globals/global_defines_app.php @@ -31,6 +31,11 @@ const ZM_ERR_METHOD_NOT_FOUND = 1; // 找不到方法 const ZM_ERR_ROUTE_NOT_FOUND = 2; // 找不到路由 const ZM_ERR_ROUTE_METHOD_NOT_ALLOWED = 3; // 路由方法不允许 +/** 定义 BotContext 下 reply 回复的模式 */ +const ZM_REPLY_NONE = 0; // 默认回复，不带任何东西 +const ZM_REPLY_MENTION = 1; // 回复时 @ 该用户 +const ZM_REPLY_QUOTE = 2; // 回复时引用该消息 + const LOAD_MODE_VENDOR = 0; // 从 vendor 加载 const LOAD_MODE_SRC = 1; // 从 src 加载 diff --git a/src/ZM/Context/BotContext.php b/src/ZM/Context/BotContext.php index c317d77c..faa0313c 100644 --- a/src/ZM/Context/BotContext.php +++ b/src/ZM/Context/BotContext.php @@ -4,9 +4,10 @@ declare(strict_types=1); namespace ZM\Context; +use DI\DependencyException; +use DI\NotFoundException; use OneBot\Driver\Event\Http\HttpRequestEvent; use OneBot\Driver\Event\WebSocket\WebSocketMessageEvent; -use OneBot\V12\Object\Action; use OneBot\V12\Object\MessageSegment; use OneBot\V12\Object\OneBotEvent; use ZM\Context\Trait\BotActionTrait; @@ -17,14 +18,16 @@ class BotContext implements ContextInterface { use BotActionTrait; + /** @var array> 记录机器人的上下文列表 */ private static array $bots = []; - private static array $echo_id_list = []; - + /** @var string[] 记录当前上下文绑定的机器人 */ private array $self; + /** @var array 如果是 BotCommand 匹配的上下文，这里会存放匹配到的参数 */ private array $params = []; + /** @var bool 用于标记当前上下文会话是否已经调用过 reply() 方法 */ private bool $replied = false; public function __construct(string $bot_id, string $platform, null|WebSocketMessageEvent|HttpRequestEvent $event = null) @@ -34,6 +37,12 @@ class BotContext implements ContextInterface $this->base_event = $event; } + /** + * 获取机器人事件对象 + * + * @throws DependencyException + * @throws NotFoundException + */ public function getEvent(): OneBotEvent { return container()->get('bot.event'); @@ -42,9 +51,10 @@ class BotContext implements ContextInterface /** * 快速回复机器人消息文本 * - * @param array|MessageSegment|string|\Stringable $message 消息内容、消息段或消息段数组 + * @param array|MessageSegment|string|\Stringable $message 消息内容、消息段或消息段数组 + * @param int $reply_mode 回复消息模式，默认为空，可选 ZM_REPLY_MENTION（at 用户）、ZM_REPLY_QUOTE（引用消息） */ - public function reply(\Stringable|MessageSegment|array|string $message) + public function reply(\Stringable|MessageSegment|array|string $message, int $reply_mode = ZM_REPLY_NONE) { if (container()->has('bot.event')) { // 这里直接使用当前上下文的事件里面的参数，不再重新挨个获取怎么发消息的参数 @@ -52,11 +62,13 @@ class BotContext implements ContextInterface $event = container()->get('bot.event'); // reply 的条件是必须 type=message - if ($event->getType() !== 'message') { + if ($event->type !== 'message') { throw new OneBot12Exception('bot()->reply() can only be used in message event.'); } $msg = (is_string($message) ? [new MessageSegment('text', ['text' => $message])] : ($message instanceof MessageSegment ? [$message] : $message)); $this->replied = true; + // 判断规则 + $this->matchReplyMode($reply_mode, $msg, $event); return $this->sendMessage($msg, $event->detail_type, $event->jsonSerialize()); } throw new OneBot12Exception('bot()->reply() can only be used in message event.'); @@ -126,13 +138,25 @@ class BotContext implements ContextInterface return $this->params; } - public function getEchoAction(mixed $echo): ?Action - { - return self::$echo_id_list[$echo] ?? null; - } - public function getSelf(): array { return $this->self; } + + /** + * 匹配更改 reply 的特殊模式 + * + * @param int $reply_mode 回复模式 + * @param array $message_segments 消息段的引用 + * @param OneBotEvent $event 事件对象 + */ + private function matchReplyMode(int $reply_mode, array &$message_segments, OneBotEvent $event) + { + if (($reply_mode & ZM_REPLY_QUOTE) === ZM_REPLY_QUOTE) { + array_unshift($message_segments, new MessageSegment('reply', ['message_id' => $event->getMessageId(), 'user_id' => $event->getUserId()])); + } + if (($reply_mode & ZM_REPLY_MENTION) === ZM_REPLY_MENTION) { + array_unshift($message_segments, new MessageSegment('mention', ['user_id' => $event->getUserId()])); + } + } }