Menu cleanup - pass 1

1 files changed, 58 insertions, 0 deletions

diff --git a/src/main/java/design/controller/userinput/menus/ManageCourses.java b/src/main/java/design/controller/userinput/menus/ManageCourses.java
new file mode 100644
index 0000000..5156a9b
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/ManageCourses.java
@@ -0,0 +1,58 @@
+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.ArrayList;
+import java.util.List;
+
+public class ManageCourses extends Menu {
+    Golfer golfer = Session.getCurrentGolfer();
+
+    @Override
+    public String getTitle() {
+        return "manage courses";
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        List<MenuOption> opts = new java.util.ArrayList<>();
+        opts.add(new MenuOption("return to main menu", () -> new MainMenu().present()));
+
+        opts.add(new MenuOption("list courses", () -> {
+            for (Course course : golfer.getCourses()) {
+                System.out.printf("- %s\n", course);
+            }
+            this.present();
+        }));
+
+        opts.add(new MenuOption("remove course...", () -> {
+            new Menu() {
+                @Override
+                public String getTitle() {
+                    return "remove course";
+                }
+
+                @Override
+                public List<MenuOption> getMenuOptions() {
+                    List<MenuOption> list = new ArrayList<>();
+                    for (Course c : golfer.getCourses()) {
+                        MenuOption menuOption = new MenuOption(c.toString(), () -> {
+                            golfer.removeCourse(c);
+                            this.present();
+                        });
+                        list.add(menuOption);
+                    }
+                    return list;
+                }
+            }.present();
+        }));
+
+        opts.add(new MenuOption("add course...", () -> new SearchMenu().present()));
+
+        return opts;
+    }
+}