Javascriptでボタンのクリックを待ち受ける方法

ボタンが押されるまで待ち受けて、押されたボタンの種類を返す処理

waitForClick() {
  return new Promise((resolve, reject) => {
    document.getElementById('button1').onclick = () => resolve('button1');
    document.getElementById('button2').onclick = () => resolve('button2');
  });
}

const result = await waitForClick();