#!/usr/bin/env python3
# Copyright 2026 The Helium Authors
# You can use, redistribute, and/or modify this source code under
# the terms of the GPL-3.0 license that can be found in the LICENSE file.
"""
Utility file for generating files for translation and
importing them into the codebase.
"""

import argparse
import sys
from pathlib import Path

REPO_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(REPO_ROOT))

PLATFORMS_DIR = Path(__file__).resolve().parent / "i18n-data"
OUT_PATH = REPO_ROOT / 'i18n' / 'source.gen.json'


def parse_args():
    """CLI arg parsing"""
    parser = argparse.ArgumentParser(description='i18n tooling for Helium')
    subparsers = parser.add_subparsers(dest='command', required=True)
    base = subparsers.add_parser('generate', help='Extract translatable strings from patches')
    base.add_argument('-p',
                      '--platforms-dir',
                      type=Path,
                      default=PLATFORMS_DIR,
                      help='Path where platform repos will be cloned')
    base.add_argument('-o',
                      '--output',
                      type=Path,
                      default=OUT_PATH,
                      help='Output path where base JSON file will be saved')

    translate = subparsers.add_parser('translate',
                                      help='Translate source strings into target languages')
    translate.add_argument('-l',
                           '--language',
                           type=str,
                           nargs='+',
                           help='Target language code(s) (e.g. "fr" "de"). '
                           'If omitted, translates all languages.')
    translate.add_argument('-f',
                           '--from-file',
                           type=Path,
                           help='Import translations from a JSON file '
                           'instead of calling the LLM.')
    translate.add_argument('--cmd',
                           help='Run translations through this command instead of '
                           'the LLM completions API. The filled prompt is appended '
                           'as the final argument and the JSON payload is passed '
                           'on stdin.')

    subparsers.add_parser('clean', help='Clean up stale translation strings')

    forage = subparsers.add_parser('forage', help='Find existing translation strings in Chromium')
    forage.add_argument('-t',
                        '--tree',
                        type=Path,
                        required=True,
                        help='Path to Chromium source tree')

    return parser.parse_args()


def main():
    """CLI entrypoint"""
    args = parse_args()

    if args.command == 'generate':
        import i18n_generate # pylint: disable=import-outside-toplevel
        return i18n_generate.run(args, REPO_ROOT)

    if args.command == 'translate':
        import i18n_translate # pylint: disable=import-outside-toplevel
        return i18n_translate.run(args)

    if args.command == 'clean':
        import i18n_clean # pylint: disable=import-outside-toplevel
        return i18n_clean.run()

    if args.command == 'forage':
        import i18n_forage # pylint: disable=import-outside-toplevel
        return i18n_forage.run(args)

    raise ValueError(f'unknown command: {args.command}')


if __name__ == '__main__':
    sys.exit(main())