From ec6e7785e5060aea6c43dac27333b915ce5bacc5 Mon Sep 17 00:00:00 2001
From: RandomityGuy <31925790+RandomityGuy@users.noreply.github.com>
Date: Sun, 18 Dec 2022 01:06:07 +0530
Subject: [PATCH] fix endgame

---
 src/gui/EndGameGui.hx   | 8 ++++++++
 src/gui/EnterNameDlg.hx | 5 +++++
 2 files changed, 13 insertions(+)

diff --git a/src/gui/EndGameGui.hx b/src/gui/EndGameGui.hx
index d29bdc5f..3029ad69 100644
--- a/src/gui/EndGameGui.hx
+++ b/src/gui/EndGameGui.hx
@@ -84,6 +84,12 @@ class EndGameGui extends GuiControl {
 		};
 		nextLevel.addChild(nextLevelBtn);
 
+		function setButtonStates(enabled:Bool) {
+			nextLevelBtn.disabled = !enabled;
+			continueButton.disabled = !enabled;
+			restartButton.disabled = !enabled;
+		}
+
 		var arial14fontdata = ResourceLoader.getFileEntry("data/font/arial.fnt");
 		var arial14b = new BitmapFont(arial14fontdata.entry);
 		@:privateAccess arial14b.loader = ResourceLoader.loader;
@@ -309,7 +315,9 @@ class EndGameGui extends GuiControl {
 		Settings.save();
 
 		if (idx <= 4) {
+			setButtonStates(false);
 			var end = new EnterNameDlg(idx, (name) -> {
+				setButtonStates(true);
 				if (scoreSubmitted)
 					return;
 
diff --git a/src/gui/EnterNameDlg.hx b/src/gui/EnterNameDlg.hx
index 0461b39e..9263ce6e 100644
--- a/src/gui/EnterNameDlg.hx
+++ b/src/gui/EnterNameDlg.hx
@@ -1,5 +1,7 @@
 package gui;
 
+import h2d.Tile;
+import hxd.BitmapData;
 import h2d.filter.DropShadow;
 import src.Settings;
 import hxd.res.BitmapFont;
@@ -57,6 +59,9 @@ class EnterNameDlg extends GuiControl {
 		enterNameEdit.position = new Vector(28, 130);
 		enterNameEdit.extent = new Vector(363, 38);
 		enterNameEdit.text.text = Settings.highscoreName;
+		haxe.Timer.delay(() -> {
+			enterNameEdit.text.focus();
+		}, 5);
 
 		var okbutton = new GuiButton(loadButtonImages("data/ui/endgame/ok"));
 		okbutton.position = new Vector(151, 184);