Added socket reconnection.

config/default.js

@@ -10,12 +10,14 @@ module.exports = {
 		birthdate: new Date(1952, 11, 10),
 		avatar: 'https://i.imgur.com/IZwrjjG.png',
 	},
+	uniqueUsername: true,
 	socket: 'ws://127.0.0.1:3000/socket',
 	api: 'http://127.0.0.1:3000/api',
 	prefix: '~',
 	style: {
 		color: 'var(--message-56)',
 	},
-    games: ['mash'],
+	games: ['mash', 'cursed'],
 	channels: ['GamesNight'],
+	reconnectDelay: 10, // seconds
 };

src/app.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const config = require('config');
+const { setTimeout: delay } = require('timers/promises');
 const bhttp = require('bhttp');
 const WebSocket = require('ws');
 const fs = require('fs').promises;
@@ -11,7 +12,10 @@ const points = {};
 async function auth() {
 	const httpSession = bhttp.session();
 
-	const res = await httpSession.post(`${config.api}/session`, config.user, {
+	const res = await httpSession.post(`${config.api}/session`, {
+		...config.user,
+		username: config.uniqueUsername ? `${config.user.username}-${new Date().getTime().toString().slice(-5)}` : config.user.username,
+	}, {
 		encodeJSON: true,
 	});
 
@@ -38,18 +42,6 @@ async function getWsId(httpSession) {
 	return res.body;
 }
 
-function connect(wsCreds, sessionCookie) {
-	const ws = new WebSocket(`${config.socket}?${new URLSearchParams({ v: wsCreds.wsId, t: wsCreds.timestamp }).toString()}`, [], {
-		headers: {
-			cookie: sessionCookie,
-		},
-	});
-
-	logger.info(`Connected to ${config.socket}`);
-
-	return ws;
-}
-
 async function setPoints(gameKey, user, value, mode = 'add') {
 	const userKey = `${user.id}:${user.username}`;
 
@@ -96,7 +88,7 @@ function getLeaderboard(game, { user, room }) {
 }
 
 function onConnect(data, bot) {
-	bot.transmit('joinRooms', { rooms: config.channels });
+	bot.socket.transmit('joinRooms', { rooms: config.channels });
 }
 
 function onRooms({ rooms, users }, bot) {
@@ -108,7 +100,7 @@ function onRooms({ rooms, users }, bot) {
 	/* eslint-enable no-param-reassign */
 
 	rooms.forEach((room) => {
-		bot.transmit('message', {
+		bot.socket.transmit('message', {
 			roomId: room.id,
 			body: `Hi, I am ${config.user.username}, your game host!`,
 			style: config.style,
@@ -184,29 +176,12 @@ async function initPoints() {
 	}
 }
 
-async function init() {
+function getGames(bot) {
-	const { user, httpSession, sessionCookie } = await auth();
-	const wsCreds = await getWsId(httpSession);
-	const ws = connect(wsCreds, sessionCookie);
-
-	ws.transmit = (domain, data) => {
-		ws.send(JSON.stringify([domain, data]));
-	};
-
-	const bot = {
-		user,
-		ws,
-		httpSession,
-		rooms: [],
-		users: [],
-		transmit: ws.transmit,
-	};
-
 	const games = config.games.reduce((acc, key) => {
 		const game = require(`./games/${key}`); // eslint-disable-line global-require, import/no-dynamic-require
 
 		const sendMessage = (body, roomId) => {
-			bot.transmit('message', {
+			bot.socket.transmit('message', {
 				roomId,
 				body: `[${game.name || key}] ${body}`,
 				style: config.style,
@@ -227,7 +202,22 @@ async function init() {
 		};
 	}, {});
 
-	ws.on('message', (msg) => {
+	return games;
+}
+
+async function connect(wsCreds, sessionCookie, bot, games) {
+	const socket = { ws: { readyState: 0 } };
+
+	socket.connect = () => {
+		logger.info(`Attempting to connect to ${config.socket}`);
+
+		socket.ws = new WebSocket(`${config.socket}?${new URLSearchParams({ v: wsCreds.wsId, t: wsCreds.timestamp }).toString()}`, [], {
+			headers: {
+				cookie: sessionCookie,
+			},
+		});
+
+		socket.ws.on('message', (msg) => {
 			const [domain, data] = JSON.parse(msg);
 
 			if (messageHandlers[domain]) {
@@ -235,6 +225,44 @@ async function init() {
 			}
 		});
 
+		socket.ws.on('close', async (info) => {
+			logger.error(`WebSocket closed, reconnecting in ${config.reconnectDelay} seconds: ${info}`);
+
+			await delay(config.reconnectDelay * 1000);
+			socket.connect();
+		});
+
+		socket.ws.on('error', async (error) => {
+			logger.error(`WebSocket error: ${error.message}`);
+		});
+
+		logger.info(`Connected to ${config.socket}`);
+	};
+
+	socket.transmit = (domain, data) => {
+		socket.ws.send(JSON.stringify([domain, data]));
+	};
+
+	socket.connect();
+
+	return socket;
+}
+
+async function init() {
+	const { user, httpSession, sessionCookie } = await auth();
+	const wsCreds = await getWsId(httpSession);
+
+	const bot = {
+		user,
+		httpSession,
+		rooms: [],
+		users: [],
+	};
+
+	const games = getGames(bot);
+
+	bot.socket = await connect(wsCreds, sessionCookie, bot, games);
+
 	await initPoints();
 }
 

src/games/mash.js

@@ -1,7 +1,9 @@
 'use strict';
 
 const config = require('config');
+
 const words = require('../../assets/mash-words.json');
+
 let mash = null;
 
 function start(length, context, attempt = 0) {
@@ -23,7 +25,7 @@ function start(length, context, attempt = 0) {
 	const wordEntries = Object.entries(lengthWords);
 	const [key, answers] = wordEntries[Math.floor(Math.random() * wordEntries.length)];
 
-    const anagram = key.split('').sort(() => Math.random() > .5 ? 1 : -1).join('');
+	const anagram = key.split('').sort(() => (Math.random() > 0.5 ? 1 : -1)).join('');
 
 	if (answers.includes(anagram)) {
 		if (attempt >= 10) {
@@ -63,7 +65,7 @@ function play(word, context) {
 	if (mash.answers.includes(word)) {
 		context.sendMessage(mash.answers.length === 1
 			? `**${word}** is the right answer, @${context.user.username} now has **${context.user.points + 1} ${context.user.points === 0 ? 'point' : 'points'}**! There were no other options for **${mash.anagram}**.`
-	    : `**${word}** is the right answer, @${context.user.username} now has **${context.user.points + 1} ${context.user.points === 0 ? 'point' : 'points'}**! Other options for **${mash.anagram}**: ${mash.answers.filter((answer) => answer !== word).map((word) => `**${word}**`).join(', ')}`, context.room.id);
+			: `**${word}** is the right answer, @${context.user.username} now has **${context.user.points + 1} ${context.user.points === 0 ? 'point' : 'points'}**! Other options for **${mash.anagram}**: ${mash.answers.filter((answer) => answer !== word).map((answer) => `**${answer}**`).join(', ')}`, context.room.id);
 
 		context.logger.info(`Mash '${mash.anagram}' guessed by '${context.user.username}' with '${word}'`);
 		context.setPoints(context.user, 1);