configurable response factory

Author: dakujem

src/JwtAuthentication.php

@@ -46,7 +46,6 @@
 use Psr\Log\LoggerInterface;
 use Psr\Log\LogLevel;
 use RuntimeException;
-use Tuupola\Middleware\DoublePassTrait;
 use Tuupola\Http\Factory\ResponseFactory;
 use Tuupola\Middleware\JwtAuthentication\RequestMethodRule;
 use Tuupola\Middleware\JwtAuthentication\RequestPathRule;
@@ -85,14 +84,20 @@ final class JwtAuthentication implements MiddlewareInterface
         "ignore" => null,
         "before" => null,
         "after" => null,
-        "error" => null
+        "error" => null,
+        "responseFactory" => null,
     ];
 
     public function __construct(array $options = [])
     {
         /* Setup stack for rules */
         $this->rules = new \SplStack;
 
+        /* Make sure the response factory option is provided a default */
+        if (!isset($options['responseFactory'])) {
+            $options['responseFactory'] = null;
+        }
+
         /* Store passed in options overwriting any defaults. */
         $this->hydrate($options);
 
@@ -139,7 +144,7 @@ public function process(ServerRequestInterface $request, RequestHandlerInterface
             $token = $this->fetchToken($request);
             $decoded = $this->decodeToken($token);
         } catch (RuntimeException | DomainException $exception) {
-            $response = (new ResponseFactory)->createResponse(401);
+            $response = call_user_func($this->options['responseFactory'], 401);
             return $this->processError($response, [
                 "message" => $exception->getMessage(),
                 "uri" => (string)$request->getUri()
@@ -158,7 +163,6 @@ public function process(ServerRequestInterface $request, RequestHandlerInterface
 
         /* Modify $request before calling next middleware. */
         if (is_callable($this->options["before"])) {
-            $response = (new ResponseFactory)->createResponse(200);
             $beforeRequest = $this->options["before"]($request, $params);
             if ($beforeRequest instanceof ServerRequestInterface) {
                 $request = $beforeRequest;
@@ -452,4 +456,21 @@ private function rules(array $rules): void
             $this->rules->push($callable);
         }
     }
+
+    /**
+     * Set the response factory.
+     */
+    private function responseFactory(callable $factory = null): void
+    {
+        if ($factory === null) {
+            $factory = function ($code, ...$args) {
+                return (new ResponseFactory)->createResponse($code, ...$args);
+            };
+        }
+        if ($factory instanceof Closure) {
+            $this->options["responseFactory"] = $factory->bindTo($this);
+        } else {
+            $this->options["responseFactory"] = $factory;
+        }
+    }
 }