try:
    from .utils import any_type
except ImportError:  # importable for tests without package context
    from utils import any_type


class ConditionalBypasser:
    """Server-side gate: passes input through when enabled, blocks downstream execution otherwise."""

    DESCRIPTION = (
        "When 'enabled' is true, passes the input through unchanged. "
        "When false, returns ExecutionBlocker so the downstream graph does not run. "
        "Accepts any data type."
    )
    SEARCH_ALIASES = ["bypass", "gate", "block", "switch", "mute", "if"]

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "value": (any_type, {"tooltip": "Any input. Type-checked dynamically."}),
                "enabled": ("BOOLEAN", {
                    "default": True,
                    "tooltip": "If false, downstream execution is blocked via ExecutionBlocker.",
                }),
            },
        }

    RETURN_TYPES = (any_type,)
    RETURN_NAMES = ("output",)
    FUNCTION = "gate"
    CATEGORY = "MoonPack/utils"

    def gate(self, value, enabled):
        if enabled:
            return (value,)
        try:
            from comfy_execution.graph import ExecutionBlocker
            return (ExecutionBlocker(None),)
        except ImportError:
            # Older ComfyUI versions without ExecutionBlocker — fall back to None.
            return (None,)


NODE_CLASS_MAPPINGS = {
    "MoonPack_ConditionalBypasser": ConditionalBypasser,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "MoonPack_ConditionalBypasser": "Conditional Bypasser",
}