Add AI to connect four

commands/games-mp/connect-four.js

@@ -1,4 +1,5 @@
 const Command = require('../../structures/Command');
+const { Connect4AI } = require('connect4-ai');
 const { stripIndents } = require('common-tags');
 const { verify, list } = require('../../util/Util');
 const { LOADING_EMOJI_ID } = process.env;
@@ -26,7 +27,7 @@ module.exports = class ConnectFourCommand extends Command {
 			args: [
 				{
 					key: 'opponent',
-					prompt: 'What user would you like to challenge?',
+					prompt: 'What user would you like to challenge? To play against AI, choose me.',
 					type: 'user'
 				},
 				{
@@ -41,7 +42,6 @@ module.exports = class ConnectFourCommand extends Command {
 	}
 
 	async run(msg, { opponent, color }) {
-		if (opponent.bot) return msg.reply('Bots may not be played against.');
 		if (opponent.id === msg.author.id) return msg.reply('You may not play against yourself.');
 		const current = this.client.games.get(msg.channel.id);
 		if (current) return msg.reply(`Please wait until the current game of \`${current.name}\` is finished.`);
@@ -49,24 +49,30 @@ module.exports = class ConnectFourCommand extends Command {
 		const playerOneEmoji = colors[color];
 		let playerTwoEmoji = color === 'yellow' ? colors.red : colors.yellow;
 		try {
-			await msg.say(`${opponent}, do you accept this challenge?`);
-			const verification = await verify(msg.channel, opponent);
-			if (!verification) {
-				this.client.games.delete(msg.channel.id);
-				return msg.say('Looks like they declined...');
-			}
 			const available = Object.keys(colors).filter(clr => {
 				if (color === 'blue' && clr === 'purple') return false;
 				if (color === 'purple' && clr === 'blue') return false;
 				return color !== clr;
 			});
-			await msg.say(`${opponent}, what color do you want to be? Either ${list(available, 'or')}.`);
+			if (opponent.bot) {
-			const filter = res => res.author.id === opponent.id && available.includes(res.content.toLowerCase());
+				playerTwoEmoji = colors[available[Math.floor(Math.random() * available.length)]];
-			const p2Color = await msg.channel.awaitMessages(filter, {
+			} else {
-				max: 1,
+				await msg.say(`${opponent}, do you accept this challenge?`);
-				time: 30000
+				const verification = await verify(msg.channel, opponent);
-			});
+				if (!verification) {
-			if (p2Color.size) playerTwoEmoji = colors[p2Color.first().content.toLowerCase()];
+					this.client.games.delete(msg.channel.id);
+					return msg.say('Looks like they declined...');
+				}
+				await msg.say(`${opponent}, what color do you want to be? Either ${list(available, 'or')}.`);
+				const filter = res => res.author.id === opponent.id && available.includes(res.content.toLowerCase());
+				const p2Color = await msg.channel.awaitMessages(filter, {
+					max: 1,
+					time: 30000
+				});
+				if (p2Color.size) playerTwoEmoji = colors[p2Color.first().content.toLowerCase()];
+			}
+			let AIEngine = null;
+			if (opponent.bot) AIEngine = new Connect4AI();
 			const board = this.generateBoard();
 			let userTurn = true;
 			let winner = null;
@@ -75,40 +81,46 @@ module.exports = class ConnectFourCommand extends Command {
 			while (!winner && board.some(row => row.includes(null))) {
 				const user = userTurn ? msg.author : opponent;
 				const sign = userTurn ? 'user' : 'oppo';
-				await msg.say(stripIndents`
+				let i;
-					${user}, which column do you pick? Type \`end\` to forefeit.
+				if (opponent.bot && !userTurn) {
+					i = AIEngine.playAI('hard');
+				} else {
+					await msg.say(stripIndents`
+						${user}, which column do you pick? Type \`end\` to forefeit.
 
-					${this.displayBoard(board, playerOneEmoji, playerTwoEmoji)}
+						${this.displayBoard(board, playerOneEmoji, playerTwoEmoji)}
-					${nums.join('')}
+						${nums.join('')}
-				`);
+					`);
-				const pickFilter = res => {
+					const pickFilter = res => {
-					if (res.author.id !== user.id) return false;
+						if (res.author.id !== user.id) return false;
-					const choice = res.content;
+						const choice = res.content;
-					if (choice.toLowerCase() === 'end') return true;
+						if (choice.toLowerCase() === 'end') return true;
-					const i = Number.parseInt(choice, 10) - 1;
+						const i = Number.parseInt(choice, 10) - 1;
-					return board[colLevels[i]] && board[colLevels[i]][i] !== undefined;
+						return board[colLevels[i]] && board[colLevels[i]][i] !== undefined;
-				};
+					};
-				const turn = await msg.channel.awaitMessages(pickFilter, {
+					const turn = await msg.channel.awaitMessages(pickFilter, {
-					max: 1,
+						max: 1,
-					time: 60000
+						time: 60000
-				});
+					});
-				if (!turn.size) {
+					if (!turn.size) {
-					await msg.say('Sorry, time is up!');
+						await msg.say('Sorry, time is up!');
-					if (lastTurnTimeout) {
+						if (lastTurnTimeout) {
-						winner = 'time';
+							winner = 'time';
-						break;
+							break;
-					} else {
+						} else {
-						lastTurnTimeout = true;
+							lastTurnTimeout = true;
-						userTurn = !userTurn;
+							userTurn = !userTurn;
-						continue;
+							continue;
+						}
 					}
+					const choice = turn.first().content;
+					if (choice.toLowerCase() === 'end') {
+						winner = userTurn ? opponent : msg.author;
+						break;
+					}
+					i = Number.parseInt(choice, 10) - 1;
+					if (AIEngine) AIEngine.play(i);
 				}
-				const choice = turn.first().content;
-				if (choice.toLowerCase() === 'end') {
-					winner = userTurn ? opponent : msg.author;
-					break;
-				}
-				const i = Number.parseInt(choice, 10) - 1;
 				board[colLevels[i]][i] = sign;
 				colLevels[i]--;
 				if (this.verifyWin(board)) winner = userTurn ? msg.author : opponent;

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "xiao",
-	"version": "124.5.1",
+	"version": "124.5.2",
 	"description": "Your personal server companion.",
 	"main": "Xiao.js",
 	"scripts": {
@@ -39,6 +39,7 @@
 		"cheerio": "^1.0.0-rc.5",
 		"cloc": "^2.7.0",
 		"common-tags": "^1.8.0",
+		"connect4-ai": "^0.1.3",
 		"custom-translate": "^2.2.8",
 		"didyoumean2": "^4.1.0",
 		"discord.js": "^12.5.1",