index.js

import e from "express";
import http from "http";
import { Server } from "socket.io";
import mongoose from "mongoose";
import { config } from "dotenv";
import cors from "cors";
import messageRouter from "./routes/messageRoutes.js";
import { messageHandler } from "./socketEvenHandlers/messages.js";
import { fileHandler } from "./socketEvenHandlers/file.js";
import userSockets from "./stores/userSocketStore.js";
config();

const MONGODB_URI = process.env.MONGODB_URI;

if (!MONGODB_URI) {
  throw new Error("Please define the MONGODB_URI environment variable in .env.local");
}

mongoose.connect(MONGODB_URI)
  .then(() => console.log("Connected to DB"))
  .catch(err => console.log("DB Error:", err));

const app = e();
const PORT = process.env.PORT || 4000;


const allowedOrigins = [
  "https://bhargav-msgapp.vercel.app",
  "http://localhost:3000" 
];

app.use(cors({
  origin: "*",
  credentials: true
}));

app.get("/", (req, res) => {
 res.send("Socket server is running");
});

const server = http.createServer(app);

const io = new Server(server,
  {
    maxHttpBufferSize: 5 * 1024 * 1024, // server can take messages till 5mb
    cors: {
      origin: allowedOrigins,
      methods: ["GET", "POST"],
      credentials: true
    }
  });

const onlineUsers = new Map();

app.use("/messages", messageRouter);
app.use("/files", messageRouter);

io.on("connection", (socket) => {
  console.log("New client connected:", socket.id);

  // Register your handlers
  messageHandler(socket);
  fileHandler(io, socket);

  socket.on("join", (userId) => {
    socket.userId = userId;
    socket.join(userId);
    userSockets.set(userId, socket.id);
    onlineUsers.set(userId, { online: true });
    io.emit("update_users", Object.fromEntries(onlineUsers));
    console.log(userId, "joined");
  });

  socket.on("disconnect", () => {
    console.log("Client disconnected:", socket.id);

    // Remove from userSockets map
    for (const [id, sock] of userSockets.entries()) {
      if (sock === socket.id) userSockets.delete(id);
    }

    // Update onlineUsers map
    if (socket.userId) {
      onlineUsers.set(socket.userId, {
        online: false,
        lastSeen: Date.now(),
      });
      io.emit("update_users", Object.fromEntries(onlineUsers));
    }
  });
});


server.listen(PORT, () => {
  console.log(`Socket server is running on Port ${PORT}`);
});