laravel-websockets/src/Router.php

100 lines
2.6 KiB
PHP

<?php
namespace BeyondCode\LaravelWebSockets;
use BeyondCode\LaravelWebSockets\Server\Logger;
use Ratchet\WebSocket\WsServer;
use Symfony\Component\Routing\Route;
use Ratchet\Http\HttpServerInterface;
use Symfony\Component\Routing\RouteCollection;
use Ratchet\WebSocket\MessageComponentInterface;
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}', LaravelEcho\WebSocket\PusherServer::class);
$this->get('/apps/{appId}/channels', LaravelEcho\Http\Controllers\FetchChannels::class);
$this->get('/apps/{appId}/channels/{channelName}', LaravelEcho\Http\Controllers\FetchChannel::class);
$this->get('/apps/{appId}/channels/{channelName}/users', LaravelEcho\Http\Controllers\FetchUsers::class);
$this->post('/apps/{appId}/events', LaravelEcho\Http\Controllers\TriggerEvent::class);
}
/**
* @param $action
* @return WsServer|HttpServerInterface
*/
protected function wrapAction($action)
{
if (is_subclass_of($action, WebSocketController::class)) {
$app = app($action);
$appThatLogs = new Logger($app);
return new WsServer($appThatLogs);
}
return app($action);
}
public function getRoutes(): RouteCollection
{
return $this->routes;
}
}