"""AuroraDSQLBackend class with methods for supported APIs."""

from collections import OrderedDict

from moto.core.base_backend import BackendDict, BaseBackend
from moto.core.common_models import BaseModel
from moto.core.utils import utcnow
from moto.moto_api._internal import mock_random
from moto.moto_api._internal.managed_state_model import ManagedState
from moto.utilities.utils import get_partition

from .exceptions import ResourceNotFoundException


class Cluster(BaseModel, ManagedState):
    """Model for an AuroraDSQL cluster."""

    def __init__(
        self,
        region_name: str,
        account_id: str,
        deletion_protection_enabled: bool | None,
        tags: dict[str, str] | None,
        client_token: str | None,
    ):
        ManagedState.__init__(
            self, "dsql::cluster", transitions=[("CREATING", "ACTIVE")]
        )
        self.region_name = region_name
        self.account_id = account_id
        self.identifier = mock_random.get_random_hex(26)
        self.arn = f"arn:{get_partition(self.region_name)}:dsql:{self.region_name}:{self.account_id}:cluster/{self.identifier}"
        self.creation_time = utcnow()
        self.deletion_protection_enabled = deletion_protection_enabled
        self.tags = tags
        self.client_token = client_token
        self.endpoint = f"{self.identifier}.{self.region_name}.on.aws"
        self.endpoint_service_name = f"com.amazonaws.{self.region_name}.dsql-7cwu"
        self.encryption_details = {
            "encryptionStatus": "ENABLED",
            "encryptionType": "AWS_OWNED_KMS_KEY",
        }


class AuroraDSQLBackend(BaseBackend):
    """Implementation of AuroraDSQL APIs."""

    def __init__(self, region_name: str, account_id: str):
        super().__init__(region_name, account_id)
        self.region_name = region_name
        self.account_id = account_id
        self.partition = get_partition(region_name)
        self.clusters: dict[str, Cluster] = OrderedDict()

    def create_cluster(
        self,
        deletion_protection_enabled: bool,
        tags: dict[str, str] | None,
        client_token: str | None,
    ) -> Cluster:
        cluster = Cluster(
            self.region_name,
            self.account_id,
            deletion_protection_enabled,
            tags,
            client_token,
        )
        self.clusters[cluster.identifier] = cluster
        return cluster

    def delete_cluster(self, identifier: str) -> Cluster:
        if identifier not in self.clusters:
            arn = f"arn:{get_partition(self.region_name)}:dsql:{self.region_name}:{self.account_id}:cluster/{identifier}"
            raise ResourceNotFoundException(arn, identifier, "cluster")
        cluster = self.clusters.pop(identifier)
        return cluster

    def get_cluster(self, identifier: str) -> Cluster:
        if identifier not in self.clusters:
            arn = f"arn:{get_partition(self.region_name)}:dsql:{self.region_name}:{self.account_id}:cluster/{identifier}"
            raise ResourceNotFoundException(arn, identifier, "cluster")
        cluster = self.clusters[identifier]
        cluster.advance()
        return cluster

    def get_vpc_endpoint_service_name(self, identifier: str) -> dict[str, str]:
        cluster = self.get_cluster(identifier=identifier)
        return {
            "serviceName": cluster.endpoint_service_name,
            "clusterVpcEndpoint": cluster.endpoint,
        }

    def list_tags_for_resource(self, identifier: str) -> dict[str, str]:
        cluster = self.get_cluster(identifier=identifier)
        return cluster.tags or {}


dsql_backends = BackendDict(
    AuroraDSQLBackend,
    "dsql",
    # currently botocore does not provide a dsql endpoint
    # https://github.com/boto/botocore/blob/e07cddc333fe4fb90efcd5d04324dd83f9cc3a57/botocore/data/endpoints.json
    use_boto3_regions=False,
    additional_regions=["us-east-1", "us-east-2"],
)
