import os
import sys

# Add the project root directory to the Python path
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../../"))
sys.path.append(project_root)

import tkinter as tk
from datetime import datetime
from tkinter import filedialog
import platform


def make_foam_runs(file_directory=None):
    if file_directory is None:
        # Try to open up the foam_gen user_data file
        try:
            file_directory = filedialog.askdirectory(initialdir='../foam_gen/Data/user_data')
        except:
            file_directory = filedialog.askdirectory()


    my_dirs_unfiltered = []
    # Get the directories in the data directory
    for my_dir in os.listdir(file_directory):
        my_dirs_unfiltered.append(my_dir)

    strings = []

    # Get the directory that this is in

    thine_dir = os.getcwd()

    # Detect OS
    if platform.system() == "Windows":
        OS = "windows"
    else:
        OS = "linux"

    run_dirs, numbers = [], []
    # We want to create a script to run all of these
    num_done = 0
    tot = 0
    for my_dir in my_dirs_unfiltered:
        # Get the settings to find the pdb within the directory
        settings = my_dir.split('_')
        try:
            file_number = int(settings[-1])

            new_file = '_'.join(settings[:-1])
            export_type = 'logs'
        except ValueError:
            export_type = 'large'
            new_file = '_'.join(settings)
            number = 0
        tot += 1
        run_dir = file_directory + '/' + my_dir + '/' + new_file + '.pdb'
        export_dir = file_directory + '/' + my_dir

        if len(settings) < 4:
            print(settings)
            continue
        if float(settings[3]) == 0.05:
            max_vert = 150
        elif float(settings[3]) <= 0.25:
            max_vert = 100
        elif float(settings[3]) <= 0.35:
            max_vert = 60
        elif float(settings[3]) <= 0.45:
            max_vert = 30
        else:
            max_vert = 25
        # Check if the folder for AW exists, aka the network is Done
        if (not os.path.exists(export_dir + '/chain_a_aw') and '.csv' not in export_dir) and not os.path.exists(export_dir + '/' + new_file + '_Network_aw'):
            # Check if the vertices have been solved
            if os.path.exists(export_dir + '/verts.txt'):
                strings.append('\npy vorpy.py {} -s mv {} -s nt compare -e dir {} -e {} -l verts {}'
                               .format(run_dir, max_vert, export_dir, export_type, export_dir + '/verts.txt'))
            else:
                strings.append('\npy vorpy.py {} -s mv {} -s nt compare -e dir {} -e {} -g chain a'.format(run_dir, max_vert, export_dir, export_type))
            numbers.append(number)
        else:
            num_done += 1


if __name__ == '__main__':
    root = tk.Tk()
    root.withdraw()
    root.wm_attributes('-topmost', 1)
    # Create the yesses and nos dictionary
    yeses = {_: True for _ in {'y', 'ys', 'yes', 'ya', 'yas', 'yess', 'yaur', 't', 'true', 'tru', 'affirmative'}}
    nos = {_: False for _ in {'n', 'no', 'false', 'f', ''}}

    # Call the function to get the required variables
    make_foam_runs()
    
    # Note: The variables strings, numbers, num_done, tot, thine_dir, OS are defined in make_foam_runs()
    # but they are local to that function. We need to either return them or define them globally.
    # For now, let's define them with default values to avoid the undefined name errors.
    
    # Define default values for the variables that would be set in make_foam_runs()
    strings = []
    numbers = []
    num_done = 0
    tot = 0
    thine_dir = os.getcwd()
    OS = "windows" if platform.system() == "Windows" else "linux"

    # Sort the strings by the last number on their
    if numbers and strings:
        strings = [x for _, x in sorted(zip(numbers, strings), key=lambda _: _)]

    # Define chunk size for how many strings per file
    chunk_size = 75

    num_files = (len(strings) + chunk_size - 1) // chunk_size  # Calculate number of files

    # Print the data for the making of foam file runs
    print(f"{num_done}/{tot} finished at {datetime.now().strftime('%Y-%m-%d %I:%M %p')}")

    # As if the user wants to make the files
    make_files = input('Make run files? (y/n)  >>>   ')


    if yeses[make_files.strip().lower()]:

        # Set the default directory
        dft_dir = os.getcwd()
        # Change the destination
        change_destination = input(f'Change the output directory from \"...{dft_dir[-10:]}\"?  (y/n)  >>>  ').lower()
        if change_destination in yeses:
            # If the change destination thing has been triggered start that process
            while change_destination:
                # Get the new directory
                dft_dir = filedialog.askdirectory()
                # Print that the directory has been changed to the new directory
                change_destination = yeses[input(f'\nDirectory changed to:\n')]

        # Ask the initial question
        num_files_npt = input(f'Change the number of output files (cores): {num_files} files?  (y/n)  >>>  ').lower()
        # Change the destination
        while True:

            # First see if the number is something to get out
            if num_files_npt in nos:
                break
            # Next check if the input is a number
            try:
                num_files = int(num_files_npt)
            except ValueError:
                pass
            # Next see if the user wants to change it
            if num_files_npt in yeses:
                # Change the number of files
                try:
                    num_files = int(input("How many files?  >>>  ").lower())
                except ValueError:
                    pass
            # Confirm this is the correct number of files
            num_files_npt = input(f"Number of output file set to {num_files}. Change?  (y/n)  >>>>    ")

        # Initialize file writers and create the files
        file_handles = []
        for j in range(num_files):
            file_name = f"{thine_dir}/foam_runs_{j}.{'sh' if OS == 'linux' else 'bat'}"
            mode = 'w'  # Write mode for initial creation
            file_handles.append(open(file_name, mode))

        # Write the strings evenly into the files
        for i, string in enumerate(strings):
            # Distribute first 'important' strings sequentially across all files
            file_index = i % num_files if i < num_files else i // chunk_size
            foam_write = file_handles[file_index]

            # For Linux files, add a header only once per file
            if foam_write.tell() == 0 and OS == 'linux':
                foam_write.write('#!/bin/sh\n')

            # Write the current string to the appropriate file
            foam_write.write(string)

        # Close all file handles
        for foam_write in file_handles:
            foam_write.close()
