#!/usr/bin/env python3

import os

START_IDENT = "// TRANSLATION-STRING-AREA-BEGIN"
END_IDENT = "// TRANSLATION-STRING-AREA-END"

src_file = os.path.join(os.path.dirname(__file__), "..", "pcsx2", "ImGui", "FullscreenUI.cpp")

with open(src_file, "r") as f:
    full_source = f.read()

strings = []
for token in ["FSUI_STR", "FSUI_CSTR", "FSUI_FSTR", "FSUI_NSTR", "FSUI_VSTR", "FSUI_ICONSTR", "FSUI_ICONSTR_S"]:
    token_len = len(token)
    last_pos = 0
    while True:
        last_pos = full_source.find(token, last_pos)
        if last_pos < 0:
            break

        if last_pos >= 8 and full_source[last_pos - 8:last_pos] == "#define ":
            last_pos += len(token)
            continue

        if full_source[last_pos + token_len] == '(':
            start_pos = last_pos + token_len + 1
            end_pos = full_source.find(")", start_pos)
            s = full_source[start_pos:end_pos]

            # Split into sting arguments, removing "
            string_args = [""]
            arg = 0;
            cpos = s.find(',')
            pos = s.find('"')
            while pos >= 0 or cpos >= 0:
                assert pos == 0 or s[pos - 1] != '\\'
                if cpos == -1 or pos < cpos:
                    epos = pos
                    while True:
                        epos = s.find('"', epos + 1)
                        # found ')' in string, extend s to next ')'
                        if epos == -1:
                            end_pos = full_source.find(")", end_pos + 1)
                            s = full_source[start_pos:end_pos]
                            epos = pos
                            continue

                        if s[epos - 1] == '\\':
                            continue
                        else:
                            break

                    assert epos > pos
                    string_args[arg] += s[pos+1:epos]
                    cpos = s.find(',', epos + 1)
                    pos = s.find('"', epos + 1)
                else:
                    arg += 1
                    string_args.append("")
                    cpos = s.find(',', cpos + 1)

            print(string_args)

            # FSUI_ICONSTR and FSUI_ICONSTR_S need to translate the only the second argument
            # other defines take only a single argument
            if len(string_args) >= 2:
                new_s = string_args[1]
            else:
                new_s = string_args[0]

            assert len(new_s) > 0

            #assert (end_pos - start_pos) < 300
            #if (end_pos - start_pos) >= 300:
            #    print("WARNING: Long string")
            #    print(new_s)
            if new_s not in strings:
                strings.append(new_s)
        last_pos += len(token)

print(f"Found {len(strings)} unique strings.")

start = full_source.find(START_IDENT)
end = full_source.find(END_IDENT)
assert start >= 0 and end > start

new_area = ""
for string in list(strings):
    new_area += f"TRANSLATE_NOOP(\"FullscreenUI\", \"{string}\");\n"

full_source = full_source[:start+len(START_IDENT)+1] + new_area + full_source[end:]
with open(src_file, "w") as f:
    f.write(full_source)