xiao/commands/games-mp/nim.js

const Command = require('../../framework/Command');
const { stripIndents } = require('common-tags');
const { verify } = require('../../util/Util');
const emoji = require('../../assets/json/nim');
const nums = ['1️⃣', '2️⃣', '3️⃣', '4️⃣', '5️⃣', '6️⃣', '7️⃣', '8️⃣', '9️⃣', '🔟'];

module.exports = class NimCommand extends Command {
	constructor(client) {
		super(client, {
			name: 'nim',
			group: 'games-mp',
			description: 'Play a game of nim with another user or the AI.',
			game: true,
			args: [
				{
					key: 'opponent',
					type: 'user'
				},
				{
					key: 'rows',
					type: 'integer',
					default: 5,
					min: 1,
					max: 10
				}
			]
		});
	}

	async run(msg, { opponent, rows }) {
		if (opponent.id === msg.author.id) return msg.reply('You may not play against yourself.');
		if (this.client.blacklist.user.includes(opponent.id)) return msg.reply('This user is blacklisted.');
		if (!opponent.bot) {
			await msg.say(`${opponent}, do you accept this challenge?`);
			const verification = await verify(msg.channel, opponent);
			if (!verification) return msg.say('Looks like they declined...');
		}
		const board = this.generateBoard(rows);
		let userTurn = true;
		let winner = null;
		let lastTurnTimeout = false;
		let firstTurn = true;
		const objectEmoji = emoji[Math.floor(Math.random() * emoji.length)];
		while (!winner) {
			const user = userTurn ? msg.author : opponent;
			if (!userTurn && opponent.bot) {
				const turn = this.computerTurn(board);
				await msg.say(`For my turn, I remove **${turn[1]}** ${objectEmoji} from **row ${turn[0] + 1}**.`);
			} else {
				await msg.say(stripIndents`
					${user}, from which row do you want to remove from? Type \`end\` to forfeit.
					After this step, you will decide how many ${objectEmoji} to remove from that row.

					${this.displayBoard(board, objectEmoji)}

					${firstTurn ? '_In Nim, you win by forcing the opponent to take the last object._' : ''}
				`);
				const pickFilter = res => {
					if (res.author.id !== user.id) return false;
					const choice = res.content;
					if (choice.toLowerCase() === 'end') return true;
					const i = Number.parseInt(choice, 10) - 1;
					return board[i] && board[i] > 0;
				};
				const turn = await msg.channel.awaitMessages({
					filter: pickFilter,
					max: 1,
					time: 60000
				});
				if (!turn.size) {
					if (lastTurnTimeout) {
						winner = 'time';
						break;
					} else {
						await msg.say('Sorry, time is up!');
						lastTurnTimeout = true;
						userTurn = !userTurn;
						continue;
					}
				}
				const choice = turn.first().content;
				const picked = Number.parseInt(choice, 10);
				if (choice.toLowerCase() === 'end') {
					winner = userTurn ? opponent : msg.author;
					break;
				}
				const row = board[picked - 1];
				let rowPicked;
				if (row === 1) {
					rowPicked = 1;
				} else {
					await msg.say(stripIndents`
						${user}, how many ${objectEmoji} do you want to take from row ${picked}? Type \`end\` to forfeit.
						If you want to go back, type \`back\`.

						${nums[picked - 1]}${objectEmoji.repeat(row)}
					`);
					const rowFilter = res => {
						if (res.author.id !== user.id) return false;
						const chosen = res.content;
						if (chosen.toLowerCase() === 'end' || chosen.toLowerCase() === 'back') return true;
						const i = Number.parseInt(chosen, 10);
						return i <= row && i > 0;
					};
					const rowTurn = await msg.channel.awaitMessages({
						filter: rowFilter,
						max: 1,
						time: 60000
					});
					if (!rowTurn.size) {
						if (lastTurnTimeout) {
							winner = 'time';
							break;
						} else {
							await msg.say('Sorry, time is up!');
							lastTurnTimeout = true;
							userTurn = !userTurn;
							continue;
						}
					}
					const rowChoice = rowTurn.first().content;
					rowPicked = Number.parseInt(rowChoice, 10);
					if (rowChoice.toLowerCase() === 'end') {
						winner = userTurn ? opponent : msg.author;
						break;
					}
					if (rowChoice.toLowerCase() === 'back') continue;
				}
				board[picked - 1] -= rowPicked;
			}
			if (!userTurn && firstTurn) firstTurn = false;
			if (!board.some(r => r !== 0)) {
				winner = userTurn ? opponent : msg.author;
				break;
			}
			userTurn = !userTurn;
			if (lastTurnTimeout) lastTurnTimeout = false;
		}
		if (winner === 'time') return msg.say('Game ended due to inactivity.');
		return msg.say(`Congrats, ${winner}! You forced your opponent to pick the final object!`);
	}

	displayBoard(board, objectEmoji) {
		return board.map((amount, i) => `${nums[i]}${objectEmoji.repeat(amount)}`).join('\n');
	}

	generateBoard(rows) {
		const board = [];
		for (let i = 0; i < rows; i++) {
			board.push(i + 1);
		}
		return board;
	}

	xOr(board) {
		let value = 0;
		for (let i = 0; i < board.length; i++) {
			value ^= board[i];
		}
		return value;
	}

	computerTurn(board) {
		let nearEnd = false;
		let countNon1 = 0;
		for (const row of board) {
			if (row > 1) countNon1++;
		}
		nearEnd = countNon1 <= 1;
		if (nearEnd) {
			let movesLeft = 0;
			for (const row of board) {
				if (row > 0) movesLeft++;
			}
			const isOdd = movesLeft % 2 === 1;
			const largest = Math.max(...board);
			const indexOfMax = board.indexOf(largest);
			if (largest === 1 && isOdd) {
				board[indexOfMax] -= 1;
				return [indexOfMax, 1];
			}
			board[indexOfMax] -= largest - Number(isOdd);
			return [indexOfMax, largest - Number(isOdd)];
		}
		for (let i = 0; i < board.length; i++) {
			if (board[i] < 0) continue;
			for (let j = 1; j <= board[i]; j++) {
				board[i] -= j;
				const sum = this.xOr(board);
				if (sum === 0) {
					return [i, j];
				} else {
					board[i] += j;
				}
			}
		}
		let i = 0;
		let keepGoing = true;
		let finalValues = null;
		while (keepGoing) {
			if (board[i] === 0) {
				i++;
				continue;
			} else {
				board[i] -= 1;
				finalValues = [i, 1];
				keepGoing = false;
			}
		}
		return finalValues;
	}
};