#!/usr/bin/env python3
"""Normalize key order and backfill defaults in applications.json."""

import json
import sys
from pathlib import Path
from typing import Any

from constants import SETTINGS_SCHEMA, SRC_FILE

KEY_ORDER = [
    "id",
    "url",
    "author",
    "name",
    "preferredApkIndex",
    "additionalSettings",
    "categories",
    "allowIdChange",
    "overrideSource",
    "meta",
]

DEFAULTS: dict[str, object] = {
    "allowIdChange": False,
}

_SETTINGS_KEY_ORDER = list(SETTINGS_SCHEMA.keys())


def _order_dict(d: dict[str, Any], key_order: list[str]) -> dict[str, Any]:
    ordered: dict[str, Any] = {}
    for key in key_order:
        if key in d:
            ordered[key] = d[key]
    for key in d:
        if key not in ordered:
            ordered[key] = d[key]
    return ordered


def normalize_app(app: dict) -> dict:
    for key, default in DEFAULTS.items():
        if key not in app:
            app[key] = default

    settings = app.get("additionalSettings")
    if isinstance(settings, dict):
        app["additionalSettings"] = _order_dict(settings, _SETTINGS_KEY_ORDER)

    return _order_dict(app, KEY_ORDER)


def normalize(input_path: str) -> int:
    path = Path(input_path)
    if not path.exists():
        print(f"Error: {path} not found.")
        return 1

    with open(path, "r", encoding="utf-8") as f:
        data = json.load(f)

    apps = data.get("apps", [])
    if not apps:
        print("No apps found in file.")
        return 1

    changes = 0
    for i, app in enumerate(apps):
        normalized = normalize_app(app)

        if list(app.keys()) != list(normalized.keys()) or app != normalized:
            changes += 1

        apps[i] = normalized

    with open(path, "w", encoding="utf-8") as f:
        json.dump(data, f, indent=2, ensure_ascii=False)
        f.write("\n")

    if changes:
        print(f"Normalized {changes} app(s) in {path}")
    else:
        print(f"All {len(apps)} apps already normalized")

    return 0


if __name__ == "__main__":
    input_file = sys.argv[1] if len(sys.argv) > 1 else SRC_FILE
    sys.exit(normalize(input_file))