Radarr/src/NzbDrone.Core/IndexerSearch/Definitions/SearchCriteriaBase.cs

40 lines
1.6 KiB
C#
Raw Normal View History

using System.Collections.Generic;
using System.Linq;
2013-04-07 07:30:37 +00:00
using System.Text.RegularExpressions;
using NzbDrone.Common.EnsureThat;
using NzbDrone.Common.Extensions;
using NzbDrone.Core.Tv;
2013-04-07 07:30:37 +00:00
namespace NzbDrone.Core.IndexerSearch.Definitions
{
public abstract class SearchCriteriaBase
2013-04-07 07:30:37 +00:00
{
private static readonly Regex SpecialCharacter = new Regex(@"[`'.]", RegexOptions.IgnoreCase | RegexOptions.Compiled);
private static readonly Regex NonWord = new Regex(@"[\W]", RegexOptions.IgnoreCase | RegexOptions.Compiled);
2013-04-07 07:30:37 +00:00
private static readonly Regex BeginningThe = new Regex(@"^the\s", RegexOptions.IgnoreCase | RegexOptions.Compiled);
public Series Series { get; set; }
public List<string> SceneTitles { get; set; }
public List<Episode> Episodes { get; set; }
public virtual bool MonitoredEpisodesOnly { get; set; }
public virtual bool UserInvokedSearch { get; set; }
2013-04-07 07:30:37 +00:00
2016-12-09 06:54:15 +00:00
public List<string> QueryTitles => SceneTitles.Select(GetQueryTitle).ToList();
2013-04-07 07:30:37 +00:00
public static string GetQueryTitle(string title)
2013-04-07 07:30:37 +00:00
{
2013-11-30 23:53:07 +00:00
Ensure.That(title,() => title).IsNotNullOrWhiteSpace();
var cleanTitle = BeginningThe.Replace(title, string.Empty);
2013-04-07 07:30:37 +00:00
cleanTitle = cleanTitle.Replace("&", "and");
cleanTitle = SpecialCharacter.Replace(cleanTitle, "");
cleanTitle = NonWord.Replace(cleanTitle, "+");
2013-04-07 07:30:37 +00:00
//remove any repeating +s
cleanTitle = Regex.Replace(cleanTitle, @"\+{2,}", "+");
cleanTitle = cleanTitle.RemoveAccent();
2013-04-07 07:30:37 +00:00
return cleanTitle.Trim('+', ' ');
}
}
}