# SPDX-License-Identifier: MIT
# Copyright (c) 2016, Elaborated Networks GmbH
# Copyright (c) 2019, Mathias Laurin

"""The cipher package provide symmetric encryption and decryption.

The API follows the recommendations from PEP 272 "API for Block
Encryption Algorithms"

"""

from __future__ import annotations

from typing import Final, Optional, Protocol, runtime_checkable

from mbedtls.cipher import (
    AES,
    ARC4,
    ARIA,
    CHACHA20,
    DES,
    DES3,
    Blowfish,
    Camellia,
    DES3dbl,
)
from mbedtls.cipher._cipher import AEADCipher as AEADCipher
from mbedtls.cipher._cipher import Cipher as Cipher
from mbedtls.cipher._cipher import Mode as Mode
from mbedtls.cipher._cipher import (
    get_supported_ciphers as get_supported_ciphers,
)

# Add module-level aliases to comply with PEP 272.
MODE_ECB: Final = Mode.ECB.value
MODE_CBC: Final = Mode.CBC.value
MODE_CFB: Final = Mode.CFB.value
MODE_OFB: Final = Mode.OFB.value
MODE_CTR: Final = Mode.CTR.value
MODE_GCM: Final = Mode.GCM.value
MODE_STREAM: Final = Mode.STREAM.value
MODE_CCM: Final = Mode.CCM.value
MODE_XTS: Final = Mode.XTS.value
MODE_CHACHAPOLY: Final = Mode.CHACHAPOLY.value


@runtime_checkable
class CipherType(Protocol):
    @property
    def __name__(self) -> str: ...

    @property
    def block_size(self) -> int: ...

    @property
    def key_size(self) -> int: ...

    def new(self, key: bytes, mode: Mode, iv: Optional[bytes]) -> Cipher: ...


@runtime_checkable
class AEADCipherType(Protocol):
    @property
    def __name__(self) -> str: ...

    @property
    def block_size(self) -> int: ...

    @property
    def key_size(self) -> int: ...

    def new(
        self, key: bytes, mode: Mode, iv: Optional[bytes], ad: Optional[bytes]
    ) -> AEADCipher: ...


__all__ = (
    "AEADCipherType",
    "AES",
    "ARC4",
    "ARIA",
    "Blowfish",
    "Camellia",
    "DES",
    "DES3",
    "DES3dbl",
    "CHACHA20",
    "CipherType",
    "get_supported_ciphers",
    "Mode",
    "MODE_ECB",
    "MODE_CBC",
    "MODE_CFB",
    "MODE_OFB",
    "MODE_CTR",
    "MODE_GCM",
    "MODE_STREAM",
    "MODE_CCM",
    "MODE_XTS",
    "MODE_CHACHAPOLY",
)