#!/usr/bin/env python3

import os

from unidecode import unidecode

import config
import parser
import flow
import fix

japanese_ranges = [
    (0x4E00, 0x9FFF),  # Kanji
    (0x3040, 0x309F),  # Hiragana
    (0x30A0, 0x30FF),  # Katakana
    (0xFF00, 0xFFEF),  # Full-width Roman characters and symbols
    (0x3000, 0x303F),  # CJK symbols and punctuation (including 「」)
    (8220, 8220), # The character “
]

debug_current_line = -1

def process_sections():
    output_filepath = os.path.join(config.get('output_path'), 'out.txt')
    outfile = open(output_filepath, 'w', encoding='shift_jisx0213')
    origfile = open(fix.open_onikakushi(), 'r', encoding='shift_jisx0213')

    sections = dict()

    for line in origfile:
        global debug_current_line
        debug_current_line += 1

        if line.startswith('*'):
            section_name = line[1:].split(' ', 1)[0].replace('\n','')

            outfile.write(line)

            if section_name in flow.onik:
                print("entering", section_name)
                write_translated(
                    outfile,
                    origfile,
                    flow.onik[section_name],
                )
        else:
            outfile.write(line)

    outfile.close()
    origfile.close()


def get_symbols(line: str) -> list[str]:
    res = []

    inbetween = 0
    i = 0
    while i < len(line):
        if line[i] in ['@', '/', '¥']:
            symbol = line[i]
            i += 1
        elif line[i:i+3] == '!sd':
            symbol = line[i:i+3]
            i += 3
        elif line[i:i+2] in ['!d', '!w', '!s']:
            x = i
            i += 2
            while i < len(line):
                if line[i].isdigit():
                    i += 1
                    continue

                symbol = line[x:i]
                break
        elif line[i] == '!':
            raise Exception('Unhandled symbol', line)
        else: # It's not a symbol, it's a regular character
            #print(line[i])
            inbetween += 1
            i += 1
            continue

        # Only reaches this if it's a symbol
        #print(symbol)
        if inbetween > 0:
            res.append(symbol)
        else:
            res[-1] += symbol
        inbetween = 0

    return res


# Given a set of translation files, the original file and the output file
# replace the japanese lines with the translated ones in a given section.
def write_translated(outfile, origfile, translation_file_paths):
    for transfilepath in translation_file_paths:
        print(f'- reading "{transfilepath}"')
        structure = parser.parse_to_structure(transfilepath)

        for line in origfile:
            global debug_current_line
            debug_current_line += 1

            # Check if the current line is a dialogue line or not
            found = False
            for start, end in japanese_ranges:
                if start <= ord(line[0]) <= end:
                    found = True

            if found:
                symbols = get_symbols(line)
                print("\n-", debug_current_line, transfilepath, symbols)

                outfile.write('`')

                _printed_line_jp = ""
                _printed_line_en = ""
                while True:
                    _printed_line_jp += structure[0].text_jp
                    _printed_line_en += structure[0].text_en
                    outfile.write(
                        unidecode(structure.pop(0).text_en)
                    )

                    if len(symbols) > 0:
                        _printed_line_jp += symbols[0]
                        _printed_line_en += symbols[0]
                        outfile.write(symbols.pop(0))

                    if len(symbols) <= 0:
                        break

                outfile.write('\n')

                print(">", _printed_line_en)
                print(">", _printed_line_jp)
                print("<", line, end='')

                if _printed_line_jp+'\n' != line:
                    raise Exception('UGABUNGA')

                # Used up all of the structures, this chapter has ended.
                # Got to the next one
                if len(structure) <= 0:
                    break

            else:
                outfile.write(line)