1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
package mx.trackermap.TrackerMap.android.units
import android.util.Log
import androidx.lifecycle.*
import kotlinx.coroutines.DelicateCoroutinesApi
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.launch
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.UnitsController
import org.koin.core.component.KoinComponent
import org.koin.core.component.inject
@DelicateCoroutinesApi
class UnitsViewModel(
savedStateHandle: SavedStateHandle
) : ViewModel(), KoinComponent {
enum class UnitsDisplayMode {
MAP, LIST
}
private val unitsController: UnitsController by inject()
private val geofenceController: GeofencesController by inject()
private var _searchQuery = savedStateHandle.getLiveData("searchQuery", "")
private var _unitsDisplayMode = MutableLiveData(UnitsDisplayMode.MAP)
private var _units = MutableLiveData<List<UnitInformation>>()
private var _selectedUnit = MutableLiveData<UnitInformation?>()
private var _mapLayerType = MutableLiveData<MapLayer.Type>()
private var _geofences = MutableLiveData<Map<Int, Geofence>>()
val searchQuery: LiveData<String> get() = _searchQuery
val unitsDisplayMode: LiveData<UnitsDisplayMode> get() = _unitsDisplayMode
val units: LiveData<List<UnitInformation>> get() = _units
val selectedUnit: LiveData<UnitInformation?> get() = _selectedUnit
val mapLayerType: LiveData<MapLayer.Type> get() = _mapLayerType
val geofences: LiveData<Map<Int, Geofence>> get() = _geofences
init {
Log.d("UnitsViewModel", "Initializing Units View Model")
viewModelScope.launch {
setupObservers()
}
viewModelScope.launch {
setupGeofenceObserver()
}
}
private suspend fun setupObservers() {
Log.d("UnitsViewModel", "Setup observers")
unitsController.displayedUnitsFlow.collect { units ->
this._units.value = units
}
}
private suspend fun setupGeofenceObserver() {
geofenceController.geofencesFlow.collect {
this._geofences.postValue(it)
}
}
fun selectUnit(unit: UnitInformation) {
Log.d("UnitsViewModel", "Selecting unit ${unit.device.name}")
_selectedUnit.postValue(unit)
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)
}
fun setMapLayerType(layer: MapLayer.Type) {
_mapLayerType.postValue(layer)
}
fun toggleDisplayMode() {
Log.d("UnitsViewModel", "Toggling Display mode")
val newDisplayMode =
if (unitsDisplayMode.value == UnitsDisplayMode.MAP) {
_selectedUnit.postValue(null)
UnitsDisplayMode.LIST
} else {
UnitsDisplayMode.MAP
}
_unitsDisplayMode.postValue(newDisplayMode)
}
fun search(query: String) {
unitsController.search(query)
}
}
|
