如何编写javascript函数来回切换按钮颜色？

<button id="toggleButton" class="btn">按钮</button>

function toggleButtonColor() {
  var button = document.getElementById("toggleButton");
  if (button.classList.contains("btn")) {
    button.classList.remove("btn");
    button.classList.add("btn-active");
  } else {
    button.classList.remove("btn-active");
    button.classList.add("btn");
  }
}

var button = document.getElementById("toggleButton");
button.addEventListener("click", toggleButtonColor);

.btn {
  background-color: red;
  color: white;
}

.btn-active {
  background-color: green;
  color: black;
}