server.ts

import { createServer } from "http";
import next from "next";
import { Server } from "socket.io";
import { generateRoomId } from "./utils";

const app = next({ dev: process.env.NODE_ENV !== "production" });
const handler = app.getRequestHandler();

const rooms: { [key: string]: string[] } = {}; // ✅ 채팅방 목록 관리

app.prepare().then(() => {
  const httpServer = createServer(handler);
  const io = new Server(httpServer, {
    path: "/socket.io",
    cors: { origin: "*" },
    transports: ["websocket"],
  });

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

    // ✅ 채팅방 입장 (유저 ID 기준으로 고유한 채팅방 ID 생성)
    socket.on("joinRoom", ({ user1, user2 }) => {
      const roomId = generateRoomId(user1, user2);

      if (!rooms[roomId]) {
        rooms[roomId] = [];
      }

      if (!rooms[roomId].includes(socket.id)) {
        rooms[roomId].push(socket.id);
      }

      socket.join(roomId);
      console.log(`🔗 ${socket.id} joined room: ${roomId}`);
    });
    socket.on("leaveRoom", (roomId) => {
      if (!rooms[roomId]) {
        console.log("❌ 존재하지 않는 채팅방입니다:", roomId);
        return;
      }

      delete rooms[roomId]; // ✅ 서버에서 채팅방 제거
      console.log(`🗑 채팅방 삭제됨: ${roomId}`);
    });   

    // ✅ 채팅 메시지 전송 (송신자를 제외하고 메시지 전달)
    socket.on("message", ({ user1, user2, msg }) => {
      const roomId = generateRoomId(user1, user2); // ✅ 올바른 채팅방 ID 유지

      if (!rooms[roomId]?.includes(socket.id)) {
        console.error("❌ User is not in the room!");
        return;
      }

      console.log(`📩 Message received in ${roomId}:`, msg);

      // ✅ 송신자에게는 메시지를 보내지 않음 (중복 출력 방지)
      socket.to(roomId).emit("message", { roomId, msg });
    });

    // ✅ 클라이언트 연결 종료 처리
    socket.on("disconnect", () => {
      console.log("❌ Client disconnected:", socket.id);
      for (const roomId in rooms) {
        rooms[roomId] = rooms[roomId].filter((id) => id !== socket.id);
      }
    });
  });

  httpServer.listen(3000, () => {
    console.log(`🚀 Server ready on http://localhost:3000`);
  });
});