feat: notifies user of bad RD download (failed for copyright, etc.)

src/Download/Action/Handler/DownloadMediaHandler.php

@@ -2,9 +2,12 @@
 
 namespace App\Download\Action\Handler;
 
+use App\Base\Enum\MediaType;
+use App\Base\Service\Broadcaster;
 use App\Download\Action\Command\DownloadMediaCommand;
 use App\Download\Action\Result\DownloadMediaResult;
 use App\Download\DownloadEvents;
+use App\Download\Framework\Entity\Download;
 use App\Download\Framework\Repository\DownloadRepository;
 use App\Download\Downloader\DownloaderInterface;
 use App\EventLog\Action\Command\AddEventLogCommand;
@@ -21,8 +24,8 @@ readonly class DownloadMediaHandler implements HandlerInterface
     public function __construct(
         private MessageBusInterface $bus,
         private DownloaderInterface $downloader,
-        private DownloadRepository $downloadRepository,
-        private UserRepository $userRepository,
+        private DownloadRepository  $downloadRepository,
+        private UserRepository      $userRepository, private Broadcaster $broadcaster,
     ) {}
 
     public function handle(CommandInterface $command): ResultInterface
@@ -49,6 +52,16 @@ readonly class DownloadMediaHandler implements HandlerInterface
             $download = $this->downloadRepository->find($command->downloadId);
         }
 
+        try {
+            $this->validateDownloadUrl($download->getUrl());
+        } catch (\Throwable $exception) {
+            $download->setProgress(100);
+            $download->setStatus('Failed');
+            $this->downloadRepository->getEntityManager()->flush();
+            $this->sendFailedDownloadAlert($download, $command->mercureAlertTopic, $exception->getMessage());
+            return new DownloadMediaResult(400, $exception->getMessage());
+        }
+
         try {
             if ($download->getStatus() !== 'Paused') {
                 $this->downloadRepository->updateStatus($download->getId(), 'In Progress');
@@ -77,4 +90,30 @@ readonly class DownloadMediaHandler implements HandlerInterface
         ));
         return new DownloadMediaResult(200, "Success.");
     }
+
+    public function validateDownloadUrl(string $downloadUrl)
+    {
+        $badFileSizes = [
+            2119075, // copyright infringement
+        ];
+
+        $badFileLocations = [
+            'https://torrentio.strem.fun/videos/failed_infringement_v2.mp4' => 'Removed for Copyright Infringement.',
+        ];
+
+        $headers = get_headers($downloadUrl, true);
+        if (array_key_exists($headers['Location'], $badFileLocations)) {
+            throw new \Exception($badFileLocations[$headers['Location']]);
+        }
+    }
+
+    private function sendFailedDownloadAlert(Download $download, string $mercureAlertTopic, ?string $message = null): void
+    {
+        $this->broadcaster->alert(
+            title: 'Download Failed',
+            message: $message ?? "{$download->getTitle()} failed to download.",
+            type: 'warning',
+            mercureAlertTopic: $mercureAlertTopic
+        );
+    }
 }