import os
from cryptography.fernet import Fernet

APP_KEY = os.getenv("APP_KEY")
API_KEY = os.getenv("API_KEY")
SESSION_KEY = os.getenv("SESSION_KEY")

if not APP_KEY or not API_KEY or not SESSION_KEY:
    raise ValueError("As variáveis APP_KEY, API_KEY e SESSION_KEY devem ser definidas.")

fernet = Fernet(APP_KEY.encode())

class EncryptionService:
    @staticmethod
    def encrypt_data(data: str) -> str: return fernet.encrypt(data.encode()).decode()
    @staticmethod
    def decrypt_data(encrypted_data: str) -> str: return fernet.decrypt(encrypted_data.encode()).decode()