pixelfederation/swoole-bundle

<?php

declare(strict_types=1);

namespace K911\Swoole\Server\Api;

use K911\Swoole\Client\Http;
use K911\Swoole\Server\RequestHandler\RequestHandlerInterface;
use Swoole\Http\Request;
use Swoole\Http\Response;

final class ApiServerRequestHandler implements RequestHandlerInterface
{
    private const SUPPORTED_HTTP_METHODS = [
        Http::METHOD_HEAD,
        Http::METHOD_GET,
        Http::METHOD_POST,
        Http::METHOD_PATCH,
        Http::METHOD_DELETE,
    ];
    private $routes;

    public function __construct(ApiServerInterface $apiServer)
    {
        $this->routes = [
            '/healthz' => [
                Http::METHOD_GET => $this->composeSimpleRouteDefinition(200, fn (): array => ['ok' => true]),
            ],
            '/api' => [
                Http::METHOD_GET => $this->composeSimpleRouteDefinition(200, [$this, 'getRouteMap']),
            ],
            '/api/server' => [
                Http::METHOD_GET => $this->composeSimpleRouteDefinition(200, [$apiServer, 'status']),
                Http::METHOD_PATCH => $this->composeSimpleRouteDefinition(204, [$apiServer, 'reload']),
                Http::METHOD_DELETE => $this->composeSimpleRouteDefinition(204, [$apiServer, 'shutdown']),
            ],
            '/api/server/metrics' => [
                Http::METHOD_GET => $this->composeSimpleRouteDefinition(200, [$apiServer, 'metrics']),
            ],
        ];
    }

    /**
     * {@inheritdoc}
     *
     * @throws \Exception
     */
    public function handle(Request $request, Response $response): void
    {
        try {
            [$method] = $this->parseRequestInfo($request);
            switch ($method) {
                case Http::METHOD_HEAD:
                    $request->server['request_method'] = Http::METHOD_GET;
                    $this->sendResponse($response, $this->handleRequest($request)[0]);

                    break;
                case Http::METHOD_GET:
                case Http::METHOD_POST:
                case Http::METHOD_PATCH:
                case Http::METHOD_DELETE:
                    [$statusCode, $data] = $this->handleRequest($request);
                    $this->sendResponse($response, $statusCode, $data);

                    return;
                default:
                    $this->sendResponse($response, 405, [
                        'error' => sprintf('Method "%s" is not supported. Supported ones are: %s.', $method, implode(', ', self::SUPPORTED_HTTP_METHODS)),
                    ]);

                    return;
            }
        } catch (\Throwable $exception) {
            $this->sendErrorExceptionResponse($response, $exception);
        }
    }

    private function composeSimpleRouteDefinition(int $code, callable $handler): array
    {
        return [
            'code' => $code,
            'handler' => $handler,
        ];
    }

    private function sendErrorExceptionResponse(Response $response, \Throwable $exception): void
    {
        $this->sendResponse($response, 500, [
            'error' => $exception->getMessage(),
            'code' => $exception->getCode(),
            'line' => $exception->getLine(),
            'file' => $exception->getFile(),
            'trace' => explode("\n", $exception->getTraceAsString()),
        ]);
    }

    private function parseRequestInfo(Request $request): array
    {
        $method = mb_strtoupper($request->server['request_method']);
        $path = mb_strtolower(rtrim($request->server['path_info'], '/'));
        $path = '' === $path ? '/' : $path;

        return [$method, $path];
    }

    private function handleRequest(Request $request): array
    {
        [$method, $path] = $this->parseRequestInfo($request);

        if (\array_key_exists($path, $this->routes)) {
            $route = $this->routes[$path];
            if (\array_key_exists($method, $route)) {
                $action = $route[$method];

                return [$action['code'], $action['handler']($request)];
            }

            return [405, [
                'error' => sprintf('Method %s for route %s is not valid. Supported ones are: %s.', $method, $path, implode(', ', array_keys($route))),
            ]];
        }

        return [404, [
            'error' => sprintf('Route %s does not exists.', $path),
            'routes' => $this->getRouteMap(),
        ]];
    }

    private function getRouteMap(): array
    {
        return array_map(fn (array $route): array => array_keys($route), $this->routes);
    }

    private function sendResponse(Response $response, int $statusCode = 200, ?array $data = []): void
    {
        if (empty($data) || 204 === $statusCode) {
            $response->status(200 === $statusCode ? 204 : $statusCode);
            $response->end();

            return;
        }

        $response->header(Http::HEADER_CONTENT_TYPE, Http::CONTENT_TYPE_APPLICATION_JSON);
        $response->status($statusCode);
        $response->end(json_encode($data, \JSON_THROW_ON_ERROR));
    }
}