add datetime override

1 files changed, 3 insertions, 10 deletions

diff --git a/src/main/java/design/controller/userinput/menus/CourseSearch.java b/src/main/java/design/controller/userinput/menus/CourseSearch.java
index 0d5efa6..9d67b51 100644
--- a/src/main/java/design/controller/userinput/menus/CourseSearch.java
+++ b/src/main/java/design/controller/userinput/menus/CourseSearch.java
@@ -21,6 +21,7 @@ import java.util.Scanner;
 public class CourseSearch extends Menu {
     private final CurrentSearchQuery query = CurrentSearchQuery.INSTANCE;
     private final PersonalDatabase personalDB = PersonalDatabase.INSTANCE;
+    private final Golfer golfer = Session.getCurrentGolfer();
 
     @Override
     public String getTitle() {
@@ -74,18 +75,10 @@ public class CourseSearch extends Menu {
                 var name = String.format("%s, %s, Difficulty: %s, %s holes, %s total par",
                         c.getName(), c.getLocation(), c.getDifficultyRating(), c.getHoleCount(), c.getTotalPar());
                 menuOptions.add(new MenuOption(name, () -> {
-                    Golfer currentGolfer = Session.getCurrentGolfer();
-                    if (currentGolfer == null) {
-                        // if we aren't logged in, notify the user.
-                        System.out.println("\n\n !!! log into a golfer account to add courses to your profile. !!! \n\n");
-                        new MainMenu().present();
-                        return;
-                    }
-
                     // add the course, try to save to DB.
-                    currentGolfer.addCourse(c);
+                    golfer.addCourse(c);
                     try {
-                        personalDB.updateGolfer(currentGolfer);
+                        personalDB.updateGolfer(golfer);
                     } catch (IOException e) {
                         e.printStackTrace(); // not sure if we should format this prettier for the user if the DB fails.
                     }