AlexisLefebvre\Bundle\AsyncTweetsBundle\Entity\TweetRepository

Complexity

Total Complexity

20

Size/Duplication

Total Lines	232
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	7

Test Coverage

Coverage

62.5%

Importance

Changes

0

12 Methods

Rating	Name	Duplication	Size	Complexity
A	getWithUsers()	0	22	1
A	getWithUsersAndMediasQuery()	0	19	1
A	getWithUsersAndMedias()	0	14	2
A	getTweetId()	0	19	2
A	getPreviousTweetId()	0	4	1
A	getNextTweetId()	0	4	1
A	countPendingTweets()	0	21	2
A	getLastTweet()	0	9	1
A	getTweetsLessThanId()	0	21	1
A	removeOrphanMedias()	0	6	2
A	removeTweet()	0	18	3
A	deleteAndHideTweetsLessThanId()	0	21	3

<?php

namespace AlexisLefebvre\Bundle\AsyncTweetsBundle\Entity;

use Doctrine\ORM\EntityRepository;

/**
 * TweetRepository.
 *
 * This class was generated by the Doctrine ORM. Add your own custom
 * repository methods below.
 */
class TweetRepository extends EntityRepository
{
    private $nbTweets = 5;

    public function getWithUsers($page = 1)
    {
        $firstResult = (($page - 1) * $this->nbTweets);

        $qb = $this->createQueryBuilder('t');

        $query = $qb
            ->select('t, user, rt, rt_user')
            ->innerJoin('t.user', 'user')
            ->leftJoin('t.retweeted_status', 'rt')
            ->leftJoin('rt.user', 'rt_user')

            // Ignore tweets that were only retweeted
            ->where($qb->expr()->eq('t.in_timeline', 'true'))

            ->orderBy('t.id', 'DESC')

            ->setFirstResult($firstResult)
            ->setMaxResults($this->nbTweets);

        return $query->getQuery()->getResult();
    }

    /**
     * @param \Doctrine\ORM\QueryBuilder $qb
     *
     * @return \Doctrine\ORM\QueryBuilder $query
     */
    private function getWithUsersAndMediasQuery($qb)
    {
        $query = $qb
            ->select('t, user, medias, rt, rt_user')
            ->innerJoin('t.user', 'user')
            ->leftJoin('t.medias', 'medias')
            ->leftJoin('t.retweeted_status', 'rt')
            ->leftJoin('rt.user', 'rt_user')

            // Ignore tweets that were only retweeted
            ->where($qb->expr()->eq('t.in_timeline', 'true'))

            ->orderBy('t.id', 'ASC')

            ->setFirstResult(0)
            ->setMaxResults($this->nbTweets);

        return $query;
    }

    public function getWithUsersAndMedias($firstTweetId = null)
    {
        $qb = $this->createQueryBuilder('t');

        $query = $this->getWithUsersAndMediasQuery($qb);

        if (!is_null($firstTweetId)) {
            $query = $query->andWhere(
                $qb->expr()->gte('t.id', $firstTweetId)
            );
        }

        return $query->getQuery()->getResult();
    }

    /**
     * @param string $condition
     * @param string $order
     * @param int    $tweetId
     *
     * @return int|null
     */
    private function getTweetId($condition, $order, $tweetId)
    {
        $qb = $this->createQueryBuilder('t')
            ->select('t.id')

            ->where('t.id '.$condition.' :tweetId')
            ->setParameter(':tweetId', $tweetId)

            ->andWhere('t.in_timeline = true')

            ->orderBy('t.id', $order)

            ->setFirstResult($this->nbTweets - 1)
            ->setMaxResults(1);

        $result = $qb->getQuery()->getOneOrNullResult();

        return is_array($result) ? $result['id'] : null;
    }

    public function getPreviousTweetId($tweetId)
    {
        return $this->getTweetId('<', 'DESC', $tweetId);
    }

    public function getNextTweetId($tweetId)
    {
        return $this->getTweetId('>', 'ASC', $tweetId);
    }

    public function countPendingTweets($lastTweetId = null): int
    {
        /** @var \Doctrine\ORM\QueryBuilder $qb */
        $qb = $this->createQueryBuilder('t');

        $query = $qb
            ->add('select', $qb->expr()->count('t.id'))
            // Ignore tweets that were only retweeted
            ->where(
                $qb->expr()->eq('t.in_timeline', 'true')
            );

        if (!is_null($lastTweetId)) {
            $query = $query->andWhere(
                $qb->expr()->gte('t.id', $lastTweetId)
            );
        }

        // return result of "COUNT()" query
        return $query->getQuery()->getSingleScalarResult();
    }

    public function getLastTweet(): ?Tweet
    {
        $qb = $this->createQueryBuilder('t')
            ->addOrderBy('t.id', 'DESC')
            ->setFirstResult(0)
            ->setMaxResults(1);

        return $qb->getQuery()->getOneOrNullResult();
    }

    /**
     * @param int $tweetId
     *
     * @return array
     */
    private function getTweetsLessThanId($tweetId)
    {
        $qb = $this->createQueryBuilder('t')
            ->select('t, m')
            ->leftJoin('t.medias', 'm')
            ->where('t.id < :tweetId')
            ->setParameter(':tweetId', $tweetId)

            // Get retweeted tweets (it would break foreign keys)
            //  http://stackoverflow.com/questions/15087933/how-to-do-left-join-in-doctrine/15088250#15088250
            ->leftJoin(
                'AsyncTweetsBundle:Tweet',
                't2',
                'WITH',
                't.id = t2.retweeted_status'
            )

            ->orderBy('t.id', 'DESC');

        return $qb->getQuery()->getResult();
    }

    /**
     * Remove Media not associated to any Tweet.
     *
     * @param Media $media
     */
    private function removeOrphanMedias(Media $media)
    {
        if (count($media->getTweets()) == 0) {
            $this->_em->remove($media);
        }
    }

    /**
     * Remove the tweet and return 1 is the deleted tweet is not a
     *  retweet.
     *
     * @param Tweet $tweet
     *
     * @return int
     */
    protected function removeTweet($tweet)
    {
        $count = 0;

        foreach ($tweet->getMedias() as $media) {
            $tweet->removeMedia($media);
            $this->removeOrphanMedias($media);
        }

        // Don't count tweets that were only retweeted
        if ($tweet->isInTimeline()) {
            $count = 1;
        }

        $this->_em->remove($tweet);

        return $count;
    }

    /**
     * Delete tweets and return the number of deleted tweets (excluding
     *  retweeted-only tweets).
     *
     * @param int $tweetId
     *
     * @return int
     */
    public function deleteAndHideTweetsLessThanId($tweetId)
    {
        $count = 0;

        $tweets = $this->getTweetsLessThanId($tweetId);

        foreach ($tweets as $tweet) {
            if ($tweet->mustBeKept($tweetId)) {
                // The Tweet is still in the timeline, it can only be hidden
                $tweet->setInTimeline(false);
                $this->_em->persist($tweet);
            } else {
                // The Tweet has not been retweeted, it can be removed
                $count += $this->removeTweet($tweet);
            }
        }

        $this->_em->flush();

        return $count;
    }
}