"""Download translated strings from GP and save as locale JSON files.

Usage:
    python download_translations.py --output path/to/locales/
"""

import argparse
import json
import sys
from pathlib import Path

from gp_client import TARGET_LANGS, get_strings


def main():
    parser = argparse.ArgumentParser(description="Download translations from GP")
    parser.add_argument("--output", required=True, help="Directory to save translated JSON files")
    args = parser.parse_args()

    output_dir = Path(args.output)
    output_dir.mkdir(parents=True, exist_ok=True)

    failed = []
    for lang in TARGET_LANGS:
        print(f"Downloading '{lang}' translations...")
        try:
            result = get_strings(lang)

            # Extract just the key:value strings from the response
            strings = {
                key: entry.get("value", "") if isinstance(entry, dict) else entry
                for key, entry in result.get("resourceStrings", {}).items()
            }

            if not strings:
                print(f"  No strings yet for '{lang}' (translation may still be in progress)")
                continue

            output_file = output_dir / f"{lang}.json"
            output_file.write_text(
                json.dumps(strings, ensure_ascii=False, indent=2),
                encoding="utf-8",
            )

            print(f"  Saved {len(strings)} strings to {output_file}")

        except Exception as e:  # noqa: BLE001
            print(f"  Error downloading '{lang}': {e}")
            failed.append(lang)

    if failed:
        print(f"\nFAILED languages: {failed}")
        sys.exit(1)

    print("\nDone.")


if __name__ == "__main__":
    main()