Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b741239194 | ||
|
|
96f8ff9332 | ||
|
|
e739affb39 | ||
|
|
92bdf14618 |
@@ -184,7 +184,7 @@ namespace UmlautAdaptarr.Controllers
|
||||
if (categories.Split(',').Any(category => AUDIO_CATEGORY_IDS.Contains(category)))
|
||||
{
|
||||
var mediaType = "audio";
|
||||
searchItem = await searchItemLookupService.GetOrFetchSearchItemByExternalId(mediaType, title.ToLower());
|
||||
searchItem = await searchItemLookupService.GetOrFetchSearchItemByExternalId(mediaType, title.GetLidarrTitleForExternalId());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -12,6 +12,7 @@ namespace UmlautAdaptarr.Models
|
||||
public bool HasUmlaut => Title?.HasUmlauts() ?? false;
|
||||
public string ExpectedTitle { get; set; }
|
||||
public string? ExpectedAuthor { get; set; }
|
||||
// TODO rename GermanTitle into Foreign or LocalTitle?
|
||||
public string? GermanTitle { get; set; }
|
||||
public string[] TitleSearchVariations { get; set; }
|
||||
public string[] TitleMatchVariations { get; set; }
|
||||
@@ -42,6 +43,11 @@ namespace UmlautAdaptarr.Models
|
||||
if (expectedTitle.Contains(expectedAuthor))
|
||||
{
|
||||
var titleWithoutAuthorName = expectedTitle.Replace(expectedAuthor, string.Empty).RemoveExtraWhitespaces().Trim();
|
||||
|
||||
if (titleWithoutAuthorName.Length < 2)
|
||||
{
|
||||
// TODO log warning that this album can't be searched for automatically
|
||||
}
|
||||
TitleMatchVariations = GenerateVariations(titleWithoutAuthorName, mediaType).ToArray();
|
||||
}
|
||||
else
|
||||
@@ -86,13 +92,18 @@ namespace UmlautAdaptarr.Models
|
||||
}
|
||||
}
|
||||
|
||||
private IEnumerable<string> GenerateVariations(string? germanTitle, string mediaType)
|
||||
private IEnumerable<string> GenerateVariations(string? title, string mediaType)
|
||||
{
|
||||
if (germanTitle == null)
|
||||
if (title == null)
|
||||
{
|
||||
return [];
|
||||
}
|
||||
var cleanTitle = title.GetCleanTitle();
|
||||
|
||||
if (cleanTitle?.Length == 0)
|
||||
{
|
||||
return [];
|
||||
}
|
||||
var cleanTitle = germanTitle.RemoveAccentButKeepGermanUmlauts().GetCleanTitle();
|
||||
|
||||
// Start with base variations including handling umlauts
|
||||
var baseVariations = new List<string>
|
||||
@@ -121,11 +132,16 @@ namespace UmlautAdaptarr.Models
|
||||
});
|
||||
}
|
||||
|
||||
// If a german title starts with der/die/das also accept variations without it
|
||||
if (mediaType != "audio" && cleanTitle.StartsWith("Der") || cleanTitle.StartsWith("Die") || cleanTitle.StartsWith("Das"))
|
||||
// If a title starts with der/die/das also accept variations without it
|
||||
// Same for english the, an, a
|
||||
if (cleanTitle.StartsWith("Der ") || cleanTitle.StartsWith("Die ") || cleanTitle.StartsWith("Das ")
|
||||
|| cleanTitle.StartsWith("The ") || cleanTitle.StartsWith("An "))
|
||||
{
|
||||
var cleanTitleWithoutArticle = germanTitle[3..].Trim();
|
||||
var cleanTitleWithoutArticle = title[3..].Trim();
|
||||
baseVariations.AddRange(GenerateVariations(cleanTitleWithoutArticle, mediaType));
|
||||
} else if (cleanTitle.StartsWith("A "))
|
||||
{
|
||||
var cleanTitleWithoutArticle = title[2..].Trim();
|
||||
}
|
||||
|
||||
// Remove multiple spaces
|
||||
|
||||
@@ -42,16 +42,17 @@ namespace UmlautAdaptarr.Providers
|
||||
|
||||
var lidarrAlbumUrl = $"{_lidarrHost}/api/v1/album?artistId={artistId}&apikey={_lidarrApiKey}";
|
||||
|
||||
if (cache.TryGetValue(lidarrAlbumUrl, out List<dynamic>? albums))
|
||||
{
|
||||
logger.LogInformation($"Using cached albums for {UrlUtilities.RedactApiKey(lidarrAlbumUrl)}");
|
||||
}
|
||||
else
|
||||
{
|
||||
// Disable cache for now as it can result in problems when adding new albums that aren't displayed on the artists page initially
|
||||
//if (cache.TryGetValue(lidarrAlbumUrl, out List<dynamic>? albums))
|
||||
//{
|
||||
// logger.LogInformation($"Using cached albums for {UrlUtilities.RedactApiKey(lidarrAlbumUrl)}");
|
||||
//}
|
||||
//else
|
||||
//{
|
||||
logger.LogInformation($"Fetching all albums from artistId {artistId} from Lidarr: {UrlUtilities.RedactApiKey(lidarrArtistsUrl)}");
|
||||
var albumApiResponse = await httpClient.GetStringAsync(lidarrAlbumUrl);
|
||||
albums = JsonConvert.DeserializeObject<List<dynamic>>(albumApiResponse);
|
||||
}
|
||||
var albums = JsonConvert.DeserializeObject<List<dynamic>>(albumApiResponse);
|
||||
//}
|
||||
|
||||
if (albums == null)
|
||||
{
|
||||
@@ -74,7 +75,7 @@ namespace UmlautAdaptarr.Providers
|
||||
string[]? aliases = null;
|
||||
|
||||
// Abuse externalId to set the search string Lidarr uses
|
||||
var externalId = expectedTitle.RemoveGermanUmlautDots().RemoveAccent().RemoveSpecialCharacters().RemoveExtraWhitespaces().ToLower();
|
||||
var externalId = expectedTitle.GetLidarrTitleForExternalId();
|
||||
|
||||
var searchItem = new SearchItem
|
||||
(
|
||||
|
||||
@@ -84,7 +84,7 @@ namespace UmlautAdaptarr.Services
|
||||
continue;
|
||||
}
|
||||
// After finding a potential item, compare normalizedTitle with each German title variation
|
||||
foreach (var variation in item?.TitleSearchVariations ?? [])
|
||||
foreach (var variation in item?.TitleMatchVariations ?? [])
|
||||
{
|
||||
var normalizedVariation = variation.RemoveAccentButKeepGermanUmlauts().ToLower();
|
||||
if (normalizedTitle.StartsWith(variation, StringComparison.OrdinalIgnoreCase))
|
||||
|
||||
@@ -20,7 +20,7 @@ namespace UmlautAdaptarr.Services
|
||||
if (titleElement != null)
|
||||
{
|
||||
var originalTitle = titleElement.Value;
|
||||
var normalizedOriginalTitle = NormalizeTitle(originalTitle);
|
||||
var cleanTitleSeperatedBySpace = ReplaceSeperatorsWithSpace(originalTitle.RemoveAccentButKeepGermanUmlauts());
|
||||
|
||||
var categoryElement = item.Element("category");
|
||||
var category = categoryElement?.Value;
|
||||
@@ -34,7 +34,7 @@ namespace UmlautAdaptarr.Services
|
||||
if (useCacheService)
|
||||
{
|
||||
// Use CacheService to find a matching SearchItem by title
|
||||
searchItem = cacheService.SearchItemByTitle(mediaType, normalizedOriginalTitle);
|
||||
searchItem = cacheService.SearchItemByTitle(mediaType, cleanTitleSeperatedBySpace);
|
||||
}
|
||||
|
||||
if (searchItem == null)
|
||||
@@ -46,10 +46,10 @@ namespace UmlautAdaptarr.Services
|
||||
switch (mediaType)
|
||||
{
|
||||
case "tv":
|
||||
FindAndReplaceForMoviesAndTV(logger, searchItem, titleElement, originalTitle, normalizedOriginalTitle!);
|
||||
FindAndReplaceForMoviesAndTV(logger, searchItem, titleElement, originalTitle, cleanTitleSeperatedBySpace!);
|
||||
break;
|
||||
case "movie":
|
||||
FindAndReplaceForMoviesAndTV(logger, searchItem, titleElement, originalTitle, normalizedOriginalTitle!);
|
||||
FindAndReplaceForMoviesAndTV(logger, searchItem, titleElement, originalTitle, cleanTitleSeperatedBySpace!);
|
||||
break;
|
||||
case "audio":
|
||||
FindAndReplaceForAudio(searchItem, titleElement, originalTitle!);
|
||||
@@ -68,13 +68,12 @@ namespace UmlautAdaptarr.Services
|
||||
var authorMatch = FindBestMatch(searchItem.AuthorMatchVariations, originalTitle.NormalizeForComparison(), originalTitle);
|
||||
var titleMatch = FindBestMatch(searchItem.TitleMatchVariations, originalTitle.NormalizeForComparison(), originalTitle);
|
||||
|
||||
if (authorMatch.Item1 && titleMatch.Item1)
|
||||
if (authorMatch.foundMatch && titleMatch.foundMatch)
|
||||
{
|
||||
int matchEndPositionInOriginal = Math.Max(authorMatch.Item3, titleMatch.Item3);
|
||||
int matchEndPositionInOriginal = Math.Max(authorMatch.bestEndInOriginal, titleMatch.bestEndInOriginal);
|
||||
|
||||
var test = originalTitle[matchEndPositionInOriginal];
|
||||
// Check and adjust for immediate following delimiter
|
||||
char[] delimiters = new char[] { ' ', '-', '_', '.' };
|
||||
char[] delimiters = [' ', '-', '_', '.'];
|
||||
if (matchEndPositionInOriginal < originalTitle.Length && delimiters.Contains(originalTitle[matchEndPositionInOriginal]))
|
||||
{
|
||||
matchEndPositionInOriginal++; // Skip the delimiter if it's immediately after the match
|
||||
@@ -97,7 +96,7 @@ namespace UmlautAdaptarr.Services
|
||||
}
|
||||
|
||||
|
||||
private Tuple<bool, int, int> FindBestMatch(string[] variations, string normalizedOriginal, string originalTitle)
|
||||
private (bool foundMatch, int bestStart, int bestEndInOriginal) FindBestMatch(string[] variations, string normalizedOriginal, string originalTitle)
|
||||
{
|
||||
bool found = false;
|
||||
int bestStart = int.MaxValue;
|
||||
@@ -120,8 +119,8 @@ namespace UmlautAdaptarr.Services
|
||||
}
|
||||
}
|
||||
|
||||
if (!found) return Tuple.Create(false, 0, 0);
|
||||
return Tuple.Create(found, bestStart, bestEndInOriginal);
|
||||
if (!found) return (false, 0, 0);
|
||||
return (found, bestStart, bestEndInOriginal);
|
||||
}
|
||||
|
||||
// Maps an index from the normalized string back to a corresponding index in the original string
|
||||
@@ -160,6 +159,7 @@ namespace UmlautAdaptarr.Services
|
||||
var titleMatchVariations = searchItem.TitleMatchVariations;
|
||||
var expectedTitle = searchItem.ExpectedTitle;
|
||||
var variationsOrderedByLength = titleMatchVariations!.OrderByDescending(variation => variation.Length);
|
||||
|
||||
// Attempt to find a variation that matches the start of the original title
|
||||
foreach (var variation in variationsOrderedByLength)
|
||||
{
|
||||
@@ -199,8 +199,8 @@ namespace UmlautAdaptarr.Services
|
||||
}
|
||||
}
|
||||
|
||||
// Clean up any leading separators from the suffix
|
||||
suffix = Regex.Replace(suffix, "^[._ ]+", "");
|
||||
// Clean up any leading separator from the suffix
|
||||
suffix = Regex.Replace(suffix, "^ +", "");
|
||||
|
||||
// TODO EVALUTE! definitely make this optional - this adds GERMAN to the title is the title is german to make sure it's recognized as german
|
||||
// can lead to problems with shows such as "dark" that have international dubs
|
||||
@@ -226,9 +226,8 @@ namespace UmlautAdaptarr.Services
|
||||
}
|
||||
}
|
||||
|
||||
private static string NormalizeTitle(string title)
|
||||
private static string ReplaceSeperatorsWithSpace(string title)
|
||||
{
|
||||
title = title.RemoveAccentButKeepGermanUmlauts();
|
||||
// Replace all known separators with space for normalization
|
||||
return WordSeperationCharRegex().Replace(title, " ".ToString());
|
||||
}
|
||||
|
||||
@@ -47,20 +47,46 @@ namespace UmlautAdaptarr.Utilities
|
||||
return stringBuilder.ToString().Normalize(NormalizationForm.FormC);
|
||||
}
|
||||
|
||||
// TODO possibly replace GetCleanTitle with RemoveSpecialCharacters
|
||||
public static string GetLidarrTitleForExternalId(this string text)
|
||||
{
|
||||
text = text.RemoveGermanUmlautDots()
|
||||
.GetCleanTitle()
|
||||
.ToLower();
|
||||
|
||||
// Lidarr removes the, an and a
|
||||
return TitlePrefixRegex()
|
||||
.Replace(text, "")
|
||||
.RemoveExtraWhitespaces()
|
||||
.Trim();
|
||||
}
|
||||
|
||||
public static string GetCleanTitle(this string text)
|
||||
{
|
||||
return text.Replace("(", "").Replace(")", "").Replace("?","").Replace(":", "").Replace("'", "");
|
||||
return text
|
||||
.Replace(".", " ")
|
||||
.Replace(":", " ")
|
||||
.RemoveAccentButKeepGermanUmlauts()
|
||||
.RemoveSpecialCharacters(removeUmlauts: false)
|
||||
.RemoveExtraWhitespaces()
|
||||
.Trim();
|
||||
}
|
||||
|
||||
public static string NormalizeForComparison(this string text)
|
||||
{
|
||||
// TODO see if we can replace RemoveGermanUmlautDots() with RemoveSpecialCharacters(removeUmlauts: false);
|
||||
return text.RemoveGermanUmlautDots().RemoveAccent().RemoveSpecialCharacters().Replace(" ", "").Trim().ToLower();
|
||||
}
|
||||
|
||||
public static string RemoveSpecialCharacters(this string text)
|
||||
public static string RemoveSpecialCharacters(this string text, bool removeUmlauts = true)
|
||||
{
|
||||
return SpecialCharactersRegex().Replace(text, "");
|
||||
if (removeUmlauts)
|
||||
{
|
||||
return NoSpecialCharactersRegex().Replace(text, "");
|
||||
}
|
||||
else
|
||||
{
|
||||
return NoSpecialCharactersExceptUmlautsRegex().Replace(text, "");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -101,9 +127,15 @@ namespace UmlautAdaptarr.Utilities
|
||||
}
|
||||
|
||||
[GeneratedRegex("[^a-zA-Z0-9 ]+", RegexOptions.Compiled)]
|
||||
private static partial Regex SpecialCharactersRegex();
|
||||
private static partial Regex NoSpecialCharactersRegex();
|
||||
|
||||
[GeneratedRegex("[^a-zA-Z0-9 öäüßÖÄÜ]+", RegexOptions.Compiled)]
|
||||
private static partial Regex NoSpecialCharactersExceptUmlautsRegex();
|
||||
|
||||
[GeneratedRegex(@"\s+")]
|
||||
private static partial Regex MultipleWhitespaceRegex();
|
||||
|
||||
[GeneratedRegex(@"\b(the|an|a)\b", RegexOptions.IgnoreCase, "de-DE")]
|
||||
private static partial Regex TitlePrefixRegex();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user