+
+ /**
+ * Count all entries for a user.
+ *
+ * @param int $userId
+ *
+ * @return int
+ */
+ public function countAllEntriesByUser($userId)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->select('count(e)')
+ ->where('e.user = :userId')->setParameter('userId', $userId)
+ ;
+
+ return (int) $qb->getQuery()->getSingleScalarResult();
+ }
+
+ /**
+ * Remove all entries for a user id.
+ * Used when a user want to reset all informations.
+ *
+ * @param int $userId
+ */
+ public function removeAllByUserId($userId)
+ {
+ $this->getEntityManager()
+ ->createQuery('DELETE FROM Wallabag\CoreBundle\Entity\Entry e WHERE e.user = :userId')
+ ->setParameter('userId', $userId)
+ ->execute();
+ }
+
+ public function removeArchivedByUserId($userId)
+ {
+ $this->getEntityManager()
+ ->createQuery('DELETE FROM Wallabag\CoreBundle\Entity\Entry e WHERE e.user = :userId AND e.isArchived = TRUE')
+ ->setParameter('userId', $userId)
+ ->execute();
+ }
+
+ /**
+ * Get id and url from all entries
+ * Used for the clean-duplicates command.
+ */
+ public function findAllEntriesIdAndUrlByUserId($userId)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->select('e.id, e.url')
+ ->where('e.user = :userid')->setParameter(':userid', $userId);
+
+ return $qb->getQuery()->getArrayResult();
+ }
+
+ /**
+ * @param int $userId
+ *
+ * @return array
+ */
+ public function findAllEntriesIdByUserId($userId = null)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->select('e.id');
+
+ if (null !== $userId) {
+ $qb->where('e.user = :userid')->setParameter(':userid', $userId);
+ }
+
+ return $qb->getQuery()->getArrayResult();
+ }
+
+ /**
+ * Find all entries by url and owner.
+ *
+ * @param string $url
+ * @param int $userId
+ *
+ * @return array
+ */
+ public function findAllByUrlAndUserId($url, $userId)
+ {
+ return $this->createQueryBuilder('e')
+ ->where('e.url = :url')->setParameter('url', urldecode($url))
+ ->andWhere('e.user = :user_id')->setParameter('user_id', $userId)
+ ->getQuery()
+ ->getResult();
+ }
+
+ /**
+ * Returns a random entry, filtering by status.
+ *
+ * @param int $userId
+ * @param string $type Can be unread, archive, starred, etc
+ *
+ * @throws NoResultException
+ *
+ * @return Entry
+ */
+ public function getRandomEntry($userId, $type = '')
+ {
+ $qb = $this->getQueryBuilderByUser($userId)
+ ->select('e.id');
+
+ switch ($type) {
+ case 'unread':
+ $qb->andWhere('e.isArchived = false');
+ break;
+ case 'archive':
+ $qb->andWhere('e.isArchived = true');
+ break;
+ case 'starred':
+ $qb->andWhere('e.isStarred = true');
+ break;
+ case 'untagged':
+ $qb->leftJoin('e.tags', 't');
+ $qb->andWhere('t.id is null');
+ break;
+ }
+
+ $ids = $qb->getQuery()->getArrayResult();
+
+ if (empty($ids)) {
+ throw new NoResultException();
+ }
+
+ // random select one in the list
+ $randomId = $ids[mt_rand(0, \count($ids) - 1)]['id'];
+
+ return $this->find($randomId);
+ }
+
+ /**
+ * Return a query builder to be used by other getBuilderFor* method.
+ *
+ * @param int $userId
+ *
+ * @return QueryBuilder
+ */
+ private function getQueryBuilderByUser($userId)
+ {
+ return $this->createQueryBuilder('e')
+ ->andWhere('e.user = :userId')->setParameter('userId', $userId);
+ }
+
+ /**
+ * Return a sorted query builder to be used by other getBuilderFor* method.
+ *
+ * @param int $userId
+ * @param string $sortBy
+ * @param string $direction
+ *
+ * @return QueryBuilder
+ */
+ private function getSortedQueryBuilderByUser($userId, $sortBy = 'createdAt', $direction = 'desc')
+ {
+ return $this->sortQueryBuilder($this->getQueryBuilderByUser($userId), $sortBy, $direction);
+ }
+
+ /**
+ * Return the given QueryBuilder with an orderBy() call.
+ *
+ * @param QueryBuilder $qb
+ * @param string $sortBy
+ * @param string $direction
+ *
+ * @return QueryBuilder
+ */
+ private function sortQueryBuilder(QueryBuilder $qb, $sortBy = 'createdAt', $direction = 'desc')
+ {
+ return $qb->orderBy(sprintf('e.%s', $sortBy), $direction);
+ }