import datetime
import typing
from dataclasses import dataclass
from uuid import UUID

from src import domain

from .analytics.get_creatives_analytics import get_creatives_analytics
from .analytics.get_external_channels_analytics import get_external_channels_analytics
from .analytics.get_placements_analytics import get_placements_analytics
from .analytics.get_spending_analytics import get_spending_analytics
from .creative.create_creative import create_creative
from .creative.delete_creative import delete_creative
from .creative.get_creative import get_creative
from .creative.get_creatives import get_creatives
from .creative.update_creative import update_creative
from .external_channel.create_external_channel import create_external_channel
from .external_channel.delete_external_channel import delete_external_channel
from .external_channel.get_external_channels import get_external_channels
from .external_channel.update_external_channel import update_external_channel
from .external_channel.update_external_channel_links import update_external_channel_links
from .get_me import get_me
from .placement.create_placement import create_placement
from .placement.delete_placement import delete_placement
from .placement.get_placement import get_placement
from .placement.get_placements import get_placements
from .placement.update_placement import update_placement
from .subscription.handle_subscription import handle_subscription
from .subscription.handle_unsubscription import handle_unsubscription
from .target_channel.connect_target_chan import connect_target_chan
from .target_channel.disconnect_target_chan import disconnect_target_chan
from .target_channel.disconnect_target_chan_by_tg_id import disconnect_target_chan_by_tg_id
from .target_channel.get_user_target_chans import get_user_target_chans
from .target_channel.update_target_chan_permissions import update_target_chan_permissions
from .telegram_login import telegram_login
from .validate_login_token import validate_login_token
from .views.fetch_views_manually import fetch_views_manually
from .views.get_views_history import get_views_history
from .views.run_views_worker_cycle import run_views_worker_cycle
from .views.update_views_manually import update_views_manually


class Database(typing.Protocol):
    def transaction(self) -> typing.AsyncContextManager[None]: ...

    async def get_user(self, *, user_id: UUID | None = None, telegram_id: int | None = None) -> domain.User | None: ...

    async def create_user(self, user: domain.User) -> domain.User: ...

    async def create_login_token(
        self, user_id: UUID, token: str, expires_at: datetime.datetime
    ) -> domain.LoginToken: ...

    async def get_login_token(self, token: str) -> domain.LoginToken | None: ...

    async def mark_token_as_used(self, token: str) -> None: ...

    async def get_target_channel(
        self, user_id: UUID, channel_id: UUID | None = None, telegram_id: int | None = None
    ) -> domain.TargetChannel | None: ...

    async def create_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...

    async def upsert_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...

    async def get_user_target_channels(self, user_id: UUID) -> list[domain.TargetChannel]: ...

    async def update_target_channel(self, channel: domain.TargetChannel) -> domain.TargetChannel: ...

    async def get_external_channel(
        self, user_id: UUID, channel_id: UUID | None = None, telegram_id: int | None = None
    ) -> domain.ExternalChannel | None: ...

    async def create_external_channel(self, channel: domain.ExternalChannel) -> domain.ExternalChannel: ...

    async def get_external_channels_for_target(self, target_channel_id: UUID) -> list[domain.ExternalChannel]: ...

    async def get_user_external_channels(self, user_id: UUID) -> list[domain.ExternalChannel]: ...

    async def add_external_channel_to_targets(
        self, external_channel_id: UUID, target_channel_ids: list[UUID]
    ) -> None: ...

    async def remove_external_channel_from_target(self, external_channel_id: UUID, target_channel_id: UUID) -> None: ...

    async def delete_external_channel(self, channel_id: UUID) -> None: ...

    async def update_external_channel(self, channel: domain.ExternalChannel) -> domain.ExternalChannel: ...

    async def get_creative(self, user_id: UUID, creative_id: UUID) -> domain.Creative | None: ...

    async def create_creative(self, creative: domain.Creative) -> domain.Creative: ...

    async def update_creative(self, creative: domain.Creative) -> domain.Creative: ...

    async def get_user_creatives(
        self, user_id: UUID, target_channel_id: UUID | None = None, include_archived: bool = False
    ) -> list[domain.Creative]: ...

    async def delete_creative(self, creative_id: UUID) -> None: ...

    async def get_placement(self, user_id: UUID, placement_id: UUID) -> domain.Placement | None: ...

    async def create_placement(self, placement: domain.Placement) -> domain.Placement: ...

    async def update_placement(self, placement: domain.Placement) -> domain.Placement: ...

    async def get_user_placements(
        self,
        user_id: UUID,
        target_channel_id: UUID | None = None,
        external_channel_id: UUID | None = None,
        creative_id: UUID | None = None,
        include_archived: bool = False,
    ) -> list[domain.Placement]: ...

    async def delete_placement(self, placement_id: UUID) -> None: ...

    async def check_creative_in_use(self, creative_id: UUID) -> bool: ...

    async def get_placement_by_invite_link(self, invite_link: str) -> domain.Placement | None: ...

    async def get_subscriber(self, telegram_id: int) -> domain.Subscriber | None: ...

    async def create_subscriber(self, subscriber: domain.Subscriber) -> domain.Subscriber: ...

    async def upsert_subscriber(self, subscriber: domain.Subscriber) -> domain.Subscriber: ...

    async def create_subscription(self, subscription: domain.Subscription) -> domain.Subscription: ...

    async def get_subscription_by_subscriber_and_placement(
        self, subscriber_id: UUID, placement_id: UUID
    ) -> domain.Subscription | None: ...

    async def update_subscription(self, subscription: domain.Subscription) -> domain.Subscription: ...

    async def get_active_subscriptions_by_subscriber_and_channel(
        self, subscriber_id: UUID, channel_telegram_id: int
    ) -> list[domain.Subscription]: ...

    async def get_active_subscription_by_subscriber_and_channel(
        self, subscriber_id: UUID, channel_telegram_id: int
    ) -> domain.Subscription | None: ...

    async def create_placement_views_history(
        self, history: domain.PlacementViewsHistory
    ) -> domain.PlacementViewsHistory: ...

    async def get_views_history(
        self,
        placement_id: UUID,
        *,
        from_date: datetime.datetime | None = None,
        to_date: datetime.datetime | None = None,
    ) -> list[domain.PlacementViewsHistory]: ...

    async def get_active_placements_with_urls(self) -> list[domain.Placement]: ...  # Для worker


