mirror of https://github.com/Ombi-app/Ombi
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
469 lines
21 KiB
469 lines
21 KiB
#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.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Nancy;
|
|
using Nancy.Responses.Negotiation;
|
|
using NLog;
|
|
using Ombi.Api.Interfaces;
|
|
using Ombi.Core;
|
|
using Ombi.Core.Models;
|
|
using Ombi.Core.SettingModels;
|
|
using Ombi.Helpers;
|
|
using Ombi.Helpers.Analytics;
|
|
using Ombi.Helpers.Permissions;
|
|
using Ombi.Services.Interfaces;
|
|
using Ombi.Services.Notification;
|
|
using Ombi.Store;
|
|
using Ombi.UI.Models;
|
|
using Ombi.UI.Models.Admin;
|
|
using Ombi.UI.Models.Requests;
|
|
using Action = Ombi.Helpers.Analytics.Action;
|
|
using ISecurityExtensions = Ombi.Core.ISecurityExtensions;
|
|
|
|
namespace Ombi.UI.Modules
|
|
{
|
|
public class RequestsModule : BaseAuthModule
|
|
{
|
|
public RequestsModule(
|
|
IRequestService service,
|
|
ISettingsService<PlexRequestSettings> prSettings,
|
|
ISettingsService<PlexSettings> plex,
|
|
INotificationService notify,
|
|
ISettingsService<SonarrSettings> sonarrSettings,
|
|
ISettingsService<SickRageSettings> sickRageSettings,
|
|
ISettingsService<CouchPotatoSettings> cpSettings,
|
|
ICouchPotatoApi cpApi,
|
|
ISonarrApi sonarrApi,
|
|
ISickRageApi sickRageApi,
|
|
ICacheProvider cache,
|
|
IAnalytics an,
|
|
INotificationEngine engine,
|
|
ISecurityExtensions security,
|
|
ISettingsService<CustomizationSettings> customSettings) : base("requests", prSettings, security)
|
|
{
|
|
Service = service;
|
|
PrSettings = prSettings;
|
|
PlexSettings = plex;
|
|
NotificationService = notify;
|
|
SonarrSettings = sonarrSettings;
|
|
SickRageSettings = sickRageSettings;
|
|
CpSettings = cpSettings;
|
|
SonarrApi = sonarrApi;
|
|
SickRageApi = sickRageApi;
|
|
CpApi = cpApi;
|
|
Cache = cache;
|
|
Analytics = an;
|
|
NotificationEngine = engine;
|
|
CustomizationSettings = customSettings;
|
|
|
|
Get["/", true] = async (x, ct) => await LoadRequests();
|
|
Get["/movies", true] = async (x, ct) => await GetMovies();
|
|
Get["/tvshows", true] = async (c, ct) => await GetTvShows();
|
|
Get["/albums", true] = async (x, ct) => await GetAlbumRequests();
|
|
Post["/delete", true] = async (x, ct) => await DeleteRequest((int)Request.Form.id);
|
|
Post["/reportissue", true] = async (x, ct) => await ReportIssue((int)Request.Form.requestId, (IssueState)(int)Request.Form.issue, null);
|
|
Post["/reportissuecomment", true] = async (x, ct) => await ReportIssue((int)Request.Form.requestId, IssueState.Other, (string)Request.Form.commentArea);
|
|
|
|
Post["/clearissues", true] = async (x, ct) => await ClearIssue((int)Request.Form.Id);
|
|
|
|
Post["/changeavailability", true] = async (x, ct) => await ChangeRequestAvailability((int)Request.Form.Id, (bool)Request.Form.Available);
|
|
|
|
Get["/UpdateFilters", true] = async (x, ct) => await GetFilterAndSortSettings();
|
|
}
|
|
|
|
private static Logger Log = LogManager.GetCurrentClassLogger();
|
|
private IRequestService Service { get; }
|
|
private IAnalytics Analytics { get; }
|
|
private INotificationService NotificationService { get; }
|
|
private ISettingsService<PlexRequestSettings> PrSettings { get; }
|
|
private ISettingsService<PlexSettings> PlexSettings { get; }
|
|
private ISettingsService<SonarrSettings> SonarrSettings { get; }
|
|
private ISettingsService<SickRageSettings> SickRageSettings { get; }
|
|
private ISettingsService<CouchPotatoSettings> CpSettings { get; }
|
|
private ISettingsService<CustomizationSettings> CustomizationSettings { get; }
|
|
private ISonarrApi SonarrApi { get; }
|
|
private ISickRageApi SickRageApi { get; }
|
|
private ICouchPotatoApi CpApi { get; }
|
|
private ICacheProvider Cache { get; }
|
|
private INotificationEngine NotificationEngine { get; }
|
|
|
|
private async Task<Negotiator> LoadRequests()
|
|
{
|
|
var settings = await PrSettings.GetSettingsAsync();
|
|
var custom = await CustomizationSettings.GetSettingsAsync();
|
|
|
|
return View["Index", new RequestsIndexViewModel { CustomizationSettings = custom, PlexRequestSettings = settings }];
|
|
}
|
|
|
|
private async Task<Response> GetMovies()
|
|
{
|
|
var allRequests = await Service.GetAllAsync();
|
|
allRequests = allRequests.Where(x => x.Type == RequestType.Movie);
|
|
|
|
var dbMovies = allRequests.ToList();
|
|
|
|
if (Security.HasPermissions(User, Permissions.UsersCanViewOnlyOwnRequests) && !IsAdmin)
|
|
{
|
|
dbMovies = dbMovies.Where(x => x.UserHasRequested(Username)).ToList();
|
|
}
|
|
|
|
List<QualityModel> qualities = new List<QualityModel>();
|
|
|
|
if (IsAdmin)
|
|
{
|
|
var cpSettings = CpSettings.GetSettings();
|
|
if (cpSettings.Enabled)
|
|
{
|
|
try
|
|
{
|
|
var result = await Cache.GetOrSetAsync(CacheKeys.CouchPotatoQualityProfiles, async () =>
|
|
{
|
|
return await Task.Run(() => CpApi.GetProfiles(cpSettings.FullUri, cpSettings.ApiKey)).ConfigureAwait(false);
|
|
});
|
|
if (result != null)
|
|
{
|
|
qualities = result.list.Select(x => new QualityModel { Id = x._id, Name = x.label }).ToList();
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Info(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
IEnumerable<RootFolderModel> rootFolders = new List<RootFolderModel>();
|
|
if (IsAdmin)
|
|
{
|
|
try
|
|
{
|
|
var sonarrSettings = await SonarrSettings.GetSettingsAsync();
|
|
if (sonarrSettings.Enabled)
|
|
{
|
|
var result = Cache.GetOrSetAsync(CacheKeys.SonarrRootFolders, async () =>
|
|
{
|
|
return await Task.Run(() => SonarrApi.GetRootFolders(sonarrSettings.ApiKey, sonarrSettings.FullUri));
|
|
});
|
|
rootFolders = result.Result.Select(x => new RootFolderModel { Id = x.id.ToString(), Path = x.path }).ToList();
|
|
}
|
|
// @TODO Sick Rage Root Folders
|
|
//else
|
|
//{
|
|
// var sickRageSettings = await SickRageSettings.GetSettingsAsync();
|
|
// if (sickRageSettings.Enabled)
|
|
// {
|
|
// qualities = sickRageSettings.Qualities.Select(x => new QualityModel { Id = x.Key, Name = x.Value }).ToList();
|
|
// }
|
|
//}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Info(e);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
var canManageRequest = Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests);
|
|
var viewModel = dbMovies.Select(movie => 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 = canManageRequest ? movie.AllUsers.ToArray() : new string[] { },
|
|
ReleaseYear = movie.ReleaseDate.Year.ToString(),
|
|
Available = movie.Available,
|
|
Admin = canManageRequest,
|
|
IssueId = movie.IssueId,
|
|
Denied = movie.Denied,
|
|
DeniedReason = movie.DeniedReason,
|
|
Qualities = qualities.ToArray(),
|
|
RootFolders = rootFolders.ToArray(),
|
|
}).ToList();
|
|
|
|
return Response.AsJson(viewModel);
|
|
}
|
|
|
|
private async Task<Response> GetTvShows()
|
|
{
|
|
var settingsTask = PrSettings.GetSettingsAsync();
|
|
|
|
var requests = await Service.GetAllAsync();
|
|
requests = requests.Where(x => x.Type == RequestType.TvShow);
|
|
|
|
var dbTv = requests;
|
|
var settings = await settingsTask;
|
|
if (Security.HasPermissions(User, Permissions.UsersCanViewOnlyOwnRequests) && !IsAdmin)
|
|
{
|
|
dbTv = dbTv.Where(x => x.UserHasRequested(Username)).ToList();
|
|
}
|
|
|
|
IEnumerable<QualityModel> qualities = new List<QualityModel>();
|
|
if (IsAdmin)
|
|
{
|
|
try
|
|
{
|
|
var sonarrSettings = await SonarrSettings.GetSettingsAsync();
|
|
if (sonarrSettings.Enabled)
|
|
{
|
|
var result = Cache.GetOrSetAsync(CacheKeys.SonarrQualityProfiles, async () =>
|
|
{
|
|
return await Task.Run(() => SonarrApi.GetProfiles(sonarrSettings.ApiKey, sonarrSettings.FullUri));
|
|
});
|
|
qualities = result.Result.Select(x => new QualityModel { Id = x.id.ToString(), Name = x.name }).ToList();
|
|
}
|
|
else
|
|
{
|
|
var sickRageSettings = await SickRageSettings.GetSettingsAsync();
|
|
if (sickRageSettings.Enabled)
|
|
{
|
|
qualities = sickRageSettings.Qualities.Select(x => new QualityModel { Id = x.Key, Name = x.Value }).ToList();
|
|
}
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Log.Info(e);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var canManageRequest = Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests);
|
|
var viewModel = dbTv.Select(tv => new RequestViewModel
|
|
{
|
|
ProviderId = tv.ProviderId,
|
|
Type = tv.Type,
|
|
Status = tv.Status,
|
|
ImdbId = tv.ImdbId,
|
|
Id = tv.Id,
|
|
PosterPath = tv.PosterPath.Contains("http:") ? tv.PosterPath.Replace("http:", "https:") : tv.PosterPath, // We make the poster path https on request, but this is just incase
|
|
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 = canManageRequest ? tv.AllUsers.ToArray() : new string[] { },
|
|
ReleaseYear = tv.ReleaseDate.Year.ToString(),
|
|
Available = tv.Available,
|
|
Admin = canManageRequest,
|
|
IssueId = tv.IssueId,
|
|
Denied = tv.Denied,
|
|
DeniedReason = tv.DeniedReason,
|
|
TvSeriesRequestType = tv.SeasonsRequested,
|
|
Qualities = qualities.ToArray(),
|
|
Episodes = tv.Episodes.ToArray(),
|
|
}).ToList();
|
|
|
|
return Response.AsJson(viewModel);
|
|
}
|
|
|
|
private async Task<Response> GetAlbumRequests()
|
|
{
|
|
var settings = PrSettings.GetSettings();
|
|
var dbAlbum = await Service.GetAllAsync();
|
|
dbAlbum = dbAlbum.Where(x => x.Type == RequestType.Album);
|
|
if (Security.HasPermissions(User, Permissions.UsersCanViewOnlyOwnRequests) && !IsAdmin)
|
|
{
|
|
dbAlbum = dbAlbum.Where(x => x.UserHasRequested(Username));
|
|
}
|
|
var canManageRequest = Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests);
|
|
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 = canManageRequest ? album.AllUsers.ToArray() : new string[] { },
|
|
ReleaseYear = album.ReleaseDate.Year.ToString(),
|
|
Available = album.Available,
|
|
Admin = canManageRequest,
|
|
IssueId = album.IssueId,
|
|
Denied = album.Denied,
|
|
DeniedReason = album.DeniedReason,
|
|
TvSeriesRequestType = album.SeasonsRequested,
|
|
MusicBrainzId = album.MusicBrainzId,
|
|
ArtistName = album.ArtistName
|
|
|
|
};
|
|
}).ToList();
|
|
|
|
return Response.AsJson(viewModel);
|
|
}
|
|
|
|
private async Task<Response> DeleteRequest(int requestid)
|
|
{
|
|
if (!Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests))
|
|
{
|
|
return Response.AsJson(new JsonResponseModel { Result = true });
|
|
}
|
|
|
|
|
|
Analytics.TrackEventAsync(Category.Requests, Action.Delete, "Delete Request", Username, CookieHelper.GetAnalyticClientId(Cookies));
|
|
|
|
var currentEntity = await Service.GetAsync(requestid);
|
|
await Service.DeleteRequestAsync(currentEntity);
|
|
return Response.AsJson(new JsonResponseModel { Result = true });
|
|
}
|
|
|
|
/// <summary>
|
|
/// Reports the issue.
|
|
/// Comment can be null if the <c>IssueState != Other</c>
|
|
/// </summary>
|
|
/// <param name="requestId">The request identifier.</param>
|
|
/// <param name="issue">The issue.</param>
|
|
/// <param name="comment">The comment.</param>
|
|
/// <returns></returns>
|
|
private async Task<Response> ReportIssue(int requestId, IssueState issue, string comment)
|
|
{
|
|
if (!Security.HasPermissions(User, Permissions.ReportIssue))
|
|
{
|
|
return Response.AsJson(new JsonResponseModel { Result = false, Message = "Sorry, you do not have the correct permissions to report an issue." });
|
|
}
|
|
var originalRequest = await Service.GetAsync(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 = await Service.UpdateRequestAsync(originalRequest);
|
|
|
|
var model = new NotificationModel
|
|
{
|
|
User = Username,
|
|
NotificationType = NotificationType.Issue,
|
|
Title = originalRequest.Title,
|
|
DateTime = DateTime.Now,
|
|
Body = issue == IssueState.Other ? comment : issue.ToString().ToCamelCaseWords(),
|
|
ImgSrc = originalRequest.Type == RequestType.Movie ? $"https://image.tmdb.org/t/p/w300/{originalRequest.PosterPath}" : originalRequest.PosterPath
|
|
};
|
|
await 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 async Task<Response> ClearIssue(int requestId)
|
|
{
|
|
if (!Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests))
|
|
{
|
|
return Response.AsJson(new JsonResponseModel { Result = false, Message = "Sorry, you do not have the correct permissions to clear an issue." });
|
|
}
|
|
|
|
var originalRequest = await Service.GetAsync(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 = await Service.UpdateRequestAsync(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 async Task<Response> ChangeRequestAvailability(int requestId, bool available)
|
|
{
|
|
if (!Security.HasAnyPermissions(User, Permissions.Administrator, Permissions.ManageRequests))
|
|
{
|
|
return Response.AsJson(new JsonResponseModel { Result = false, Message = "Sorry, you do not have the correct permissions to change a request." });
|
|
}
|
|
|
|
Analytics.TrackEventAsync(Category.Requests, Action.Update, available ? "Make request available" : "Make request unavailable", Username, CookieHelper.GetAnalyticClientId(Cookies));
|
|
var originalRequest = await Service.GetAsync(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 = await Service.UpdateRequestAsync(originalRequest);
|
|
var plexService = await PlexSettings.GetSettingsAsync();
|
|
await NotificationEngine.NotifyUsers(originalRequest, plexService.PlexAuthToken, available ? NotificationType.RequestAvailable : NotificationType.RequestDeclined);
|
|
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 async Task<Response> GetFilterAndSortSettings()
|
|
{
|
|
var s = await CustomizationSettings.GetSettingsAsync();
|
|
|
|
var sortVal = EnumHelper<SortOptions>.GetDisplayValue((SortOptions)s.DefaultSort);
|
|
var filterVal = EnumHelper<FilterOptions>.GetDisplayValue((FilterOptions)s.DefaultFilter);
|
|
|
|
var vm = new
|
|
{
|
|
DefaultSort = sortVal,
|
|
DefaultFilter = filterVal
|
|
};
|
|
|
|
return Response.AsJson(vm);
|
|
}
|
|
}
|
|
}
|