From fd37c9d60ea818dc9a562fa88ff5f9a50132506f Mon Sep 17 00:00:00 2001
From: Igor Scheller <igor.scheller@igorshp.de>
Date: Sat, 27 Oct 2018 11:11:37 +0100
Subject: Added additional headers

---
 src/Middleware/AddHeaders.php | 46 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 src/Middleware/AddHeaders.php

(limited to 'src/Middleware/AddHeaders.php')

diff --git a/src/Middleware/AddHeaders.php b/src/Middleware/AddHeaders.php
new file mode 100644
index 00000000..75383a90
--- /dev/null
+++ b/src/Middleware/AddHeaders.php
@@ -0,0 +1,46 @@
+<?php
+
+namespace Engelsystem\Middleware;
+
+use Engelsystem\Config\Config;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Server\MiddlewareInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+
+class AddHeaders implements MiddlewareInterface
+{
+    /** @var Config */
+    protected $config;
+
+    /**
+     * @param Config $config
+     */
+    public function __construct(Config $config)
+    {
+        $this->config = $config;
+    }
+
+    /**
+     * Process an incoming server request and setting the locale if required
+     *
+     * @param ServerRequestInterface  $request
+     * @param RequestHandlerInterface $handler
+     * @return ResponseInterface
+     */
+    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
+    {
+        $response = $handler->handle($request);
+        if (!$this->config->get('add_headers', true)) {
+            return $response;
+        }
+
+        $headers = $this->config->get('headers', []);
+
+        foreach ($headers as $name => $value) {
+            $response = $response->withAddedHeader($name, $value);
+        }
+
+        return $response;
+    }
+}
-- 
cgit v1.2.3-54-g00ecf