"""Shared script schema for writers, renderers, mixers, and UI."""

from __future__ import annotations

from dataclasses import dataclass, field
from enum import StrEnum
from typing import Any


class Genre(StrEnum):
    NOIR = "noir"
    WEIRD = "weird"
    WESTERN = "western"
    ROMANCE = "romance"
    COMEDY = "comedy"
    HINDI_MELODRAMA = "hindi_melodrama"


class Delivery(StrEnum):
    NEUTRAL = "neutral"
    SLOW = "slow"
    URGENT = "urgent"
    WHISPER = "whisper"
    BOOMING = "booming"
    DEADPAN = "deadpan"
    AGITATED = "agitated"


class VoiceID(StrEnum):
    """Locked Kokoro voice roster verified against hexgrad/Kokoro-82M."""

    ANNOUNCER = "am_michael"
    DETECTIVE = "am_onyx"
    INGENUE = "af_bella"
    COMIC = "am_puck"
    VILLAIN = "am_fenrir"
    ELDER = "bm_george"
    HINDI_FEMALE = "hf_alpha"
    HINDI_MALE = "hm_omega"


@dataclass(frozen=True)
class CastMember:
    id: str
    name: str
    voice: VoiceID
    description: str


@dataclass(frozen=True)
class Line:
    cast: str
    text: str
    delivery: Delivery = Delivery.NEUTRAL


@dataclass(frozen=True)
class Scene:
    title: str
    sfx: list[str]
    lines: list[Line]


@dataclass(frozen=True)
class MusicPlan:
    genre: Genre
    opening_sting: str
    bed: str
    closing_sting: str


@dataclass(frozen=True)
class Script:
    title: str
    logline: str
    genre: Genre
    cast: list[CastMember]
    scenes: list[Scene]
    music: MusicPlan
    estimated_seconds: int = field(default=75)

    def validate(self) -> None:
        if not self.title.strip():
            raise ValueError("script title is required")
        if not self.logline.strip():
            raise ValueError("script logline is required")
        if not 30 <= self.estimated_seconds <= 120:
            raise ValueError("estimated_seconds must be between 30 and 120")
        if len(self.cast) < 2:
            raise ValueError("script must define at least two cast members")
        cast_ids = {member.id for member in self.cast}
        if len(cast_ids) != len(self.cast):
            raise ValueError("cast member ids must be unique")
        if not self.scenes:
            raise ValueError("script must contain at least one scene")
        for scene in self.scenes:
            if not scene.lines:
                raise ValueError(f"scene '{scene.title}' must contain at least one line")
            for line in scene.lines:
                if line.cast not in cast_ids:
                    raise ValueError(f"line references unknown cast id '{line.cast}'")
                if not line.text.strip():
                    raise ValueError("line text cannot be empty")
        if self.music.genre != self.genre:
            raise ValueError("music genre must match script genre")


def validate_script(script: Script) -> Script:
    script.validate()
    return script


def script_from_json_dict(payload: dict[str, Any]) -> Script:
    """Build and validate a Script from a JSON-compatible dict."""
    script = Script(
        title=str(payload["title"]),
        logline=str(payload["logline"]),
        genre=Genre(payload["genre"]),
        cast=[
            CastMember(
                id=str(member["id"]),
                name=str(member["name"]),
                voice=VoiceID(member["voice"]),
                description=str(member.get("description", "")),
            )
            for member in payload["cast"]
        ],
        scenes=[
            Scene(
                title=str(scene["title"]),
                sfx=[str(item) for item in scene.get("sfx", [])],
                lines=[
                    Line(
                        cast=str(line["cast"]),
                        text=str(line["text"]),
                        delivery=Delivery(line.get("delivery", Delivery.NEUTRAL)),
                    )
                    for line in scene["lines"]
                ],
            )
            for scene in payload["scenes"]
        ],
        music=MusicPlan(
            genre=Genre(payload["music"]["genre"]),
            opening_sting=str(payload["music"]["opening_sting"]),
            bed=str(payload["music"]["bed"]),
            closing_sting=str(payload["music"]["closing_sting"]),
        ),
        estimated_seconds=int(payload.get("estimated_seconds", 75)),
    )
    script.validate()
    return script