(#11) Updated encrypt method to work with different algorithms

Author: Simon Hofmann

lib/encrypt.function.spec.ts

@@ -1,3 +1,4 @@
+import {Algorithm} from "./algorithm.enum";
 import {encrypt} from "./encrypt.function";
 
 describe("secret", () => {
@@ -9,6 +10,6 @@ describe("secret", () => {
         // WHEN
 
         // THEN
-        await expect(() => encrypt(input, key)).not.toThrow();
+        await expect(() => encrypt(input, key, Algorithm.AES128CBC)).not.toThrow();
     });
 });

lib/encrypt.function.ts

@@ -1,16 +1,27 @@
 import * as crypto from "crypto";
+import {Algorithm} from "./algorithm.enum";
 import {base64EncodedBufferFromString} from "./buffer.function";
-import {getAlgorithm} from "./get-algorithm.function";
+import {getCipherSpec} from "./get-cipher-spec.function";
 import {generateRandom} from "./random.function";
 
-export const encrypt = async (plainText: string, keyBase64: string): Promise<string> => {
-    const cipherSpec = getAlgorithm(keyBase64);
+export const encrypt = async (
+    plainText: string,
+    keyBase64: string,
+    algorithm: Algorithm,
+    useIV: boolean = true): Promise<string> => {
+    const cipherSpec = getCipherSpec(algorithm);
     const key = base64EncodedBufferFromString(keyBase64);
-    const iv = await generateRandom({
-        byteLength: cipherSpec.blockSize,
-    });
 
-    const cipher = crypto.createCipheriv(cipherSpec.cipher, key, iv);
-    const part = cipher.update(plainText);
-    return Buffer.concat([iv, part, cipher.final()]).toString("base64");
+    if (useIV) {
+        const iv = await generateRandom({
+            byteLength: cipherSpec.blockSize,
+        });
+        const cipher = crypto.createCipheriv(algorithm, key, iv);
+        const part = cipher.update(plainText);
+        return Buffer.concat([iv, part, cipher.final()]).toString("base64");
+    } else {
+        const cipher = crypto.createCipher(algorithm, key);
+        const part = cipher.update(plainText);
+        return Buffer.concat([part, cipher.final()]).toString("base64");
+    }
 };