From db9f06156425782c79a2cfce8d944d867a4be929 Mon Sep 17 00:00:00 2001
From: Bogdan <mynameisbogdan@users.noreply.github.com>
Date: Fri, 20 Oct 2023 00:13:55 +0300
Subject: [PATCH] Return TV category if season/episode detected in title for
 PTP

---
 .../PassThePopcorn/PassThePopcorn.cs          |  2 +-
 .../PassThePopcorn/PassThePopcornParser.cs    | 20 +++++++++++++------
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcorn.cs b/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcorn.cs
index c18a07a36..046e0e250 100644
--- a/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcorn.cs
+++ b/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcorn.cs
@@ -34,7 +34,7 @@ namespace NzbDrone.Core.Indexers.Definitions.PassThePopcorn
 
         public override IParseIndexerResponse GetParser()
         {
-            return new PassThePopcornParser(Settings, _logger);
+            return new PassThePopcornParser(Settings);
         }
 
         private IndexerCapabilities SetCapabilities()
diff --git a/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcornParser.cs b/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcornParser.cs
index c5680d6f1..bc4d11463 100644
--- a/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcornParser.cs
+++ b/src/NzbDrone.Core/Indexers/Definitions/PassThePopcorn/PassThePopcornParser.cs
@@ -2,7 +2,7 @@ using System;
 using System.Collections.Generic;
 using System.Globalization;
 using System.Net;
-using NLog;
+using System.Text.RegularExpressions;
 using NzbDrone.Common.Extensions;
 using NzbDrone.Common.Http;
 using NzbDrone.Common.Serializer;
@@ -14,12 +14,12 @@ namespace NzbDrone.Core.Indexers.Definitions.PassThePopcorn
     public class PassThePopcornParser : IParseIndexerResponse
     {
         private readonly PassThePopcornSettings _settings;
-        private readonly Logger _logger;
 
-        public PassThePopcornParser(PassThePopcornSettings settings, Logger logger)
+        private static Regex SeasonRegex => new (@"\bS\d{2,3}(E\d{2,3})?\b", RegexOptions.Compiled);
+
+        public PassThePopcornParser(PassThePopcornSettings settings)
         {
             _settings = settings;
-            _logger = logger;
         }
 
         public IList<ReleaseInfo> ParseResponse(IndexerResponse indexerResponse)
@@ -57,6 +57,7 @@ namespace NzbDrone.Core.Indexers.Definitions.PassThePopcorn
                 foreach (var torrent in result.Torrents)
                 {
                     var id = torrent.Id;
+                    var title = torrent.ReleaseName;
 
                     var flags = new HashSet<IndexerFlag>();
 
@@ -70,14 +71,21 @@ namespace NzbDrone.Core.Indexers.Definitions.PassThePopcorn
                         flags.Add(PassThePopcornFlag.Approved);
                     }
 
+                    var categories = new List<IndexerCategory> { NewznabStandardCategory.Movies };
+
+                    if (title != null && SeasonRegex.Match(title).Success)
+                    {
+                        categories.Add(NewznabStandardCategory.TV);
+                    }
+
                     torrentInfos.Add(new TorrentInfo
                     {
                         Guid = $"PassThePopcorn-{id}",
-                        Title = torrent.ReleaseName,
+                        Title = title,
                         Year = int.Parse(result.Year),
                         InfoUrl = GetInfoUrl(result.GroupId, id),
                         DownloadUrl = GetDownloadUrl(id, jsonResponse.AuthKey, jsonResponse.PassKey),
-                        Categories = new List<IndexerCategory> { NewznabStandardCategory.Movies },
+                        Categories = categories,
                         Size = long.Parse(torrent.Size),
                         Grabs = int.Parse(torrent.Snatched),
                         Seeders = int.Parse(torrent.Seeders),