from collections.abc import Sequence

from anydi import singleton

from dmp.tiktok.dtos import AssignTikTokUserAdAccountLabelData
from dmp.tiktok.models import TikTokUserAdAccount
from dmp.tiktok.repositories import TikTokUserAdAccountRepository


@singleton
class TikTokUserAdAccountService:
    def __init__(
        self,
        user_ad_account_repository: TikTokUserAdAccountRepository,
    ) -> None:
        self.user_ad_account_repository = user_ad_account_repository

    def create_or_update_user_ad_accounts_for_user(
        self, assignments: list[AssignTikTokUserAdAccountLabelData], identity_id: str
    ) -> list[TikTokUserAdAccount]:
        existed_user_ad_accounts = (
            self.user_ad_account_repository.find_by_ad_accounts_ids_and_identity_id(
                ad_accounts_ids=[
                    assignment.ad_account_id for assignment in assignments
                ],
                identity_id=identity_id,
            )
        )
        existed_user_ad_accounts_by_id = {
            existed_user_ad_account.ad_account_id: existed_user_ad_account
            for existed_user_ad_account in existed_user_ad_accounts
        }
        user_ad_accounts = []
        for update_assignment in assignments:
            if update_assignment.ad_account_id in existed_user_ad_accounts_by_id.keys():
                existed_user_ad_account = existed_user_ad_accounts_by_id[
                    update_assignment.ad_account_id
                ]
                existed_user_ad_account.vendor_id = update_assignment.vendor_id
                existed_user_ad_account.subaccount_id = update_assignment.subaccount_id
                user_ad_accounts.append(existed_user_ad_account)
            else:
                new_user_ad_account = TikTokUserAdAccount(
                    ad_account_id=update_assignment.ad_account_id,
                    identity_id=identity_id,
                    vendor_id=update_assignment.vendor_id,
                    subaccount_id=update_assignment.subaccount_id,
                )
                user_ad_accounts.append(new_user_ad_account)

        self.user_ad_account_repository.add_all(user_ad_accounts)

        return user_ad_accounts

    def get_labels_ad_accounts(
        self, vendor_ids: list[int], subaccount_ids: list[int]
    ) -> Sequence[TikTokUserAdAccount]:
        return self.user_ad_account_repository.find_by_vendor_and_subaccount_ids(
            vendor_ids=vendor_ids, subaccount_ids=subaccount_ids
        )

    def get_ad_accounts(
        self, ad_accounts_ids: set[str]
    ) -> Sequence[TikTokUserAdAccount]:
        return self.user_ad_account_repository.find_by_ad_accounts_external_ids(
            ad_accounts_ids=ad_accounts_ids
        )