2018-02-19 07:18:14 +00:00
|
|
|
using System.Collections.Generic;
|
2017-01-14 00:03:37 +00:00
|
|
|
using System.IO;
|
|
|
|
using NLog;
|
2018-11-23 07:03:32 +00:00
|
|
|
using Radarr.Http;
|
|
|
|
using Radarr.Http.REST;
|
2017-01-14 00:03:37 +00:00
|
|
|
using NzbDrone.Core.Datastore.Events;
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2018-03-14 20:41:36 +00:00
|
|
|
using NzbDrone.Core.Movies;
|
2017-01-14 00:03:37 +00:00
|
|
|
using NzbDrone.Core.DecisionEngine;
|
|
|
|
using NzbDrone.SignalR;
|
|
|
|
|
2018-02-19 07:18:14 +00:00
|
|
|
namespace NzbDrone.Api.MovieFiles
|
2017-01-14 00:03:37 +00:00
|
|
|
{
|
2018-11-23 07:03:32 +00:00
|
|
|
public class MovieFileModule : RadarrRestModuleWithSignalR<MovieFileResource, MovieFile>, IHandle<MovieFileAddedEvent>
|
2017-01-14 00:03:37 +00:00
|
|
|
{
|
|
|
|
private readonly IMediaFileService _mediaFileService;
|
|
|
|
private readonly IRecycleBinProvider _recycleBinProvider;
|
2017-02-26 04:16:18 +00:00
|
|
|
private readonly IMovieService _movieService;
|
2017-01-14 00:03:37 +00:00
|
|
|
private readonly IQualityUpgradableSpecification _qualityUpgradableSpecification;
|
|
|
|
private readonly Logger _logger;
|
|
|
|
|
|
|
|
public MovieFileModule(IBroadcastSignalRMessage signalRBroadcaster,
|
|
|
|
IMediaFileService mediaFileService,
|
|
|
|
IRecycleBinProvider recycleBinProvider,
|
2017-02-26 04:16:18 +00:00
|
|
|
IMovieService movieService,
|
2017-01-14 00:03:37 +00:00
|
|
|
IQualityUpgradableSpecification qualityUpgradableSpecification,
|
|
|
|
Logger logger)
|
|
|
|
: base(signalRBroadcaster)
|
|
|
|
{
|
|
|
|
_mediaFileService = mediaFileService;
|
|
|
|
_recycleBinProvider = recycleBinProvider;
|
2017-02-26 04:16:18 +00:00
|
|
|
_movieService = movieService;
|
2017-01-14 00:03:37 +00:00
|
|
|
_qualityUpgradableSpecification = qualityUpgradableSpecification;
|
|
|
|
_logger = logger;
|
2017-01-28 16:35:51 +00:00
|
|
|
GetResourceById = GetMovieFile;
|
|
|
|
UpdateResource = SetQuality;
|
2017-02-26 04:16:18 +00:00
|
|
|
DeleteResource = DeleteMovieFile;
|
2017-01-14 00:03:37 +00:00
|
|
|
}
|
|
|
|
|
2017-01-28 16:35:51 +00:00
|
|
|
private MovieFileResource GetMovieFile(int id)
|
2017-01-14 00:03:37 +00:00
|
|
|
{
|
2017-02-26 04:16:18 +00:00
|
|
|
var movie = _mediaFileService.GetMovie(id);
|
2017-01-14 00:03:37 +00:00
|
|
|
|
2017-02-26 04:16:18 +00:00
|
|
|
return movie.ToResource();
|
2017-01-14 00:03:37 +00:00
|
|
|
}
|
|
|
|
|
2017-04-26 11:31:55 +00:00
|
|
|
|
2017-02-26 04:16:18 +00:00
|
|
|
private void SetQuality(MovieFileResource movieFileResource)
|
2018-09-10 19:25:10 +00:00
|
|
|
{
|
2017-02-26 04:16:18 +00:00
|
|
|
var movieFile = _mediaFileService.GetMovie(movieFileResource.Id);
|
|
|
|
movieFile.Quality = movieFileResource.Quality;
|
|
|
|
_mediaFileService.Update(movieFile);
|
2017-04-26 11:31:55 +00:00
|
|
|
|
|
|
|
BroadcastResourceChange(ModelAction.Updated, movieFile.Id);
|
2017-01-28 16:35:51 +00:00
|
|
|
}
|
2017-01-14 00:03:37 +00:00
|
|
|
|
2017-02-26 04:16:18 +00:00
|
|
|
private void DeleteMovieFile(int id)
|
2017-01-14 00:03:37 +00:00
|
|
|
{
|
2017-02-26 04:16:18 +00:00
|
|
|
var movieFile = _mediaFileService.GetMovie(id);
|
|
|
|
var movie = _movieService.GetMovie(movieFile.MovieId);
|
|
|
|
var fullPath = Path.Combine(movie.Path, movieFile.RelativePath);
|
2017-01-14 00:03:37 +00:00
|
|
|
|
2017-02-26 04:16:18 +00:00
|
|
|
_logger.Info("Deleting movie file: {0}", fullPath);
|
2017-01-14 00:03:37 +00:00
|
|
|
_recycleBinProvider.DeleteFile(fullPath);
|
2017-02-26 04:16:18 +00:00
|
|
|
_mediaFileService.Delete(movieFile, DeleteMediaFileReason.Manual);
|
2017-01-14 00:03:37 +00:00
|
|
|
}
|
|
|
|
|
2017-02-26 04:16:18 +00:00
|
|
|
public void Handle(MovieFileAddedEvent message)
|
2017-01-14 00:03:37 +00:00
|
|
|
{
|
2017-02-26 04:16:18 +00:00
|
|
|
BroadcastResourceChange(ModelAction.Updated, message.MovieFile.Id);
|
2017-01-14 00:03:37 +00:00
|
|
|
}
|
|
|
|
}
|
2018-02-19 07:18:14 +00:00
|
|
|
}
|