50 lines
1.6 KiB
TypeScript
50 lines
1.6 KiB
TypeScript
import { Server as SocketIOServer } from 'socket.io';
|
|
import type { HttpServer } from 'vite';
|
|
// Don't try to replace with $lib alias. Since this
|
|
// file gets loaded as a vite plugin, it will crash
|
|
import { v4 as uuidv4 } from 'uuid';
|
|
import { db } from '../server/db';
|
|
import { authdb } from '../server/db/sqlite';
|
|
|
|
let io: SocketIOServer | undefined;
|
|
|
|
export function startupSocketIOServer(httpServer: HttpServer | null) {
|
|
if (io) return;
|
|
console.log('\x1b[35m[WS]\x1b[0m setup');
|
|
|
|
if (!httpServer) {
|
|
console.error('Error: httpServer is null. Cannot start Socket.IO server.');
|
|
return;
|
|
}
|
|
|
|
io = new SocketIOServer(httpServer);
|
|
|
|
io.on('connection', async (socket) => {
|
|
// Runs on client connection
|
|
console.log(`\x1b[35m[WS]\x1b[0m client connected (${socket.id})`);
|
|
// Runs on message received
|
|
socket.on('message', async (msg) => {
|
|
// If message not empty
|
|
if (msg.content !== '') {
|
|
console.log(`\x1b[35m[WS]\x1b[0m message from ${socket.id}: ${msg.content}`);
|
|
// Store the message in the database
|
|
const timestamp = new Date();
|
|
await db.sendMessage(msg.channel, msg.content, msg.id, uuidv4(), timestamp);
|
|
const sender = authdb.getUser(msg.id);
|
|
io!.emit('message', {
|
|
user: sender.username,
|
|
message: msg.content,
|
|
imageSrc: sender.image,
|
|
channel: msg.channel,
|
|
timestamp: timestamp,
|
|
});
|
|
}
|
|
});
|
|
|
|
// Runs on client disconnect
|
|
socket.on('disconnect', () => {
|
|
console.log(`\x1b[35m[WS]\x1b[0m client disconnected (${socket.id})`);
|
|
});
|
|
});
|
|
}
|