mirror of https://github.com/Radarr/Radarr
77 lines
2.4 KiB
C#
77 lines
2.4 KiB
C#
|
using System.Collections.Generic;
|
|||
|
using System.Linq;
|
|||
|
using FluentValidation.Results;
|
|||
|
using NzbDrone.Core.Parser.Model;
|
|||
|
using NzbDrone.Core.Tv;
|
|||
|
|
|||
|
namespace NzbDrone.Core.Organizer
|
|||
|
{
|
|||
|
public interface IFilenameValidationService
|
|||
|
{
|
|||
|
ValidationFailure ValidateStandardFilename(SampleResult sampleResult);
|
|||
|
ValidationFailure ValidateDailyFilename(SampleResult sampleResult);
|
|||
|
}
|
|||
|
|
|||
|
public class FilenameValidationService : IFilenameValidationService
|
|||
|
{
|
|||
|
private const string ERROR_MESSAGE = "Produces invalid file names";
|
|||
|
|
|||
|
public ValidationFailure ValidateStandardFilename(SampleResult sampleResult)
|
|||
|
{
|
|||
|
var validationFailure = new ValidationFailure("StandardEpisodeFormat", ERROR_MESSAGE);
|
|||
|
var parsedEpisodeInfo = Parser.Parser.ParseTitle(sampleResult.Filename);
|
|||
|
|
|||
|
if (parsedEpisodeInfo == null)
|
|||
|
{
|
|||
|
return validationFailure;
|
|||
|
}
|
|||
|
|
|||
|
if (!ValidateSeasonAndEpisodeNumbers(sampleResult.Episodes, parsedEpisodeInfo))
|
|||
|
{
|
|||
|
return validationFailure;
|
|||
|
}
|
|||
|
|
|||
|
return null;
|
|||
|
}
|
|||
|
|
|||
|
public ValidationFailure ValidateDailyFilename(SampleResult sampleResult)
|
|||
|
{
|
|||
|
var validationFailure = new ValidationFailure("DailyEpisodeFormat", ERROR_MESSAGE);
|
|||
|
var parsedEpisodeInfo = Parser.Parser.ParseTitle(sampleResult.Filename);
|
|||
|
|
|||
|
if (parsedEpisodeInfo == null)
|
|||
|
{
|
|||
|
return validationFailure;
|
|||
|
}
|
|||
|
|
|||
|
if (parsedEpisodeInfo.IsDaily())
|
|||
|
{
|
|||
|
if (!parsedEpisodeInfo.AirDate.Equals(sampleResult.Episodes.Single().AirDate))
|
|||
|
{
|
|||
|
return validationFailure;
|
|||
|
}
|
|||
|
|
|||
|
return null;
|
|||
|
}
|
|||
|
|
|||
|
if (!ValidateSeasonAndEpisodeNumbers(sampleResult.Episodes, parsedEpisodeInfo))
|
|||
|
{
|
|||
|
return validationFailure;
|
|||
|
}
|
|||
|
|
|||
|
return null;
|
|||
|
}
|
|||
|
|
|||
|
private bool ValidateSeasonAndEpisodeNumbers(List<Episode> episodes, ParsedEpisodeInfo parsedEpisodeInfo)
|
|||
|
{
|
|||
|
if (parsedEpisodeInfo.SeasonNumber != episodes.First().SeasonNumber ||
|
|||
|
!parsedEpisodeInfo.EpisodeNumbers.OrderBy(e => e).SequenceEqual(episodes.Select(e => e.EpisodeNumber).OrderBy(e => e)))
|
|||
|
{
|
|||
|
return false;
|
|||
|
}
|
|||
|
|
|||
|
return true;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|