Update movie sets in update-data CLI command

Author: Programie

config/defaults.yaml

@@ -9,6 +9,7 @@ parameters:
   tvdb.shows.language: eng
   scrobbler.use-queue: false
   show-fetch-interval: 86400
+  movie-set-fetch-interval: 86400
   image-fetcher.storage-path: "%kernel.cache_dir%/images"
   image-fetcher.download-all-on-update: true
   user.enable-register: true

config/services.yaml

@@ -24,6 +24,7 @@ services:
   tracky\console\UpdateDataCommand:
     arguments:
       $showFetchInterval: "%show-fetch-interval%"
+      $movieSetFetchInterval: "%movie-set-fetch-interval%"
       $downloadAllImages: "%image-fetcher.download-all-on-update%"
   tracky\controller\UserController:
     arguments:

migrations/Version20260715190056.php

@@ -0,0 +1,23 @@
+<?php
+namespace DoctrineMigrations;
+
+use Doctrine\DBAL\Schema\Schema;
+use Doctrine\Migrations\AbstractMigration;
+
+final class Version20260715190056 extends AbstractMigration
+{
+    public function getDescription(): string
+    {
+        return "Add lastupdate column to moviesets";
+    }
+
+    public function up(Schema $schema): void
+    {
+        $this->addSql("ALTER TABLE moviesets ADD lastUpdate DATETIME DEFAULT NULL");
+    }
+
+    public function down(Schema $schema): void
+    {
+        $this->addSql("ALTER TABLE moviesets DROP lastUpdate");
+    }
+}

src/main/php/tracky/console/FetchDataCommand.php

@@ -120,7 +120,7 @@ public function execute(InputInterface $input, OutputInterface $output): int
                 if ($this->downloadAllImages) {
                     $output->writeln(sprintf("Fetching images for movie set %d (%s)", $id, $movieSet->getTitle()));
 
-                    $movieSet->fetchPosterImage($this->imageFetcher);
+                    $movieSet->fetchPosterImages($this->imageFetcher, true);
                 }
                 break;
             default:

src/main/php/tracky/console/UpdateDataCommand.php

@@ -9,18 +9,22 @@
 use Symfony\Component\Console\Output\OutputInterface;
 use tracky\dataprovider\Helper;
 use tracky\ImageFetcher;
+use tracky\model\MovieSet;
 use tracky\model\Show;
+use tracky\orm\MovieSetRepository;
 use tracky\orm\ShowRepository;
 
-#[AsCommand(name: "update-data", description: "Fetch new episodes and seasons for all shows needing an update")]
+#[AsCommand(name: "update-data", description: "Fetch updates of shows and movie sets needing an update")]
 class UpdateDataCommand extends Command
 {
     public function __construct(
         private readonly Helper                 $dataProviderHelper,
         private readonly ShowRepository         $showRepository,
+        private readonly MovieSetRepository     $movieSetRepository,
         private readonly ImageFetcher           $imageFetcher,
         private readonly EntityManagerInterface $entityManager,
         private readonly int                    $showFetchInterval,
+        private readonly int                    $movieSetFetchInterval,
         private readonly bool                   $downloadAllImages
     )
     {
@@ -29,7 +33,7 @@ public function __construct(
 
     protected function configure(): void
     {
-        $this->addOption("force", "f", InputOption::VALUE_NONE, "Update all shows even if they don't need an update");
+        $this->addOption("force", "f", InputOption::VALUE_NONE, "Update all shows and movie sets even if they don't need an update");
     }
 
     public function execute(InputInterface $input, OutputInterface $output): int
@@ -55,6 +59,27 @@ public function execute(InputInterface $input, OutputInterface $output): int
             }
         }
 
+        /**
+         * @var MovieSet
+         */
+        foreach ($this->movieSetRepository->findAll() as $movieSet) {
+            if (!$input->getOption("force") and !$movieSet->needsUpdate($this->movieSetFetchInterval)) {
+                continue;
+            }
+
+            $output->writeln(sprintf("Fetching data for movie set: %s", $movieSet->getTitle()));
+
+            $dataProvider = $this->dataProviderHelper->getProviderByEntry($movieSet);
+            $dataProvider->fetchMovieSet($movieSet);
+
+            $this->entityManager->persist($movieSet);
+            $this->entityManager->flush();
+
+            if ($this->downloadAllImages) {
+                $movieSet->fetchPosterImages($this->imageFetcher, true);
+            }
+        }
+
         return self::SUCCESS;
     }
 }

src/main/php/tracky/dataprovider/TMDB.php

@@ -288,6 +288,8 @@ public function fetchMovieSet(MovieSet $movieSet): bool
             $this->fetchMovie($movie);
         }
 
+        $movieSet->setLastUpdate(new DateTime);
+
         return true;
     }
 

src/main/php/tracky/dataprovider/TVDB.php

@@ -403,6 +403,8 @@ public function fetchMovieSet(MovieSet $movieSet): bool
             $movieSet->setPlot($translationData["overview"] ?? null);
         }
 
+        $movieSet->setLastUpdate(new DateTime);
+
         return true;
     }
 

src/main/php/tracky/model/MovieSet.php

@@ -2,6 +2,8 @@
 namespace tracky\model;
 
 use Doctrine\ORM\Mapping as ORM;
+use tracky\datetime\DateTime;
+use tracky\ImageFetcher;
 use tracky\model\traits\DataProvider;
 use tracky\model\traits\Plot;
 use tracky\model\traits\PosterImage;
@@ -18,6 +20,9 @@ class MovieSet extends BaseEntity
     #[ORM\Column(type: "string")]
     private string $title;
 
+    #[ORM\Column(name: "lastUpdate", type: "datetime", nullable: true)]
+    private ?DateTime $lastUpdate;
+
     #[ORM\OneToMany(mappedBy: "movieSet", targetEntity: Movie::class, cascade: ["persist"])]
     #[ORM\OrderBy(["year" => "ASC"])]
     private mixed $movies = [];
@@ -33,6 +38,31 @@ public function setTitle(string $title): MovieSet
         return $this;
     }
 
+    public function setLastUpdate(?DateTime $lastUpdate): self
+    {
+        $this->lastUpdate = $lastUpdate;
+        return $this;
+    }
+
+    public function getLastUpdate(): ?DateTime
+    {
+        return $this->lastUpdate;
+    }
+
+    public function needsUpdate(int $maxAge): bool
+    {
+        // Update if never updated before
+        if ($this->getLastUpdate() === null) {
+            return true;
+        }
+
+        // Update if max age passed
+        $now = new DateTime;
+        $diff = $now->getTimestamp() - $this->getLastUpdate()->getTimestamp();
+
+        return $diff >= $maxAge;
+    }
+
     /**
      * @return Movie[]
      */
@@ -67,4 +97,15 @@ public function getRuntime(): int
 
         return $runtime;
     }
+
+    public function fetchPosterImages(ImageFetcher $imageFetcher, bool $includeMovies): void
+    {
+        $this->fetchPosterImage($imageFetcher);
+
+        if ($includeMovies) {
+            foreach ($this->getMovies() as $movie) {
+                $movie->fetchPosterImage($imageFetcher);
+            }
+        }
+    }
 }