From 916a06b9289ab5d0fd79e91623faa9a085e93cd7 Mon Sep 17 00:00:00 2001
From: Harrison Bruhl <harrisonbruhl@gmail.com>
Date: Thu, 5 Mar 2026 21:01:31 -0500
Subject: [PATCH 1/2] feat(ui): add game over restart screen controller

---
 Assets/_Project/Art/UI/GameOverScreen.meta    |   8 ++
 .../Art/UI/GameOverScreen/RestartButtonUI.png | Bin 0 -> 1139 bytes
 .../GameOverScreen/RestartButtonUI.png.meta   | 127 ++++++++++++++++++
 Assets/_Project/Scenes/MainPrototype.unity    |  30 ++++-
 Assets/_Project/Scenes/SampleScene.unity      |  20 ++-
 .../_Project.Gameplay/Managers/GameManager.cs |  55 +++++++-
 .../UI/GameOverScreenController.cs            | 123 +++++++++++++++++
 .../UI/GameOverScreenController.cs.meta       |  11 ++
 8 files changed, 363 insertions(+), 11 deletions(-)
 create mode 100644 Assets/_Project/Art/UI/GameOverScreen.meta
 create mode 100644 Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png
 create mode 100644 Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png.meta
 create mode 100644 Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs
 create mode 100644 Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs.meta

