#!/usr/bin/env python3

import argparse
import re
import subprocess
from pathlib import Path

REPO_DIR = Path(__file__).parent.parent
TRANS_DIR = REPO_DIR / "po"
TRANS_TEMPLATE = TRANS_DIR / "gajim_plugins.pot"
BUILD_DIR = REPO_DIR / "plugins_translations"
TRANSLATABLE_FILES = [
    "*.py",
    "*.ui",
]
PLUGIN_DIRS = [
    "acronyms_expander",
    "anti_spam",
    "clients_icons",
    "length_notifier",
    "message_box_size",
    "now_listen",
    "openpgp",
    "pgp",
    "plugins_translations",
    "quick_replies",
    "triggers",
]


def template_is_equal(old_template_path: Path, new_template: str) -> bool:
    with open(old_template_path) as f:
        old_template = f.read()

    pattern = r'"POT-Creation-Date: .*\n"'

    old_template = re.sub(pattern, "", old_template, count=1)
    new_template = re.sub(pattern, "", new_template, count=1)

    return old_template == new_template


def update_translation_template() -> bool:
    paths: list[Path] = []
    for file_path in TRANSLATABLE_FILES:
        for plugin_dir in PLUGIN_DIRS:
            plugin_path = REPO_DIR / plugin_dir
            paths += list(plugin_path.rglob(file_path))

    cmd = [
        "xgettext",
        "-o",
        "-",
        "-c#",
        "--from-code=utf-8",
        "--keyword=Q_",
        "--no-location",
        "--sort-output",
        "--package-name=Gajim Plugins",
    ]

    for path in paths:
        cmd.append(str(path))

    result = subprocess.run(
        cmd, cwd=REPO_DIR, text=True, check=True, capture_output=True
    )

    template = result.stdout

    if TRANS_TEMPLATE.exists() and template_is_equal(TRANS_TEMPLATE, template):
        # No new strings were discovered
        return False

    with open(TRANS_TEMPLATE, "w") as f:
        f.write(template)
    return True


def update_translation_files() -> None:
    for file in TRANS_DIR.glob("*.po"):
        subprocess.run(
            ["msgmerge", "-U", "--sort-output", str(file), TRANS_TEMPLATE],
            cwd=REPO_DIR,
            check=True,
        )


def build_translations() -> None:
    for po_file in TRANS_DIR.glob("*.po"):
        lang = po_file.stem
        po_file = TRANS_DIR / f"{lang}.po"
        mo_file = BUILD_DIR / f"{po_file.stem}.mo"

        subprocess.run(
            ["msgfmt", str(po_file), "-o", str(mo_file)], cwd=REPO_DIR, check=True
        )


def cleanup_translations() -> None:
    for po_file in TRANS_DIR.glob("*.po"):
        subprocess.run(
            ["msgattrib", "--output-file", str(po_file), "--no-obsolete", str(po_file)],
            cwd=REPO_DIR,
            check=True,
        )


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Update Translations")
    parser.add_argument("command", choices=["update", "build", "cleanup"])
    args = parser.parse_args()

    if args.command == "cleanup":
        cleanup_translations()

    elif args.command == "update":
        update_translation_template()
        update_translation_files()

    elif args.command == "build":
        update_translation_template()
        update_translation_files()
        build_translations()