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.
119 lines
3.8 KiB
119 lines
3.8 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Core.MediaFiles.Events;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Core.Tv.Events;
|
|
|
|
namespace NzbDrone.Core.Extras.Files
|
|
{
|
|
public interface IExtraFileService<TExtraFile>
|
|
where TExtraFile : ExtraFile, new()
|
|
{
|
|
List<TExtraFile> GetFilesBySeries(int seriesId);
|
|
List<TExtraFile> GetFilesByEpisodeFile(int episodeFileId);
|
|
TExtraFile FindByPath(string path);
|
|
void Upsert(TExtraFile extraFile);
|
|
void Upsert(List<TExtraFile> extraFiles);
|
|
void Delete(int id);
|
|
void DeleteMany(IEnumerable<int> ids);
|
|
}
|
|
|
|
public abstract class ExtraFileService<TExtraFile> : IExtraFileService<TExtraFile>,
|
|
IHandleAsync<SeriesDeletedEvent>,
|
|
IHandleAsync<EpisodeFileDeletedEvent>
|
|
where TExtraFile : ExtraFile, new()
|
|
{
|
|
private readonly IExtraFileRepository<TExtraFile> _repository;
|
|
private readonly ISeriesService _seriesService;
|
|
private readonly IDiskProvider _diskProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public ExtraFileService(IExtraFileRepository<TExtraFile> repository,
|
|
ISeriesService seriesService,
|
|
IDiskProvider diskProvider,
|
|
Logger logger)
|
|
{
|
|
_repository = repository;
|
|
_seriesService = seriesService;
|
|
_diskProvider = diskProvider;
|
|
_logger = logger;
|
|
}
|
|
|
|
public List<TExtraFile> GetFilesBySeries(int seriesId)
|
|
{
|
|
return _repository.GetFilesBySeries(seriesId);
|
|
}
|
|
|
|
public List<TExtraFile> GetFilesByEpisodeFile(int episodeFileId)
|
|
{
|
|
return _repository.GetFilesByEpisodeFile(episodeFileId);
|
|
}
|
|
|
|
public TExtraFile FindByPath(string path)
|
|
{
|
|
return _repository.FindByPath(path);
|
|
}
|
|
|
|
public void Upsert(TExtraFile extraFile)
|
|
{
|
|
Upsert(new List<TExtraFile> { extraFile });
|
|
}
|
|
|
|
public void Upsert(List<TExtraFile> extraFiles)
|
|
{
|
|
extraFiles.ForEach(m =>
|
|
{
|
|
m.LastUpdated = DateTime.UtcNow;
|
|
|
|
if (m.Id == 0)
|
|
{
|
|
m.Added = m.LastUpdated;
|
|
}
|
|
});
|
|
|
|
_repository.InsertMany(extraFiles.Where(m => m.Id == 0).ToList());
|
|
_repository.UpdateMany(extraFiles.Where(m => m.Id > 0).ToList());
|
|
}
|
|
|
|
public void Delete(int id)
|
|
{
|
|
_repository.Delete(id);
|
|
}
|
|
|
|
public void DeleteMany(IEnumerable<int> ids)
|
|
{
|
|
_repository.DeleteMany(ids);
|
|
}
|
|
|
|
public void HandleAsync(SeriesDeletedEvent message)
|
|
{
|
|
_logger.Debug("Deleting Extra from database for series: {0}", message.Series);
|
|
_repository.DeleteForSeries(message.Series.Id);
|
|
}
|
|
|
|
public void HandleAsync(EpisodeFileDeletedEvent message)
|
|
{
|
|
var episodeFile = message.EpisodeFile;
|
|
var series = _seriesService.GetSeries(message.EpisodeFile.SeriesId);
|
|
|
|
foreach (var extra in _repository.GetFilesByEpisodeFile(episodeFile.Id))
|
|
{
|
|
var path = Path.Combine(series.Path, extra.RelativePath);
|
|
|
|
if (_diskProvider.FileExists(path))
|
|
{
|
|
_diskProvider.DeleteFile(path);
|
|
}
|
|
}
|
|
|
|
_logger.Debug("Deleting Extra from database for episode file: {0}", episodeFile);
|
|
_repository.DeleteForEpisodeFile(episodeFile.Id);
|
|
}
|
|
}
|
|
}
|