from __future__ import annotations

from dataclasses import dataclass


@dataclass(frozen=True)
class CaptionTemplate:
    key: str
    label: str
    font_size: int
    primary_color: str
    secondary_color: str
    outline_color: str = "&H000000"
    back_color: str = "&H80000000"
    alignment: int = 2
    margin_v: int = 180
    bold: bool = True
    effect: str = "none"

    def ass_style(self) -> str:
        bold = -1 if self.bold else 0
        return (
            "Style: Default,DejaVu Sans,"
            f"{self.font_size},{self.primary_color},{self.secondary_color},"
            f"{self.outline_color},{self.back_color},{bold},0,0,0,100,100,0,0,3,3,1,"
            f"{self.alignment},80,80,{self.margin_v},1"
        )


TEMPLATES: dict[str, CaptionTemplate] = {
    "tiktok_classic": CaptionTemplate("tiktok_classic", "TikTok Classic", 64, "&H00FFFFFF", "&H0000FFFF", effect="karaoke"),
    "tiktok_zoom": CaptionTemplate("tiktok_zoom", "TikTok Zoom", 72, "&H00FFFFFF", "&H0000E5FF", effect="zoom"),
    "alex_hormozi": CaptionTemplate("alex_hormozi", "Alex Hormozi", 70, "&H0000FFFF", "&H00FFFFFF", effect="bounce"),
    "modern_minimal": CaptionTemplate("modern_minimal", "Modern Minimal", 52, "&H00FFFFFF", "&H00DDDDDD", margin_v=240),
    "youtube_shorts": CaptionTemplate("youtube_shorts", "YouTube Shorts", 62, "&H00FFFFFF", "&H000000FF", effect="karaoke"),
    "podcast_style": CaptionTemplate("podcast_style", "Podcast Style", 48, "&H00F5F5F5", "&H0099CCFF", margin_v=120),
    "news_style": CaptionTemplate("news_style", "News Style", 46, "&H00FFFFFF", "&H0000FFFF", alignment=2, margin_v=100),
    "neon_pop": CaptionTemplate("neon_pop", "Neon Pop", 76, "&H00FFFFFF", "&H0000E5FF", outline_color="&H00FF2BD6", effect="karaoke"),
    "product_demo": CaptionTemplate("product_demo", "Product Demo", 54, "&H00FFFFFF", "&H00C7F9CC", margin_v=210, effect="zoom"),
    "cinematic_gold": CaptionTemplate("cinematic_gold", "Cinematic Gold", 50, "&H00F4E7B2", "&H00FFFFFF", outline_color="&H00111111", margin_v=180),
    "creator_clean": CaptionTemplate("creator_clean", "Creator Clean", 58, "&H00FFFFFF", "&H00BCE7FD", margin_v=220, effect="bounce"),
}


def get_template(key: str) -> CaptionTemplate:
    return TEMPLATES.get(key, TEMPLATES["tiktok_classic"])


def list_templates() -> list[str]:
    return list(TEMPLATES.keys())