https://github.com/TheChance101/beep-beep/blob/develop/client_end_user/shared/src/commonMain/kotlin/presentation/meals/MealsScreenModel.kt

MealsScreenModel.kt

class MealsScreenModel(
    private val cuisineId: String,
    private val manageRestaurant: IExploreRestaurantUseCase,
) : BaseScreenModel<MealsUiState, MealsUiEffect>(MealsUiState()), MealsInteractionListener,
    MealInteractionListener {

    init {
        getData()
    }

    private fun getData() {
        updateState { it.copy(isLoading = true, cuisineName = cuisineName) }
        tryToExecute(
            { manageRestaurant.getMealsInCuisine(cuisineId) },
            ::onGetMealsSuccess,
            ::onError
        )
    }

    private fun onError(errorState: ErrorState) {
        updateState { it.copy(isLoading = false) }
        when (errorState) {
            is ErrorState.NoInternet -> {
                updateState { it.copy(error = errorState) }
            }

            else -> {
                updateState { it.copy(error = errorState) }
            }
        }
    }

    private fun onGetMealsSuccess(meals: Flow<PagingData<Meal>>) {
        updateState { it.copy(meals = meals.toUIState(), isLoading = false) }
    }