mirror of https://github.com/Radarr/Radarr
107 lines
3.6 KiB
C#
107 lines
3.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Ical.Net;
|
|
using Ical.Net.CalendarComponents;
|
|
using Ical.Net.DataTypes;
|
|
using Ical.Net.Serialization;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.Tags;
|
|
using Radarr.Http;
|
|
|
|
namespace Radarr.Api.V3.Calendar
|
|
{
|
|
[V3FeedController("calendar")]
|
|
public class CalendarFeedController : Controller
|
|
{
|
|
private readonly IMovieService _movieService;
|
|
private readonly ITagService _tagService;
|
|
|
|
public CalendarFeedController(IMovieService movieService, ITagService tagService)
|
|
{
|
|
_movieService = movieService;
|
|
_tagService = tagService;
|
|
}
|
|
|
|
[HttpGet("Radarr.ics")]
|
|
public IActionResult GetCalendarFeed(int pastDays = 7, int futureDays = 28, string tagList = "", bool unmonitored = false)
|
|
{
|
|
var start = DateTime.Today.AddDays(-pastDays);
|
|
var end = DateTime.Today.AddDays(futureDays);
|
|
var tags = new List<int>();
|
|
|
|
if (tagList.IsNotNullOrWhiteSpace())
|
|
{
|
|
tags.AddRange(tagList.Split(',').Select(_tagService.GetTag).Select(t => t.Id));
|
|
}
|
|
|
|
var movies = _movieService.GetMoviesBetweenDates(start, end, unmonitored);
|
|
var calendar = new Ical.Net.Calendar
|
|
{
|
|
ProductId = "-//radarr.video//Radarr//EN"
|
|
};
|
|
|
|
var calendarName = "Radarr Movies Calendar";
|
|
calendar.AddProperty(new CalendarProperty("NAME", calendarName));
|
|
calendar.AddProperty(new CalendarProperty("X-WR-CALNAME", calendarName));
|
|
|
|
foreach (var movie in movies.OrderBy(v => v.Added))
|
|
{
|
|
if (tags.Any() && tags.None(movie.Tags.Contains))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
CreateEvent(calendar, movie, "cinematic");
|
|
CreateEvent(calendar, movie, "digital");
|
|
CreateEvent(calendar, movie, "physical");
|
|
}
|
|
|
|
var serializer = (IStringSerializer)new SerializerFactory().Build(calendar.GetType(), new SerializationContext());
|
|
var icalendar = serializer.SerializeToString(calendar);
|
|
|
|
return Content(icalendar, "text/calendar");
|
|
}
|
|
|
|
private void CreateEvent(Ical.Net.Calendar calendar, Movie movie, string releaseType)
|
|
{
|
|
var date = movie.InCinemas;
|
|
string eventType = "_cinemas";
|
|
string summaryText = "(Theatrical Release)";
|
|
|
|
if (releaseType == "digital")
|
|
{
|
|
date = movie.DigitalRelease;
|
|
eventType = "_digital";
|
|
summaryText = "(Digital Release)";
|
|
}
|
|
else if (releaseType == "physical")
|
|
{
|
|
date = movie.PhysicalRelease;
|
|
eventType = "_physical";
|
|
summaryText = "(Physical Release)";
|
|
}
|
|
|
|
if (!date.HasValue)
|
|
{
|
|
return;
|
|
}
|
|
|
|
var occurrence = calendar.Create<CalendarEvent>();
|
|
occurrence.Uid = "Radarr_movie_" + movie.Id + eventType;
|
|
occurrence.Status = movie.Status == MovieStatusType.Announced ? EventStatus.Tentative : EventStatus.Confirmed;
|
|
|
|
occurrence.Start = new CalDateTime(date.Value);
|
|
occurrence.End = occurrence.Start;
|
|
occurrence.IsAllDay = true;
|
|
|
|
occurrence.Description = movie.Overview;
|
|
occurrence.Categories = new List<string>() { movie.Studio };
|
|
|
|
occurrence.Summary = $"{movie.Title} " + summaryText;
|
|
}
|
|
}
|
|
}
|