1 files changed, 35 insertions, 6 deletions

diff --git a/src/Wallabag/CoreBundle/Twig/WallabagExtension.php b/src/Wallabag/CoreBundle/Twig/WallabagExtension.php
index a305c53f..034b3fa2 100644
--- a/src/Wallabag/CoreBundle/Twig/WallabagExtension.php
+++ b/src/Wallabag/CoreBundle/Twig/WallabagExtension.php
@@ -3,7 +3,9 @@
 namespace Wallabag\CoreBundle\Twig;
 
 use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;
+use Wallabag\CoreBundle\Notifications\NotificationInterface;
 use Wallabag\CoreBundle\Repository\EntryRepository;
+use Wallabag\CoreBundle\Repository\NotificationRepository;
 use Wallabag\CoreBundle\Repository\TagRepository;
 use Symfony\Component\Translation\TranslatorInterface;
 
@@ -12,15 +14,19 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
     private $tokenStorage;
     private $entryRepository;
     private $tagRepository;
+    private $notificationRepository;
     private $lifeTime;
+    private $nbNotifications;
     private $translator;
 
-    public function __construct(EntryRepository $entryRepository, TagRepository $tagRepository, TokenStorageInterface $tokenStorage, $lifeTime, TranslatorInterface $translator)
+    public function __construct(EntryRepository $entryRepository, TagRepository $tagRepository, NotificationRepository $notificationRepository, TokenStorageInterface $tokenStorage, $lifeTime, $nbNotifications, TranslatorInterface $translator)
     {
         $this->entryRepository = $entryRepository;
         $this->tagRepository = $tagRepository;
+        $this->notificationRepository = $notificationRepository;
         $this->tokenStorage = $tokenStorage;
         $this->lifeTime = $lifeTime;
+        $this->nbNotifications = $nbNotifications;
         $this->translator = $translator;
     }
 
@@ -28,6 +34,7 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
     {
         return [
             new \Twig_SimpleFilter('removeWww', [$this, 'removeWww']),
+            new \Twig_SimpleFilter('unread_notif', [$this, 'unreadNotif']),
         ];
     }
 
@@ -37,6 +44,7 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
             new \Twig_SimpleFunction('count_entries', [$this, 'countEntries']),
             new \Twig_SimpleFunction('count_tags', [$this, 'countTags']),
             new \Twig_SimpleFunction('display_stats', [$this, 'displayStats']),
+            new \Twig_SimpleFunction('get_notifications', [$this, 'getNotifications'])
         );
     }
 
@@ -46,6 +54,17 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
     }
 
     /**
+     * @param $notifs
+     * @return array
+     */
+    public function unreadNotif($notifs)
+    {
+        return array_filter($notifs, function (NotificationInterface $notif) {
+            return !$notif->isRead();
+        });
+    }
+
+    /**
      * Return number of entries depending of the type (unread, archive, starred or all).
      *
      * @param string $type Type of entries to count
@@ -111,6 +130,21 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
         return $this->tagRepository->countAllTags($user->getId());
     }
 
+    public function getNotifications()
+    {
+        $user = $this->tokenStorage->getToken() ? $this->tokenStorage->getToken()->getUser() : null;
+
+        if (null === $user || !is_object($user)) {
+            return 0;
+        }
+
+        return $this->notificationRepository->findBy(
+            ['user' => $user->getId()],
+            ['timestamp' => 'DESC'],
+            $this->nbNotifications
+        );
+    }
+
     /**
      * Display a single line about reading stats.
      *
@@ -147,9 +181,4 @@ class WallabagExtension extends \Twig_Extension implements \Twig_Extension_Globa
             '%per_day%' => round($nbArchives / $nbDays, 2),
         ]);
     }
-
-    public function getName()
-    {
-        return 'wallabag_extension';
-    }
 }