New: Custom Formats

Co-Authored-By: ta264 <ta264@users.noreply.github.com>
2026-04-20 21:54:58 -04:00 · 2022-01-23 23:42:41 -06:00
parent 909af6c874
commit b04b4000b8
173 changed files with 6401 additions and 1347 deletions
@@ -0,0 +1,148 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Blocklisting;
+using NzbDrone.Core.History;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Parser;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.CustomFormats
+{
+    public interface ICustomFormatCalculationService
+    {
+        List<CustomFormat> ParseCustomFormat(ParsedEpisodeInfo movieInfo);
+        List<CustomFormat> ParseCustomFormat(EpisodeFile movieFile);
+        List<CustomFormat> ParseCustomFormat(Blocklist blocklist);
+        List<CustomFormat> ParseCustomFormat(EpisodeHistory history);
+    }
+
+    public class CustomFormatCalculationService : ICustomFormatCalculationService
+    {
+        private readonly ICustomFormatService _formatService;
+        private readonly IParsingService _parsingService;
+        private readonly ISeriesService _seriesService;
+
+        public CustomFormatCalculationService(ICustomFormatService formatService,
+                                              IParsingService parsingService,
+                                              ISeriesService seriesService)
+        {
+            _formatService = formatService;
+            _parsingService = parsingService;
+            _seriesService = seriesService;
+        }
+
+        public static List<CustomFormat> ParseCustomFormat(ParsedEpisodeInfo episodeInfo, List<CustomFormat> allCustomFormats)
+        {
+            var matches = new List<CustomFormat>();
+
+            foreach (var customFormat in allCustomFormats)
+            {
+                var specificationMatches = customFormat.Specifications
+                    .GroupBy(t => t.GetType())
+                    .Select(g => new SpecificationMatchesGroup
+                    {
+                        Matches = g.ToDictionary(t => t, t => t.IsSatisfiedBy(episodeInfo))
+                    })
+                    .ToList();
+
+                if (specificationMatches.All(x => x.DidMatch))
+                {
+                    matches.Add(customFormat);
+                }
+            }
+
+            return matches;
+        }
+
+        public static List<CustomFormat> ParseCustomFormat(EpisodeFile episodeFile, List<CustomFormat> allCustomFormats)
+        {
+            var sceneName = string.Empty;
+            if (episodeFile.SceneName.IsNotNullOrWhiteSpace())
+            {
+                sceneName = episodeFile.SceneName;
+            }
+            else if (episodeFile.OriginalFilePath.IsNotNullOrWhiteSpace())
+            {
+                sceneName = episodeFile.OriginalFilePath;
+            }
+            else if (episodeFile.RelativePath.IsNotNullOrWhiteSpace())
+            {
+                sceneName = Path.GetFileName(episodeFile.RelativePath);
+            }
+
+            var info = new ParsedEpisodeInfo
+            {
+                SeriesTitle = episodeFile.Series.Value.Title,
+                ReleaseTitle  = sceneName,
+                Quality = episodeFile.Quality,
+                Language = episodeFile.Language,
+                ReleaseGroup = episodeFile.ReleaseGroup,
+                ExtraInfo = new Dictionary<string, object>
+                {
+                    { "Size", episodeFile.Size },
+                    { "Filename", Path.GetFileName(episodeFile.RelativePath) }
+                }
+            };
+
+            return ParseCustomFormat(info, allCustomFormats);
+        }
+
+        public List<CustomFormat> ParseCustomFormat(ParsedEpisodeInfo movieInfo)
+        {
+            return ParseCustomFormat(movieInfo, _formatService.All());
+        }
+
+        public List<CustomFormat> ParseCustomFormat(EpisodeFile wpisodeFile)
+        {
+            return ParseCustomFormat(wpisodeFile, _formatService.All());
+        }
+
+        public List<CustomFormat> ParseCustomFormat(Blocklist blocklist)
+        {
+            var movie = _seriesService.GetSeries(blocklist.SeriesId);
+            var parsed = Parser.Parser.ParseTitle(blocklist.SourceTitle);
+
+            var info = new ParsedEpisodeInfo
+            {
+                SeriesTitle = movie.Title,
+                ReleaseTitle = parsed?.ReleaseTitle ?? blocklist.SourceTitle,
+                Quality = blocklist.Quality,
+                Language = blocklist.Language,
+                ReleaseGroup = parsed?.ReleaseGroup,
+                ExtraInfo = new Dictionary<string, object>
+                {
+                    { "Size", blocklist.Size }
+                }
+            };
+
+            return ParseCustomFormat(info);
+        }
+
+        public List<CustomFormat> ParseCustomFormat(EpisodeHistory history)
+        {
+            var movie = _seriesService.GetSeries(history.SeriesId);
+            var parsed = Parser.Parser.ParseTitle(history.SourceTitle);
+
+            long.TryParse(history.Data.GetValueOrDefault("size"), out var size);
+
+            var info = new ParsedEpisodeInfo
+            {
+                SeriesTitle = movie.Title,
+                ReleaseTitle = parsed?.ReleaseTitle ?? history.SourceTitle,
+                Quality = history.Quality,
+                Language = history.Language,
+                ReleaseGroup = parsed?.ReleaseGroup,
+                ExtraInfo = new Dictionary<string, object>
+                {
+                    { "Size", size }
+                }
+            };
+
+            return ParseCustomFormat(info);
+        }
+    }
+}