import os
import json
import yaml

from markdownify import markdownify

from utils.qualities import QUALITIES
from utils.strings import get_name


def collect_profile_formats(
    service, trash_score_name, format_items, trash_id_to_scoring_mapping
):
    profile_formats = []
    for name, trash_id in format_items.items():
        scoring = trash_id_to_scoring_mapping[trash_id]
        score = scoring.get(trash_score_name, scoring.get("default", 0))
        if score == 0:
            continue

        profile_formats.append({"name": get_name(service, name), "score": score})
    return sorted(
        profile_formats,
        key=lambda profile_format: (
            -profile_format["score"],
            profile_format["name"].lower(),
        ),
        reverse=False,
    )


def get_quality_id(quality_name):
    return next(
        (quality["id"] for quality in QUALITIES if quality["name"] == quality_name),
        None,
    )


def collect_qualities(items):
    qualities = []
    quality_collection_id = -1
    for item in items:
        if item.get("allowed", False) is False:
            continue

        quality = {
            "id": get_quality_id(item.get("name", "")),
            "name": item.get("name", ""),
        }
        if item.get("items") is not None:
            quality["id"] = quality_collection_id
            quality_collection_id -= 1
            quality["description"] = ""
            quality["qualities"] = []
            for sub_item in item["items"]:
                quality["qualities"].append(
                    {"id": get_quality_id(sub_item), "name": sub_item}
                )
        qualities.append(quality)

    return list(reversed(qualities))


def get_upgrade_until(quality_name, profile_qualities):
    found_quality = next(
        quality for quality in profile_qualities if quality["name"] == quality_name
    )
    if found_quality:
        found_quality = found_quality.copy()
        if found_quality.get("description", "") == "":
            found_quality.pop("description", None)
        found_quality.pop("qualities", None)
    return found_quality


def collect_profile(service, input_json, output_dir, trash_id_to_scoring_mapping):
    # Compose YAML structure
    name = input_json.get("name", "")
    profile_qualities = collect_qualities(input_json.get("items", []))
    yml_data = {
        "name": get_name(service, name),
        "description": f"""[Profile from TRaSH-Guides.](https://trash-guides.info/{service.capitalize()}/{service}-setup-quality-profiles)

{markdownify(input_json.get('trash_description', ''))}""".strip(),
        "tags": [service.capitalize()],
        "upgradesAllowed": input_json.get("upgradeAllowed", True),
        "minCustomFormatScore": input_json.get("minFormatScore", 0),
        "upgradeUntilScore": input_json.get("cutoffFormatScore", 0),
        "minScoreIncrement": input_json.get("minUpgradeFormatScore", 0),
        "custom_formats": collect_profile_formats(
            service,
            input_json.get("trash_score_set"),
            input_json.get("formatItems", {}),
            trash_id_to_scoring_mapping,
        ),
        "qualities": profile_qualities,
        "upgrade_until": get_upgrade_until(input_json.get("cutoff"), profile_qualities),
        "language": input_json.get("language", "any").lower(),
    }

    # Output path
    output_path = os.path.join(output_dir, f"{get_name(service, name)}.yml")
    with open(output_path, "w", encoding="utf-8") as f:
        yaml.dump(yml_data, f, sort_keys=False, allow_unicode=True)
    print(f"Generated: {output_path}")


def collect_profiles(
    service,
    input_dir,
    output_dir,
    trash_id_to_scoring_mapping,
):
    for root, _, files in os.walk(input_dir):
        for filename in files:
            if not filename.endswith(".json"):
                continue

            file_path = os.path.join(root, filename)
            with open(file_path, "r", encoding="utf-8") as f:
                data = json.load(f)
            collect_profile(service, data, output_dir, trash_id_to_scoring_mapping)