Add persistence

Signed-off-by: Yuuta Liang <yuutaw@student.cs.ubc.ca>

1 files changed, 3 insertions, 0 deletions

diff --git a/src/main/ui/MgmtScreen.java b/src/main/ui/MgmtScreen.java
index 0a25bfe..c630a34 100644
--- a/src/main/ui/MgmtScreen.java
+++ b/src/main/ui/MgmtScreen.java
@@ -73,6 +73,7 @@ public class MgmtScreen implements UIHandler {
         try {
             CertificationRequest req = session.getCa().signCSR();
             System.out.println(Utils.toPEM(req.encodeDER(), "CERTIFICATE REQUEST"));
+            session.setUnsaved(true);
         } catch (Throwable e) {
             System.out.println(e.getMessage());
         }
@@ -90,6 +91,7 @@ public class MgmtScreen implements UIHandler {
             final Byte[] in = session.handleInputPEM("CERTIFICATE");
             final Certificate cert = new Certificate(new BytesReader(in), false);
             session.getCa().installCertificate(cert);
+            session.setUnsaved(true);
         } catch (InvalidCAException | ParseException e) {
             System.out.println(e.getMessage());
         }
@@ -105,6 +107,7 @@ public class MgmtScreen implements UIHandler {
         }
         try {
             session.getCa().generateKey();
+            session.setUnsaved(true);
         } catch (NoSuchAlgorithmException e) {
             System.out.println(e.getMessage());
         }