Remove Remaining TV Code

2026-04-22 22:14:44 -04:00 · 2017-10-18 21:42:57 -04:00
parent 81716762d8
commit a80360f6fd
98 changed files with 135 additions and 6151 deletions
@@ -0,0 +1,87 @@
+using System;
+using System.Collections.Generic;
+using System.Text.RegularExpressions;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Music;
+
+namespace NzbDrone.Core.MetadataSource
+{
+    public class SearchArtistComparer : IComparer<Artist>
+    {
+        private static readonly Regex RegexCleanPunctuation = new Regex("[-._:]", RegexOptions.Compiled);
+        private static readonly Regex RegexCleanCountryYearPostfix = new Regex(@"(?<=.+)( \([A-Z]{2}\)| \(\d{4}\)| \([A-Z]{2}\) \(\d{4}\))$", RegexOptions.Compiled);
+        private static readonly Regex ArticleRegex = new Regex(@"^(a|an|the)\s", RegexOptions.IgnoreCase | RegexOptions.Compiled);
+
+        public string SearchQuery { get; private set; }
+
+        private readonly string _searchQueryWithoutYear;
+        private int? _year;
+
+        public SearchArtistComparer(string searchQuery)
+        {
+            SearchQuery = searchQuery;
+            
+            var match = Regex.Match(SearchQuery, @"^(?<query>.+)\s+(?:\((?<year>\d{4})\)|(?<year>\d{4}))$");
+            if (match.Success)
+            {
+                _searchQueryWithoutYear = match.Groups["query"].Value.ToLowerInvariant();
+                _year = int.Parse(match.Groups["year"].Value);
+            }
+            else
+            {
+                _searchQueryWithoutYear = searchQuery.ToLowerInvariant();
+            }
+        }
+
+        public int Compare(Artist x, Artist y)
+        {
+            int result = 0;
+
+            // Prefer exact matches
+            result = Compare(x, y, s => CleanPunctuation(s.Name).Equals(CleanPunctuation(SearchQuery)));
+            if (result != 0) return -result;
+
+            // Remove Articles (a/an/the)
+            result = Compare(x, y, s => CleanArticles(s.Name).Equals(CleanArticles(SearchQuery)));
+            if (result != 0) return -result;
+
+            // Prefer close matches
+            result = Compare(x, y, s => CleanPunctuation(s.Name).LevenshteinDistance(CleanPunctuation(SearchQuery)) <= 1);
+            if (result != 0) return -result;
+         
+            return Compare(x, y, s => SearchQuery.LevenshteinDistanceClean(s.Name));
+        }
+        
+        public int Compare<T>(Artist x, Artist y, Func<Artist, T> keySelector)
+            where T : IComparable<T>
+        {
+            var keyX = keySelector(x);
+            var keyY = keySelector(y);
+
+            return keyX.CompareTo(keyY);
+        }
+
+        private string CleanPunctuation(string title)
+        {
+            title = RegexCleanPunctuation.Replace(title, "");
+
+            return title.ToLowerInvariant();
+        }
+
+        private string CleanTitle(string title)
+        {
+            title = RegexCleanPunctuation.Replace(title, "");
+            title = RegexCleanCountryYearPostfix.Replace(title, "");
+
+            return title.ToLowerInvariant();
+        }
+
+        private string CleanArticles(string title)
+        {
+            title = ArticleRegex.Replace(title, "");
+
+            return title.Trim().ToLowerInvariant();
+        }
+
+    }
+}