Sonarr/NzbDrone.Core/Tv/SeasonService.cs

85 lines
6.3 KiB
C#
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

using System.Collections.Generic;
using System.Linq;
using NLog;
using NzbDrone.Common.Eventing;
using NzbDrone.Core.Tv.Events;
namespace NzbDrone.Core.Tv
{
public interface ISeasonService
{
void SetIgnore(int seriesId, int seasonNumber, bool isIgnored);
List<Season> GetSeasonsBySeries(int seriesId);
}
public class SeasonService : ISeasonService,
IHandle<EpisodeInfoAddedEvent>,
IHandle<EpisodeInfoUpdatedEvent>
{
private readonly ISeasonRepository _seasonRepository;
private readonly Logger _logger;
public SeasonService(ISeasonRepository seasonRepository, Logger logger)
{
_seasonRepository = seasonRepository;
_logger = logger;
}
public void SetIgnore(int seriesId, int seasonNumber, bool isIgnored)
{
var season = _seasonRepository.Get(seriesId, seasonNumber);
_logger.Trace("Setting ignore flag on Series:{0} Season:{1} to {2}", seriesId, seasonNumber, isIgnored);
season.Ignored = isIgnored;
season.Episodes.ForEach(e => e.Ignored = isIgnored);
_seasonRepository.Update(season);
_logger.Info("Ignore flag for Series:{0} Season:{1} successfully set to {2}", seriesId, seasonNumber, isIgnored);
}
public List<Season> GetSeasonsBySeries(int seriesId)
{
return _seasonRepository.GetSeasonBySeries(seriesId);
}
public void Handle(EpisodeInfoAddedEvent message)
{
EnsureSeasons(message.Episodes);
}
public void Handle(EpisodeInfoUpdatedEvent message)
{
EnsureSeasons(message.Episodes);
}
private void EnsureSeasons(IEnumerable<Episode> episodes)
{
var seriesGroup = episodes.GroupBy(c => c.SeriesId);
foreach (var group in seriesGroup)
{
var seriesId = group.Key;
var existingSeasons = _seasonRepository.GetSeasonNumbers(seriesId);
var seasonNumbers = group.Select(c => c.SeasonNumber).Distinct();
var missingSeasons = seasonNumbers.Where(seasonNumber => !existingSeasons.Contains(seasonNumber));
var seasonToAdd = missingSeasons.Select(c => new Season()
{
SeriesId = seriesId,
SeasonNumber = c,
Ignored = false
});
_seasonRepository.InsertMany(seasonToAdd.ToList());
}
}
}
}