namespace Wallabag\CoreBundle\Repository;
use Doctrine\ORM\EntityRepository;
+use Doctrine\ORM\NoResultException;
use Doctrine\ORM\QueryBuilder;
use Pagerfanta\Adapter\DoctrineORMAdapter;
use Pagerfanta\Pagerfanta;
use Wallabag\CoreBundle\Entity\Entry;
use Wallabag\CoreBundle\Entity\Tag;
+use Wallabag\CoreBundle\Helper\UrlHasher;
class EntryRepository extends EntityRepository
{
->andWhere('t.id is null');
}
+ /**
+ * Retrieve the number of untagged entries for a user.
+ *
+ * @param int $userId
+ *
+ * @return int
+ */
+ public function countUntaggedEntriesByUser($userId)
+ {
+ return (int) $this->getRawBuilderForUntaggedByUser($userId)
+ ->select('count(e.id)')
+ ->getQuery()
+ ->getSingleScalarResult();
+ }
+
/**
* Find Entries.
*
* @param string $order
* @param int $since
* @param string $tags
+ * @param string $detail 'metadata' or 'full'. Include content field if 'full'
+ *
+ * @todo Breaking change: replace default detail=full by detail=metadata in a future version
*
* @return Pagerfanta
*/
- public function findEntries($userId, $isArchived = null, $isStarred = null, $isPublic = null, $sort = 'created', $order = 'asc', $since = 0, $tags = '')
+ public function findEntries($userId, $isArchived = null, $isStarred = null, $isPublic = null, $sort = 'created', $order = 'asc', $since = 0, $tags = '', $detail = 'full')
{
+ if (!\in_array(strtolower($detail), ['full', 'metadata'], true)) {
+ throw new \Exception('Detail "' . $detail . '" parameter is wrong, allowed: full or metadata');
+ }
+
$qb = $this->createQueryBuilder('e')
->leftJoin('e.tags', 't')
->where('e.user = :userId')->setParameter('userId', $userId);
+ if ('metadata' === $detail) {
+ $fieldNames = $this->getClassMetadata()->getFieldNames();
+ $fields = array_filter($fieldNames, function ($k) {
+ return 'content' !== $k;
+ });
+ $qb->select(sprintf('partial e.{%s}', implode(',', $fields)));
+ }
+
if (null !== $isArchived) {
$qb->andWhere('e.isArchived = :isArchived')->setParameter('isArchived', (bool) $isArchived);
}
* 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)
{
* Find an entry by its url and its owner.
* If it exists, return the entry otherwise return false.
*
- * @param $url
- * @param $userId
+ * @param string $url
+ * @param int $userId
*
- * @return Entry|bool
+ * @return Entry|false
*/
public function findByUrlAndUserId($url, $userId)
{
+ return $this->findByHashedUrlAndUserId(
+ UrlHasher::hashUrl($url),
+ $userId
+ );
+ }
+
+ /**
+ * Find an entry by its hashed url and its owner.
+ * If it exists, return the entry otherwise return false.
+ *
+ * @param string $hashedUrl Url hashed using sha1
+ * @param int $userId
+ *
+ * @return Entry|false
+ */
+ public function findByHashedUrlAndUserId($hashedUrl, $userId)
+ {
+ // try first using hashed_url (to use the database index)
$res = $this->createQueryBuilder('e')
- ->where('e.url = :url')->setParameter('url', urldecode($url))
+ ->where('e.hashedUrl = :hashed_url')->setParameter('hashed_url', $hashedUrl)
+ ->andWhere('e.user = :user_id')->setParameter('user_id', $userId)
+ ->getQuery()
+ ->getResult();
+
+ if (\count($res)) {
+ return current($res);
+ }
+
+ // then try using hashed_given_url (to use the database index)
+ $res = $this->createQueryBuilder('e')
+ ->where('e.hashedGivenUrl = :hashed_given_url')->setParameter('hashed_given_url', $hashedUrl)
->andWhere('e.user = :user_id')->setParameter('user_id', $userId)
->getQuery()
->getResult();
/**
* Find all entries by url and owner.
*
- * @param $url
- * @param $userId
+ * @param string $url
+ * @param int $userId
*
* @return array
*/
/**
* Returns a random entry, filtering by status.
*
- * @param $userId
- * @param string $status can be unread, archive or starred
- *
- * @throws \Doctrine\ORM\NoResultException
- * @throws \Doctrine\ORM\NonUniqueResultException
+ * @param int $userId
+ * @param string $type Can be unread, archive, starred, etc
*
- * @see https://github.com/doctrine/doctrine2/issues/5479#issuecomment-403862934
+ * @throws NoResultException
*
* @return Entry
*/
- public function getRandomEntry($userId, $status = '')
+ public function getRandomEntry($userId, $type = '')
{
$qb = $this->getQueryBuilderByUser($userId)
- ->select('MIN(e.id)', 'MAX(e.id)');
-
- if ('unread' === $status) {
- $qb->andWhere('e.isArchived = false');
- }
+ ->select('e.id');
- if ('archive' === $status) {
- $qb->andWhere('e.isArchived = true');
+ 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;
}
- if ('starred' === $status) {
- $qb->andWhere('e.isStarred = true');
- }
+ $ids = $qb->getQuery()->getArrayResult();
- if ('untagged' === $status) {
- $qb->leftJoin('e.tags', 't');
- $qb->andWhere('t.id is null');
+ if (empty($ids)) {
+ throw new NoResultException();
}
- $idLimits = $qb
- ->getQuery()
- ->getOneOrNullResult();
- $randomPossibleIds = rand($idLimits[1], $idLimits[2]);
+ // random select one in the list
+ $randomId = $ids[mt_rand(0, \count($ids) - 1)]['id'];
- return $qb
- ->select('e')
- ->andWhere('e.id >= :random_id')
- ->setParameter('random_id', $randomPossibleIds)
- ->setMaxResults(1)
- ->getQuery()
- ->getSingleResult();
+ return $this->find($randomId);
}
/**
/**
* Return the given QueryBuilder with an orderBy() call.
*
- * @param QueryBuilder $qb
- * @param string $sortBy
- * @param string $direction
+ * @param string $sortBy
+ * @param string $direction
*
* @return QueryBuilder
*/