"""
* Version: 0.1.0.0.0

* Purposes:
    * Translate Markdown files from English to Japanese.

* System parameters:
    * sys_parameter 1: The path to the input Markdown folder.
    * sys_parameter 2: The path to the output Markdown folder.
    * sys_parameter 3: The AWS profile capable of invoking translate service.
    * sys_parameter 4: PingCAP team tag (pe, doc, or tse, ... etc), it decides which terminology dictionary to use.
"""

import sys
from pingcap.pe.flash.util import load_1gram_terms_en
from pingcap.pe.flash.markdown_processor import (
    parse_markdown_folder,
    dump_markdown_folder,
    translate_chapter_v2,
    write_markdown_folder,
)

if __name__ == "__main__":

    input_md_folder_path = sys.argv[1]
    output_md_folder_path = sys.argv[2]
    skip_count = 0
    mode = "standard"
    if len(sys.argv) > 3:
        skip_count = int(sys.argv[3])
    if len(sys.argv) > 4:
        mode = sys.argv[4]

    if mode != "silent":
        print(
            "Translating markdown files in:",
            input_md_folder_path,
            "to",
            output_md_folder_path,
            "for",
            "pe",
        )

    en_1gram_list = load_1gram_terms_en("pe")

    translated_chapter = translate_chapter_v2(
        input_md_folder_path,
        "ja2en",
        "terms" + "-" + "pe" + "-ja-en",
        en_1gram_list,
        skip_count=skip_count,
        mode=mode,
    )
    write_markdown_folder(translated_chapter, output_md_folder_path, mode)