108 lines
4.7 KiB
PHP
108 lines
4.7 KiB
PHP
<?php
|
|
|
|
namespace App\Monitor\Action\Handler;
|
|
|
|
use Aimeos\Map;
|
|
use App\Monitor\Action\Command\MonitorMovieCommand;
|
|
use App\Monitor\Action\Command\MonitorTvEpisodeCommand;
|
|
use App\Monitor\Action\Result\MonitorTvShowResult;
|
|
use App\Monitor\Framework\Entity\Monitor;
|
|
use App\Monitor\Framework\Repository\MonitorRepository;
|
|
use App\Monitor\Service\MediaFiles;
|
|
use App\Tmdb\Tmdb;
|
|
use DateTimeImmutable;
|
|
use Doctrine\ORM\EntityManagerInterface;
|
|
use Nihilarr\PTN;
|
|
use OneToMany\RichBundle\Contract\CommandInterface;
|
|
use OneToMany\RichBundle\Contract\HandlerInterface;
|
|
use OneToMany\RichBundle\Contract\ResultInterface;
|
|
use Psr\Log\LoggerInterface;
|
|
|
|
/** @implements HandlerInterface<MonitorMovieCommand> */
|
|
readonly class MonitorTvShowHandler implements HandlerInterface
|
|
{
|
|
public function __construct(
|
|
private MonitorRepository $monitorRepository,
|
|
private EntityManagerInterface $entityManager,
|
|
private MonitorTvEpisodeHandler $monitorTvEpisodeHandler,
|
|
private MediaFiles $mediaFiles,
|
|
private LoggerInterface $logger,
|
|
private Tmdb $tmdb,
|
|
) {}
|
|
|
|
public function handle(CommandInterface $command): ResultInterface
|
|
{
|
|
$this->logger->info('> [MonitorTvShowHandler] Executing MonitorTvShowHandler');
|
|
$monitor = $this->monitorRepository->find($command->monitorId);
|
|
|
|
// Check current episodes
|
|
$downloadedEpisodes = $this->mediaFiles
|
|
->getEpisodes($monitor->getTitle())
|
|
->map(fn($episode) => (object) (new PTN())->parse($episode));
|
|
$this->logger->info('> [MonitorTvShowHandler] Found ' . count($downloadedEpisodes) . ' downloaded episodes for title: ' . $monitor->getTitle());
|
|
|
|
// Compare against list from TMDB
|
|
$episodesInShow = Map::from(
|
|
$this->tmdb->tvDetails($monitor->getTmdbId())->episodes
|
|
)->flat(1);
|
|
$this->logger->info('> [MonitorTvShowHandler] Found ' . count($episodesInShow) . ' episodes for title: ' . $monitor->getTitle());
|
|
|
|
if ($downloadedEpisodes->count() !== $episodesInShow->count()) {
|
|
// Since $monitor has children monitors, set the status
|
|
// to Active, so it will be re-executed.
|
|
$monitor->setStatus('Active');
|
|
|
|
// Dispatch Episode commands for each missing Episode
|
|
foreach ($episodesInShow as $episode) {
|
|
$monitorCheck = $this->monitorRepository->findOneBy([
|
|
'imdbId' => $monitor->getImdbId(),
|
|
'title' => $monitor->getTitle(),
|
|
'monitorType' => 'tvepisode',
|
|
'season' => $monitor->getSeason(),
|
|
'episode' => $episode['episode_number'],
|
|
'status' => ['New', 'Active', 'In Progress']
|
|
]);
|
|
|
|
$this->logger->info('> [MonitorTvShowHandler] Monitor exists for season ' . $monitor->getSeason() . ' episode ' . $episode['episode_number'] . ' for title: ' . $monitor->getTitle() . ' ? ' . (null !== $monitorCheck ? 'YES' : 'NO'));
|
|
|
|
if (!array_key_exists($episode['episode_number'], $downloadedEpisodes->toArray())
|
|
&& null === $monitorCheck
|
|
) {
|
|
$episodeMonitor = (new Monitor())
|
|
->setParent($monitor)
|
|
->setUser($monitor->getUser())
|
|
->setTmdbId($monitor->getTmdbId())
|
|
->setImdbId($monitor->getImdbId())
|
|
->setTitle($monitor->getTitle())
|
|
->setMonitorType('tvepisode')
|
|
->setSeason($monitor->getSeason())
|
|
->setEpisode($episode['episode_number'])
|
|
->setCreatedAt(new DateTimeImmutable())
|
|
->setSearchCount(0)
|
|
->setStatus('New');
|
|
|
|
$this->monitorRepository->getEntityManager()->persist($episodeMonitor);
|
|
$this->monitorRepository->getEntityManager()->flush();
|
|
|
|
$command = new MonitorTvEpisodeCommand($episodeMonitor->getId());
|
|
$this->monitorTvEpisodeHandler->handle($command);
|
|
$this->logger->info('> [MonitorTvShowHandler] Dispatching MonitorTvEpisodeCommand for season ' . $episodeMonitor->getSeason() . ' episode ' . $episodeMonitor->getEpisode() . ' for title: ' . $monitor->getTitle());
|
|
}
|
|
}
|
|
} else {
|
|
$monitor->setStatus('Complete');
|
|
}
|
|
|
|
$monitor->setLastSearch(new DateTimeImmutable());
|
|
$monitor->setSearchCount($monitor->getSearchCount() + 1);
|
|
$this->entityManager->flush();
|
|
|
|
return new MonitorTvShowResult(
|
|
status: 'OK',
|
|
result: [
|
|
'monitor' => $monitor,
|
|
]
|
|
);
|
|
}
|
|
}
|