mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-07-22 16:15:34 +08:00
update to 3.0.0-alpha2 (build 610): refactor driver
This commit is contained in:
crazywhalecc
@@ -1,368 +0,0 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
namespace ZM\Context;
|
||||
|
||||
use Closure;
|
||||
use Exception;
|
||||
use Stringable;
|
||||
use Swoole\Coroutine;
|
||||
use Swoole\Http\Request;
|
||||
use Swoole\WebSocket\Frame;
|
||||
use Swoole\WebSocket\Server;
|
||||
use ZM\API\ZMRobot;
|
||||
use ZM\Config\ZMConfig;
|
||||
use ZM\ConnectionManager\ConnectionObject;
|
||||
use ZM\ConnectionManager\ManagerGM;
|
||||
use ZM\Event\EventDispatcher;
|
||||
use ZM\Exception\InterruptException;
|
||||
use ZM\Exception\InvalidArgumentException;
|
||||
use ZM\Exception\WaitTimeoutException;
|
||||
use ZM\Exception\ZMKnownException;
|
||||
use ZM\Http\Response;
|
||||
use ZM\Utils\CoMessage;
|
||||
use ZM\Utils\MessageUtil;
|
||||
|
||||
class Context implements ContextInterface
|
||||
{
|
||||
public static $context = [];
|
||||
|
||||
private $cid;
|
||||
|
||||
public function __construct($cid)
|
||||
{
|
||||
$this->cid = $cid;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Server
|
||||
*/
|
||||
public function getServer(): ?Server
|
||||
{
|
||||
return self::$context[$this->cid]['server'] ?? server();
|
||||
}
|
||||
|
||||
public function getFrame(): ?Frame
|
||||
{
|
||||
return self::$context[$this->cid]['frame'] ?? null;
|
||||
}
|
||||
|
||||
public function getFd(): ?int
|
||||
{
|
||||
return self::$context[$this->cid]['fd'] ?? $this->getFrame()->fd ?? null;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return mixed
|
||||
*/
|
||||
public function getData()
|
||||
{
|
||||
return self::$context[$this->cid]['data'] ?? null;
|
||||
}
|
||||
|
||||
public function setData($data)
|
||||
{
|
||||
self::$context[$this->cid]['data'] = $data;
|
||||
}
|
||||
|
||||
public function getRequest(): ?Request
|
||||
{
|
||||
return self::$context[$this->cid]['request'] ?? null;
|
||||
}
|
||||
|
||||
public function getResponse(): ?Response
|
||||
{
|
||||
return self::$context[$this->cid]['response'] ?? null;
|
||||
}
|
||||
|
||||
public function getConnection(): ?ConnectionObject
|
||||
{
|
||||
return ManagerGM::get($this->getFd());
|
||||
}
|
||||
|
||||
public function getCid(): ?int
|
||||
{
|
||||
return $this->cid;
|
||||
}
|
||||
|
||||
public function getRobot(): ?ZMRobot
|
||||
{
|
||||
$conn = ManagerGM::get($this->getFrame()->fd);
|
||||
return $conn instanceof ConnectionObject ? new ZMRobot($conn) : null;
|
||||
}
|
||||
|
||||
public function getMessage()
|
||||
{
|
||||
if ((ZMConfig::get('global', 'onebot')['message_convert_string'] ?? true) === true && is_array($msg = $this->getOriginMessage())) {
|
||||
return MessageUtil::arrayToStr($msg);
|
||||
}
|
||||
return self::$context[$this->cid]['data']['message'] ?? null;
|
||||
}
|
||||
|
||||
public function setMessage($msg)
|
||||
{
|
||||
if (is_string($msg) && is_array($this->getOriginMessage())) {
|
||||
$msg = MessageUtil::strToArray($msg);
|
||||
}
|
||||
self::$context[$this->cid]['data']['message'] = $msg;
|
||||
}
|
||||
|
||||
public function getUserId()
|
||||
{
|
||||
return $this->getData()['user_id'] ?? null;
|
||||
}
|
||||
|
||||
public function setUserId($id)
|
||||
{
|
||||
self::$context[$this->cid]['data']['user_id'] = $id;
|
||||
}
|
||||
|
||||
public function getGroupId()
|
||||
{
|
||||
return $this->getData()['group_id'] ?? null;
|
||||
}
|
||||
|
||||
public function setGroupId($id)
|
||||
{
|
||||
self::$context[$this->cid]['data']['group_id'] = $id;
|
||||
}
|
||||
|
||||
public function getDiscussId()
|
||||
{
|
||||
return $this->getData()['discuss_id'] ?? null;
|
||||
}
|
||||
|
||||
public function setDiscussId($id)
|
||||
{
|
||||
self::$context[$this->cid]['data']['discuss_id'] = $id;
|
||||
}
|
||||
|
||||
public function getMessageType(): ?string
|
||||
{
|
||||
return $this->getData()['message_type'] ?? null;
|
||||
}
|
||||
|
||||
public function setMessageType($type)
|
||||
{
|
||||
self::$context[$this->cid]['data']['message_type'] = $type;
|
||||
}
|
||||
|
||||
public function getRobotId()
|
||||
{
|
||||
return $this->getData()['self_id'] ?? null;
|
||||
}
|
||||
|
||||
public function getCache($key)
|
||||
{
|
||||
return self::$context[$this->cid]['cache'][$key] ?? null;
|
||||
}
|
||||
|
||||
public function setCache($key, $value)
|
||||
{
|
||||
self::$context[$this->cid]['cache'][$key] = $value;
|
||||
}
|
||||
|
||||
public function getCQResponse()
|
||||
{
|
||||
return self::$context[$this->cid]['cq_response'] ?? null;
|
||||
}
|
||||
|
||||
/**
|
||||
* only can used by cq->message event function
|
||||
* @param array|string $msg 要回复的消息
|
||||
* @param bool|callable|Closure $yield 是否协程挂起（true），是否绑定异步事件（Closure）
|
||||
* @return array|bool 返回API调用结果
|
||||
*/
|
||||
public function reply($msg, $yield = false)
|
||||
{
|
||||
$data = $this->getData();
|
||||
$conn = $this->getConnection();
|
||||
if (!is_array($msg)) {
|
||||
switch ($this->getData()['message_type']) {
|
||||
case 'group':
|
||||
case 'private':
|
||||
case 'discuss':
|
||||
$this->setCache('has_reply', true);
|
||||
$operation['reply'] = $msg;
|
||||
$operation['at_sender'] = false;
|
||||
return (new ZMRobot($conn))->setCallback($yield)->callExtendedAPI('.handle_quick_operation', [
|
||||
'context' => $data,
|
||||
'operation' => $operation,
|
||||
]);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
$operation = $msg;
|
||||
return (new ZMRobot($conn))->setCallback(false)->callExtendedAPI('.handle_quick_operation', [
|
||||
'context' => $data,
|
||||
'operation' => $operation,
|
||||
]);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array|string $msg 要回复的消息
|
||||
* @param bool $yield 是否协程挂起（true），是否绑定异步事件（Closure）
|
||||
* @throws InterruptException 阻止消息被后续插件处理
|
||||
*/
|
||||
public function finalReply($msg, $yield = false)
|
||||
{
|
||||
self::$context[$this->cid]['cache']['block_continue'] = true;
|
||||
if ($msg != '') {
|
||||
$this->reply($msg, $yield);
|
||||
}
|
||||
EventDispatcher::interrupt();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $prompt
|
||||
* @param int $timeout
|
||||
* @param string $timeout_prompt
|
||||
* @throws WaitTimeoutException
|
||||
* @throws InvalidArgumentException
|
||||
* @return string 返回用户输入的内容
|
||||
*/
|
||||
public function waitMessage($prompt = '', $timeout = 600, $timeout_prompt = '')
|
||||
{
|
||||
if (!isset($this->getData()['user_id'], $this->getData()['message'], $this->getData()['self_id'])) {
|
||||
throw new InvalidArgumentException('协程等待参数缺失');
|
||||
}
|
||||
|
||||
logger()->debug('==== 开始等待输入 ====');
|
||||
if ($prompt != '') {
|
||||
$this->reply($prompt);
|
||||
}
|
||||
|
||||
try {
|
||||
$r = CoMessage::yieldByWS($this->getData(), ['user_id', 'self_id', 'message_type', get_onebot_target_id_name($this->getMessageType())], $timeout);
|
||||
} catch (Exception $e) {
|
||||
$r = false;
|
||||
}
|
||||
if ($r === false) {
|
||||
throw new WaitTimeoutException($this, $timeout_prompt);
|
||||
}
|
||||
if (is_array($r['message']) && (ZMConfig::get('global', 'onebot')['message_convert_string'] ?? true) === true) {
|
||||
return MessageUtil::arrayToStr($r['message']);
|
||||
}
|
||||
return $r['message'];
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据选定的模式获取消息参数
|
||||
* @param int|string $mode 获取的模式
|
||||
* @param string|Stringable $prompt_msg 提示语回复
|
||||
* @throws InvalidArgumentException
|
||||
* @throws ZMKnownException
|
||||
* @throws WaitTimeoutException
|
||||
* @return float|int|string
|
||||
*/
|
||||
public function getArgs($mode, $prompt_msg)
|
||||
{
|
||||
$arg = ctx()->getCache('match') ?? [];
|
||||
switch ($mode) {
|
||||
case ZM_MATCH_ALL:
|
||||
$p = $arg;
|
||||
return trim(implode(' ', $p)) == '' ? $this->waitMessage($prompt_msg) : trim(implode(' ', $p));
|
||||
case ZM_MATCH_NUMBER:
|
||||
foreach ($arg as $k => $v) {
|
||||
if (is_numeric($v)) {
|
||||
array_splice($arg, $k, 1);
|
||||
ctx()->setCache('match', $arg);
|
||||
return $v;
|
||||
}
|
||||
}
|
||||
return $this->waitMessage($prompt_msg);
|
||||
case ZM_MATCH_FIRST:
|
||||
if (isset($arg[0])) {
|
||||
$a = $arg[0];
|
||||
array_splice($arg, 0, 1);
|
||||
ctx()->setCache('match', $arg);
|
||||
return $a;
|
||||
}
|
||||
return $this->waitMessage($prompt_msg);
|
||||
}
|
||||
throw new InvalidArgumentException();
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下一个参数
|
||||
* @param string $prompt_msg 提示语回复
|
||||
* @throws InvalidArgumentException
|
||||
* @throws ZMKnownException
|
||||
* @throws WaitTimeoutException
|
||||
* @return int|mixed|string 返回获取的参数
|
||||
*/
|
||||
public function getNextArg($prompt_msg = '')
|
||||
{
|
||||
return $this->getArgs(ZM_MATCH_FIRST, $prompt_msg);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取接下来所有的消息当成一个完整的参数（包含空格）
|
||||
* @param string $prompt_msg 提示语回复
|
||||
* @throws InvalidArgumentException
|
||||
* @throws ZMKnownException
|
||||
* @throws WaitTimeoutException
|
||||
* @return int|mixed|string 返回获取的参数
|
||||
*/
|
||||
public function getFullArg($prompt_msg = '')
|
||||
{
|
||||
return $this->getArgs(ZM_MATCH_ALL, $prompt_msg);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取下一个数字类型的参数
|
||||
* @param string $prompt_msg 提示语回复
|
||||
* @throws InvalidArgumentException
|
||||
* @throws ZMKnownException
|
||||
* @throws WaitTimeoutException
|
||||
* @return int|mixed|string 返回获取的参数
|
||||
*/
|
||||
public function getNumArg($prompt_msg = '')
|
||||
{
|
||||
return $this->getArgs(ZM_MATCH_NUMBER, $prompt_msg);
|
||||
}
|
||||
|
||||
/**
|
||||
* @throws ZMKnownException
|
||||
* @return ContextInterface 返回上下文
|
||||
*/
|
||||
public function cloneFromParent(): ContextInterface
|
||||
{
|
||||
set_coroutine_params(self::$context[Coroutine::getPcid()] ?? self::$context[$this->cid]);
|
||||
return context();
|
||||
}
|
||||
|
||||
public function copy()
|
||||
{
|
||||
return self::$context[$this->cid];
|
||||
}
|
||||
|
||||
public function getOption()
|
||||
{
|
||||
return self::getCache('match');
|
||||
}
|
||||
|
||||
public function getOriginMessage()
|
||||
{
|
||||
return self::$context[$this->cid]['data']['message'] ?? null;
|
||||
}
|
||||
|
||||
public function getArrayMessage(): array
|
||||
{
|
||||
$msg = $this->getOriginMessage();
|
||||
if (is_array($msg)) {
|
||||
return $msg;
|
||||
}
|
||||
return MessageUtil::strToArray($msg);
|
||||
}
|
||||
|
||||
public function getStringMessage(): string
|
||||
{
|
||||
$msg = $this->getOriginMessage();
|
||||
if (is_string($msg)) {
|
||||
return $msg;
|
||||
}
|
||||
return MessageUtil::arrayToStr($msg);
|
||||
}
|
||||
}
|
||||
@@ -1,100 +0,0 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
namespace ZM\Context;
|
||||
|
||||
use Swoole\Http\Request;
|
||||
use Swoole\WebSocket\Frame;
|
||||
use Swoole\WebSocket\Server;
|
||||
use ZM\API\ZMRobot;
|
||||
use ZM\ConnectionManager\ConnectionObject;
|
||||
use ZM\Http\Response;
|
||||
|
||||
interface ContextInterface
|
||||
{
|
||||
public function __construct($cid);
|
||||
|
||||
/** @return Server */
|
||||
public function getServer();
|
||||
|
||||
/** @return Frame */
|
||||
public function getFrame();
|
||||
|
||||
/** @return mixed */
|
||||
public function getData();
|
||||
|
||||
public function setData($data);
|
||||
|
||||
/** @return ConnectionObject */
|
||||
public function getConnection();
|
||||
|
||||
/** @return null|int */
|
||||
public function getFd();
|
||||
|
||||
/** @return int */
|
||||
public function getCid();
|
||||
|
||||
/** @return Response */
|
||||
public function getResponse();
|
||||
|
||||
/** @return Request */
|
||||
public function getRequest();
|
||||
|
||||
/** @return ZMRobot */
|
||||
public function getRobot();
|
||||
|
||||
/** @return mixed */
|
||||
public function getUserId();
|
||||
|
||||
/** @return mixed */
|
||||
public function getGroupId();
|
||||
|
||||
/** @return mixed */
|
||||
public function getDiscussId();
|
||||
|
||||
/** @return string */
|
||||
public function getMessageType();
|
||||
|
||||
/** @return mixed */
|
||||
public function getRobotId();
|
||||
|
||||
/** @return mixed */
|
||||
public function getMessage();
|
||||
|
||||
public function setMessage($msg);
|
||||
|
||||
public function setUserId($id);
|
||||
|
||||
public function setGroupId($id);
|
||||
|
||||
public function setDiscussId($id);
|
||||
|
||||
public function setMessageType($type);
|
||||
|
||||
public function getCQResponse();
|
||||
|
||||
public function reply($msg, $yield = false);
|
||||
|
||||
public function finalReply($msg, $yield = false);
|
||||
|
||||
public function waitMessage($prompt = '', $timeout = 600, $timeout_prompt = '');
|
||||
|
||||
public function getArgs($mode, $prompt_msg);
|
||||
|
||||
public function getNextArg($prompt_msg = '');
|
||||
|
||||
public function getFullArg($prompt_msg = '');
|
||||
|
||||
public function setCache($key, $value);
|
||||
|
||||
public function getCache($key);
|
||||
|
||||
public function cloneFromParent();
|
||||
|
||||
public function getNumArg($prompt_msg = '');
|
||||
|
||||
public function copy();
|
||||
|
||||
public function getOption();
|
||||
}
|
||||
Reference in New Issue
Block a user