laravel-websockets/src/Server/Router.php

<?php

namespace BeyondCode\LaravelWebSockets\Server;

use BeyondCode\LaravelWebSockets\LaravelEcho;
use BeyondCode\LaravelWebSockets\Server\Logger\MessageLogger;
use BeyondCode\LaravelWebSockets\WebSockets\Controllers\FetchChannel;
use BeyondCode\LaravelWebSockets\WebSockets\Controllers\FetchChannels;
use BeyondCode\LaravelWebSockets\WebSockets\Controllers\FetchUsers;
use BeyondCode\LaravelWebSockets\WebSockets\Controllers\TriggerEvent;
use BeyondCode\LaravelWebSockets\WebSockets\Pusher\PusherController;
use Ratchet\WebSocket\WsServer;
use Symfony\Component\Routing\Route;
use Ratchet\Http\HttpServerInterface;
use Symfony\Component\Routing\RouteCollection;
use BeyondCode\LaravelWebSockets\Exceptions\InvalidWebSocketController;

class Router
{
    /** @var RouteCollection */
    protected $routes;

    public function __construct()
    {
        $this->routes = new RouteCollection;
    }

    public function websocket(string $uri, $action)
    {
        if (!is_subclass_of($action, WebSocketController::class)) {
            throw InvalidWebSocketController::withController($action);
        }

        $this->get($uri, $action);
    }

    public function get(string $uri, $action)
    {
        $this->addRoute('GET', $uri, $action);
    }

    public function post(string $uri, $action)
    {
        $this->addRoute('POST', $uri, $action);
    }

    public function put(string $uri, $action)
    {
        $this->addRoute('PUT', $uri, $action);
    }

    public function patch(string $uri, $action)
    {
        $this->addRoute('PATCH', $uri, $action);
    }

    public function delete(string $uri, $action)
    {
        $this->addRoute('DELETE', $uri, $action);
    }

    public function addRoute(string $method, string $uri, $action)
    {
        $this->routes->add($uri, $this->getRoute($method, $uri, $action));
    }

    protected function getRoute(string $method, string $uri, $action): Route
    {
        return new Route($uri, ['_controller' => $this->wrapAction($action)], [], [], null, [], [$method]);
    }

    public function echo()
    {
        $this->get('/app/{appKey}', PusherController::class);

        $this->get('/apps/{appId}/channels', FetchChannels::class);
        $this->get('/apps/{appId}/channels/{channelName}', FetchChannel::class);
        $this->get('/apps/{appId}/channels/{channelName}/users', FetchUsers::class);

        $this->post('/apps/{appId}/events', TriggerEvent::class);
    }

    /**
     * @param $action
     * @return WsServer|HttpServerInterface
     */
    protected function wrapAction($action)
    {
        if (is_subclass_of($action, WebSocketController::class)) {
            $app = app($action);

            if (MessageLogger::isEnabled()) {
                $app = MessageLogger::decorate($app);
            }

            return new WsServer($app);
        }

        return app($action);
    }

    public function getRoutes(): RouteCollection
    {
        return $this->routes;
    }
}