mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-07-20 23:25:35 +08:00
enhancement for BotCommand and CommandArgument
This commit is contained in:
@@ -15,7 +15,7 @@ const ZM_VERSION_ID = Framework::VERSION_ID;
|
||||
const ZM_VERSION = Framework::VERSION;
|
||||
|
||||
/** 机器人用的,用于判断二元语意 */
|
||||
const TRUE_LIST = ['yes', 'y', 'true', 'on', '是', '对', true];
|
||||
const TRUE_LIST = ['yes', 'ok', 'y', 'true', 'on', '是', '对', true];
|
||||
const FALSE_LIST = ['no', 'n', 'false', 'off', '否', '错', false];
|
||||
|
||||
/** 定义多进程的全局变量 */
|
||||
|
||||
@@ -103,6 +103,54 @@ function is_assoc_array(array $array): bool
|
||||
return !empty($array) && array_keys($array) !== range(0, count($array) - 1);
|
||||
}
|
||||
|
||||
/**
|
||||
* 格式匹配
|
||||
*/
|
||||
function match_pattern(string $pattern, string $subject): bool
|
||||
{
|
||||
$pattern = str_replace(['\*', '\\\\.*'], ['.*', '\*'], preg_quote($pattern, '/'));
|
||||
$pattern = '/^' . $pattern . '$/i';
|
||||
return preg_match($pattern, $subject) === 1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 匹配参数
|
||||
*
|
||||
* @return array|false 成功时返回匹配到的参数数组,失败时返回false
|
||||
*/
|
||||
function match_args(string $pattern, string $subject)
|
||||
{
|
||||
$result = [];
|
||||
if (match_pattern($pattern, $subject)) {
|
||||
if (mb_strpos($pattern, '*') === false) {
|
||||
return [];
|
||||
}
|
||||
$exp = explode('*', $pattern);
|
||||
$i = 0;
|
||||
foreach ($exp as $k => $v) {
|
||||
if (empty($v) && $k === 0) {
|
||||
continue;
|
||||
}
|
||||
if (empty($v) && $k === count($exp) - 1) {
|
||||
$subject .= '^EOL';
|
||||
$v = '^EOL';
|
||||
}
|
||||
$cur_var = '';
|
||||
$ori = $i;
|
||||
while (($a = mb_substr($subject, $i, mb_strlen($v))) !== $v && !empty($a)) {
|
||||
$cur_var .= mb_substr($subject, $i, 1);
|
||||
++$i;
|
||||
}
|
||||
if ($i !== $ori || $k === 1 || $k === count($exp) - 1) {
|
||||
$result[] = $cur_var;
|
||||
}
|
||||
$i += mb_strlen($v);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 构建消息段的助手函数
|
||||
*
|
||||
|
||||
@@ -35,7 +35,7 @@ class AnnotationMap
|
||||
public static function loadAnnotationByParser(AnnotationParser $parser): void
|
||||
{
|
||||
// 生成后加入到全局list中
|
||||
self::$_list = array_merge(self::$_list, $parser->generateAnnotationList());
|
||||
self::$_list = array_merge_recursive(self::$_list, $parser->generateAnnotationList());
|
||||
self::$_map = $parser->getAnnotationMap();
|
||||
}
|
||||
|
||||
|
||||
@@ -28,34 +28,32 @@ class BotCommand extends AnnotationBase implements Level
|
||||
* @param string[] $alias
|
||||
*/
|
||||
public function __construct(
|
||||
public $name = '',
|
||||
public $match = '',
|
||||
public $pattern = '',
|
||||
public $regex = '',
|
||||
public $start_with = '',
|
||||
public $end_with = '',
|
||||
public $keyword = '',
|
||||
public $alias = [],
|
||||
public $detail_type = '',
|
||||
public $user_id = '',
|
||||
public $group_id = '',
|
||||
public $level = 20
|
||||
public string $name = '',
|
||||
public string $match = '',
|
||||
public string $pattern = '',
|
||||
public string $regex = '',
|
||||
public string $start_with = '',
|
||||
public string $end_with = '',
|
||||
public string $keyword = '',
|
||||
public array $alias = [],
|
||||
public string $detail_type = '',
|
||||
public string $prefix = '',
|
||||
public int $level = 20
|
||||
) {
|
||||
}
|
||||
|
||||
public static function make(
|
||||
$name = '',
|
||||
$match = '',
|
||||
$pattern = '',
|
||||
$regex = '',
|
||||
$start_with = '',
|
||||
$end_with = '',
|
||||
$keyword = '',
|
||||
$alias = [],
|
||||
$message_type = '',
|
||||
$user_id = '',
|
||||
$group_id = '',
|
||||
$level = 20
|
||||
string $name = '',
|
||||
string $match = '',
|
||||
string $pattern = '',
|
||||
string $regex = '',
|
||||
string $start_with = '',
|
||||
string $end_with = '',
|
||||
string $keyword = '',
|
||||
array $alias = [],
|
||||
string $detail_type = '',
|
||||
string $prefix = '',
|
||||
int $level = 20
|
||||
): BotCommand {
|
||||
return new static(...func_get_args());
|
||||
}
|
||||
|
||||
@@ -23,13 +23,14 @@ class CommandArgument extends AnnotationBase implements ErgodicAnnotation
|
||||
public string $type = 'string';
|
||||
|
||||
/**
|
||||
* @param string $name 参数名称(可以是中文)
|
||||
* @param string $description 参数描述(默认为空)
|
||||
* @param bool $required 参数是否必需,如果是必需,为true(默认为false)
|
||||
* @param string $prompt 当参数为必需且缺失时,返回给用户的提示输入的消息(默认为"请输入$name")
|
||||
* @param string $default 当required为false时,未匹配到参数将自动使用default值(默认为空)
|
||||
* @param int $timeout prompt超时时间(默认为60秒)
|
||||
* @throws InvalidArgumentException|ZMKnownException
|
||||
* @param string $name 参数名称(可以是中文)
|
||||
* @param string $description 参数描述(默认为空)
|
||||
* @param bool $required 参数是否必需,如果是必需,为true(默认为false)
|
||||
* @param string $prompt 当参数为必需且缺失时,返回给用户的提示输入的消息(默认为"请输入$name")
|
||||
* @param null|array|\Closure|float|int|string $default 当required为false时,未匹配到参数将自动使用default值(默认为空)
|
||||
* @param int $timeout prompt超时时间(默认为60秒)
|
||||
* @throws InvalidArgumentException
|
||||
* @throws ZMKnownException
|
||||
*/
|
||||
public function __construct(
|
||||
/**
|
||||
@@ -40,7 +41,7 @@ class CommandArgument extends AnnotationBase implements ErgodicAnnotation
|
||||
string $type = 'string',
|
||||
public bool $required = false,
|
||||
public string $prompt = '',
|
||||
public string $default = '',
|
||||
public string|\Closure|array|int|float|null $default = '',
|
||||
public int $timeout = 60,
|
||||
public int $error_prompt_policy = 1
|
||||
) {
|
||||
@@ -97,6 +98,10 @@ class CommandArgument extends AnnotationBase implements ErgodicAnnotation
|
||||
if (array_key_exists($type, $table)) {
|
||||
return $table[$type];
|
||||
}
|
||||
// 说明是自定义类型
|
||||
if (mb_strpos($type, '.') === 0 && mb_strlen($type) >= 2) {
|
||||
return $type;
|
||||
}
|
||||
throw new ZMKnownException(zm_internal_errcode('E00077') . 'Invalid argument type: ' . $type . ', only support any, string, number and bool !');
|
||||
}
|
||||
}
|
||||
|
||||
@@ -24,6 +24,10 @@ class BotContext implements ContextInterface
|
||||
|
||||
private array $self;
|
||||
|
||||
private array $params = [];
|
||||
|
||||
private bool $replied = false;
|
||||
|
||||
public function __construct(string $bot_id, string $platform)
|
||||
{
|
||||
$this->self = ['user_id' => $bot_id, 'platform' => $platform];
|
||||
@@ -55,6 +59,7 @@ class BotContext implements ContextInterface
|
||||
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;
|
||||
return $this->sendMessage($msg, $event->detail_type, $event->jsonSerialize());
|
||||
}
|
||||
throw new OneBot12Exception('bot()->reply() can only be used in message event.');
|
||||
@@ -65,8 +70,7 @@ class BotContext implements ContextInterface
|
||||
*/
|
||||
public function hasReplied(): bool
|
||||
{
|
||||
// TODO: 完成是否已经回复的记录和返回
|
||||
return false;
|
||||
return $this->replied;
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -93,6 +97,39 @@ class BotContext implements ContextInterface
|
||||
return $this->sendAction(Utils::camelToSeparator(__FUNCTION__), $params, $this->self);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置该消息下解析出来的参数列表
|
||||
*
|
||||
* @param array $params 参数列表
|
||||
*/
|
||||
public function setParams(array $params): void
|
||||
{
|
||||
$this->params = $params;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取单个参数值,不存在则返回 null
|
||||
*
|
||||
* @param int|string $name 参数名称或索引
|
||||
*/
|
||||
public function getParam(string|int $name): mixed
|
||||
{
|
||||
return $this->params[$name] ?? null;
|
||||
}
|
||||
|
||||
public function getParamString(string|int $name): ?string
|
||||
{
|
||||
return MessageUtil::getAltMessage($this->params[$name] ?? null);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有参数
|
||||
*/
|
||||
public function getParams(): array
|
||||
{
|
||||
return $this->params;
|
||||
}
|
||||
|
||||
public function getEchoAction(mixed $echo): ?Action
|
||||
{
|
||||
return self::$echo_id_list[$echo] ?? null;
|
||||
|
||||
16
src/ZM/Exception/WaitTimeoutException.php
Normal file
16
src/ZM/Exception/WaitTimeoutException.php
Normal file
@@ -0,0 +1,16 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
namespace ZM\Exception;
|
||||
|
||||
class WaitTimeoutException extends ZMException
|
||||
{
|
||||
public $module;
|
||||
|
||||
public function __construct($module, $message = '', $code = 0, \Throwable $previous = null)
|
||||
{
|
||||
parent::__construct($message, '', $code, $previous);
|
||||
$this->module = $module;
|
||||
}
|
||||
}
|
||||
@@ -10,8 +10,11 @@ use OneBot\Driver\Event\WebSocket\WebSocketMessageEvent;
|
||||
use OneBot\Driver\Event\WebSocket\WebSocketOpenEvent;
|
||||
use OneBot\V12\Exception\OneBotException;
|
||||
use OneBot\V12\Object\ActionResponse;
|
||||
use OneBot\V12\Object\MessageSegment;
|
||||
use OneBot\V12\Object\OneBotEvent;
|
||||
use OneBot\V12\Validator;
|
||||
use Psr\Container\ContainerExceptionInterface;
|
||||
use Psr\Container\NotFoundExceptionInterface;
|
||||
use ZM\Annotation\AnnotationHandler;
|
||||
use ZM\Annotation\AnnotationMap;
|
||||
use ZM\Annotation\AnnotationParser;
|
||||
@@ -21,10 +24,24 @@ use ZM\Annotation\OneBot\BotEvent;
|
||||
use ZM\Annotation\OneBot\CommandArgument;
|
||||
use ZM\Container\ContainerRegistrant;
|
||||
use ZM\Context\BotContext;
|
||||
use ZM\Exception\InterruptException;
|
||||
use ZM\Exception\OneBot12Exception;
|
||||
use ZM\Exception\WaitTimeoutException;
|
||||
use ZM\Utils\ConnectionUtil;
|
||||
use ZM\Utils\MessageUtil;
|
||||
|
||||
class OneBot12Adapter extends ZMPlugin
|
||||
{
|
||||
/**
|
||||
* 缓存待询问参数的队列
|
||||
*
|
||||
* 0: 代表 OneBotEvent 对象,即用于判断是否为同一会话环境
|
||||
* 1: \Generator 生成器,协程,不多讲
|
||||
* 2: BotCommand 注解对象
|
||||
* 3: match_result(array)匹配到一半的结果
|
||||
*/
|
||||
private static array $prompt_queue = [];
|
||||
|
||||
public function __construct(string $submodule = '', ?AnnotationParser $parser = null)
|
||||
{
|
||||
parent::__construct(__DIR__);
|
||||
@@ -35,8 +52,9 @@ class OneBot12Adapter extends ZMPlugin
|
||||
$this->addEvent(WebSocketOpenEvent::class, [$this, 'handleWSReverseOpen']);
|
||||
$this->addEvent(WebSocketMessageEvent::class, [$this, 'handleWSReverseMessage']);
|
||||
// 在 BotEvent 内处理 BotCommand
|
||||
// $cmd_event = BotEvent::make(type: 'message', level: 15)->on([$this, 'handleBotCommand']);
|
||||
// $this->addBotEvent($cmd_event);
|
||||
$this->addBotEvent(BotEvent::make(type: 'message', level: 15)->on([$this, 'handleBotCommand']));
|
||||
// 在 BotEvent 内处理需要等待回复的 CommandArgument
|
||||
$this->addBotEvent(BotEvent::make(type: 'message', level: 20)->on([$this, 'handlePrompt']));
|
||||
// 处理和声明所有 BotCommand 下的 CommandArgument
|
||||
$parser->addSpecialParser(BotCommand::class, [$this, 'parseBotCommand']);
|
||||
// 不需要给列表写入 CommandArgument
|
||||
@@ -77,30 +95,98 @@ class OneBot12Adapter extends ZMPlugin
|
||||
}
|
||||
|
||||
/**
|
||||
* 调用 BotCommand 注解的方法
|
||||
* [CALLBACK] 调用 BotCommand 注解的方法
|
||||
*
|
||||
* @param BotEvent $event BotEvent 事件
|
||||
* @param BotContext $ctx 机器人环境上下文
|
||||
* @param BotContext $ctx 机器人环境上下文
|
||||
* @throws ContainerExceptionInterface
|
||||
* @throws NotFoundExceptionInterface
|
||||
* @throws \Throwable
|
||||
* @throws OneBot12Exception
|
||||
*/
|
||||
public function handleBotCommand(BotEvent $event, BotContext $ctx)
|
||||
public function handleBotCommand(BotContext $ctx)
|
||||
{
|
||||
$handler = new AnnotationHandler(BotCommand::class);
|
||||
$handler->setReturnCallback(function ($result) use ($ctx) {
|
||||
if (is_string($result)) {
|
||||
$ctx->reply($result);
|
||||
return;
|
||||
}
|
||||
try {
|
||||
Validator::validateMessageSegment($result);
|
||||
$ctx->reply($result);
|
||||
} catch (\Throwable) {
|
||||
}
|
||||
if ($ctx->hasReplied()) {
|
||||
AnnotationHandler::interrupt();
|
||||
}
|
||||
});
|
||||
if ($ctx->hasReplied()) {
|
||||
return;
|
||||
}
|
||||
// 匹配消息
|
||||
$match_result = $this->matchBotCommand($ctx->getEvent());
|
||||
if ($match_result === null) {
|
||||
return;
|
||||
}
|
||||
// 匹配成功,检查 CommandArguments
|
||||
/** @var BotCommand $command */
|
||||
$command = $match_result[0];
|
||||
$arguments = $this->matchCommandArguments($match_result[1], $command);
|
||||
// 返回的是生成器,说明有需要询问的参数
|
||||
if ($arguments instanceof \Generator) {
|
||||
/** @var null|CommandArgument $argument */
|
||||
$argument = $arguments->current();
|
||||
if ($argument === null) {
|
||||
// 是 null 表明返回了空生成器,说明参数都已经匹配完毕
|
||||
$ctx->setParams($arguments->getReturn());
|
||||
$this->callBotCommand($ctx, $command);
|
||||
return;
|
||||
}
|
||||
$message = MessageSegment::text($argument->prompt === '' ? ('请输入' . $argument->name) : $argument->prompt);
|
||||
$ctx->reply([$message]);
|
||||
// 然后将此事件放入等待队列
|
||||
self::$prompt_queue[] = [$ctx->getEvent(), $arguments, $command, $match_result];
|
||||
return;
|
||||
}
|
||||
$ctx->setParams($arguments);
|
||||
// 调用方法
|
||||
$this->callBotCommand($ctx, $command);
|
||||
}
|
||||
|
||||
/**
|
||||
* [CALLBACK] 处理需要等待回复的 CommandArgument
|
||||
*
|
||||
* @param BotContext $ctx 机器人上下文
|
||||
* @throws ContainerExceptionInterface
|
||||
* @throws NotFoundExceptionInterface
|
||||
* @throws OneBot12Exception
|
||||
* @throws \Throwable
|
||||
*/
|
||||
public function handlePrompt(BotContext $ctx)
|
||||
{
|
||||
// 需要先从队列里找到定义当前会话的 prompt
|
||||
// 定义一个会话的标准是:事件的 detail_type,user_id,[group_id],[guild_id,channel_id] 全部相同
|
||||
$new_event = $ctx->getEvent();
|
||||
foreach (self::$prompt_queue as $k => $v) {
|
||||
/** @var OneBotEvent $old_event */
|
||||
$old_event = $v[0];
|
||||
if ($old_event->detail_type !== $new_event->detail_type) {
|
||||
continue;
|
||||
}
|
||||
$matched = match ($old_event->detail_type) {
|
||||
'private' => $new_event->getUserId() === $old_event->getUserId(),
|
||||
'group' => $new_event->getGroupId() === $old_event->getGroupId(),
|
||||
'guild' => $new_event->getGuildId() === $old_event->getGuildId() && $new_event->getChannelId() === $old_event->getChannelId(),
|
||||
default => false,
|
||||
};
|
||||
if (!$matched) {
|
||||
continue;
|
||||
}
|
||||
array_splice(self::$prompt_queue, $k, 1);
|
||||
// 找到了,开始处理
|
||||
/** @var \Generator $arguments */
|
||||
$arguments = $v[1];
|
||||
$new_arguments = $arguments->send($new_event->getMessage());
|
||||
if ($new_arguments !== null || $arguments->valid()) {
|
||||
// 还有需要询问的参数
|
||||
/** @var CommandArgument $argument */
|
||||
$argument = $arguments->current();
|
||||
$message = MessageSegment::text($argument->prompt === '' ? ('请输入' . $argument->name) : $argument->prompt);
|
||||
$ctx->reply([$message]);
|
||||
// 然后将此事件放入等待队列
|
||||
self::$prompt_queue[] = [$ctx->getEvent(), $arguments, $v[2], $v[3]];
|
||||
} else {
|
||||
// 所有参数都已经获取到了,调用方法
|
||||
$ctx->setParams($arguments->getReturn());
|
||||
$this->callBotCommand($ctx, $v[2]);
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -118,8 +204,10 @@ class OneBot12Adapter extends ZMPlugin
|
||||
}
|
||||
|
||||
/**
|
||||
* 接入和认证反向 WS 的连接
|
||||
* [CALLBACK] 接入和认证反向 WS 的连接
|
||||
*
|
||||
* @throws StopException
|
||||
* @throws \JsonException
|
||||
*/
|
||||
public function handleWSReverseOpen(WebSocketOpenEvent $event): void
|
||||
{
|
||||
@@ -151,7 +239,7 @@ class OneBot12Adapter extends ZMPlugin
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理 WebSocket 消息
|
||||
* [CALLBACK] 处理 WebSocket 消息
|
||||
*
|
||||
* @param WebSocketMessageEvent $event 事件对象
|
||||
* @throws OneBotException
|
||||
@@ -177,7 +265,7 @@ class OneBot12Adapter extends ZMPlugin
|
||||
// 如果含有 type,detail_type 字段,表明是 event
|
||||
try {
|
||||
$obj = new OneBotEvent($body);
|
||||
} catch (OneBotException $e) {
|
||||
} catch (OneBotException) {
|
||||
logger()->debug('收到非 OneBot 12 标准的消息,已忽略');
|
||||
return;
|
||||
}
|
||||
@@ -192,7 +280,11 @@ class OneBot12Adapter extends ZMPlugin
|
||||
&& ($event->sub_type === null || $event->sub_type === $obj->sub_type)
|
||||
&& ($event->detail_type === null || $event->detail_type === $obj->detail_type);
|
||||
});
|
||||
$handler->handleAll($obj);
|
||||
try {
|
||||
$handler->handleAll($obj);
|
||||
} catch (WaitTimeoutException $e) {
|
||||
bot()->reply($e->getMessage());
|
||||
}
|
||||
} elseif (isset($body['status'], $body['retcode'])) {
|
||||
// 如果含有 status,retcode 字段,表明是 action 的 response
|
||||
$resp = new ActionResponse();
|
||||
@@ -212,11 +304,325 @@ class OneBot12Adapter extends ZMPlugin
|
||||
}
|
||||
}
|
||||
|
||||
private function matchBotCommand(OneBotEvent $event): array
|
||||
/**
|
||||
* 根据事件匹配规则
|
||||
*
|
||||
* @param OneBotEvent $event 事件对象
|
||||
*/
|
||||
public function matchBotCommand(OneBotEvent $event): ?array
|
||||
{
|
||||
/** @var BotCommand[] $ls */
|
||||
$ls = AnnotationMap::$_list[BotCommand::class] ?? [];
|
||||
$msg = $event->getMessageString();
|
||||
// TODO: 还没写完匹配 BotCommand
|
||||
return [];
|
||||
$msgs = $event->getMessage();
|
||||
$head = '';
|
||||
$cmd_explode = [];
|
||||
$full_str = '';
|
||||
foreach ($msgs as $segment) {
|
||||
/** @param \MessageSegment $segment */
|
||||
if ($segment->type !== 'text') {
|
||||
if ($head === '') {
|
||||
continue;
|
||||
}
|
||||
$cmd_explode[] = $segment;
|
||||
continue;
|
||||
}
|
||||
// 当没识别到命令头的时候,就当作命令头识别
|
||||
if ($head === '') {
|
||||
$text = $segment->data['text'];
|
||||
$full_str .= $text;
|
||||
// 切分字符串
|
||||
$nlp = MessageUtil::splitMessage(str_replace("\r", '', $text));
|
||||
// 啥也没有,分个锤子
|
||||
if (empty($nlp)) {
|
||||
continue;
|
||||
}
|
||||
// 先预留一个给分组而配置的空间
|
||||
$cmd_explode = $nlp;
|
||||
$head = $nlp[0];
|
||||
} else {
|
||||
$full_str .= $segment->data['text'];
|
||||
// 如果已经识别到了命令头,就当作命令体识别
|
||||
$nlp = MessageUtil::splitMessage(str_replace("\r", '', $segment->data['text']));
|
||||
if (empty($nlp)) {
|
||||
continue;
|
||||
}
|
||||
$cmd_explode = array_merge($cmd_explode, $nlp);
|
||||
}
|
||||
// 先匹配
|
||||
}
|
||||
if ($head === '') {
|
||||
return null;
|
||||
}
|
||||
// 遍历所有 BotCommand 注解
|
||||
foreach ($ls as $v) {
|
||||
/** @var BotCommand $v */
|
||||
// 测试 deatil_type
|
||||
if ($v->detail_type !== '' && $v->detail_type !== $event->detail_type) {
|
||||
continue;
|
||||
}
|
||||
// 测试 prefix
|
||||
if ($v->prefix !== '' && mb_strpos($full_str, $v->prefix) !== 0) {
|
||||
continue;
|
||||
}
|
||||
// 测试 match
|
||||
if ($v->match !== '' && $v->match === $head) {
|
||||
array_shift($cmd_explode);
|
||||
return [$v, $cmd_explode, $full_str];
|
||||
}
|
||||
// 测试 alias
|
||||
if ($v->match !== '' && $v->alias !== [] && in_array($head, $v->alias, true)) {
|
||||
array_shift($cmd_explode);
|
||||
return [$v, $cmd_explode, $full_str];
|
||||
}
|
||||
// 测试 pattern
|
||||
if ($v->pattern !== '' && ($args = match_args($v->pattern, $full_str)) !== false) {
|
||||
return [$v, $args, $full_str];
|
||||
}
|
||||
// 测试 regex
|
||||
if ($v->regex !== '' && preg_match('/' . $v->regex . '/u', $full_str, $match) !== 0) {
|
||||
return [$v, $match, $full_str];
|
||||
}
|
||||
// 测试 start_with
|
||||
if ($v->start_with !== '' && mb_strpos($full_str, $v->start_with) === 0) {
|
||||
return [$v, [mb_substr($full_str, mb_strlen($v->start_with))], $full_str];
|
||||
}
|
||||
// 测试 end_with
|
||||
if ($v->end_with !== '' && mb_substr($full_str, 0 - mb_strlen($v->end_with)) === $v->end_with) {
|
||||
return [$v, [mb_substr($full_str, 0, 0 - mb_strlen($v->end_with))], $full_str];
|
||||
}
|
||||
// 测试 keyword
|
||||
if ($v->keyword !== '' && mb_strpos($full_str, $v->keyword) !== false) {
|
||||
return [$v, explode($v->keyword, $full_str), $full_str];
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据匹配结果和 CommandArgument 进行匹配
|
||||
*
|
||||
* @param array $match_result 匹配结果的引用
|
||||
* @param BotCommand $cmd 注解对象
|
||||
* @return array|\Generator 返回 array 时为匹配结果,返回 Generator 时为等待结果
|
||||
* @throws WaitTimeoutException
|
||||
*/
|
||||
private function matchCommandArguments(array $match_result, BotCommand $cmd): array|\Generator
|
||||
{
|
||||
$arguments = [];
|
||||
/** @var CommandArgument $argument */
|
||||
foreach ($cmd->getArguments() as $argument) {
|
||||
switch ($argument->type) {
|
||||
case 'string':
|
||||
case 'any':
|
||||
case 'str':
|
||||
$cnt = count($match_result);
|
||||
for ($k = 0; $k < $cnt; ++$k) {
|
||||
$v = $match_result[$k];
|
||||
if (is_string($v)) {
|
||||
array_splice($match_result, $k, 1);
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
if ($argument->required) {
|
||||
// 不够用,且是必需的,就询问用户(这里可能还需要考虑没有协程环境怎么处理)
|
||||
$g = yield $argument;
|
||||
foreach ($g as $v) {
|
||||
if ($v instanceof MessageSegment && $v->type === 'text') {
|
||||
$arguments[$argument->name] = $v->data['text'];
|
||||
break 2;
|
||||
}
|
||||
if (is_string($v)) {
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
if ($argument->error_prompt_policy === 1) {
|
||||
$prompt = $argument->getTypeErrorPrompt() . "\n" . $argument->prompt;
|
||||
$clone_argument = clone $argument;
|
||||
$clone_argument->prompt = $prompt;
|
||||
$g = yield $clone_argument;
|
||||
foreach ($g as $v) {
|
||||
if ($v instanceof MessageSegment && $v->type === 'text') {
|
||||
$arguments[$argument->name] = $v->data['text'];
|
||||
break 2;
|
||||
}
|
||||
if (is_string($v)) {
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
throw new WaitTimeoutException($cmd->name, $argument->getErrorQuitPrompt());
|
||||
} else {
|
||||
// 非必需,就使用缺省值
|
||||
$arguments[$argument->name] = $argument->default;
|
||||
}
|
||||
break;
|
||||
case 'number':
|
||||
$cnt = count($match_result);
|
||||
// 遍历现存的参数列表,找到第一个数字
|
||||
for ($k = 0; $k < $cnt; ++$k) {
|
||||
$v = $match_result[$k];
|
||||
if (is_numeric($v)) {
|
||||
array_splice($match_result, $k, 1);
|
||||
$arguments[$argument->name] = $v / 1;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
// 找不到就看看是不是必需的,如果不是必需的,且缺省值是数字,那么就顶上
|
||||
if (!$argument->required && is_numeric($argument->default)) {
|
||||
$arguments[$argument->name] = $argument->default / 1;
|
||||
break;
|
||||
}
|
||||
// 到这里还没找到,就说明需要询问用户了
|
||||
$g = yield $argument;
|
||||
foreach ($g as $v) {
|
||||
if (is_numeric($v)) {
|
||||
$arguments[$argument->name] = $v / 1;
|
||||
break 2;
|
||||
}
|
||||
if ($v instanceof MessageSegment && $v->type === 'text') {
|
||||
if (is_numeric($v->data['text'])) {
|
||||
$arguments[$argument->name] = $v->data['text'] / 1;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
if ($argument->error_prompt_policy === 1) {
|
||||
$prompt = $argument->getTypeErrorPrompt() . "\n" . $argument->prompt;
|
||||
$clone_argument = clone $argument;
|
||||
$clone_argument->prompt = $prompt;
|
||||
$g = yield $clone_argument;
|
||||
foreach ($g as $v) {
|
||||
if (is_numeric($v)) {
|
||||
$arguments[$argument->name] = $v / 1;
|
||||
break 2;
|
||||
}
|
||||
if ($v instanceof MessageSegment && $v->type === 'text') {
|
||||
if (is_numeric($v->data['text'])) {
|
||||
$arguments[$argument->name] = $v->data['text'] / 1;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
throw new WaitTimeoutException($cmd->name, $argument->getErrorQuitPrompt());
|
||||
case 'bool':
|
||||
// 先遍历参数,找到具有布尔值参数的语言
|
||||
$cnt = count($match_result);
|
||||
for ($k = 0; $k < $cnt; ++$k) {
|
||||
$v = $match_result[$k];
|
||||
// 看看有没有true值
|
||||
if (in_array(strtolower($v), TRUE_LIST, true)) {
|
||||
array_splice($match_result, $k, 1);
|
||||
$arguments[$argument->name] = true;
|
||||
break 2;
|
||||
}
|
||||
// 看看有没有false值
|
||||
if (in_array(strtolower($v), FALSE_LIST, true)) {
|
||||
array_splice($match_result, $k, 1);
|
||||
$arguments[$argument->name] = false;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
// 如果不是必需的,那就使用缺省值
|
||||
if (!$argument->required) {
|
||||
$arguments[$argument->name] = in_array($argument->default === '' ? true : 'true', TRUE_LIST);
|
||||
break;
|
||||
}
|
||||
// 到这里还没找到,就说明需要询问用户了
|
||||
$g = yield $argument;
|
||||
if (in_array(strtolower($g), TRUE_LIST, true)) {
|
||||
$arguments[$argument->name] = true;
|
||||
} elseif (in_array(strtolower($g), FALSE_LIST, true)) {
|
||||
$arguments[$argument->name] = false;
|
||||
} else {
|
||||
if ($argument->error_prompt_policy === 1) {
|
||||
$prompt = $argument->getTypeErrorPrompt() . "\n" . $argument->prompt;
|
||||
$clone_argument = clone $argument;
|
||||
$clone_argument->prompt = $prompt;
|
||||
$g = yield $clone_argument;
|
||||
if (in_array(strtolower($g), TRUE_LIST, true)) {
|
||||
$arguments[$argument->name] = true;
|
||||
} elseif (in_array(strtolower($g), FALSE_LIST, true)) {
|
||||
$arguments[$argument->name] = false;
|
||||
} else {
|
||||
throw new WaitTimeoutException($cmd->name, $argument->getErrorQuitPrompt());
|
||||
}
|
||||
} else {
|
||||
throw new WaitTimeoutException($cmd->name, $argument->getErrorQuitPrompt());
|
||||
}
|
||||
}
|
||||
break;
|
||||
default:
|
||||
// 其他类型,处理富文本
|
||||
$msg_type = mb_substr($argument->type, 1);
|
||||
$cnt = count($match_result);
|
||||
for ($k = 0; $k < $cnt; ++$k) {
|
||||
$v = $match_result[$k];
|
||||
if ($v instanceof MessageSegment && $v->type === $msg_type) {
|
||||
array_splice($match_result, $k, 1);
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
// 如果不是必需的,那就使用缺省值
|
||||
if (!$argument->required && is_array($argument->default)) {
|
||||
// 生成一个消息段的段
|
||||
$a = [new MessageSegment($msg_type, $argument->default)];
|
||||
$arguments[$argument->name] = $a;
|
||||
break;
|
||||
}
|
||||
// 到这里还没找到,就说明需要询问用户了
|
||||
$g = yield $argument;
|
||||
foreach ($g as $v) {
|
||||
if ($v instanceof MessageSegment && $v->type === $msg_type) {
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
if ($argument->error_prompt_policy === 1) {
|
||||
$prompt = $argument->getTypeErrorPrompt() . "\n" . $argument->prompt;
|
||||
$clone_argument = clone $argument;
|
||||
$clone_argument->prompt = $prompt;
|
||||
$g = yield $clone_argument;
|
||||
foreach ($g as $v) {
|
||||
if ($v instanceof MessageSegment && $v->type === $msg_type) {
|
||||
$arguments[$argument->name] = $v;
|
||||
break 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
throw new WaitTimeoutException($cmd->name, $argument->getErrorQuitPrompt());
|
||||
}
|
||||
}
|
||||
$arguments['.unnamed'] = $match_result;
|
||||
return $arguments;
|
||||
}
|
||||
|
||||
/**
|
||||
* @throws InterruptException
|
||||
* @throws \Throwable
|
||||
*/
|
||||
private function callBotCommand(BotContext $ctx, BotCommand $cmd)
|
||||
{
|
||||
$handler = new AnnotationHandler(BotCommand::class);
|
||||
$handler->setReturnCallback(function ($result) use ($ctx) {
|
||||
if (is_string($result)) {
|
||||
$ctx->reply($result);
|
||||
return;
|
||||
}
|
||||
try {
|
||||
Validator::validateMessageSegment($result);
|
||||
$ctx->reply($result);
|
||||
} catch (\Throwable) {
|
||||
}
|
||||
if ($ctx->hasReplied()) {
|
||||
AnnotationHandler::interrupt();
|
||||
}
|
||||
});
|
||||
$handler->handle($cmd);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -67,7 +67,7 @@ class PluginGenerator
|
||||
'{class}' => $this->convertClassName(),
|
||||
];
|
||||
$main_php = str_replace(array_keys($replace), array_values($replace), $template);
|
||||
file_put_contents(zm_dir($plugin_base_dir . '/src/PluginMain.php'), $main_php);
|
||||
file_put_contents(zm_dir($plugin_base_dir . '/src/' . $this->convertClassName() . '.php'), $main_php);
|
||||
// 写入 composer.json
|
||||
$composer_json = [
|
||||
'autoload' => [
|
||||
|
||||
@@ -64,6 +64,11 @@ class MessageUtil
|
||||
public static function convertToArr(MessageSegment|\Stringable|array|string $message)
|
||||
{
|
||||
if (is_array($message)) {
|
||||
foreach ($message as $k => $v) {
|
||||
if (is_string($v)) {
|
||||
$message[$k] = new MessageSegment('text', ['text' => $v]);
|
||||
}
|
||||
}
|
||||
return $message;
|
||||
}
|
||||
if ($message instanceof MessageSegment) {
|
||||
@@ -74,4 +79,48 @@ class MessageUtil
|
||||
}
|
||||
return new MessageSegment('text', ['text' => $message]);
|
||||
}
|
||||
|
||||
/**
|
||||
* 分割消息字符串
|
||||
*
|
||||
* @param array $includes 需要进行切割的字符串,默认包含空格及制表符(\t)
|
||||
*/
|
||||
public static function splitMessage(string $msg, array $includes = [' ', "\t"]): array
|
||||
{
|
||||
$msg = trim($msg);
|
||||
foreach ($includes as $v) {
|
||||
$msg = str_replace($v, "\n", $msg);
|
||||
}
|
||||
$msg_seg = explode("\n", $msg);
|
||||
$ls = [];
|
||||
foreach ($msg_seg as $v) {
|
||||
if (empty(trim($v))) {
|
||||
continue;
|
||||
}
|
||||
$ls[] = trim($v);
|
||||
}
|
||||
return $ls;
|
||||
}
|
||||
|
||||
public static function getAltMessage(null|array|string|MessageSegment $message): string
|
||||
{
|
||||
if ($message === null) {
|
||||
return '';
|
||||
}
|
||||
if (is_string($message)) {
|
||||
return $message;
|
||||
}
|
||||
if ($message instanceof MessageSegment) {
|
||||
$message = [$message];
|
||||
}
|
||||
$message_string = '';
|
||||
foreach ($message as $segment) {
|
||||
if ($segment->type === 'text') {
|
||||
$message_string .= $segment->data['text'];
|
||||
} else {
|
||||
$message_string .= '[富文本:' . $segment->type . ']';
|
||||
}
|
||||
}
|
||||
return $message_string;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user