from .nodes import VACE_MODES


class VACEModeSelect:
    """Select a VACE mode by integer index (0-9)."""

    @classmethod
    def INPUT_TYPES(cls):
        mode_list = " | ".join(f"{i}={m}" for i, m in enumerate(VACE_MODES))
        return {
            "required": {
                "index": ("INT", {
                    "default": 0,
                    "min": 0,
                    "max": len(VACE_MODES) - 1,
                    "step": 1,
                    "tooltip": f"Mode index: {mode_list}",
                }),
            },
        }

    RETURN_TYPES = (VACE_MODES,)
    RETURN_NAMES = ("mode",)
    FUNCTION = "select"
    CATEGORY = "VACE Tools"
    DESCRIPTION = "Select a VACE mode by integer index (0-9). Useful when driving mode selection from another node's integer output instead of a dropdown."

    def select(self, index):
        index = max(0, min(index, len(VACE_MODES) - 1))
        return (VACE_MODES[index],)


NODE_CLASS_MAPPINGS = {
    "VACEModeSelect": VACEModeSelect,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "VACEModeSelect": "VACE Mode Select",
}