server.js

// const WebSocket = require("ws");
// const express = require("express");
// const path = require("path");
// const http = require("http");
// const cors = require("cors");

// const app = express();
// app.use(cors({ origin: "*" }));
// const server = http.createServer(app);
// const wss = new WebSocket.Server({ server });

// app.use(express.static(path.join(__dirname, "."))); // Serve Frontend

// // Store users in rooms
// let users = {};
// let rooms = {}; // { roomName: [clients] }

// wss.on("connection", (ws) => {
//     ws.on("message", (message) => {
//         const data = JSON.parse(message);

//         if (data.type === "join") {
//             ws.room = data.room;
//             ws.username = data.username;

//             if (!users[data.room]) {
//                 users[data.room] = [];
//             }

//             users[data.room].push({ username: data.username, ws });

//             // Notify all clients about updated user list
//             broadcastUsersList(data.room);
//         }

//         if (data.type === "leave") {
//             users[data.room] = users[data.room].filter(user => user.username !== data.username);
//             broadcastUsersList(data.room);
//         }
//     });
//     function broadcastUsersList(room) {
//         const userList = users[room]?.map(user => user.username) || [];
//         const userListMessage = JSON.stringify({ type: "user_list", users: userList, room });
    
//         users[room]?.forEach(user => {
//             if (user.ws.readyState === WebSocket.OPEN) {
//                 user.ws.send(userListMessage);
//             }
//         });
//     }


//         if (data.type === "message") {
//             // Broadcast message to all users in the same room
//             rooms[data.room]?.forEach(client => {
//                 if (client.readyState === WebSocket.OPEN) {
//                     client.send(JSON.stringify(data));
//                 }
//             });
//         } else if (data.type === "typing" || data.type === "stop_typing") {
//             rooms[data.room]?.forEach(client => {
//                 if (client !== ws && client.readyState === WebSocket.OPEN) {
//                     client.send(JSON.stringify(data));
//                 }  });
//             }
//             else if (data.type === "file") {
//                 if (rooms[data.room]) {
//                     rooms[data.room].forEach(client => {
//                         if (client.readyState === WebSocket.OPEN) {
//                             client.send(JSON.stringify(data));
//                         }
//                     });
//                 }
//             } 
//             else if (data.type === "seen") {
//                     rooms[data.room]?.forEach(client => {
//                         if (client !== ws && client.readyState === WebSocket.OPEN) {
//                             client.send(JSON.stringify(data));
//                         }  });
//                     }
//              else if (data.type === "join") {
//                 ws.room = data.room;

//             if (!rooms[data.room]) {
//                 rooms[data.room] = [];
//             }
//             rooms[data.room].push(ws);

//             console.log(`${data.username} joined room: ${data.room}`);

//             // Notify all users in the room
//             const joinMsg = {
//                 type: "system",
//                 message: `${data.username} has joined the chat!`,
//                 room: data.room
//             };
//             rooms[data.room].forEach(client => {
//                 if (client.readyState === WebSocket.OPEN) {
//                     client.send(JSON.stringify(joinMsg));
//                 }
//             });
//         }   
//     });    

//     ws.on("close", () => {
//         if (ws.room && ws.username) {
//             users[ws.room] = users[ws.room].filter(user => user.username !== ws.username);
//             broadcastUsersList(ws.room);
//         }
// });

// server.listen(3000, '192.168.0.101', () => {
//     console.log("Server running on all network interfaces");
// })   


const WebSocket = require("ws");
const express = require("express");
const path = require("path");
const http = require("http");
const cors = require("cors");

const app = express();
app.use(cors({ origin: "*" }));
const server = http.createServer(app);
const wss = new WebSocket.Server({ server });

app.use(express.static(path.join(__dirname, "."))); // Serve Frontend

// Store users in rooms
let users = {};
let rooms = {}; // { roomName: [{ ws, username }] }

wss.on("connection", (ws) => {
    ws.on("message", (message) => {
        const data = JSON.parse(message);

        if (data.type === "join") {
            ws.room = data.room;
            ws.username = data.username;

            if (!users[data.room]) {
                users[data.room] = [];
            }
            if (!rooms[data.room]) {
                rooms[data.room] = [];
            }

            // ✅ Store user in `users` and `rooms`
            users[data.room].push({ username: data.username, ws });
            rooms[data.room].push({ username: data.username, ws });

            console.log(`${data.username} joined room: ${data.room}`);

            // ✅ Notify all users in the room
            const joinMsg = {
                type: "system",
                message: `${data.username} has joined the chat!`,
                room: data.room
            };
            broadcast(data.room, joinMsg);

            // ✅ Broadcast updated user list
            broadcastUsersList(data.room);
        }
        else if (data.type === "leave") {
            // ✅ Notify all users in the room
            const leaveMsg = {
                type: "system",
                message: `${data.username} has left the chat.`,
                room: data.room
            };
            broadcast(data.room, leaveMsg);

            // ✅ Remove user from room list
            users[data.room] = users[data.room].filter(user => user.username !== data.username);
            rooms[data.room] = rooms[data.room].filter(client => client.username !== data.username);

            // ✅ Update user list for everyone
            broadcastUsersList(data.room);
        }
        else if (data.type === "message") {
            // ✅ Broadcast message to all users in the same room
            broadcast(data.room, data);
        }
        else if (data.type === "typing" || data.type === "stop_typing") {
            rooms[data.room]?.forEach(client => {
                if (client.ws !== ws && client.ws.readyState === WebSocket.OPEN) {
                    client.ws.send(JSON.stringify(data));
                }
            });
        }
        else if (data.type === "file") {
            broadcast(data.room, data);
        }
        else if (data.type === "seen") {
            rooms[data.room]?.forEach(client => {
                if (client.ws !== ws && client.ws.readyState === WebSocket.OPEN) {
                    client.ws.send(JSON.stringify(data));
                }
            });
        }
    });

    // ✅ Function to Broadcast Messages in a Room
    function broadcast(room, data) {
        rooms[room]?.forEach(client => {
            if (client.ws.readyState === WebSocket.OPEN) {
                client.ws.send(JSON.stringify(data));
            }
        });
    }

    // ✅ Function to Broadcast Active Users List
    function broadcastUsersList(room) {
        const userList = users[room]?.map(user => user.username) || [];
        const userListMessage = JSON.stringify({ type: "user_list", users: userList, room });

        users[room]?.forEach(user => {
            if (user.ws.readyState === WebSocket.OPEN) {
                user.ws.send(userListMessage);
            }
        });
    }

    // ✅ Handle Disconnection
    ws.on("close", () => {
        if (ws.room && ws.username) {
            // ✅ Send system message that user left
            const disconnectMsg = {
                type: "system",
                message: `${ws.username} has left the chat.`,
                room: ws.room
            };
            broadcast(ws.room, disconnectMsg);

            // ✅ Remove user from lists
            users[ws.room] = users[ws.room].filter(user => user.username !== ws.username);
            rooms[ws.room] = rooms[ws.room].filter(client => client.username !== ws.username);

            // ✅ Update user list for everyone
            broadcastUsersList(ws.room);
        }
    });
});

server.listen(3000, '192.168.0.101', () => {
    console.log("Server running on 192.168.0.101:3000");
});