2019-12-13 20:11:37 +01:00
|
|
|
#pragma warning disable CS1591
|
|
|
|
#pragma warning disable SA1600
|
2020-01-22 22:18:56 +01:00
|
|
|
#nullable enable
|
2019-12-13 20:11:37 +01:00
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
using System;
|
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Globalization;
|
|
|
|
using System.Linq;
|
2019-01-13 20:17:29 +01:00
|
|
|
using Emby.Naming.Common;
|
2018-09-12 19:26:21 +02:00
|
|
|
|
|
|
|
namespace Emby.Naming.TV
|
|
|
|
{
|
|
|
|
public class EpisodePathParser
|
|
|
|
{
|
|
|
|
private readonly NamingOptions _options;
|
|
|
|
|
|
|
|
public EpisodePathParser(NamingOptions options)
|
|
|
|
{
|
|
|
|
_options = options;
|
|
|
|
}
|
|
|
|
|
2019-05-10 20:37:42 +02:00
|
|
|
public EpisodePathParserResult Parse(string path, bool isDirectory, bool? isNamed = null, bool? isOptimistic = null, bool? supportsAbsoluteNumbers = null, bool fillExtendedInfo = true)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
// Added to be able to use regex patterns which require a file extension.
|
|
|
|
// There were no failed tests without this block, but to be safe, we can keep it until
|
|
|
|
// the regex which require file extensions are modified so that they don't need them.
|
2019-05-10 20:37:42 +02:00
|
|
|
if (isDirectory)
|
2019-03-01 17:12:22 +01:00
|
|
|
{
|
2018-09-12 19:26:21 +02:00
|
|
|
path += ".mp4";
|
2019-03-01 17:12:22 +01:00
|
|
|
}
|
2018-09-12 19:26:21 +02:00
|
|
|
|
2020-01-22 22:18:56 +01:00
|
|
|
EpisodePathParserResult? result = null;
|
2018-09-12 19:26:21 +02:00
|
|
|
|
|
|
|
foreach (var expression in _options.EpisodeExpressions)
|
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
if (supportsAbsoluteNumbers.HasValue
|
|
|
|
&& expression.SupportsAbsoluteEpisodeNumbers != supportsAbsoluteNumbers.Value)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
continue;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
2019-03-01 17:12:22 +01:00
|
|
|
|
2019-05-10 20:37:42 +02:00
|
|
|
if (isNamed.HasValue && expression.IsNamed != isNamed.Value)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
continue;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
2019-03-01 17:12:22 +01:00
|
|
|
|
2019-05-10 20:37:42 +02:00
|
|
|
if (isOptimistic.HasValue && expression.IsOptimistic != isOptimistic.Value)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
continue;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
var currentResult = Parse(path, expression);
|
|
|
|
if (currentResult.Success)
|
|
|
|
{
|
|
|
|
result = currentResult;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (result != null && fillExtendedInfo)
|
|
|
|
{
|
|
|
|
FillAdditional(path, result);
|
|
|
|
|
|
|
|
if (!string.IsNullOrEmpty(result.SeriesName))
|
|
|
|
{
|
|
|
|
result.SeriesName = result.SeriesName
|
|
|
|
.Trim()
|
|
|
|
.Trim(new[] { '_', '.', '-' })
|
|
|
|
.Trim();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return result ?? new EpisodePathParserResult();
|
|
|
|
}
|
|
|
|
|
2019-01-06 21:50:43 +01:00
|
|
|
private static EpisodePathParserResult Parse(string name, EpisodeExpression expression)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
var result = new EpisodePathParserResult();
|
|
|
|
|
|
|
|
// This is a hack to handle wmc naming
|
|
|
|
if (expression.IsByDate)
|
|
|
|
{
|
|
|
|
name = name.Replace('_', '-');
|
|
|
|
}
|
|
|
|
|
|
|
|
var match = expression.Regex.Match(name);
|
|
|
|
|
|
|
|
// (Full)(Season)(Episode)(Extension)
|
|
|
|
if (match.Success && match.Groups.Count >= 3)
|
|
|
|
{
|
|
|
|
if (expression.IsByDate)
|
|
|
|
{
|
|
|
|
DateTime date;
|
|
|
|
if (expression.DateTimeFormats.Length > 0)
|
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
if (DateTime.TryParseExact(
|
|
|
|
match.Groups[0].Value,
|
2018-09-12 19:26:21 +02:00
|
|
|
expression.DateTimeFormats,
|
|
|
|
CultureInfo.InvariantCulture,
|
|
|
|
DateTimeStyles.None,
|
|
|
|
out date))
|
|
|
|
{
|
|
|
|
result.Year = date.Year;
|
|
|
|
result.Month = date.Month;
|
|
|
|
result.Day = date.Day;
|
|
|
|
result.Success = true;
|
|
|
|
}
|
|
|
|
}
|
2019-05-10 20:37:42 +02:00
|
|
|
else if (DateTime.TryParse(match.Groups[0].Value, out date))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-05-10 20:37:42 +02:00
|
|
|
result.Year = date.Year;
|
|
|
|
result.Month = date.Month;
|
|
|
|
result.Day = date.Day;
|
|
|
|
result.Success = true;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// TODO: Only consider success if date successfully parsed?
|
|
|
|
result.Success = true;
|
|
|
|
}
|
|
|
|
else if (expression.IsNamed)
|
|
|
|
{
|
2019-01-13 21:46:33 +01:00
|
|
|
if (int.TryParse(match.Groups["seasonnumber"].Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out var num))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
result.SeasonNumber = num;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (int.TryParse(match.Groups["epnumber"].Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out num))
|
|
|
|
{
|
|
|
|
result.EpisodeNumber = num;
|
|
|
|
}
|
|
|
|
|
2019-01-13 21:37:13 +01:00
|
|
|
var endingNumberGroup = match.Groups["endingepnumber"];
|
2018-09-12 19:26:21 +02:00
|
|
|
if (endingNumberGroup.Success)
|
|
|
|
{
|
2020-01-18 16:18:55 +01:00
|
|
|
// Will only set EndingEpisodeNumber if the captured number is not followed by additional numbers
|
2018-09-12 19:26:21 +02:00
|
|
|
// or a 'p' or 'i' as what you would get with a pixel resolution specification.
|
|
|
|
// It avoids erroneous parsing of something like "series-s09e14-1080p.mkv" as a multi-episode from E14 to E108
|
|
|
|
int nextIndex = endingNumberGroup.Index + endingNumberGroup.Length;
|
2019-05-10 20:37:42 +02:00
|
|
|
if (nextIndex >= name.Length
|
2020-01-22 22:18:56 +01:00
|
|
|
|| !"0123456789iIpP".Contains(name[nextIndex], StringComparison.Ordinal))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
if (int.TryParse(endingNumberGroup.Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out num))
|
|
|
|
{
|
|
|
|
result.EndingEpsiodeNumber = num;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
result.SeriesName = match.Groups["seriesname"].Value;
|
|
|
|
result.Success = result.EpisodeNumber.HasValue;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2019-01-13 21:46:33 +01:00
|
|
|
if (int.TryParse(match.Groups[1].Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out var num))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
|
|
|
result.SeasonNumber = num;
|
|
|
|
}
|
2019-05-10 20:37:42 +02:00
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
if (int.TryParse(match.Groups[2].Value, NumberStyles.Integer, CultureInfo.InvariantCulture, out num))
|
|
|
|
{
|
|
|
|
result.EpisodeNumber = num;
|
|
|
|
}
|
|
|
|
|
|
|
|
result.Success = result.EpisodeNumber.HasValue;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Invalidate match when the season is 200 through 1927 or above 2500
|
|
|
|
// because it is an error unless the TV show is intentionally using false season numbers.
|
|
|
|
// It avoids erroneous parsing of something like "Series Special (1920x1080).mkv" as being season 1920 episode 1080.
|
2019-05-10 20:37:42 +02:00
|
|
|
if ((result.SeasonNumber >= 200 && result.SeasonNumber < 1928)
|
|
|
|
|| result.SeasonNumber > 2500)
|
|
|
|
{
|
2018-09-12 19:26:21 +02:00
|
|
|
result.Success = false;
|
2019-05-10 20:37:42 +02:00
|
|
|
}
|
2018-09-12 19:26:21 +02:00
|
|
|
|
|
|
|
result.IsByDate = expression.IsByDate;
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void FillAdditional(string path, EpisodePathParserResult info)
|
|
|
|
{
|
|
|
|
var expressions = _options.MultipleEpisodeExpressions.ToList();
|
|
|
|
|
|
|
|
if (string.IsNullOrEmpty(info.SeriesName))
|
|
|
|
{
|
|
|
|
expressions.InsertRange(0, _options.EpisodeExpressions.Where(i => i.IsNamed));
|
|
|
|
}
|
|
|
|
|
|
|
|
FillAdditional(path, info, expressions);
|
|
|
|
}
|
|
|
|
|
|
|
|
private void FillAdditional(string path, EpisodePathParserResult info, IEnumerable<EpisodeExpression> expressions)
|
|
|
|
{
|
2019-03-01 17:12:22 +01:00
|
|
|
foreach (var i in expressions)
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-03-01 17:12:22 +01:00
|
|
|
if (!i.IsNamed)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
var result = Parse(path, i);
|
|
|
|
|
|
|
|
if (!result.Success)
|
|
|
|
{
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2018-09-12 19:26:21 +02:00
|
|
|
if (string.IsNullOrEmpty(info.SeriesName))
|
|
|
|
{
|
|
|
|
info.SeriesName = result.SeriesName;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!info.EndingEpsiodeNumber.HasValue && info.EpisodeNumber.HasValue)
|
|
|
|
{
|
|
|
|
info.EndingEpsiodeNumber = result.EndingEpsiodeNumber;
|
|
|
|
}
|
|
|
|
|
2019-03-01 17:12:22 +01:00
|
|
|
if (!string.IsNullOrEmpty(info.SeriesName)
|
|
|
|
&& (!info.EpisodeNumber.HasValue || info.EndingEpsiodeNumber.HasValue))
|
2018-09-12 19:26:21 +02:00
|
|
|
{
|
2019-03-01 17:12:22 +01:00
|
|
|
break;
|
2018-09-12 19:26:21 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|