1 files changed, 26 insertions, 0 deletions

diff --git a/src/Wallabag/CoreBundle/Repository/NotificationRepository.php b/src/Wallabag/CoreBundle/Repository/NotificationRepository.php
new file mode 100644
index 00000000..6d6938ae
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Repository/NotificationRepository.php
@@ -0,0 +1,26 @@
+<?php
+
+namespace Wallabag\CoreBundle\Repository;
+
+use Doctrine\ORM\EntityRepository;
+
+class NotificationRepository extends EntityRepository
+{
+    public function markAllAsReadForUser($userId)
+    {
+        return $this->getEntityManager()->createQueryBuilder()
+        ->update('WallabagCoreBundle:Notification', 'n')
+        ->set('n.read', true)
+        ->where('n.user = :userId')->setParameter('userId', $userId)
+        ->getQuery()
+        ->getResult();
+    }
+
+    public function getBuilderForNotificationsByUser($userId)
+    {
+        return $this->createQueryBuilder('n')
+            ->andWhere('n.user = :userId')->setParameter('userId', $userId)
+            ->orderBy('n.timestamp', 'desc')
+            ;
+    }
+}