jellyfin/Emby.Naming/Video/CleanDateTimeParser.cs

90 lines
2.9 KiB
C#
Raw Normal View History

#pragma warning disable CS1591
#pragma warning disable SA1600
using System;
2018-09-12 19:26:21 +02:00
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text.RegularExpressions;
2019-01-13 20:17:29 +01:00
using Emby.Naming.Common;
2018-09-12 19:26:21 +02:00
namespace Emby.Naming.Video
{
/// <summary>
2019-10-25 12:47:20 +02:00
/// <see href="http://kodi.wiki/view/Advancedsettings.xml#video" />.
2018-09-12 19:26:21 +02:00
/// </summary>
public class CleanDateTimeParser
{
private readonly NamingOptions _options;
public CleanDateTimeParser(NamingOptions options)
{
_options = options;
}
public CleanDateTimeResult Clean(string name)
{
var originalName = name;
try
{
var extension = Path.GetExtension(name) ?? string.Empty;
// Check supported extensions
2019-05-10 20:37:42 +02:00
if (!_options.VideoFileExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase)
&& !_options.AudioFileExtensions.Contains(extension, StringComparer.OrdinalIgnoreCase))
2018-09-12 19:26:21 +02:00
{
// Dummy up a file extension because the expressions will fail without one
// This is tricky because we can't just check Path.GetExtension for empty
// If the input is "St. Vincent (2014)", it will produce ". Vincent (2014)" as the extension
name += ".mkv";
}
}
catch (ArgumentException)
{
}
var result = _options.CleanDateTimeRegexes.Select(i => Clean(name, i))
.FirstOrDefault(i => i.HasChanged) ??
new CleanDateTimeResult { Name = originalName };
if (result.HasChanged)
{
return result;
}
// Make a second pass, running clean string first
2020-01-11 20:25:06 +01:00
var cleanStringResult = CleanStringParser.Clean(name, _options.CleanStringRegexes);
2018-09-12 19:26:21 +02:00
if (!cleanStringResult.HasChanged)
{
return result;
}
return _options.CleanDateTimeRegexes.Select(i => Clean(cleanStringResult.Name, i))
.FirstOrDefault(i => i.HasChanged) ??
result;
}
private static CleanDateTimeResult Clean(string name, Regex expression)
2018-09-12 19:26:21 +02:00
{
var result = new CleanDateTimeResult();
var match = expression.Match(name);
2019-05-10 20:37:42 +02:00
if (match.Success
2020-01-11 20:25:06 +01:00
&& match.Groups.Count == 5
2019-05-10 20:37:42 +02:00
&& match.Groups[1].Success
&& match.Groups[2].Success
&& int.TryParse(match.Groups[2].Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out var year))
2018-09-12 19:26:21 +02:00
{
2020-01-11 20:25:06 +01:00
name = match.Groups[1].Value.TrimEnd();
2019-05-10 20:37:42 +02:00
result.Year = year;
result.HasChanged = true;
2018-09-12 19:26:21 +02:00
}
result.Name = name;
return result;
}
}
}