button.tapHandler = { ... } を状態遷移に連動して書き換えるとかやらないこともないよ。