switch.py

from __future__ import annotations

from enum import Enum

from api import KeyInfo
from nodes.groups import optional_list_group
from nodes.properties.inputs import AnyInput, EnumInput
from nodes.properties.outputs import BaseOutput
from nodes.utils.utils import ALPHABET

from .. import value_group


class ValueIndex(Enum):
    A = 0
    B = 1
    C = 2
    D = 3
    E = 4
    F = 5
    G = 6
    H = 7
    I = 8
    J = 9


@value_group.register(
    schema_id="chainner:utility:switch",
    name="Switch",
    description="Allows you to pass in multiple inputs and then change which one passes through to the output.",
    icon="BsShuffle",
    inputs=[
        EnumInput(ValueIndex).with_id(0),
        AnyInput("Value A").make_optional(),
        AnyInput("Value B").make_optional(),
        optional_list_group(
            *[AnyInput(f"Value {letter}").make_optional() for letter in ALPHABET[2:10]],
        ),
    ],
    outputs=[
        BaseOutput(
            output_type="""
            let value = match Input0 {
                ValueIndex::A => Input1,
                ValueIndex::B => Input2,
                ValueIndex::C => Input3,
                ValueIndex::D => Input4,
                ValueIndex::E => Input5,
                ValueIndex::F => Input6,
                ValueIndex::G => Input7,
                ValueIndex::H => Input8,
                ValueIndex::I => Input9,
                ValueIndex::J => Input10,
                _ => never
            };

            match value {
                null => never,
                _ => value
            }
            """,
            label="Value",
        ).with_never_reason("The selected value should have a connection.")
    ],
    key_info=KeyInfo.enum(0),
    see_also=["chainner:utility:pass_through"],
)
def switch_node(selection: ValueIndex, *args: object | None) -> object:
    if args[selection.value] is not None:
        return args[selection.value]
    raise RuntimeError(
        "Invalid value selected. The selected value should have a connection."
    )