#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: RequestsModule.cs // Created By: Jamie Rees // // Permission is hereby granted, free of charge, to any person obtaining // a copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to // permit persons to whom the Software is furnished to do so, subject to // the following conditions: // // The above copyright notice and this permission notice shall be // included in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // ************************************************************************/ #endregion using System; using System.Linq; using Nancy; using Nancy.Responses.Negotiation; using Nancy.Security; using PlexRequests.Core; using PlexRequests.Core.SettingModels; using PlexRequests.Services.Interfaces; using PlexRequests.Services.Notification; using PlexRequests.Store; using PlexRequests.UI.Models; using PlexRequests.Helpers; using PlexRequests.UI.Helpers; using System.Collections.Generic; using PlexRequests.Api.Interfaces; using System.Threading.Tasks; namespace PlexRequests.UI.Modules { public class RequestsModule : BaseAuthModule { public RequestsModule( IRequestService service, ISettingsService prSettings, ISettingsService plex, INotificationService notify, ISettingsService sonarrSettings, ISettingsService sickRageSettings, ISettingsService cpSettings, ICouchPotatoApi cpApi, ISonarrApi sonarrApi, ISickRageApi sickRageApi, ICacheProvider cache) : base("requests", prSettings) { Service = service; PrSettings = prSettings; PlexSettings = plex; NotificationService = notify; SonarrSettings = sonarrSettings; SickRageSettings = sickRageSettings; CpSettings = cpSettings; SonarrApi = sonarrApi; SickRageApi = sickRageApi; CpApi = cpApi; Cache = cache; Get["/"] = _ => LoadRequests(); Get["/movies"] = _ => GetMovies(); Get["/tvshows"] = _ => GetTvShows(); Get["/albums"] = _ => GetAlbumRequests(); Post["/delete"] = _ => DeleteRequest((int)Request.Form.id); Post["/reportissue"] = _ => ReportIssue((int)Request.Form.requestId, (IssueState)(int)Request.Form.issue, null); Post["/reportissuecomment"] = _ => ReportIssue((int)Request.Form.requestId, IssueState.Other, (string)Request.Form.commentArea); Post["/clearissues"] = _ => ClearIssue((int)Request.Form.Id); Post["/changeavailability"] = _ => ChangeRequestAvailability((int)Request.Form.Id, (bool)Request.Form.Available); Post["/addnote"] = _ => AddNote((int)Request.Form.requestId, (string)Request.Form.noteArea); } private IRequestService Service { get; } private INotificationService NotificationService { get; } private ISettingsService PrSettings { get; } private ISettingsService PlexSettings { get; } private ISettingsService SonarrSettings { get; } private ISettingsService SickRageSettings { get; } private ISettingsService CpSettings { get; } private ISonarrApi SonarrApi { get; } private ISickRageApi SickRageApi { get; } private ICouchPotatoApi CpApi { get; } private ICacheProvider Cache { get; } private Negotiator LoadRequests() { var settings = PrSettings.GetSettings(); return View["Index", settings]; } private Response GetMovies() // TODO: async await the API calls { var settings = PrSettings.GetSettings(); List taskList = new List(); List dbMovies = new List(); taskList.Add(Task.Factory.StartNew(() => { return Service.GetAll().Where(x => x.Type == RequestType.Movie); }).ContinueWith((t) => { dbMovies = t.Result.ToList(); if (settings.UsersCanViewOnlyOwnRequests && !IsAdmin) { dbMovies = dbMovies.Where(x => x.UserHasRequested(Username)).ToList(); } })); List qualities = new List(); if (IsAdmin) { var cpSettings = CpSettings.GetSettings(); if (cpSettings.Enabled) { taskList.Add(Task.Factory.StartNew(() => { return Cache.GetOrSet(CacheKeys.CouchPotatoQualityProfiles, () => { return CpApi.GetProfiles(cpSettings.FullUri, cpSettings.ApiKey); // TODO: cache this! }); }).ContinueWith((t) => { qualities = t.Result.list.Select(x => new QualityModel() { Id = x._id, Name = x.label }).ToList(); })); } } Task.WaitAll(taskList.ToArray()); var viewModel = dbMovies.Select(movie => { return new RequestViewModel { ProviderId = movie.ProviderId, Type = movie.Type, Status = movie.Status, ImdbId = movie.ImdbId, Id = movie.Id, PosterPath = movie.PosterPath, ReleaseDate = movie.ReleaseDate, ReleaseDateTicks = movie.ReleaseDate.Ticks, RequestedDate = movie.RequestedDate, Released = DateTime.Now > movie.ReleaseDate, RequestedDateTicks = DateTimeHelper.OffsetUTCDateTime(movie.RequestedDate, DateTimeOffset).Ticks, Approved = movie.Available || movie.Approved, Title = movie.Title, Overview = movie.Overview, RequestedUsers = IsAdmin ? movie.AllUsers.ToArray() : new string[] { }, ReleaseYear = movie.ReleaseDate.Year.ToString(), Available = movie.Available, Admin = IsAdmin, Issues = movie.Issues.ToString().CamelCaseToWords(), OtherMessage = movie.OtherMessage, AdminNotes = movie.AdminNote, Qualities = qualities.ToArray() }; }).ToList(); return Response.AsJson(viewModel); } private Response GetTvShows() // TODO: async await the API calls { var settings = PrSettings.GetSettings(); List taskList = new List(); List dbTv = new List(); taskList.Add(Task.Factory.StartNew(() => { return Service.GetAll().Where(x => x.Type == RequestType.TvShow); }).ContinueWith((t) => { dbTv = t.Result.ToList(); if (settings.UsersCanViewOnlyOwnRequests && !IsAdmin) { dbTv = dbTv.Where(x => x.UserHasRequested(Username)).ToList(); } })); List qualities = new List(); if (IsAdmin) { var sonarrSettings = SonarrSettings.GetSettings(); if (sonarrSettings.Enabled) { taskList.Add(Task.Factory.StartNew(() => { return Cache.GetOrSet(CacheKeys.SonarrQualityProfiles, () => { return SonarrApi.GetProfiles(sonarrSettings.ApiKey, sonarrSettings.FullUri); // TODO: cache this! }); }).ContinueWith((t) => { qualities = t.Result.Select(x => new QualityModel() { Id = x.id.ToString(), Name = x.name }).ToList(); })); } else { var sickRageSettings = SickRageSettings.GetSettings(); if (sickRageSettings.Enabled) { qualities = sickRageSettings.Qualities.Select(x => new QualityModel() { Id = x.Key, Name = x.Value }).ToList(); } } } Task.WaitAll(taskList.ToArray()); var viewModel = dbTv.Select(tv => { return new RequestViewModel { ProviderId = tv.ProviderId, Type = tv.Type, Status = tv.Status, ImdbId = tv.ImdbId, Id = tv.Id, PosterPath = tv.PosterPath, ReleaseDate = tv.ReleaseDate, ReleaseDateTicks = tv.ReleaseDate.Ticks, RequestedDate = tv.RequestedDate, RequestedDateTicks = DateTimeHelper.OffsetUTCDateTime(tv.RequestedDate, DateTimeOffset).Ticks, Released = DateTime.Now > tv.ReleaseDate, Approved = tv.Available || tv.Approved, Title = tv.Title, Overview = tv.Overview, RequestedUsers = IsAdmin ? tv.AllUsers.ToArray() : new string[] { }, ReleaseYear = tv.ReleaseDate.Year.ToString(), Available = tv.Available, Admin = IsAdmin, Issues = tv.Issues.ToString().CamelCaseToWords(), OtherMessage = tv.OtherMessage, AdminNotes = tv.AdminNote, TvSeriesRequestType = tv.SeasonsRequested, Qualities = qualities.ToArray() }; }).ToList(); return Response.AsJson(viewModel); } private Response GetAlbumRequests() { var settings = PrSettings.GetSettings(); var dbAlbum = Service.GetAll().Where(x => x.Type == RequestType.Album); if (settings.UsersCanViewOnlyOwnRequests && !IsAdmin) { dbAlbum = dbAlbum.Where(x => x.UserHasRequested(Username)); } var viewModel = dbAlbum.Select(album => { return new RequestViewModel { ProviderId = album.ProviderId, Type = album.Type, Status = album.Status, ImdbId = album.ImdbId, Id = album.Id, PosterPath = album.PosterPath, ReleaseDate = album.ReleaseDate, ReleaseDateTicks = album.ReleaseDate.Ticks, RequestedDate = album.RequestedDate, RequestedDateTicks = DateTimeHelper.OffsetUTCDateTime(album.RequestedDate, DateTimeOffset).Ticks, Released = DateTime.Now > album.ReleaseDate, Approved = album.Available || album.Approved, Title = album.Title, Overview = album.Overview, RequestedUsers = IsAdmin ? album.AllUsers.ToArray() : new string[] { }, ReleaseYear = album.ReleaseDate.Year.ToString(), Available = album.Available, Admin = IsAdmin, Issues = album.Issues.ToString().CamelCaseToWords(), OtherMessage = album.OtherMessage, AdminNotes = album.AdminNote, TvSeriesRequestType = album.SeasonsRequested, MusicBrainzId = album.MusicBrainzId, ArtistName = album.ArtistName }; }).ToList(); return Response.AsJson(viewModel); } private Response DeleteRequest(int requestid) { this.RequiresClaims (UserClaims.Admin); var currentEntity = Service.Get(requestid); Service.DeleteRequest(currentEntity); return Response.AsJson(new JsonResponseModel { Result = true }); } /// /// Reports the issue. /// Comment can be null if the IssueState != Other /// /// The request identifier. /// The issue. /// The comment. /// private Response ReportIssue(int requestId, IssueState issue, string comment) { var originalRequest = Service.Get(requestId); if (originalRequest == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not add issue, please try again or contact the administrator!" }); } originalRequest.Issues = issue; originalRequest.OtherMessage = !string.IsNullOrEmpty(comment) ? $"{Username} - {comment}" : string.Empty; var result = Service.UpdateRequest(originalRequest); var model = new NotificationModel { User = Username, NotificationType = NotificationType.Issue, Title = originalRequest.Title, DateTime = DateTime.Now, Body = issue == IssueState.Other ? comment : issue.ToString().CamelCaseToWords() }; NotificationService.Publish(model); return Response.AsJson(result ? new JsonResponseModel { Result = true } : new JsonResponseModel { Result = false, Message = "Could not add issue, please try again or contact the administrator!" }); } private Response ClearIssue(int requestId) { this.RequiresClaims ( UserClaims.Admin); var originalRequest = Service.Get(requestId); if (originalRequest == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Request does not exist to clear it!" }); } originalRequest.Issues = IssueState.None; originalRequest.OtherMessage = string.Empty; var result = Service.UpdateRequest(originalRequest); return Response.AsJson(result ? new JsonResponseModel { Result = true } : new JsonResponseModel { Result = false, Message = "Could not clear issue, please try again or check the logs" }); } private Response ChangeRequestAvailability(int requestId, bool available) { this.RequiresClaims (UserClaims.Admin); var originalRequest = Service.Get(requestId); if (originalRequest == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Request does not exist to change the availability!" }); } originalRequest.Available = available; var result = Service.UpdateRequest(originalRequest); return Response.AsJson(result ? new { Result = true, Available = available, Message = string.Empty } : new { Result = false, Available = false, Message = "Could not update the availability, please try again or check the logs" }); } private Response AddNote(int requestId, string noteArea) { this.RequiresClaims (UserClaims.Admin); var originalRequest = Service.Get(requestId); if (originalRequest == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Request does not exist to add a note!" }); } originalRequest.AdminNote = noteArea; var result = Service.UpdateRequest(originalRequest); return Response.AsJson(result ? new JsonResponseModel { Result = true } : new JsonResponseModel { Result = false, Message = "Could not update the notes, please try again or check the logs" }); } } }