New redirect strategy

config/module.config.php

@@ -16,14 +16,17 @@
         ),
     ),
     'controllers' => array(
-        'invokables' => array(
-            'zfcuser' => 'ZfcUser\Controller\UserController',
+        'factories' => array(
+            'zfcuser' => 'ZfcUser\Factory\Controller\UserControllerFactory',
         ),
     ),
     'service_manager' => array(
         'aliases' => array(
             'zfcuser_zend_db_adapter' => 'Zend\Db\Adapter\Adapter',
         ),
+        'factories' => array(
+            'zfcuser_redirect_callback' => 'ZfcUser\Factory\Controller\RedirectCallbackFactory'
+        )
     ),
     'router' => array(
         'routes' => array(

src/ZfcUser/Controller/RedirectCallback.php

@@ -0,0 +1,94 @@
+<?php
+
+namespace ZfcUser\Controller;
+
+use Zend\Mvc\Controller\Plugin\Redirect;
+use Zend\Mvc\Router\RouteInterface;
+use Zend\Mvc\Router\Exception;
+use Zend\Http\PhpEnvironment\Request;
+use Zend\Http\PhpEnvironment\Response;
+use ZfcUser\Options\ModuleOptions;
+
+class RedirectCallback
+{
+    /** @var RouteInterface */
+    protected $router;
+
+    /** @var Response */
+    protected $response;
+
+    /** @var Request */
+    protected $request;
+
+    /** @var ModuleOptions */
+    protected $options;
+
+    /**
+     * @param RouteInterface $router
+     * @param Response $response
+     * @param Request $request
+     * @param ModuleOptions $options
+     */
+    public function __construct(RouteInterface $router, Response $response, Request $request, ModuleOptions $options)
+    {
+        $this->router = $router;
+        $this->request = $request;
+        $this->response = $response;
+        $this->options = $options;
+    }
+
+    /**
+     * @return Response
+     */
+    public function __invoke()
+    {
+        $routeMatch = $this->router->match($this->request);
+        $response = $this->response;
+        $response->getHeaders()->addHeaderLine('Location', $this->getRedirect($routeMatch->getMatchedRouteName(), $routeMatch->getParam('redirect', false)));
+        $response->setStatusCode(302);
+        return $response;
+    }
+
+    /**
+     * @param $route
+     * @return bool
+     */
+    protected function routeExists($route)
+    {
+        try{
+            $this->router->assemble($route);
+        } catch (Exception\RuntimeException $e) {
+            return false;
+        }
+        return true;
+    }
+
+    /**
+     * Returns the url to redirect to based on current route.
+     * If $redirect is set and the option to use redirect is set to true, it will return the $redirect url.
+     *
+     * @param string $currentRoute
+     * @param bool $redirect
+     * @return mixed
+     */
+    protected function getRedirect($currentRoute, $redirect = false)
+    {
+        if (!$this->options->getUseRedirectParameterIfPresent() || ($redirect && !$this->routeExists($redirect))) {
+            $redirect = false;
+        }
+
+        switch ($currentRoute) {
+            case 'zfcuser/login':
+                $route = ($redirect) ?: $this->options->getLoginRedirectRoute();
+                return $this->router->assemble([], ['name' => $route]);
+                break;
+            case 'zfcuser/logout':
+                $route = ($redirect) ?: $this->options->getLogoutRedirectRoute();
+                return $this->router->assemble([], ['name' => $route]);
+                break;
+            default:
+                return $this->router->assemble([], ['name' => 'zfcuser']);
+        }
+    }
+
+}

src/ZfcUser/Controller/UserController.php

@@ -55,6 +55,16 @@ class UserController extends AbstractActionController
      */
     protected $options;
 
+    /**
+     * @var Callable $redirectCallback
+     */
+    protected $redirectCallback;
+
+    public function __construct($redirectCallback)
+    {
+        $this->redirectCallback = $redirectCallback;
+    }
+
     /**
      * User page
      */
@@ -116,13 +126,8 @@ public function logoutAction()
         $this->zfcUserAuthentication()->getAuthAdapter()->logoutAdapters();
         $this->zfcUserAuthentication()->getAuthService()->clearIdentity();
 
-        $redirect = $this->params()->fromPost('redirect', $this->params()->fromQuery('redirect', false));
-
-        if ($this->getOptions()->getUseRedirectParameterIfPresent() && $redirect) {
-            return $this->redirect()->toRoute($redirect);
-        }
-
-        return $this->redirect()->toRoute($this->getOptions()->getLogoutRedirectRoute());
+        $redirect = $this->redirectCallback;
+        return $redirect();
     }
 
     /**
@@ -165,7 +170,8 @@ public function authenticateAction()
             $route = $route($this->zfcUserAuthentication()->getIdentity());
         }
 
-        return $this->redirect()->toRoute($route);
+        $redirect = $this->redirectCallback;
+        return $redirect();
     }
 
     /**

src/ZfcUser/Factory/Controller/RedirectCallbackFactory.php

@@ -0,0 +1,21 @@
+<?php
+namespace ZfcUser\Factory\Controller;
+
+use Zend\ServiceManager\FactoryInterface;
+use Zend\ServiceManager\ServiceLocatorInterface;
+use ZfcUser\Controller\RedirectCallback;
+
+class RedirectCallbackFactory implements FactoryInterface
+{
+    /**
+     * {@inheritDoc}
+     */
+    public function createService(ServiceLocatorInterface $serviceLocator)
+    {
+        $router = $serviceLocator->get('Router');
+        $response = $serviceLocator->get('Response');
+        $request = $serviceLocator->get('Request');
+        $options = $serviceLocator->get('zfcuser_module_options');
+        return new RedirectCallback($router, $response, $request, $options);
+    }
+}

src/ZfcUser/Factory/Controller/UserControllerFactory.php

@@ -0,0 +1,25 @@
+<?php
+namespace ZfcUser\Factory\Controller;
+
+use Zend\Mvc\Controller\ControllerManager;
+use Zend\ServiceManager\FactoryInterface;
+use Zend\ServiceManager\ServiceLocatorInterface;
+use ZfcUser\Authentication\Adapter;
+use ZfcUser\Controller\UserController;
+
+class UserControllerFactory implements FactoryInterface
+{
+    /**
+     * {@inheritDoc}
+     */
+    public function createService(ServiceLocatorInterface $controllerManager)
+    {
+        /** @var ControllerManager $controllerManager*/
+        $serviceManager = $controllerManager->getServiceLocator();
+
+        $redirectCallback = $serviceManager->get('zfcuser_redirect_callback');
+        $controller = new UserController($redirectCallback);
+
+        return $controller;
+    }
+}