From 938a247e9ea1ec6143882462e96f51494d6358ba Mon Sep 17 00:00:00 2001
From: macmade <macmade@xs-labs.com>
Date: Sun, 18 Aug 2024 20:02:54 +0200
Subject: [PATCH] feat: Refresh timer

---
 SDO/Classes/MainWindowController.swift | 36 +++++++++++++++++++++-----
 1 file changed, 29 insertions(+), 7 deletions(-)

diff --git a/SDO/Classes/MainWindowController.swift b/SDO/Classes/MainWindowController.swift
index 39427cb..c4d9cad 100644
--- a/SDO/Classes/MainWindowController.swift
+++ b/SDO/Classes/MainWindowController.swift
@@ -48,8 +48,11 @@ public class MainWindowController: NSWindowController, NSCollectionViewDataSourc
     @IBOutlet private var collectionView: NSCollectionView?
     @IBOutlet private var slider:         Slider?
 
-    private var imageSizeObserver: NSKeyValueObservation?
-    private var imagesObserver:    NSKeyValueObservation?
+    private var imageSizeObserver:        NSKeyValueObservation?
+    private var imagesObserver:           NSKeyValueObservation?
+    private var refreshIntervalObserver:  NSKeyValueObservation?
+    private var automaticRefreshObserver: NSKeyValueObservation?
+    private var refreshTimer:             Timer?
 
     public init()
     {
@@ -69,6 +72,7 @@ public class MainWindowController: NSWindowController, NSCollectionViewDataSourc
     public override func windowDidLoad()
     {
         super.windowDidLoad()
+        self.updateRefreshTimer()
 
         self.imageSizeObserver = Preferences.shared.observe( \.imageSize )
         {
@@ -96,6 +100,16 @@ public class MainWindowController: NSWindowController, NSCollectionViewDataSourc
             self.collectionView?.reloadData()
         }
 
+        self.refreshIntervalObserver = Preferences.shared.observe( \.refreshInterval )
+        {
+            [ weak self ] _, _ in self?.updateRefreshTimer()
+        }
+
+        self.automaticRefreshObserver = Preferences.shared.observe( \.automaticRefresh )
+        {
+            [ weak self ] _, _ in self?.updateRefreshTimer()
+        }
+
         self.collectionView?.register( NSNib( nibNamed: "ImageItem", bundle: nil ), forItemWithIdentifier: self.itemID )
         self.refresh( nil )
         self.updateLayout()
@@ -109,6 +123,19 @@ public class MainWindowController: NSWindowController, NSCollectionViewDataSourc
         }
     }
 
+    private func updateRefreshTimer()
+    {
+        self.refreshTimer?.invalidate()
+
+        if Preferences.shared.automaticRefresh
+        {
+            self.refreshTimer = Timer.scheduledTimer( withTimeInterval: TimeInterval( Preferences.shared.refreshInterval ), repeats: true )
+            {
+                [ weak self ] _ in self?.refresh( nil )
+            }
+        }
+    }
+
     @IBAction
     public func resetImageSize( _ sender: Any? )
     {
@@ -158,11 +185,6 @@ public class MainWindowController: NSWindowController, NSCollectionViewDataSourc
                 self.lastRefresh               = "Last Refreshed: \( fmt.string( from: Date() ) )"
 
                 self.collectionView?.reloadData()
-
-                DispatchQueue.main.asyncAfter( deadline: .now() + .seconds( 60 * 5 ) )
-                {
-                    self.refresh( nil )
-                }
             }
         }
     }