import uuid


class TaskIdGenerate:
    """TaskID生成器：用户可传入或自动生成TaskID"""

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {},
            "optional": {
                "custom_task_id": ("STRING", {"default": "", "placeholder": "留空则自动生成"}),
            },
        }

    RETURN_TYPES = ("STRING",)
    RETURN_NAMES = ("task_id",)
    FUNCTION = "generate_task_id"
    OUTPUT_NODE = False
    CATEGORY = "utils"

    def generate_task_id(self, custom_task_id=""):
        if custom_task_id and custom_task_id.strip():
            # 用户输入了自定义ID
            task_id = custom_task_id.strip()
            print(f"📝 使用自定义TaskID: {task_id}")
        else:
            # 自动生成UUID
            task_id = str(uuid.uuid4())
            print(f"🎲 自动生成TaskID: {task_id}")
        
        return (task_id,)


NODE_CLASS_MAPPINGS = {
    "TaskIdGenerate": TaskIdGenerate
}
NODE_DISPLAY_NAME_MAPPINGS = {
    "TaskIdGenerate": "TaskID生成器"
}