StoreViewModelAfter.kt

StoreViewModel.kt

import androidx.lifecycle.ViewModel
import com.messiaslima.promogamer.domain.Store
import kotlinx.coroutines.FlowPreview
import kotlinx.coroutines.flow.catch
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.flow.onStart

@FlowPreview
class StoreViewModel(private val storeOrchestrator: StoreOrchestrator) : ViewModel() {
    private val retryTrigger = RetryTrigger()

    val uiState = retryableFlow(retryTrigger) {
        storeOrchestrator.getStores()
            .map<List<Store>, UiState> { UiState.Success(it) }
            .catch { emit(UiState.Error(it))}
            .onStart { emit(UiState.Loading) }
    }

    fun tryAgain() {
        retryTrigger.retry()
    }

    sealed class UiState {
        object Loading : UiState()
        class Success(stores: List<Store>) : UiState()
        class Error(throwable: Throwable) : UiState()
        object Idle : UiState()
    }
}