Radarr/NzbDrone.Api/Series/SeriesModule.cs

133 lines
4.7 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Globalization;
2013-02-09 10:04:18 +00:00
using System.Linq;
using AutoMapper;
2013-01-28 18:06:54 +00:00
using FluentValidation;
2013-02-23 20:35:26 +00:00
using NzbDrone.Api.Extensions;
2013-01-28 18:06:54 +00:00
using NzbDrone.Common;
2013-04-20 23:36:23 +00:00
using NzbDrone.Core.SeriesStats;
using NzbDrone.Core.Tv;
using NzbDrone.Core.Model;
2013-04-20 22:14:41 +00:00
using NzbDrone.Api.Validation;
2013-01-20 01:37:48 +00:00
namespace NzbDrone.Api.Series
{
2013-04-20 22:14:41 +00:00
public class SeriesModule : NzbDroneRestModule<SeriesResource>
2013-01-20 01:37:48 +00:00
{
private readonly ISeriesService _seriesService;
2013-04-20 23:36:23 +00:00
private readonly ISeriesStatisticsService _seriesStatisticsService;
2013-01-20 01:37:48 +00:00
2013-04-20 23:36:23 +00:00
public SeriesModule(ISeriesService seriesService, ISeriesStatisticsService seriesStatisticsService)
2013-01-20 01:37:48 +00:00
: base("/Series")
{
_seriesService = seriesService;
2013-04-20 23:36:23 +00:00
_seriesStatisticsService = seriesStatisticsService;
2013-04-20 22:14:41 +00:00
GetResourceAll = AllSeries;
GetResourceById = GetSeries;
CreateResource = AddSeries;
UpdateResource = UpdateSeries;
DeleteResource = DeleteSeries;
SharedValidator.RuleFor(s => s.RootFolderId).ValidId();
SharedValidator.RuleFor(s => s.QualityProfileId).ValidId();
PostValidator.RuleFor(s => s.Title).NotEmpty();
2013-01-20 01:37:48 +00:00
}
2013-04-20 22:14:41 +00:00
private List<SeriesResource> AllSeries()
2013-02-09 10:04:18 +00:00
{
2013-04-12 00:36:47 +00:00
var series = _seriesService.GetAllSeries().ToList();
2013-04-20 23:36:23 +00:00
var seriesStats = _seriesStatisticsService.SeriesStatistics();
var seriesModels = Mapper.Map<List<Core.Tv.Series>, List<SeriesResource>>(series);
2013-02-09 10:04:18 +00:00
2013-04-20 20:09:12 +00:00
foreach (var s in seriesModels)
{
var stats = seriesStats.SingleOrDefault(ss => ss.SeriesId == s.Id);
if (stats == null) continue;
s.EpisodeCount = stats.EpisodeCount;
s.EpisodeFileCount = stats.EpisodeFileCount;
s.NumberOfSeasons = stats.NumberOfSeasons;
s.NextAiring = stats.NextAiring;
}
2013-04-20 22:14:41 +00:00
return seriesModels;
2013-02-09 10:04:18 +00:00
}
2013-01-20 01:37:48 +00:00
2013-04-20 22:14:41 +00:00
private SeriesResource GetSeries(int id)
2013-02-10 03:42:44 +00:00
{
2013-04-12 00:36:47 +00:00
var series = _seriesService.GetSeries(id);
var seriesModels = Mapper.Map<Core.Tv.Series, SeriesResource>(series);
2013-04-20 22:14:41 +00:00
return seriesModels;
2013-02-10 03:42:44 +00:00
}
2013-04-20 22:14:41 +00:00
private SeriesResource AddSeries(SeriesResource seriesResource)
2013-01-20 01:37:48 +00:00
{
2013-03-31 21:45:16 +00:00
var newSeries = Request.Body.FromJson<Core.Tv.Series>();
2013-01-28 18:06:54 +00:00
//Todo: Alert the user if this series already exists
//Todo: We need to create the folder if the user is adding a new series
//(we can just create the folder and it won't blow up if it already exists)
2013-02-09 10:04:18 +00:00
//We also need to remove any special characters from the filename before attempting to create it
2013-04-20 22:14:41 +00:00
var series = Mapper.Map<SeriesResource, Core.Tv.Series>(seriesResource);
_seriesService.AddSeries(series);
return Mapper.Map<Core.Tv.Series, SeriesResource>(series);
2013-01-28 18:06:54 +00:00
}
2013-02-10 03:42:44 +00:00
2013-04-20 22:14:41 +00:00
private SeriesResource UpdateSeries(SeriesResource seriesResource)
{
2013-04-20 22:14:41 +00:00
var series = _seriesService.GetSeries(seriesResource.Id);
2013-04-20 22:14:41 +00:00
series.Monitored = seriesResource.Monitored;
series.SeasonFolder = seriesResource.SeasonFolder;
series.QualityProfileId = seriesResource.QualityProfileId;
//Todo: Do we want to force a scan when this path changes? Can we use events instead?
2013-04-20 22:14:41 +00:00
series.RootFolderId = seriesResource.RootFolderId;
series.FolderName = seriesResource.FolderName;
2013-04-20 22:14:41 +00:00
series.BacklogSetting = (BacklogSettingType)seriesResource.BacklogSetting;
2013-04-20 22:14:41 +00:00
if (!String.IsNullOrWhiteSpace(seriesResource.CustomStartDate))
series.CustomStartDate = DateTime.Parse(seriesResource.CustomStartDate, null, DateTimeStyles.RoundtripKind);
else
series.CustomStartDate = null;
2013-04-12 00:36:47 +00:00
_seriesService.UpdateSeries(series);
2013-04-20 22:14:41 +00:00
return Mapper.Map<Core.Tv.Series, SeriesResource>(series);
}
2013-04-20 22:14:41 +00:00
private void DeleteSeries(int id)
2013-02-10 03:42:44 +00:00
{
var deleteFiles = Convert.ToBoolean(Request.Headers["deleteFiles"].FirstOrDefault());
_seriesService.DeleteSeries(id, deleteFiles);
2013-02-10 03:42:44 +00:00
}
2013-01-28 18:06:54 +00:00
}
public class SeriesValidator : AbstractValidator<Core.Tv.Series>
2013-01-28 18:06:54 +00:00
{
private readonly DiskProvider _diskProvider;
public SeriesValidator(DiskProvider diskProvider)
{
_diskProvider = diskProvider;
2013-01-20 01:37:48 +00:00
}
2013-01-28 18:06:54 +00:00
public SeriesValidator()
{
RuleSet("POST", () =>
{
2013-04-01 06:22:16 +00:00
RuleFor(s => s.Id).GreaterThan(0);
2013-01-28 18:06:54 +00:00
RuleFor(s => s.Path).NotEmpty().Must(_diskProvider.FolderExists);
RuleFor(s => s.QualityProfileId).GreaterThan(0);
});
}
2013-01-20 01:37:48 +00:00
}
}