45 lines
1.8 KiB
C#
45 lines
1.8 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.DecisionEngine.Specifications;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.SignalR;
|
|
using Sonarr.Api.V3.Episodes;
|
|
using Sonarr.Http.Extensions;
|
|
|
|
namespace Sonarr.Api.V3.Calendar
|
|
{
|
|
public class CalendarModule : EpisodeModuleWithSignalR
|
|
{
|
|
public CalendarModule(IEpisodeService episodeService,
|
|
ISeriesService seriesService,
|
|
IUpgradableSpecification ugradableSpecification,
|
|
IBroadcastSignalRMessage signalRBroadcaster)
|
|
: base(episodeService, seriesService, ugradableSpecification, signalRBroadcaster, "calendar")
|
|
{
|
|
GetResourceAll = GetCalendar;
|
|
}
|
|
|
|
private List<EpisodeResource> GetCalendar()
|
|
{
|
|
var start = DateTime.Today;
|
|
var end = DateTime.Today.AddDays(2);
|
|
var includeUnmonitored = Request.GetBooleanQueryParameter("unmonitored");
|
|
var includeSeries = Request.GetBooleanQueryParameter("includeSeries");
|
|
var includeEpisodeFile = Request.GetBooleanQueryParameter("includeEpisodeFile");
|
|
var includeEpisodeImages = Request.GetBooleanQueryParameter("includeEpisodeImages");
|
|
|
|
var queryStart = Request.Query.Start;
|
|
var queryEnd = Request.Query.End;
|
|
|
|
if (queryStart.HasValue) start = DateTime.Parse(queryStart.Value);
|
|
if (queryEnd.HasValue) end = DateTime.Parse(queryEnd.Value);
|
|
|
|
var resources = MapToResource(_episodeService.EpisodesBetweenDates(start, end, includeUnmonitored), includeSeries, includeEpisodeFile, includeEpisodeImages);
|
|
|
|
return resources.OrderBy(e => e.AirDateUtc).ToList();
|
|
}
|
|
}
|
|
}
|