added the ability for users to save courses to their profile

1 files changed, 14 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
index 6bba2ef..7170b5d 100644
--- a/src/main/java/design/controller/userinput/menus/CourseSearch.java
+++ b/src/main/java/design/controller/userinput/menus/CourseSearch.java
@@ -5,7 +5,9 @@ import design.controller.userinput.MenuOption;
 import design.runtime.*;
 import design.model.*;
 import design.model.course_search.*;
+import design.persistence.PersonalDatabase;
 
+import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Scanner;
@@ -15,6 +17,7 @@ import java.util.Scanner;
  */
 public class CourseSearch extends Menu {
     CurrentSearchQuery query = CurrentSearchQuery.INSTANCE;
+    PersonalDatabase GolferDB = PersonalDatabase.INSTANCE;
 
     @Override
     public String getTitle() {
@@ -81,8 +84,19 @@ public class CourseSearch extends Menu {
                         new MainMenu().present();
                     }
 
+                    // add the course, try to save to DB.
                     currentGolfer.addCourse(c);
+                    try
+                    {
+                        GolferDB.updateGolfer(currentGolfer);
+                    }
+                    catch(IOException e)
+                    {
+                        System.out.println(e); // 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(); 
                 }
             ));
             }