Room.ts 1.98 KB
import { Connection } from "../connection/Connection";
import { v4 as uuidv4 } from "uuid";
import { RoomDescription, RoomInfo } from "./types";
import {
  Message,
  RoomChatMessage,
  RoomUserUpdateMessage,
} from "../message/types";
import { UserData } from "../user/types";
import { User } from "../user/User";

export class Room {
  public readonly uuid: string;

  public name: string;
  public readonly maxUsers: number;

  private users: User[] = [];

  private closed: boolean = false;

  constructor(name: string, maxUsers: number = 8) {
    this.uuid = uuidv4();
    this.name = name;
    this.maxUsers = maxUsers;
  }

  public connect(user: User): void {
    if (this.users.includes(user) || this.users.length >= this.maxUsers) {
      return;
    }

    this.broadcast(new RoomUserUpdateMessage("added", user.getData()));

    this.users.push(user);
    user.room = this; // TODO: 더 나은 관리
  }

  public disconnect(user: User): void {
    const index = this.users.indexOf(user);
    if (index > -1) {
      this.users.splice(index, 1);
      user.room = undefined;

      this.broadcast(new RoomUserUpdateMessage("removed", user.getData()));
    }
  }

  public sendChat(user: User, message: string): void {
    this.broadcast(new RoomChatMessage(message, user.username), user);
  }

  public getDescription(): RoomDescription {
    return {
      uuid: this.uuid,
      name: this.name,
      currentUsers: this.users.length,
      maxUsers: this.maxUsers,
    };
  }

  public getInfo(): RoomInfo {
    var users: UserData[] = this.users.map((u) => u.getData());
    return {
      uuid: this.uuid,
      name: this.name,
      maxUsers: this.maxUsers,
      users: users,
    };
  }

  public broadcast(message: Message, except?: User): void {
    this.users.forEach((u) => {
      if (u !== except) {
        u.connection.send(message);
      }
    });
  }

  public close(): void {
    if (!this.closed) {
      this.users.forEach((u) => this.disconnect(u));
      this.closed = true;
    }
  }
}