1 files changed, 25 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
new file mode 100644
index 0000000..11df89b
--- /dev/null
+++ b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/units/UnitsViewModel.kt
@@ -0,0 +1,25 @@
+package mx.trackermap.TrackerMap.android.units
+
+import android.util.Log
+import androidx.lifecycle.MutableLiveData
+import androidx.lifecycle.ViewModel
+
+class UnitsViewModel : ViewModel() {
+
+    enum class UnitsDisplayMode {
+        MAP, LIST
+    }
+
+    var unitsDisplayMode = MutableLiveData<UnitsDisplayMode>(UnitsDisplayMode.MAP)
+
+    fun toggleDisplayMode() {
+        Log.d("UnitsViewModel", "Toggling Display mode")
+        val newDisplayMode =
+            if (unitsDisplayMode.value == UnitsDisplayMode.MAP) {
+                UnitsDisplayMode.LIST
+            } else {
+                UnitsDisplayMode.MAP
+            }
+        unitsDisplayMode.postValue(newDisplayMode)
+    }
+}
+\ No newline at end of file