From 05f91ffebc5a91da1a6c32f7d990178d6756cede Mon Sep 17 00:00:00 2001
From: Kayne Ruse <kayneruse@gmail.com>
Date: Thu, 18 Mar 2021 02:41:10 +1100
Subject: [PATCH] Added chatlog and backlog features

---
 server/chat/index.js              | 75 ++++++++++++++++++++++++++++---
 server/database/index.js          |  2 +-
 server/database/models/chatlog.js | 39 ++++++++++++++++
 server/database/models/index.js   |  2 +-
 4 files changed, 111 insertions(+), 7 deletions(-)
 create mode 100644 server/database/models/chatlog.js

diff --git a/server/chat/index.js b/server/chat/index.js
index 88a2cba..22251ec 100644
--- a/server/chat/index.js
+++ b/server/chat/index.js
@@ -1,5 +1,6 @@
-const { request } = require('express');
 const jwt = require('jsonwebtoken');
+const { chatlog } = require('../database/models');
+const { Op } = require('sequelize');
 
 const chat = io => {
 	io.on('connection', socket => {
@@ -47,6 +48,38 @@ const chat = io => {
 
 			//broadcast to this room
 			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} entered chat` });
+
+			//log
+			chatlog.create({
+				username: socket.user.username,
+				text: `${socket.user.username} entered chat`,
+				room: socket.user.room,
+				emphasis: true
+			});
+
+			//send backlog to the user
+			chatlog.findAll({
+				where: {
+					room: {
+						[Op.eq]: socket.user.room
+					}
+				},
+				order: [
+					['id', 'ASC']
+				]
+			})
+				.then(rows => rows.map(row => row.dataValues))
+				.then(rows => rows.filter(row => {
+					//emphasis and strong don't use usernames
+					return !(row.emphasis || row.strong);
+				}))
+				.then(rows => socket.emit('backlog', rows))
+				.then(() => {
+					//send a # to the user
+					const count = io.sockets.size;
+					socket.emit('message', { emphasis: true, text: count == 1 ? `${count} person in the chat` : `${count} people in the chat` });
+				})
+			;
 		});
 
 		socket.on('message', message => {
@@ -57,6 +90,13 @@ const chat = io => {
 
 			//broadcast to this room
 			socket.broadcast.to(socket.user.room).emit('message', { username: socket.user.username, text: message.text });
+
+			//log
+			chatlog.create({
+				username: socket.user.username,
+				text: message.text,
+				room: socket.user.room
+			});
 		});
 
 		socket.on('disconnect', reason => {
@@ -66,6 +106,14 @@ const chat = io => {
 			}
 
 			socket.broadcast.to(socket.user.room || '.error').emit('message', { emphasis: true, text: `${socket.user.username} left chat` });
+
+			//log
+			chatlog.create({
+				username: socket.user.username,
+				text: `${socket.user.username} left chat`,
+				room: socket.user.room,
+				emphasis: true
+			});
 		});
 	});
 };
@@ -82,14 +130,33 @@ const executeCommand = (socket, command) => {
 			}
 
 			//broadcast to the old room
-			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} left chat` });
+			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} left the room (going to ${room})` });
 
+			//log
+			chatlog.create({
+				username: socket.user.username,
+				text: `${socket.user.username} left the room`,
+				room: socket.user.room,
+				emphasis: true
+			});
+
+			//move
 			socket.leave(socket.user.room);
 			socket.user.room = room;
 			socket.join(socket.user.room);
 
 			//broadcast to the new room
-			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} entered chat` });
+			socket.broadcast.to(socket.user.room).emit('message', { emphasis: true, text: `${socket.user.username} entered the room` });
+
+			//log
+			chatlog.create({
+				username: socket.user.username,
+				text: `${socket.user.username} entered the room`,
+				room: socket.user.room,
+				emphasis: true
+			});
+
+			//update the user
 			socket.emit('message', { emphasis: true, text: `Entered room ${socket.user.room}` });
 			break;
 
@@ -100,6 +167,4 @@ const executeCommand = (socket, command) => {
 
 module.exports = chat;
 
-//TODO: record messages in a database
-//TODO: handle message backlog on connection
 //TODO: add banning and muting controls
diff --git a/server/database/index.js b/server/database/index.js
index 7d01b93..d098b9f 100644
--- a/server/database/index.js
+++ b/server/database/index.js
@@ -4,7 +4,7 @@ const sequelize = new Sequelize(process.env.DB_DATABASE, process.env.DB_USERNAME
 	host: process.env.DB_HOSTNAME,
 	dialect: 'mariadb',
 	timezone: process.env.DB_TIMEZONE,
-	logging: false
+	logging: process.env.DB_LOGGING ? console.log : false
 });
 
 sequelize.sync();
diff --git a/server/database/models/chatlog.js b/server/database/models/chatlog.js
new file mode 100644
index 0000000..5cdcb7f
--- /dev/null
+++ b/server/database/models/chatlog.js
@@ -0,0 +1,39 @@
+const Sequelize = require('sequelize');
+const sequelize = require('..');
+
+module.exports = sequelize.define('chatlog', {
+	id: {
+		type: Sequelize.INTEGER(11),
+		allowNull: false,
+		autoIncrement: true,
+		primaryKey: true,
+		unique: true
+	},
+
+	username: {
+		type: 'varchar(320)',
+		allowNull: false
+	},
+
+	text: {
+		type: Sequelize.TEXT,
+		allowNull: false,
+	},
+
+	room: {
+		type: Sequelize.TEXT,
+		allowNull: false
+	},
+
+	emphasis: {
+		type: Sequelize.BOOLEAN,
+		allowNull: false,
+		defaultValue: false
+	},
+
+	strong: {
+		type: Sequelize.BOOLEAN,
+		allowNull: false,
+		defaultValue: false
+	},
+});
diff --git a/server/database/models/index.js b/server/database/models/index.js
index aabbaee..231b939 100644
--- a/server/database/models/index.js
+++ b/server/database/models/index.js
@@ -1,3 +1,3 @@
 module.exports = {
-	//TODO: models
+	chatlog: require('./chatlog')
 };
\ No newline at end of file