80 lines
2.6 KiB
JavaScript
80 lines
2.6 KiB
JavaScript
const path = require("path");
|
|
require("dotenv").config({
|
|
path: path.resolve(process.cwd(), `.env.${process.env.NODE_ENV || "development"}`)
|
|
});
|
|
const logger = require("../utils/logger");
|
|
const { admin } = require("../firebase/firebase-handler");
|
|
|
|
// Logging helper functions
|
|
function createLogEvent(socket, level, message) {
|
|
console.log(`[IOREDIS LOG EVENT] - ${socket.user.email} - ${socket.id} - ${message}`);
|
|
logger.log("ioredis-log-event", level, socket.user.email, null, { wsmessage: message });
|
|
}
|
|
|
|
const redisSocketEvents = (io) => {
|
|
// Room management and broadcasting events
|
|
function registerRoomAndBroadcastEvents(socket) {
|
|
socket.on("join-bodyshop-room", async (bodyshopUUID) => {
|
|
socket.join(bodyshopUUID);
|
|
createLogEvent(socket, "DEBUG", `Client joined bodyshop room: ${bodyshopUUID}`);
|
|
});
|
|
|
|
socket.on("leave-bodyshop-room", async (bodyshopUUID) => {
|
|
socket.leave(bodyshopUUID);
|
|
createLogEvent(socket, "DEBUG", `Client left bodyshop room: ${bodyshopUUID}`);
|
|
});
|
|
|
|
socket.on("broadcast-to-bodyshop", async (bodyshopUUID, message) => {
|
|
io.to(bodyshopUUID).emit("bodyshop-message", message);
|
|
createLogEvent(socket, "INFO", `Broadcast message to bodyshop ${bodyshopUUID}`);
|
|
});
|
|
}
|
|
|
|
// Register all socket events for a given socket connection
|
|
function registerSocketEvents(socket) {
|
|
createLogEvent(socket, "DEBUG", `Connected and Authenticated.`);
|
|
|
|
// Register room and broadcasting events
|
|
registerRoomAndBroadcastEvents(socket);
|
|
|
|
// Handle socket disconnection
|
|
socket.on("disconnect", async () => {
|
|
createLogEvent(socket, "DEBUG", `User disconnected.`);
|
|
});
|
|
}
|
|
|
|
const authMiddleware = (socket, next) => {
|
|
try {
|
|
if (socket.handshake.auth.token) {
|
|
admin
|
|
.auth()
|
|
.verifyIdToken(socket.handshake.auth.token)
|
|
.then((user) => {
|
|
socket.user = user;
|
|
next();
|
|
})
|
|
.catch((error) => {
|
|
next(new Error("Authentication error", JSON.stringify(error)));
|
|
});
|
|
} else {
|
|
next(new Error("Authentication error - no authorization token."));
|
|
}
|
|
} catch (error) {
|
|
console.log("Uncaught connection error:::", error);
|
|
logger.log("websocket-connection-error", "error", null, null, {
|
|
token: socket.handshake.auth.token,
|
|
...error
|
|
});
|
|
next(new Error(`Authentication error ${error}`));
|
|
}
|
|
};
|
|
|
|
// Socket.IO Middleware and Connection
|
|
io.use(authMiddleware);
|
|
io.on("connection", registerSocketEvents);
|
|
};
|
|
|
|
module.exports = {
|
|
redisSocketEvents
|
|
};
|