/**
 * TrackerMap
 * Copyright (C) 2021-2022  Iván Ávalos <avalos@disroot.org>, Henoch Ojeda <imhenoch@protonmail.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package mx.trackermap.TrackerMap.android.session

import android.util.Log
import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import com.zhuinden.eventemitter.EventEmitter
import com.zhuinden.eventemitter.EventSource
import kotlinx.coroutines.DelicateCoroutinesApi
import kotlinx.coroutines.flow.collect
import kotlinx.coroutines.launch
import mx.trackermap.TrackerMap.client.models.User
import mx.trackermap.TrackerMap.controllers.SessionController
import org.koin.core.component.KoinComponent
import org.koin.core.component.inject

@DelicateCoroutinesApi
class UserInformationViewModel : ViewModel(), KoinComponent {

    private val sessionController: SessionController by inject()

    init {
        Log.d("UserInformationVM", "Initializing User Information View Model")
    }

    private val _user = MutableLiveData<User>()
    val user: LiveData<User> get() = _user

    private val loginStateEmitter = EventEmitter<SessionController.LoginState>()
    val loginState: EventSource<SessionController.LoginState> = loginStateEmitter

    init {
        viewModelScope.launch {
            setupLoginStateObserver()
        }
        viewModelScope.launch {
            setupUserObserver()
        }
    }

    private suspend fun setupLoginStateObserver() {
        sessionController.loginStateFlow.collect {
            it?.let {
                loginStateEmitter.emit(it)
            }
        }
    }

    private suspend fun setupUserObserver() {
        sessionController.userFlow.collect {
            it?.let {
                _user.postValue(it)
            }
        }
    }

    fun fetchUserInfo() {
        viewModelScope.launch {
            loginStateEmitter.emit(SessionController.LoginState.Loading)
            try {
                sessionController.getSession()
                loginStateEmitter.emit(SessionController.LoginState.Success)
            } catch (e: Exception) {
                loginStateEmitter.emit(SessionController.LoginState.Failure)
            }
        }
    }

    fun signOut(token: String?) {
        viewModelScope.launch {
            sessionController.logout(token)
        }
    }
}