From f47f263fc3f5ae6747c35b5050f22ab3d2a59581 Mon Sep 17 00:00:00 2001
From: Hypolite Petovan <hypolite@mrpetovan.com>
Date: Sat, 13 Jun 2020 13:16:35 -0400
Subject: [PATCH] Add ZrlMiddleware class

---
 src/classes/Middleware/ZrlMiddleware.php | 40 ++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 src/classes/Middleware/ZrlMiddleware.php

diff --git a/src/classes/Middleware/ZrlMiddleware.php b/src/classes/Middleware/ZrlMiddleware.php
new file mode 100644
index 0000000..e880481
--- /dev/null
+++ b/src/classes/Middleware/ZrlMiddleware.php
@@ -0,0 +1,40 @@
+<?php
+
+namespace Friendica\Directory\Middleware;
+
+use Friendica\Directory\Views\PhpRenderer;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+
+/**
+ * The ZRL middleware ensures the Renderer ZRL attribute is set if the query string parameter is present
+ *
+ * @author Hypolite Petovan <hypolite@mrpetovan.com>
+ * @package Friendica\Directory\Middleware
+ */
+class ZrlMiddleware
+{
+	/**
+	 * @var PhpRenderer
+	 */
+	private $phpRenderer;
+
+	public function __construct(PhpRenderer $phpRenderer)
+	{
+		$this->phpRenderer = $phpRenderer;
+	}
+
+	/**
+	 * @param ServerRequestInterface $request  PSR7 request
+	 * @param ResponseInterface      $response PSR7 response
+	 * @param callable               $next     Next middleware
+	 *
+	 * @return ResponseInterface
+	 */
+	public function __invoke(ServerRequestInterface $request, ResponseInterface $response, callable $next)
+	{
+		$this->phpRenderer->addAttribute('zrl', $request->getQueryParams()['zrl'] ?? null);
+
+		return $next($request, $response);
+	}
+}