diff --git a/app/src/main/java/com/example/cupcake/model/OrderViewModel.kt b/app/src/main/java/com/example/cupcake/model/OrderViewModel.kt index 4b8e9dbb9c77e7c2fbf35640b4d6ae452ecff254..67b4a831d9ea255b28b99fcfb6a7ecce51424660 100644 --- a/app/src/main/java/com/example/cupcake/model/OrderViewModel.kt +++ b/app/src/main/java/com/example/cupcake/model/OrderViewModel.kt @@ -7,6 +7,8 @@ import java.text.SimpleDateFormat import java.util.Calendar import java.util.Locale +private const val PRICE_PER_CUPCAKE = 2.00 + class OrderViewModel : ViewModel() { private val _quantity = MutableLiveData<Int>() val quantity: LiveData<Int> = _quantity @@ -26,6 +28,10 @@ class OrderViewModel : ViewModel() { resetOrder() } + private fun updatePrice() { + _price.value = (quantity.value ?: 0) * PRICE_PER_CUPCAKE + } + private fun getPickupOptions(): List<String> { val options = mutableListOf<String>() val formatter = SimpleDateFormat("E MMM d", Locale.getDefault()) @@ -47,6 +53,7 @@ class OrderViewModel : ViewModel() { fun setQuantity(numberCupcakes: Int) { _quantity.value = numberCupcakes + updatePrice() } fun setFlavor(desiredFlavor: String) {