Finished #6

PlexRequests.Core/SettingModels/PlexRequestSettings.cs

@@ -32,6 +32,13 @@ namespace PlexRequests.Core.SettingModels
 {
     public class PlexRequestSettings : Settings
     {
+        public PlexRequestSettings()
+        {
+            TvWeeklyRequestLimit = 0;
+            MovieWeeklyRequestLimit = 0;
+            AlbumWeeklyRequestLimit = 0;
+        }
+
         public int Port { get; set; }
         public string BaseUrl { get; set; }
         public bool SearchForMovies { get; set; }
@@ -42,7 +49,9 @@ public class PlexRequestSettings : Settings
         public bool RequireMusicApproval { get; set; }
         public bool UsersCanViewOnlyOwnRequests { get; set; }
         public bool UsersCanViewOnlyOwnIssues { get; set; }
-        public int WeeklyRequestLimit { get; set; }
+        public int MovieWeeklyRequestLimit { get; set; }
+        public int TvWeeklyRequestLimit { get; set; }
+        public int AlbumWeeklyRequestLimit { get; set; }
         public string NoApprovalUsers { get; set; }
         public bool CollectAnalyticData { get; set; }
         public bool IgnoreNotifyForAutoApprovedRequests { get; set; }

PlexRequests.Core/SettingModels/ScheduledJobsSettings.cs

@@ -36,12 +36,14 @@ public ScheduledJobsSettings()
             CouchPotatoCacher = 10;
             StoreBackup = 24;
             StoreCleanup = 24;
+            UserRequestLimitResetter = 12;
         }
         public int PlexAvailabilityChecker { get; set; }
         public int SickRageCacher { get; set; }
         public int SonarrCacher { get; set; }
         public int CouchPotatoCacher { get; set; }
         public int StoreBackup { get; set; }
         public int StoreCleanup { get; set; }
+        public int UserRequestLimitResetter { get; set; }
     }
 }

PlexRequests.Core/Setup.cs

@@ -100,7 +100,6 @@ private void CreateDefaultSettingsPage(string baseUrl)
                 RequireMovieApproval = true,
                 SearchForMovies = true,
                 SearchForTvShows = true,
-                WeeklyRequestLimit = 0,
                 BaseUrl = baseUrl ?? string.Empty,
                 CollectAnalyticData = true,
             };

PlexRequests.Services/Jobs/JobNames.cs

@@ -34,5 +34,6 @@ public static class JobNames
         public const string SrCacher = "SickRage Cacher";
         public const string PlexChecker = "Plex Availability Cacher";
         public const string StoreCleanup = "Database Cleanup";
+        public const string RequestLimitReset = "Request Limit Reset";
     }
 }

PlexRequests.Services/Jobs/UserRequestLimitResetter.cs

@@ -0,0 +1,119 @@
+#region Copyright
+// /************************************************************************
+//    Copyright (c) 2016 Jamie Rees
+//    File: UserRequestLimitResetter.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 NLog;
+
+using PlexRequests.Core;
+using PlexRequests.Core.SettingModels;
+using PlexRequests.Services.Interfaces;
+using PlexRequests.Store;
+using PlexRequests.Store.Models;
+using PlexRequests.Store.Repository;
+
+using Quartz;
+
+namespace PlexRequests.Services.Jobs
+{
+    public class UserRequestLimitResetter : IJob
+    {
+        public UserRequestLimitResetter(IJobRecord record, IRepository<RequestLimit> repo, ISettingsService<PlexRequestSettings> settings)
+        {
+            Record = record;
+            Repo = repo;
+            Settings = settings;
+        }
+        private IJobRecord Record { get; }
+        private IRepository<RequestLimit> Repo { get; }
+        private ISettingsService<PlexRequestSettings> Settings { get; }
+
+        private static Logger Log = LogManager.GetCurrentClassLogger();
+
+        public void Execute(IJobExecutionContext context)
+        {
+            try
+            {
+                var settings = Settings.GetSettings();
+                var users = Repo.GetAll();
+                var requestLimits = users as RequestLimit[] ?? users.ToArray();
+
+                MovieLimit(settings, requestLimits);
+                TvLimit(settings, requestLimits);
+                AlbumLimit(settings, requestLimits);
+            }
+            catch (Exception e)
+            {
+                Log.Error(e);
+            }
+            finally
+            {
+                Record.Record(JobNames.RequestLimitReset);
+            }
+        }
+
+        public void MovieLimit(PlexRequestSettings s, IEnumerable<RequestLimit> allUsers)
+        {
+            if (s.MovieWeeklyRequestLimit == 0)
+            {
+                return; // The limit has not been set
+            }
+            CheckAndDelete(allUsers, RequestType.Movie);
+        }
+
+        public void TvLimit(PlexRequestSettings s, IEnumerable<RequestLimit> allUsers)
+        {
+            if (s.TvWeeklyRequestLimit == 0)
+            {
+                return; // The limit has not been set
+            }
+            CheckAndDelete(allUsers, RequestType.TvShow);
+        }
+
+        public void AlbumLimit(PlexRequestSettings s, IEnumerable<RequestLimit> allUsers)
+        {
+            if (s.AlbumWeeklyRequestLimit == 0)
+            {
+                return; // The limit has not been set
+            }
+            CheckAndDelete(allUsers, RequestType.Album);
+        }
+
+        private void CheckAndDelete(IEnumerable<RequestLimit> allUsers, RequestType type)
+        {
+            var users = allUsers.Where(x => x.RequestType == type);
+            foreach (var u in users)
+            {
+                if (u.FirstRequestDate > DateTime.UtcNow.AddDays(-7))
+                {
+                    Repo.Delete(u);
+                }
+            }
+        }
+    }
+}