from pathlib import Path

from cryptography.fernet import Fernet


class CryptoManager:
    def __init__(self, key_path: Path) -> None:
        self.key_path = key_path

    def _load_or_create_key(self) -> bytes:
        if self.key_path.exists():
            return self.key_path.read_bytes().strip()

        self.key_path.parent.mkdir(parents=True, exist_ok=True)
        key = Fernet.generate_key()
        self.key_path.write_bytes(key)
        return key

    def _fernet(self) -> Fernet:
        return Fernet(self._load_or_create_key())

    def encrypt_text(self, value: str) -> str:
        token = self._fernet().encrypt(value.encode("utf-8"))
        return token.decode("utf-8")

    def decrypt_text(self, token: str) -> str:
        value = self._fernet().decrypt(token.encode("utf-8"))
        return value.decode("utf-8")