SubprotocolHandler prototype

examples/echoserver.php

@@ -31,6 +31,7 @@
     $server
         ->addMiddleware(new \WebSocket\Middleware\CloseHandler())
         ->addMiddleware(new \WebSocket\Middleware\PingResponder())
+        ->addMiddleware(new \WebSocket\Middleware\SubprotocolHandler(['soap', 'towel']))
         ;
 
     // If debug mode and logger is available

examples/send.php

@@ -33,6 +33,7 @@
     $client
         ->addMiddleware(new \WebSocket\Middleware\CloseHandler())
         ->addMiddleware(new \WebSocket\Middleware\PingResponder())
+        ->addMiddleware(new \WebSocket\Middleware\SubprotocolHandler(['towel', 'soap']))
         ->onText(function ($client, $connection, $message) {
             echo "> Received '{$message->getContent()}' [opcode: {$message->getOpcode()}]\n";
             echo "< Closing client\n";

src/Http/HttpHandler.php

@@ -57,6 +57,7 @@ public function pull(): MessageInterface
             $buffer = $this->stream->readLine(1024);
             $data .= $buffer;
         } while (substr_count($data, "\r\n\r\n") == 0);
+echo "---------- PULL: $data \n";
 
         list ($head, $body) = explode("\r\n\r\n", $data);
         $headers = array_filter(explode("\r\n", $head));
@@ -85,7 +86,7 @@ public function pull(): MessageInterface
         foreach ($headers as $header) {
             $parts = explode(':', $header, 2);
             if (count($parts) == 2) {
-                $message = $message->withHeader($parts[0], $parts[1]);
+                $message = $message->withAddedHeader($parts[0], $parts[1]);
             }
         }
         if ($message instanceof Request) {
@@ -99,6 +100,7 @@ public function pull(): MessageInterface
     public function push(MessageInterface $message): MessageInterface
     {
         $data = implode("\r\n", $message->getAsArray()) . "\r\n\r\n";
+echo "---------- PUSH: $data \n";
         $this->stream->write($data);
         return $message;
     }

src/Middleware/Callback.php

@@ -16,10 +16,7 @@
 };
 use Stringable;
 use WebSocket\Connection;
-use WebSocket\Http\{
-    HttpHandler,
-    Message as HttpMessage
-};
+use WebSocket\Http\Message as HttpMessage;
 use WebSocket\Message\Message;
 use WebSocket\Trait\StringableTrait;
 

src/Middleware/SubprotocolHandler.php

@@ -0,0 +1,71 @@
+<?php
+
+/**
+ * Copyright (C) 2014-2023 Textalk and contributors.
+ *
+ * This file is part of Websocket PHP and is free software under the ISC License.
+ * License text: https://raw.githubusercontent.com/sirn-se/websocket-php/master/COPYING.md
+ */
+
+namespace WebSocket\Middleware;
+
+use Psr\Log\{
+    LoggerAwareInterface,
+    LoggerAwareTrait
+};
+use WebSocket\Connection;
+use WebSocket\Http\Message;
+use WebSocket\Trait\StringableTrait;
+
+/**
+ * WebSocket\Middleware\CloseHandler class.
+ * Handles close procedure.
+ */
+class SubprotocolHandler implements LoggerAwareInterface, ProcessHttpOutgoingInterface, ProcessHttpIncomingInterface
+{
+    use LoggerAwareTrait;
+    use StringableTrait;
+
+    private $subprotocols;
+    private $selected = null;
+
+    public function __construct(array $subprotocols)
+    {
+        $this->subprotocols = $subprotocols;
+    }
+
+    public function processHttpOutgoing(ProcessHttpStack $stack, Connection $connection, Message $message): Message
+    {
+        if ($message instanceof \WebSocket\Http\Request) {
+            // Outgoing requests on Client
+            foreach ($this->subprotocols as $subprotocol) {
+                $message = $message->withAddedHeader('Sec-WebSocket-Protocol', $subprotocol);
+            }
+        } elseif ($message instanceof \WebSocket\Http\Response) {
+            // Outgoing Response
+            if ($this->selected) {
+                $message = $message->withHeader('Sec-WebSocket-Protocol', $this->selected);
+            }
+        }
+
+        return $stack->handleHttpOutgoing($message);
+    }
+
+    public function processHttpIncoming(ProcessHttpStack $stack, Connection $connection): Message
+    {
+        $this->selected = null;
+        $message = $stack->handleHttpIncoming();
+
+        if ($message instanceof \WebSocket\Http\ServerRequest) {
+            // Incoming requests on Server
+            foreach ($message->getHeader('Sec-WebSocket-Protocol') as $subprotocol) {
+                if (in_array($subprotocol, $this->subprotocols)) {
+                    $this->selected = $subprotocol;
+                    return $message;
+                }
+            }
+        }
+
+        return $message;
+    }
+}