diff --git a/Assets/_Project/Art/UI/GameOverScreen.meta b/Assets/_Project/Art/UI/GameOverScreen.meta
new file mode 100644
index 0000000..ab700c5
--- /dev/null
+++ b/Assets/_Project/Art/UI/GameOverScreen.meta
@@ -0,0 +1,8 @@
+fileFormatVersion: 2
+guid: 366ce872bdd59184ba04a24fd12b4677
+folderAsset: yes
+DefaultImporter:
+  externalObjects: {}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png b/Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png
new file mode 100644
index 0000000000000000000000000000000000000000..6f2227706c68ff178147b71f72728a042271cda2
GIT binary patch
literal 1139
zcmV-(1dRKMP)<h;3K|Lk000e1NJLTq00961002M;1^@s6^<AmH00009a7bBm001r{
z001r{0eGc9b^rhbB}qg<RCodH-9JndVI0QsBWWTNgQH<^z(fa)4lV?u3?>d`(&)xW
zA_)<RgK=;)3<fbIXc*ZvI@uTvgBXpAF`6(yf`bDygA*J2oWAuaw7o0$=l%2jEVZqH
za`*n8_r2?T@71C|-n`q~ivW0SX>oqv^G0L2ucqSv{OM!rAM*X%m+1V3i+O)GXXx)Y
zLf#Ica=HCaP0#KNfqH#%uU6sbagKKWkFAa9`i+U`<%^a0HCG!0vU9ok+vzC)eB2P-
zB#U44m>dnjZ)X4d*6uS0z}ot1w7av@eD2!@>BJPjp#~u#ZG#ZChQCiT1%Nh$ivP~9
z_tE2lv1sVTx%l-E3gEE6(Z>dXgUr6yVV`gNJaqt=6zo5KJT87cZtreK(bnJhLkIzc
z%h&duX#}8xpP9IS=zAfa7XVzBDF6oh2Rpt`2*77Pw(Z=$2GqvktT^D7OaU-->P%6e
z13(2*2SC~$p2GmZ9n1s-0+wq4m{9_w0Kg5I<^iNZ;REi-Gy<Twf!?%sptjT!=Kz2b
zv=)4~5p`JsktzT<rU&xXt)9wScY+ZBU?CN~S#v_(es~@~UcGcP*FfkR0L<<Ks$s4?
zR1oOmJ6(xuf*`3~Ahg!*@W|+a|C4G2!0;caj)Gur;ck2xzPvttyZN4G=XUC(yVY*T
z0N~gku*1EXN73BCSR4Z5n@{4`bXk6UN&Y|PV!bSD7PJijhW>zko}RoG-TL{e`Mllb
z`Dt|=0nqyccF-Y!kvs%As<r_@@dI`-=KwawpT(<l>9F9aoctP)a}x-aieK_to6xxc
zp!gB8yxYE00l<k#gi0%LtM?B&0EFHj$d5Atz|bF|Qrp*rpCa1769K@`AE8p)*Ysaq
z15iXK=tKYry+1-?Iz1J?xf7Q{AaBkD01yIG`1+n-(|mOi(6I+!<HsTp0DvXT?f0bh
zdwxy%E#j8+)(3#Ew`B$l)9b5^m)*{-4*(zpX$s$r3i3P|NB{s8N%s6zu{r?E@ZYz4
zvIEj<;RkgNAS)YE6>x>s0kF2d+Vz&?Nz4`mvavb<B)Q~KJM{WMZ>$Xf6}47D?KS~|
z;s<(XZ2%;vKqf7$JZv5o6m@|F5DIGpz)%QiHPqxtfWimzVPyblSg(RNy}$l_`Ro;n
z9>|$>0bq{lRq#U51I4i}01OHrD4vx8K=A|RurdGul*6h3U?v|>E^7jSb^n0!SrY(4
z?+=vAiU45f4^+d7002;p$^syH%x`P&&&UM6t0Vxpf)B7mX#wCuj}P}TYRjgS0C2I#
zXE|~=nRWHqZ|M{s04^6li;=oV`>co?X!xqOv^XD4P0tqg3x-EV3;Wqcaq%<bKlb0@
zRO_v3?cM<3Kjr}d0V4o_fYlm}<-U4-a<A$S0-#)~{oleNjJQwFE@%J%002ovPDHLk
FV1o3B6LbIo

literal 0
HcmV?d00001

diff --git a/Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png.meta b/Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png.meta
new file mode 100644
index 0000000..2784b50
--- /dev/null
+++ b/Assets/_Project/Art/UI/GameOverScreen/RestartButtonUI.png.meta
@@ -0,0 +1,127 @@
+fileFormatVersion: 2
+guid: b2831b7f6899ee944a13fa562e3b84df
+TextureImporter:
+  internalIDToNameTable: []
+  externalObjects: {}
+  serializedVersion: 13
+  mipmaps:
+    mipMapMode: 0
+    enableMipMap: 0
+    sRGBTexture: 1
+    linearTexture: 0
+    fadeOut: 0
+    borderMipMap: 0
+    mipMapsPreserveCoverage: 0
+    alphaTestReferenceValue: 0.5
+    mipMapFadeDistanceStart: 1
+    mipMapFadeDistanceEnd: 3
+  bumpmap:
+    convertToNormalMap: 0
+    externalNormalMap: 0
+    heightScale: 0.25
+    normalMapFilter: 0
+    flipGreenChannel: 0
+  isReadable: 0
+  streamingMipmaps: 0
+  streamingMipmapsPriority: 0
+  vTOnly: 0
+  ignoreMipmapLimit: 0
+  grayScaleToAlpha: 0
+  generateCubemap: 6
+  cubemapConvolution: 0
+  seamlessCubemap: 0
+  textureFormat: 1
+  maxTextureSize: 2048
+  textureSettings:
+    serializedVersion: 2
+    filterMode: 0
+    aniso: 1
+    mipBias: 0
+    wrapU: 1
+    wrapV: 1
+    wrapW: 1
+  nPOTScale: 0
+  lightmap: 0
+  compressionQuality: 50
+  spriteMode: 1
+  spriteExtrude: 1
+  spriteMeshType: 1
+  alignment: 0
+  spritePivot: {x: 0.5, y: 0.5}
+  spritePixelsToUnits: 32
+  spriteBorder: {x: 0, y: 0, z: 0, w: 0}
+  spriteGenerateFallbackPhysicsShape: 1
+  alphaUsage: 1
+  alphaIsTransparency: 1
+  spriteTessellationDetail: -1
+  textureType: 8
+  textureShape: 1
+  singleChannelComponent: 0
+  flipbookRows: 1
+  flipbookColumns: 1
+  maxTextureSizeSet: 0
+  compressionQualitySet: 0
+  textureFormatSet: 0
+  ignorePngGamma: 0
+  applyGammaDecoding: 0
+  swizzle: 50462976
+  cookieLightType: 0
+  platformSettings:
+  - serializedVersion: 3
+    buildTarget: DefaultTexturePlatform
+    maxTextureSize: 2048
+    resizeAlgorithm: 0
+    textureFormat: -1
+    textureCompression: 0
+    compressionQuality: 50
+    crunchedCompression: 0
+    allowsAlphaSplitting: 0
+    overridden: 0
+    ignorePlatformSupport: 0
+    androidETC2FallbackOverride: 0
+    forceMaximumCompressionQuality_BC6H_BC7: 0
+  - serializedVersion: 3
+    buildTarget: Standalone
+    maxTextureSize: 2048
+    resizeAlgorithm: 0
+    textureFormat: -1
+    textureCompression: 1
+    compressionQuality: 50
+    crunchedCompression: 0
+    allowsAlphaSplitting: 0
+    overridden: 0
+    ignorePlatformSupport: 0
+    androidETC2FallbackOverride: 0
+    forceMaximumCompressionQuality_BC6H_BC7: 0
+  - serializedVersion: 3
+    buildTarget: Android
+    maxTextureSize: 2048
+    resizeAlgorithm: 0
+    textureFormat: -1
+    textureCompression: 1
+    compressionQuality: 50
+    crunchedCompression: 0
+    allowsAlphaSplitting: 0
+    overridden: 0
+    ignorePlatformSupport: 0
+    androidETC2FallbackOverride: 0
+    forceMaximumCompressionQuality_BC6H_BC7: 0
+  spriteSheet:
+    serializedVersion: 2
+    sprites: []
+    outline: []
+    physicsShape: []
+    bones: []
+    spriteID: 5e97eb03825dee720800000000000000
+    internalID: 0
+    vertices: []
+    indices: 
+    edges: []
+    weights: []
+    secondaryTextures: []
+    nameFileIdTable: {}
+  mipmapLimitGroupName: 
+  pSDRemoveMatte: 0
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/_Project/Scenes/MainPrototype.unity b/Assets/_Project/Scenes/MainPrototype.unity
index cd76348..66ee5cd 100644
--- a/Assets/_Project/Scenes/MainPrototype.unity
+++ b/Assets/_Project/Scenes/MainPrototype.unity
@@ -2327,6 +2327,7 @@ GameObject:
   - component: {fileID: 904751786}
   - component: {fileID: 904751788}
   - component: {fileID: 904751787}
+  - component: {fileID: 904751789}
   m_Layer: 5
   m_Name: GameOver
   m_TagString: Untagged
@@ -2379,7 +2380,7 @@ MonoBehaviour:
 
 
 
-    Press Space - Restart'
+    Tap Restart (Press Space)'
   m_isRightToLeft: 0
   m_fontAsset: {fileID: 11400000, guid: 8f586378b4e144a9851e7b34d9b748ee, type: 2}
   m_sharedMaterial: {fileID: 2180264, guid: 8f586378b4e144a9851e7b34d9b748ee, type: 2}
@@ -2393,8 +2394,8 @@ MonoBehaviour:
   m_enableVertexGradient: 1
   m_colorMode: 3
   m_fontColorGradient:
-    topLeft: {r: 1, g: 1, b: 1, a: 1}
-    topRight: {r: 1, g: 1, b: 1, a: 1}
+    topLeft: {r: 0, g: 0, b: 0, a: 0.87058824}
+    topRight: {r: 0.8113208, g: 0.16456035, b: 0.16456035, a: 1}
     bottomLeft: {r: 0.4528302, g: 0.41651833, b: 0.41651833, a: 1}
     bottomRight: {r: 0.5188679, g: 0.5017355, b: 0.5017355, a: 1}
   m_fontColorGradientPreset: {fileID: 0}
@@ -2456,6 +2457,22 @@ CanvasRenderer:
   m_PrefabAsset: {fileID: 0}
   m_GameObject: {fileID: 904751785}
   m_CullTransparentMesh: 1
+--- !u!114 &904751789
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 904751785}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: cf6ef566c2404d7caf4f7f8838fd3e9d, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  restartButton: {fileID: 0}
+  restartButtonSprite: {fileID: 21300000, guid: b2831b7f6899ee944a13fa562e3b84df, type: 3}
+  restartButtonText: Rise Again
+  restartButtonTextColor: {r: 0.84, g: 0.95, b: 1, a: 1}
 --- !u!1 &977390122
 GameObject:
   m_ObjectHideFlags: 0
@@ -3787,8 +3804,8 @@ MonoBehaviour:
   m_Enabled: 1
   m_EditorHideFlags: 0
   m_Script: {fileID: 11500000, guid: fe87c0e1cc204ed48ad3b37840f39efc, type: 3}
-  m_Name:
-  m_EditorClassIdentifier:
+  m_Name: 
+  m_EditorClassIdentifier: 
   m_Material: {fileID: 0}
   m_Color: {r: 0.8018868, g: 0.22316661, b: 0.22316661, a: 0}
   m_RaycastTarget: 0
@@ -4694,6 +4711,8 @@ MonoBehaviour:
   movementZone: {fileID: 3001000004}
   aimAttackZone: {fileID: 3002000004}
   repairButton: {fileID: 3003000004}
+  playerInput: {fileID: 0}
+  baseAttackRate: 1.5
 --- !u!114 &3000000003
 MonoBehaviour:
   m_ObjectHideFlags: 0
@@ -4797,6 +4816,7 @@ MonoBehaviour:
   m_Script: {fileID: 11500000, guid: d7f83b5a6c1a8934d915b999993b5326, type: 3}
   m_Name: 
   m_EditorClassIdentifier: 
+  maxRadius: 75
 --- !u!1 &3002000000
 GameObject:
   m_ObjectHideFlags: 0
diff --git a/Assets/_Project/Scenes/SampleScene.unity b/Assets/_Project/Scenes/SampleScene.unity
index 436999a..81bb9f9 100644
--- a/Assets/_Project/Scenes/SampleScene.unity
+++ b/Assets/_Project/Scenes/SampleScene.unity
@@ -1259,6 +1259,7 @@ GameObject:
   - component: {fileID: 904751786}
   - component: {fileID: 904751788}
   - component: {fileID: 904751787}
+  - component: {fileID: 904751789}
   m_Layer: 5
   m_Name: GameOver
   m_TagString: Untagged
@@ -1311,7 +1312,7 @@ MonoBehaviour:
 
 
 
-    Press Space - Restart'
+    Tap Restart (Space also works)'
   m_isRightToLeft: 0
   m_fontAsset: {fileID: 11400000, guid: 8f586378b4e144a9851e7b34d9b748ee, type: 2}
   m_sharedMaterial: {fileID: 2180264, guid: 8f586378b4e144a9851e7b34d9b748ee, type: 2}
@@ -1388,6 +1389,22 @@ CanvasRenderer:
   m_PrefabAsset: {fileID: 0}
   m_GameObject: {fileID: 904751785}
   m_CullTransparentMesh: 1
+--- !u!114 &904751789
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 904751785}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: cf6ef566c2404d7caf4f7f8838fd3e9d, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  restartButton: {fileID: 0}
+  restartButtonSprite: {fileID: 0}
+  restartButtonText: Rise Again
+  restartButtonTextColor: {r: 0.84, g: 0.95, b: 1, a: 1}
 --- !u!1 &969171467
 GameObject:
   m_ObjectHideFlags: 0
@@ -3207,3 +3224,4 @@ SceneRoots:
   - {fileID: 833814061}
   - {fileID: 1027969717}
   - {fileID: 751719342}
+
diff --git a/Assets/_Project/Scripts/_Project.Gameplay/Managers/GameManager.cs b/Assets/_Project/Scripts/_Project.Gameplay/Managers/GameManager.cs
index 0610749..ca2eb45 100644
--- a/Assets/_Project/Scripts/_Project.Gameplay/Managers/GameManager.cs
+++ b/Assets/_Project/Scripts/_Project.Gameplay/Managers/GameManager.cs
@@ -1,5 +1,9 @@
 using UnityEngine;
 using UnityEngine.SceneManagement;
+using Castlebound.Gameplay.UI;
+#if ENABLE_INPUT_SYSTEM
+using UnityEngine.InputSystem;
+#endif
 using Random = System.Random;
 
 public class GameManager : MonoBehaviour
@@ -10,31 +14,72 @@ public class GameManager : MonoBehaviour
     public Random LootRng { get; private set; }
 
     bool gameOver;
+    GameOverScreenController gameOverScreenController;
 
     void Awake()
     {
         if (I && I != this) { Destroy(gameObject); return; }
         I = this;
         LootRng = new Random();
+        ResolveGameOverScreenController();
     }
 
     public void OnPlayerDied()
     {
         if (gameOver) return;
         gameOver = true;
-        if (gameOverUI) gameOverUI.SetActive(true);
+
+        if (gameOverScreenController != null)
+        {
+            gameOverScreenController.Show();
+        }
+        else if (gameOverUI)
+        {
+            gameOverUI.SetActive(true);
+        }
+
         // Freeze time if you like:
         // Time.timeScale = 0f;
     }
 
+    public void RequestRestart()
+    {
+        // Time.timeScale = 1f;
+        SceneManager.LoadScene(SceneManager.GetActiveScene().buildIndex);
+    }
+
     void Update()
     {
         if (!gameOver) return;
-        // Space = restart scene
-        if (Input.GetKeyDown(KeyCode.Space))
+
+        if (WasRestartKeyPressed())
         {
-            // Time.timeScale = 1f;
-            SceneManager.LoadScene(SceneManager.GetActiveScene().buildIndex);
+            RequestRestart();
         }
     }
+
+    bool WasRestartKeyPressed()
+    {
+#if ENABLE_INPUT_SYSTEM
+        if (Keyboard.current != null && Keyboard.current.spaceKey.wasPressedThisFrame)
+            return true;
+#endif
+#if ENABLE_LEGACY_INPUT_MANAGER
+        if (Input.GetKeyDown(KeyCode.Space))
+            return true;
+#endif
+        return false;
+    }
+
+    void ResolveGameOverScreenController()
+    {
+        if (gameOverUI == null)
+            return;
+
+        gameOverScreenController = gameOverUI.GetComponent<GameOverScreenController>();
+        if (gameOverScreenController == null)
+            gameOverScreenController = gameOverUI.AddComponent<GameOverScreenController>();
+
+        gameOverScreenController.SetRestartHandler(RequestRestart);
+    }
 }
diff --git a/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs b/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs
new file mode 100644
index 0000000..8775ffe
--- /dev/null
+++ b/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs
@@ -0,0 +1,123 @@
+using TMPro;
+using UnityEngine;
+using UnityEngine.Events;
+using UnityEngine.UI;
+
+namespace Castlebound.Gameplay.UI
+{
+    public class GameOverScreenController : MonoBehaviour
+    {
+        [SerializeField] private Button restartButton;
+        [SerializeField] private Sprite restartButtonSprite;
+        [SerializeField] private string restartButtonText = "Rise Again";
+        [SerializeField] private Color restartButtonTextColor = new Color(0.84f, 0.95f, 1f, 1f);
+
+        private UnityAction restartHandler;
+
+        private void OnEnable()
+        {
+            EnsureRestartButton();
+            ApplyVisuals();
+            BindRestartHandler();
+        }
+
+        private void OnDisable()
+        {
+            if (restartButton != null && restartHandler != null)
+            {
+                restartButton.onClick.RemoveListener(restartHandler);
+            }
+        }
+
+        public void SetRestartHandler(UnityAction handler)
+        {
+            if (restartButton != null && restartHandler != null)
+            {
+                restartButton.onClick.RemoveListener(restartHandler);
+            }
+
+            restartHandler = handler;
+            EnsureRestartButton();
+            ApplyVisuals();
+            BindRestartHandler();
+        }
+
+        public void Show()
+        {
+            gameObject.SetActive(true);
+        }
+
+        private void BindRestartHandler()
+        {
+            if (restartButton == null || restartHandler == null)
+            {
+                return;
+            }
+
+            restartButton.onClick.RemoveListener(restartHandler);
+            restartButton.onClick.AddListener(restartHandler);
+        }
+
+        private void EnsureRestartButton()
+        {
+            if (restartButton == null)
+            {
+                restartButton = GetComponentInChildren<Button>(true);
+            }
+
+            if (restartButton == null)
+            {
+                restartButton = CreateDefaultRestartButton(transform);
+            }
+        }
+
+        private static Button CreateDefaultRestartButton(Transform parent)
+        {
+            var buttonObject = new GameObject("RestartButton", typeof(RectTransform), typeof(Image), typeof(Button));
+            buttonObject.transform.SetParent(parent, false);
+
+            var rect = buttonObject.GetComponent<RectTransform>();
+            rect.anchorMin = new Vector2(0.5f, 0.2f);
+            rect.anchorMax = new Vector2(0.5f, 0.2f);
+            rect.pivot = new Vector2(0.5f, 0.5f);
+            rect.sizeDelta = new Vector2(300f, 92f);
+
+            return buttonObject.GetComponent<Button>();
+        }
+
+        private void ApplyVisuals()
+        {
+            if (restartButton == null)
+            {
+                return;
+            }
+
+            var image = restartButton.GetComponent<Image>();
+            if (image != null && restartButtonSprite != null)
+            {
+                image.sprite = restartButtonSprite;
+                image.type = Image.Type.Simple;
+                image.preserveAspect = true;
+            }
+
+            var label = restartButton.GetComponentInChildren<TextMeshProUGUI>(true);
+            if (label == null)
+            {
+                var labelObject = new GameObject("Label", typeof(RectTransform), typeof(TextMeshProUGUI));
+                labelObject.transform.SetParent(restartButton.transform, false);
+                label = labelObject.GetComponent<TextMeshProUGUI>();
+            }
+
+            var labelRect = label.rectTransform;
+            labelRect.anchorMin = Vector2.zero;
+            labelRect.anchorMax = Vector2.one;
+            labelRect.offsetMin = Vector2.zero;
+            labelRect.offsetMax = Vector2.zero;
+
+            label.alignment = TextAlignmentOptions.Center;
+            label.fontSize = 30f;
+            label.text = string.IsNullOrWhiteSpace(restartButtonText) ? "Rise Again" : restartButtonText;
+            label.color = restartButtonTextColor;
+        }
+    }
+}
diff --git a/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs.meta b/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs.meta
new file mode 100644
index 0000000..d3ef420
--- /dev/null
+++ b/Assets/_Project/Scripts/_Project.Gameplay/UI/GameOverScreenController.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: cf6ef566c2404d7caf4f7f8838fd3e9d
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 

From 3478212666e70355b231be72c34bf751e2e9cfb8 Mon Sep 17 00:00:00 2001
From: Harrison Bruhl <harrisonbruhl@gmail.com>
Date: Thu, 5 Mar 2026 21:03:40 -0500
Subject: [PATCH 2/2] test(ui): add game over restart regression coverage

---
 ...ameManagerKeyboardRestartContractsTests.cs | 21 +++++++
 ...nagerKeyboardRestartContractsTests.cs.meta | 11 ++++
 .../UI/GameOverRestartContractsTests.cs       | 53 ++++++++++++++++++
 .../UI/GameOverRestartContractsTests.cs.meta  | 11 ++++
 .../Smoke/DeathScreenRestartPlayTests.cs      | 56 +++++++++++++++++++
 .../Smoke/DeathScreenRestartPlayTests.cs.meta | 11 ++++
 Docs/TEST_LOG.md                              | 18 ++++++
 7 files changed, 181 insertions(+)
 create mode 100644 Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs
 create mode 100644 Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs.meta
 create mode 100644 Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs
 create mode 100644 Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs.meta
 create mode 100644 Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs
 create mode 100644 Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs.meta

diff --git a/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs b/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs
new file mode 100644
index 0000000..bbb7fd4
--- /dev/null
+++ b/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs
@@ -0,0 +1,21 @@
+using System.IO;
+using NUnit.Framework;
+
+namespace Castlebound.Tests.Input
+{
+    public class GameManagerKeyboardRestartContractsTests
+    {
+        private const string GameManagerPath = "Assets/_Project/Scripts/_Project.Gameplay/Managers/GameManager.cs";
+
+        [Test]
+        public void SpaceRestartContract_IsPresentForInputSystemAndLegacyInput()
+        {
+            var source = File.ReadAllText(GameManagerPath);
+
+            StringAssert.Contains("Keyboard.current.spaceKey.wasPressedThisFrame", source,
+                "GameManager should support Space restart through the new Input System.");
+            StringAssert.Contains("Input.GetKeyDown(KeyCode.Space)", source,
+                "GameManager should retain legacy Space restart fallback.");
+        }
+    }
+}
diff --git a/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs.meta b/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs.meta
new file mode 100644
index 0000000..f044ae1
--- /dev/null
+++ b/Assets/_Project/_Tests/EditMode/Input/GameManagerKeyboardRestartContractsTests.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: f9cd42d968bff1e40a4a10d44597822e
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs b/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs
new file mode 100644
index 0000000..c6c1f71
--- /dev/null
+++ b/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs
@@ -0,0 +1,53 @@
+using System.IO;
+using System.Reflection;
+using NUnit.Framework;
+
+namespace Castlebound.Tests.UI
+{
+    public class GameOverRestartContractsTests
+    {
+        private static readonly string[] ScenePaths =
+        {
+            "Assets/_Project/Scenes/MainPrototype.unity",
+            "Assets/_Project/Scenes/SampleScene.unity"
+        };
+
+        [Test]
+        public void GameManager_ExposesUiRestartEntryPoint()
+        {
+            var method = typeof(GameManager).GetMethod(
+                "RequestRestart",
+                BindingFlags.Instance | BindingFlags.Public);
+
+            Assert.IsNotNull(method,
+                "GameManager must expose RequestRestart() so UI buttons and keyboard input can trigger one shared restart path.");
+        }
+
+        [Test]
+        public void DeathScreenPrompt_DoesNotUseSpaceOnlyCopy()
+        {
+            foreach (var scenePath in ScenePaths)
+            {
+                var text = File.ReadAllText(scenePath);
+                StringAssert.DoesNotContain("Press Space - Restart", text,
+                    $"Scene '{scenePath}' should not ship with a spacebar-only restart prompt.");
+            }
+        }
+
+        [Test]
+        public void DeathScreenPrompt_UsesTouchFriendlyRestartCopy()
+        {
+            foreach (var scenePath in ScenePaths)
+            {
+                var text = File.ReadAllText(scenePath);
+                var hasTouchCopy =
+                    text.Contains("Tap Restart") ||
+                    text.Contains("Tap to Restart") ||
+                    text.Contains("Tap the Restart button");
+
+                Assert.IsTrue(hasTouchCopy,
+                    $"Scene '{scenePath}' should include touch-friendly restart copy.");
+            }
+        }
+    }
+}
diff --git a/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs.meta b/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs.meta
new file mode 100644
index 0000000..5f043de
--- /dev/null
+++ b/Assets/_Project/_Tests/EditMode/UI/GameOverRestartContractsTests.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: 6905730abdc33ad4a804a5c51281d6bc
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs b/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs
new file mode 100644
index 0000000..0ea8d31
--- /dev/null
+++ b/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs
@@ -0,0 +1,56 @@
+using System.Collections;
+using System.Linq;
+using System.Reflection;
+using NUnit.Framework;
+using UnityEngine;
+using UnityEngine.SceneManagement;
+using UnityEngine.TestTools;
+
+namespace Castlebound.Tests.PlayMode.UI
+{
+    public class DeathScreenRestartPlayTests
+    {
+        [UnityTest]
+        public IEnumerator OnPlayerDied_GameOverUiContainsTappableRestartButton()
+        {
+            yield return LoadMainPrototype();
+
+            var manager = Object.FindObjectOfType<GameManager>();
+            Assert.NotNull(manager, "Expected GameManager in MainPrototype.");
+
+            manager.OnPlayerDied();
+
+            var gameOverUi = GetGameOverUi(manager);
+            Assert.NotNull(gameOverUi, "GameManager should reference a game-over UI object.");
+            Assert.IsTrue(gameOverUi.activeSelf, "Game-over UI should be active after player death.");
+
+            var buttonComponents = gameOverUi
+                .GetComponentsInChildren<MonoBehaviour>(true)
+                .Where(component => component.GetType().Name == "Button")
+                .ToArray();
+
+            Assert.That(buttonComponents.Length, Is.GreaterThan(0),
+                "Death screen should include a tappable Restart button.");
+        }
+
+        private static IEnumerator LoadMainPrototype()
+        {
+            var load = SceneManager.LoadSceneAsync("MainPrototype", LoadSceneMode.Single);
+            while (!load.isDone)
+            {
+                yield return null;
+            }
+
+            yield return null;
+        }
+
+        private static GameObject GetGameOverUi(GameManager manager)
+        {
+            var field = typeof(GameManager).GetField(
+                "gameOverUI",
+                BindingFlags.Instance | BindingFlags.NonPublic);
+
+            return field?.GetValue(manager) as GameObject;
+        }
+    }
+}
diff --git a/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs.meta b/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs.meta
new file mode 100644
index 0000000..7f5fc81
--- /dev/null
+++ b/Assets/_Project/_Tests/PlayMode/Smoke/DeathScreenRestartPlayTests.cs.meta
@@ -0,0 +1,11 @@
+fileFormatVersion: 2
+guid: a8e74c53ea5c49d43971b4fa1595cf4c
+MonoImporter:
+  externalObjects: {}
+  serializedVersion: 2
+  defaultReferences: []
+  executionOrder: 0
+  icon: {instanceID: 0}
+  userData: 
+  assetBundleName: 
+  assetBundleVariant: 
diff --git a/Docs/TEST_LOG.md b/Docs/TEST_LOG.md
index d94a8df..8b8d1e8 100644
--- a/Docs/TEST_LOG.md
+++ b/Docs/TEST_LOG.md
@@ -4,6 +4,24 @@
 
 ---
 
+## 2026-03-05 - feat(ui): death-screen tappable restart button (#139)
+
+### Summary
+- Replaced spacebar-only death prompt with touch-friendly restart messaging on the game-over screen.
+- Added `GameOverScreenController` so death-screen UI (restart button + label) is owned by canvas UI, while `GameManager` keeps restart flow/state orchestration.
+- Preserved keyboard restart behavior by supporting `Space` via both Input System and legacy input fallback.
+
+### New or Updated Tests
+**EditMode**
+- `GameOverRestartContractsTests` — verifies shared UI restart entrypoint contract and touch-friendly death-screen copy in scenes.
+- `GameManagerKeyboardRestartContractsTests` — verifies `Space` restart code path remains present for Input System and legacy input.
+
+**PlayMode**
+- `DeathScreenRestartPlayTests` — verifies game-over UI appears on death and includes a tappable restart button.
+
+### Notes
+- Manual validation confirmed restart button works on death screen and `Space` restart works on PC after input-backend fallback update.
+
 ## 2026-03-05 - refactor/143-split-player-controller (#145 attack speed wiring)
 
 ### Summary