Fix/Feature : Fixing ability to add custom handlers to a route by adding a custom routes method to the router (#150)

Fixing Coding Standards

Fixing Coding Standards

Author: lukepolo

src/Console/StartWebSocketServer.php

@@ -44,6 +44,7 @@ public function handle()
             ->configureMessageLogger()
             ->configureConnectionLogger()
             ->registerEchoRoutes()
+            ->registerCustomRoutes()
             ->startWebSocketServer();
     }
 
@@ -110,6 +111,13 @@ protected function registerEchoRoutes()
         return $this;
     }
 
+    protected function registerCustomRoutes()
+    {
+        WebSocketsRouter::customRoutes();
+
+        return $this;
+    }
+
     protected function startWebSocketServer()
     {
         $this->info("Starting the WebSocket server on port {$this->option('port')}...");

src/Server/Router.php

@@ -3,6 +3,7 @@
 namespace BeyondCode\LaravelWebSockets\Server;
 
 use Ratchet\WebSocket\WsServer;
+use Illuminate\Support\Collection;
 use Symfony\Component\Routing\Route;
 use Symfony\Component\Routing\RouteCollection;
 use Ratchet\WebSocket\MessageComponentInterface;
@@ -18,10 +19,12 @@ class Router
 {
     /** @var \Symfony\Component\Routing\RouteCollection */
     protected $routes;
+    protected $customRoutes;
 
     public function __construct()
     {
         $this->routes = new RouteCollection;
+        $this->customRoutes = new Collection();
     }
 
     public function getRoutes(): RouteCollection
@@ -39,6 +42,13 @@ public function echo()
         $this->get('/apps/{appId}/channels/{channelName}/users', FetchUsersController::class);
     }
 
+    public function customRoutes()
+    {
+        $this->customRoutes->each(function ($action, $uri) {
+            $this->get($uri, $action);
+        });
+    }
+
     public function get(string $uri, $action)
     {
         $this->addRoute('GET', $uri, $action);
@@ -70,7 +80,7 @@ public function webSocket(string $uri, $action)
             throw InvalidWebSocketController::withController($action);
         }
 
-        $this->get($uri, $action);
+        $this->customRoutes->put($uri, $action);
     }
 
     public function addRoute(string $method, string $uri, $action)