2018-03-14 20:41:36 +00:00
|
|
|
using System.Collections.Generic;
|
2014-07-23 23:43:54 +00:00
|
|
|
using System.IO;
|
2013-07-19 05:05:07 +00:00
|
|
|
using System.Linq;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Common;
|
2013-03-07 04:34:56 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.Events;
|
2013-09-14 06:36:07 +00:00
|
|
|
using NzbDrone.Core.Messaging.Events;
|
2018-03-14 20:41:36 +00:00
|
|
|
using NzbDrone.Core.Movies;
|
|
|
|
using NzbDrone.Core.Movies.Events;
|
2013-03-01 07:03:41 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
|
|
{
|
|
|
|
public interface IMediaFileService
|
|
|
|
{
|
2017-01-08 22:01:37 +00:00
|
|
|
MovieFile Add(MovieFile movieFile);
|
|
|
|
void Update(MovieFile movieFile);
|
2018-11-23 07:03:32 +00:00
|
|
|
void Update(List<MovieFile> movieFile);
|
2017-01-08 22:01:37 +00:00
|
|
|
void Delete(MovieFile movieFile, DeleteMediaFileReason reason);
|
2017-01-03 19:24:55 +00:00
|
|
|
List<MovieFile> GetFilesByMovie(int movieId);
|
2018-03-14 20:41:36 +00:00
|
|
|
List<MovieFile> GetFilesWithoutMediaInfo();
|
2017-01-02 17:05:55 +00:00
|
|
|
List<string> FilterExistingFiles(List<string> files, Movie movie);
|
2017-01-14 00:03:37 +00:00
|
|
|
MovieFile GetMovie(int id);
|
2017-01-09 03:16:14 +00:00
|
|
|
List<MovieFile> GetMovies(IEnumerable<int> ids);
|
2021-11-29 04:06:24 +00:00
|
|
|
List<MovieFile> GetFilesWithRelativePath(int movieIds, string relativePath);
|
2013-03-01 07:03:41 +00:00
|
|
|
}
|
|
|
|
|
2020-06-05 03:35:02 +00:00
|
|
|
public class MediaFileService : IMediaFileService, IHandleAsync<MoviesDeletedEvent>
|
2013-03-01 07:03:41 +00:00
|
|
|
{
|
2019-12-18 02:17:47 +00:00
|
|
|
private readonly IMediaFileRepository _mediaFileRepository;
|
2019-12-18 21:56:41 +00:00
|
|
|
private readonly IMovieRepository _movieRepository;
|
|
|
|
private readonly IEventAggregator _eventAggregator;
|
2013-03-01 07:03:41 +00:00
|
|
|
|
2019-07-27 04:33:04 +00:00
|
|
|
public MediaFileService(IMediaFileRepository mediaFileRepository,
|
2019-12-18 21:56:41 +00:00
|
|
|
IMovieRepository movieRepository,
|
|
|
|
IEventAggregator eventAggregator)
|
2013-03-01 07:03:41 +00:00
|
|
|
{
|
|
|
|
_mediaFileRepository = mediaFileRepository;
|
2019-12-18 21:56:41 +00:00
|
|
|
_movieRepository = movieRepository;
|
2013-09-14 06:36:07 +00:00
|
|
|
_eventAggregator = eventAggregator;
|
2013-03-01 07:03:41 +00:00
|
|
|
}
|
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
public MovieFile Add(MovieFile movieFile)
|
2013-03-01 07:03:41 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
var addedFile = _mediaFileRepository.Insert(movieFile);
|
2019-12-18 21:56:41 +00:00
|
|
|
if (addedFile.Movie == null)
|
2018-03-14 20:41:36 +00:00
|
|
|
{
|
2019-12-18 21:56:41 +00:00
|
|
|
addedFile.Movie = _movieRepository.Get(movieFile.MovieId);
|
2018-03-14 20:41:36 +00:00
|
|
|
}
|
2019-12-22 22:08:53 +00:00
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
_eventAggregator.PublishEvent(new MovieFileAddedEvent(addedFile));
|
2013-03-01 07:03:41 +00:00
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
return addedFile;
|
2013-03-01 07:03:41 +00:00
|
|
|
}
|
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
public void Update(MovieFile movieFile)
|
2013-03-01 07:03:41 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
_mediaFileRepository.Update(movieFile);
|
2013-03-01 07:03:41 +00:00
|
|
|
}
|
|
|
|
|
2018-11-23 07:03:32 +00:00
|
|
|
public void Update(List<MovieFile> movieFiles)
|
|
|
|
{
|
|
|
|
_mediaFileRepository.UpdateMany(movieFiles);
|
|
|
|
}
|
|
|
|
|
2017-03-04 02:32:52 +00:00
|
|
|
public void Delete(MovieFile movieFile, DeleteMediaFileReason reason)
|
2017-01-03 19:24:55 +00:00
|
|
|
{
|
2022-11-20 18:27:45 +00:00
|
|
|
// Little hack so we have the movie attached for the event consumers
|
2019-12-18 21:56:41 +00:00
|
|
|
if (movieFile.Movie == null)
|
|
|
|
{
|
|
|
|
movieFile.Movie = _movieRepository.Get(movieFile.MovieId);
|
|
|
|
}
|
|
|
|
|
|
|
|
movieFile.Path = Path.Combine(movieFile.Movie.Path, movieFile.RelativePath);
|
2017-01-03 19:24:55 +00:00
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
_mediaFileRepository.Delete(movieFile);
|
2017-03-04 02:32:52 +00:00
|
|
|
_eventAggregator.PublishEvent(new MovieFileDeletedEvent(movieFile, reason));
|
2017-01-03 19:24:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public List<MovieFile> GetFilesByMovie(int movieId)
|
2017-01-02 17:05:55 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
return _mediaFileRepository.GetFilesByMovie(movieId);
|
2017-01-02 17:05:55 +00:00
|
|
|
}
|
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
public List<MovieFile> GetFilesWithoutMediaInfo()
|
2014-04-10 17:58:50 +00:00
|
|
|
{
|
|
|
|
return _mediaFileRepository.GetFilesWithoutMediaInfo();
|
|
|
|
}
|
|
|
|
|
2017-01-02 17:05:55 +00:00
|
|
|
public List<string> FilterExistingFiles(List<string> files, Movie movie)
|
|
|
|
{
|
2017-01-20 00:58:57 +00:00
|
|
|
var movieFiles = GetFilesByMovie(movie.Id).Select(f => Path.Combine(movie.Path, f.RelativePath)).ToList();
|
2017-01-02 17:05:55 +00:00
|
|
|
|
2019-12-22 22:08:53 +00:00
|
|
|
if (!movieFiles.Any())
|
|
|
|
{
|
|
|
|
return files;
|
|
|
|
}
|
2017-01-02 17:05:55 +00:00
|
|
|
|
2017-01-20 00:58:57 +00:00
|
|
|
return files.Except(movieFiles, PathEqualityComparer.Instance).ToList();
|
2017-01-02 17:05:55 +00:00
|
|
|
}
|
|
|
|
|
2017-01-09 03:16:14 +00:00
|
|
|
public List<MovieFile> GetMovies(IEnumerable<int> ids)
|
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
return _mediaFileRepository.Get(ids).ToList();
|
2017-01-09 03:16:14 +00:00
|
|
|
}
|
2017-01-08 22:01:37 +00:00
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
public MovieFile GetMovie(int id)
|
2013-03-06 21:20:33 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
return _mediaFileRepository.Get(id);
|
2013-03-06 21:20:33 +00:00
|
|
|
}
|
2017-01-03 19:24:55 +00:00
|
|
|
|
2021-11-29 04:06:24 +00:00
|
|
|
public List<MovieFile> GetFilesWithRelativePath(int movieId, string relativePath)
|
|
|
|
{
|
|
|
|
return _mediaFileRepository.GetFilesWithRelativePath(movieId, relativePath);
|
|
|
|
}
|
|
|
|
|
2020-06-05 03:35:02 +00:00
|
|
|
public void HandleAsync(MoviesDeletedEvent message)
|
2017-01-03 19:24:55 +00:00
|
|
|
{
|
2020-06-05 03:35:02 +00:00
|
|
|
_mediaFileRepository.DeleteForMovies(message.Movies.Select(m => m.Id).ToList());
|
2017-01-14 00:03:37 +00:00
|
|
|
}
|
2021-11-28 21:42:44 +00:00
|
|
|
|
|
|
|
public static List<string> FilterExistingFiles(List<string> files, List<MovieFile> movieFiles, Movie movie)
|
|
|
|
{
|
|
|
|
var seriesFilePaths = movieFiles.Select(f => Path.Combine(movie.Path, f.RelativePath)).ToList();
|
|
|
|
|
|
|
|
if (!seriesFilePaths.Any())
|
|
|
|
{
|
|
|
|
return files;
|
|
|
|
}
|
|
|
|
|
|
|
|
return files.Except(seriesFilePaths, PathEqualityComparer.Instance).ToList();
|
|
|
|
}
|
2018-03-14 20:41:36 +00:00
|
|
|
}
|
2017-02-25 21:38:52 +00:00
|
|
|
}
|