Menu cleanup - pass 2

1 files changed, 34 insertions, 0 deletions

diff --git a/src/main/java/design/controller/userinput/menus/SelectClub.java b/src/main/java/design/controller/userinput/menus/SelectClub.java
new file mode 100644
index 0000000..09facdb
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/SelectClub.java
@@ -0,0 +1,34 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.Club;
+import design.model.Golfer;
+
+import java.util.Arrays;
+import java.util.List;
+
+public class SelectClub extends Menu {
+    private final Golfer golfer;
+    private Club result;
+
+    public SelectClub(Golfer golfer) {
+        this.golfer = golfer;
+    }
+
+    @Override
+    public String getTitle() {
+        return "select club";
+    }
+
+    public Club getResult() {
+        return result;
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        return Arrays.stream(golfer.getClubs())
+                .map(g -> new MenuOption(g.toString(), () -> result = g))
+                .toList();
+    }
+}