import os

def save_file(filename, output_dir, content, mode='Overwrite', file_mode='Windows'):
    os.makedirs(output_dir, exist_ok=True)
    file_path = os.path.join(output_dir, filename)

    if file_mode == 'Linux|Unix':
        content = content.replace('\r\n', '\n').replace('\r', '\n')

    if mode == 'Overwrite':
        with open(file_path, 'w') as file:
            file.write(content)
    elif mode == 'Merge' and os.path.exists(file_path):
        with open(file_path, 'a') as file:
            file.write(content)
    else:
        counter = 0
        while os.path.exists(file_path):
            counter += 1
            file_path = os.path.join(output_dir, f"{os.path.splitext(filename)[0]}_{counter}{os.path.splitext(filename)[1]}")
        with open(file_path, 'w') as file:
            file.write(content)
    
    print(f"File saved successfully at {file_path}")

class DataSet_TextFilesSave:

    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "TextFileNames": ("STRING",{"forceInput": True}),
                "TextFileContents": ("STRING",{"forceInput": True}),
                "destination": ("STRING", {"default": "directory path"}),
                "save_mode": (['Overwrite','Merge','SaveNew','MergeAndSaveNew'],),
                "file_mode": (['Windows','Linux|Unix'],),
            },
        }

    INPUT_IS_LIST = True
    RETURN_TYPES = ()
    FUNCTION = "SaveIT"
    OUTPUT_NODE = True

    CATEGORY = "🔶DATASET🔶"

    def SaveIT(self, TextFileNames, TextFileContents, destination, save_mode, file_mode):
        try:

            directory = destination[0]
            mode = save_mode[0]
            fmode = file_mode[0]

            if not os.path.exists(directory):
                os.makedirs(directory)

            for i in range(0, len(TextFileContents)):
                text = TextFileContents[i]
                file_name = TextFileNames[i]
                save_file(f"{file_name}.txt", directory, text, mode, fmode)

        except Exception as e:
            print(f"Error saving: {e}")

        return ()

N_CLASS_MAPPINGS = {
    "DataSet_TextFilesSave": DataSet_TextFilesSave,
}

N_DISPLAY_NAME_MAPPINGS = {
    "DataSet_TextFilesSave": "DataSet_TextFilesSave",
}