GH-354: Disable all buttons when game is over

utarwyn · Oct 17, 2022 · 5010bdb · 5010bdb
1 parent 2a4963d
commit 5010bdb
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/src/__tests__/GameBoardButtonBuilder.test.ts b/src/__tests__/GameBoardButtonBuilder.test.ts
@@ -68,4 +68,17 @@ describe('GameBoardButtonBuilder', () => {
         expect((options.components![1].components[0] as MessageButton).disabled).toBeFalsy();
         expect((options.components![1].components[1] as MessageButton).disabled).toBeFalsy();
     });
+
+    it('should disable all buttons if game has ended', () => {
+        const options = builder
+            .withButtonsDisabledAfterUse()
+            .withBoard(2, [Player.First, Player.Second, Player.None, Player.None])
+            .withEndingMessage()
+            .toMessageOptions();
+
+        expect((options.components![0].components[0] as MessageButton).disabled).toBeTruthy();
+        expect((options.components![0].components[1] as MessageButton).disabled).toBeTruthy();
+        expect((options.components![1].components[0] as MessageButton).disabled).toBeTruthy();
+        expect((options.components![1].components[1] as MessageButton).disabled).toBeTruthy();
+    });
 });
diff --git a/src/bot/builder/GameBoardButtonBuilder.ts b/src/bot/builder/GameBoardButtonBuilder.ts
@@ -36,6 +36,11 @@ export default class GameBoardButtonBuilder extends GameBoardBuilder {
      * @private
      */
     private disableButtonsAfterUsed = false;
+    /**
+     * Stores if game has ended or not.
+     * @protected
+     */
+    private gameEnded = false;
 
     /**
      * Should disable buttons after been used.
@@ -60,6 +65,15 @@ export default class GameBoardButtonBuilder extends GameBoardBuilder {
         }
     }
 
+    /**
+     * @inheritdoc
+     * @override
+     */
+    override withEndingMessage(winner?: Entity): GameBoardBuilder {
+        this.gameEnded = true;
+        return super.withEndingMessage(winner);
+    }
+
     /**
      * @inheritdoc
      * @override
@@ -108,6 +122,10 @@ export default class GameBoardButtonBuilder extends GameBoardBuilder {
             }
         } else {
             button.setLabel(' ');
+
+            if (this.gameEnded && this.disableButtonsAfterUsed) {
+                button.setDisabled(true);
+            }
         }
 
         return button.setCustomId(buttonIndex.toString()).setStyle(this.buttonStyles[buttonData]);