1 files changed, 13 insertions, 1 deletions
diff --git a/iosApp/iosApp/Units/UnitsViewModel.swift b/iosApp/iosApp/Units/UnitsViewModel.swift
index 7fcc80e..52e0c39 100644
--- a/iosApp/iosApp/Units/UnitsViewModel.swift
+++ b/iosApp/iosApp/Units/UnitsViewModel.swift
@@ -21,6 +21,7 @@ import shared
 
 @MainActor
 class UnitsViewModel: ObservableObject {
+    @Inject var networkController: NetworkController
     @Inject var unitsController: UnitsController
     @Inject var geofenceController: GeofencesController
     
@@ -34,6 +35,7 @@ class UnitsViewModel: ObservableObject {
     var detailsUnit: UnitInformation? = nil
     var detailsAction = DeviceRow.Action.details
     
+    @Published var networkAvailable: Bool? = nil
     @Published var searchQuery = "" {
         didSet {
             unitsDisplayMode = .list
@@ -83,6 +85,9 @@ class UnitsViewModel: ObservableObject {
     }
     
     private func setupObservers() {
+        let networkCollector = Collector<Bool?>(callback: setNetworkState)
+        networkController.networkAvailable.collect(collector: networkCollector) { _ in }
+        
         let unitsCollector = Collector<[UnitInformation]>(callback: setUnits)
         unitsController.displayedUnitsFlow.collect(collector: unitsCollector) { _ in }
         
@@ -90,12 +95,19 @@ class UnitsViewModel: ObservableObject {
         geofenceController.geofencesFlow.collect(collector: geofencesCollector) { _ in }
     }
     
+    private func setNetworkState(state: Bool?) {
+        print("Network state is: \(state?.description ?? "")")
+        Task { @MainActor in
+            self.networkAvailable = state
+        }
+    }
+    
     private func setUnits(units: [UnitInformation]) {
         print("Positions")
         Task { @MainActor in
             self.units = units
+            updateSelectedUnit()
         }
-        updateSelectedUnit()
     }
     
     private func setGeofences(geofences: [Int: Geofence]) {