如何在unity中切换Input.GetKeyDown？

enum InputState
{
    StateA,
    StateB,
    StateC
}

InputState currentInputState;

void Update()
{
    if (Input.GetKeyDown(KeyCode.Space))
    {
        // 切换到下一个输入状态
        switch (currentInputState)
        {
            case InputState.StateA:
                currentInputState = InputState.StateB;
                break;
            case InputState.StateB:
                currentInputState = InputState.StateC;
                break;
            case InputState.StateC:
                currentInputState = InputState.StateA;
                break;
        }
    }

    // 根据当前的输入状态执行相应的操作
    switch (currentInputState)
    {
        case InputState.StateA:
            // 执行状态A的操作
            break;
        case InputState.StateB:
            // 执行状态B的操作
            break;
        case InputState.StateC:
            // 执行状态C的操作
            break;
    }
}