Add websocket handler

Signed-off-by: Marcos Candeia <[email protected]>

examples/cf/counter-actor/package.json

@@ -15,7 +15,7 @@
     "wrangler": "^3.91.0"
   },
   "dependencies": {
-    "@deco/actors": "npm:@jsr/deco__actors@^0.14.0-beta.7",
+    "@deco/actors": "npm:@jsr/deco__actors@^0.14.0-beta.8",
     "hono": "^4.6.12"
   }
 }

examples/cf/counter-actor/src/counter.ts

@@ -1,5 +1,5 @@
 import { ActorState } from "@deco/actors";
-import { WatchTarget } from "@deco/actors/watch";
+import { ChannelUpgrader, WatchTarget } from "@deco/actors/watch";
 
 export class Counter {
   private count: number;
@@ -34,4 +34,14 @@ export class Counter {
   watch(): AsyncIterableIterator<number> {
     return this.watchTarget.subscribe();
   }
+  chan(name: string): ChannelUpgrader<string, string> {
+    return (async ({ send, recv }) => {
+      await send(`Hello ${name}`);
+      for await (const str of recv()) {
+        if (str === "PING") {
+          await send("PONG");
+        }
+      }
+    });
+  }
 }

examples/cf/counter-actor/src/index.ts

@@ -6,7 +6,11 @@ export { ActorDurableObject } from "@deco/actors/cf";
 const app = new Hono<{ Bindings: Env }>();
 
 const runtime = new ActorCfRuntime([Counter]);
-app.use(withActors(runtime));
+const mid = withActors(runtime)
+app.use(async (ctx, next) => {
+  await mid(ctx, next);
+  console.log(await ctx.res.text())
+});
 
 app.get("/", (c) => c.text("Hello Cloudflare Workers!"));
 

src/actors/runtimes/cf/fetcher.ts

@@ -10,7 +10,18 @@ export interface Env {
 
 export class ActorCfRuntime implements ActorFetcher<Env> {
   constructor(protected actorsConstructors: Array<ActorConstructor>) {
-    registerActors(actorsConstructors);
+    registerActors(actorsConstructors, () => {
+      const webSocketPair = new WebSocketPair();
+      const [client, server] = Object.values(webSocketPair);
+      return {
+        socket: server,
+        response: new Response(null, {
+          status: 101,
+          // @ts-ignore: webSocket is not part of the Response type
+          webSocket: client,
+        }),
+      };
+    });
   }
   fetch(request: Request, env?: Env | undefined): Promise<Response> | Response {
     if (!env) {