from fansifter_common.adapters.encrypter import EncryptedStr, Encrypter

from ows_text_campaigns.adapters.aws.kms import KMSClient


class KMSEncrypter(Encrypter):
    def __init__(self, kms_client: KMSClient) -> None:
        self.kms_client = kms_client

    def encrypt(self, value: str) -> EncryptedStr:
        return self.kms_client.encrypt(value)

    def decrypt(self, value: EncryptedStr) -> str:
        return self.kms_client.decrypt(value)