zhamao-framework/src/ZM/Context/Trait/BotActionTrait.php

<?php

declare(strict_types=1);

namespace ZM\Context\Trait;

use OneBot\Driver\Coroutine\Adaptive;
use OneBot\Util\Utils;
use OneBot\V12\Object\Action;
use OneBot\V12\Object\ActionResponse;
use OneBot\V12\Object\MessageSegment;
use ZM\Annotation\AnnotationHandler;
use ZM\Annotation\OneBot\BotAction;
use ZM\Context\BotConnectContext;
use ZM\Exception\OneBot12Exception;
use ZM\Plugin\OneBot\BotMap;
use ZM\Utils\MessageUtil;

trait BotActionTrait
{
    /**
     * @internal 只允许内部调用
     * @param ActionResponse $response 尝试调用看看有没有协程等待的
     */
    public static function tryResume(ActionResponse $response): void
    {
        if (($co = Adaptive::getCoroutine()) !== null && isset(BotMap::$bot_coroutines[$response->echo ?? ''])) {
            $co->resume(BotMap::$bot_coroutines[$response->echo ?? ''], $response);
        }
    }

    /**
     * 发送一条机器人消息
     *
     * @param  array|MessageSegment|string|\Stringable $message 消息内容，可以是消息段、字符串
     * @throws \Throwable
     */
    public function sendMessage(array|MessageSegment|string|\Stringable $message, string $detail_type, array $params = []): ActionResponse|bool
    {
        $message = MessageUtil::convertToArr($message);
        $params['message'] = $message;
        $params['detail_type'] = $detail_type;
        return $this->sendAction(Utils::camelToSeparator(__FUNCTION__), $params, $this->self);
    }

    /**
     * 发送机器人动作
     *
     * @throws \Throwable
     */
    public function sendAction(string $action, array $params = [], ?array $self = null): ActionResponse|bool
    {
        if ($self === null && $this->self !== null) {
            $self = $this->self;
        }
        // 声明 Action 对象
        $a = new Action($action, $params, ob_uuidgen(), $self);
        // 调用事件在回复之前的回调
        $handler = new AnnotationHandler(BotAction::class);
        container()->set(Action::class, $a);
        $handler->setRuleCallback(fn (BotAction $act) => ($act->action === '' || $act->action === $action) && !$act->need_response);
        $handler->handleAll();
        // 被阻断时候，就不发送了
        if ($handler->getStatus() === AnnotationHandler::STATUS_INTERRUPTED) {
            return false;
        }

        // 获取机器人的 BotMap 对应连接（前提是当前上下文有 self）
        if ($self !== null) {
            $fd_map = BotMap::getBotFd($self['user_id'], $self['platform']);
            if ($fd_map === null) {
                logger()->error("机器人 [{$self['platform']}:{$self['user_id']}] 没有连接或未就绪，无法发送数据");
                return false;
            }
            $result = ws_socket($fd_map[0])->send(json_encode($a->jsonSerialize()), $fd_map[1]);
        } elseif ($this instanceof BotConnectContext) {
            // self 为空，说明可能是发送的元动作，需要通过 fd 来查找对应的 connect 连接
            $flag = $this->getFlag();
            $fd = $this->getFd();
            $result = ws_socket($flag)->send(json_encode($a->jsonSerialize()), $fd);
        } elseif (method_exists($this, 'emitSendAction')) {
            $result = $this->emitSendAction($a);
        } else {
            logger()->error('未匹配到任何机器人连接');
            return false;
        }

        // 如果开启了协程，并且成功发送，那就进入协程等待，挂起等待结果返回一个 ActionResponse 对象
        if (($result ?? false) === true && ($co = Adaptive::getCoroutine()) !== null) {
            BotMap::$bot_coroutines[$a->echo] = $co->getCid();
            $response = $co->suspend();
            if ($response instanceof ActionResponse) {
                $handler = new AnnotationHandler(BotAction::class);
                $handler->setRuleCallback(fn (BotAction $act) => ($act->action === '' || $act->action === $action) && $act->need_response);
                container()->set(ActionResponse::class, $response);
                $handler->handleAll();
                return $response;
            }
            return false;
        }
        if (isset($result)) {
            return $result;
        }
        // 到这里表明你调用时候不在 WS 或 HTTP 上下文
        throw new OneBot12Exception('No bot connection found.');
    }
}