alexislefebvre / AsyncTweetsBundle

src/Command/StatusesHomeTimelineCommand.php

<?php

namespace AlexisLefebvre\Bundle\AsyncTweetsBundle\Command;

use Abraham\TwitterOAuth\TwitterOAuth;
use AlexisLefebvre\Bundle\AsyncTweetsBundle\Utils\PersistTweet;
use Symfony\Component\Console\Helper\ProgressBar;
use Symfony\Component\Console\Helper\Table;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;

class StatusesHomeTimelineCommand extends BaseCommand
{
    private $displayTable;
    private $table;
    private $progress;

    /** @see https://dev.twitter.com/rest/reference/get/statuses/home_timeline */
    private $parameters = [
        'count'           => 200,
        'exclude_replies' => true,
    ];

    protected function configure()
    {
        parent::configure();

        $this
            ->setName('statuses:hometimeline')
            ->setDescription('Fetch home timeline')
            // http://symfony.com/doc/2.3/cookbook/console/console_command.html#automatically-registering-commands
            ->addOption(
                'table',
                null,
                InputOption::VALUE_NONE,
                'Display a table with tweets'
            );
    }

    /**
     * @param InputInterface  $input
     * @param OutputInterface $output
     *
     * @return int|void
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $this->setAndDisplayLastTweet($output);

        $content = $this->getContent($input);

        if (!is_array($content)) {
            $this->displayContentNotArrayError($output, $content);

            return 1;
        }

        $numberOfTweets = count($content);

        if ($numberOfTweets == 0) {
            $output->writeln('<comment>No new tweet.</comment>');

            return 0;
        }

        $this->addAndDisplayTweets($input, $output, $content, $numberOfTweets);
    }

    /**
     * @param OutputInterface $output
     */
    protected function setAndDisplayLastTweet(OutputInterface $output)
    {
        // Get the last tweet
        $lastTweet = $this->em

It seems like you code against a concrete implementation and not the interface Doctrine\Persistence\ObjectRepository as the method getLastTweet() does only exist in the following implementations of said interface: AlexisLefebvre\Bundle\AsyncTweetsBundle\Entity\TweetRepository.

            ->getRepository('AsyncTweetsBundle:Tweet')
            ->getLastTweet();

        // And use it in the request if it exists
        if ($lastTweet) {
            $this->parameters['since_id'] = $lastTweet->getId();

            $comment = 'last tweet = '.$this->parameters['since_id'];
        } else {
            $comment = 'no last tweet';
        }

        $output->writeln('<comment>'.$comment.'</comment>');
    }

    /**
     * @param InputInterface $input
     */
    protected function getContent(InputInterface $input)
    {
        $connection = new TwitterOAuth(
            $this->container->getParameter('twitter_consumer_key'),
            $this->container->getParameter('twitter_consumer_secret'),
            $this->container->getParameter('twitter_token'),
            $this->container->getParameter('twitter_token_secret')
        );

        return $connection->get(
            'statuses/home_timeline',
            $this->parameters
        );
    }

    /**
     * @param OutputInterface $output
     * @param null|object     $content
     */
    protected function displayContentNotArrayError(
        OutputInterface $output,
        $content
    ) {
        $formatter = $this->getHelper('formatter');

        $errorMessages = ['Error!', 'Something went wrong, $content is not an array.'];
        $formattedBlock = $formatter->formatBlock($errorMessages, 'error');
        $output->writeln($formattedBlock);
        $output->writeln(print_r($content, true));
    }

    /**
     * @param InputInterface  $input
     * @param OutputInterface $output
     * @param array           $content
     * @param int             $numberOfTweets
     */
    protected function addAndDisplayTweets(
        InputInterface $input,
        OutputInterface $output,
        $content,
        $numberOfTweets
    ) {
        $output->writeln('<comment>Number of tweets: '.$numberOfTweets.'</comment>');

        // Iterate through $content in order to add the oldest tweet first,
        //  if there is an error the oldest tweet will still be saved
        //  and newer tweets can be saved next time the command is launched
        $tweets = array_reverse($content);

        $this->setProgressBar($output, $numberOfTweets);
        $this->setTable($input, $output);
        $this->iterateTweets($tweets);

        $this->progress->finish();
        $output->writeln('');

        if ($this->displayTable) {
            $this->table->render();
        }
    }

    /**
     * @param OutputInterface $output
     * @param int             $numberOfTweets
     */
    protected function setProgressBar(
        OutputInterface $output,
        $numberOfTweets
    ) {
        $this->progress = new ProgressBar($output, $numberOfTweets);
        $this->progress->setBarCharacter('<comment>=</comment>');
        $this->progress->start();
    }

    /**
     * @param InputInterface  $input
     * @param OutputInterface $output
     */
    protected function setTable(
        InputInterface $input,
        OutputInterface $output
    ) {
        $this->displayTable = $input->getOption('table');

        // Display
        if ($this->displayTable) {
            $this->table = new Table($output);
            $this->table
                ->setHeaders(['Datetime', 'Text excerpt', 'Name']);
        }
    }

    /**
     * @param array $tweets
     */
    protected function iterateTweets($tweets)
    {
        $persistTweet = new PersistTweet(
            $this->em,
            $this->displayTable,
            $this->table
        );

        foreach ($tweets as $tweetTmp) {
            $persistTweet->addTweet($tweetTmp, true);

            $this->progress->advance();
        }
    }
}