You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
4.4 KiB
129 lines
4.4 KiB
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Core.Tv.Events;
|
|
|
|
namespace NzbDrone.Core.MediaFiles
|
|
{
|
|
public interface IMediaFileService
|
|
{
|
|
EpisodeFile Add(EpisodeFile episodeFile);
|
|
void Update(EpisodeFile episodeFile);
|
|
void Update(List<EpisodeFile> episodeFiles);
|
|
void Delete(EpisodeFile episodeFile, DeleteMediaFileReason reason);
|
|
List<EpisodeFile> GetFilesBySeries(int seriesId);
|
|
List<EpisodeFile> GetFilesBySeason(int seriesId, int seasonNumber);
|
|
List<EpisodeFile> GetFiles(IEnumerable<int> ids);
|
|
List<EpisodeFile> GetFilesWithoutMediaInfo();
|
|
List<string> FilterExistingFiles(List<string> files, Series series);
|
|
EpisodeFile Get(int id);
|
|
List<EpisodeFile> Get(IEnumerable<int> ids);
|
|
List<EpisodeFile> GetFilesWithRelativePath(int seriesId, string relativePath);
|
|
}
|
|
|
|
public class MediaFileService : IMediaFileService, IHandleAsync<SeriesDeletedEvent>
|
|
{
|
|
private readonly IEventAggregator _eventAggregator;
|
|
private readonly IMediaFileRepository _mediaFileRepository;
|
|
private readonly Logger _logger;
|
|
|
|
public MediaFileService(IMediaFileRepository mediaFileRepository, IEventAggregator eventAggregator, Logger logger)
|
|
{
|
|
_mediaFileRepository = mediaFileRepository;
|
|
_eventAggregator = eventAggregator;
|
|
_logger = logger;
|
|
}
|
|
|
|
public EpisodeFile Add(EpisodeFile episodeFile)
|
|
{
|
|
var addedFile = _mediaFileRepository.Insert(episodeFile);
|
|
_eventAggregator.PublishEvent(new EpisodeFileAddedEvent(addedFile));
|
|
return addedFile;
|
|
}
|
|
|
|
public void Update(EpisodeFile episodeFile)
|
|
{
|
|
_mediaFileRepository.Update(episodeFile);
|
|
}
|
|
|
|
public void Update(List<EpisodeFile> episodeFiles)
|
|
{
|
|
_mediaFileRepository.UpdateMany(episodeFiles);
|
|
}
|
|
|
|
public void Delete(EpisodeFile episodeFile, DeleteMediaFileReason reason)
|
|
{
|
|
// Little hack so we have the episodes and series attached for the event consumers
|
|
episodeFile.Episodes.LazyLoad();
|
|
episodeFile.Path = Path.Combine(episodeFile.Series.Value.Path, episodeFile.RelativePath);
|
|
|
|
_mediaFileRepository.Delete(episodeFile);
|
|
_eventAggregator.PublishEvent(new EpisodeFileDeletedEvent(episodeFile, reason));
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesBySeries(int seriesId)
|
|
{
|
|
return _mediaFileRepository.GetFilesBySeries(seriesId);
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesBySeason(int seriesId, int seasonNumber)
|
|
{
|
|
return _mediaFileRepository.GetFilesBySeason(seriesId, seasonNumber);
|
|
}
|
|
|
|
public List<EpisodeFile> GetFiles(IEnumerable<int> ids)
|
|
{
|
|
return _mediaFileRepository.Get(ids).ToList();
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesWithoutMediaInfo()
|
|
{
|
|
return _mediaFileRepository.GetFilesWithoutMediaInfo();
|
|
}
|
|
|
|
public List<string> FilterExistingFiles(List<string> files, Series series)
|
|
{
|
|
var seriesFiles = GetFilesBySeries(series.Id);
|
|
|
|
return FilterExistingFiles(files, seriesFiles, series);
|
|
}
|
|
|
|
public EpisodeFile Get(int id)
|
|
{
|
|
return _mediaFileRepository.Get(id);
|
|
}
|
|
|
|
public List<EpisodeFile> Get(IEnumerable<int> ids)
|
|
{
|
|
return _mediaFileRepository.Get(ids).ToList();
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesWithRelativePath(int seriesId, string relativePath)
|
|
{
|
|
return _mediaFileRepository.GetFilesWithRelativePath(seriesId, relativePath);
|
|
}
|
|
|
|
public void HandleAsync(SeriesDeletedEvent message)
|
|
{
|
|
_mediaFileRepository.DeleteForSeries(message.Series.Select(s => s.Id).ToList());
|
|
}
|
|
|
|
public static List<string> FilterExistingFiles(List<string> files, List<EpisodeFile> seriesFiles, Series series)
|
|
{
|
|
var seriesFilePaths = seriesFiles.Select(f => Path.Combine(series.Path, f.RelativePath)).ToList();
|
|
|
|
if (!seriesFilePaths.Any())
|
|
{
|
|
return files;
|
|
}
|
|
|
|
return files.Except(seriesFilePaths, PathEqualityComparer.Instance).ToList();
|
|
}
|
|
}
|
|
}
|