From dfc11d7a7fe79e9d2d457605205de4c18b61ed37 Mon Sep 17 00:00:00 2001
From: Dan Jones <danjones@goodevilgenius.org>
Date: Sun, 8 Oct 2023 23:01:28 -0500
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20Export=20films?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/Data/DataPipes/GetSeasonEp.php    | 42 ++++++++++++++++++++++-----
 app/Data/DataPipes/GetShowTmdb.php    |  4 +++
 app/Data/DataPipes/ParseWatchFile.php | 22 ++++++++++----
 app/Data/Enums/Type.php               | 11 +++++++
 app/Data/WatchFile.php                |  6 ++++
 5 files changed, 71 insertions(+), 14 deletions(-)
 create mode 100644 app/Data/Enums/Type.php

diff --git a/app/Data/DataPipes/GetSeasonEp.php b/app/Data/DataPipes/GetSeasonEp.php
index 57c713c..e62dd3f 100644
--- a/app/Data/DataPipes/GetSeasonEp.php
+++ b/app/Data/DataPipes/GetSeasonEp.php
@@ -1,7 +1,10 @@
 <?php
 
+declare(strict_types=1);
+
 namespace App\Data\DataPipes;
 
+use App\Data\Enums\Type;
 use FFMpeg\FFMpeg;
 use Illuminate\Support\Collection;
 use SimpleXMLElement;
@@ -23,26 +26,49 @@ class GetSeasonEp implements DataPipe
         $nfo = "{$pi['dirname']}/{$pi['filename']}.nfo";
         if (file_exists($nfo)) {
             $properties['epNfo'] = $nfo;
+            $xml = simplexml_load_file($nfo);
+            $type = $xml->getName();
 
-            return $this->getEpFromNfo($properties, $nfo);
+            if ($type === 'episodedetails') {
+                return $this->getEpFromNfo($properties, $xml);
+            } else if ($type === 'movie') {
+                return $this->getFilmFromNfo($properties, $xml);
+            }
         }
 
         return $this->getEpFromFfprobe($properties);
     }
 
-    protected function getEpFromNfo(Collection $properties, string $nfo): Collection
+    protected function getFilmFromNfo(Collection $properties, SimpleXMLElement $xml): Collection
     {
-        $xml = simplexml_load_file($nfo);
-        if ($xml->season) {
-            $properties['season'] = (int) $xml->season;
-        }
-        if ($xml->episode) {
-            $properties['episode'] = (int) $xml->episode;
+        $properties['type'] = Type::Movie;
+        $properties['showTitle'] = $this->getProp($xml, 'title');
+        $properties['movieTmdb'] = $this->getProp($xml, 'tmdbid');
+        $properties['movieImdb'] = $this->getProp($xml, 'imdbid');
+        if (!is_null($year = $this->getProp($xml, 'year'))) {
+            $properties['movieYear'] = (int) $year;
         }
 
         return $properties;
     }
 
+    protected function getEpFromNfo(Collection $properties, SimpleXMLElement $xml): Collection
+    {
+        if (!is_null($seas = $this->getProp($xml, 'season'))) {
+            $properties['season'] = (int) $seas;
+        }
+        if (!is_null($ep = $this->getProp($xml, 'episode'))) {
+            $properties['episode'] = (int) $ep;
+        }
+
+        return $properties;
+    }
+
+    protected function getProp(SimpleXMLElement $xml, string $prop): ?string
+    {
+        return $xml->$prop ? (string) $xml->$prop : null;
+    }
+
     protected function getEpFromFfprobe(Collection $properties): Collection
     {
         $ffmpeg  = FFMpeg::create();
diff --git a/app/Data/DataPipes/GetShowTmdb.php b/app/Data/DataPipes/GetShowTmdb.php
index 192d729..154c2b1 100644
--- a/app/Data/DataPipes/GetShowTmdb.php
+++ b/app/Data/DataPipes/GetShowTmdb.php
@@ -1,7 +1,10 @@
 <?php
 
+declare(strict_types=1);
+
 namespace App\Data\DataPipes;
 
+use App\Data\Enums\Type;
 use Illuminate\Support\Collection;
 use SimpleXMLElement;
 use Spatie\LaravelData\DataPipes\DataPipe;
@@ -28,6 +31,7 @@ class GetShowTmdb implements DataPipe
         }
 
         $properties['showNfo'] = $nfo;
+        $properties['type'] = Type::Episode;
 
         $xml = simplexml_load_file($nfo);
         $properties['showTmdb'] = $this->getProp($xml, 'tmdbid');
diff --git a/app/Data/DataPipes/ParseWatchFile.php b/app/Data/DataPipes/ParseWatchFile.php
index c625eaa..14ac0d9 100644
--- a/app/Data/DataPipes/ParseWatchFile.php
+++ b/app/Data/DataPipes/ParseWatchFile.php
@@ -1,7 +1,10 @@
 <?php
 
+declare(strict_types=1);
+
 namespace App\Data\DataPipes;
 
+use App\Data\Enums\Type;
 use App\Data\FileData;
 use Illuminate\Support\Carbon;
 use Illuminate\Support\Collection;
@@ -18,12 +21,19 @@ class ParseWatchFile implements DataPipe
     public function handle(mixed $payload, DataClass $class, Collection $properties): Collection
     {
         $properties['watched'] = Carbon::now();
-        $properties['output'] = sprintf(
-            '%s-%dx%02d.json',
-            Str::slug($properties['showTitle'] ?? uniqid()),
-            $properties['season'],
-            $properties['episode']
-        );
+        $properties['output'] = match ($properties['type']) {
+            Type::Movie => sprintf(
+                '%s-%d.json',
+                Str::slug($properties['showTitle'] ?? uniqid()),
+                $properties['movieYear'],
+            ),
+            default => sprintf(
+                '%s-%dx%02d.json',
+                Str::slug($properties['showTitle'] ?? uniqid()),
+                $properties['season'],
+                $properties['episode']
+            ),
+        };
 
         return $properties;
     }
diff --git a/app/Data/Enums/Type.php b/app/Data/Enums/Type.php
new file mode 100644
index 0000000..3d2f302
--- /dev/null
+++ b/app/Data/Enums/Type.php
@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+namespace App\Data\Enums;
+
+enum Type: string {
+    case Unknown = "unknown";
+    case Episode = "episode";
+    case Movie = "movie";
+}
diff --git a/app/Data/WatchFile.php b/app/Data/WatchFile.php
index 8495324..4e23739 100644
--- a/app/Data/WatchFile.php
+++ b/app/Data/WatchFile.php
@@ -1,5 +1,7 @@
 <?php
 
+declare(strict_types=1);
+
 namespace App\Data;
 
 use Illuminate\Support\Carbon;
@@ -15,10 +17,14 @@ class WatchFile extends Data
     public ?string $showTmdb = null;
     public ?string $showImdb = null;
     public ?string $showTitle = null;
+    public ?string $movieTmdb = null;
+    public ?string $movieImdb = null;
+    public int $movieYear = 1900;
     public ?string $epNfo = null;
     public int $season = 0;
     public int $episode = 0;
     public Carbon $watched;
+    public Enums\Type $type = Enums\Type::Unknown;
 
     public static function fromPath(string $path): static
     {