aboutsummaryrefslogtreecommitdiff
path: root/shared/src/commonMain/kotlin/mx/trackermap/TrackerMap/controllers/UnitsController.kt
blob: 584f656409b7b16d609c99234822d355d9c35517 (plain)
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
package mx.trackermap.TrackerMap.controllers

import kotlinx.coroutines.DelicateCoroutinesApi
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.flow.MutableStateFlow
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.flow.combine
import kotlinx.coroutines.launch
import mx.trackermap.TrackerMap.client.apis.DevicesApi
import mx.trackermap.TrackerMap.client.apis.PositionsApi
import mx.trackermap.TrackerMap.client.models.Position
import mx.trackermap.TrackerMap.client.models.UnitInformation

@DelicateCoroutinesApi
class UnitsController(
    private val devicesApi: DevicesApi,
    private val positionsApi: PositionsApi
) {

    val displayedUnitsFlow = MutableStateFlow<List<UnitInformation>>(emptyList())
    private val unitsFlow = MutableStateFlow<List<UnitInformation>>(emptyList())
    private val queryFlow = MutableStateFlow("")

    init {
        GlobalScope.launch {
            fetchUnits()
        }
        GlobalScope.launch {
            setupFlows()
        }
    }

    private suspend fun fetchUnits() {
        val devices = devicesApi.devicesGet()
        val positions = positionsApi.positionsGet()
        val positionsMap: MutableMap<Int, Position> = mutableMapOf()

        positions.forEach { position ->
            position.deviceId?.let {
                positionsMap[it] = position
            }
        }

        val units = devices.map {
            UnitInformation(it, positionsMap[it.id])
        }
        unitsFlow.value = units
    }

    fun search(query: String) {
        queryFlow.value = query
    }

    suspend fun getUnit(deviceId: Int): UnitInformation {
        val device = devicesApi.devicesGet(id = deviceId).first()
        val position = positionsApi.positionsGet(id = device.positionId).first()

        return UnitInformation(device, position)
    }

    private suspend fun setupFlows() {
        unitsFlow.combine(queryFlow) { units, query ->
            units.filter { unit ->
                unit.device.name.contains(query, true) ||
                        unit.device.contact?.contains(query, true) == true ||
                        unit.position?.address?.contains(query, true) == true
            }
        }.collect { units ->
            this.displayedUnitsFlow.value = units
        }
    }
}