From 803836a8cd52d76522efb90aa44bca022e2cfe21 Mon Sep 17 00:00:00 2001
From: Dragon Fire <danielbodendahl@gmail.com>
Date: Fri, 22 Jan 2021 20:50:29 -0500
Subject: [PATCH] Clean up AI

---
 commands/games-mp/nim.js | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/commands/games-mp/nim.js b/commands/games-mp/nim.js
index 883ea885..63a23672 100644
--- a/commands/games-mp/nim.js
+++ b/commands/games-mp/nim.js
@@ -185,6 +185,17 @@ module.exports = class NimCommand extends Command {
 	}
 
 	computerTurn(board) {
+		let clearRows = 0;
+		const unclearRows = [];
+		for (const row of board) {
+			if (row === 0) clearRows++;
+			else unclearRows.push(row);
+		}
+		if (clearRows === board.length - 1) {
+			const amount = board[unclearRows[0]] - 1;
+			board[unclearRows[0]] -= amount;
+			return [unclearRows[0], amount];
+		}
 		for (let i = 0; i < board.length; i++) {
 			if (board[i] > 0) {
 				for (let j = 1; j <= board[i]; j++) {
@@ -199,8 +210,8 @@ module.exports = class NimCommand extends Command {
 			}
 		}
 		const randomRow = Math.floor(Math.random() * board.length);
-		const randomAmount = Math.floor(Math.random() * board[randomRow]) + 1;
-		board[randomRow] -= randomAmount;
-		return [randomRow, randomAmount];
+		const amount = board[randomRow] - 1;
+		board[randomRow] -= amount;
+		return [randomRow, amount];
 	}
 };