from concurrent.futures import ThreadPoolExecutor, as_completed
import sys
import traceback
from colorama import Fore
import os

from modules.rpgmakermvmz import handleMVMZ
from modules.rpgmakerace import handleACE
from modules.csvtl import handleCSV
from modules.textfile import handleTextfile

THREADS = 20

# Info Message
print(Fore.LIGHTYELLOW_EX + "WARNING: Once a translation starts do not close it unless you want to lose your\
translated data. If a file fails or gets stuck, translated lines will remain translated so you don't have \
to worry about being charged twice. You can simply copy the file generated in /translations back over to \
/files and start the script again. It will skip over any translated text." + Fore.RESET, end='\n\n')

def main():
    estimate = ''
    while estimate == '':
        estimate = input('Select Translation or Cost Estimation:\n\n1. Translate\n2. Estimate\n')
        match estimate:
            case '1': estimate = False
            case '2': estimate = True
            case _: estimate = ''

    totalCost = 0
    version = ''
    while version == '':
        version = input('Select the RPGMaker Version:\n\n1. MV/MZ\n2. ACE\n3. CSV (From Translator++)\n4. Text (Custom)\n')
        match version:
            case '1':
                # Open File (Threads)
                with ThreadPoolExecutor(max_workers=THREADS) as executor:
                    futures = [executor.submit(handleMVMZ, filename, estimate) \
                                for filename in os.listdir("files") if filename.endswith('json')]
                    
                    for future in as_completed(futures):
                        try:
                            totalCost = future.result()
                        except Exception as e:
                            tracebackLineNo = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno)
                            print(Fore.RED + str(e) + '|' + tracebackLineNo + Fore.RESET)

            case '2':
                # Open File (Threads)
                with ThreadPoolExecutor(max_workers=THREADS) as executor:
                    futures = [executor.submit(handleACE, filename, estimate) \
                                for filename in os.listdir("files") if filename.endswith('yaml')]
                    
                    for future in as_completed(futures):
                        try:
                            totalCost = future.result()
                        except Exception as e:
                            tracebackLineNo = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno)
                            print(Fore.RED + str(e) + '|' + tracebackLineNo + Fore.RESET)

            case '3':
                # Open File (Threads)
                with ThreadPoolExecutor(max_workers=THREADS) as executor:
                    futures = [executor.submit(handleCSV, filename, estimate) \
                                for filename in os.listdir("files") if filename.endswith('csv')]
                    
                    for future in as_completed(futures):
                        try:
                            totalCost = future.result()
                            
                        except Exception as e:
                            tracebackLineNo = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno)
                            print(Fore.RED + str(e) + '|' + tracebackLineNo + Fore.RESET)

            case '4':
                # Open File (Threads)
                with ThreadPoolExecutor(max_workers=THREADS) as executor:
                    futures = [executor.submit(handleTextfile, filename, estimate) \
                                for filename in os.listdir("files") if filename.endswith('txt')]
                    
                    for future in as_completed(futures):
                        try:
                            totalCost = future.result()
                            
                        except Exception as e:
                            tracebackLineNo = str(traceback.extract_tb(sys.exc_info()[2])[-1].lineno)
                            print(Fore.RED + str(e) + '|' + tracebackLineNo + Fore.RESET)

            case _:
                version = ''
        
    if estimate == False:
        # This is to encourage people to grab what's in /translated instead
        deleteFolderFiles('files')

    # Prevent immediately closing of CLI
    print(totalCost)
    input('Done! Press Enter to close.')

def deleteFolderFiles(folderPath):
    for filename in os.listdir(folderPath):
        file_path = os.path.join(folderPath, filename)
        if file_path.endswith('.json') or file_path.endswith('.yaml'):
            os.remove(file_path)