using System; using System.Collections.Generic; using System.IO; using System.Linq; using Ninject; using NLog; using NzbDrone.Core.Model.Notification; using NzbDrone.Core.Repository; namespace NzbDrone.Core.Providers.Jobs { /// /// This job processes newly added jobs by downloading their info /// from TheTVDB.org and doing a disk scan. this job should only /// public class ImportNewSeriesJob : IJob { private readonly SeriesProvider _seriesProvider; private readonly EpisodeProvider _episodeProvider; private readonly MediaFileProvider _mediaFileProvider; private readonly UpdateInfoJob _updateInfoJob; private readonly DiskScanJob _diskScanJob; private static readonly Logger Logger = LogManager.GetCurrentClassLogger(); private List _attemptedSeries; [Inject] public ImportNewSeriesJob(SeriesProvider seriesProvider, EpisodeProvider episodeProvider, MediaFileProvider mediaFileProvider, UpdateInfoJob updateInfoJob, DiskScanJob diskScanJob) { _seriesProvider = seriesProvider; _episodeProvider = episodeProvider; _mediaFileProvider = mediaFileProvider; _updateInfoJob = updateInfoJob; _diskScanJob = diskScanJob; } public string Name { get { return "New Series Update"; } } public int DefaultInterval { get { return 1; } } public void Start(ProgressNotification notification, int targetId, int secondaryTargetId ) { _attemptedSeries = new List(); ScanSeries(notification); } private void ScanSeries(ProgressNotification notification) { var syncList = _seriesProvider.GetAllSeries().Where(s => s.LastInfoSync == null && !_attemptedSeries.Contains(s.SeriesId)).ToList(); if (syncList.Count == 0) { return; } foreach (var currentSeries in syncList) { try { _attemptedSeries.Add(currentSeries.SeriesId); notification.CurrentMessage = String.Format("Searching for '{0}'", new DirectoryInfo(currentSeries.Path).Name); _updateInfoJob.Start(notification, currentSeries.SeriesId, 0); _diskScanJob.Start(notification, currentSeries.SeriesId, 0); var updatedSeries = _seriesProvider.GetSeries(currentSeries.SeriesId); AutoIgnoreSeasons(updatedSeries.SeriesId); notification.CurrentMessage = String.Format("{0} was successfully imported", updatedSeries.Title); } catch (Exception e) { Logger.ErrorException(e.Message, e); } } //Keep scanning until there no more shows left. ScanSeries(notification); } public void AutoIgnoreSeasons(int seriesId) { var episodeFiles = _mediaFileProvider.GetSeriesFiles(seriesId); if (episodeFiles.Count() != 0) { var seasons = _episodeProvider.GetSeasons(seriesId); var currentSeasons = seasons.Max(); foreach (var season in seasons) { if (season != currentSeasons && !episodeFiles.Any(e => e.SeasonNumber == season)) { _episodeProvider.SetSeasonIgnore(seriesId, season, true); } } } } } }