2018-02-13 18:10:49 +00:00
|
|
|
using System;
|
2013-02-23 17:42:15 +00:00
|
|
|
using System.Collections.Generic;
|
2013-06-04 03:59:24 +00:00
|
|
|
using System.Linq;
|
2017-02-08 00:09:36 +00:00
|
|
|
using Nancy;
|
2018-02-19 07:18:14 +00:00
|
|
|
using NzbDrone.Api.Movies;
|
2017-01-16 21:40:59 +00:00
|
|
|
using NzbDrone.Core.MediaCover;
|
2018-03-14 20:41:36 +00:00
|
|
|
using NzbDrone.Core.Movies;
|
2014-08-26 04:01:32 +00:00
|
|
|
using NzbDrone.SignalR;
|
2019-09-08 01:58:45 +00:00
|
|
|
using Radarr.Http;
|
2013-02-23 17:42:15 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Api.Calendar
|
|
|
|
{
|
2019-09-08 01:58:45 +00:00
|
|
|
public class CalendarModule : RadarrRestModuleWithSignalR<MovieResource, Movie>
|
2013-02-23 17:42:15 +00:00
|
|
|
{
|
2019-09-08 01:58:45 +00:00
|
|
|
protected readonly IMovieService _moviesService;
|
|
|
|
private readonly IMapCoversToLocal _coverMapper;
|
|
|
|
|
2017-01-16 21:40:59 +00:00
|
|
|
public CalendarModule(IBroadcastSignalRMessage signalR,
|
|
|
|
IMovieService moviesService,
|
|
|
|
IMapCoversToLocal coverMapper)
|
2019-09-08 01:58:45 +00:00
|
|
|
: base(signalR, "calendar")
|
2013-02-23 17:42:15 +00:00
|
|
|
{
|
2019-09-08 01:58:45 +00:00
|
|
|
_moviesService = moviesService;
|
|
|
|
_coverMapper = coverMapper;
|
|
|
|
|
2013-11-30 10:53:53 +00:00
|
|
|
GetResourceAll = GetCalendar;
|
2013-12-02 00:30:30 +00:00
|
|
|
}
|
|
|
|
|
2017-01-16 21:40:59 +00:00
|
|
|
private List<MovieResource> GetCalendar()
|
2013-02-23 17:42:15 +00:00
|
|
|
{
|
2013-10-04 02:25:07 +00:00
|
|
|
var start = DateTime.Today;
|
|
|
|
var end = DateTime.Today.AddDays(2);
|
2015-03-28 16:10:30 +00:00
|
|
|
var includeUnmonitored = false;
|
2013-02-23 23:08:22 +00:00
|
|
|
|
2013-02-26 03:30:24 +00:00
|
|
|
var queryStart = Request.Query.Start;
|
|
|
|
var queryEnd = Request.Query.End;
|
2015-03-28 16:10:30 +00:00
|
|
|
var queryIncludeUnmonitored = Request.Query.Unmonitored;
|
2013-02-23 23:08:22 +00:00
|
|
|
|
2019-12-22 22:08:53 +00:00
|
|
|
if (queryStart.HasValue)
|
|
|
|
{
|
|
|
|
start = DateTime.Parse(queryStart.Value);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (queryEnd.HasValue)
|
|
|
|
{
|
|
|
|
end = DateTime.Parse(queryEnd.Value);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (queryIncludeUnmonitored.HasValue)
|
|
|
|
{
|
|
|
|
includeUnmonitored = Convert.ToBoolean(queryIncludeUnmonitored.Value);
|
|
|
|
}
|
2013-02-23 23:08:22 +00:00
|
|
|
|
2017-01-16 21:40:59 +00:00
|
|
|
var resources = _moviesService.GetMoviesBetweenDates(start, end, includeUnmonitored).Select(MapToResource);
|
2013-06-04 03:59:24 +00:00
|
|
|
|
2017-01-16 21:40:59 +00:00
|
|
|
return resources.OrderBy(e => e.InCinemas).ToList();
|
2013-02-23 17:42:15 +00:00
|
|
|
}
|
2019-09-08 01:58:45 +00:00
|
|
|
|
|
|
|
protected MovieResource MapToResource(Movie movie)
|
|
|
|
{
|
2019-12-22 22:08:53 +00:00
|
|
|
if (movie == null)
|
|
|
|
{
|
|
|
|
return null;
|
|
|
|
}
|
2019-09-08 01:58:45 +00:00
|
|
|
|
|
|
|
var resource = movie.ToResource();
|
|
|
|
|
|
|
|
return resource;
|
|
|
|
}
|
2013-02-23 17:42:15 +00:00
|
|
|
}
|
2013-05-31 00:12:20 +00:00
|
|
|
}
|