Fix the endpoints

Author: TanPingZhi

collab-service/app/controller/collab-controller.js

@@ -1,7 +1,66 @@
-export async function create_room(req, res) {
-    // this function is to create a connection between two users
-    // the connection is store in mongodb
+import { createRoom, get_roomID, heartbeat, get_all_rooms } from '../model/repository.js';
+import crypto from 'crypto';
 
+// Create a room between two users
+export async function create_room(req, res) {
     const { user1, user2 } = req.body;
 
-}
+    if (!user1 || !user2) {
+        return res.status(400).json({ error: 'Both user1 and user2 are required' });
+    }
+
+    // Generate a unique room ID by hashing the two user IDs
+    const roomId = crypto.createHash('sha256').update(user1 + user2).digest('hex');
+    const room = await createRoom(user1, user2, roomId);
+
+    if (room) {
+        res.status(201).json(room);
+    } else {
+        res.status(500).json({ error: 'Failed to create room' });
+    }
+}
+
+// Get room ID by user
+export async function get_room_by_user(req, res) {
+    const { user } = req.params;
+
+    if (!user) {
+        return res.status(400).json({ error: 'User is required' });
+    }
+
+    const room = await get_roomID(user);
+
+    if (room) {
+        res.status(200).json(room);
+    } else {
+        res.status(404).json({ error: `Room not found for user: ${user}` });
+    }
+}
+
+// Update heartbeat for a room
+export async function update_heartbeat(req, res) {
+    const { roomId } = req.params;
+
+    if (!roomId) {
+        return res.status(400).json({ error: 'Room ID is required' });
+    }
+
+    const updatedRoom = await heartbeat(roomId);
+
+    if (updatedRoom) {
+        res.status(200).json(updatedRoom);
+    } else {
+        res.status(404).json({ error: `Room with ID ${roomId} not found` });
+    }
+}
+
+// Get all rooms
+export async function get_all_rooms_controller(req, res) {
+    const rooms = await get_all_rooms();
+
+    if (rooms) {
+        res.status(200).json(rooms);
+    } else {
+        res.status(500).json({ error: 'Failed to retrieve rooms' });
+    }
+}

collab-service/app/index.js

@@ -30,6 +30,9 @@ app.use((req, res, next) => {
 });
 
 
+app.use("/collab", collabRoutes);
+
+
 app.get("/", (req, res, next) => {
   console.log("Sending Greetings!");
   res.json({
@@ -53,6 +56,5 @@ app.use((error, req, res, next) => {
   });
 });
 
-app.use("/collab", collabRoutes);
 
 export default app;

collab-service/app/model/repository.js

@@ -1,6 +1,5 @@
 import { connect } from 'mongoose';
-import UsersSession from './usersSession-model';
-import mongoose from 'mongoose';
+import UsersSession from "./usersSession-model.js";
 
 export async function connectToMongo() {
     await connect('mongodb+srv://admin:[email protected]/');
@@ -30,4 +29,26 @@ export async function get_roomID(user) {
         console.error('Error finding room for ${user}:', error);
         return null;
     }
+}
+
+export async function heartbeat(roomId) {
+    try {
+        const room = await UsersSession.findOne({ roomId: roomId });
+        room.lastUpdated = new Date();
+        await room.save();
+        return room;
+    } catch (error) {
+        console.error('Error updating room ${roomId}:', error);
+        return null;
+    }  
+}
+
+export async function get_all_rooms() {
+    try {
+        const rooms = await UsersSession.find({});
+        return rooms;
+    } catch (error) {
+        console.error('Error getting all rooms:', error);
+        return null;
+    }
 }

collab-service/app/routes/collab-routes.js

@@ -1,9 +1,19 @@
 import express from "express";
-
-import { create_room } from "../controller/collab-controller.js";
+import {
+    create_room,
+    get_room_by_user,
+    update_heartbeat,
+    get_all_rooms_controller
+} from "../controller/collab-controller.js";
 
 const router = express.Router();
 
 router.post("/create-room", create_room);
 
-export default router;
+router.get("/user/:user", get_room_by_user);
+
+router.patch("/heartbeat/:roomId", update_heartbeat);
+
+router.get("/rooms", get_all_rooms_controller);
+
+export default router;

collab-service/app/server.js

@@ -2,4 +2,12 @@ import { connectToMongo } from "./model/repository.js";
 import http from "http";
 import index from "./index.js";
 const PORT = process.env.PORT || 5000;
-const server = http.createServer(index);
+const server = http.createServer(index);
+
+connectToMongo().then(() => {
+    server.listen(PORT, () => {
+        console.log(`Server running on http://localhost:${PORT}`);
+    });
+}).catch((error) => {
+    console.error("Error connecting to MongoDB:", error);
+});

collab-service/package.json

@@ -2,11 +2,11 @@
   "name": "collab-service",
   "version": "1.0.0",
   "description": "",
-  "main": "app/index.js",
+  "main": "app/server.js",
   "type": "module",
   "scripts": {
-    "dev": "nodemon -L app/index.js",
-    "start": "node app/index.js",
+    "dev": "nodemon -L app/server.js",
+    "start": "node app/server.js",
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "keywords": [],
@@ -20,4 +20,4 @@
   "devDependencies": {
     "nodemon": "^3.1.4"
   }
-}
+}