Added constants for redirection values

[github/wallabag/wallabag.git] / src / Wallabag / CoreBundle / Helper / Redirect.php

diff --git a/src/Wallabag/CoreBundle/Helper/Redirect.php b/src/Wallabag/CoreBundle/Helper/Redirect.php
index c14c79d11bd1f6a46312160174c177ae0fca3e14..59172db40110d628a223a750af6a88fc718ded36 100644 (file)
--- a/src/Wallabag/CoreBundle/Helper/Redirect.php
+++ b/src/Wallabag/CoreBundle/Helper/Redirect.php
@@ -3,6 +3,8 @@
 namespace Wallabag\CoreBundle\Helper;
 
 use Symfony\Component\Routing\Router;
+use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorage;
+use Wallabag\CoreBundle\Entity\Config;
 
 /**
  * Manage redirections to avoid redirecting to empty routes.
@@ -10,10 +12,12 @@ use Symfony\Component\Routing\Router;
 class Redirect
 {
     private $router;
+    private $actionMarkAsRead;
 
-    public function __construct(Router $router)
+    public function __construct(Router $router, TokenStorage $token)
     {
         $this->router = $router;
+        $this->actionMarkAsRead = $token->getToken()->getUser()->getConfig()->getActionMarkAsRead();
     }
 
     /**
@@ -24,6 +28,10 @@ class Redirect
      */
     public function to($url, $fallback = '')
     {
+        if (Config::REDIRECT_TO_HOMEPAGE === $this->actionMarkAsRead) {
+            return $this->router->generate('homepage');
+        }
+
         if (null !== $url) {
             return $url;
         }