#!/usr/bin/env python3

import json
import os


# Function to load the contents of a file into a line list
def read_file_lines(file_path):
    with open(file_path, 'r', encoding='utf-8') as file:
        return file.readlines()


# Function to load the contents of a file into a line list
def read_file_json(file_path):
    with open(file_path, 'r', encoding='utf-8') as file:
        return json.load(file)


# Function to save a list of lines to a file
def write_to_file(file_path, text):
    with open(file_path, 'w', encoding='utf-8') as file:
        file.write(text)


# Move keys in target into same order as in source.
# Keys not present in source are removed from target.
# Returns a new value of target; does not mutate arguments.
def equalize_key_order(source, target):
    if isinstance(source, dict) and isinstance(target, dict):
        target_result = {}
        for key in source.keys():
            if key in target and target[key] is not None:
                source_value = source[key]
                target_value = target[key]
                target_result[key] = equalize_key_order(source_value, target_value)

            else:
                if key.startswith("@"):
                    target_result[key] = source[key]
                else:
                    target_result[key] = None
        return target_result

    elif isinstance(source, list) and isinstance(target, list):
        return [
            equalize_key_order(source_item, target_item)
            for source_item, target_item in zip(source, target, strict=True)
        ]

    else:
        return target


# Function to update the ARB files
def update_arb_file(source_path, target_path, language_code):
    source_lines = read_file_lines(source_path)
    source_json = read_file_json(source_path)
    target_json = read_file_json(target_path)

    target_reordered = equalize_key_order(source_json, target_json)
    target_text = json.dumps(target_reordered, indent=4, ensure_ascii=False)
    target_lines = target_text.splitlines()

    for i, line in enumerate(source_lines):
        if line.strip() == "":
            target_lines.insert(i, "")

    write_to_file(target_path, "\n".join(target_lines).strip() + "\n")


if __name__ == "__main__":
    source_file_path = 'lib/l10n/app_en.arb'
    target_directory = 'lib/l10n'

    for file_name in os.listdir(target_directory):
        if file_name.startswith('app_') and file_name.endswith('.arb') and file_name != os.path.basename(source_file_path):
            target_file_path = os.path.join(target_directory, file_name)
            language_code = file_name.split('_')[1].split('.')[0]
            update_arb_file(source_file_path, target_file_path, language_code)
            print(f'File updated: {file_name}')