import base64
import re
from datetime import datetime
from typing import Final, NotRequired, TypedDict
from urllib.parse import quote

import pydash
from adapters.services.screenscraper import ScreenScraperService
from adapters.services.screenscraper_types import SSGame, SSGameDate
from config import SCREENSCRAPER_PASSWORD, SCREENSCRAPER_USER
from unidecode import unidecode as uc

from .base_hander import (
    PS2_OPL_REGEX,
    SONY_SERIAL_REGEX,
    SWITCH_PRODUCT_ID_REGEX,
    SWITCH_TITLEDB_REGEX,
    MetadataHandler,
)

# Used to display the Screenscraper API status in the frontend
SS_API_ENABLED: Final = bool(SCREENSCRAPER_USER) and bool(SCREENSCRAPER_PASSWORD)
SS_DEV_ID: Final = base64.b64decode("enVyZGkxNQ==").decode()
SS_DEV_PASSWORD: Final = base64.b64decode("eFRKd29PRmpPUUc=").decode()

PS1_SS_ID: Final = 57
PS2_SS_ID: Final = 58
PSP_SS_ID: Final = 61
SWITCH_SS_ID: Final = 225
ARCADE_SS_IDS: Final = [
    6,
    7,
    8,
    47,
    49,
    52,
    53,
    54,
    55,
    56,
    68,
    69,
    75,
    112,
    142,
    147,
    148,
    149,
    150,
    151,
    152,
    153,
    154,
    155,
    156,
    157,
    158,
    159,
    160,
    161,
    162,
    163,
    164,
    165,
    166,
    167,
    168,
    169,
    170,
    173,
    174,
    175,
    176,
    177,
    178,
    179,
    180,
    181,
    182,
    183,
    184,
    185,
    186,
    187,
    188,
    189,
    190,
    191,
    192,
    193,
    194,
    195,
    196,
    209,
    227,
    130,
    158,
    269,
]


class SSPlatform(TypedDict):
    slug: str
    ss_id: int | None
    name: NotRequired[str]


class SSAgeRating(TypedDict):
    rating: str
    category: str
    rating_cover_url: str


class SSMetadata(TypedDict):
    ss_score: str
    first_release_date: int | None
    alternative_names: list[str]
    companies: list[str]
    franchises: list[str]
    game_modes: list[str]
    genres: list[str]


class SSRom(TypedDict):
    ss_id: int | None
    name: NotRequired[str]
    summary: NotRequired[str]
    url_cover: NotRequired[str]
    url_manual: NotRequired[str]
    url_screenshots: NotRequired[list[str]]
    ss_metadata: NotRequired[SSMetadata]


def build_ss_rom(game: SSGame) -> SSRom:
    res_name = next(
        (name["text"] for name in game.get("noms", []) if name.get("region") == "ss"),
        "",
    )
    res_summary = next(
        (
            synopsis["text"]
            for synopsis in game.get("synopsis", [])
            if synopsis.get("langue") == "en"
        ),
        "",
    )

    cover_preferred_regions = ["us", "ss"]
    url_cover = ""
    for region in cover_preferred_regions:
        url_cover = next(
            (
                media["url"]
                for media in game.get("medias", [])
                if media.get("region") == region
                and media.get("type") == "box-2D"
                and media.get("parent") == "jeu"
            ),
            "",
        )
        if url_cover:
            break

    manual_preferred_regions = ["us", "eu"]
    url_manual: str = ""
    for region in manual_preferred_regions:
        url_manual = next(
            (
                media["url"]
                for media in game.get("medias", [])
                if media.get("region") == region
                and media.get("type") == "manuel"
                and media.get("parent") == "jeu"
                and media.get("format") == "pdf"
            ),
            "",
        )
        if url_manual:
            break

    ss_id = int(game["id"]) if game.get("id") is not None else None
    rom: SSRom = {
        "ss_id": ss_id,
        "name": res_name,
        "summary": res_summary,
        "url_cover": url_cover,
        "url_manual": url_manual,
        "url_screenshots": [],
        "ss_metadata": extract_metadata_from_ss_rom(game),
    }

    return SSRom({k: v for k, v in rom.items() if v})  # type: ignore[misc]


