2013-11-20 00:42:17 +00:00
|
|
|
|
using System.Collections.Generic;
|
2014-07-23 23:43:54 +00:00
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
using NLog;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
using NzbDrone.Api.REST;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
using NzbDrone.Core.Datastore.Events;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
2013-08-26 05:14:55 +00:00
|
|
|
|
using NzbDrone.Api.Mapping;
|
2013-10-02 05:20:30 +00:00
|
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2014-07-23 23:43:54 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
2014-08-18 22:24:44 +00:00
|
|
|
|
using NzbDrone.Core.DecisionEngine;
|
2014-08-26 04:01:32 +00:00
|
|
|
|
using NzbDrone.SignalR;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Api.EpisodeFiles
|
|
|
|
|
{
|
2013-10-02 05:20:30 +00:00
|
|
|
|
public class EpisodeModule : NzbDroneRestModuleWithSignalR<EpisodeFileResource, EpisodeFile>,
|
|
|
|
|
IHandle<EpisodeFileAddedEvent>
|
2013-08-21 06:46:23 +00:00
|
|
|
|
{
|
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
private readonly IRecycleBinProvider _recycleBinProvider;
|
2014-07-23 23:43:54 +00:00
|
|
|
|
private readonly ISeriesService _seriesService;
|
2014-08-18 22:24:44 +00:00
|
|
|
|
private readonly IQualityUpgradableSpecification _qualityUpgradableSpecification;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
private readonly Logger _logger;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
|
2014-08-26 04:01:32 +00:00
|
|
|
|
public EpisodeModule(IBroadcastSignalRMessage signalRBroadcaster,
|
2013-11-29 06:31:46 +00:00
|
|
|
|
IMediaFileService mediaFileService,
|
|
|
|
|
IRecycleBinProvider recycleBinProvider,
|
2014-07-23 23:43:54 +00:00
|
|
|
|
ISeriesService seriesService,
|
2014-08-18 22:24:44 +00:00
|
|
|
|
IQualityUpgradableSpecification qualityUpgradableSpecification,
|
2013-11-29 06:31:46 +00:00
|
|
|
|
Logger logger)
|
2014-08-26 04:01:32 +00:00
|
|
|
|
: base(signalRBroadcaster)
|
2013-08-21 06:46:23 +00:00
|
|
|
|
{
|
|
|
|
|
_mediaFileService = mediaFileService;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
_recycleBinProvider = recycleBinProvider;
|
2014-07-23 23:43:54 +00:00
|
|
|
|
_seriesService = seriesService;
|
2014-08-18 22:24:44 +00:00
|
|
|
|
_qualityUpgradableSpecification = qualityUpgradableSpecification;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
_logger = logger;
|
2013-08-22 03:24:50 +00:00
|
|
|
|
GetResourceById = GetEpisodeFile;
|
2013-08-22 00:56:25 +00:00
|
|
|
|
GetResourceAll = GetEpisodeFiles;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
UpdateResource = SetQuality;
|
2013-11-29 06:31:46 +00:00
|
|
|
|
DeleteResource = DeleteEpisodeFile;
|
2013-08-21 06:46:23 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-08-22 03:24:50 +00:00
|
|
|
|
private EpisodeFileResource GetEpisodeFile(int id)
|
|
|
|
|
{
|
2014-07-23 23:43:54 +00:00
|
|
|
|
var episodeFile = _mediaFileService.Get(id);
|
|
|
|
|
var series = _seriesService.GetSeries(episodeFile.SeriesId);
|
|
|
|
|
|
|
|
|
|
return MapToResource(series, episodeFile);
|
2013-08-22 03:24:50 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-08-22 00:56:25 +00:00
|
|
|
|
private List<EpisodeFileResource> GetEpisodeFiles()
|
|
|
|
|
{
|
|
|
|
|
var seriesId = (int?)Request.Query.SeriesId;
|
|
|
|
|
|
|
|
|
|
if (seriesId == null)
|
|
|
|
|
{
|
|
|
|
|
throw new BadRequestException("seriesId is missing");
|
|
|
|
|
}
|
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
var series = _seriesService.GetSeries(seriesId.Value);
|
|
|
|
|
|
|
|
|
|
return _mediaFileService.GetFilesBySeries(seriesId.Value)
|
|
|
|
|
.Select(f => MapToResource(series, f)).ToList();
|
2013-08-22 00:56:25 +00:00
|
|
|
|
}
|
|
|
|
|
|
2013-08-26 05:14:55 +00:00
|
|
|
|
private void SetQuality(EpisodeFileResource episodeFileResource)
|
2013-08-21 06:46:23 +00:00
|
|
|
|
{
|
|
|
|
|
var episodeFile = _mediaFileService.Get(episodeFileResource.Id);
|
|
|
|
|
episodeFile.Quality = episodeFileResource.Quality;
|
|
|
|
|
_mediaFileService.Update(episodeFile);
|
|
|
|
|
}
|
2013-10-02 05:20:30 +00:00
|
|
|
|
|
2013-11-29 06:31:46 +00:00
|
|
|
|
private void DeleteEpisodeFile(int id)
|
|
|
|
|
{
|
|
|
|
|
var episodeFile = _mediaFileService.Get(id);
|
2014-07-23 23:43:54 +00:00
|
|
|
|
var series = _seriesService.GetSeries(episodeFile.SeriesId);
|
|
|
|
|
var fullPath = Path.Combine(series.Path, episodeFile.RelativePath);
|
2013-11-29 06:31:46 +00:00
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
_logger.Info("Deleting episode file: {0}", fullPath);
|
|
|
|
|
_recycleBinProvider.DeleteFile(fullPath);
|
2014-08-21 06:29:34 +00:00
|
|
|
|
_mediaFileService.Delete(episodeFile, DeleteMediaFileReason.Manual);
|
2013-11-29 06:31:46 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-08-18 22:24:44 +00:00
|
|
|
|
private EpisodeFileResource MapToResource(Core.Tv.Series series, EpisodeFile episodeFile)
|
2014-07-23 23:43:54 +00:00
|
|
|
|
{
|
|
|
|
|
var resource = episodeFile.InjectTo<EpisodeFileResource>();
|
|
|
|
|
resource.Path = Path.Combine(series.Path, episodeFile.RelativePath);
|
|
|
|
|
|
2014-08-18 22:24:44 +00:00
|
|
|
|
resource.QualityCutoffNotMet = _qualityUpgradableSpecification.CutoffNotMet(series.Profile.Value, episodeFile.Quality);
|
|
|
|
|
|
2014-07-23 23:43:54 +00:00
|
|
|
|
return resource;
|
|
|
|
|
}
|
|
|
|
|
|
2013-10-02 05:20:30 +00:00
|
|
|
|
public void Handle(EpisodeFileAddedEvent message)
|
|
|
|
|
{
|
|
|
|
|
BroadcastResourceChange(ModelAction.Updated, message.EpisodeFile.Id);
|
|
|
|
|
}
|
2013-08-21 06:46:23 +00:00
|
|
|
|
}
|
|
|
|
|
}
|