From 4e45988ddff84205d2c5c81466ae86f57b25de17 Mon Sep 17 00:00:00 2001
From: Michael Lizzio <mjl2396@rit.edu>
Date: Mon, 10 Nov 2025 15:13:27 -0500
Subject: Used UndoManager in Menus to Capture Sanpshots

---
 src/main/java/design/controller/userinput/menus/ManageCourses.java | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'src/main/java/design/controller/userinput/menus/ManageCourses.java')

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) {
-- 
cgit v1.2.3