mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-16 11:37:58 +02:00
157 lines
6.5 KiB
C#
157 lines
6.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Marr.Data.QGen;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Datastore;
|
|
|
|
|
|
namespace NzbDrone.Core.Tv
|
|
{
|
|
public interface IEpisodeRepository : IBasicRepository<Episode>
|
|
{
|
|
Episode Get(int seriesId, int season, int episodeNumber);
|
|
Episode Find(int seriesId, int season, int episodeNumber);
|
|
Episode Get(int seriesId, DateTime date);
|
|
Episode Find(int seriesId, DateTime date);
|
|
List<Episode> GetEpisodes(int seriesId);
|
|
List<Episode> GetEpisodes(int seriesId, int seasonNumber);
|
|
List<Episode> GetEpisodeByFileId(int fileId);
|
|
PagingSpec<Episode> EpisodesWithoutFiles(PagingSpec<Episode> pagingSpec, bool includeSpecials);
|
|
Episode GetEpisodeBySceneNumbering(int seriesId, int seasonNumber, int episodeNumber);
|
|
Episode FindEpisodeBySceneNumbering(int seriesId, int seasonNumber, int episodeNumber);
|
|
List<Episode> EpisodesWithFiles();
|
|
List<Episode> EpisodesBetweenDates(DateTime startDate, DateTime endDate);
|
|
void SetMonitoredFlat(Episode episode, bool monitored);
|
|
void SetMonitoredBySeason(int seriesId, int seasonNumber, bool monitored);
|
|
void SetFileId(int episodeId, int fileId);
|
|
}
|
|
|
|
public class EpisodeRepository : BasicRepository<Episode>, IEpisodeRepository
|
|
{
|
|
private readonly IDatabase _database;
|
|
|
|
public EpisodeRepository(IDatabase database, IMessageAggregator messageAggregator)
|
|
: base(database, messageAggregator)
|
|
{
|
|
_database = database;
|
|
}
|
|
|
|
public Episode Get(int seriesId, int season, int episodeNumber)
|
|
{
|
|
return Query.Single(s => s.SeriesId == seriesId && s.SeasonNumber == season && s.EpisodeNumber == episodeNumber);
|
|
}
|
|
|
|
public Episode Find(int seriesId, int season, int episodeNumber)
|
|
{
|
|
return Query.SingleOrDefault(s => s.SeriesId == seriesId && s.SeasonNumber == season && s.EpisodeNumber == episodeNumber);
|
|
}
|
|
|
|
public Episode Get(int seriesId, DateTime date)
|
|
{
|
|
return Query.Single(s => s.SeriesId == seriesId && s.AirDate.HasValue && s.AirDate.Value.Date == date.Date);
|
|
}
|
|
|
|
public Episode Find(int seriesId, DateTime date)
|
|
{
|
|
return Query.SingleOrDefault(s => s.SeriesId == seriesId && s.AirDate.HasValue && s.AirDate.Value.Date == date.Date);
|
|
}
|
|
|
|
public List<Episode> GetEpisodes(int seriesId)
|
|
{
|
|
return Query.Where(s => s.SeriesId == seriesId).ToList();
|
|
}
|
|
|
|
public List<Episode> GetEpisodes(int seriesId, int seasonNumber)
|
|
{
|
|
return Query.Where(s => s.SeriesId == seriesId && s.SeasonNumber == seasonNumber).ToList();
|
|
}
|
|
|
|
public List<Episode> GetEpisodeByFileId(int fileId)
|
|
{
|
|
return Query.Where(e => e.EpisodeFileId == fileId).ToList();
|
|
}
|
|
|
|
public PagingSpec<Episode> EpisodesWithoutFiles(PagingSpec<Episode> pagingSpec, bool includeSpecials)
|
|
{
|
|
var currentTime = DateTime.UtcNow;
|
|
var startingSeasonNumber = 1;
|
|
|
|
if (includeSpecials)
|
|
{
|
|
startingSeasonNumber = 0;
|
|
}
|
|
|
|
pagingSpec.Records = GetEpisodesWithoutFilesQuery(pagingSpec, currentTime, startingSeasonNumber).ToList();
|
|
pagingSpec.TotalRecords = GetEpisodesWithoutFilesQuery(pagingSpec, currentTime, startingSeasonNumber).GetRowCount();
|
|
|
|
return pagingSpec;
|
|
}
|
|
|
|
public Episode GetEpisodeBySceneNumbering(int seriesId, int seasonNumber, int episodeNumber)
|
|
{
|
|
return Query.Single(s => s.SeriesId == seriesId && s.SeasonNumber == seasonNumber && s.SceneEpisodeNumber == episodeNumber);
|
|
}
|
|
|
|
public Episode FindEpisodeBySceneNumbering(int seriesId, int seasonNumber, int episodeNumber)
|
|
{
|
|
return Query.SingleOrDefault(s => s.SeriesId == seriesId && s.SeasonNumber == seasonNumber && s.SceneEpisodeNumber == episodeNumber);
|
|
}
|
|
|
|
public List<Episode> EpisodesWithFiles()
|
|
{
|
|
return Query.Where(s => s.EpisodeFileId != 0).ToList();
|
|
}
|
|
|
|
public List<Episode> EpisodesBetweenDates(DateTime startDate, DateTime endDate)
|
|
{
|
|
return Query.Join<Episode, Series>(JoinType.Inner, e => e.Series, (e, s) => e.SeriesId == s.Id)
|
|
.Where<Episode>(e => e.AirDateUtc >= startDate)
|
|
.AndWhere(e => e.AirDateUtc <= endDate)
|
|
.AndWhere(e => e.Monitored)
|
|
.AndWhere(e => e.Series.Monitored)
|
|
.ToList();
|
|
}
|
|
|
|
public void SetMonitoredFlat(Episode episode, bool monitored)
|
|
{
|
|
episode.Monitored = monitored;
|
|
SetFields(episode, p => p.Monitored);
|
|
}
|
|
|
|
public void SetMonitoredBySeason(int seriesId, int seasonNumber, bool monitored)
|
|
{
|
|
var mapper = _database.GetDataMapper();
|
|
|
|
mapper.AddParameter("seriesId", seriesId);
|
|
mapper.AddParameter("seasonNumber", seasonNumber);
|
|
mapper.AddParameter("monitored", monitored);
|
|
|
|
const string sql = "UPDATE Episodes " +
|
|
"SET Monitored = @monitored " +
|
|
"WHERE SeriesId = @seriesId " +
|
|
"AND SeasonNumber = @seasonNumber";
|
|
|
|
mapper.ExecuteNonQuery(sql);
|
|
}
|
|
|
|
public void SetFileId(int episodeId, int fileId)
|
|
{
|
|
SetFields(new Episode { Id = episodeId, EpisodeFileId = fileId }, episode => episode.EpisodeFileId);
|
|
}
|
|
|
|
private SortBuilder<Episode> GetEpisodesWithoutFilesQuery(PagingSpec<Episode> pagingSpec, DateTime currentTime, int startingSeasonNumber)
|
|
{
|
|
return Query.Join<Episode, Series>(JoinType.Inner, e => e.Series, (e, s) => e.SeriesId == s.Id)
|
|
.Where(e => e.EpisodeFileId == 0)
|
|
.AndWhere(e => e.SeasonNumber >= startingSeasonNumber)
|
|
.AndWhere(e => e.AirDateUtc <= currentTime)
|
|
.AndWhere(e => e.Monitored)
|
|
.AndWhere(e => e.Series.Monitored)
|
|
.OrderBy(pagingSpec.OrderByClause(), pagingSpec.ToSortDirection())
|
|
.Skip(pagingSpec.PagingOffset())
|
|
.Take(pagingSpec.PageSize);
|
|
}
|
|
}
|
|
}
|