Merge pull request #12 from RIT-SWEN-262/menu-reorganize

Menu reorganize

1 files changed, 46 insertions, 0 deletions

diff --git a/src/main/java/design/controller/userinput/menus/SelectCourse.java b/src/main/java/design/controller/userinput/menus/SelectCourse.java
new file mode 100644
index 0000000..efbdef5
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/SelectCourse.java
@@ -0,0 +1,46 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.Course;
+import design.model.Golfer;
+import design.runtime.Session;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.Objects;
+
+public class SelectCourse extends Menu {
+    private final Golfer golfer = Session.getCurrentGolfer();
+    private Course result;
+
+    @Override
+    public String getTitle() {
+        return "select course";
+    }
+
+    public Course getResult() {
+        Objects.requireNonNull(result);
+        return result;
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        return Arrays.stream(golfer.getCourses())
+                .map(c -> new MenuOption(c.toString(), () -> result = c))
+                .toList();
+    }
+
+    @Override
+    public void present() {
+        // Checks user has courses of not informs them to add one to their profile
+        if (!golfer.hasCourses()) {
+            System.out.println("You don't have any courses saved to your profile yet.");
+            System.out.println("Use the 'Add Course to Profile' menu first.");
+            new MainMenu().present();
+            return;
+        }
+
+        super.present();
+    }
+}