1 files changed, 14 insertions, 0 deletions
diff --git a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/units/UnitsViewModel.kt b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/units/UnitsViewModel.kt
index b5ed78d..05962e9 100644
--- a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/units/UnitsViewModel.kt
+++ b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/units/UnitsViewModel.kt
@@ -27,6 +27,7 @@ import mx.trackermap.TrackerMap.client.models.Geofence
 import mx.trackermap.TrackerMap.client.models.MapLayer
 import mx.trackermap.TrackerMap.client.models.UnitInformation
 import mx.trackermap.TrackerMap.controllers.GeofencesController
+import mx.trackermap.TrackerMap.controllers.NetworkController
 import mx.trackermap.TrackerMap.controllers.UnitsController
 import org.koin.core.component.KoinComponent
 import org.koin.core.component.inject
@@ -48,9 +49,11 @@ class UnitsViewModel(
         val animated: Boolean,
     )
 
+    private val networkController: NetworkController by inject()
     private val unitsController: UnitsController by inject()
     private val geofenceController: GeofencesController by inject()
 
+    private var _networkAvailable = MutableLiveData<Boolean?>()
     private var _searchQuery = savedStateHandle.getLiveData("searchQuery", "")
     private var _unitsDisplayMode = MutableLiveData(UnitsDisplayMode.MAP)
     private var _units = MutableLiveData<List<UnitInformation>>()
@@ -61,6 +64,7 @@ class UnitsViewModel(
     private var _geofences = MutableLiveData<Map<Int, Geofence>>()
     private val _camera = MutableLiveData<Camera?>()
 
+    val networkAvailable: LiveData<Boolean?> get() = _networkAvailable
     val searchQuery: LiveData<String> get() = _searchQuery
     val unitsDisplayMode: LiveData<UnitsDisplayMode> get() = _unitsDisplayMode
     val units: LiveData<List<UnitInformation>> get() = _units
@@ -75,6 +79,9 @@ class UnitsViewModel(
         Log.d("UnitsViewModel", "Initializing Units View Model")
         unitsController.fetchUnits(viewModelScope)
         viewModelScope.launch {
+            setupNetworkObserver()
+        }
+        viewModelScope.launch {
             setupUnitsObserver()
         }
         viewModelScope.launch {
@@ -86,6 +93,13 @@ class UnitsViewModel(
         }
     }
 
+    private suspend fun setupNetworkObserver() {
+        networkController.networkAvailable.collect { available ->
+            Log.d("UnitsViewModel", "Collecting network state")
+            _networkAvailable.value = available
+        }
+    }
+
     private suspend fun setupUnitsObserver() {
         (unitsController.displayedUnitsFlow as StateFlow<List<UnitInformation>>).collect { units ->
             Log.d("UnitsViewModel", "Collecting units")