/*
 * Copyright (c) 2022 New Vector Ltd
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package io.element.android.x

import com.airbnb.mvrx.MavericksState
import com.airbnb.mvrx.MavericksViewModel
import com.airbnb.mvrx.MavericksViewModelFactory
import dagger.assisted.Assisted
import dagger.assisted.AssistedInject
import io.element.android.x.anvilannotations.ContributesViewModel
import io.element.android.x.core.di.daggerMavericksViewModelFactory
import io.element.android.x.di.AppScope
import io.element.android.x.di.SessionComponentsOwner
import io.element.android.x.matrix.Matrix
import kotlinx.coroutines.flow.first
import kotlinx.coroutines.launch

data class MainState(val fake: Boolean = false) : MavericksState

@ContributesViewModel(AppScope::class)
class MainViewModel @AssistedInject constructor(
    private val matrix: Matrix,
    private val sessionComponentsOwner: SessionComponentsOwner,
    @Assisted initialState: MainState
) : MavericksViewModel<MainState>(initialState) {

    companion object :
        MavericksViewModelFactory<MainViewModel, MainState> by daggerMavericksViewModelFactory()

    suspend fun isLoggedIn(): Boolean {
        return matrix.isLoggedIn().first()
    }

    fun startSyncIfLogged() {
        viewModelScope.launch {
            if (!isLoggedIn()) return@launch
        }
    }

    fun stopSyncIfLogged() {
        viewModelScope.launch {
            if (!isLoggedIn()) return@launch
        }
    }

    suspend fun restoreSession() {
        val matrixClient = matrix.restoreSession()
        if (matrixClient == null) {
            throw IllegalStateException("Couldn't restore session...")
        } else {
            sessionComponentsOwner.create(matrixClient)
            matrixClient.startSync()
        }
    }
}