open-source-mirrors
/
Prowlarr
mirror of https://github.com/Prowlarr/Prowlarr
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

minor cleanup.

Browse Source
pull/2/head
kayone 11 years ago committed by Mark McDowall
parent 21af4bbdfa
commit 4cfb2f271d
4 changed files with 59 additions and 76 deletions
Show Stats Download Patch File Download Diff File

1
src/NzbDrone.Core/NzbDrone.Core.csproj
Unescape View File

@ -327,6 +327,7 @@
<Compile Include="Organizer\EpisodeFormat.cs" /> <Compile Include="Organizer\EpisodeFormat.cs" />
<Compile Include="Organizer\Exception.cs" /> <Compile Include="Organizer\Exception.cs" />
<Compile Include="Organizer\FilenameBuilderTokenEqualityComparer.cs" /> <Compile Include="Organizer\FilenameBuilderTokenEqualityComparer.cs" />
<Compile Include="Organizer\NamingConfigService.cs" />
<Compile Include="Parser\InvalidDateException.cs" /> <Compile Include="Parser\InvalidDateException.cs" />
<Compile Include="Parser\Model\SeriesTitleInfo.cs" /> <Compile Include="Parser\Model\SeriesTitleInfo.cs" />
<Compile Include="ProgressMessaging\CommandUpdatedEvent.cs" /> <Compile Include="ProgressMessaging\CommandUpdatedEvent.cs" />

87
src/NzbDrone.Core/Organizer/FileNameBuilder.cs
Unescape View File

