From 2b11620b5fce7abca3b7eaee580d096a280ff4b4 Mon Sep 17 00:00:00 2001
From: He4eT <He4eT@oddsquat.org>
Date: Sat, 7 Jun 2025 08:14:45 +0200
Subject: [PATCH] game: improve startscreen and gameover

---
 game.js | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/game.js b/game.js
index fa6f0c7..8424377 100644
--- a/game.js
+++ b/game.js
@@ -13,13 +13,13 @@ function TIC() {
 /* Stages */
 
 const gameStages = {
-  mainMenu,
-  gameover,
+  startscreen,
   gameplay,
+  gameover,
 }
 
 /** @type {keyof typeof gameStages} */
-let currentStage = 'mainMenu'
+let currentStage = 'startscreen'
 
 /* State */
 
@@ -57,8 +57,8 @@ let effects = []
 
 /* Main Menu */
 
-function mainMenu() {
-  if ([BTN_A, BTN_B, BTN_X, BTN_Y].map(btn).some(Boolean)) {
+function startscreen() {
+  if ([BTN_A, BTN_B, BTN_X, BTN_Y].map((b) => btnp(b)).some(Boolean)) {
     currentStage = 'gameplay'
   }
 
@@ -76,11 +76,15 @@ function mainMenu() {
 function gameover() {
   cls(0)
 
+  if ([BTN_A, BTN_B, BTN_X, BTN_Y].map((b) => btnp(b)).some(Boolean)) {
+    reset()
+  }
+
   drawEnemies()
   drawPlayer()
 
   const title = 'Game Over'
-  print(title, 12, 12, 10, false, 2)
+  print(title, 12, SCREEN_H - 24, 10, false, 2)
 }
 
 /* Gameplay */