from dataclasses import dataclass

from anydi import singleton

from dmp.adapters.db import DefaultDB, ReportingDB, transactional
from dmp.adapters.fivetran import FivetranClient
from dmp.app_connections.enums import AppConnectionStatus
from dmp.google.exceptions import GoogleAdReportingConnectionNotFoundError
from dmp.google.models import GoogleAdReportingConnection
from dmp.google.repositories import GoogleAdReportingConnectionRepository
from dmp.google.services import GoogleAdReportingConnectionService


@dataclass
class RefreshGoogleAdReportingConnectionRequest:
    fivetran_connection_id: str


@singleton
class RefreshGoogleAdReportingConnectionHandler:
    def __init__(
        self,
        db: DefaultDB,
        reporting_db: ReportingDB,
        fivetran_client: FivetranClient,
        ad_reporting_connection_repository: GoogleAdReportingConnectionRepository,
        ad_reporting_connection_service: GoogleAdReportingConnectionService,
    ) -> None:
        self.db = db
        self.reporting_db = reporting_db
        self.fivetran_client = fivetran_client
        self.ad_reporting_connection_repository = ad_reporting_connection_repository
        self.ad_reporting_connection_service = ad_reporting_connection_service

    @transactional
    def handle(
        self, request: RefreshGoogleAdReportingConnectionRequest
    ) -> GoogleAdReportingConnection:
        connection = (
            self.ad_reporting_connection_repository.get_by_fivetran_connection_id(
                fivetran_connection_id=request.fivetran_connection_id
            )
        )

        if not connection or connection.status == AppConnectionStatus.DELETED:
            raise GoogleAdReportingConnectionNotFoundError

        connection = self.ad_reporting_connection_service.refresh_connection(connection)

        return connection