def extract_metadata_from_ss_rom(rom: SSGame) -> SSMetadata:
    def _normalize_score(score: str) -> str:
        """Normalize the score to be between 0 and 10 because for some reason Screenscraper likes to rate over 20."""
        try:
            return str(int(score) / 2)
        except (ValueError, TypeError):
            return ""

    def _get_lowest_date(dates: list[SSGameDate]) -> int | None:
        lowest_date = min(dates, default=None, key=lambda v: v.get("text", ""))
        if not lowest_date:
            return None

        try:
            return int(datetime.strptime(lowest_date["text"], "%Y-%m-%d").timestamp())
        except ValueError:
            try:
                return int(datetime.strptime(lowest_date["text"], "%Y").timestamp())
            except ValueError:
                return None

    def _get_genres(rom: SSGame) -> list[str]:
        return [
            genre_name["text"]
            for genre in rom.get("genres", [])
            for genre_name in genre.get("noms", [])
            if genre_name.get("langue") == "en"
        ]

    def _get_franchises(rom: SSGame) -> list[str]:
        preferred_languages = ["en", "fr"]
        for lang in preferred_languages:
            franchises = [
                franchise_name["text"]
                for franchise in rom.get("familles", [])
                for franchise_name in franchise.get("noms", [])
                if franchise_name.get("langue") == lang
            ]
            if franchises:
                return franchises
        return []

    def _get_game_modes(rom: SSGame) -> list[str]:
        preferred_languages = ["en", "fr"]
        for lang in preferred_languages:
            modes = [
                mode_name["text"]
                for mode in rom.get("modes", [])
                for mode_name in mode.get("noms", [])
                if mode_name.get("langue") == lang
            ]
            if modes:
                return modes
        return []

    return SSMetadata(
        {
            "ss_score": _normalize_score(rom.get("note", {}).get("text", "")),
            "alternative_names": [name["text"] for name in rom.get("noms", [])],
            "companies": pydash.compact(
                [
                    rom.get("editeur", {}).get("text"),
                    rom.get("developpeur", {}).get("text"),
                ]
            ),
            "genres": _get_genres(rom),
            "first_release_date": _get_lowest_date(rom.get("dates", [])),
            "franchises": _get_franchises(rom),
            "game_modes": _get_game_modes(rom),
        }
    )