class TelegramBotWriter(typing.Protocol):
    async def send_message(self, text: str, chat_id: int) -> None: ...

    async def send_message_with_button(self, text: str, chat_id: int, button_text: str, button_url: str) -> None: ...

    async def get_chat_member(self, chat_id: int, user_id: int) -> dict[str, typing.Any]: ...

    async def create_chat_invite_link(self, chat_id: int, requires_approval: bool = False) -> str: ...


class TelegramParser(typing.Protocol):
    async def get_post_views(self, post_url: str) -> int | None: ...


class JWTEncoder(typing.Protocol):
    def encode_access_token(self, user_id: UUID, telegram_id: int, username: str | None = None) -> str: ...


@dataclass
class Usecase:
    database: Database
    telegram_bot: TelegramBotWriter
    telegram_parser: TelegramParser
    jwt_encoder: JWTEncoder

    validate_login_token = validate_login_token
    telegram_login = telegram_login
    get_me = get_me
    connect_target_chan = connect_target_chan
    get_user_target_chans = get_user_target_chans
    disconnect_target_chan = disconnect_target_chan
    disconnect_target_chan_by_tg_id = disconnect_target_chan_by_tg_id
    update_target_chan_permissions = update_target_chan_permissions
    get_external_channels = get_external_channels
    create_external_channel = create_external_channel
    delete_external_channel = delete_external_channel
    update_external_channel = update_external_channel
    update_external_channel_links = update_external_channel_links
    get_creatives = get_creatives
    get_creative = get_creative
    create_creative = create_creative
    update_creative = update_creative
    delete_creative = delete_creative
    get_placements = get_placements
    get_placement = get_placement
    create_placement = create_placement
    update_placement = update_placement
    delete_placement = delete_placement
    handle_subscription = handle_subscription
    handle_unsubscription = handle_unsubscription
    fetch_views = fetch_views_manually
    run_views_worker_cycle = run_views_worker_cycle
    get_views_history = get_views_history
    update_views_manually = update_views_manually
    get_placements_analytics = get_placements_analytics
    get_creatives_analytics = get_creatives_analytics
    get_external_channels_analytics = get_external_channels_analytics
    get_spending_analytics = get_spending_analytics