from dataclasses import dataclass
from typing import Any

from anydi import singleton
from fansifter_common.auth.services import AuthService
from fansifter_common.auth.types import Permission
from pydantic import TypeAdapter

from dmp.adapters.db import DefaultDB
from dmp.app_connections.enums import AppConnectionStatus
from dmp.tiktok.dtos import TikTokAdAccountOverview
from dmp.tiktok.enums import TikTokUserConnectionStatus
from dmp.tiktok.services import TikTokAdAccountService


@dataclass(frozen=True)
class GetTikTokAdAccountsRequest:
    identity_id: str


TikTokAdAccountOverviewListValidator = TypeAdapter(list[TikTokAdAccountOverview])


@singleton
class GetTikTokAdAccountsHandler:
    permission = Permission("ad_account", "view")

    def __init__(
        self,
        db: DefaultDB,
        auth_service: AuthService,
        ad_account_service: TikTokAdAccountService,
    ) -> None:
        self.db = db
        self.auth_service = auth_service
        self.ad_account_service = ad_account_service

    def handle(
        self, request: GetTikTokAdAccountsRequest
    ) -> list[TikTokAdAccountOverview]:
        account_access = self.auth_service.authorize_for_permission(
            request.identity_id, permission=self.permission
        )
        return self._get_overview(
            identity_id=request.identity_id,
            vendor_ids=account_access.vendor_ids,
            subaccount_ids=account_access.subaccount_ids,
        )

    def _get_overview(
        self,
        identity_id: str,
        vendor_ids: list[int],
        subaccount_ids: list[int],
    ) -> list[TikTokAdAccountOverview]:
        query_context: dict[str, Any] = {
            "vendor_ids": vendor_ids,
            "subaccount_ids": subaccount_ids,
            "identity_id": identity_id,
            "ordered_reporting_enum_type": AppConnectionStatus.ordered(),
            "ordered_sharing_enum_type": TikTokUserConnectionStatus.ordered(),
        }
        query = self.db.query_from_template(
            "tiktok-ad-account/find-by-account.sql",
            context=query_context,
        )
        ad_accounts = TikTokAdAccountOverviewListValidator.validate_python(
            self.db.session.execute(query).mappings(),
        )
        return list(self.ad_account_service.fill_ad_account_empty_data(ad_accounts))