@colyseus/core 0.16.0-preview.3 → 0.16.0-preview.31

package/build/Room.mjs

@@ -1,57 +1,96 @@
-import { unpack } from "msgpackr";
+// packages/core/src/Room.ts
+import { unpack } from "@colyseus/msgpackr";
 import { decode, $changes } from "@colyseus/schema";
-import Clock from "@gamestdio/timer";
+import Clock from "@colyseus/timer";
 import { EventEmitter } from "events";
-import { logger } from "./Logger";
-import { NoneSerializer } from "./serializer/NoneSerializer";
-import { SchemaSerializer } from "./serializer/SchemaSerializer";
-import { ErrorCode, getMessageBytes, Protocol } from "./Protocol";
-import { Deferred, generateId } from "./utils/Utils";
-import { isDevMode } from "./utils/DevMode";
-import { debugAndPrintError, debugMessage } from "./Debug";
-import { ServerError } from "./errors/ServerError";
-import { ClientArray, ClientState } from "./Transport";
-const DEFAULT_PATCH_RATE = 1e3 / 20;
-const DEFAULT_SIMULATION_INTERVAL = 1e3 / 60;
-const noneSerializer = new NoneSerializer();
-const DEFAULT_SEAT_RESERVATION_TIME = Number(process.env.COLYSEUS_SEAT_RESERVATION_TIME || 15);
+import { logger } from "./Logger.mjs";
+import { NoneSerializer } from "./serializer/NoneSerializer.mjs";
+import { SchemaSerializer } from "./serializer/SchemaSerializer.mjs";
+import { ErrorCode, getMessageBytes, Protocol } from "./Protocol.mjs";
+import { Deferred, generateId } from "./utils/Utils.mjs";
+import { isDevMode } from "./utils/DevMode.mjs";
+import { debugAndPrintError, debugMessage } from "./Debug.mjs";
+import { ServerError } from "./errors/ServerError.mjs";
+import { ClientArray, ClientState } from "./Transport.mjs";
+var DEFAULT_PATCH_RATE = 1e3 / 20;
+var DEFAULT_SIMULATION_INTERVAL = 1e3 / 60;
+var noneSerializer = new NoneSerializer();
+var DEFAULT_SEAT_RESERVATION_TIME = Number(process.env.COLYSEUS_SEAT_RESERVATION_TIME || 15);
 var RoomInternalState = /* @__PURE__ */ ((RoomInternalState2) => {
   RoomInternalState2[RoomInternalState2["CREATING"] = 0] = "CREATING";
   RoomInternalState2[RoomInternalState2["CREATED"] = 1] = "CREATED";
   RoomInternalState2[RoomInternalState2["DISPOSING"] = 2] = "DISPOSING";
   return RoomInternalState2;
 })(RoomInternalState || {});
