From cc88660c078ba2830fae80e221b218105af7d4f3 Mon Sep 17 00:00:00 2001
From: Brock H Caldwell <brock@caldwell.digital>
Date: Sun, 8 Feb 2026 15:11:14 -0600
Subject: [PATCH] fix(DownloadSeasonHandler): captures episode id

---
 .../Action/Handler/DownloadSeasonHandler.php  | 59 +++++++++++--------
 1 file changed, 35 insertions(+), 24 deletions(-)

diff --git a/src/Download/Action/Handler/DownloadSeasonHandler.php b/src/Download/Action/Handler/DownloadSeasonHandler.php
index 0cb0fca..233d6f5 100644
--- a/src/Download/Action/Handler/DownloadSeasonHandler.php
+++ b/src/Download/Action/Handler/DownloadSeasonHandler.php
@@ -10,6 +10,8 @@ use App\Download\Action\Command\DownloadSeasonCommand;
 use App\Download\Action\Result\DownloadMediaResult;
 use App\Download\Action\Result\DownloadSeasonResult;
 use App\Download\DownloadOptionEvaluator;
+use App\Download\Framework\Entity\Download;
+use App\Download\Framework\Repository\DownloadRepository;
 use App\Tmdb\TmdbClient;
 use App\Torrentio\Action\Command\GetTvShowOptionsCommand;
 use App\Torrentio\Action\Handler\GetTvShowOptionsHandler;
@@ -26,13 +28,13 @@ use Symfony\Component\Messenger\MessageBusInterface;
 readonly class DownloadSeasonHandler implements HandlerInterface
 {
     public function __construct(
-        private MediaFiles $mediaFiles,
-        private LoggerInterface $logger,
-        private TmdbClient $tmdb,
-        private MessageBusInterface $bus,
+        private MediaFiles              $mediaFiles,
+        private LoggerInterface         $logger,
+        private TmdbClient              $tmdb,
+        private MessageBusInterface     $bus,
         private DownloadOptionEvaluator $downloadOptionEvaluator,
         private GetTvShowOptionsHandler $getTvShowOptionsHandler,
-        private UserRepository $userRepository,
+        private UserRepository          $userRepository, private DownloadRepository $downloadRepository,
     ) {}
 
     public function handle(CommandInterface $command): ResultInterface
@@ -68,13 +70,15 @@ readonly class DownloadSeasonHandler implements HandlerInterface
             if (null !== $result) {
                 $this->logger->info('> [DownloadTvSeasonHandler] ......Found 1 matching result');
                 $this->logger->info('> [DownloadTvSeasonHandler] ......Dispatching DownloadMediaCommand for "' . $series->title . '" season ' . $command->season . ' episode ' . $episode->episodeNumber);
+                $download = $this->createDownload($command, $result->url, $series->title, $result->filename, $episode->episodeNumber);
                 $downloadCommand = new DownloadMediaCommand(
-                    $result->url,
-                    $series->title,
-                    $result->filename,
-                    'tvshows',
-                    $command->imdbId,
-                    $command->userId,
+                    $download->getUrl(),
+                    $download->getTitle(),
+                    $download->getFilename(),
+                    $download->getMediaType(),
+                    $download->getImdbId(),
+                    $download->getUser()->getId(),
+                    $download->getId()
                 );
                 $this->bus->dispatch($downloadCommand);
                 $downloadCommands[] = $downloadCommand;
@@ -90,19 +94,26 @@ readonly class DownloadSeasonHandler implements HandlerInterface
         );
     }
 
-    private function getDownloadedEpisodes(string $title)
+    private function createDownload(DownloadSeasonCommand $command, string $url, string $title, string $filename, int $episodeNumber): Download
     {
-        // Check current episodes
-        $downloadedEpisodes = $this->mediaFiles
-            ->getEpisodes($title)
-            ->map(fn($episode) => (object) (new PTN())->parse($episode))
-            ->filter(fn ($episode) =>
-                property_exists($episode, 'episode')
-                && property_exists($episode, 'season')
-                && null !== $episode->episodeNumber
-                && null !== $episode->season
-            )
-            ->rekey(fn($episode) => $episode->episodeNumber);
-        $this->logger->info('> [MonitorTvSeasonHandler] Found ' . count($downloadedEpisodes) . ' downloaded episodes for title: ' . $monitor->getTitle());
+        $download = new Download();
+        $download->setUrl($url);
+        $download->setTitle($title);
+        $download->setFilename($filename);
+        $download->setImdbId($command->imdbId);
+        $download->setMediaType(MediaType::TvShow->value);
+        $download->setEpisodeId($this->getEpisodeNumber($command->season, $episodeNumber));
+        $download->setUser($this->userRepository->find($command->userId));
+        $this->downloadRepository->getEntityManager()->persist($download);
+        return $download;
+    }
+
+    private function getEpisodeNumber(int $season, int $episode): string
+    {
+        return sprintf(
+            "S%sE%s",
+            str_pad($season, 2, "0", STR_PAD_LEFT),
+            str_pad($episode, 2, "0", STR_PAD_LEFT)
+        );
     }
 }