FoodCategoriesViewModel.kt

class FoodCategoriesViewModel constructor(private val repository: FoodMenuRepository) :
    BaseViewModel<
        FoodCategoriesContract.Event, 
        FoodCategoriesContract.State,
        FoodCategoriesContract.Effect>() {

    init {
        viewModelScope.launch { getFoodCategories() }
    }

    override fun setInitialState() =
        FoodCategoriesContract.State(categories = listOf(), isLoading = true)

    override fun handleEvents(event: FoodCategoriesContract.Event) {
        when (event) {
            is FoodCategoriesContract.Event.CategorySelection -> {
                setEffect { 
                  FoodCategoriesContract.Effect.Navigation.ToCategoryDetails(event.categoryName)
                }
            }
        }
    }

    private suspend fun getFoodCategories() {
        val categories = repository.getFoodCategories()
        setState {
            copy(categories = categories, isLoading = false)
        }
        setEffect { FoodCategoriesContract.Effect.ToastDataWasLoaded }
    }
}