basic add headers to response (#13)

mattvb91 · Sep 23, 2022 · 5fb4a46 · 5fb4a46
1 parent cc3603c
commit 5fb4a46
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/src/Config/Apps/Http/Server/Routes/Handle/Headers/Response.php b/src/Config/Apps/Http/Server/Routes/Handle/Headers/Response.php
@@ -11,6 +11,8 @@ class Response implements Arrayable
      */
     private ?array $_delete;
 
+    private ?array $_add;
+
     private bool $_deferred = false;
 
     public function addDeleteHeader(string $header): static
@@ -24,6 +26,13 @@ public function addDeleteHeader(string $header): static
         return $this;
     }
 
+    public function addHeader(string $name, string $value): static
+    {
+        $this->_add[$name] = [$value];
+
+        return $this;
+    }
+
     public function setDeferred(bool $deferred): static
     {
         $this->_deferred = $deferred;
@@ -39,6 +48,10 @@ public function toArray(): array
             $array['delete'] = $this->_delete;
         }
 
+        if (isset($this->_add)) {
+            $array['add'] = $this->_add;
+        }
+
         $array['deferred'] = $this->_deferred;
 
         return $array;

diff --git a/tests/Integration/Apps/Http/HandlerTest.php b/tests/Integration/Apps/Http/HandlerTest.php
@@ -100,4 +100,38 @@ public function test_static_file_server_handler()
         $request = $client->request('GET', 'caddy/static.txt');
         $this->assertEquals(200, $request->getStatusCode());
     }
+
+    /**
+     * @coversNothing
+     */
+    public function test_header_response()
+    {
+        $caddy = new Caddy();
+        $caddy->addApp((new Http())
+            ->addServer('staticFileServer', (new Http\Server())
+                ->addRoute((new Http\Server\Route())
+                    ->addHandle((new Http\Server\Routes\Handle\Headers())
+                        ->setResponse((new Http\Server\Routes\Handle\Headers\Response())
+                            ->addHeader('Cache-Status', 'public, max-age=3600')
+                        ))->addHandle((new Http\Server\Routes\Handle\FileServer())
+                        ->setRoot('/var/files')
+                    )
+                )
+            )
+        );
+
+        $this->assertCaddyConfigLoaded($caddy);
+
+        $client = new Client([
+            'base_uri'    => 'caddy',
+            'http_errors' => false,
+            'headers'     => [
+                'Host' => 'localhost',
+            ],
+        ]);
+
+        $request = $client->request('GET', 'caddy/static.txt');
+        $this->assertEquals(200, $request->getStatusCode());
+        $this->assertEquals('public, max-age=3600', $request->getHeader('Cache-Status')[0]);
+    }
 }