Cancel requests

Author: lennardv2

src/ApiClient.php

@@ -49,6 +49,10 @@ public function addResultHandler(callable $handler)
      * @return int The length of the handled data.
      */
     protected function handleStream($curl, $streamData) {
+        if ($this->shouldAbort) {
+            return -1;
+        }
+
         foreach(explode("data: ", $streamData) as $bodyParsed) {
             if (!$bodyParsed) continue;
 

src/Client.php

@@ -106,6 +106,16 @@ public function combineStreamMessageToResultMessage() : ChatMessage
         return $message;
     }
 
+    public function cancelRequest()
+    {
+        $this->driver->cancelRequest();
+    }
+
+    public function isBusy() : bool
+    {
+        return $this->driver->isBusy();
+    }
+
     /**
      * Sets the system message in the chat history.
      *

src/Traits/WithCurlRequests.php

@@ -4,6 +4,9 @@
 
 trait WithCurlRequests
 {
+    public bool $shouldAbort = false;
+    public bool $isBusy = false;
+
     /**
      * Sends a JSON request to a specified URL.
      *
@@ -14,6 +17,8 @@ trait WithCurlRequests
      * @return array|null The parsed JSON response, or null on failure.
      */
     protected function sendJsonRequest(string $method, string $url, array $jsonData, array $headers = []) {
+        $this->isBusy = true;
+
         $curlOptions = [
             CURLOPT_URL => $url,
             CURLOPT_RETURNTRANSFER => true,
@@ -25,6 +30,11 @@ protected function sendJsonRequest(string $method, string $url, array $jsonData,
             CURLOPT_CUSTOMREQUEST => $method,
             CURLOPT_POSTFIELDS => json_encode($jsonData),
             CURLOPT_HTTPHEADER => $headers,
+            CURLOPT_WRITEFUNCTION => function ($curl, $streamData) {
+                if ($this->shouldAbort) {
+                    return -1;
+                }
+            }
         ];
 
         if ($jsonData === []) {
@@ -52,6 +62,18 @@ protected function sendJsonRequest(string $method, string $url, array $jsonData,
             $bodyParsed = null;
         }
 
+        $this->isBusy = false;
+
         return $bodyParsed;
     }
+
+    public function cancelRequest()
+    {
+        $this->shouldAbort = true;
+    }
+
+    public function isBusy() : bool
+    {
+        return $this->isBusy;
+    }
 }