Redirect to homepage if referer is null

Fix #1924

1 files changed, 36 insertions, 0 deletions

diff --git a/src/Wallabag/CoreBundle/Helper/Redirect.php b/src/Wallabag/CoreBundle/Helper/Redirect.php
new file mode 100644
index 00000000..0921c3f9
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Helper/Redirect.php
@@ -0,0 +1,36 @@
+<?php
+
+namespace Wallabag\CoreBundle\Helper;
+
+use Symfony\Component\Routing\Router;
+
+class Redirect
+{
+    private $router;
+
+    public function __construct(Router $router)
+    {
+        $this->router = $router;
+    }
+
+    /**
+     * @param string $url      URL to redirect
+     * @param string $fallback Fallback URL if $url is null
+     *
+     * @return string
+     */
+    public function to($url, $fallback = '')
+    {
+        $returnUrl = $url;
+
+        if (null === $url) {
+            if ('' !== $fallback) {
+                $returnUrl = $fallback;
+            } else {
+                $returnUrl = $this->router->generate('homepage');
+            }
+        }
+
+        return $returnUrl;
+    }
+}