Add websockets for chathistory

Author: TanPingZhi

collab-service/app/model/repository.js

@@ -1,4 +1,4 @@
-import { connect } from 'mongoose';
+import { connect, mongoose } from 'mongoose';
 import UsersSession from "./usersSession-model.js";
 
 export async function connectToMongo() {
@@ -51,4 +51,55 @@ export async function get_all_rooms() {
         console.error('Error getting all rooms:', error);
         return null;
     }
-}
+}
+
+
+// Function to add a new message to chatHistory with transaction support
+export async function addMessageToChat(roomId, userId, text) {
+    // Start a session for the transaction
+    const session = await mongoose.startSession();
+  
+    try {
+      session.startTransaction();
+  
+      // Find the session document by roomId within the transaction
+      const sessionDoc = await UsersSession.findOne({ roomId }).session(session);
+  
+      if (!sessionDoc) {
+        throw new Error('Room not found');
+      }
+  
+      // Determine the next message index within the transaction
+      const lastMessageIndex = sessionDoc.chatHistory.length > 0
+        ? sessionDoc.chatHistory[sessionDoc.chatHistory.length - 1].messageIndex
+        : -1;
+  
+      // Create the new message with incremented messageIndex
+      const newMessage = {
+        messageIndex: lastMessageIndex + 1,
+        userId,
+        text,
+        timestamp: new Date()
+      };
+  
+      // Add the new message to chatHistory within the transaction
+      sessionDoc.chatHistory.push(newMessage);
+  
+      // Save the document within the transaction
+      await sessionDoc.save({ session });
+  
+      // Commit the transaction
+      await session.commitTransaction();
+  
+      // End the session and return the new message
+      session.endSession();
+      return newMessage;
+  
+    } catch (error) {
+      // If an error occurs, abort the transaction
+      await session.abortTransaction();
+      session.endSession();
+      console.error('Error adding message to chat:', error);
+      throw error;
+    }
+  }

collab-service/app/model/usersSession-model.js

@@ -2,6 +2,27 @@ import mongoose from 'mongoose';
 
 const Schema = mongoose.Schema;
 
+
+const messageSchema = new Schema({
+    messageIndex: {
+      type: Number,
+      required: true
+    },
+    userId: {
+      type: String,
+      required: true
+    },
+    text: {
+      type: String,
+      required: true
+    },
+    // timestamp: {
+    //   type: Date,
+    //   default: Date.now,
+    //   required: true
+    // }
+  });
+
 const usersSessionSchema = new Schema({
     users: {
         type: [String],
@@ -15,7 +36,11 @@ const usersSessionSchema = new Schema({
         type: Date,
         required: true,
         default: Date.now
-    }
+    },
+    chatHistory: {
+        type: [messageSchema],
+        default: []
+      }
 });
 
 export default mongoose.model('UsersSession', usersSessionSchema);

collab-service/app/server.js

@@ -1,9 +1,45 @@
 import { connectToMongo } from "./model/repository.js";
 import http from "http";
 import index from "./index.js";
+import { Server } from "socket.io";
+import { addMessageToChat } from "./model/repository.js";
+
 const PORT = process.env.PORT || 5000;
 const server = http.createServer(index);
 
+const io = new Server(server, {
+    cors: {
+        origin: "*",  // Allow all origins; replace with specific origin if needed
+        methods: ["GET", "POST"],
+        allowedHeaders: ["Content-Type", "Authorization"],
+        credentials: true
+    }
+});
+
+io.on('connection', (socket) => {
+    console.log('User connected to Socket.IO');
+
+    // Join a room based on roomId
+    socket.on('joinRoom', (roomId) => {
+        socket.join(roomId);
+        console.log(`User joined room: ${roomId}`);
+    });
+
+    // Handle incoming chat messages
+    socket.on('sendMessage', async (data) => {
+        const { roomId, userId, text } = data;
+        const newMessage = await addMessageToChat(roomId, userId, text);
+
+        // Broadcast the message to all clients in the same room
+        io.to(roomId).emit('chatMessage', newMessage);
+    });
+
+    socket.on('disconnect', () => {
+        console.log('User disconnected from Socket.IO');
+    });
+});
+
+
 connectToMongo().then(() => {
     server.listen(PORT, () => {
         console.log(`Server running on http://localhost:${PORT}`);