from __future__ import annotations

import logging
from dataclasses import dataclass
from typing import TYPE_CHECKING, Any, cast

from django.db.models import Count, Prefetch, Q
from django.db.models.query import QuerySet
from django.shortcuts import render
from django.utils import timezone
from django.views.generic import DetailView, ListView

from twitch.models import DropCampaign, Game, Organization, TimeBasedDrop

if TYPE_CHECKING:
    import datetime

    from django.db.models import QuerySet
    from django.http import HttpRequest, HttpResponse

logger: logging.Logger = logging.getLogger(__name__)


class DropCampaignListView(ListView):
    """List view for drop campaigns."""

    model = DropCampaign
    template_name = "twitch/campaign_list.html"
    context_object_name = "campaigns"
    paginate_by = 100

    def get_queryset(self) -> QuerySet[DropCampaign]:
        """Get queryset of drop campaigns.

        Returns:
            QuerySet: Filtered drop campaigns.
        """
        queryset: QuerySet[DropCampaign] = super().get_queryset()
        game_filter: str | None = self.request.GET.get("game")

        if game_filter:
            queryset = queryset.filter(game__id=game_filter)

        return queryset.select_related("game", "owner").order_by("-start_at")

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        """Add additional context data.

        Args:
            **kwargs: Additional arguments.

        Returns:
            dict: Context data.
        """
        kwargs = cast("dict[str, Any]", kwargs)
        context: dict[str, datetime.datetime | str | int | QuerySet[Game, Game] | None] = super().get_context_data(**kwargs)

        context["games"] = Game.objects.all().order_by("display_name")

        context["now"] = timezone.now()
        context["selected_game"] = str(self.request.GET.get(key="game", default=""))
        context["selected_per_page"] = self.paginate_by
        context["selected_status"] = self.request.GET.get(key="status", default="")

        return context


class DropCampaignDetailView(DetailView):
    """Detail view for a drop campaign."""

    model = DropCampaign
    template_name = "twitch/campaign_detail.html"
    context_object_name = "campaign"

    def get_object(self, queryset: QuerySet[DropCampaign] | None = None) -> DropCampaign:
        """Get the campaign object with related data prefetched.

        Args:
            queryset: Optional queryset to use.

        Returns:
            DropCampaign: The campaign object with prefetched relations.
        """
        if queryset is None:
            queryset = self.get_queryset()

        queryset = queryset.select_related("game", "owner")

        return super().get_object(queryset=queryset)

    def get_context_data(self, **kwargs: object) -> dict[str, Any]:
        """Add additional context data.

        Args:
            **kwargs: Additional arguments.

        Returns:
            dict: Context data.
        """
        context: dict[str, Any] = super().get_context_data(**kwargs)
        campaign = context["campaign"]

        context["now"] = timezone.now()
        context["drops"] = (
            TimeBasedDrop.objects.filter(campaign=campaign)
            .select_related("campaign")
            .prefetch_related("benefits")
            .order_by("required_minutes_watched")
        )

        return context


class GameListView(ListView):
    """List view for games grouped by organization."""

    model = Game
    template_name = "twitch/game_list.html"
    context_object_name = "games"

    def get_queryset(self) -> QuerySet[Game]:
        """Get queryset of games, annotated with campaign counts to avoid N+1 queries.

        Returns:
            QuerySet[Game]: Queryset of games with annotations.
        """
        now = timezone.now()
        return (
            super()
            .get_queryset()
            .annotate(
                campaign_count=Count("drop_campaigns", distinct=True),
                active_count=Count(
                    "drop_campaigns",
                    filter=Q(
                        drop_campaigns__start_at__lte=now,
                        drop_campaigns__end_at__gte=now,
                    ),
                    distinct=True,
                ),
            )
            .order_by("display_name")
        )

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        """Add additional context data with games grouped by organization.

        Args:
            **kwargs: Additional keyword arguments.

        Returns:
            dict: Context data with games grouped by organization.
        """

        @dataclass(frozen=True)
        class OrganizationData:
            id: str
            name: str

        context: dict[str, Any] = super().get_context_data(**kwargs)

        games_by_org: dict[OrganizationData, list[dict[str, Game | dict[str, int]]]] = {}
        now: datetime.datetime = timezone.now()

        organizations_with_games: QuerySet[Organization, Organization] = (
            Organization.objects.filter(drop_campaigns__isnull=False).distinct().order_by("name")
        )

        game_org_relations: QuerySet[DropCampaign, dict[str, Any]] = DropCampaign.objects.values(
            "game_id", "owner_id", "owner__name"
        ).annotate(
            campaign_count=Count("id", distinct=True),
            active_count=Count("id", filter=Q(start_at__lte=now, end_at__gte=now), distinct=True),
        )

        all_games: dict[str, Game] = {game.id: game for game in Game.objects.all()}
        org_names: dict[str, str] = {org.id: org.name for org in organizations_with_games}

        game_org_map: dict[str, dict[str, Any]] = {}
        for relation in game_org_relations:
            org_id: str = relation["owner_id"]
            game_id: str = relation["game_id"]

            if org_id not in game_org_map:
                game_org_map[org_id] = {}

            if game_id not in game_org_map[org_id]:
                game: Game | None = all_games.get(game_id)
                if game:
                    game_org_map[org_id][game_id] = {
                        "game": game,
                        "campaign_count": relation["campaign_count"],
                        "active_count": relation["active_count"],
                    }

        for org_id, games in game_org_map.items():
            if org_id in org_names:
                org_obj = OrganizationData(id=org_id, name=org_names[org_id])
                games_by_org[org_obj] = list(games.values())

        games_with_counts: list[dict[str, Game | dict[str, int]]] = []

        for org_games in games_by_org.values():
            games_with_counts.extend(org_games)

        context["games_with_counts"] = games_with_counts
        context["games_by_org"] = games_by_org

        return context


