Merge branch 'main' into statistic-dev

4 files changed, 193 insertions, 0 deletions

diff --git a/src/main/java/design/controller/userinput/menus/CourseSearch.java b/src/main/java/design/controller/userinput/menus/CourseSearch.java
new file mode 100644
index 0000000..019965c
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/CourseSearch.java
@@ -0,0 +1,104 @@
+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.model.course_search.CourseList;
+import design.model.course_search.CurrentSearchQuery;
+import design.model.course_search.ICourse;
+import design.persistence.PersonalDatabase;
+import design.runtime.Session;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Scanner;
+
+/*
+ * The actual SEARCH feature of course searching.
+ */
+public class CourseSearch extends Menu {
+    CurrentSearchQuery query = CurrentSearchQuery.INSTANCE;
+    PersonalDatabase GolferDB = PersonalDatabase.INSTANCE;
+
+    @Override
+    public String getTitle() {
+        return "select course";
+    }
+
+    /*
+     * Prompt for input and search.
+     */
+    public void search() {
+        System.out.print("Enter search term (blank for all): ");
+        Scanner sc = new Scanner(System.in);
+        String searchTerm = sc.nextLine();
+
+        // search and present 
+        query.search(searchTerm);
+        this.present();
+
+        // reset the query after we're done.
+        query.reset();
+    }
+
+    /*
+     * Display the results of our search.
+     */
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        var l = new ArrayList<MenuOption>();
+        List<ICourse> queryResult = query.getQueryResult().getCourses();
+
+        // 0 - return to main menu
+        l.add(new MenuOption("return to main menu", () -> new MainMenu().present()));
+
+        // if we find no results, let the user know.
+        if (queryResult.isEmpty()) {
+            System.out.println("\nNo matching courses found.\n");
+        }
+
+        // traverse the course list tree and add menu options for each leaf (course)
+        addCoursesRecursive(l, query.getQueryResult());
+        return l;
+    }
+
+
+    // recursively go through tree structure of courselist to make menu options.
+    // this is all for displaying the menu options, not the actual sorting. 
+    private void addCoursesRecursive(List<MenuOption> menuOptions, CourseList list) {
+        for (ICourse icourse : list.getCourses()) {
+            // if we find a leaf (course), display it as a menu option
+            if (icourse instanceof Course c) {
+                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);
+                    try {
+                        GolferDB.updateGolfer(currentGolfer);
+                    } catch (IOException e) {
+                        e.printStackTrace(); // not sure if we should format this prettier for the user if the DB fails.
+                    }
+
+                    System.out.println("\n Course added to profile. \n");
+                    new MainMenu().present();
+                }
+                ));
+            }
+            // if not, we need to traverse another courselist
+            else if (icourse instanceof CourseList sublist) {
+                addCoursesRecursive(menuOptions, sublist);
+            }
+        }
+    }
+}
diff --git a/src/main/java/design/controller/userinput/menus/FiltersMenu.java b/src/main/java/design/controller/userinput/menus/FiltersMenu.java
new file mode 100644
index 0000000..9f2d8df
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/FiltersMenu.java
@@ -0,0 +1,53 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.course_search.*;
+
+import java.util.List;
+
+import design.model.course_search.CurrentSearchQuery;
+
+/*
+ * Tracks the user's current filters on their search.
+ */
+public class FiltersMenu extends Menu {
+    CurrentSearchQuery query = CurrentSearchQuery.INSTANCE;
+
+    @Override
+    public String getTitle() {
+        return "filter menu";
+    }
+
+    /*
+     * Add a new filter to our filter list
+     */
+    public void addFilter(CourseSorter filter) {
+        CurrentSearchQuery.INSTANCE.addFilter(filter);
+        System.out.println("\nAdded filter.\n");
+        new SearchMenu().present();
+    }
+
+    /*
+     * clear all of our filters.
+     */
+    public void clearFilters() {
+        CurrentSearchQuery.INSTANCE.clearFilters();
+        System.out.println("\nCleared all filters.\n");
+        new SearchMenu().present();        
+    }
+
+    /*
+     * show our current filters.
+     */
+    @Override
+    public List<MenuOption> getMenuOptions() {
+
+        System.out.println("\nYou are filtering by ");
+        System.out.println(query.printFilters());
+
+        return List.of(
+                new MenuOption("return to search", () -> new SearchMenu().present())
+        );
+    }
+}
diff --git a/src/main/java/design/controller/userinput/menus/MainMenu.java b/src/main/java/design/controller/userinput/menus/MainMenu.java
index 1d6b72d..71a93b0 100644
--- a/src/main/java/design/controller/userinput/menus/MainMenu.java
+++ b/src/main/java/design/controller/userinput/menus/MainMenu.java
@@ -15,6 +15,7 @@ public class MainMenu extends Menu {
     public List<MenuOption> getMenuOptions() {
         return List.of(
                 new MenuOption("user settings...", () -> new UserSettings().present()),
+                new MenuOption("add course to profile...", () -> new SearchMenu().present()),
                 new MenuOption("statistics...", () -> new StatisticsMenu().present()),
                 new MenuOption("log round...", () -> new CourseSelectMenu().present()),
                 new MenuOption("quit", () -> {}));
diff --git a/src/main/java/design/controller/userinput/menus/SearchMenu.java b/src/main/java/design/controller/userinput/menus/SearchMenu.java
new file mode 100644
index 0000000..8b50265
--- /dev/null
+++ b/src/main/java/design/controller/userinput/menus/SearchMenu.java
@@ -0,0 +1,35 @@
+package design.controller.userinput.menus;
+
+import design.controller.userinput.Menu;
+import design.controller.userinput.MenuOption;
+import design.model.course_search.SortByDifficulty;
+import design.model.course_search.SortByHoles;
+import design.model.course_search.SortByLocation;
+import design.model.course_search.SortByPar;
+
+import java.util.List;
+
+/*
+ * The main control panel for course searching.
+ */
+public class SearchMenu extends Menu {
+
+    @Override
+    public String getTitle() {
+        return "search menu";
+    }
+
+    @Override
+    public List<MenuOption> getMenuOptions() {
+        return List.of(
+                new MenuOption("return to main menu", () -> new MainMenu().present()),
+                new MenuOption("search...", () -> new CourseSearch().search()),
+                new MenuOption("add difficulty filter...", () -> new FiltersMenu().addFilter(new SortByDifficulty())),
+                new MenuOption("add hole count filter...", () -> new FiltersMenu().addFilter(new SortByHoles())),
+                new MenuOption("add location filter...", () -> new FiltersMenu().addFilter(new SortByLocation())),
+                new MenuOption("add par filter...", () -> new FiltersMenu().addFilter(new SortByPar())),
+                new MenuOption("list filters", () -> new FiltersMenu().present()),
+                new MenuOption("clear all filters", () -> new FiltersMenu().clearFilters())
+        );
+    }
+}