from typing import Any, Self

from fansifter_common.adapters.aws.secretsmanager import SecretsManager
from fansifter_common.constants import PROD_ENVIRONMENT, QA_ENVIRONMENT
from fansifter_common.logging import LogFormat
from fansifter_common.logging.utils import update_logging_config
from pydantic import (
    Field,
    ValidationError,
    model_validator,
)
from pydantic_settings import BaseSettings, SettingsConfigDict
from sqlalchemy import URL

from app.exceptions import ImproperlyConfigured


class Settings(BaseSettings):
    environment: str
    debug: bool = Field(validation_alias="APP_DEBUG", default=False)

    aws_region_name: str = "us-east-1"

    service_name: str = "lambda-audience-test-audiences-cleanup"
    service_version: str = "0.1.0"

    # KMS
    kms_key_id: str = Field(default="", validation_alias="DMP_KMS_KEY_ID")

    # Postgres
    postgres_host: str = "localhost"
    postgres_port: int = 5432
    postgres_name: str = "dmp"
    postgres_username: str = "postgres"

    postgres_password: str = "postgres"
    postgres_password_secret_name: str = ""

    ssl_mode: bool = True

    # API urls
    meta_api_url: str = "https://graph.facebook.com/v21.0"
    tiktok_api_url: str = "https://business-api.tiktok.com/open_api/v1.3"

    @property
    def postgres_url(self) -> URL:
        if self.environment in [QA_ENVIRONMENT, PROD_ENVIRONMENT]:
            secrets_manager = SecretsManager(region_name=self.aws_region_name)
            postgres_password = secrets_manager.get_secret(
                self.postgres_password_secret_name
            )
        else:
            postgres_password = self.postgres_password

        return URL.create(
            drivername="postgresql",
            username=self.postgres_username,
            password=postgres_password,
            host=self.postgres_host,
            port=self.postgres_port,
            database=self.postgres_name,
        )

    # Logging
    log_format: LogFormat = "json"

    logging_config: dict[str, Any] = {
        "version": 1,
        "disable_existing_loggers": True,
        "handlers": {
            "stream": {
                "level": "INFO",
                "class": "fansifter_common.logging.OwsStreamHandler",
            },
        },
        "loggers": {
            "app": {
                "handlers": ["stream"],
                "level": "INFO",
                "propagate": False,
            },
            "httpx": {
                "handlers": ["stream"],
                "level": "WARNING",
                "propagate": False,
            },
            "owsclient": {
                "handlers": ["stream"],
                "level": "WARNING",
                "propagate": False,
            },
        },
    }

    @model_validator(mode="after")
    def _setup_logging_config(self) -> Self:
        self.logging_config = update_logging_config(
            logging_config=self.logging_config,
            environment=self.environment,
            service_name=self.service_name,
            service_version=self.service_version,
            log_format=self.log_format,
            debug=self.debug,
        )
        return self

    # Sentry
    sentry_dsn: str | None = None

    @model_validator(mode="after")
    def _setup_sentry_dsn(self) -> Self:
        if not self.sentry_dsn and self.environment in [
            QA_ENVIRONMENT,
            PROD_ENVIRONMENT,
        ]:
            raise ValueError(f"Sentry is not configured in {self.environment}.")
        return self

    model_config = SettingsConfigDict(
        env_file=".env",
        env_file_encoding="utf-8",
        case_sensitive=False,
        extra="allow",
    )


def initialize_settings(**defaults: Any) -> Settings:
    try:
        return Settings(**defaults)
    except ValidationError as exc:
        errors = "\n".join([str(e) for e in exc.errors()])
        raise ImproperlyConfigured(
            f"Failed to initialize settings:\n {errors}"
        ) from exc