Add updateSite method to Sites

DefinedNet · Feb 20, 2025 · e80a10e · e80a10e
1 parent 0166bd3
commit e80a10e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/ios/Runner/AppDelegate.swift b/ios/Runner/AppDelegate.swift
@@ -34,14 +34,8 @@ func MissingArgumentError(message: String, details: Any?) -> FlutterError {
         container!.updater.update(connected: site.connected ?? false, replaceSite: site)
       }
         Task {
-            await dnUpdater.updateAllLoop { @MainActor site in
-                // Signal the site has changed in case the current site details screen is active
-                let container = self.sites?.getContainer(id: site.id)
-                if container != nil {
-                    // Update references to the site with the new site config
-                    container!.site = site
-                    container!.updater.update(connected: site.connected ?? false, replaceSite: site)
-                }
+            await dnUpdater.updateAllLoop { site in
+                self.sites?.updateSite(site: site)
 
                 // Signal to the main screen to reload
                 self.ui?.invokeMethod("refreshSites", arguments: nil)

diff --git a/ios/Runner/Sites.swift b/ios/Runner/Sites.swift
@@ -76,6 +76,16 @@ class Sites {
   func getContainer(id: String) -> SiteContainer? {
     return containers[id]
   }
+
+  func updateSite(site: Site) {
+    // Signal the site has changed in case the current site details screen is active
+    let container = getContainer(id: site.id)
+    if container != nil {
+      // Update references to the site with the new site config
+      container!.site = site
+      container!.updater.update(connected: site.connected ?? false, replaceSite: site)
+    }
+  }
 }
 
 class SiteUpdater: NSObject, FlutterStreamHandler, @unchecked Sendable {