import datetime
import os
import subprocess
import sys
import atexit
import threading
import re

# Logger setup
if os.path.exists("comfyui.log"):
    if os.path.exists("comfyui.prev.log"):
       os.remove("comfyui.prev.log")
    os.rename("comfyui.log", "comfyui.prev.log")

original_stdout = sys.stdout
original_stderr = sys.stderr

tqdm = r'\d+%.*\[(.*?)\]'

class Logger:
    def __init__(self, filename):
        self.file = open(filename, "w", encoding="utf-8")

    def write(self, message):
        self.file.write(message)
        self.file.flush()

        match = re.search(tqdm, message)
        if match:
            message = re.sub(r'([#|])\d', r'\1▌', message)
            message = re.sub('#', '█', message)
            original_stderr.write(message)
            original_stderr.flush()
        else:
            original_stdout.write(message)
            original_stdout.flush()

    def flush(self):
        self.file.flush()
        original_stdout.flush()

    def close_file(self):
        self.file.close()


def handle_stream(stream, prefix):
    for line in stream:
        print(prefix, line, end="")


sys.stdout = Logger("comfyui.log")
sys.stderr = sys.stdout

atexit.register(sys.stdout.close_file)

print("** ComfyUI start up time:", datetime.datetime.now())


# Perform install
script_list_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "startup-scripts", "install-scripts.txt")

# Check if script_list_path exists
if os.path.exists(script_list_path):
    print("\n#######################################################################")
    print("[ComfyUI-Manager] Starting dependency installation/(de)activation for the extension\n")

    executed = set()
    # Read each line from the file and convert it to a list using eval
    with open(script_list_path, 'r') as file:
        for line in file:
            if line in executed:
                continue

            executed.add(line)

            try:
                script = eval(line)
                print(f"\n## ComfyUI-Manager: EXECUTE => {script[1:]}")

                print(f"\n## Execute install/(de)activation script for '{script[0]}'")
                process = subprocess.Popen(script[1:], cwd=script[0], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)

                stdout_thread = threading.Thread(target=handle_stream, args=(process.stdout, ""))
                stderr_thread = threading.Thread(target=handle_stream, args=(process.stderr, "[!]"))

                stdout_thread.start()
                stderr_thread.start()

                stdout_thread.join()
                stderr_thread.join()

                exit_code = process.wait()

                if exit_code != 0:
                    print(f"install/(de)activation script failed: {script[0]}")
            except Exception as e:
                print(f"[ERROR] Failed to execute install/(de)activation script: {line} / {e}")

    # Remove the script_list_path file
    if os.path.exists(script_list_path):
        os.remove(script_list_path)
        
    print("\n[ComfyUI-Manager] Startup script completed.")
    print("#######################################################################\n")