import glob
import json
import os
import shutil


def main():
    # 1. Ensure ga directories exist and have translations.json
    packages = [
        "invenio_communities",
        "invenio_search_ui",
        "invenio_app_rdm",
        "invenio_requests",
        "invenio_jobs",
        "invenio_rdm_records",
        "invenio_administration",
    ]
    bases = [
        "/opt/invenio/src/.venv/lib/python3.12/site-packages",
        "/opt/invenio/var/instance/assets/translations",
    ]
    for pkg in packages:
        for base in bases:
            for root, dirs, files_in_dir in os.walk(base):
                if root.endswith("messages") and pkg in root:
                    ga_dir = os.path.join(root, "ga")
                    os.makedirs(ga_dir, exist_ok=True)
                    json_path = os.path.join(ga_dir, "translations.json")
                    if not os.path.exists(json_path) or os.path.getsize(json_path) <= 2:
                        with open(json_path, "w") as fp:
                            fp.write("{}")
                        print(f"Created empty translations.json: {json_path}")

    # 2. Copy the actual compiled translations if they exist
    copies = [
        (
            "/opt/invenio/src/.venv/lib/python3.12/site-packages/invenio_communities/assets/semantic-ui/translations/invenio_communities/messages/ga/translations.json",
            "/opt/invenio/var/instance/assets/translations/invenio_communities/messages/ga/translations.json",
        ),
        (
            "/opt/invenio/src/.venv/lib/python3.12/site-packages/invenio_search_ui/assets/semantic-ui/translations/invenio_search_ui/messages/ga/translations.json",
            "/opt/invenio/var/instance/assets/translations/invenio_search_ui/messages/ga/translations.json",
        ),
    ]
    for src, dst in copies:
        if os.path.exists(src):
            os.makedirs(os.path.dirname(dst), exist_ok=True)
            shutil.copy2(src, dst)
            print(f"Copied compiled translations: {src} -> {dst}")

    # 3. Extract app_rdm translations from global ga.json
    global_ga = "/opt/invenio/var/instance/translations/ga.json"
    app_rdm_dst = "/opt/invenio/var/instance/assets/translations/invenio_app_rdm/messages/ga/translations.json"
    if os.path.exists(global_ga):
        try:
            with open(global_ga, "r") as fp:
                ga_data = json.load(fp)
            app_rdm_data = ga_data.get("invenio_app_rdm", {})
            os.makedirs(os.path.dirname(app_rdm_dst), exist_ok=True)
            with open(app_rdm_dst, "w") as fp:
                json.dump(app_rdm_data, fp, indent=2)
            print(f"Extracted invenio_app_rdm translations to {app_rdm_dst}")

            # Also extract search_ui and communities if they were not copied
            search_ui_dst = "/opt/invenio/var/instance/assets/translations/invenio_search_ui/messages/ga/translations.json"
            if "invenio_search_ui" in ga_data:
                os.makedirs(os.path.dirname(search_ui_dst), exist_ok=True)
                with open(search_ui_dst, "w") as fp:
                    json.dump(ga_data["invenio_search_ui"], fp, indent=2)
                print(f"Extracted invenio_search_ui translations to {search_ui_dst}")

            communities_dst = "/opt/invenio/var/instance/assets/translations/invenio_communities/messages/ga/translations.json"
            if "invenio_communities" in ga_data:
                os.makedirs(os.path.dirname(communities_dst), exist_ok=True)
                with open(communities_dst, "w") as fp:
                    json.dump(ga_data["invenio_communities"], fp, indent=2)
                print(
                    f"Extracted invenio_communities translations to {communities_dst}"
                )
        except Exception as e:
            print(f"Error extracting translations: {e}")

    # 4. Patch all _generatedTranslations.js files
    files = glob.glob(
        "/opt/invenio/src/.venv/lib/python3.12/site-packages/**/_generatedTranslations.js",
        recursive=True,
    ) + glob.glob(
        "/opt/invenio/var/instance/assets/**/_generatedTranslations.js", recursive=True
    )
    for f in files:
        with open(f, "r") as fp:
            content = fp.read()
        if "TRANSLATE_GA" not in content:
            lines = content.split("\n")
            last_import = -1
            for i, line in enumerate(lines):
                if line.startswith("import "):
                    last_import = i
            if last_import != -1:
                lines.insert(
                    last_import + 1,
                    'import TRANSLATE_GA from "./ga/translations.json";',
                )
            content = "\n".join(lines)
            if "export const translations = {" in content:
                content = content.replace(
                    "export const translations = {",
                    "export const translations = {\n  ga: { translation: TRANSLATE_GA },",
                )
            with open(f, "w") as fp:
                fp.write(content)
            print(f"Patched _generatedTranslations.js: {f}")


if __name__ == "__main__":
    main()