mercureact/src/Http/Server.php

<?php

namespace NoccyLabs\Mercureact\Http;

use Monolog\Handler\StreamHandler;
use Monolog\Logger;
use NoccyLabs\Mercureact\Broker\TopicManager;
use NoccyLabs\Mercureact\Configuration;
use NoccyLabs\Mercureact\Http\Middleware\ApiHandler;
use NoccyLabs\Mercureact\Http\Middleware\MercureHandler;
use NoccyLabs\Mercureact\Http\Middleware\NotFoundHandler;
use NoccyLabs\Mercureact\Http\Middleware\ResponseMiddleware;
use NoccyLabs\Mercureact\Http\Middleware\SecurityMiddleware;
use NoccyLabs\Mercureact\Http\Middleware\WebSocketHandler;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use React\EventLoop\Loop;
use React\EventLoop\LoopInterface;
use React\Http\HttpServer;
use React\Socket\SecureServer;
use React\Socket\ServerInterface;
use SplObjectStorage;

class Server
{
    private Configuration $config;

    private LoopInterface $loop;

    private HttpServer $server;

    private SplObjectStorage $webSocketClients;

    private TopicManager $topicManager;
    
    private Logger $logger;

    private ResponseMiddleware $responseMiddleware;
    private SecurityMiddleware $securityMiddleware;
    private WebSocketHandler $webSocketHandler;
    private MercureHandler $mercureHandler;
    private ApiHandler $apiRequestHandler;
    private NotFoundHandler $notFoundHandler;


    /**
     * 
     * 
     */
    public function __construct(Configuration $config, ?LoggerInterface $logger, ?LoopInterface $loop=null)
    {
        $this->loop = $loop??Loop::get();

        $this->config = $config;

        $this->logger = $logger ?? new NullLogger();
        if ($logger instanceof Logger) {
            $topicLogger = $logger->withName("broker");
        } else {
            $topicLogger = $this->logger;
        }

        $this->topicManager = new TopicManager($topicLogger);
        $this->loop->addPeriodicTimer(30, function () {
            $this->topicManager->garbageCollect();
        });
        
        $this->webSocketClients = new SplObjectStorage();

        $this->server = $this->createHttpServer();
    }

    /**
     * 
     * 
     * @return void
     */
    public function listen(ServerInterface $socket): void
    {
        $this->server->listen($socket);
        $this->logger->info(sprintf(
            "Listening on %s",
            str_replace("tcp://",($socket instanceof SecureServer?"https://":"http://"),$socket->getAddress())
        ));
    }

    /**
     * 
     * @return HttpServer
     */
    private function createHttpServer(): HttpServer
    {
        $stack = [
            $this->responseMiddleware = new ResponseMiddleware(
                config: $this->config,
                logger: $this->logger->withName("http"),
            ),
            $this->securityMiddleware = new SecurityMiddleware(
                config: $this->config
            ),
        ];
        if ($this->config->getEnableWebSockets()) {
            $stack = [ ...$stack, 
                $this->webSocketHandler   = new WebSocketHandler(
                    config: $this->config, 
                    webSocketClients: $this->webSocketClients, 
                    topicManager: $this->topicManager
                ),
            ];
            $this->logger->warning("The WebSocket support is incomplete and insecure, but enabling it as requested.");
        }
        $stack = [ ...$stack, 
            $this->mercureHandler     = new MercureHandler(
                config: $this->config,
                topicManager: $this->topicManager
            ),
            $this->apiRequestHandler  = new ApiHandler(
                config: $this->config,
                topicManager: $this->topicManager
            ),
            $this->notFoundHandler    = new NotFoundHandler()
        ];

        return new HttpServer(...$stack);
    }


}