class SSHandler(MetadataHandler):
    def __init__(self) -> None:
        self.ss_service = ScreenScraperService()

    async def _search_rom(self, search_term: str, platform_ss_id: int) -> SSGame | None:
        if not platform_ss_id:
            return None

        search_term = uc(search_term)
        roms = await self.ss_service.search_games(
            term=quote(search_term, safe="/ "),
            system_id=platform_ss_id,
        )
        return roms[0] if roms else None

    def get_platform(self, slug: str) -> SSPlatform:
        platform = SCREENSAVER_PLATFORM_LIST.get(slug, None)

        if not platform:
            return SSPlatform(ss_id=None, slug=slug)

        return SSPlatform(
            ss_id=platform["id"],
            slug=slug,
            name=platform["name"],
        )

    async def get_rom(self, file_name: str, platform_ss_id: int) -> SSRom:
        from handler.filesystem import fs_rom_handler

        if not SS_API_ENABLED:
            return SSRom(ss_id=None)

        if not platform_ss_id:
            return SSRom(ss_id=None)

        search_term = fs_rom_handler.get_file_name_with_no_tags(file_name)
        fallback_rom = SSRom(ss_id=None)

        # Support for PS2 OPL filename format
        match = PS2_OPL_REGEX.match(file_name)
        if platform_ss_id == PS2_SS_ID and match:
            search_term = await self._ps2_opl_format(match, search_term)
            fallback_rom = SSRom(ss_id=None, name=search_term)

        # Support for sony serial filename format (PS, PS3, PS3)
        match = SONY_SERIAL_REGEX.search(file_name, re.IGNORECASE)
        if platform_ss_id == PS1_SS_ID and match:
            search_term = await self._ps1_serial_format(match, search_term)
            fallback_rom = SSRom(ss_id=None, name=search_term)

        if platform_ss_id == PS2_SS_ID and match:
            search_term = await self._ps2_serial_format(match, search_term)
            fallback_rom = SSRom(ss_id=None, name=search_term)

        if platform_ss_id == PSP_SS_ID and match:
            search_term = await self._psp_serial_format(match, search_term)
            fallback_rom = SSRom(ss_id=None, name=search_term)

        # Support for switch titleID filename format
        match = SWITCH_TITLEDB_REGEX.search(file_name)
        if platform_ss_id == SWITCH_SS_ID and match:
            search_term, index_entry = await self._switch_titledb_format(
                match, search_term
            )
            if index_entry:
                fallback_rom = SSRom(
                    ss_id=None,
                    name=index_entry["name"],
                    summary=index_entry.get("description", ""),
                    url_cover=index_entry.get("iconUrl", ""),
                    url_manual=index_entry.get("iconUrl", ""),
                    url_screenshots=index_entry.get("screenshots", None) or [],
                )

        # Support for switch productID filename format
        match = SWITCH_PRODUCT_ID_REGEX.search(file_name)
        if platform_ss_id == SWITCH_SS_ID and match:
            search_term, index_entry = await self._switch_productid_format(
                match, search_term
            )
            if index_entry:
                fallback_rom = SSRom(
                    ss_id=None,
                    name=index_entry["name"],
                    summary=index_entry.get("description", ""),
                    url_cover=index_entry.get("iconUrl", ""),
                    url_manual=index_entry.get("iconUrl", ""),
                    url_screenshots=index_entry.get("screenshots", None) or [],
                )

        # Support for MAME arcade filename format
        if platform_ss_id in ARCADE_SS_IDS:
            search_term = await self._mame_format(search_term)
            fallback_rom = SSRom(ss_id=None, name=search_term)

        search_term = self.normalize_search_term(search_term)
        res = await self._search_rom(search_term, platform_ss_id)

        # Some MAME games have two titles split by a slash
        if not res and "/" in search_term:
            for term in search_term.split("/"):
                res = await self._search_rom(term.strip(), platform_ss_id)
                if res:
                    break

        if not res or not res.get("id"):
            return fallback_rom

        return build_ss_rom(res)

    async def get_rom_by_id(self, ss_id: int) -> SSRom:
        if not SS_API_ENABLED:
            return SSRom(ss_id=None)

        res = await self.ss_service.get_game_info(game_id=ss_id)
        if not res:
            return SSRom(ss_id=None)

        return build_ss_rom(res)

    async def get_matched_rom_by_id(self, ss_id: int) -> SSRom | None:
        if not SS_API_ENABLED:
            return None

        rom = await self.get_rom_by_id(ss_id)
        return rom if rom.get("ss_id", "") else None

    async def get_matched_roms_by_name(
        self, search_term: str, platform_ss_id: int | None
    ) -> list[SSRom]:
        if not SS_API_ENABLED:
            return []

        if not platform_ss_id:
            return []

        search_term = uc(search_term)
        matched_roms = await self.ss_service.search_games(
            term=quote(search_term, safe="/ "),
            system_id=platform_ss_id,
        )

        def _is_ss_region(rom: SSGame) -> bool:
            return any(name.get("region") == "ss" for name in rom.get("noms", []))

        return [
            build_ss_rom(rom)
            for rom in matched_roms
            if _is_ss_region(rom) and rom.get("id")
        ]


class SlugToSSId(TypedDict):
    id: int
    name: str


