Updated to store headers

Author: qschmick

src/DTOs/PushPullBatchJobResponse.php

@@ -2,11 +2,14 @@
 
 namespace AlwaysOpen\OxylabsApi\DTOs;
 
+use AlwaysOpen\OxylabsApi\Traits\Headers;
 use Spatie\LaravelData\Attributes\DataCollectionOf;
 use Spatie\LaravelData\Data;
 
 class PushPullBatchJobResponse extends Data
 {
+    use Headers;
+
     public function __construct(
         /* @var PushPullJobResponse[] $queries */
         #[DataCollectionOf(PushPullJobResponse::class)]

src/DTOs/PushPullJob.php

@@ -2,6 +2,7 @@
 
 namespace AlwaysOpen\OxylabsApi\DTOs;
 
+use AlwaysOpen\OxylabsApi\Traits\Headers;
 use Illuminate\Support\Carbon;
 use Spatie\LaravelData\Attributes\DataCollectionOf;
 use Spatie\LaravelData\Attributes\WithCast;
@@ -10,6 +11,8 @@
 
 class PushPullJob extends Data
 {
+    use Headers;
+
     public function __construct(
         #[WithCast(DateTimeInterfaceCast::class, format: ['Y-m-d H:i:s', 'Y-m-d\TH:i:s\+H:i', 'Y-m-d H:i:s.u'])]
         public readonly Carbon $created_at,

src/OxylabsApiClient.php

@@ -101,7 +101,7 @@ public function makePostRequest(
                 throw new RuntimeException('API request failed: '.$response->body(), $response->getStatusCode());
             }
 
-            return PushPullJob::from($response->json());
+            return PushPullJob::from($response->json())->setHeaders($response->getHeaders());
         } catch (Throwable $e) {
             throw new RuntimeException('API request failed: '.$e->getMessage(), $e->getCode(), $e);
         }
@@ -342,7 +342,7 @@ public function makeBatchRequest(
             throw new RuntimeException('API request failed: '.$response->body());
         }
 
-        return PushPullBatchJobResponse::from($response->json());
+        return PushPullBatchJobResponse::from($response->json())->setHeaders($response->getHeaders());
     }
 
     /**

src/Traits/Headers.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace AlwaysOpen\OxylabsApi\Traits;
+
+trait Headers
+{
+    protected ?array $__headers = null;
+
+    public function getHeaderValue(string $key): mixed
+    {
+        if ($this->__headers) {
+            return $this->__headers[$key] ?? null;
+        }
+
+        return null;
+    }
+
+    public function getHeaders(): ?array
+    {
+        return $this->__headers;
+    }
+
+    public function setHeaders(?array $headers = null): self
+    {
+        $this->__headers = $headers;
+
+        return $this;
+    }
+}