1 files changed, 30 insertions, 9 deletions

diff --git a/src/main/java/design/controller/userinput/menus/MainMenu.java b/src/main/java/design/controller/userinput/menus/MainMenu.java
index 9d5204f..0c42422 100644
--- a/src/main/java/design/controller/userinput/menus/MainMenu.java
+++ b/src/main/java/design/controller/userinput/menus/MainMenu.java
@@ -2,6 +2,8 @@ package design.controller.userinput.menus;
 
 import design.controller.userinput.Menu;
 import design.controller.userinput.MenuOption;
+import design.controller.userinput.UndoActions;
+import design.runtime.Session;
 
 import java.util.List;
 
@@ -13,14 +15,33 @@ public class MainMenu extends Menu {
 
     @Override
     public List<MenuOption> getMenuOptions() {
-        return List.of(
-                new MenuOption("quit", () -> System.exit(0)),
-                new MenuOption("user settings...", () -> new UserSettings().present()),
-                new MenuOption("manage courses...", () -> new ManageCourses().present()),
-                new MenuOption("manage clubs...", () -> new ManageClubs().present()),
-                new MenuOption("statistics...", () -> new StatisticsMenu().present()),
-                new MenuOption("log round...", () -> new HolePlayMenu().present()),
-                new MenuOption("manage data...", () -> new ImportExportMenu().present())
-        );
+        boolean guest = Session.isGuest();
+        if (guest) {
+            return List.of(
+                    new MenuOption("quit", () -> System.exit(0)),
+                    new MenuOption("guest settings...", () -> new UserSettings().present()),
+                    new MenuOption("browse courses...", () -> new SearchMenu().present())
+
+            );
+        } else {
+            return List.of(
+                    new MenuOption("quit", () -> System.exit(0)),
+                    new MenuOption("undo", () -> {
+                        UndoActions.undoWithSave();
+                        this.present();
+                    }),
+                    new MenuOption("redo", () -> {
+                        UndoActions.redoWithSave();
+                        this.present();
+                    }),
+                    new MenuOption("user settings...", () -> new UserSettings().present()),
+                    new MenuOption("manage courses...", () -> new ManageCourses().present()),
+                    new MenuOption("manage clubs...", () -> new ManageClubs().present()),
+                    new MenuOption("statistics...", () -> new StatisticsMenu().present()),
+                    new MenuOption("log round...", () -> new HolePlayMenu().present()),
+                    new MenuOption("League play...", () -> new SelectLeague().present()),
+                    new MenuOption("manage data...", () -> new ImportExportMenu().present())
+            );
+        }
     }
 }