Adding mysql support.

app/api/agent/[agentId]/backup/helpers.ts

@@ -2,7 +2,7 @@ import fs from "node:fs";
 import forge from "node-forge";
 
 
-export async function decryptedDump(file: File, aesKeyHex: string, ivHex: string): Promise<File> {
+export async function decryptedDump(file: File, aesKeyHex: string, ivHex: string, fileExtension: string ): Promise<File> {
     const privateKeyPem = fs.readFileSync("private/keys/server_private.pem", "utf8");
     const privateKey = forge.pki.privateKeyFromPem(privateKeyPem);
 
@@ -10,7 +10,7 @@ export async function decryptedDump(file: File, aesKeyHex: string, ivHex: string
     const encryptedAesKey = forge.util.hexToBytes(aesKeyHex);
     const aesKey = privateKey.decrypt(encryptedAesKey, "RSA-OAEP", {
         md: forge.md.sha256.create(),
-        mgf1: { md: forge.md.sha256.create() },
+        mgf1: {md: forge.md.sha256.create()},
     });
 
     // Read encrypted file content
@@ -19,7 +19,7 @@ export async function decryptedDump(file: File, aesKeyHex: string, ivHex: string
 
     // AES decryption
     const decipher = forge.cipher.createDecipher("AES-CBC", aesKey);
-    decipher.start({ iv });
+    decipher.start({iv});
     decipher.update(forge.util.createBuffer(encryptedBuffer.toString("binary")));
     const success = decipher.finish();
 
@@ -33,8 +33,19 @@ export async function decryptedDump(file: File, aesKeyHex: string, ivHex: string
     // Return a File so you can use file.arrayBuffer() later
     return new File(
         [decryptedBuffer],
-        file.name.replace(/\.enc$/, ".dump"), // rename if needed
+        file.name.replace(/\.enc$/, fileExtension),
         {type: "application/octet-stream"}
     );
 }
 
+
+export function getFileExtension(dbType: string) {
+    switch (dbType) {
+        case "postgresql":
+            return ".dump";
+        case "mysql":
+            return ".sql";
+        default:
+            return ".dump";
+    }
+}