Menu cleanup - pass 2

1 files changed, 34 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..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();
+    }
+}