@pluv/io 0.20.0 → 0.21.1

package/src/IORoom.ts

@@ -18,16 +18,21 @@ import type {
 import colors from "kleur";
 import type {
     AbstractPlatform,
-    InferPlatformEventContextType,
-    InferPlatformRoomContextType,
+    InferPlatformWebSocketSource,
     InferPlatformWebSocketType,
+    InferRoomContextType,
 } from "./AbstractPlatform";
-import { AbstractMessageEvent } from "./AbstractWebSocket";
+import { AbstractCloseEvent, AbstractErrorEvent, AbstractMessageEvent, AbstractWebSocket } from "./AbstractWebSocket";
 import type { PluvRouter, PluvRouterEventConfig } from "./PluvRouter";
 import { authorize } from "./authorize";
-import type { EventResolverContext, IORoomListenerEvent, SendMessageOptions, WebSocketSession } from "./types";
-
-const PING_TIMEOUT_MS = 30_000;
+import { PING_TIMEOUT_MS } from "./constants";
+import type {
+    EventResolverContext,
+    IORoomListenerEvent,
+    SendMessageOptions,
+    WebSocketSession,
+    WebSocketType,
+} from "./types";
 
 type BroadcastMessage<TIO extends IORoom<any, any, any, any>> =
     | InferEventMessage<InferIOInput<TIO>>
@@ -51,12 +56,12 @@ export type BroadcastProxy<TIO extends IORoom<any, any, any, any>> = (<TEvent ex
 
 export type IORoomConfig<
     TPlatform extends AbstractPlatform<any> = AbstractPlatform<any>,
-    TAuthorize extends IOAuthorize<any, any, InferPlatformRoomContextType<TPlatform>> = BaseIOAuthorize,
+    TAuthorize extends IOAuthorize<any, any, InferRoomContextType<TPlatform>> = BaseIOAuthorize,
     TContext extends JsonObject = {},
     TEvents extends PluvRouterEventConfig<TPlatform, TAuthorize, TContext> = {},
 > = Partial<IORoomListeners<TPlatform>> & {
     authorize?: TAuthorize;
-    context: TContext & InferPlatformRoomContextType<TPlatform>;
+    context: TContext & InferRoomContextType<TPlatform>;
     crdt?: { doc: (value: any) => AbstractCrdtDocFactory<any> };
     debug: boolean;
     platform: TPlatform;
@@ -64,30 +69,29 @@ export type IORoomConfig<
 };
 
 interface SendMessageSender {
-    id: string | null;
+    sessionId: string | null;
     user: JsonObject | null;
 }
 
-export type WebsocketRegisterOptions<TPlatform extends AbstractPlatform> = {
+export interface WebsocketRegisterOptions {
     token?: string | null;
-} & InferPlatformRoomContextType<TPlatform> &
-    InferPlatformEventContextType<TPlatform>;
+}
 
 export class IORoom<
     TPlatform extends AbstractPlatform<any> = AbstractPlatform<any>,
-    TAuthorize extends IOAuthorize<any, any, InferPlatformRoomContextType<TPlatform>> = BaseIOAuthorize,
+    TAuthorize extends IOAuthorize<any, any, InferRoomContextType<TPlatform>> = BaseIOAuthorize,
     TContext extends Record<string, any> = {},
     TEvents extends PluvRouterEventConfig<TPlatform, TAuthorize, TContext> = {},
 > implements IOLike<TAuthorize, TEvents>
 {
-    private readonly _context: TContext & InferPlatformRoomContextType<TPlatform>;
+    private readonly _context: TContext & InferRoomContextType<TPlatform>;
     private readonly _debug: boolean;
     private readonly _docFactory: AbstractCrdtDocFactory<any>;
     private readonly _platform: TPlatform;
 
     private _doc: AbstractCrdtDoc<any>;
     private _listeners: IORoomListeners<TPlatform>;
-    private _sessions = new Map<string, WebSocketSession<TAuthorize>>();
+    private _sessions = new Map<string, AbstractWebSocket>();
     private _uninitialize: (() => Promise<void>) | null = null;
 
     readonly _authorize: TAuthorize | null = null;
@@ -124,7 +128,7 @@ export class IORoom<
         this._docFactory = crdt.doc(() => ({}));
         this._doc = this._docFactory.getEmpty();
         this._router = router;
-        this._platform = platform;
+        this._platform = platform.initialize({ context });
 
         this.id = id;
 
@@ -133,6 +137,20 @@ export class IORoom<
         };
 
         if (authorize) this._authorize = authorize;
+
+        const webSockets = this._platform.getWebSockets() as readonly InferPlatformWebSocketSource<TPlatform>[];
+
+        if (!webSockets.length) return;
+
+        webSockets.forEach((webSocket) => {
+            const deserialized = this._platform.getSerializedState(webSocket);
+            const sessionId = this._platform.getSessionId(webSocket);
+
+            if (!deserialized) return;
+            if (typeof sessionId !== "string") return;
+
+            this._sessions.set(sessionId, this._platform.convertWebSocket(webSocket, { room: this.id }));
+        });
     }
 
     public getSize(): number {
@@ -144,147 +162,177 @@ export class IORoom<
          * @author leedavidcs
          * @date December 21, 2022
          */
-        return Array.from(this._sessions.values()).reduce((count, session) => {
-            if (session.quit) return count;
+        return Array.from(this._sessions.values()).reduce((count, pluvWs) => {
+            if (pluvWs.state.quit) return count;
+
+            const pingTime = this._platform.getLastPing(pluvWs) ?? pluvWs.state.timers.ping;
 
-            return currentTime - session.timers.ping > PING_TIMEOUT_MS ? count : count + 1;
+            return currentTime - pingTime > PING_TIMEOUT_MS ? count : count + 1;
         }, 0);
     }
 
+    public onClose(webSocket: WebSocketType<TPlatform>): (event: AbstractCloseEvent) => void {
+        this._ensureDetached();
+
+        const wsSession = this._getAbstractWs(webSocket);
+
+        if (!wsSession) return (): void => undefined;
+
+        return this._onClose(wsSession);
+    }
+
+    public onError(webSocket: WebSocketType<TPlatform>): (event: AbstractErrorEvent) => void {
+        this._ensureDetached();
+
+        const wsSession = this._getAbstractWs(webSocket);
+
+        if (!wsSession) return (): void => undefined;
+
+        return this._onClose(wsSession);
+    }
+
+    public onMessage(webSocket: WebSocketType<TPlatform>): (event: AbstractMessageEvent) => void {
+        this._ensureDetached();
+
+        const wsSession = this._getAbstractWs(webSocket);
+
+        if (!wsSession) return (): void => undefined;
+
+        return this._onMessage(wsSession);
+    }
+
     public async register(
-        webSocket: InferPlatformWebSocketType<TPlatform>,
-        options: WebsocketRegisterOptions<TPlatform>,
+        webSocket: InferPlatformWebSocketSource<TPlatform>,
+        options: WebsocketRegisterOptions = {},
     ): Promise<void> {
-        const { token, ..._platformEventContext } = options;
-        const platformEventContext = _platformEventContext as InferPlatformRoomContextType<TPlatform> &
-            InferPlatformEventContextType<TPlatform>;
+        const { token } = options;
 
         if (!this._uninitialize) await this._initialize();
 
-        const user = await this._getAuthorizedUser(token, options);
+        const user = await this._getAuthorizedUser(token);
+        const ioAuthorize = this._getIOAuthorize();
+        const isUnauthorized = !!ioAuthorize?.required && !user;
 
-        const sessionId = this._platform.randomUUID();
-        const pluvWs = this._platform.convertWebSocket(webSocket, {
-            room: this.id,
-            sessionId,
-            userId: user?.id ?? null,
-        });
+        // There is an attempt for multiple of the same identities. Probably malicious.
+        const isTokenInUse: boolean =
+            !!user &&
+            (await this._getSessions().then((sessions) => sessions.some((session) => session.user?.id === user.id)));
 
-        this._logDebug(`${colors.blue(`Registering connection for room ${this.id}:`)} ${sessionId}`);
+        const pluvWs = await this._platform.convertWebSocket(webSocket, { room: this.id });
 
-        const uninitializeWs = await pluvWs.initialize();
-        const ioAuthorize = this._getIOAuthorize(options);
+        this._logDebug(`${colors.blue(`Registering connection for room ${this.id}:`)} ${pluvWs.sessionId}`);
 
-        const isUnauthorized = !!ioAuthorize?.required && !user;
-        // There is an attempt for multiple of the same identities. Probably malicious.
-        const isTokenInUse =
-            !!user && Array.from(this._sessions.values()).some((session) => session.user?.id === user.id);
+        await this._platform.acceptWebSocket(pluvWs);
 
         if (isUnauthorized || isTokenInUse) {
-            this._logDebug(`${colors.blue("Authorization failed for connection:")} ${sessionId}`);
+            this._logDebug(`${colors.blue("Authorization failed for connection:")} ${pluvWs.sessionId}`);
 
-            pluvWs.handleError({ error: new Error("Not authorized") });
+            pluvWs.handleError({ error: new Error("Not authorized"), room: this.id });
             pluvWs.close(3000, "WebSocket unauthorized.");
 
             return;
         }
 
-        const currentTime = new Date().getTime();
-        const session: WebSocketSession<TAuthorize> = {
-            id: sessionId,
-            presence: null,
-            quit: false,
-            room: this.id,
-            timers: {
-                ping: currentTime,
-            },
-            webSocket: pluvWs,
-            user,
-        };
-
-        this._sessions.set(session.id, session);
+        this._sessions.set(pluvWs.sessionId, pluvWs);
 
-        await this._platform.persistance.addUser(this.id, sessionId, user ?? {});
+        await this._platform.persistance.addUser(this.id, pluvWs.sessionId, user ?? {});
 
-        const onClose = this._onClose(session, uninitializeWs).bind(this);
-        const onMessage = this._onMessage(session, platformEventContext).bind(this);
+        if (this._platform._registrationMode === "attached") {
+            const onClose = this._onClose(pluvWs).bind(this);
+            const onMessage = this._onMessage(pluvWs).bind(this);
 
-        pluvWs.addEventListener("close", onClose);
-        pluvWs.addEventListener("error", onClose);
-        pluvWs.addEventListener("message", onMessage);
+            pluvWs.addEventListener("close", onClose);
+            pluvWs.addEventListener("error", onClose);
+            pluvWs.addEventListener("message", onMessage);
+        }
 
-        this._emitRegistered(session);
+        await this._emitRegistered(pluvWs);
 
-        this._logDebug(`${colors.blue(`Registered connection for room ${this.id}:`)} ${sessionId}`);
+        this._logDebug(`${colors.blue(`Registered connection for room ${this.id}:`)} ${pluvWs.sessionId}`);
 
         const size = this.getSize();
 
         this._logDebug(`${colors.blue(`Room ${this.id} size:`)} ${size}`);
     }
 
-    private _broadcast(params: BroadcastParams<this>): void {
+    private async _broadcast(params: BroadcastParams<this>): Promise<void> {
         const { message, senderId } = params;
 
-        const currentTime = new Date().getTime();
-        const sender = senderId ? this._sessions.get(senderId) : null;
-
-        const quitters = Array.from(this._sessions.values()).filter(
-            (session) => session.quit || currentTime - session.timers.ping > PING_TIMEOUT_MS,
-        );
-
-        quitters.forEach((session) => {
-            this._sessions.delete(session.id);
-        });
-
-        this._emitQuitters(quitters);
+        const sender = senderId ? (this._sessions.get(senderId) ?? null) : null;
+        const session = (await sender?.getSession()) ?? null;
+        const user = session?.user ?? null;
 
         this._platform.pubSub.publish(this.id, {
             connectionId: senderId ?? null,
             room: this.id,
-            user: sender?.user ?? null,
+            user,
             ...message,
         });
     }
 
-    private _emitQuitters(quitters: readonly WebSocketSession<TAuthorize>[]): void {
-        quitters.forEach((quitter) => {
-            this._broadcast({
+    private async _emitQuitters(): Promise<void> {
+        const currentTime = new Date().getTime();
+        const quitters = Array.from(this._sessions.values()).filter(
+            (pluvWs) => pluvWs.state.quit || currentTime - pluvWs.state.timers.ping > PING_TIMEOUT_MS,
+        );
+
+        quitters.forEach((pluvWs) => {
+            this._sessions.delete(pluvWs.sessionId);
+        });
+
+        const promises = quitters.map(async (quitter) => {
+            await this._broadcast({
                 message: {
                     type: "$EXIT",
-                    data: { sessionId: quitter.id },
+                    data: { sessionId: quitter.sessionId },
                 },
-                senderId: quitter.id,
+                senderId: quitter.sessionId,
             });
         });
+
+        await Promise.all(promises);
     }
 
-    private _emitRegistered(session: WebSocketSession<TAuthorize>): void {
-        this._sendSelfMessage(
-            {
-                type: "$REGISTERED",
-                data: { sessionId: session.id },
-            },
-            session,
-        );
+    private async _emitRegistered(pluvWs: AbstractWebSocket): Promise<void> {
+        const session = await pluvWs.getSession();
+        const sessionId = session.id;
+        const user = session.user;
+
+        await this._sendSelfMessage({ type: "$REGISTERED", data: { sessionId } }, { sessionId, user });
     }
 
-    private _getIOAuthorize(context: InferPlatformRoomContextType<TPlatform>) {
-        if (typeof this._authorize === "function") {
-            return this._authorize(context);
+    private _ensureDetached(): void {
+        if (this._platform._registrationMode === "detached") return;
+
+        throw new Error("Platform must use detached mode");
+    }
+
+    private _getAbstractWs(webSocket: WebSocketType<TPlatform>): AbstractWebSocket | null {
+        if ((webSocket as unknown as any) instanceof AbstractWebSocket) return webSocket;
+
+        const sessionId = this._platform.getSessionId(webSocket);
+
+        if (typeof sessionId === "string") {
+            const session = this._sessions.get(sessionId) ?? null;
+
+            if (session) return session;
         }
 
-        return this._authorize as {
-            required: any;
-            secret: string;
-            user: InputZodLike<any>;
-        } | null;
+        const sessions = Array.from(this._sessions.values());
+
+        return sessions.find((pluvWs) => pluvWs.webSocket === webSocket) ?? null;
     }
 
-    private async _getAuthorizedUser(
-        token: Maybe<string>,
-        context: InferPlatformRoomContextType<TPlatform>,
-    ): Promise<InferIOAuthorizeUser<InferIOAuthorize<this>>> {
-        const ioAuthorize = this._getIOAuthorize(context);
+    private async _getSession(webSocket: WebSocketType<TPlatform>): Promise<WebSocketSession<TAuthorize>> {
+        const pluvWs = this._getAbstractWs(webSocket);
+
+        if (!pluvWs) throw new Error("Session could not be found");
+
+        return await pluvWs.getSession<TAuthorize>();
+    }
+
+    private async _getAuthorizedUser(token: Maybe<string>): Promise<InferIOAuthorizeUser<InferIOAuthorize<this>>> {
+        const ioAuthorize = this._getIOAuthorize();
 
         if (!ioAuthorize) return null as InferIOAuthorizeUser<InferIOAuthorize<this>>;
         if (!token) return null as InferIOAuthorizeUser<InferIOAuthorize<this>>;
@@ -318,6 +366,18 @@ export class IORoom<
         }
     }
 
+    private _getIOAuthorize() {
+        if (typeof this._authorize === "function") {
+            return this._authorize(this._context);
+        }
+
+        return this._authorize as {
+            required: any;
+            secret: string;
+            user: InputZodLike<any>;
+        } | null;
+    }
+
     private _getProcedure(
         message: EventMessage<string, any>,
     ): (typeof this._router)["_events"][keyof (typeof this._router)["_events"]] | null {
@@ -332,6 +392,12 @@ export class IORoom<
         return procedure.config.input ? procedure.config.input.parse(message.data) : message.data;
     }
 
+    private async _getSessions() {
+        const promises = Array.from(this._sessions.values()).map(async (pluvWs) => await pluvWs.getSession());
+
+        return await Promise.all(promises);
+    }
+
     private async _initialize(): Promise<void> {
         this._logDebug(`${colors.blue(`Initializing room ${this.id}:`)}`);
 
@@ -341,14 +407,15 @@ export class IORoom<
             this._uninitialize = null;
         }
 
-        const pubSubId = await this._platform.pubSub.subscribe(this.id, ({ options = {}, ...message }) => {
-            const session = {
-                id: message.connectionId,
-                user: message.user,
-            };
+        const pubSubId = await this._platform.pubSub.subscribe(
+            this.id,
+            async ({ options = {}, ...message }): Promise<void> => {
+                const sessionId = message.connectionId;
+                const user = message.user;
 
-            this._sendMessage(message, session, options);
-        });
+                await this._sendMessage(message, { sessionId, user }, options);
+            },
+        );
 
         this._uninitialize = async () => {
             this._platform.pubSub.unsubscribe(pubSubId);
@@ -372,29 +439,27 @@ export class IORoom<
         this._debug && console.log(...data);
     }
 
-    private _onClose(session: WebSocketSession<TAuthorize>, callback?: () => void): () => void {
+    private _onClose(webSocket: AbstractWebSocket): () => void {
         return (): void => {
             if (!this._uninitialize) return;
 
-            session.quit = true;
+            webSocket.state.quit = true;
 
-            this._logDebug(`${colors.blue(`(Unregistering connection for room ${this.id}:`)} ${session.id}`);
-            this._sessions.delete(session.id);
+            this._logDebug(`${colors.blue(`Unregistering connection for room ${this.id}:`)} ${webSocket.sessionId}`);
+            this._sessions.delete(webSocket.sessionId);
 
-            this._platform.persistance.deleteUser(this.id, session.id).finally(() => {
-                this._broadcast({
+            this._platform.persistance.deleteUser(this.id, webSocket.sessionId).finally(async () => {
+                await this._broadcast({
                     message: {
                         type: "$EXIT",
-                        data: { sessionId: session.id },
+                        data: { sessionId: webSocket.sessionId },
                     },
-                    senderId: session.id,
+                    senderId: webSocket.sessionId,
                 });
 
-                callback?.();
-
                 const size = this.getSize();
 
-                this._logDebug(`${colors.blue(`Unregistered connection for room ${this.id}:`)} ${session.id}`);
+                this._logDebug(`${colors.blue(`Unregistered connection for room ${this.id}:`)} ${webSocket.sessionId}`);
                 this._logDebug(`${colors.blue(`Room ${this.id} size:`)} ${size}`);
 
                 if (size) return;
@@ -404,23 +469,29 @@ export class IORoom<
         };
     }
 
-    private _onMessage(
-        session: WebSocketSession<TAuthorize>,
-        platformEventContext: InferPlatformEventContextType<TPlatform>,
-    ): (event: AbstractMessageEvent) => void {
-        return (event: AbstractMessageEvent): void => {
+    private _onMessage(webSocket: AbstractWebSocket): (event: AbstractMessageEvent) => void {
+        return async (event: AbstractMessageEvent): Promise<void> => {
+            const pluvWs = this._getAbstractWs(webSocket as WebSocketType<TPlatform>);
+
+            if (!pluvWs) throw new Error("Could not get session");
+
+            const [session, sessions] = await Promise.all([
+                this._getSession(pluvWs as WebSocketType<TPlatform>),
+                this._getSessions(),
+            ]);
+
             const baseContext: EventResolverContext<TPlatform, TAuthorize, TContext> = {
                 context: this._context,
                 doc: this._doc,
                 room: this.id,
                 session,
-                sessions: this._sessions,
+                sessions,
             };
 
             if (!this._uninitialize) return;
 
-            if (session.quit) {
-                session.webSocket.close(1011, "WebSocket broken.");
+            if (pluvWs.state.quit) {
+                pluvWs.close(1011, "WebSocket broken.");
 
                 return;
             }
@@ -438,9 +509,10 @@ export class IORoom<
             try {
                 inputs = this._getProcedureInputs(message);
             } catch (error) {
-                session?.webSocket.handleError({
+                pluvWs.handleError({
                     error,
                     message: "Invalid input",
+                    room: this.id,
                     session,
                 });
 
@@ -450,30 +522,32 @@ export class IORoom<
             const extendedContext: EventResolverContext<
                 TPlatform,
                 TAuthorize,
-                TContext & InferPlatformRoomContextType<TPlatform> & InferPlatformEventContextType<TPlatform>
-            > = {
-                ...baseContext,
-                context: { ...this._context, ...platformEventContext },
-            };
+                TContext & InferRoomContextType<TPlatform>
+            > = { ...baseContext, context: { ...this._context } };
 
             Promise.all([
                 procedure.config.broadcast?.(inputs, extendedContext),
                 procedure.config.self?.(inputs, extendedContext),
                 procedure.config.sync?.(inputs, baseContext),
-            ]).then(([broadcast, self, sync]) => {
+            ]).then(async ([broadcast, self, sync]) => {
+                const sessionId = session.id;
+                const user = session.user;
+
                 if (broadcast) {
                     Object.entries(broadcast).forEach(([type, data]: any) => {
                         this._broadcast({
                             message: { data, type },
-                            senderId: session.id,
+                            senderId: sessionId,
                         });
                     });
                 }
 
                 if (self) {
-                    Object.entries(self).forEach(([type, data]) => {
-                        this._sendSelfMessage({ data, type }, session);
-                    });
+                    await Promise.all(
+                        Object.entries(self).map(async ([type, data]) => {
+                            await this._sendSelfMessage({ data, type }, { sessionId, user });
+                        }),
+                    );
                 }
 
                 if (sync) {
@@ -481,13 +555,15 @@ export class IORoom<
                         connectionId: session.id,
                         options: { type: "sync" },
                         room: this.id,
-                        user: session.user,
+                        user,
                         ...message,
                     });
 
-                    Object.entries(sync).forEach(([type, data]) => {
-                        this._sendSelfMessage({ data, type }, session);
-                    });
+                    await Promise.all(
+                        Object.entries(sync).map(async ([type, data]) => {
+                            await this._sendSelfMessage({ data, type }, { sessionId, user });
+                        }),
+                    );
                 }
             });
         };
@@ -507,19 +583,19 @@ export class IORoom<
         }
     }
 
-    private _sendMessage(
+    private async _sendMessage(
         message: EventMessage<string, any>,
         sender: SendMessageSender | null,
         options: SendMessageOptions = {},
-    ): void {
+    ): Promise<void> {
         switch (options.type) {
             case "self": {
-                this._sendSelfMessage(message, sender);
+                await this._sendSelfMessage(message, sender);
 
                 return;
             }
             case "sync": {
-                this._sendSyncMessage(message, sender);
+                await this._sendSyncMessage(message, sender);
 
                 return;
             }
@@ -527,63 +603,73 @@ export class IORoom<
             default: {
                 const sessionIds = options.sessionIds;
 
-                this._sendBroadcastMessage(message, sender, sessionIds);
+                await this._sendBroadcastMessage(message, sender, sessionIds);
             }
         }
     }
 
-    private _sendBroadcastMessage(
+    private async _sendBroadcastMessage(
         message: EventMessage<string, any>,
         sender: SendMessageSender | null,
         sessionIds?: readonly string[],
-    ): void {
-        const senderId = sender?.id ?? null;
+    ): Promise<void> {
+        const { data, type } = message;
+        const connectionId = sender?.sessionId ?? null;
+        const room = this.id;
 
-        const sessions =
+        const webSockets =
             sessionIds?.reduce((dict, id) => {
-                const _session = this._sessions.get(id);
+                const pluvWs = this._sessions.get(id);
 
-                return _session ? dict.set(id, _session) : dict;
-            }, new Map<string, WebSocketSession<TAuthorize>>()) ?? this._sessions;
+                return pluvWs ? dict.set(id, pluvWs) : dict;
+            }, new Map<string, AbstractWebSocket>()) ?? this._sessions;
 
-        Array.from(sessions.values()).forEach((_session) => {
-            _session.webSocket.sendMessage({
-                connectionId: senderId,
-                room: this.id,
-                user: _session.user,
-                ...message,
-            });
+        const promises = Array.from(webSockets.values()).map(async (pluvWs) => {
+            const session = await pluvWs.getSession();
+            const user = session.user ?? null;
+
+            pluvWs.sendMessage({ connectionId, data, room, type, user });
         });
+
+        await Promise.all(promises);
     }
 
-    private _sendSelfMessage(message: EventMessage<string, any>, sender: SendMessageSender | null): void {
-        const senderId = sender?.id;
+    private async _sendSelfMessage(
+        message: EventMessage<string, any>,
+        sender: SendMessageSender | null,
+    ): Promise<void> {
+        const senderId = sender?.sessionId ?? null;
+
+        if (typeof senderId !== "string") return;
 
-        if (!senderId) return;
+        const pluvWs = this._sessions.get(senderId);
 
-        const session = this._sessions.get(senderId);
+        if (!pluvWs) return;
+        if (message.type === "$PONG") await this._emitQuitters();
 
-        if (!session) return;
+        const session = await pluvWs.getSession();
+        const user = session.user ?? null;
 
-        session.webSocket.sendMessage({
-            connectionId: sender.id,
+        pluvWs.sendMessage({
+            connectionId: senderId,
             room: this.id,
-            user: sender.user,
+            user,
             ...message,
         });
     }
 
-    private _sendSyncMessage(message: EventMessage<string, any>, sender: SendMessageSender | null): void {
-        const senderId = sender?.id;
-
-        if (!senderId) return;
+    private async _sendSyncMessage(
+        message: EventMessage<string, any>,
+        sender: SendMessageSender | null,
+    ): Promise<void> {
+        if (!sender) return;
 
         const context: EventResolverContext<TPlatform, TAuthorize, TContext> = {
             context: this._context,
             doc: this._doc,
             room: this.id,
             session: null,
-            sessions: this._sessions,
+            sessions: await this._getSessions(),
         };
 
         const resolver = this._getProcedure(message)?.config.sync;
@@ -605,7 +691,7 @@ export class IORoom<
                 const data = output[type] ?? {};
 
                 this._platform.pubSub.publish(this.id, {
-                    connectionId: senderId,
+                    connectionId: sender.sessionId,
                     data,
                     options: { type: "self" },
                     room: this.id,