import time
import comfy.utils

class AnyType(str):
    def __ne__(self, __value: object) -> bool:
        return False

any_type = AnyType("*")

class add_delay_node:
    """Node that adds a configurable delay between operations"""

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "input": (any_type, {}),
                "delay_seconds": ("FLOAT", {
                    "default": 1.0,
                    "min": 0.0,
                    "step": 0.1
                }),
            },
        }

    RETURN_TYPES = (any_type,)
    RETURN_NAMES = ("output",)
    FUNCTION = "add_delay"
    CATEGORY = "utils"

    def add_delay(self, input, delay_seconds):
        delay_text = f"{delay_seconds:.1f} second{'s' if delay_seconds != 1 else ''}"
        print(f"[Delay Node] Starting delay of {delay_text}")

        progress_bar = comfy.utils.ProgressBar(delay_seconds)
        update_interval_seconds = 1.0

        start_time = time.monotonic()
        while True:
            current_elapsed = time.monotonic() - start_time

            progress_bar.update_absolute(min(current_elapsed, delay_seconds))

            if current_elapsed >= delay_seconds:
                # time is up
                break

            # Sleep one entire update interval, or the remaining fraction if we are at the end
            time_to_sleep = min(update_interval_seconds, delay_seconds - current_elapsed)
            time.sleep(time_to_sleep)

        print(f"[Delay Node] Delay of {delay_text} completed")
        return (input,)

NODE_CLASS_MAPPINGS = {
    "Add Delay": add_delay_node
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "Add Delay": "Add Delay"
}