base

plane-src/apps/api/plane/license/utils/encryption.py

@@ -0,0 +1,44 @@
+# Copyright (c) 2023-present Plane Software, Inc. and contributors
+# SPDX-License-Identifier: AGPL-3.0-only
+# See the LICENSE file for details.
+
+import base64
+import hashlib
+from django.conf import settings
+from cryptography.fernet import Fernet
+
+from plane.utils.exception_logger import log_exception
+
+
+def derive_key(secret_key):
+    # Use a key derivation function to get a suitable encryption key
+    dk = hashlib.pbkdf2_hmac("sha256", secret_key.encode(), b"salt", 100000)
+    return base64.urlsafe_b64encode(dk)
+
+
+# Encrypt data
+def encrypt_data(data):
+    try:
+        if data:
+            cipher_suite = Fernet(derive_key(settings.SECRET_KEY))
+            encrypted_data = cipher_suite.encrypt(data.encode())
+            return encrypted_data.decode()  # Convert bytes to string
+        else:
+            return ""
+    except Exception as e:
+        log_exception(e)
+        return ""
+
+
+# Decrypt data
+def decrypt_data(encrypted_data):
+    try:
+        if encrypted_data:
+            cipher_suite = Fernet(derive_key(settings.SECRET_KEY))
+            decrypted_data = cipher_suite.decrypt(encrypted_data.encode())  # Convert string back to bytes
+            return decrypted_data.decode()
+        else:
+            return ""
+    except Exception as e:
+        log_exception(e)
+        return ""