Used UndoManager in Menus to Capture Sanpshots

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/java/design/controller/userinput/menus/ManageCourses.java b/src/main/java/design/controller/userinput/menus/ManageCourses.java
index b592cd2..4f3efe5 100644
--- a/src/main/java/design/controller/userinput/menus/ManageCourses.java
+++ b/src/main/java/design/controller/userinput/menus/ManageCourses.java
@@ -6,6 +6,7 @@ import design.model.Course;
 import design.model.Golfer;
 import design.persistence.PersonalDatabase;
 import design.runtime.Session;
+import design.model.undo.UndoManager;
 
 import java.io.IOException;
 import java.util.List;
@@ -34,7 +35,11 @@ public class ManageCourses extends Menu {
         opts.add(new MenuOption("remove course...", () -> {
             var selector = new SelectCourse();
             selector.present();
-            golfer.removeCourse(selector.getResult());
+            Course course = selector.getResult();
+
+            UndoManager.instance().capture(golfer, "Remove course " + course.getName());
+
+            golfer.removeCourse(course);
             try {
                 personalDB.updateGolfer(golfer);
             } catch (IOException ex) {