Improve websocket connection

Author: AlboCode

src/CCatClient.php

@@ -39,18 +39,16 @@ public function sendMessage(Message $message, ?\Closure $closure = null): Respon
 
             $response = $client->receive();
             $message = $response->getContent();
-            if (str_contains($message, "\"type\":\"notification\"") || str_contains($message, "\"type\":\"chat_token\"")) {
+            if (!str_contains($message, "\"type\":\"chat\"")) {
                 $closure?->call($this, $message);
                 continue;
             }
-            if (str_contains($message, "\"type\":\"chat\"")) {
-                break;
-            }
-
-
+            break;
 
         }
 
+        $client->disconnect();
+
         return $this->jsonToResponse($message);
     }
 

src/Clients/WSClient.php

@@ -4,6 +4,8 @@
 
 use Phrity\Net\Uri;
 use WebSocket\Client;
+use WebSocket\Middleware\CloseHandler;
+use WebSocket\Middleware\PingResponder;
 
 class WSClient
 {
@@ -34,8 +36,14 @@ public function getWsClient(string $userid = 'user'): Client
             ->withPath(sprintf('ws/%s', $userid))
             ->withPort($this->port)
         ;
-        return new Client($wsUri);
-
+        $client = new Client($wsUri);
+        $client->setPersistent(true)
+            ->setTimeout(100000)
+            // Add CloseHandler middleware
+            ->addMiddleware(new CloseHandler())
+            // Add PingResponder middleware
+            ->addMiddleware(new PingResponder());
+        return $client;
     }