From 5624f86e383d452325607de658cb367109d6727d Mon Sep 17 00:00:00 2001
From: Oliver Davies <oliver@oliverdavies.uk>
Date: Wed, 16 Jan 2019 00:11:21 +0000
Subject: [PATCH] Fetch untweeted tweets, store retweet time

---
 src/Command/RetweetTweetsCommand.php | 30 +++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/src/Command/RetweetTweetsCommand.php b/src/Command/RetweetTweetsCommand.php
index 247a57b..02d6be7 100644
--- a/src/Command/RetweetTweetsCommand.php
+++ b/src/Command/RetweetTweetsCommand.php
@@ -6,6 +6,7 @@ use App\Entity\Tweet;
 use App\Repository\TweetRepository;
 use App\Service\Retweeter;
 use App\Service\TweetFetcher;
+use Doctrine\ORM\EntityManagerInterface;
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Input\InputOption;
@@ -19,25 +20,48 @@ class RetweetTweetsCommand extends Command
 
     private $retweeter;
 
-    public function __construct(TweetFetcher $tweetFetcher, Retweeter $retweeter)
-    {
+    /**
+     * @var \App\Repository\TweetRepository
+     */
+    private $tweetRepository;
+
+    /**
+     * @var \Doctrine\ORM\EntityManagerInterface
+     */
+    private $entityManager;
+
+    public function __construct(
+        TweetFetcher $tweetFetcher,
+        TweetRepository $tweetRepository,
+        Retweeter $retweeter,
+        EntityManagerInterface $entityManager
+    ) {
         parent::__construct();
 
         $this->tweetFetcher = $tweetFetcher;
         $this->retweeter = $retweeter;
+        $this->tweetRepository = $tweetRepository;
+        $this->entityManager = $entityManager;
     }
 
     protected function configure()
     {
         $this
             ->setDescription('Add a short description for your command')
+            ->addOption('number', null, InputOption::VALUE_OPTIONAL, 'Specify how many tweets to retweet.', 1)
         ;
     }
 
     protected function execute(InputInterface $input, OutputInterface $output)
     {
-        $this->tweetFetcher->getTweets()->each(function (Tweet $tweet) {
+        $this->tweetRepository->getUntweetedTweets($input->getOption('number'))->each(function (Tweet $tweet) {
             $this->retweeter->retweet($tweet);
+
+            $tweet->setRetweeted(time());
+
+            $this->entityManager->persist($tweet);
         });
+
+        $this->entityManager->flush();
     }
 }