mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 16:02:29 +01:00
101 lines
3.5 KiB
C#
101 lines
3.5 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
using NLog;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.MediaFiles.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 Delete(EpisodeFile episodeFile);
|
|
bool Exists(string path);
|
|
EpisodeFile GetFileByPath(string path);
|
|
List<EpisodeFile> GetFilesBySeries(int seriesId);
|
|
List<EpisodeFile> GetFilesBySeason(int seriesId, int seasonNumber);
|
|
}
|
|
|
|
public class MediaFileService : IMediaFileService, IHandleAsync<SeriesDeletedEvent>
|
|
{
|
|
private readonly IConfigService _configService;
|
|
private readonly IEpisodeService _episodeService;
|
|
private readonly IMessageAggregator _messageAggregator;
|
|
private readonly Logger _logger;
|
|
private readonly IMediaFileRepository _mediaFileRepository;
|
|
|
|
public MediaFileService(IMediaFileRepository mediaFileRepository, IConfigService configService, IEpisodeService episodeService, IMessageAggregator messageAggregator, Logger logger)
|
|
{
|
|
_mediaFileRepository = mediaFileRepository;
|
|
_configService = configService;
|
|
_episodeService = episodeService;
|
|
_messageAggregator = messageAggregator;
|
|
_logger = logger;
|
|
}
|
|
|
|
public EpisodeFile Add(EpisodeFile episodeFile)
|
|
{
|
|
var addedFile = _mediaFileRepository.Insert(episodeFile);
|
|
_messageAggregator.PublishEvent(new EpisodeFileAddedEvent(addedFile));
|
|
return addedFile;
|
|
}
|
|
|
|
public void Update(EpisodeFile episodeFile)
|
|
{
|
|
_mediaFileRepository.Update(episodeFile);
|
|
}
|
|
|
|
public void Delete(EpisodeFile episodeFile)
|
|
{
|
|
_mediaFileRepository.Delete(episodeFile);
|
|
_messageAggregator.PublishEvent(new EpisodeFileDeletedEvent(episodeFile));
|
|
}
|
|
|
|
public bool Exists(string path)
|
|
{
|
|
return GetFileByPath(path) != null;
|
|
}
|
|
|
|
public EpisodeFile GetFileByPath(string path)
|
|
{
|
|
return _mediaFileRepository.GetFileByPath(path.Normalize());
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesBySeries(int seriesId)
|
|
{
|
|
return _mediaFileRepository.GetFilesBySeries(seriesId);
|
|
}
|
|
|
|
public List<EpisodeFile> GetFilesBySeason(int seriesId, int seasonNumber)
|
|
{
|
|
return _mediaFileRepository.GetFilesBySeason(seriesId, seasonNumber);
|
|
}
|
|
|
|
public void HandleAsync(SeriesDeletedEvent message)
|
|
{
|
|
var files = GetFilesBySeries(message.Series.Id);
|
|
_mediaFileRepository.DeleteMany(files);
|
|
}
|
|
|
|
public FileInfo CalculateFilePath(Series series, int seasonNumber, string fileName, string extension)
|
|
{
|
|
string path = series.Path;
|
|
if (series.SeasonFolder)
|
|
{
|
|
var seasonFolder = _configService.SortingSeasonFolderFormat
|
|
.Replace("%0s", seasonNumber.ToString("00"))
|
|
.Replace("%s", seasonNumber.ToString());
|
|
|
|
path = Path.Combine(path, seasonFolder);
|
|
}
|
|
|
|
path = Path.Combine(path, fileName + extension);
|
|
|
|
return new FileInfo(path);
|
|
}
|
|
}
|
|
} |