import { Room } from "../../room/Room";
import { RoomManager } from "../../room/RoomManager";
import { User } from "../../user/User";
import { SocketTester } from "./SocketTester";

/**
 * 유저들을 만듭니다.
 * @param userCount 만들어지는 유저 수입니다.
 * @returns 만들어진 유저들을 반환합니다.
 */
export function prepareUsers(userCount: number): {
  sockets: SocketTester[];
  users: User[];
  roomManager: RoomManager;
} {
  const roomManager = new RoomManager();
  let sockets = [];
  let users = [];
  for (let i = 0; i < userCount; i++) {
    const socket = new SocketTester(roomManager);
    sockets.push(socket);

    socket.login(`guest${i + 1}`);
    if (!socket.connection.user) {
      throw new Error("User is not initialized.");
    }
    users.push(socket.connection.user);
  }
  return { sockets, users, roomManager };
}

/**
 * 유저들과 빈 방들을 만듭니다.
 * @param userCount 만들어지는 유저 수입니다.
 * @param roomCount 만들어지는 방 수입니다.
 * @param roomMaxConnections 각 방의 최대 접속 인원 수 입니다.
 * @returns 생성된 소켓, 유저, 방을 반환합니다.
 */
export function prepareUsersEmptyRooms(
  userCount: number,
  roomCount: number,
  roomMaxConnections: number = 2
): {
  sockets: SocketTester[];
  users: User[];
  rooms: Room[];
  roomManager: RoomManager;
} {
  const { sockets, users, roomManager } = prepareUsers(userCount);

  let rooms = [];
  for (let i = 0; i < roomCount; i++) {
    rooms.push(roomManager.create(`room${i + 1}`, roomMaxConnections));
  }
  return { sockets, users, rooms, roomManager };
}

/**
 * 방 하나를 만들고 유저들을 접속시킵니다. 이때 첫번째 유저가 방장이 됩니다.
 * @param userCount 방에 접속하는 유저 수 입니다.
 * @param roomMaxConnections 방에 최대 접속할 수 있는 인원 수 입니다.만약 userCount보다 작다면 userCount값으로 변경됩니다.
 * @param emptyAdmin 참이면 방장이 없는 상태로 방이 생성됩니다.
 * @returns 생성된 소켓, 유저, 방을 반환합니다.
 */
export function prepareJoinedRoom(
  userCount: number,
  roomMaxConnections: number = 2,
  emptyAdmin: boolean = false
): { sockets: SocketTester[]; users: User[]; room: Room } {
  roomMaxConnections = Math.max(roomMaxConnections, userCount);
  const roomManager = new RoomManager();
  let sockets = [];
  let users = [];
  for (let i = 0; i < userCount; i++) {
    const socket = new SocketTester(roomManager);
    sockets.push(socket);

    socket.login(`guest${i + 1}`);
    if (!socket.connection.user) {
      throw new Error("User is not initialized.");
    }
    users.push(socket.connection.user);
  }

  const room = roomManager.create(
    "room1",
    roomMaxConnections,
    emptyAdmin ? undefined : users[0]
  );
  for (let i = emptyAdmin ? 0 : 1; i < userCount; i++) {
    sockets[i].testOk("joinRoom", { uuid: room.uuid });
  }
  return { sockets, users, room };
}