feat: add item check

Hamedblue1381 · Feb 20, 2024 · b93506b · b93506b
1 parent 16fe70d
commit b93506b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/state/item_request_state.go b/state/item_request_state.go
@@ -13,6 +13,18 @@ func (i *ItemRequestState) InsertCoin() error {
 }
 
 func (i *ItemRequestState) ChooseItem(id int) error {
+	itemExists := false
+	for _, item := range i.vendingMachine.Items {
+		if item.ID == id {
+			itemExists = true
+			break
+		}
+	}
+
+	if !itemExists {
+		return fmt.Errorf("item with ID %d does not exist", id)
+	}
+
 	if i.vendingMachine.Stock > 0 {
 		i.vendingMachine.SetState(i.vendingMachine.Dispensing)
 		return nil