namespace Wallabag\CoreBundle\Repository;
use Doctrine\ORM\EntityRepository;
-use Doctrine\ORM\Tools\Pagination\Paginator;
+use Doctrine\ORM\Query;
+use Pagerfanta\Adapter\DoctrineORMAdapter;
+use Pagerfanta\Pagerfanta;
+use Wallabag\CoreBundle\Entity\Tag;
class EntryRepository extends EntityRepository
{
/**
- * Retrieves unread entries for a user
+ * Return a query builder to used by other getBuilderFor* method.
*
* @param int $userId
- * @param int $firstResult
- * @param int $maxResults
*
- * @return Paginator
+ * @return QueryBuilder
*/
- public function findUnreadByUser($userId, $firstResult, $maxResults = 12)
+ private function getBuilderByUser($userId)
{
- $qb = $this->createQueryBuilder('e')
- ->setFirstResult($firstResult)
- ->setMaxResults($maxResults)
- ->leftJoin('e.user', 'u')
- ->where('e.isArchived = false')
- ->andWhere('u.id =:userId')->setParameter('userId', $userId)
- ->andWhere('e.isDeleted=false')
+ return $this->createQueryBuilder('e')
+ ->andWhere('e.user = :userId')->setParameter('userId', $userId)
->orderBy('e.createdAt', 'desc')
- ->getQuery();
-
- $paginator = new Paginator($qb);
-
- return $paginator;
+ ;
}
/**
- * Retrieves read entries for a user
+ * Retrieves all entries for a user.
*
* @param int $userId
- * @param int $firstResult
- * @param int $maxResults
*
- * @return Paginator
+ * @return QueryBuilder
*/
- public function findArchiveByUser($userId, $firstResult, $maxResults = 12)
+ public function getBuilderForAllByUser($userId)
{
- $qb = $this->createQueryBuilder('e')
- ->select('e')
- ->setFirstResult($firstResult)
- ->setMaxResults($maxResults)
- ->leftJoin('e.user', 'u')
- ->where('e.isArchived = true')
- ->andWhere('u.id =:userId')->setParameter('userId', $userId)
- ->andWhere('e.isDeleted=false')
- ->orderBy('e.createdAt', 'desc')
- ->getQuery();
+ return $this
+ ->getBuilderByUser($userId)
+ ;
+ }
- $paginator = new Paginator($qb);
+ /**
+ * Retrieves unread entries for a user.
+ *
+ * @param int $userId
+ *
+ * @return QueryBuilder
+ */
+ public function getBuilderForUnreadByUser($userId)
+ {
+ return $this
+ ->getBuilderByUser($userId)
+ ->andWhere('e.isArchived = false')
+ ;
+ }
- return $paginator;
+ /**
+ * Retrieves read entries for a user.
+ *
+ * @param int $userId
+ *
+ * @return QueryBuilder
+ */
+ public function getBuilderForArchiveByUser($userId)
+ {
+ return $this
+ ->getBuilderByUser($userId)
+ ->andWhere('e.isArchived = true')
+ ;
}
/**
- * Retrieves starred entries for a user
+ * Retrieves starred entries for a user.
*
* @param int $userId
- * @param int $firstResult
- * @param int $maxResults
*
- * @return Paginator
+ * @return QueryBuilder
*/
- public function findStarredByUser($userId, $firstResult, $maxResults = 12)
+ public function getBuilderForStarredByUser($userId)
{
- $qb = $this->createQueryBuilder('e')
- ->select('e')
- ->setFirstResult($firstResult)
- ->setMaxResults($maxResults)
- ->leftJoin('e.user', 'u')
- ->where('e.isStarred = true')
- ->andWhere('u.id =:userId')->setParameter('userId', $userId)
- ->andWhere('e.isDeleted = false')
- ->orderBy('e.createdAt', 'desc')
- ->getQuery();
+ return $this
+ ->getBuilderByUser($userId)
+ ->andWhere('e.isStarred = true')
+ ;
+ }
- $paginator = new Paginator($qb);
+ /**
+ * Retrieves entries filtered with a search term for a user.
+ *
+ * @param int $userId
+ * @param string $term
+ * @param strint $currentRoute
+ *
+ * @return QueryBuilder
+ */
+ public function getBuilderForSearchByUser($userId, $term, $currentRoute)
+ {
+ $qb = $this
+ ->getBuilderByUser($userId);
- return $paginator;
+ if ('starred' === $currentRoute) {
+ $qb->andWhere('e.isStarred = true');
+ } elseif ('unread' === $currentRoute) {
+ $qb->andWhere('e.isArchived = false');
+ } elseif ('archive' === $currentRoute) {
+ $qb->andWhere('e.isArchived = true');
+ }
+
+ // We lower() all parts here because PostgreSQL 'LIKE' verb is case-sensitive
+ $qb
+ ->andWhere('lower(e.content) LIKE lower(:term) OR lower(e.title) LIKE lower(:term) OR lower(e.url) LIKE lower(:term)')->setParameter('term', '%'.$term.'%')
+ ->leftJoin('e.tags', 't')
+ ->groupBy('e.id');
+
+ return $qb;
+ }
+
+ /**
+ * Retrieves untagged entries for a user.
+ *
+ * @param int $userId
+ *
+ * @return QueryBuilder
+ */
+ public function getBuilderForUntaggedByUser($userId)
+ {
+ return $this
+ ->getBuilderByUser($userId)
+ ->andWhere('size(e.tags) = 0');
}
/**
- * Find Entries
+ * Find Entries.
*
* @param int $userId
* @param bool $isArchived
* @param bool $isStarred
- * @param bool $isDeleted
+ * @param bool $isPublic
* @param string $sort
* @param string $order
+ * @param int $since
+ * @param string $tags
*
* @return array
*/
- public function findEntries($userId, $isArchived = null, $isStarred = null, $isDeleted = null, $sort = 'created', $order = 'ASC')
+ public function findEntries($userId, $isArchived = null, $isStarred = null, $isPublic = null, $sort = 'created', $order = 'ASC', $since = 0, $tags = '')
{
$qb = $this->createQueryBuilder('e')
+ ->leftJoin('e.tags', 't')
->where('e.user =:userId')->setParameter('userId', $userId);
if (null !== $isArchived) {
- $qb->andWhere('e.isArchived =:isArchived')->setParameter('isArchived', (bool) $isArchived);
+ $qb->andWhere('e.isArchived = :isArchived')->setParameter('isArchived', (bool) $isArchived);
}
if (null !== $isStarred) {
- $qb->andWhere('e.isStarred =:isStarred')->setParameter('isStarred', (bool) $isStarred);
+ $qb->andWhere('e.isStarred = :isStarred')->setParameter('isStarred', (bool) $isStarred);
}
- if (null !== $isDeleted) {
- $qb->andWhere('e.isDeleted =:isDeleted')->setParameter('isDeleted', (bool) $isDeleted);
+ if (null !== $isPublic) {
+ $qb->andWhere('e.uid IS '.(true === $isPublic ? 'NOT' : '').' NULL');
+ }
+
+ if ($since > 0) {
+ $qb->andWhere('e.updatedAt > :since')->setParameter('since', new \DateTime(date('Y-m-d H:i:s', $since)));
+ }
+
+ if ('' !== $tags) {
+ foreach (explode(',', $tags) as $tag) {
+ $qb->andWhere('t.label = :label')->setParameter('label', $tag);
+ }
}
if ('created' === $sort) {
- $qb->orderBy('e.createdAt', $order);
+ $qb->orderBy('e.id', $order);
} elseif ('updated' === $sort) {
$qb->orderBy('e.updatedAt', $order);
}
- return $qb
+ $pagerAdapter = new DoctrineORMAdapter($qb, true, false);
+
+ return new Pagerfanta($pagerAdapter);
+ }
+
+ /**
+ * Fetch an entry with a tag. Only used for tests.
+ *
+ * @param int $userId
+ *
+ * @return Entry
+ */
+ public function findOneWithTags($userId)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->innerJoin('e.tags', 't')
+ ->innerJoin('e.user', 'u')
+ ->addSelect('t', 'u')
+ ->where('e.user=:userId')->setParameter('userId', $userId)
+ ;
+
+ return $qb->getQuery()->getResult();
+ }
+
+ /**
+ * Find distinct language for a given user.
+ * Used to build the filter language list.
+ *
+ * @param int $userId User id
+ *
+ * @return array
+ */
+ public function findDistinctLanguageByUser($userId)
+ {
+ $results = $this->createQueryBuilder('e')
+ ->select('e.language')
+ ->where('e.user = :userId')->setParameter('userId', $userId)
+ ->andWhere('e.language IS NOT NULL')
+ ->groupBy('e.language')
+ ->orderBy('e.language', ' ASC')
+ ->getQuery()
+ ->getResult();
+
+ $languages = [];
+ foreach ($results as $result) {
+ $languages[$result['language']] = $result['language'];
+ }
+
+ return $languages;
+ }
+
+ /**
+ * Used only in test case to get the right entry associated to the right user.
+ *
+ * @param string $username
+ *
+ * @return Entry
+ */
+ public function findOneByUsernameAndNotArchived($username)
+ {
+ return $this->createQueryBuilder('e')
+ ->leftJoin('e.user', 'u')
+ ->where('u.username = :username')->setParameter('username', $username)
+ ->andWhere('e.isArchived = false')
+ ->setMaxResults(1)
+ ->getQuery()
+ ->getSingleResult();
+ }
+
+ /**
+ * Remove a tag from all user entries.
+ *
+ * We need to loop on each entry attached to the given tag to remove it, since Doctrine doesn't know EntryTag entity because it's a ManyToMany relation.
+ * It could be faster with one query but I don't know how to retrieve the table name `entry_tag` which can have a prefix:
+ *
+ * DELETE et FROM entry_tag et WHERE et.entry_id IN ( SELECT e.id FROM entry e WHERE e.user_id = :userId ) AND et.tag_id = :tagId
+ *
+ * @param int $userId
+ * @param Tag $tag
+ */
+ public function removeTag($userId, Tag $tag)
+ {
+ $entries = $this->getBuilderByUser($userId)
+ ->innerJoin('e.tags', 't')
+ ->andWhere('t.id = :tagId')->setParameter('tagId', $tag->getId())
+ ->getQuery()
+ ->getResult();
+
+ foreach ($entries as $entry) {
+ $entry->removeTag($tag);
+ }
+
+ $this->getEntityManager()->flush();
+ }
+
+ /**
+ * Remove tags from all user entries.
+ *
+ * @param int $userId
+ * @param Array<Tag> $tags
+ */
+ public function removeTags($userId, $tags)
+ {
+ foreach ($tags as $tag) {
+ $this->removeTag($userId, $tag);
+ }
+ }
+
+ /**
+ * Find all entries that are attached to a give tag id.
+ *
+ * @param int $userId
+ * @param int $tagId
+ *
+ * @return array
+ */
+ public function findAllByTagId($userId, $tagId)
+ {
+ return $this->getBuilderByUser($userId)
+ ->innerJoin('e.tags', 't')
+ ->andWhere('t.id = :tagId')->setParameter('tagId', $tagId)
+ ->getQuery()
+ ->getResult();
+ }
+
+ /**
+ * Find an entry by its url and its owner.
+ * If it exists, return the entry otherwise return false.
+ *
+ * @param $url
+ * @param $userId
+ *
+ * @return Entry|bool
+ */
+ public function findByUrlAndUserId($url, $userId)
+ {
+ $res = $this->createQueryBuilder('e')
+ ->where('e.url = :url')->setParameter('url', urldecode($url))
+ ->andWhere('e.user = :user_id')->setParameter('user_id', $userId)
+ ->getQuery()
+ ->getResult();
+
+ if (count($res)) {
+ return current($res);
+ }
+
+ return false;
+ }
+
+ /**
+ * 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 $qb->getQuery()->getSingleScalarResult();
+ }
+
+ /**
+ * Count all entries for a tag and a user.
+ *
+ * @param int $userId
+ * @param int $tagId
+ *
+ * @return int
+ */
+ public function countAllEntriesByUserIdAndTagId($userId, $tagId)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->select('count(e.id)')
+ ->leftJoin('e.tags', 't')
+ ->where('e.user=:userId')->setParameter('userId', $userId)
+ ->andWhere('t.id=:tagId')->setParameter('tagId', $tagId)
+ ;
+
+ return $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 getAllEntriesIdAndUrl($userId)
+ {
+ $qb = $this->createQueryBuilder('e')
+ ->select('e.id, e.url')
+ ->where('e.user = :userid')->setParameter(':userid', $userId);
+
+ return $qb->getQuery()->getArrayResult();
+ }
+
+ /**
+ * Find all entries by url and owner.
+ *
+ * @param $url
+ * @param $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();
}