using System.Collections.Generic; using System.Linq; using System; using NLog; using NzbDrone.Common.Eventing; using NzbDrone.Core.Download; using NzbDrone.Core.ExternalNotification; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.Tv; using NzbDrone.Core.Model.Notification; using NzbDrone.Core.Providers; namespace NzbDrone.Core.Jobs { public class RenameSeasonJob : IJob { private readonly IMediaFileService _mediaFileService; private readonly DiskScanProvider _diskScanProvider; private readonly ISeriesService _seriesService; private readonly MetadataProvider _metadataProvider; private readonly ISeriesRepository _seriesRepository; private readonly IEventAggregator _eventAggregator; private static readonly Logger logger = LogManager.GetCurrentClassLogger(); public RenameSeasonJob(IMediaFileService mediaFileService, DiskScanProvider diskScanProvider, ISeriesService seriesService, MetadataProvider metadataProvider, ISeriesRepository seriesRepository, IEventAggregator eventAggregator) { _mediaFileService = mediaFileService; _diskScanProvider = diskScanProvider; _seriesService = seriesService; _metadataProvider = metadataProvider; _seriesRepository = seriesRepository; _eventAggregator = eventAggregator; } public string Name { get { return "Rename Season"; } } public TimeSpan DefaultInterval { get { return TimeSpan.FromTicks(0); } } public void Start(ProgressNotification notification, dynamic options) { if (options == null || options.SeriesId <= 0) throw new ArgumentException("options"); if (options.SeasonNumber < 0) throw new ArgumentException("options.SeasonNumber"); var series = _seriesRepository.Get((int)options.SeriesId); notification.CurrentMessage = String.Format("Renaming episodes for {0} Season {1}", series.Title, options.SeasonNumber); logger.Debug("Getting episodes from database for series: {0} and season: {1}", options.SeriesId, options.SeasonNumber); IList episodeFiles = _mediaFileService.GetFilesBySeason((int)options.SeriesId, (int)options.SeasonNumber); if (episodeFiles == null || !episodeFiles.Any()) { logger.Warn("No episodes in database found for series: {0} and season: {1}.", options.SeriesId, options.SeasonNumber); return; } var newEpisodeFiles = new List(); var oldEpisodeFiles = new List(); foreach (var episodeFile in episodeFiles) { try { var oldFile = new EpisodeFile(episodeFile); var newFile = _diskScanProvider.MoveEpisodeFile(episodeFile); if (newFile != null) { newEpisodeFiles.Add(newFile); oldEpisodeFiles.Add(oldFile); } } catch (Exception e) { logger.WarnException("An error has occurred while renaming file", e); } } if (!oldEpisodeFiles.Any()) { logger.Trace("No episodes were renamed for: {0} Season {1}, no changes were made", series.Title, options.SeasonNumber); notification.CurrentMessage = String.Format("Rename completed for: {0} Season {1}, no changes were made", series.Title, options.SeasonNumber); return; } //Remove & Create Metadata for episode files //Todo: Add a metadata manager to avoid this hack _metadataProvider.RemoveForEpisodeFiles(oldEpisodeFiles); _metadataProvider.CreateForEpisodeFiles(newEpisodeFiles); //Start AfterRename var message = String.Format("Renamed: Series {0}, Season: {1}", series.Title, options.SeasonNumber); _eventAggregator.Publish(new SeriesRenamedEvent(series)); notification.CurrentMessage = String.Format("Rename completed for {0} Season {1}", series.Title, options.SeasonNumber); } } }