Prompt for password and cleanup

1 files changed, 15 insertions, 8 deletions

diff --git a/src/main/java/design/controller/userinput/menus/SelectUser.java b/src/main/java/design/controller/userinput/menus/SelectUser.java
index 8dee860..ffae559 100644
--- a/src/main/java/design/controller/userinput/menus/SelectUser.java
+++ b/src/main/java/design/controller/userinput/menus/SelectUser.java
@@ -15,6 +15,7 @@ import java.util.Scanner;
 public class SelectUser extends Menu {
     private final PersonalDatabase personalDB = PersonalDatabase.INSTANCE;
     private final Action doAfter;
+    private final Scanner scanner = new Scanner(System.in);
 
     public SelectUser(Action doAfter) {
         this.doAfter = doAfter;
@@ -31,25 +32,31 @@ public class SelectUser extends Menu {
 
         for (Golfer i : personalDB.getGolfers()) {
             MenuOption menuOption = new MenuOption(i.getUsername(), () -> {
-                // Sets user in Session too
-                Session.setCurrentGolfer(i);
-                System.out.printf("Loaded user: %s%n", i.getUsername());
-                doAfter.onCommand();
+                System.out.print("Enter password: ");
+                var password = scanner.nextLine();
+                if (i.checkPassword(password)) {
+                    // Sets user in Session too
+                    Session.setCurrentGolfer(i);
+                    System.out.printf("Loaded user: %s%n", i.getUsername());
+                    doAfter.onAction();
+                } else {
+                    System.out.println("Incorrect password. ");
+                    this.present();
+                }
             });
             opts.add(menuOption);
         }
 
         opts.add(new MenuOption("<create>", () -> {
-            Scanner sc = new Scanner(System.in);
 
             System.out.print("Enter username: ");
-            var username = sc.nextLine();
+            var username = scanner.nextLine();
 
             System.out.print("Enter full name: ");
-            var fullName = sc.nextLine();
+            var fullName = scanner.nextLine();
 
             System.out.print("Enter password: ");
-            var password = sc.nextLine();
+            var password = scanner.nextLine();
 
             try {
                 personalDB.addGolfer(new Golfer(fullName, username, password));