zhamao-framework/src/ZM/Container/ContainerRegistrant.php

<?php

declare(strict_types=1);

namespace ZM\Container;

use DI;
use OneBot\Driver\Event\Http\HttpRequestEvent;
use OneBot\Driver\Event\WebSocket\WebSocketCloseEvent;
use OneBot\Driver\Event\WebSocket\WebSocketMessageEvent;
use OneBot\Driver\Event\WebSocket\WebSocketOpenEvent;
use OneBot\V12\Object\ActionResponse;
use OneBot\V12\Object\OneBotEvent;
use Psr\Http\Message\ServerRequestInterface;
use ZM\Context\BotContext;
use ZM\Context\Context;
use ZM\Context\ContextInterface;

class ContainerRegistrant
{
    /**
     * 应在收到 OneBot 事件时调用
     */
    public static function registerOBEventServices(OneBotEvent $event): void
    {
        self::addServices([
            OneBotEvent::class => $event,
            'bot.event' => DI\get(OneBotEvent::class),
            BotContext::class => fn () => bot(),
        ]);
    }

    /**
     * 应在收到 OneBot 动作响应时调用
     */
    public static function registerOBActionResponseServices(ActionResponse $response): void
    {
        self::addServices([
            ActionResponse::class => $response,
            'bot.action.response' => DI\get(ActionResponse::class),
        ]);
    }

    /**
     * 应在收到 HTTP 请求时调用
     */
    public static function registerHttpRequestServices(HttpRequestEvent $event): void
    {
        self::addServices([
            HttpRequestEvent::class => $event,
            'http.request.event' => DI\get(HttpRequestEvent::class),
            ServerRequestInterface::class => fn () => $event->getRequest(),
            'http.request' => DI\get(ServerRequestInterface::class),
            ContextInterface::class => Context::class,
        ]);
    }

    /**
     * 应在收到 WebSocket 连接时调用
     */
    public static function registerWSOpenServices(WebSocketOpenEvent $event): void
    {
        self::addServices([
            WebSocketOpenEvent::class => $event,
            'ws.open.event' => DI\get(WebSocketOpenEvent::class),
        ]);
    }

    /**
     * 应在收到 WebSocket 消息时调用
     */
    public static function registerWSMessageServices(WebSocketMessageEvent $event): void
    {
        self::addServices([
            WebSocketMessageEvent::class => $event,
            'ws.message.event' => DI\get(WebSocketMessageEvent::class),
        ]);
    }

    /**
     * 应在收到 WebSocket 关闭时调用
     */
    public static function registerWSCloseServices(WebSocketCloseEvent $event): void
    {
        self::addServices([
            WebSocketCloseEvent::class => $event,
            'ws.close.event' => DI\get(WebSocketCloseEvent::class),
        ]);
    }

    private static function addServices(array $services): void
    {
        foreach ($services as $name => $service) {
            ContainerHolder::getEventContainer()->set($name, $service);
        }
    }
}