fix: allows encryption with specified key

open-amt-cloud-toolkit · Oct 22, 2024 · 4350ac9 · 4350ac9
1 parent 9e27ab0
commit 4350ac9
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/security/encrypt.go b/pkg/security/encrypt.go
@@ -10,7 +10,12 @@ import (
 
 // Encrypt encrypts a string.
 func (c Crypto) Encrypt(plainText string) (string, error) {
-	block, err := aes.NewCipher([]byte(c.EncryptionKey))
+	return c.EncryptWithKey(plainText, c.EncryptionKey)
+}
+
+// Encrypt encrypts a string with provided key.
+func (c Crypto) EncryptWithKey(plainText string, key string) (string, error) {
+	block, err := aes.NewCipher([]byte(key))
 	if err != nil {
 		return "", err
 	}