package mx.trackermap.TrackerMap.android.units

import android.util.Log
import androidx.lifecycle.*
import com.mousebird.maply.RemoteTileInfoNew
import com.mousebird.maply.TileInfoNew
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.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 _mapTileInfo = MutableLiveData<TileInfoNew>()
    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 mapTileInfo: LiveData<TileInfoNew> get() = _mapTileInfo
    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 setMapLayer(url: String, minZoom: Int = 0, maxZoom: Int = 21) {
        val tileInfo = RemoteTileInfoNew(url, minZoom, maxZoom)
        _mapTileInfo.postValue(tileInfo)
    }

    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)
    }
}