@ -5,7 +5,6 @@ using System.Linq;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
using NLog; using NLog;
using NzbDrone.Core.Configuration; using NzbDrone.Core.Configuration;
using NzbDrone.Core.Datastore;
using NzbDrone.Core.MediaFiles; using NzbDrone.Core.MediaFiles;
using NzbDrone.Core.Tv; using NzbDrone.Core.Tv;
@ -18,39 +17,7 @@ namespace NzbDrone.Core.Organizer
string BuildFilePath(Series series, int seasonNumber, string fileName, string extension); string BuildFilePath(Series series, int seasonNumber, string fileName, string extension);
} }
public interface INamingConfigService
{
NamingConfig GetConfig();
NamingConfig Save(NamingConfig namingConfig);
}
public class NamingConfigService : INamingConfigService
{
private readonly IBasicRepository<NamingConfig> _repository;
public NamingConfigService(IBasicRepository<NamingConfig> repository)
{
_repository = repository;
}
public NamingConfig GetConfig()
{
var config = _repository.SingleOrDefault();
if (config == null)
{
_repository.Insert(NamingConfig.Default);
config = _repository.Single();
}
return config;
}
public NamingConfig Save(NamingConfig namingConfig)
{
return _repository.Upsert(namingConfig);
}
}
public class FileNameBuilder : IBuildFileNames public class FileNameBuilder : IBuildFileNames
{ {
@ -84,9 +51,9 @@ namespace NzbDrone.Core.Organizer
return BuildFilename(episodes, series, episodeFile, nameSpec); return BuildFilename(episodes, series, episodeFile, nameSpec);
} }
public string BuildFilename(IList<Episode> episodes, Series series, EpisodeFile episodeFile, NamingConfig nameSpec) public string BuildFilename(IList<Episode> episodes, Series series, EpisodeFile episodeFile, NamingConfig namingConfig)
{ {
if (!nameSpec.RenameEpisodes) if (!namingConfig.RenameEpisodes)
{ {
if (String.IsNullOrWhiteSpace(episodeFile.SceneName)) if (String.IsNullOrWhiteSpace(episodeFile.SceneName))
{ {
@ -96,29 +63,31 @@ namespace NzbDrone.Core.Organizer
return episodeFile.SceneName; return episodeFile.SceneName;
} }
if (String.IsNullOrWhiteSpace(nameSpec.StandardEpisodeFormat) && series.SeriesType == SeriesTypes.Standard) if (String.IsNullOrWhiteSpace(namingConfig.StandardEpisodeFormat) && series.SeriesType == SeriesTypes.Standard)
{ {
throw new NamingFormatException("Standard episode format cannot be null"); throw new NamingFormatException("Standard episode format cannot be null");
} }
if (String.IsNullOrWhiteSpace(nameSpec.DailyEpisodeFormat) && series.SeriesType == SeriesTypes.Daily) if (String.IsNullOrWhiteSpace(namingConfig.DailyEpisodeFormat) && series.SeriesType == SeriesTypes.Daily)
{ {
throw new NamingFormatException("Daily episode format cannot be null"); throw new NamingFormatException("Daily episode format cannot be null");
} }
var sortedEpisodes = episodes.OrderBy(e => e.EpisodeNumber).ToList(); var sortedEpisodes = episodes.OrderBy(e => e.EpisodeNumber).ToList();
var pattern = nameSpec.StandardEpisodeFormat; var pattern = namingConfig.StandardEpisodeFormat;
var episodeTitles = new List<string> var episodeTitles = new List<string>
{ {
Parser.Parser.CleanupEpisodeTitle(sortedEpisodes.First().Title) Parser.Parser.CleanupEpisodeTitle(sortedEpisodes.First().Title)
}; };
var tokenValues = new Dictionary<string, string>(new FilenameBuilderTokenEqualityComparer()); var tokenValues = new Dictionary<string, string>(FilenameBuilderTokenEqualityComparer.Instance)
tokenValues.Add("{Series Title}", series.Title); {
{"{Series Title}", series.Title}
};
if (series.SeriesType == SeriesTypes.Daily) if (series.SeriesType == SeriesTypes.Daily)
{ {
pattern = nameSpec.DailyEpisodeFormat; pattern = namingConfig.DailyEpisodeFormat;
if (!String.IsNullOrWhiteSpace(episodes.First().AirDate)) if (!String.IsNullOrWhiteSpace(episodes.First().AirDate))
{ {
@ -146,7 +115,7 @@ namespace NzbDrone.Core.Organizer
foreach (var episode in sortedEpisodes.Skip(1)) foreach (var episode in sortedEpisodes.Skip(1))
{ {
switch ((MultiEpisodeStyle)nameSpec.MultiEpisodeStyle) switch ((MultiEpisodeStyle)namingConfig.MultiEpisodeStyle)
{ {
case MultiEpisodeStyle.Duplicate: case MultiEpisodeStyle.Duplicate:
seasonEpisodePattern += episodeFormat.Separator + episodeFormat.SeasonEpisodePattern; seasonEpisodePattern += episodeFormat.Separator + episodeFormat.SeasonEpisodePattern;
@ -194,7 +163,7 @@ namespace NzbDrone.Core.Organizer
else else
{ {
var tokenValues = new Dictionary<string, string>(new FilenameBuilderTokenEqualityComparer()); var tokenValues = new Dictionary<string, string>(FilenameBuilderTokenEqualityComparer.Instance);
tokenValues.Add("{Series Title}", series.Title); tokenValues.Add("{Series Title}", series.Title);
seasonFolder = ReplaceSeasonTokens(_configService.SeasonFolderFormat, seasonNumber); seasonFolder = ReplaceSeasonTokens(_configService.SeasonFolderFormat, seasonNumber);
@ -275,38 +244,6 @@ namespace NzbDrone.Core.Organizer
return value.ToString().PadLeft(zeroCount + 1, '0'); return value.ToString().PadLeft(zeroCount + 1, '0');
} }
private static readonly List<EpisodeSortingType> MultiEpisodeStyles = new List<EpisodeSortingType>
{
new EpisodeSortingType
{
Id = 0,
Name = "Extend",
Pattern = "-%0e"
},
new EpisodeSortingType
{
Id = 1,
Name = "Duplicate",
Pattern = "%p%0s%x%0e"
},
new EpisodeSortingType
{
Id = 2,
Name = "Repeat",
Pattern = "%x%0e"
},
new EpisodeSortingType
{
Id = 3,
Name = "Scene",
Pattern = "-%x%0e"
}
};
private static EpisodeSortingType GetMultiEpisodeStyle(int id)
{
return MultiEpisodeStyles.Single(s => s.Id == id);
}
} }
public enum MultiEpisodeStyle public enum MultiEpisodeStyle

9
src/NzbDrone.Core/Organizer/FilenameBuilderTokenEqualityComparer.cs
Unescape View File

@ -6,8 +6,15 @@ namespace NzbDrone.Core.Organizer
{ {
public class FilenameBuilderTokenEqualityComparer : IEqualityComparer<String> public class FilenameBuilderTokenEqualityComparer : IEqualityComparer<String>
{ {
public static readonly FilenameBuilderTokenEqualityComparer Instance = new FilenameBuilderTokenEqualityComparer();
private static readonly Regex SimpleTokenRegex = new Regex(@"\s|_|\W", RegexOptions.Compiled | RegexOptions.IgnoreCase); private static readonly Regex SimpleTokenRegex = new Regex(@"\s|_|\W", RegexOptions.Compiled | RegexOptions.IgnoreCase);
private FilenameBuilderTokenEqualityComparer()
{
}
public bool Equals(String s1, String s2) public bool Equals(String s1, String s2)
{ {
return SimplifyToken(s1).Equals(SimplifyToken(s2)); return SimplifyToken(s1).Equals(SimplifyToken(s2));
@ -18,7 +25,7 @@ namespace NzbDrone.Core.Organizer
return SimplifyToken(str).GetHashCode(); return SimplifyToken(str).GetHashCode();
} }
private string SimplifyToken(string token) private static string SimplifyToken(string token)
{ {
return SimpleTokenRegex.Replace(token, String.Empty).ToLower(); return SimpleTokenRegex.Replace(token, String.Empty).ToLower();
} }

38
src/NzbDrone.Core/Organizer/NamingConfigService.cs
Unescape View File

@ -0,0 +1,38 @@
using NzbDrone.Core.Datastore;
namespace NzbDrone.Core.Organizer
{
public interface INamingConfigService
{
NamingConfig GetConfig();
void Save(NamingConfig namingConfig);
}
public class NamingConfigService : INamingConfigService
{
private readonly IBasicRepository<NamingConfig> _repository;
public NamingConfigService(IBasicRepository<NamingConfig> repository)
{
_repository = repository;
}
public NamingConfig GetConfig()
{
var config = _repository.SingleOrDefault();
if (config == null)
{
_repository.Insert(NamingConfig.Default);
config = _repository.Single();
}
return config;
}
public void Save(NamingConfig namingConfig)
{
_repository.Upsert(namingConfig);
}
}
}
Write Preview
Loading…
Cancel
Save