Tool-Plaid

"""Encryption utilities for token storage""" from cryptography.fernet import Fernet import base64 import logging import hashlib logger = logging.getLogger(__name__) class Encryptor: """AES-256 encryption using Fernet symmetric encryption.""" def __init__(self, key: str): """ Initialize encryptor with encryption key. Args: key: At least 32-byte encryption key (will be hashed to create Fernet key) """ if len(key) < 32: raise ValueError("Encryption key must be at least 32 bytes") # Derive a 32-byte Fernet key from the input key try: key_bytes = key.encode() hash_obj = hashlib.sha256(key_bytes) fernet_key = base64.urlsafe_b64encode(hash_obj.digest()) self.fernet = Fernet(fernet_key) logger.info("Encryptor initialized") except Exception as e: logger.error(f"Failed to initialize Fernet: {e}") raise def encrypt(self, data: str) -> str: """ Encrypt string data. Args: data: Plain text string to encrypt Returns: Base64 encoded encrypted string """ try: encrypted = self.fernet.encrypt(data.encode()) return base64.urlsafe_b64encode(encrypted).decode() except Exception as e: logger.error(f"Encryption failed: {e}") raise def decrypt(self, encrypted_data: str) -> str: """ Decrypt string data. Args: encrypted_data: Base64 encoded encrypted string Returns: Decrypted plain text string """ try: encrypted = base64.urlsafe_b64decode(encrypted_data.encode()) decrypted = self.fernet.decrypt(encrypted) return decrypted.decode() except Exception as e: logger.error(f"Decryption failed: {e}") raise