From fac7fb71260b901a893b5b90aae23e86d6c23550 Mon Sep 17 00:00:00 2001
From: sowgro <tpoke.ferrari@gmail.com>
Date: Sat, 11 Oct 2025 22:24:54 -0400
Subject: Menu cleanup - pass 2

---
 .../controller/userinput/menus/SelectCourse.java   | 34 ++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 src/main/java/design/controller/userinput/menus/SelectCourse.java

(limited to 'src/main/java/design/controller/userinput/menus/SelectCourse.java')

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..811586f
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/SelectCourse.java
@@ -0,0 +1,34 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.Course;
+import design.model.Golfer;
+
+import java.util.Arrays;
+import java.util.List;
+
+public class SelectCourse extends Menu {
+    private final Golfer golfer;
+    private Course result;
+
+    public SelectCourse(Golfer golfer) {
+        this.golfer = golfer;
+    }
+
+    @Override
+    public String getTitle() {
+        return "select course";
+    }
+
+    public Course getResult() {
+        return result;
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        return Arrays.stream(golfer.getCourses())
+                .map(c -> new MenuOption(c.toString(), () -> result = c))
+                .toList();
+    }
+}
-- 
cgit v1.2.3


From df8adcc59fad860445527c8a4da820a73036b542 Mon Sep 17 00:00:00 2001
From: sowgro <tpoke.ferrari@gmail.com>
Date: Sat, 11 Oct 2025 23:49:49 -0400
Subject: Menu cleanup - pass 3

---
 .../controller/userinput/menus/SelectCourse.java   | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

(limited to 'src/main/java/design/controller/userinput/menus/SelectCourse.java')

diff --git a/src/main/java/design/controller/userinput/menus/SelectCourse.java b/src/main/java/design/controller/userinput/menus/SelectCourse.java
index 811586f..efbdef5 100644
--- a/src/main/java/design/controller/userinput/menus/SelectCourse.java
+++ b/src/main/java/design/controller/userinput/menus/SelectCourse.java
@@ -4,24 +4,23 @@ 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;
+    private final Golfer golfer = Session.getCurrentGolfer();
     private Course result;
 
-    public SelectCourse(Golfer golfer) {
-        this.golfer = golfer;
-    }
-
     @Override
     public String getTitle() {
         return "select course";
     }
 
     public Course getResult() {
+        Objects.requireNonNull(result);
         return result;
     }
 
@@ -31,4 +30,17 @@ public class SelectCourse extends Menu {
                 .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();
+    }
 }
-- 
cgit v1.2.3