from dataclasses import dataclass

from anydi import singleton
from fansifter_common.auth.requests import AuthRequest
from fansifter_common.auth.services import AuthService
from fansifter_common.auth.types import Permission

from dmp.adapters.db import DefaultDB, transactional
from dmp.audiences.enums import AudienceExportStatus
from dmp.google.exceptions import GoogleAudienceNotFoundError
from dmp.google.repositories import GoogleAudienceRepository


@dataclass
class GetGoogleAudienceShareStatusRequest(AuthRequest):
    google_audience_id: str


@dataclass
class GetGoogleAudienceShareStatusResponse:
    audience_id: str
    audience_name: str
    ad_account_name: str
    status: AudienceExportStatus
    fans_count: int
    processed_fans: int
    progress: float
    seconds_remaining: int | None
    url: str


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

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

    @transactional
    def handle(
        self, request: GetGoogleAudienceShareStatusRequest
    ) -> GetGoogleAudienceShareStatusResponse:
        google_audience = self.google_audience_repository.get(
            request.google_audience_id
        )

        if google_audience is None:
            raise GoogleAudienceNotFoundError

        self.auth_service.authorize_account(
            request.identity_id,
            permission=self.permission,
            account=google_audience.share.audience.account,
        )

        return GetGoogleAudienceShareStatusResponse(
            audience_id=google_audience.id,
            audience_name=google_audience.name,
            ad_account_name=google_audience.ad_account.display_name,
            status=google_audience.share.status,
            fans_count=google_audience.fans_count,
            processed_fans=google_audience.share.processed_fans,
            progress=google_audience.share.progress,
            seconds_remaining=google_audience.share.seconds_remaining,
            url=google_audience.url,
        )