Back to the original MVIViewModel!

MVIViewModel.kt

abstract class MVIViewModel<State, Event>(
    protected val savedStateHandle: SavedStateHandle? = null,
) : ViewModel() {

    private val _uiState by lazy { MutableStateFlow(initialState()) }
    val uiState get() = _uiState.asStateFlow()

    fun update(block: State.() -> State) {
        _uiState.update(block)
    }

    abstract fun initialState(): State
    abstract fun handleEvent(event: Event)
}