ChapletFlowViewModel.kt

class ChapletFlowViewModel(
    private val challengeRepository: ChallengeRepository
) : ViewModel() {

    private val _shouldCloseFlow = MutableSharedFlow<Boolean>()
    val shouldCloseFlow: SharedFlow<Boolean> = _shouldCloseFlow

    val backStack = mutableStateListOf<ChapletRoutes>()

    init {
        if (backStack.isEmpty()) {
            backStack.add(ChapletRoutes.Start)
        }
    }

    fun navigateTo(route: ChapletRoutes) {
        backStack.add(route)
    }

    fun onChapletDone() {
        viewModelScope.launch {
            challengeRepository.updateChallengeV2(ChallengeId.CHAPLETS)
            _shouldCloseFlow.emit(true)
        }
    }
}