@pluv/platform-cloudflare 0.19.0 → 0.21.0

package/src/createPluvHandler.ts

@@ -52,19 +52,37 @@ export const createPluvHandler = <TPluvServer extends PluvServer<CloudflarePlatf
     const { authorize, binding, endpoint = "/api/pluv", modify, io } = config;
 
     const DurableObject = class implements DurableObject {
-        private _env: Id<InferCloudflarePluvHandlerEnv<TPluvServer>>;
-        private _io: IORoom<CloudflarePlatform>;
+        private _room: IORoom<CloudflarePlatform>;
 
         constructor(state: DurableObjectState, env: Id<InferCloudflarePluvHandlerEnv<TPluvServer>>) {
-            this._env = env;
-            this._io = io.getRoom(state.id.toString(), { env });
+            this._room = io.getRoom(state.id.toString(), { env, state });
         }
 
-        webSocketClose(ws: WebSocket, code: number, reason: string, wasClean: boolean): void | Promise<void> {}
+        public webSocketClose(ws: WebSocket, code: number, reason: string, wasClean: boolean): void | Promise<void> {
+            if (io._registrationMode !== "detached") return;
 
-        webSocketError(ws: WebSocket, error: unknown): void | Promise<void> {}
+            const handler = this._room.onClose(ws);
 
-        webSocketMessage(ws: WebSocket, message: string | ArrayBuffer): void | Promise<void> {}
+            handler({ code, reason });
+        }
+
+        public webSocketError(ws: WebSocket, error: unknown): void | Promise<void> {
+            if (io._registrationMode !== "detached") return;
+
+            const handler = this._room.onError(ws);
+
+            const eventError = error instanceof Error ? error : new Error("Internal Error");
+
+            handler({ error: eventError, message: eventError.message });
+        }
+
+        public webSocketMessage(ws: WebSocket, message: string | ArrayBuffer): void | Promise<void> {
+            if (io._registrationMode !== "detached") return;
+
+            const handler = this._room.onMessage(ws);
+
+            handler({ data: message });
+        }
 
         async fetch(request: Request<any, CfProperties<any>>): Promise<Response> {
             const isWSRequest = request.headers.get("Upgrade") === "websocket";
@@ -77,11 +95,7 @@ export const createPluvHandler = <TPluvServer extends PluvServer<CloudflarePlatf
 
             const token = new URL(request.url).searchParams.get("token");
 
-            await this._io.register(server, {
-                env: this._env,
-                request,
-                token,
-            });
+            await this._room.register(server, { token });
 
             return new Response(null, { status: 101, webSocket: client });
         }

package/src/index.ts

@@ -1,3 +1,5 @@
+export type { CloudflarePlatformConfig } from "./CloudflarePlatform";
+export { createPluvHandler } from "./createPluvHandler";
 export type {
     AuthorizeFunction,
     AuthorizeFunctionContext,
@@ -5,5 +7,4 @@ export type {
     CreatePluvHandlerResult,
     PluvHandlerFetch,
 } from "./createPluvHandler";
-export { createPluvHandler } from "./createPluvHandler";
 export { platformCloudflare } from "./platformCloudflare";

package/src/platformCloudflare.ts

@@ -1,5 +1,6 @@
+import type { CloudflarePlatformConfig } from "./CloudflarePlatform";
 import { CloudflarePlatform } from "./CloudflarePlatform";
 
-export const platformCloudflare = <TEnv extends Record<string, any> = {}>(): CloudflarePlatform<TEnv> => {
-    return new CloudflarePlatform<TEnv>();
+export const platformCloudflare = <TEnv extends Record<string, any> = {}>(config: CloudflarePlatformConfig<TEnv> = {}): CloudflarePlatform<TEnv> => {
+    return new CloudflarePlatform<TEnv>(config);
 };

package/src/utils/index.ts

@@ -0,0 +1 @@
+export { partitionByLength } from "./partitionByLength";

package/src/utils/partitionByLength.ts

@@ -0,0 +1,8 @@
+export const partitionByLength = <T>(arr: readonly T[] | T[], length: number): readonly (readonly T[])[] => {
+    if (!arr.length) return [];
+
+    const head = arr.slice(0, length);
+    const tail = arr.slice(length);
+
+    return [head, ...partitionByLength(tail, length)];
+};