[5.x] Add --headers option to static warm command

Author: ChristianPraiss

src/Console/Commands/StaticWarm.php

@@ -42,6 +42,7 @@ class StaticWarm extends Command
         {--include= : Only warm specific URLs}
         {--exclude= : Exclude specific URLs}
         {--max-requests= : Maximum number of requests to warm}
+        {--headers= : Set custom headers (e.g. --headers="Authorization: Bearer your_token", can be used multiple times)}
     ';
 
     protected $description = 'Warms the static cache by visiting all URLs';
@@ -167,8 +168,9 @@ private function getRelativeUri(int $index): string
 
     private function requests()
     {
-        return $this->uris()->map(function ($uri) {
-            return new Request('GET', $uri);
+        $headers = $this->parseHeaders($this->option('headers'));
+        return $this->uris()->map(function ($uri) use ($headers) {
+            return new Request('GET', $uri, $headers);
         })->all();
     }
 
@@ -374,4 +376,24 @@ protected function additionalUris(): Collection
 
         return $uris->map(fn ($uri) => URL::makeAbsolute($uri));
     }
+
+    private function parseHeaders($headerOptions): array
+    {
+        $headers = [];
+        if (empty($headerOptions)) {
+            return $headers;
+        }
+        if (!is_array($headerOptions)) {
+            $headerOptions = [$headerOptions];
+        }
+        foreach ($headerOptions as $header) {
+            if (strpos($header, ':') !== false) {
+                [$key, $value] = explode(':', $header, 2);
+                $headers[trim($key)] = trim($value);
+            } else {
+                $this->line("<fg=yellow;options=bold>Warning:</> Invalid header format: '$header'. Headers should be in 'Key: Value' format.");
+            }
+        }
+        return $headers;
+    }
 }

tests/Console/Commands/StaticWarmTest.php

@@ -271,6 +271,24 @@ public static function queueConnectionsProvider()
         ];
     }
 
+    #[Test]
+    public function it_sets_custom_headers_on_requests()
+    {
+        config(['statamic.static_caching.strategy' => 'half']);
+
+        $mock = Mockery::mock(\GuzzleHttp\Client::class);
+        $mock->shouldReceive('send')->andReturnUsing(function ($request) {
+            $this->assertEquals('Bearer testtoken', $request->getHeaderLine('Authorization'));
+
+            return Mockery::mock(\GuzzleHttp\Psr7\Response::class);
+        });
+        $this->app->instance(\GuzzleHttp\Client::class, $mock);
+
+        $this->artisan('statamic:static:warm', [
+            '--headers' => ['Authorization: Bearer testtoken'],
+        ])->assertExitCode(0);
+    }
+
     private function createPage($slug, $attributes = [])
     {
         $this->makeCollection()->save();