Radarr/src/Radarr.Api.V3/Calendar/CalendarFeedController.cs

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;
}
}
}