#!/usr/bin/env python3
import os
import sys
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from cryptography.fernet import Fernet
import config_i

# Encryption helper
def encrypt(data: str) -> str:
    """Encrypts any string with Fernet encryption"""
    fernet = Fernet(config_i.config.key.encode())
    return fernet.encrypt(data.encode()).decode()

def decrypt(encrypted_text: str) -> str:
    """ Decrypts any string with Fernet encryption"""
    key = config_i.config.key.encode()

    # Ensure correct padding for base64 key
    padded_key = key + b'=' * (-len(key) % 4)
    fernet = Fernet(padded_key)

    decrypted_text = fernet.decrypt(encrypted_text.encode())
    return decrypted_text.decode()