pull/687/head
Jamie.Rees 8 years ago
parent 8a61371048
commit cb50fc61fb

@ -112,6 +112,7 @@
<Compile Include="PlexReadOnlyDatabase.cs" /> <Compile Include="PlexReadOnlyDatabase.cs" />
<Compile Include="Queue\ITransientFaultQueue.cs" /> <Compile Include="Queue\ITransientFaultQueue.cs" />
<Compile Include="Queue\TransientFaultQueue.cs" /> <Compile Include="Queue\TransientFaultQueue.cs" />
<Compile Include="RequestHelper.cs" />
<Compile Include="SettingModels\AuthenticationSettings.cs" /> <Compile Include="SettingModels\AuthenticationSettings.cs" />
<Compile Include="SettingModels\ExternalSettings.cs" /> <Compile Include="SettingModels\ExternalSettings.cs" />
<Compile Include="SettingModels\HeadphonesSettings.cs" /> <Compile Include="SettingModels\HeadphonesSettings.cs" />

@ -0,0 +1,82 @@
#region Copyright
// /************************************************************************
// Copyright (c) 2016 Jamie Rees
// File: RequestHelper.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 PlexRequests.Core.SettingModels;
using PlexRequests.Store;
namespace PlexRequests.Core
{
public static class RequestHelper
{
public static bool ShouldAutoApprove(this RequestType requestType, PlexRequestSettings prSettings, bool isAdmin, string username)
{
// if the user is an admin or they are whitelisted, they go ahead and allow auto-approval
if (isAdmin || prSettings.ApprovalWhiteList.Any(x => x.Equals(username, StringComparison.OrdinalIgnoreCase))) return true;
// check by request type if the category requires approval or not
switch (requestType)
{
case RequestType.Movie:
return !prSettings.RequireMovieApproval;
case RequestType.TvShow:
return !prSettings.RequireTvShowApproval;
case RequestType.Album:
return !prSettings.RequireMusicApproval;
default:
return false;
}
}
public static bool ShouldAutoApprove(this RequestType requestType, PlexRequestSettings prSettings, bool isAdmin, List<string> username)
{
// if the user is an admin or they are whitelisted, they go ahead and allow auto-approval
if (isAdmin) return true;
if(prSettings.ApprovalWhiteList.Intersect(username).Any())
{
return true;
}
// check by request type if the category requires approval or not
switch (requestType)
{
case RequestType.Movie:
return !prSettings.RequireMovieApproval;
case RequestType.TvShow:
return !prSettings.RequireTvShowApproval;
case RequestType.Album:
return !prSettings.RequireMusicApproval;
default:
return false;
}
}
}
}

