Return 404 when no $next callable is passed in

Author: WyriHaximus

src/WebSocketMiddleware.php

@@ -22,11 +22,15 @@ public function __construct(array $paths = [], callable $connectionHandler = nul
         $this->subProtocols      = $subProtocols;
     }
 
-    public function __invoke(ServerRequestInterface $request, callable $next)
+    public function __invoke(ServerRequestInterface $request, callable $next = null)
     {
         // check path at some point - for now we just go go ws
         if (count($this->paths) > 0) {
             if (!in_array($request->getUri()->getPath(), $this->paths)) {
+                if ($next === null) {
+                    return new Response(404);
+                }
+
                 return $next($request);
             }
         }
@@ -38,6 +42,10 @@ public function __invoke(ServerRequestInterface $request, callable $next)
         $response = $negotiator->handshake($request);
 
         if ($response->getStatusCode() != '101') {
+            if ($next === null) {
+                return new Response(404);
+            }
+
             // TODO: this should return an error or something not continue the chain
             return $next($request);
         }