Finishes the markers functionality

1 files changed, 13 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 d8d7ea1..ca05736 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
@@ -51,6 +51,18 @@ class UnitsViewModel(
         setDisplayMode(UnitsDisplayMode.MAP)
     }
 
+    fun selectUnitWith(positionId: Int?) {
+        if (positionId == null) {
+            Log.d("UnitsViewModel", "Deselecting unit")
+            _selectedUnit.postValue(null)
+            return
+        }
+
+        Log.d("UnitsViewModel", "Selecting unit with position id: $positionId")
+        val unit = _units.value?.find { it.position?.id == positionId }
+        _selectedUnit.postValue(unit)
+    }
+
     fun setDisplayMode(displayMode: UnitsDisplayMode) {
         Log.d("UnitsViewModel", "Setting Display mode to $displayMode")
         _unitsDisplayMode.postValue(displayMode)
@@ -60,6 +72,7 @@ class UnitsViewModel(
         Log.d("UnitsViewModel", "Toggling Display mode")
         val newDisplayMode =
             if (unitsDisplayMode.value == UnitsDisplayMode.MAP) {
+                _selectedUnit.postValue(null)
                 UnitsDisplayMode.LIST
             } else {
                 UnitsDisplayMode.MAP