-class Room {
+var Room = class _Room {
   constructor() {
+    /**
+     * Timing events tied to the room instance.
+     * Intervals and timeouts are cleared when the room is disposed.
+     */
     this.clock = new Clock();
-    this.#_autoDispose = true;
+    /**
+     * Maximum number of clients allowed to connect into the room. When room reaches this limit,
+     * it is locked automatically. Unless the room was explicitly locked by you via `lock()` method,
+     * the room will be unlocked as soon as a client disconnects from it.
+     */
     this.maxClients = Infinity;
+    /**
+     * Automatically dispose the room when last client disconnects.
+     *
+     * @default true
+     */
+    this.autoDispose = true;
+    /**
+     * Frequency to send the room state to connected clients, in milliseconds.
+     *
+     * @default 50ms (20fps)
+     */
     this.patchRate = DEFAULT_PATCH_RATE;
+    /**
+     * The array of connected clients.
+     *
+     * @see {@link https://docs.colyseus.io/colyseus/server/room/#client|Client instance}
+     */
     this.clients = new ClientArray();
+    /** @internal */
     this._events = new EventEmitter();
+    // seat reservation & reconnection
     this.seatReservationTime = DEFAULT_SEAT_RESERVATION_TIME;
     this.reservedSeats = {};
     this.reservedSeatTimeouts = {};
     this._reconnections = {};
     this._reconnectingSessionId = /* @__PURE__ */ new Map();
-    this.onMessageHandlers = {};
+    this.onMessageHandlers = {
+      "__no_message_handler": {
+        callback: (client, messageType, _) => {
+          const errorMessage = `onMessage for "${messageType}" not registered.`;
+          debugAndPrintError(errorMessage);
+          if (isDevMode) {
+            client.error(ErrorCode.INVALID_PAYLOAD, errorMessage);
+          } else {
+            client.leave(Protocol.WS_CLOSE_WITH_ERROR, errorMessage);
+          }
+        }
+      }
+    };
     this._serializer = noneSerializer;
     this._afterNextPatchQueue = [];
     this._internalState = 0 /* CREATING */;
     this._locked = false;
     this._lockedExplicitly = false;
     this._maxClientsReached = false;
-    this._events.once("dispose", async () => {
-      try {
-        await this._dispose();
-      } catch (e) {
-        debugAndPrintError(`onDispose error: ${e && e.message || e || "promise rejected"}`);
-      }
-      this._events.emit("disconnect");
+    this._events.once("dispose", () => {
+      this._dispose().catch((e) => debugAndPrintError(`onDispose error: ${e && e.message || e || "promise rejected"}`)).finally(() => this._events.emit("disconnect"));
     });
-    this.setPatchRate(this.patchRate);
-    this.resetAutoDisposeTimeout(this.seatReservationTime);
   }
+  /**
+   * This property will change on these situations:
+   * - The maximum number of allowed clients has been reached (`maxClients`)
+   * - You manually locked, or unlocked the room using lock() or `unlock()`.
+   *
+   * @readonly
+   */
   get locked() {
     return this._locked;
   }
@@ -61,64 +100,149 @@ class Room {
   #_roomId;
   #_roomName;
   #_autoDispose;
-  get autoDispose() {
-    return this.#_autoDispose;
-  }
-  set autoDispose(value) {
-    if (value !== this.#_autoDispose && this._internalState !== 2 /* DISPOSING */) {
-      this.#_autoDispose = value;
-      this.resetAutoDisposeTimeout();
-    }
+  #_patchRate;
+  #_patchInterval;
+  __init() {
+    if (this.state) {
+      this.setState(this.state);
+    }
+    this.#_autoDispose = this.autoDispose;
+    this.#_patchRate = this.patchRate;
+    Object.defineProperties(this, {
+      autoDispose: {
+        enumerable: true,
+        get: () => this.#_autoDispose,
+        set: (value) => {
+          if (value !== this.#_autoDispose && this._internalState !== 2 /* DISPOSING */) {
+            this.#_autoDispose = value;
+            this.resetAutoDisposeTimeout();
+          }
+        }
+      },
+      patchRate: {
+        enumerable: true,
+        get: () => this.#_patchRate,
+        set: (milliseconds) => {
+          this.#_patchRate = milliseconds;
+          if (this.#_patchInterval) {
+            clearInterval(this.#_patchInterval);
+            this.#_patchInterval = void 0;
+          }
+          if (milliseconds !== null && milliseconds !== 0) {
+            this.#_patchInterval = setInterval(() => this.broadcastPatch(), milliseconds);
+          }
+        }
+      }
+    });
+    this.patchRate = this.#_patchRate;
+    this.resetAutoDisposeTimeout(this.seatReservationTime);
   }
+  /**
+   * The name of the room you provided as first argument for `gameServer.define()`.
+   *
+   * @returns roomName string
+   */
   get roomName() {
     return this.#_roomName;
   }
+  /**
+   * Setting the name of the room. Overwriting this property is restricted.
+   *
+   * @param roomName
+   */
   set roomName(roomName) {
     if (this.#_roomName) {
       throw new ServerError(ErrorCode.APPLICATION_ERROR, "'roomName' cannot be overwritten.");
     }
     this.#_roomName = roomName;
   }
+  /**
+   * A unique, auto-generated, 9-character-long id of the room.
+   * You may replace `this.roomId` during `onCreate()`.
+   *
+   * @returns roomId string
+   */
   get roomId() {
     return this.#_roomId;
   }
+  /**
+   * Setting the roomId, is restricted in room lifetime except upon room creation.
+   *
+   * @param roomId
+   * @returns roomId string
+   */
   set roomId(roomId) {
     if (this._internalState !== 0 /* CREATING */ && !isDevMode) {
       throw new ServerError(ErrorCode.APPLICATION_ERROR, "'roomId' can only be overridden upon room creation.");
     }
     this.#_roomId = roomId;
   }
+  // TODO: flag as @deprecated on v0.16
+  // TOOD: remove instance level `onAuth` on 1.0
+  /**
+   * onAuth at the instance level will be deprecated in the future.
+   * Please use "static onAuth(token, req) instead
+   */
   onAuth(client, options, request) {
     return true;
   }
   static async onAuth(token, req) {
     return true;
   }
+  /**
+   * Returns whether the sum of connected clients and reserved seats exceeds maximum number of clients.
+   *
+   * @returns boolean
+   */
   hasReachedMaxClients() {
     return this.clients.length + Object.keys(this.reservedSeats).length >= this.maxClients || this._internalState === 2 /* DISPOSING */;
   }
+  /**
+   * Set the number of seconds a room can wait for a client to effectively join the room.
+   * You should consider how long your `onAuth()` will have to wait for setting a different seat reservation time.
+   * The default value is 15 seconds. You may set the `COLYSEUS_SEAT_RESERVATION_TIME`
+   * environment variable if you'd like to change the seat reservation time globally.
+   *
+   * @default 15 seconds
+   *
+   * @param seconds - number of seconds.
+   * @returns The modified Room object.
+   */
   setSeatReservationTime(seconds) {
     this.seatReservationTime = seconds;
     return this;
   }
   hasReservedSeat(sessionId, reconnectionToken) {
-    if (reconnectionToken) {
-      const reconnection = this._reconnections[reconnectionToken];
-      return reconnection && reconnection[0] === sessionId && this.reservedSeats[sessionId] !== void 0 && this._reconnectingSessionId.has(sessionId);
+    const reservedSeat = this.reservedSeats[sessionId];
+    if (reservedSeat === void 0) {
+      return false;
+    }
+    if (reservedSeat[3]) {
+      return reconnectionToken && this._reconnections[reconnectionToken]?.[0] === sessionId && this._reconnectingSessionId.has(sessionId);
     } else {
-      return this.reservedSeats[sessionId] !== void 0 && (!this._reconnectingSessionId.has(sessionId) || this._reconnectingSessionId.get(sessionId) === sessionId);
+      return reservedSeat[2] === false;
     }
   }
   checkReconnectionToken(reconnectionToken) {
-    const reconnection = this._reconnections[reconnectionToken];
-    const sessionId = reconnection && reconnection[0];
-    if (this.hasReservedSeat(sessionId)) {
+    const sessionId = this._reconnections[reconnectionToken]?.[0];
+    const reservedSeat = this.reservedSeats[sessionId];
+    if (reservedSeat && reservedSeat[3]) {
       this._reconnectingSessionId.set(sessionId, reconnectionToken);
       return sessionId;
     } else {
       return void 0;
     }
   }
+  /**
+   * (Optional) Set a simulation interval that can change the state of the game.
+   * The simulation interval is your game loop.
+   *
+   * @default 16.6ms (60fps)
+   *
+   * @param onTickCallback - You can implement your physics or world updates here!
+   *  This is a good place to update the room state.
+   * @param delay - Interval delay on executing `onTickCallback` in milliseconds.
+   */
   setSimulationInterval(onTickCallback, delay = DEFAULT_SIMULATION_INTERVAL) {
     if (this._simulationInterval) {
       clearInterval(this._simulationInterval);
@@ -130,15 +254,11 @@ class Room {
       }, delay);
     }
   }
+  /**
+   * @deprecated Use `.patchRate=` instead.
+   */
   setPatchRate(milliseconds) {
     this.patchRate = milliseconds;
-    if (this._patchInterval) {
-      clearInterval(this._patchInterval);
-      this._patchInterval = void 0;
-    }
-    if (milliseconds !== null && milliseconds !== 0) {
-      this._patchInterval = setInterval(() => this.broadcastPatch(), milliseconds);
-    }
   }
   setState(newState) {
     this.clock.start();
@@ -172,14 +292,16 @@ class Room {
     }
   }
   async setPrivate(bool = true) {
-    if (this.listing.private === bool)
-      return;
+    if (this.listing.private === bool) return;
     this.listing.private = bool;
     if (this._internalState === 1 /* CREATED */) {
       await this.listing.save();
     }
     this._events.emit("visibility-change", bool);
   }
+  /**
+   * Locking the room will remove it from the pool of available rooms for new clients to connect to.
+   */
   async lock() {
     this._lockedExplicitly = arguments[0] === void 0;
     if (this._locked) {
@@ -191,6 +313,9 @@ class Room {
     });
     this._events.emit("lock");
   }
+  /**
+   * Unlocking the room returns it to the pool of available rooms for new clients to connect to.
+   */
   async unlock() {
     if (arguments[0] === void 0) {
       this._lockedExplicitly = false;
@@ -208,16 +333,28 @@ class Room {
     logger.warn("DEPRECATION WARNING: use client.send(...) instead of this.send(client, ...)");
     client.send(messageOrType, messageOrOptions, options);
   }
-  broadcast(typeOrSchema, messageOrOptions, options) {
-    const isSchema = typeof typeOrSchema === "object";
-    const opts = isSchema ? messageOrOptions : options;
-    if (opts && opts.afterNextPatch) {
-      delete opts.afterNextPatch;
+  broadcast(type, message, options) {
+    if (options && options.afterNextPatch) {
+      delete options.afterNextPatch;
       this._afterNextPatchQueue.push(["broadcast", arguments]);
       return;
     }
-    this.broadcastMessageType(typeOrSchema, messageOrOptions, opts);
+    this.broadcastMessageType(type, message, options);
+  }
+  /**
+   * Broadcast bytes (UInt8Arrays) to a particular room
+   */
+  broadcastBytes(type, message, options) {
+    if (options && options.afterNextPatch) {
+      delete options.afterNextPatch;
+      this._afterNextPatchQueue.push(["broadcastBytes", arguments]);
+      return;
+    }
+    this.broadcastMessageType(type, message, options);
   }
+  /**
+   * Checks whether mutations have occurred in the state, and broadcast them to all connected clients.
+   */
   broadcastPatch() {
     if (this.onBeforePatch) {
       this.onBeforePatch(this.state);
@@ -232,10 +369,16 @@ class Room {
     this._dequeueAfterPatchMessages();
     return hasChanges;
   }
-  onMessage(messageType, callback) {
-    this.onMessageHandlers[messageType] = callback;
+  onMessage(messageType, callback, validate) {
+    this.onMessageHandlers[messageType] = { callback, validate };
     return () => delete this.onMessageHandlers[messageType];
   }
+  /**
+   * Disconnect all connected clients, and then dispose the room.
+   *
+   * @param closeCode WebSocket close code (default = 4000, which is a "consented leave")
+   * @returns Promise<void>
+   */
   disconnect(closeCode = Protocol.WS_CLOSE_CONSENTED) {
     if (this._internalState === 2 /* DISPOSING */) {
       return Promise.resolve(`disconnect() ignored: room (${this.roomId}) is already disposing.`);
@@ -261,7 +404,7 @@ class Room {
   }
   async ["_onJoin"](client, req) {
     const sessionId = client.sessionId;
-    client._reconnectionToken = generateId();
+    client.reconnectionToken = generateId();
     if (this.reservedSeatTimeouts[sessionId]) {
       clearTimeout(this.reservedSeatTimeouts[sessionId]);
       delete this.reservedSeatTimeouts[sessionId];
@@ -270,23 +413,28 @@ class Room {
       clearTimeout(this._autoDisposeTimeout);
       this._autoDisposeTimeout = void 0;
     }
-    const [joinOptions, authData] = this.reservedSeats[sessionId];
-    if (this.reservedSeats[sessionId].length > 2) {
+    const [joinOptions, authData, isConsumed, isWaitingReconnection] = this.reservedSeats[sessionId];
+    if (isConsumed) {
       throw new ServerError(ErrorCode.MATCHMAKE_EXPIRED, "already consumed");
     }
-    this.reservedSeats[sessionId].push(true);
+    this.reservedSeats[sessionId][2] = true;
     client._afterNextPatchQueue = this._afterNextPatchQueue;
     client.ref["onleave"] = (_) => client.state = ClientState.LEAVING;
     client.ref.once("close", client.ref["onleave"]);
-    const previousReconnectionToken = this._reconnectingSessionId.get(sessionId);
-    if (previousReconnectionToken) {
-      this.clients.push(client);
-      this._reconnections[previousReconnectionToken]?.[1].resolve(client);
+    if (isWaitingReconnection) {
+      const previousReconnectionToken = this._reconnectingSessionId.get(sessionId);
+      if (previousReconnectionToken) {
+        this.clients.push(client);
+        await this._reconnections[previousReconnectionToken]?.[1].resolve(client);
+      } else {
+        const errorMessage = process.env.NODE_ENV === "production" ? "already consumed" : "bad reconnection token";
+        throw new ServerError(ErrorCode.MATCHMAKE_EXPIRED, errorMessage);
+      }
     } else {
       try {
         if (authData) {
           client.auth = authData;
-        } else if (this.onAuth !== Room.prototype.onAuth) {
+        } else if (this.onAuth !== _Room.prototype.onAuth) {
           client.auth = await this.onAuth(client, joinOptions, req);
           if (!client.auth) {
             throw new ServerError(ErrorCode.AUTH_FAILED, "onAuth failed");
@@ -324,15 +472,25 @@ class Room {
       client.ref.once("close", client.ref["onleave"]);
       client.ref.on("message", this._onMessage.bind(this, client));
       client.raw(getMessageBytes[Protocol.JOIN_ROOM](
-        client._reconnectionToken,
+        client.reconnectionToken,
         this._serializer.id,
         this._serializer.handshake && this._serializer.handshake()
       ));
     }
   }
+  /**
+   * Allow the specified client to reconnect into the room. Must be used inside `onLeave()` method.
+   * If seconds is provided, the reconnection is going to be cancelled after the provided amount of seconds.
+   *
+   * @param previousClient - The client which is to be waiting until re-connection happens.
+   * @param seconds - Timeout period on re-connection in seconds.
+   *
+   * @returns Deferred<Client> - The differed is a promise like type.
+   *  This type can forcibly reject the promise by calling `.reject()`.
+   */
   allowReconnection(previousClient, seconds) {
     if (previousClient._enqueuedMessages !== void 0) {
-      return;
+      return Deferred.reject("client not joined");
     }
     if (seconds === void 0) {
       console.warn('DEPRECATED: allowReconnection() requires a second argument. Using "manual" mode.');
@@ -343,10 +501,10 @@ class Room {
     }
     if (this._internalState === 2 /* DISPOSING */) {
       this._disposeIfEmpty();
-      throw new Error("disconnecting");
+      return Deferred.reject("disconnecting");
     }
     const sessionId = previousClient.sessionId;
-    const reconnectionToken = previousClient._reconnectionToken;
+    const reconnectionToken = previousClient.reconnectionToken;
     this._reserveSeat(sessionId, true, previousClient.auth, seconds, true);
     const reconnection = new Deferred();
     this._reconnections[reconnectionToken] = [sessionId, reconnection];
@@ -384,7 +542,7 @@ class Room {
   }
   broadcastMessageType(type, message, options = {}) {
     debugMessage("broadcast: %O", message);
-    const encodedMessage = getMessageBytes.raw(Protocol.ROOM_DATA, type, message);
+    const encodedMessage = message instanceof Uint8Array ? getMessageBytes.raw(Protocol.ROOM_DATA_BYTES, type, void 0, message) : getMessageBytes.raw(Protocol.ROOM_DATA, type, message);
     const except = typeof options.except !== "undefined" ? Array.isArray(options.except) ? options.except : [options.except] : void 0;
     let numClients = this.clients.length;
     while (numClients--) {
@@ -395,7 +553,7 @@ class Room {
     }
   }
   sendFullState(client) {
-    client.enqueueRaw(this._serializer.getFullState(client));
+    client.raw(this._serializer.getFullState(client));
   }
   _dequeueAfterPatchMessages() {
     const length = this._afterNextPatchQueue.length;
@@ -415,7 +573,7 @@ class Room {
     if (!allowReconnection && this.hasReachedMaxClients()) {
       return false;
     }
-    this.reservedSeats[sessionId] = [joinOptions, authData];
+    this.reservedSeats[sessionId] = [joinOptions, authData, false, allowReconnection];
     if (!allowReconnection) {
       await this._incrementClientCount();
       this.reservedSeatTimeouts[sessionId] = setTimeout(async () => {
@@ -439,14 +597,14 @@ class Room {
   }
   async _dispose() {
     this._internalState = 2 /* DISPOSING */;
-    await this.listing.remove();
+    this.listing.remove();
     let userReturnData;
     if (this.onDispose) {
       userReturnData = this.onDispose();
     }
-    if (this._patchInterval) {
-      clearInterval(this._patchInterval);
-      this._patchInterval = void 0;
+    if (this.#_patchInterval) {
+      clearInterval(this.#_patchInterval);
+      this.#_patchInterval = void 0;
     }
     if (this._simulationInterval) {
       clearInterval(this._simulationInterval);
@@ -460,59 +618,52 @@ class Room {
     this.clock.stop();
     return await (userReturnData || Promise.resolve());
   }
-  _onMessage(client, bytes) {
+  _onMessage(client, buffer) {
     if (client.state === ClientState.LEAVING) {
       return;
     }
-    const it = { offset: 0 };
-    const code = decode.uint8(bytes, it);
-    if (!bytes) {
-      debugAndPrintError(`${this.roomName} (${this.roomId}), couldn't decode message: ${bytes}`);
+    const it = { offset: 1 };
+    const code = buffer[0];
+    if (!buffer) {
+      debugAndPrintError(`${this.roomName} (${this.roomId}), couldn't decode message: ${buffer}`);
       return;
     }
     if (code === Protocol.ROOM_DATA) {
-      const messageType = decode.stringCheck(bytes, it) ? decode.string(bytes, it) : decode.number(bytes, it);
+      const messageType = decode.stringCheck(buffer, it) ? decode.string(buffer, it) : decode.number(buffer, it);
+      const messageTypeHandler = this.onMessageHandlers[messageType];
       let message;
       try {
-        message = bytes.length > it.offset ? unpack(new Uint8Array(bytes.slice(it.offset, bytes.length))) : void 0;
+        message = buffer.byteLength > it.offset ? unpack(buffer.subarray(it.offset, buffer.byteLength)) : void 0;
         debugMessage("received: '%s' -> %j", messageType, message);
+        if (messageTypeHandler?.validate !== void 0) {
+          message = messageTypeHandler.validate(message);
+        }
       } catch (e) {
         debugAndPrintError(e);
         client.leave(Protocol.WS_CLOSE_WITH_ERROR);
         return;
       }
-      if (this.onMessageHandlers[messageType]) {
-        this.onMessageHandlers[messageType](client, message);
-      } else if (this.onMessageHandlers["*"]) {
-        this.onMessageHandlers["*"](client, messageType, message);
+      if (messageTypeHandler) {
+        messageTypeHandler.callback(client, message);
       } else {
-        const errorMessage = `onMessage for "${messageType}" not registered.`;
-        debugAndPrintError(errorMessage);
-        if (isDevMode) {
-          client.error(ErrorCode.INVALID_PAYLOAD, errorMessage);
-        } else {
-          client.leave(Protocol.WS_CLOSE_WITH_ERROR, errorMessage);
-        }
+        (this.onMessageHandlers["*"] || this.onMessageHandlers["__no_message_handler"]).callback(client, messageType, message);
       }
     } else if (code === Protocol.ROOM_DATA_BYTES) {
-      const messageType = decode.stringCheck(bytes, it) ? decode.string(bytes, it) : decode.number(bytes, it);
-      const message = bytes.slice(it.offset, bytes.length);
+      const messageType = decode.stringCheck(buffer, it) ? decode.string(buffer, it) : decode.number(buffer, it);
+      const messageTypeHandler = this.onMessageHandlers[messageType];
+      let message = buffer.subarray(it.offset, buffer.byteLength);
       debugMessage("received: '%s' -> %j", messageType, message);
-      if (this.onMessageHandlers[messageType]) {
-        this.onMessageHandlers[messageType](client, message);
-      } else if (this.onMessageHandlers["*"]) {
-        this.onMessageHandlers["*"](client, messageType, message);
+      if (messageTypeHandler?.validate !== void 0) {
+        message = messageTypeHandler.validate(message);
+      }
+      if (messageTypeHandler) {
+        messageTypeHandler.callback(client, message);
       } else {
-        const errorMessage = `onMessage for "${messageType}" not registered.`;
-        debugAndPrintError(errorMessage);
-        if (isDevMode) {
-          client.error(ErrorCode.INVALID_PAYLOAD, errorMessage);
-        } else {
-          client.leave(Protocol.WS_CLOSE_WITH_ERROR, errorMessage);
-        }
+        (this.onMessageHandlers["*"] || this.onMessageHandlers["__no_message_handler"]).callback(client, messageType, message);
       }
     } else if (code === Protocol.JOIN_ROOM && client.state === ClientState.JOINING) {
       client.state = ClientState.JOINED;
+      client._joinedAt = this.clock.elapsedTime;
       if (this.state) {
         this.sendFullState(client);
       }
@@ -541,11 +692,18 @@ class Room {
         }
       }
     }
-    if (client.state !== ClientState.RECONNECTED) {
-      const willDispose = await this._decrementClientCount();
-      if (this.reservedSeats[client.sessionId] === void 0) {
-        this._events.emit("leave", client, willDispose);
-      }
+    if (this._reconnections[client.reconnectionToken]) {
+      this._reconnections[client.reconnectionToken][1].catch(async () => {
+        await this._onAfterLeave(client);
+      });
+    } else if (client.state !== ClientState.RECONNECTED) {
+      await this._onAfterLeave(client);
+    }
+  }
+  async _onAfterLeave(client) {
+    const willDispose = await this._decrementClientCount();
+    if (this.reservedSeats[client.sessionId] === void 0) {
+      this._events.emit("leave", client, willDispose);
     }
   }
   async _incrementClientCount() {
@@ -575,7 +733,7 @@ class Room {
     }
     return willDispose;
   }
-}
+};
 export {
   DEFAULT_SEAT_RESERVATION_TIME,
   Room,