@edge-base/server 0.2.4 → 0.2.6

package/src/durable-objects/rooms-do.ts

@@ -1,12 +1,11 @@
 import {
-  getRoomHooks,
   type AuthContext as SharedAuthContext,
   type RoomMemberInfo,
+  type RoomNamespaceConfig,
   type RoomSender,
   type RoomServerAPI,
 } from '@edge-base/shared';
 import {
-  createCloudflareRealtimeClient,
   type CloudflareRealtimeCloseTracksRequest,
   type CloudflareRealtimeNewSessionRequest,
   type CloudflareRealtimeNewSessionResponse,
@@ -14,7 +13,6 @@ import {
   type CloudflareRealtimeTracksRequest,
   type CloudflareRealtimeTracksResponse,
 } from '../lib/cloudflare-realtime.js';
-import { resolveAuthContextFromToken } from '../middleware/auth.js';
 import type { Env } from '../types.js';
 import { RoomRuntimeBaseDO, type RoomWSMeta } from './room-runtime-base.js';
 
@@ -96,6 +94,17 @@ interface RoomMemberPresence {
   state: Record<string, unknown>;
 }
 
+interface RoomSummaryResponse {
+  namespace: string;
+  roomId: string;
+  metadata: Record<string, unknown>;
+  occupancy: {
+    activeMembers: number;
+    activeConnections: number;
+  };
+  updatedAt: string;
+}
+
 interface RoomMemberRealtimeSession {
   sessionId: string;
   connectionId?: string;
@@ -117,6 +126,10 @@ const MEDIA_DENIED = Symbol('rooms.media.denied');
 const WEBSOCKET_OPEN = 1;
 const CLOUDFLARE_REALTIME_KIT_MEETING_STORAGE_KEY = 'cloudflareRealtimeKitMeetingId';
 
+function getRoomHooks(namespaceConfig?: RoomNamespaceConfig | null) {
+  return namespaceConfig?.hooks;
+}
+
 function computeStateDelta(
   previous: Record<string, unknown>,
   next: Record<string, unknown>,
@@ -153,6 +166,10 @@ export class RoomsDO extends RoomRuntimeBaseDO {
   override async fetch(request: Request): Promise<Response> {
     const url = new URL(request.url);
 
+    if (url.pathname === '/summary' && request.method === 'GET') {
+      return this.handleSummaryGet(url);
+    }
+
     if (url.pathname === '/media/cloudflare_realtimekit/session' && request.method === 'POST') {
       return this.handleCloudflareRealtimeKitSessionCreate(request, url);
     }
@@ -182,6 +199,22 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     return super.fetch(request);
   }
 
+  private async handleSummaryGet(url: URL): Promise<Response> {
+    this.hydrateRoomIdentityFromUrl(url);
+    const metadata = await this.getRoomMetadataSnapshot();
+
+    return this.jsonResponse<RoomSummaryResponse>(200, {
+      namespace: this.namespace ?? '',
+      roomId: this.roomId ?? '',
+      metadata,
+      occupancy: {
+        activeMembers: this.members.size,
+        activeConnections: this.joinedConnectionIds.size,
+      },
+      updatedAt: new Date().toISOString(),
+    });
+  }
+
   private async handleCloudflareRealtimeKitSessionCreate(request: Request, url: URL): Promise<Response> {
     try {
       const body = await this.readJsonBody<{
@@ -250,7 +283,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
         });
       }
 
-      const client = this.buildRealtimeClient();
+      const client = await this.buildRealtimeClient();
       const response = await client.createSession(
         {
           sessionDescription: body.sessionDescription,
@@ -311,7 +344,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     try {
       const body = await this.readJsonBody<{ ttl?: number }>(request);
       await this.authenticateRealtimeRequest(request, url);
-      const client = this.buildRealtimeClient();
+      const client = await this.buildRealtimeClient();
       const ttl = typeof body.ttl === 'number' && Number.isFinite(body.ttl) && body.ttl > 0
         ? Math.floor(body.ttl)
         : 3600;
@@ -354,7 +387,8 @@ export class RoomsDO extends RoomRuntimeBaseDO {
         throw new Error('tracks is required');
       }
 
-      const response = await this.buildRealtimeClient().addTracks(sessionId, {
+      const client = await this.buildRealtimeClient();
+      const response = await client.addTracks(sessionId, {
         sessionDescription: body.sessionDescription,
         tracks: body.tracks,
         autoDiscover: body.autoDiscover === true,
@@ -410,7 +444,8 @@ export class RoomsDO extends RoomRuntimeBaseDO {
         throw new Error('sessionDescription is required');
       }
 
-      const response = await this.buildRealtimeClient().renegotiate(sessionId, {
+      const client = await this.buildRealtimeClient();
+      const response = await client.renegotiate(sessionId, {
         sessionDescription: body.sessionDescription,
       });
       this.assertRealtimeTracksResponseSuccess(response);
@@ -447,7 +482,8 @@ export class RoomsDO extends RoomRuntimeBaseDO {
         throw new Error('tracks is required');
       }
 
-      const response = await this.buildRealtimeClient().closeTracks(sessionId, {
+      const client = await this.buildRealtimeClient();
+      const response = await client.closeTracks(sessionId, {
         sessionDescription: body.sessionDescription,
         tracks: body.tracks,
         force: body.force === true,
@@ -472,7 +508,8 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     }
   }
 
-  private buildRealtimeClient() {
+  private async buildRealtimeClient() {
+    const { createCloudflareRealtimeClient } = await import('../lib/cloudflare-realtime.js');
     return createCloudflareRealtimeClient(this.env as unknown as Env);
   }
 
@@ -633,9 +670,10 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     const token = this.extractBearerToken(request);
     if (!token) {
-      throw new Error('Authentication required');
+      throw new Error('Authentication required before the room connection could be established.');
     }
 
+    const { resolveAuthContextFromToken } = await import('../middleware/auth.js');
     const auth = await resolveAuthContextFromToken(this.env, token, request);
     const memberId = auth.id;
     const member = this.members.get(memberId);
@@ -758,7 +796,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
       const event = typeof msg.event === 'string' ? msg.event : '';
       const requestId = typeof msg.requestId === 'string' ? msg.requestId : undefined;
-      if (!this.checkRateLimit(meta.connectionId)) {
+      if (!this.checkRateLimit(meta.connectionId, 'signals')) {
         this.safeSend(ws, {
           type: 'signal_error',
           event,
@@ -806,7 +844,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
       const operation = typeof msg.operation === 'string' ? msg.operation : '';
       const requestId = typeof msg.requestId === 'string' ? msg.requestId : undefined;
-      if (!this.checkRateLimit(meta.connectionId)) {
+      if (!this.checkRateLimit(meta.connectionId, 'admin')) {
         this.safeSend(ws, {
           type: 'admin_error',
           operation,
@@ -833,7 +871,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
       const operation = this.normalizeMediaOperation(msg.operation);
       const kind = this.normalizeMediaKind(msg.kind);
       const requestId = typeof msg.requestId === 'string' ? msg.requestId : undefined;
-      if (!this.checkRateLimit(meta.connectionId)) {
+      if (!this.checkRateLimit(meta.connectionId, 'media')) {
         this.safeSend(ws, {
           type: 'media_error',
           operation: operation ?? '',
@@ -1450,7 +1488,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     const beforeSend = getRoomHooks(this.namespaceConfig ?? undefined)?.signals?.beforeSend;
     if (!beforeSend) return signal.payload;
 
-    const ctx = this.buildHandlerContext();
+    const ctx = await this.buildHandlerContext();
     const result = await Promise.resolve(
       beforeSend(signal.event, signal.payload, signal.sender, signal.roomApi, ctx),
     );
@@ -1468,7 +1506,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     if (!onSend) return;
 
     try {
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onSend(event, payload, sender, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] signal.onSend error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1481,7 +1519,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     try {
       const roomApi = this.buildRoomServerAPI();
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onJoin(member, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] members.onJoin error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1497,7 +1535,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     try {
       const roomApi = this.buildRoomServerAPI();
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onLeave(member, roomApi, ctx, reason));
     } catch (err) {
       console.error(`[Rooms] members.onLeave error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1513,7 +1551,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     try {
       const roomApi = this.buildRoomServerAPI();
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onStateChange(member, state, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] members.onStateChange error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1528,7 +1566,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     if (!onStateChange) return;
 
     try {
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onStateChange(delta, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] state.onStateChange error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1541,7 +1579,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     try {
       const roomApi = this.buildRoomServerAPI();
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onReconnect(sender, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] session.onReconnect error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1554,7 +1592,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
 
     try {
       const roomApi = this.buildRoomServerAPI();
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onDisconnectTimeout(sender, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] session.onDisconnectTimeout error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1919,7 +1957,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
       return undefined;
     }
 
-    const ctx = this.buildHandlerContext();
+    const ctx = await this.buildHandlerContext();
     const result = await Promise.resolve(beforePublish(kind, sender, roomApi, ctx));
     if (result === false) {
       return MEDIA_DENIED;
@@ -1936,7 +1974,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     if (!onPublished) return;
 
     try {
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onPublished(kind, sender, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] media.onPublished error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1952,7 +1990,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     if (!onUnpublished) return;
 
     try {
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onUnpublished(kind, sender, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] media.onUnpublished error: ${err instanceof Error ? err.message : String(err)}`);
@@ -1969,7 +2007,7 @@ export class RoomsDO extends RoomRuntimeBaseDO {
     if (!onMuteChange) return;
 
     try {
-      const ctx = this.buildHandlerContext();
+      const ctx = await this.buildHandlerContext();
       await Promise.resolve(onMuteChange(kind, sender, muted, roomApi, ctx));
     } catch (err) {
       console.error(`[Rooms] media.onMuteChange error: ${err instanceof Error ? err.message : String(err)}`);