from server import PromptServer  # type:ignore
import folder_paths  # type:ignore
import os
import threading
import time
import datetime


CATEGORY = "HotReload"


# ------------ Terminal ------------
class HotReload_Terminal:

    CATEGORY = CATEGORY

    @classmethod
    def INPUT_TYPES(s):
        return {"optional": {}}

    RETURN_TYPES = ()
    FUNCTION = "Func"

    def Func(self):
        pass


# file terminal checker
CHECKER_THREAD_NAME = "HotReload.LogFileChecker"


# func checker
def LogFileChecker(selfStamp):

    CheckedLine = 0
    CheckedDateSec = 0.0

    while True:
        try:

            # check for stop
            for thread in threading.enumerate():
                if thread.name == CHECKER_THREAD_NAME:
                    stamp = thread.__dict__.get("stamp")
                    if stamp > selfStamp:
                        return

            # check file
            port = PromptServer.instance.port if hasattr(PromptServer.instance, "port") else 8000

            logFilePath = os.path.join(folder_paths.get_user_directory(), f"comfyui_{port}.log")
            if os.path.isfile(logFilePath):
                fileDateSec = os.path.getmtime(logFilePath)
                if fileDateSec > CheckedDateSec + 0.1:
                    requareClear = CheckedLine == 0
                    CheckedDateSec = fileDateSec
                    resultText = ""
                    try:
                        # Try UTF-8 encoding first
                        with open(logFilePath, "r", encoding="utf-8") as file:
                            currentLine = 0
                            for line in file:
                                if currentLine > CheckedLine:
                                    CheckedLine = currentLine
                                    # 处理时间格式，移除毫秒部分
                                    if line.startswith("[20"):
                                        try:
                                            time_end = line.index("]")
                                            if time_end > 0:
                                                time_str = line[1:time_end]
                                                if "." in time_str:
                                                    time_str = time_str.split(".")[0]
                                                line = "[" + time_str + "]" + line[time_end+1:]
                                        except:
                                            pass
                                    resultText += line + "\n"
                                currentLine += 1
                    except UnicodeDecodeError:
                        # If UTF-8 fails, try with error handling
                        with open(logFilePath, "r", encoding="utf-8", errors="replace") as file:
                            currentLine = 0
                            for line in file:
                                if currentLine > CheckedLine:
                                    CheckedLine = currentLine
                                    # 处理时间格式，移除毫秒部分
                                    if line.startswith("[20"):
                                        try:
                                            time_end = line.index("]")
                                            if time_end > 0:
                                                time_str = line[1:time_end]
                                                if "." in time_str:
                                                    time_str = time_str.split(".")[0]
                                                line = "[" + time_str + "]" + line[time_end+1:]
                                        except:
                                            pass
                                    resultText += line + "\n"
                                currentLine += 1
                    PromptServer.instance.send_sync(
                        "/hotreload.terminal.log", {"text": resultText, "clear": requareClear}
                    )
        finally:
            # loop
            time.sleep(1)


# get current largest stamp
currentMaxStamp = 0
for thread in threading.enumerate():
    if thread.name == CHECKER_THREAD_NAME:
        stamp = thread.__dict__.get("stamp")
        currentMaxStamp = max(currentMaxStamp, stamp)

# start new checker
timestamp = max(int(datetime.datetime.now().timestamp()), currentMaxStamp + 1)
newThread = threading.Thread(target=LogFileChecker, name=CHECKER_THREAD_NAME, args=(timestamp,))
newThread.__dict__["stamp"] = timestamp
newThread.start()