class GameDetailView(DetailView):
    """Detail view for a game."""

    model = Game
    template_name = "twitch/game_detail.html"
    context_object_name = "game"

    def get_context_data(self, **kwargs: object) -> dict[str, Any]:
        """Add additional context data.

        Args:
            **kwargs: Additional arguments.

        Returns:
            dict: Context data with active, upcoming, and expired campaigns.
                 Expired campaigns are filtered based on either end date or status.
        """
        context: dict[str, Any] = super().get_context_data(**kwargs)
        game: Game = self.get_object()

        now: datetime.datetime = timezone.now()
        all_campaigns: QuerySet[DropCampaign, DropCampaign] = (
            DropCampaign.objects.filter(game=game).select_related("owner").order_by("-end_at")
        )

        active_campaigns: list[DropCampaign] = [
            campaign for campaign in all_campaigns if campaign.start_at <= now and campaign.end_at >= now
        ]
        active_campaigns.sort(key=lambda c: c.end_at)

        upcoming_campaigns: list[DropCampaign] = [campaign for campaign in all_campaigns if campaign.start_at > now]
        upcoming_campaigns.sort(key=lambda c: c.start_at)

        expired_campaigns: list[DropCampaign] = [campaign for campaign in all_campaigns if campaign.end_at < now]

        context.update({
            "active_campaigns": active_campaigns,
            "upcoming_campaigns": upcoming_campaigns,
            "expired_campaigns": expired_campaigns,
            "now": now,
        })

        return context


def dashboard(request: HttpRequest) -> HttpResponse:
    """Dashboard view showing active campaigns and progress.

    Args:
        request: The HTTP request.

    Returns:
        HttpResponse: The rendered dashboard template.
    """
    now: datetime.datetime = timezone.now()
    active_campaigns: QuerySet[DropCampaign, DropCampaign] = (
        DropCampaign.objects.filter(start_at__lte=now, end_at__gte=now)
        .select_related("game", "owner")
        .prefetch_related(Prefetch("time_based_drops", queryset=TimeBasedDrop.objects.prefetch_related("benefits")))
    )

    campaigns_by_org_game: dict[str, Any] = {}

    for campaign in active_campaigns:
        org_id: str = campaign.owner.id
        org_name: str = campaign.owner.name
        game_id: str = campaign.game.id
        game_name: str = campaign.game.display_name

        if org_id not in campaigns_by_org_game:
            campaigns_by_org_game[org_id] = {"name": org_name, "games": {}}

        if game_id not in campaigns_by_org_game[org_id]["games"]:
            campaigns_by_org_game[org_id]["games"][game_id] = {"name": game_name, "campaigns": []}

        campaigns_by_org_game[org_id]["games"][game_id]["campaigns"].append(campaign)

    sorted_campaigns_by_org_game: dict[str, Any] = {
        org_id: campaigns_by_org_game[org_id]
        for org_id in sorted(campaigns_by_org_game.keys(), key=lambda k: campaigns_by_org_game[k]["name"])
    }

    for org_data in sorted_campaigns_by_org_game.values():
        org_data["games"] = {
            game_id: org_data["games"][game_id] for game_id in sorted(org_data["games"].keys(), key=lambda k: org_data["games"][k]["name"])
        }

    return render(
        request,
        "twitch/dashboard.html",
        {
            "active_campaigns": active_campaigns,
            "campaigns_by_org_game": sorted_campaigns_by_org_game,
            "now": now,
        },
    )