From 14d7ce3be94c2727a78bd6d24190abb8699392c8 Mon Sep 17 00:00:00 2001
From: Agnibho Mondal <mondal@agnibho.com>
Date: Wed, 13 Sep 2023 01:10:59 +0530
Subject: [PATCH] Enabled signature removal

---
 filehandler.py | 7 +++++++
 setting.py     | 2 +-
 window.py      | 8 ++++++++
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/filehandler.py b/filehandler.py
index 943b976..48aa5d5 100644
--- a/filehandler.py
+++ b/filehandler.py
@@ -81,5 +81,12 @@ class FileHandler():
         except FileNotFoundError as e:
             print(e)
 
+    def delete_sign(self):
+        try:
+            os.unlink(os.path.join(self.directory.name, "certificate.pem"))
+            os.unlink(os.path.join(self.directory.name, "signature.p7m"))
+        except Exception as e:
+            raise(e)
+
     def is_signed(self):
         return(os.path.exists(os.path.join(self.directory.name, "certificate.pem")) and (os.path.exists(os.path.join(self.directory.name, "signature.p7m"))))
diff --git a/setting.py b/setting.py
index 3728214..b6e8dba 100644
--- a/setting.py
+++ b/setting.py
@@ -24,7 +24,7 @@ class EditConfiguration(QMainWindow):
     def select_key(self):
         f=QFileDialog.getOpenFileName(self, "Select Private Key", os.path.expanduser("~"), "PEM (*.pem);; All Files (*)")[0]
         if(f):
-            self.input_prescriber.setText(f)
+            self.input_key.setText(f)
     def select_certificate(self):
         f=QFileDialog.getOpenFileName(self, "Select Certificate", os.path.expanduser("~"), "PEM (*.pem);; All Files (*)")[0]
         if(f):
diff --git a/window.py b/window.py
index a671234..7a1941e 100644
--- a/window.py
+++ b/window.py
@@ -133,6 +133,11 @@ class MainWindow(QMainWindow):
         else:
            QMessageBox.information(self, "Save first", "Please save the file before signing.")
 
+    def cmd_unsign(self):
+        self.current_file.delete_sign()
+        self.cmd_save()
+        self.refresh()
+
     def cmd_verify(self):
         try:
             result=self.current_file.verify()
@@ -417,6 +422,8 @@ class MainWindow(QMainWindow):
         action_render2.triggered.connect(self.cmd_render)
         action_sign=QAction("Sign", self)
         action_sign.triggered.connect(self.cmd_sign)
+        action_unsign=QAction("Unsign", self)
+        action_unsign.triggered.connect(self.cmd_unsign)
         action_verify=QAction("Verify", self)
         action_verify.triggered.connect(self.cmd_verify)
         action_configuration=QAction("Configuration", self)
@@ -441,6 +448,7 @@ class MainWindow(QMainWindow):
         menu_prepare.addAction(action_render)
         menu_prepare.addAction(action_refresh)
         menu_prepare.addAction(action_sign)
+        menu_prepare.addAction(action_unsign)
         menu_prepare.addAction(action_verify)
         menu_settings=menubar.addMenu("Settings")
         menu_settings.addAction(action_configuration)
-- 
2.39.2