from typing import Final, TypeVar

from .models import Character, StarrailInfoParsed
from .models.v1 import Character, StarrailInfoParsedV1

RawData = TypeVar("RawData")
ParsedData = TypeVar("ParsedData", StarrailInfoParsed, StarrailInfoParsedV1)

ASSET_URL: Final[str] = "https://raw.githubusercontent.com/Mar-7th/StarRailRes/master"


def remove_empty_dict(data: RawData) -> RawData:
    """
    Recursively removes empty dictionaries from the given raw data.

    Args:
        - data (`RawData`): The input raw data.

    Returns:
        - `RawData`: The data with empty dictionaries removed.
    """
    if isinstance(data, dict):
        for key in data.keys():
            data[key] = None if (data[key] == {}) else remove_empty_dict(data[key])
    elif isinstance(data, list):
        for i in range(len(data)):
            data[i] = remove_empty_dict(data[i])
    return data


def replace_icon_name_with_url(data: RawData) -> RawData:
    """
    Replaces icon file names with asset URLs in the given raw data.

    Example: Replace "/icon/avatar/1201.png" with
    "https://raw.githubusercontent.com/Mar-7th/StarRailRes/master/icon/avatar/1201.png"

    Args:
        - data (`RawData`): The input raw data.

    Returns:
        - `RawData`: The data with icon file names replaced by asset URLs.
    """
    if isinstance(data, dict):
        for key in data.keys():
            data[key] = replace_icon_name_with_url(data[key])
    elif isinstance(data, list):
        for i in range(len(data)):
            data[i] = replace_icon_name_with_url(data[i])
    elif isinstance(data, str):
        if ".png" in data:
            data = ASSET_URL + "/" + data
    return data


def replace_trailblazer_name(data: StarrailInfoParsedV1) -> StarrailInfoParsedV1:
    """
    Replaces the trailblazer name with the player's name.

    Args:
        - data (`StarrailInfoParsed`): The input StarrailInfoParsed data.

    Returns:
        - `StarrailInfoParsed`: The updated StarrailInfoParsed data.
    """
    for i in range(len(data.characters)):
        if data.characters[i].name == r"{NICKNAME}":
            data.characters[i].name = data.player.name
    return data


def remove_duplicate_character(data: ParsedData) -> ParsedData:
    """
    Removes duplicate characters from the given StarrailInfoParsed data.

    Args:
        - data (`ParsedData`): The input StarrailInfoParsed data.

    Returns:
        - `ParsedData`: The updated StarrailInfoParsed data without duplicate characters.
    """
    new_characters = []
    characters_ids: set[str] = set()
    for character in data.characters:
        if character.id not in characters_ids:
            new_characters.append(character)
            characters_ids.add(character.id)
    data.characters = new_characters
    return data


def merge_character_data(new_data: ParsedData, old_data: ParsedData) -> ParsedData:
    """
    Append the old data characters to the list of new data characters.
    The player's info from the old data will be omitted/discarded.

    Args:
        - new_data (`ParsedData`): The new data to be merged.
        - old_data (`ParsedData`): The old data to merge into.

    Returns:
        - `ParsedData`: The merged new data.
    """
    for character in old_data.characters:
        new_data.characters.append(character)
    new_data = remove_duplicate_character(new_data)
    return new_data