Merge pull request #9 from RIT-SWEN-262/lizzio-SelectClub

Lizzio select club

3 files changed, 110 insertions, 6 deletions

diff --git a/src/main/java/design/controller/userinput/menus/AddClubMenu.java b/src/main/java/design/controller/userinput/menus/AddClubMenu.java
new file mode 100644
index 0000000..ea18ed5
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/AddClubMenu.java
@@ -0,0 +1,77 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.Club;
+import design.model.Golfer;
+import design.persistence.PersonalDatabase;
+import design.runtime.Session;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+import java.util.Scanner;
+
+public class AddClubMenu extends Menu {
+
+    @Override
+    public String getTitle() {
+        return "add a club";
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        List<MenuOption> opts = new ArrayList<>();
+
+        opts.add(new MenuOption("create new club", () -> {
+            Golfer g = Session.getCurrentGolfer();
+            if (g == null) {
+                System.out.println("No user loaded.");
+                new UserSettings().present();
+                return;
+            }
+
+            Scanner sc = new Scanner(System.in);
+
+            System.out.print("Manufacturer: ");
+            String manufacture = sc.nextLine().trim();
+
+            System.out.print("Nickname: ");
+            String nickname = sc.nextLine().trim();
+
+            // Pick type
+            Club.ClubType[] types = Club.ClubType.values();
+            System.out.println("Club type:");
+            for (int i = 0; i < types.length; i++) {
+                System.out.printf("%d: %s%n", i + 1, types[i]);
+            }
+
+            Club.ClubType type = null;
+            while (type == null) {
+                System.out.print("Select (1.." + types.length + "): ");
+                String line = sc.nextLine().trim();
+                int idx = Integer.parseInt(line);
+                if (idx < 1 || idx > types.length) {
+                    System.out.println("Out of range. Try again.");
+                    continue;
+                }
+                type = types[idx - 1];
+            }
+
+            g.addClub(manufacture, nickname, type);
+
+            // Add club to JSON
+            try {
+                PersonalDatabase.INSTANCE.updateGolfer(g);
+                System.out.println("Club added and saved.");
+            } catch (IOException e) {
+                throw new RuntimeException("Failed to save club", e);
+            }
+            new UserSettings().present();
+        }));
+
+        opts.add(new MenuOption("cancel", () -> new UserSettings().present()));
+        return opts;
+    }
+}
diff --git a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
index e12bdfa..14e5345 100644
--- a/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
+++ b/src/main/java/design/controller/userinput/menus/HolePlayMenu.java
@@ -44,13 +44,38 @@ public class HolePlayMenu extends Menu {
         } else {
             // 0) Take a shot
             opts.add(new MenuOption("take a shot", () -> {
-                // Gets club
+                // loads golfers clubs
+                var golfer = Session.getCurrentGolfer();
+                Club[] clubs = (golfer == null) ? new Club[0] : golfer.getClubs();
+
+                if (clubs.length == 0) {
+                    System.out.println("You don't have any clubs yet. Add one first.");
+                    new AddClubMenu().present();
+                    this.present();
+                    return;
+                }
+
+                // list clubs
+                System.out.println("-- YOUR CLUBS --");
+                for (int i = 0; i < clubs.length; i++) {
+                    Club c = clubs[i];
+                    System.out.printf("%d: #%d %s - %s (%s)%n",
+                            i + 1, c.getId(), c.getNickname(), c.getManufacture(), c.getClubType());
+                }
+
+                // user selects one of their clubs
                 Scanner sc = new Scanner(System.in);
-                System.out.print("Club nickname: ");
-                String nick = sc.nextLine().trim();
-                // Defaulted to Driver for now
-                // TODO update logic
-                Club club = new Club("Generic", nick, Club.ClubType.DRIVER);
+                Club club = null;
+                while (club == null) {
+                    System.out.print("Select club # (1.." + clubs.length + "): ");
+                    String line = sc.nextLine().trim();
+                    int idx = Integer.parseInt(line);
+                    if (idx < 1 || idx > clubs.length) {
+                        System.out.println("Out of range. Try again.");
+                        continue;
+                    }
+                    club = clubs[idx - 1];
+                }
 
                 // Get shot distance (defaults to 0 of not stated)
                 System.out.print("Distance (yds, blank=0): ");
diff --git a/src/main/java/design/controller/userinput/menus/UserSettings.java b/src/main/java/design/controller/userinput/menus/UserSettings.java
index 04ec4d0..34b645d 100644
--- a/src/main/java/design/controller/userinput/menus/UserSettings.java
+++ b/src/main/java/design/controller/userinput/menus/UserSettings.java
@@ -60,6 +60,8 @@ public class UserSettings extends Menu {
             this.present();
         }));
 
+        opts.add(new MenuOption("add a club...", () -> new AddClubMenu().present()));
+
         return opts;
     }
 }