added cooldown on refresh

MilchRatchet · Nov 14, 2019 · eddc08c · eddc08c
1 parent 335889f
commit eddc08c
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 6 deletions.
diff --git a/SubBox/Controllers/ValuesController.cs b/SubBox/Controllers/ValuesController.cs
@@ -77,8 +77,6 @@ public string[] GetSettings()
         [HttpPost("refresh")]
         public void Refresh()
         {
-            Console.WriteLine("Refreshing...");
-
             DataRetriever fetcher = new DataRetriever();
 
             fetcher.UpdateVideoList();

diff --git a/SubBox/Models/AppSettings.cs b/SubBox/Models/AppSettings.cs
@@ -1,4 +1,5 @@
-using System.IO;
+using System;
+using System.IO;
 
 namespace SubBox.Models
 {
@@ -10,12 +11,15 @@ public class AppSettings
         public static int PlaylistPlaybackSize { get; set; }   
         public static string Color { get; set; }
         public static bool NightMode { get; set; }
+        public static DateTime LastRefresh { get; set; }
+
+        //Runtime vars
         public static bool GCMode { get; set; }
 
-        public static void Save()
+        public static async void Save()
         {
-            string[] options = new string[] {RetrievalTimeFrame.ToString(), NewChannelTimeFrame.ToString(), DeletionTimeFrame.ToString(), PlaylistPlaybackSize.ToString(), Color, NightMode.ToString() };
-            File.WriteAllLines("settings.txt", options);
+            string[] options = new string[] {RetrievalTimeFrame.ToString(), NewChannelTimeFrame.ToString(), DeletionTimeFrame.ToString(), PlaylistPlaybackSize.ToString(), Color, NightMode.ToString(), LastRefresh.ToString("O") };
+            await File.WriteAllLinesAsync("settings.txt", options);
         }
     }
 }
diff --git a/SubBox/Models/DataRetriever.cs b/SubBox/Models/DataRetriever.cs
@@ -17,6 +17,8 @@ public class DataRetriever
 
         private static int LifeTime = 7;
 
+        private readonly int RefreshLimit = 120;
+
         public DataRetriever()
         {
             service = new YouTubeService(new BaseClientService.Initializer()
@@ -200,8 +202,23 @@ private List<string> CreateRequestList(List<string> videoIds)
 
         public void UpdateVideoList()
         {
+            int diffSec = (int) (DateTime.Now - AppSettings.LastRefresh).TotalSeconds;
+
+            if (diffSec < RefreshLimit)
+            {
+                Console.WriteLine("Can only refresh every " + RefreshLimit + " seconds! " + (RefreshLimit - diffSec) + " seconds remaining");
+
+                return;
+            }
+
+            AppSettings.LastRefresh = DateTime.Now;
+
+            AppSettings.Save();
+
             GarbageCollector();
 
+            Console.WriteLine("Refreshing...");
+
             LifeTime = AppSettings.RetrievalTimeFrame;
 
             List<Channel> Channels;

diff --git a/SubBox/Startup.cs b/SubBox/Startup.cs
@@ -84,6 +84,7 @@ public void Configure(IApplicationBuilder app, IHostingEnvironment env)
                 {
                     AppSettings.NightMode = false;
                 }
+                AppSettings.LastRefresh = DateTime.ParseExact(options[6], "O", CultureInfo.InvariantCulture);
             }
             catch (Exception)
             {
@@ -93,6 +94,7 @@ public void Configure(IApplicationBuilder app, IHostingEnvironment env)
                 AppSettings.PlaylistPlaybackSize = 50;
                 AppSettings.Color = "DB4437";
                 AppSettings.NightMode = false;
+                AppSettings.LastRefresh = BuildTime;
                 AppSettings.Save();
             }