caught numberFormatException

AY2324S2-CS2113-T11-3 · Apr 9, 2024 · 3554131 · 3554131
1 parent 56a762b
commit 3554131
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/main/java/florizz/core/Parser.java b/src/main/java/florizz/core/Parser.java
@@ -240,7 +240,12 @@ private static AddFlowerCommand handleAddFlower(String argument, boolean enableU
         String flowerName = argument.substring(0,quantityIndex).trim().toLowerCase();
         String quantityString = removePrefix(argument.substring(quantityIndex, prefixIndex), QUANTITY).trim();
         // [WARNING] might need to check if it's a valid integer
-        int quantity = Integer.parseInt(quantityString);
+        int quantity;
+        try {
+            quantity = Integer.parseInt(quantityString);
+        }catch(NumberFormatException error){
+            throw new FlorizzException("Invalid number inputted, please enter a sensible number next time!");
+        }
         String bouquetName = removePrefix(argument.substring(prefixIndex), ADD_FLOWER_PREFIX).trim();
         if (includeColour){
             int colourIndex = argument.indexOf(COLOUR);
@@ -285,7 +290,12 @@ private static RemoveFlowerCommand handleRemoveFlower(String argument) throws Fl
         String flowerName = argument.substring(0, quantityIndex).trim().toLowerCase();
         String quantityString = removePrefix(argument.substring(quantityIndex, prefixIndex), QUANTITY).trim();
         // [WARNING] might need to check if it's a valid integer
-        int quantity = Integer.parseInt(quantityString);
+        int quantity;
+        try {
+            quantity = Integer.parseInt(quantityString);
+        }catch(NumberFormatException error){
+            throw new FlorizzException("Invalid number inputted, please enter a sensible number next time!");
+        }
         String bouquetName = removePrefix(argument.substring(prefixIndex), REMOVE_FLOWER_PREFIX).trim();
         if (includeColour){
             int colourIndex = argument.indexOf(COLOUR);