Refactor file handling in Encryption and Decryption classes; add deleteFile method in TempFile class and clean up unused comments

Author: Anof-cyber

.gitignore

@@ -3,7 +3,7 @@
 ##############################
 .mtj.tmp/
 *.class
-*.jar
+
 *.war
 *.ear
 *.nar

PyCript-Websocket.jar

@@ -10,19 +10,21 @@ public ByteArray decryptAndPassToTempFile(ByteArray content) {
         TempFile tempFile = new TempFile();
         String tempFilePath = tempFile.processData(decryptedContent);
 
-        // Get the decryption file path from the UI class
         String decryptionFilePath = UI.getInstance().getDecryptionFilePath();
 
-        // Call the Execution class to run the system command
         Execution execution = new Execution();
         boolean success = execution.runCommand(decryptionFilePath, tempFilePath);
 
         if (success) {
-            // Read the updated content from the temp file
             String updatedContent = tempFile.readFileContent(tempFilePath);
+            boolean isDeleted = tempFile.deleteFile(tempFilePath);
+            if (isDeleted) {
+                System.out.println("Temporary file deleted successfully.");
+            } else {
+                System.out.println("Failed to delete the temporary file.");
+            }
             return ByteArray.byteArray(updatedContent);
         } else {
-            // Return the original content if the execution was not successful
             return content;
         }
     }

src/main/java/com/pycriptsocket/Decryption.java

@@ -9,20 +9,22 @@ public ByteArray encryptAndPassToTempFile(ByteArray content) {
 
         TempFile tempFile = new TempFile();
         String tempFilePath = tempFile.processData(encryptedContent);
-
-        // Get the encryption file path from the UI class
         String encryptionFilePath = UI.getInstance().getEncryptionFilePath();
 
         // Call the Execution class to run the system command
         Execution execution = new Execution();
         boolean success = execution.runCommand(encryptionFilePath, tempFilePath);
 
         if (success) {
-            // Read the updated content from the temp file
             String updatedContent = tempFile.readFileContent(tempFilePath);
+            boolean isDeleted = tempFile.deleteFile(tempFilePath);
+            if (isDeleted) {
+                System.out.println("Temporary file deleted successfully.");
+            } else {
+                System.out.println("Failed to delete the temporary file.");
+            }
             return ByteArray.byteArray(updatedContent);
         } else {
-            // Return the original content if the execution was not successful
             return content;
         }
     }

src/main/java/com/pycriptsocket/Encryption.java

@@ -17,15 +17,13 @@ public class PyCript implements BurpExtension
     @Override
     public void initialize(MontoyaApi api)
     {
-        // set extension name
         api.extension().setName("PyCript WebSocket");
 
         Logging logging = api.logging();
         logging.logToOutput("Author: Sourav Kalal");
         logging.logToOutput("VERSION: 0.1");
         logging.logToOutput("GitHub - https://github.com/Anof-cyber/PyCript-WebSocket");
         logging.logToOutput("Website - https://souravkalal.tech/");
-        logging.logToOutput("Documentation - https://pycript.souravkalal.tech/");
         api.userInterface().registerSuiteTab("PyCript WebSocket", new UI(api));
 
         api.userInterface().registerWebSocketMessageEditorProvider(new WebSocketEditorProvider(api));

src/main/java/com/pycriptsocket/PyCript.java

@@ -30,4 +30,9 @@ public String readFileContent(String filePath) {
             return null;
         }
     }
+
+    public boolean deleteFile(String filePath) {
+        File file = new File(filePath);
+        return file.delete();
+    }
 }