@ -37,6 +37,7 @@ using PlexRequests.Api.Interfaces;
using PlexRequests.Core; using PlexRequests.Core;
using PlexRequests.Core.SettingModels; using PlexRequests.Core.SettingModels;
using PlexRequests.Helpers; using PlexRequests.Helpers;
using PlexRequests.Helpers.Permissions;
using PlexRequests.Services.Interfaces; using PlexRequests.Services.Interfaces;
using PlexRequests.Store; using PlexRequests.Store;
using PlexRequests.Store.Models; using PlexRequests.Store.Models;
@ -53,7 +54,7 @@ namespace PlexRequests.Services.Jobs
public FaultQueueHandler(IJobRecord record, IRepository<RequestQueue> repo, ISonarrApi sonarrApi, public FaultQueueHandler(IJobRecord record, IRepository<RequestQueue> repo, ISonarrApi sonarrApi,
ISickRageApi srApi, ISettingsService<SonarrSettings> sonarrSettings, ISettingsService<SickRageSettings> srSettings, ISickRageApi srApi, ISettingsService<SonarrSettings> sonarrSettings, ISettingsService<SickRageSettings> srSettings,
ICouchPotatoApi cpApi, ISettingsService<CouchPotatoSettings> cpsettings, IRequestService requestService, ICouchPotatoApi cpApi, ISettingsService<CouchPotatoSettings> cpsettings, IRequestService requestService,
ISettingsService<HeadphonesSettings> hpSettings, IHeadphonesApi headphonesApi) ISettingsService<HeadphonesSettings> hpSettings, IHeadphonesApi headphonesApi, ISettingsService<PlexRequestSettings> prSettings)
{ {
Record = record; Record = record;
Repo = repo; Repo = repo;
@ -68,6 +69,7 @@ namespace PlexRequests.Services.Jobs
SonarrSettings = sonarrSettings; SonarrSettings = sonarrSettings;
CpSettings = cpsettings; CpSettings = cpsettings;
HeadphoneSettings = hpSettings; HeadphoneSettings = hpSettings;
PrSettings = prSettings.GetSettings();
} }
private IRepository<RequestQueue> Repo { get; } private IRepository<RequestQueue> Repo { get; }
@ -77,6 +79,7 @@ namespace PlexRequests.Services.Jobs
private ICouchPotatoApi CpApi { get; } private ICouchPotatoApi CpApi { get; }
private IHeadphonesApi HpApi { get; } private IHeadphonesApi HpApi { get; }
private IRequestService RequestService { get; } private IRequestService RequestService { get; }
private PlexRequestSettings PrSettings { get; }
private ISettingsService<SonarrSettings> SonarrSettings { get; } private ISettingsService<SonarrSettings> SonarrSettings { get; }
private ISettingsService<SickRageSettings> SickrageSettings { get; } private ISettingsService<SickRageSettings> SickrageSettings { get; }
private ISettingsService<CouchPotatoSettings> CpSettings { get; } private ISettingsService<CouchPotatoSettings> CpSettings { get; }
@ -108,13 +111,14 @@ namespace PlexRequests.Services.Jobs
private void ProcessMissingInformation(List<RequestQueue> requests) private void ProcessMissingInformation(List<RequestQueue> requests)
{ {
var sonarrSettings = SonarrSettings.GetSettings();
var sickrageSettings = SickrageSettings.GetSettings();
if (!requests.Any()) if (!requests.Any())
{ {
return; return;
} }
var sonarrSettings = SonarrSettings.GetSettings();
var sickrageSettings = SickrageSettings.GetSettings();
var tv = requests.Where(x => x.Type == RequestType.TvShow); var tv = requests.Where(x => x.Type == RequestType.TvShow);
// TV // TV
@ -122,7 +126,7 @@ namespace PlexRequests.Services.Jobs
foreach (var t in tv) foreach (var t in tv)
{ {
var providerId = int.Parse(t.PrimaryIdentifier); var providerId = int.Parse(t.PrimaryIdentifier);
var showInfo = tvApi.ShowLookupByTheTvDbId(providerId); var showInfo = tvApi.ShowLookup(providerId);
if (showInfo.externals?.thetvdb != null) if (showInfo.externals?.thetvdb != null)
{ {
@ -227,8 +231,10 @@ namespace PlexRequests.Services.Jobs
if (result) if (result)
{ {
// Approve it now
model.Approved = true; if (model.Type.ShouldAutoApprove(PrSettings, false, model.RequestedUsers))
// Approve it now
model.Approved = true;
RequestService.UpdateRequest(model); RequestService.UpdateRequest(model);
}; };
@ -258,7 +264,7 @@ namespace PlexRequests.Services.Jobs
foreach (var request in requests) foreach (var request in requests)
{ {
var model = ByteConverterHelper.ReturnObject<RequestedModel>(request.Content); var model = ByteConverterHelper.ReturnObject<RequestedModel>(request.Content);
var result = false; bool result;
switch (request.Type) switch (request.Type)
{ {
case RequestType.Movie: case RequestType.Movie:

@ -226,7 +226,8 @@ namespace PlexRequests.UI.Jobs
var fault = var fault =
TriggerBuilder.Create() TriggerBuilder.Create()
.WithIdentity("FaultQueueHandler", "Fault") .WithIdentity("FaultQueueHandler", "Fault")
.StartAt(DateBuilder.FutureDate(10, IntervalUnit.Minute)) //.StartAt(DateBuilder.FutureDate(10, IntervalUnit.Minute))
.StartNow()
.WithSimpleSchedule(x => x.WithIntervalInHours(s.FaultQueueHandler).RepeatForever()) .WithSimpleSchedule(x => x.WithIntervalInHours(s.FaultQueueHandler).RepeatForever())
.Build(); .Build();

@ -878,7 +878,7 @@ namespace PlexRequests.UI.Modules
try try
{ {
if (ShouldAutoApprove(RequestType.TvShow, settings)) if (RequestType.TvShow.ShouldAutoApprove(settings, IsAdmin, Username))
{ {
model.Approved = true; model.Approved = true;
var s = await sonarrSettings; var s = await sonarrSettings;
@ -974,7 +974,7 @@ namespace PlexRequests.UI.Modules
private bool ShouldSendNotification(RequestType type, PlexRequestSettings prSettings) private bool ShouldSendNotification(RequestType type, PlexRequestSettings prSettings)
{ {
var sendNotification = ShouldAutoApprove(type, prSettings) var sendNotification = type.ShouldAutoApprove(prSettings, IsAdmin, Username)
? !prSettings.IgnoreNotifyForAutoApprovedRequests ? !prSettings.IgnoreNotifyForAutoApprovedRequests
: true; : true;
var claims = Context.CurrentUser?.Claims; var claims = Context.CurrentUser?.Claims;
@ -1073,7 +1073,7 @@ namespace PlexRequests.UI.Modules
try try
{ {
if (ShouldAutoApprove(RequestType.Album, settings)) if (RequestType.Album.ShouldAutoApprove(settings, IsAdmin, Username))
{ {
model.Approved = true; model.Approved = true;
var hpSettings = HeadphonesService.GetSettings(); var hpSettings = HeadphonesService.GetSettings();
@ -1127,25 +1127,6 @@ namespace PlexRequests.UI.Modules
return img; return img;
} }
private bool ShouldAutoApprove(RequestType requestType, PlexRequestSettings prSettings)
{
// if the user is an admin or they are whitelisted, they go ahead and allow auto-approval
if (IsAdmin || prSettings.ApprovalWhiteList.Any(x => x.Equals(Username, StringComparison.OrdinalIgnoreCase))) return true;
// check by request type if the category requires approval or not
switch (requestType)
{
case RequestType.Movie:
return !prSettings.RequireMovieApproval;
case RequestType.TvShow:
return !prSettings.RequireTvShowApproval;
case RequestType.Album:
return !prSettings.RequireMusicApproval;
default:
return false;
}
}
private async Task<Response> NotifyUser(bool notify) private async Task<Response> NotifyUser(bool notify)
{ {
Analytics.TrackEventAsync(Category.Search, Action.Save, "NotifyUser", Username, CookieHelper.GetAnalyticClientId(Cookies), notify ? 1 : 0); Analytics.TrackEventAsync(Category.Search, Action.Save, "NotifyUser", Username, CookieHelper.GetAnalyticClientId(Cookies), notify ? 1 : 0);

Loading…
Cancel
Save