AlexisLefebvre\Bundle\AsyncTweetsBundle\Entity\TweetRepository

Complexity

Total Complexity

20

Size/Duplication

Total Lines	235
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Test Coverage

Coverage

100%

Importance

Changes	24
Bugs	12	Features	10

12 Methods

Rating	Name	Duplication	Size	Complexity
A	getWithUsers()	0	23	1
A	getWithUsersAndMediasQuery()	0	20	1
A	getWithUsersAndMedias()	0	15	2
A	getTweetId()	0	20	1
A	getIdOrNull()	0	4	2
A	getPreviousAndNextTweetId()	0	7	1
A	countPendingTweets()	0	22	2
A	getLastTweet()	0	10	1
A	getTweetsLessThanId()	0	22	1
A	removeOrphanMedias()	0	7	2
A	removeTweet()	0	18	3
A	deleteAndHideTweetsLessThanId()	0	22	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
     */
    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
     */
    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($this->getIdOrNull($result));
    }
    
    private function getIdOrNull($result)
    {
        return(is_array($result) ? $result['id'] : null);
    }
    
    public function getPreviousAndNextTweetId($tweetId)
    {
        return(array(
            $this->getTweetId('<', 'DESC', $tweetId),
            $this->getTweetId('>', 'ASC', $tweetId)
        ));
    }
    
    public function countPendingTweets($lastTweetId = null)
    {
        $qb = $this->createQueryBuilder('t');
        
        $query = $qb
            ->add('select', $qb->expr()->count('t.id'))

$qb->expr()->count('t.id') is of type object<Doctrine\ORM\Query\Expr\Func>, but the function expects a object<Doctrine\ORM\Query\Expr\Base>.

            // 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()
    {
        $qb = $this->createQueryBuilder('t')
            ->addOrderBy('t.id', 'DESC')
            ->setFirstResult(0)
            ->setMaxResults(1)
        ;
        
        return $qb->getQuery()->getOneOrNullResult();
    }
    
    /**
     * @param integer $tweetId
     */
    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
     */
    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 integer
     */
    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 integer $tweetId
     * 
     * @return integer
     */
    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);
    }
}