"""Shared helpers for turning ORM rows into response schemas."""

from __future__ import annotations

from app.core.ids import aurora_id
from app.models.artwork import Artwork
from app.models.credit import Credit
from app.models.enums import ArtworkType
from app.models.video import Video
from app.schemas.common import (
    ArtworkOut,
    CastCreditOut,
    CrewCreditOut,
    ImagesOut,
    PersonBriefOut,
    VideoOut,
)


def artwork_url(artwork: Artwork) -> str:
    return artwork.local_path or artwork.original_url


def to_artwork_out(artwork: Artwork) -> ArtworkOut:
    return ArtworkOut(
        id=aurora_id("artwork", str(artwork.id)),
        type=artwork.artwork_type.value,
        language=artwork.language_code,
        width=artwork.width,
        height=artwork.height,
        aspect_ratio=artwork.aspect_ratio,
        url=artwork_url(artwork),
        provider=artwork.provider.slug if artwork.provider else None,
        is_primary=artwork.is_primary,
        vote_average=artwork.vote_average,
    )


def to_images_out(artwork_list: list[Artwork]) -> ImagesOut:
    images = ImagesOut()
    for artwork in artwork_list:
        out = to_artwork_out(artwork)
        if artwork.artwork_type == ArtworkType.POSTER:
            images.posters.append(out)
        elif artwork.artwork_type == ArtworkType.BACKDROP:
            images.backdrops.append(out)
        elif artwork.artwork_type == ArtworkType.LOGO:
            images.logos.append(out)
    return images


def primary_artwork_path(artwork_list: list[Artwork], artwork_type: ArtworkType) -> str | None:
    for artwork in artwork_list:
        if artwork.artwork_type == artwork_type:
            return artwork_url(artwork)
    return None


def to_video_out(video: Video) -> VideoOut:
    return VideoOut(
        id=aurora_id("video", str(video.id)),
        type=video.video_type.value,
        site=video.site,
        url=video.url,
        language=video.language_code,
        country=video.country_code,
        is_official=video.is_official,
        published_at=video.published_at,
        provider=video.provider.slug if video.provider else None,
    )


def split_credits(
    credits_: list[Credit], profile_paths: dict | None = None
) -> tuple[list[CastCreditOut], list[CrewCreditOut]]:
    profile_paths = profile_paths or {}
    cast: list[CastCreditOut] = []
    crew: list[CrewCreditOut] = []
    for credit in credits_:
        person_brief = PersonBriefOut(
            id=aurora_id("person", str(credit.person_id)),
            name=credit.person.name,
            profile_path=profile_paths.get(credit.person_id),
        )
        if credit.department == "Acting":
            cast.append(
                CastCreditOut(
                    id=aurora_id("credit", str(credit.id)),
                    person=person_brief,
                    character=credit.character,
                    billing_order=credit.billing_order,
                    episode_count=credit.episode_count,
                )
            )
        else:
            crew.append(
                CrewCreditOut(
                    id=aurora_id("credit", str(credit.id)),
                    person=person_brief,
                    department=credit.department,
                    job=credit.job,
                    episode_count=credit.episode_count,
                )
            )
    return cast, crew