From 79610b958b5072c890425fadb7c3063f83474331 Mon Sep 17 00:00:00 2001
From: Joel Therrien <joel@joeltherrien.ca>
Date: Mon, 25 Sep 2023 20:50:58 -0700
Subject: [PATCH] Fix end-of-game UI bugs

---
 ui/src/Game.tsx | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/ui/src/Game.tsx b/ui/src/Game.tsx
index 27ba994..51f476a 100644
--- a/ui/src/Game.tsx
+++ b/ui/src/Game.tsx
@@ -370,11 +370,11 @@ export function Game(props: {
     }
 
     useEffect(() => {
+        trayDispatch({action: TileDispatchActionType.RETRIEVE});
+        setConfirmedScorePoints(-1);
         if(!isGameOver){
             logDispatch(<h4>Turn {turnCount}</h4>);
-            logDispatch(<div>{playerTurnName}'s turn</div>)
-            setConfirmedScorePoints(-1);
-            trayDispatch({action: TileDispatchActionType.RETRIEVE});
+            logDispatch(<div>{playerTurnName}'s turn</div>);
             if(playerTurnName != props.settings.playerName && !isGameOver) {
                 runAI();
             }
@@ -421,7 +421,7 @@ export function Game(props: {
                     {props.settings.aiName} has {remainingAITiles} tiles
                 </div>
                 <button
-                    disabled={remainingTiles == 0}
+                    disabled={remainingTiles == 0 || isGameOver}
                     onClick={() => {
                     trayDispatch({action: TileDispatchActionType.RETURN}); // want all tiles back on tray for tile exchange
                     setIsTileExchangeOpen(true);
@@ -430,7 +430,10 @@ export function Game(props: {
         
             <TileTray letters={playerLetters} trayLength={props.settings.trayLength} trayDispatch={trayDispatch}/>
             <div className="player-controls">
-                <button className="check" onClick={() => {
+                <button
+                    className="check"
+                    disabled={isGameOver}
+                    onClick={() => {
                     const playedTiles = playerLetters.map((i) => {
                         if (i === undefined) {
                             return null;
@@ -485,10 +488,16 @@ export function Game(props: {
 
 
                 }}>{confirmedScorePoints > -1 ? `Score ${confirmedScorePoints} points` : "Check"}</button>
-                <button className="return" onClick={() => {
+                <button
+                    className="return"
+                    disabled={isGameOver}
+                    onClick={() => {
                     trayDispatch({action: TileDispatchActionType.RETURN});
                 }}>Return Tiles</button>
-                <button className="pass" onClick={() => {
+                <button
+                    className="pass"
+                    disabled={isGameOver}
+                    onClick={() => {
                     if (window.confirm("Are you sure you want to pass?")) {
                         const result = props.wasm.skip_turn() as MyResult<string>;
                         handlePlayerAction({type: "Pass"}, props.settings.playerName);