SCREENSAVER_PLATFORM_LIST: dict[str, SlugToSSId] = {
    "3do": {"id": 29, "name": "3DO"},
    "amiga": {"id": 64, "name": "Amiga"},
    "amiga-cd32": {"id": 134, "name": "Amiga CD"},
    "cpc": {"id": 60, "name": "CPC"},
    "acpc": {"id": 60, "name": "CPC"},  # IGDB
    "adventure-vision": {"id": 78, "name": "Entex Adventure Vision"},
    "amstrad-gx4000": {"id": 87, "name": "Amstrad GX4000"},
    "android": {"id": 63, "name": "Android"},
    "apple2": {"id": 86, "name": "Apple II"},
    "appleii": {"id": 86, "name": "Apple II"},  # IGDB
    "apple2gs": {"id": 217, "name": "Apple IIGS"},
    "apple-iigs": {"id": 51, "name": "Apple IIGS"},  # IGDB
    "arcadia-2001": {"id": 94, "name": "Arcadia 2001"},
    "arduboy": {"id": 263, "name": "Arduboy"},
    "atari-2600": {"id": 26, "name": "Atari 2600"},
    "atari2600": {"id": 26, "name": "Atari 2600"},  # IGDB
    "atari-5200": {"id": 40, "name": "Atari 5200"},
    "atari5200": {"id": 40, "name": "Atari 5200"},  # IGDB
    "atari-7800": {"id": 41, "name": "Atari 7800"},
    "atari7800": {"id": 41, "name": "Atari 7800"},  # IGDB
    "atari-8-bit": {"id": 43, "name": "Atari 8bit"},
    "atari8bit": {"id": 43, "name": "Atari 8bit"},  # IGDB
    "atari-st": {"id": 42, "name": "Atari ST"},
    "atom": {"id": 36, "name": "Atom"},
    "bbc-micro": {"id": 37, "name": "BBC Micro"},
    "bbcmicro": {"id": 37, "name": "BBC Micro"},  # IGDB
    "bally-astrocade": {"id": 44, "name": "Astrocade"},
    "astrocade": {"id": 44, "name": "Astrocade"},  # IGDB
    "cd-i": {"id": 133, "name": "CD-i"},
    "philips-cd-i": {"id": 133, "name": "CD-i"},  # IGDB
    "cdtv": {"id": 129, "name": "Amiga CDTV"},
    "commodore-cdtv": {"id": 129, "name": "Amiga CDTV"},  # IGDB
    "camputers-lynx": {"id": 88, "name": "Camputers Lynx"},
    "casio-loopy": {"id": 98, "name": "Loopy"},
    "casio-pv-1000": {"id": 74, "name": "PV-1000"},
    "channel-f": {"id": 80, "name": "Channel F"},
    "fairchild-channel-f": {"id": 80, "name": "Channel F"},  # IGDB
    "colecoadam": {"id": 89, "name": "Adam"},
    "colecovision": {"id": 48, "name": "Colecovision"},
    "colour-genie": {"id": 92, "name": "EG2000 Colour Genie"},
    "c128": {"id": 66, "name": "Commodore 64"},
    "commodore-16-plus4": {"id": 99, "name": "Plus/4"},
    "c-plus-4": {"id": 99, "name": "Plus/4"},  # IGDB
    "c16": {"id": 99, "name": "Plus/4"},  # IGDB
    "c64": {"id": 66, "name": "Commodore 64"},
    "pet": {"id": 240, "name": "PET"},
    "cpet": {"id": 240, "name": "PET"},  # IGDB
    "creativision": {"id": 241, "name": "CreatiVision"},
    "dos": {"id": 135, "name": "PC Dos"},
    "dragon-3264": {"id": 91, "name": "Dragon 32/64"},
    "dragon-32-slash-64": {"id": 91, "name": "Dragon 32/64"},  # IGDB
    "dreamcast": {"id": 23, "name": "Dreamcast"},
    "dc": {"id": 23, "name": "Dreamcast"},  # IGDB
    "electron": {"id": 85, "name": "Electron"},
    "acorn-electron": {"id": 85, "name": "Electron"},  # IGDB
    "epoch-game-pocket-computer": {"id": 95, "name": "Game Pocket Computer"},
    "epoch-super-cassette-vision": {"id": 67, "name": "Super Cassette Vision"},
    "exelvision": {"id": 96, "name": "EXL 100"},
    "exidy-sorcerer": {"id": 165, "name": "Exidy"},
    "fmtowns": {"id": 253, "name": "FM Towns"},
    "fm-towns": {"id": 253, "name": "FM Towns"},  # IGDB
    "fm-7": {"id": 97, "name": "FM-7"},
    "g-and-w": {"id": 52, "name": "Game & Watch"},  # IGDB (Game & Watch)
    "gp32": {"id": 101, "name": "GP32"},
    "gameboy": {"id": 9, "name": "Game Boy"},
    "gb": {"id": 9, "name": "Game Boy"},  # IGDB
    "gameboy-advance": {"id": 12, "name": "Game Boy Advance"},
    "gba": {"id": 12, "name": "Game Boy Advance"},  # IGDB
    "gameboy-color": {"id": 10, "name": "Game Boy Color"},
    "gbc": {"id": 10, "name": "Game Boy Color"},  # IGDB
    "game-gear": {"id": 21, "name": "Game Gear"},
    "gamegear": {"id": 21, "name": "Game Gear"},  # IGDB
    "game-com": {"id": 121, "name": "Game.com"},
    "game-dot-com": {"id": 121, "name": "Game.com"},  # IGDB
    "gamecube": {"id": 13, "name": "GameCube"},
    "ngc": {"id": 13, "name": "GameCube"},  # IGDB
    "genesis": {"id": 1, "name": "Megadrive"},
    "genesis-slash-megadrive": {"id": 1, "name": "Megadrive"},
    "hartung": {"id": 103, "name": "Game Master"},
    "intellivision": {"id": 115, "name": "Intellivision"},
    "jaguar": {"id": 27, "name": "Jaguar"},
    "jupiter-ace": {"id": 126, "name": "Jupiter Ace"},
    "linux": {"id": 145, "name": "Linux"},
    "lynx": {"id": 28, "name": "Lynx"},
    "msx": {"id": 113, "name": "MSX"},
    "msx-turbo": {"id": 118, "name": "MSX Turbo R"},  # IGDB
    "macintosh": {"id": 146, "name": "Mac OS"},
    "mac": {"id": 146, "name": "Mac OS"},  # IGDB
    "ngage": {"id": 30, "name": "N-Gage"},
    "nes": {"id": 3, "name": "NES"},
    "fds": {"id": 106, "name": "Famicom"},
    "neo-geo": {"id": 142, "name": "Neo-Geo"},
    "neogeoaes": {"id": 142, "name": "Neo-Geo"},  # IGDB
    "neogeomvs": {"id": 68, "name": "Neo-Geo MVS"},  # IGDB
    "neo-geo-cd": {"id": 70, "name": "Neo-Geo CD"},
    "neo-geo-pocket": {"id": 25, "name": "Neo-Geo Pocket"},
    "neo-geo-pocket-color": {"id": 82, "name": "Neo-Geo Pocket Color"},
    "3ds": {"id": 17, "name": "Nintendo 3DS"},
    "n64": {"id": 14, "name": "Nintendo 64"},
    "nintendo-ds": {"id": 15, "name": "Nintendo DS"},
    "nds": {"id": 15, "name": "Nintendo DS"},  # IGDB
    "nintendo-dsi": {"id": 15, "name": "Nintendo DS"},
    "switch": {"id": 225, "name": "Switch"},
    "odyssey-2": {"id": 104, "name": "Videopac G7000"},
    "odyssey-2-slash-videopac-g7000": {"id": 104, "name": "Videopac G7000"},
    "oric": {"id": 131, "name": "Oric 1 / Atmos"},
    "pc88": {"id": 221, "name": "NEC PC-8801"},
    "pc-8800-series": {"id": 221, "name": "NEC PC-8801"},  # IGDB
    "pc98": {"id": 208, "name": "NEC PC-9801"},
    "pc-9800-series": {"id": 208, "name": "NEC PC-9801"},  # IGDB
    "pc-fx": {"id": 72, "name": "PC-FX"},
    "pico": {"id": 234, "name": "Pico-8"},
    "ps-vita": {"id": 62, "name": "PS Vita"},
    "psvita": {"id": 62, "name": "PS Vita"},  # IGDB
    "psp": {"id": 61, "name": "PSP"},
    "palmos": {"id": 219, "name": "Palm OS"},
    "palm-os": {"id": 219, "name": "Palm OS"},  # IGDB
    "philips-vg-5000": {"id": 261, "name": "Philips VG 5000"},
    "playstation": {"id": 57, "name": "Playstation"},
    "ps": {"id": 57, "name": "Playstation"},  # IGDB
    "ps2": {"id": 58, "name": "Playstation 2"},
    "ps3": {"id": 59, "name": "Playstation 3"},
    "playstation-4": {"id": 60, "name": "Playstation 4"},
    "ps4--1": {"id": 60, "name": "Playstation 4"},  # IGDB
    "playstation-5": {"id": 284, "name": "Playstation 5"},
    "ps5": {"id": 284, "name": "Playstation 5"},  # IGDB
    "pokemon-mini": {"id": 211, "name": "Pokémon mini"},
    "sam-coupe": {"id": 213, "name": "MGT SAM Coupé"},
    "sega-32x": {"id": 19, "name": "Megadrive 32X"},
    "sega32": {"id": 19, "name": "Megadrive 32X"},  # IGDB
    "sega-cd": {"id": 20, "name": "Mega-CD"},
    "segacd": {"id": 20, "name": "Mega-CD"},  # IGDB
    "sega-master-system": {"id": 2, "name": "Master System"},
    "sms": {"id": 2, "name": "Master System"},  # IGDB
    "sega-pico": {"id": 250, "name": "Sega Pico"},
    "sega-saturn": {"id": 22, "name": "Saturn"},
    "saturn": {"id": 22, "name": "Saturn"},  # IGDB
    "sg-1000": {"id": 109, "name": "SG-1000"},
    "snes": {"id": 4, "name": "Super Nintendo"},
    "sharp-x1": {"id": 220, "name": "Sharp X1"},
    "x1": {"id": 220, "name": "Sharp X1"},  # IGDB
    "sharp-x68000": {"id": 79, "name": "Sharp X68000"},
    "spectravideo": {"id": 218, "name": "Spectravideo"},
    "sufami-turbo": {"id": 108, "name": "Sufami Turbo"},
    "super-acan": {"id": 100, "name": "Super A'can"},
    "supergrafx": {"id": 105, "name": "PC Engine SuperGrafx"},
    "supervision": {"id": 207, "name": "Watara Supervision"},
    "ti-99": {"id": 205, "name": "TI-99/4A"},  # IGDB
    "trs-80-coco": {"id": 144, "name": "TRS-80 Color Computer"},
    "trs-80-color-computer": {"id": 144, "name": "TRS-80 Color Computer"},  # IGDB
    "taito-x-55": {"id": 112, "name": "Type X"},
    "thomson-mo": {"id": 141, "name": "Thomson MO/TO"},
    "thomson-mo5": {"id": 141, "name": "Thomson MO/TO"},
    "thomson-to": {"id": 141, "name": "Thomson MO/TO"},
    "turbografx-cd": {"id": 114, "name": "PC Engine CD-Rom"},
    "turbografx-16-slash-pc-engine-cd": {"id": 114, "name": "PC Engine CD-Rom"},
    "turbo-grafx": {"id": 31, "name": "PC Engine"},
    "turbografx16--1": {"id": 31, "name": "PC Engine"},  # IGDB
    "uzebox": {"id": 216, "name": "UzeBox"},
    "vsmile": {"id": 120, "name": "V.Smile"},
    "vic-20": {"id": 73, "name": "Vic-20"},
    "vectrex": {"id": 102, "name": "Vectrex"},
    "videopac-g7400": {"id": 104, "name": "Videopac G7000"},
    "virtual-boy": {"id": 11, "name": "Virtual Boy"},
    "virtualboy": {"id": 11, "name": "Virtual Boy"},
    "wii": {"id": 16, "name": "Wii"},
    "wii-u": {"id": 18, "name": "Wii U"},
    "wiiu": {"id": 18, "name": "Wii U"},
    "windows": {"id": 3, "name": "Windows"},
    "win": {"id": 138, "name": "PC Windows"},  # IGDB
    "win3x": {"id": 136, "name": "PC Win3.xx"},
    "wonderswan": {"id": 45, "name": "WonderSwan"},
    "wonderswan-color": {"id": 46, "name": "WonderSwan Color"},
    "xbox": {"id": 32, "name": "Xbox"},
    "xbox360": {"id": 33, "name": "Xbox 360"},
    "xbox-one": {"id": 34, "name": "Xbox One"},
    "xboxone": {"id": 34, "name": "Xbox One"},
    "z-machine": {"id": 215, "name": "Z-Machine"},
    "zx-spectrum": {"id": 76, "name": "ZX Spectrum"},
    "zx81": {"id": 77, "name": "ZX81"},
    "sinclair-zx81": {"id": 77, "name": "ZX81"},  # IGDB
}

# Reverse lookup
SS_ID_TO_SLUG = {v["id"]: k for k, v in SCREENSAVER_PLATFORM_LIST.items()}