Merge pull request #172 from samuelory/Samuel-v2.6

Caught NoSuchElementException and IllegalStateException
AY2324S2-CS2113-T11-3 · Apr 14, 2024 · 37be645 · 37be645
2 parents a350bac + 3fc60d5
commit 37be645
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/florizz/core/Ui.java b/src/main/java/florizz/core/Ui.java
@@ -5,6 +5,7 @@
 
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.NoSuchElementException;
 import java.util.Scanner;
 
 public class Ui {
@@ -42,7 +43,11 @@ public static void printBreakLine(){
      * @return Returns the user input as one String.
      */
     public String getInput(){
-        return inputScanner.nextLine();
+        try{
+            return inputScanner.nextLine();
+        } catch(NoSuchElementException | IllegalStateException error){
+            return "bye";
+        }
     }
 
     /**