"Close mappings" option now check if you have change your mappings and if you want to save them

1 files changed, 24 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
index 038698b..befe129 100644
--- a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
+++ b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
@@ -146,7 +146,30 @@ public class MenuBar extends JMenuBar {
             {
                 JMenuItem item = new JMenuItem("Close Mappings");
                 menu.add(item);
-                item.addActionListener(event -> this.gui.getController().closeMappings());
+                item.addActionListener(event -> {
+                    if (this.gui.getController().isDirty())
+                    {
+                        this.gui.showDiscardDiag((response -> {
+                            if (response == JOptionPane.YES_OPTION)
+                            {
+                                try
+                                {
+                                    gui.saveMapping();
+                                    this.gui.getController().closeMappings();
+                                } catch (IOException e)
+                                {
+                                    throw new Error(e);
+                                }
+                            }
+                            else if (response == JOptionPane.NO_OPTION)
+                                this.gui.getController().closeMappings();
+                            return null;
+                        }), "Save and close", "Discard changes", "Cancel");
+                    }
+                    else
+                        this.gui.getController().closeMappings();
+
+                });
                 this.closeMappingsMenu = item;
             }
             menu.addSeparator();