cycle pieces/markers when clicking the hide pieces button

1 files changed, 11 insertions, 1 deletions

diff --git a/play.js b/play.js
index 6948f1d..81531da 100644
--- a/play.js
+++ b/play.js
@@ -5,7 +5,17 @@
 */
 
 function toggle_pieces() {
-	document.getElementById("pieces").classList.toggle("hide")
+	// Cycle between showing everything, only pieces, and nothing.
+	let hidden_pieces = ui.pieces_element.classList.contains("hide")
+	let hidden_markers = ui.markers_element.classList.contains("hide")
+	if (hidden_pieces && hidden_markers) {
+		ui.pieces_element.classList.remove("hide")
+		ui.markers_element.classList.remove("hide")
+	} else if (hidden_pieces) {
+		ui.markers_element.classList.add("hide")
+	} else {
+		ui.pieces_element.classList.add("hide")
+	}
 }
 
 /* DATA */