mirror of https://github.com/lidarr/Lidarr
51 lines
1.9 KiB
C#
51 lines
1.9 KiB
C#
using System.Collections.Generic;
|
||
using System.Text.RegularExpressions;
|
||
using NzbDrone.Common.EnsureThat;
|
||
using NzbDrone.Common.Extensions;
|
||
using NzbDrone.Core.Music;
|
||
|
||
namespace NzbDrone.Core.IndexerSearch.Definitions
|
||
{
|
||
public abstract class SearchCriteriaBase
|
||
{
|
||
private static readonly Regex SpecialCharacter = new Regex(@"[`'’.]", RegexOptions.IgnoreCase | RegexOptions.Compiled);
|
||
private static readonly Regex NonWord = new Regex(@"[\W]", RegexOptions.IgnoreCase | RegexOptions.Compiled);
|
||
private static readonly Regex BeginningThe = new Regex(@"^the\s", RegexOptions.IgnoreCase | RegexOptions.Compiled);
|
||
|
||
public virtual bool MonitoredEpisodesOnly { get; set; }
|
||
public virtual bool UserInvokedSearch { get; set; }
|
||
public virtual bool InteractiveSearch { get; set; }
|
||
|
||
public Artist Artist { get; set; }
|
||
public List<Album> Albums { get; set; }
|
||
public List<Track> Tracks { get; set; }
|
||
|
||
public string ArtistQuery => Artist.Name;
|
||
public string CleanArtistQuery => GetQueryTitle(ArtistQuery);
|
||
|
||
public static string GetQueryTitle(string title)
|
||
{
|
||
Ensure.That(title, () => title).IsNotNullOrWhiteSpace();
|
||
|
||
// Most VA albums are listed as VA, not Various Artists
|
||
if (title == "Various Artists")
|
||
{
|
||
title = "VA";
|
||
}
|
||
|
||
var cleanTitle = BeginningThe.Replace(title, string.Empty);
|
||
|
||
cleanTitle = cleanTitle.Replace(" & ", " ");
|
||
cleanTitle = SpecialCharacter.Replace(cleanTitle, "");
|
||
cleanTitle = NonWord.Replace(cleanTitle, "+");
|
||
|
||
// remove any repeating +s
|
||
cleanTitle = Regex.Replace(cleanTitle, @"\+{2,}", "+");
|
||
cleanTitle = cleanTitle.RemoveAccent();
|
||
cleanTitle = cleanTitle.Trim('+', ' ');
|
||
|
||
return cleanTitle.Length == 0 ? title : cleanTitle;
|
||
}
|
||
}
|
||
}
|