crossws/playground/_common.ts

import type { WebSocketHooks, WebSocketAdapter } from "../src";

export const getIndexHTML = (params) =>
  import("../examples/h3/index.html.ts").then((r) => r.html(params));

export function createDemo<T extends WebSocketAdapter>(
  adapter: T,
  opts?: Parameters<T>[1],
): ReturnType<T> {
  const hooks = createWebSocketDebugHooks({
    open(peer) {
      peer.send("Hello!");
    },
    message(peer, message) {
      if (message.text() === "ping") {
        peer.send("pong");
      }
    },
  });

  return adapter(hooks, opts);
}

function createWebSocketDebugHooks(
  hooks: Partial<WebSocketHooks>,
): Partial<WebSocketHooks> {
  const createDebugHook =
    (name: keyof WebSocketHooks) =>
    (peer, ...args: any[]) => {
      console.log(
        `[ws] [${name}]`,
        peer,
        [...args].map((arg, i) => `\n - arg#${i} ${arg}`).join(""),
      );
      hooks[name]?.(peer, ...args);
    };

  return new Proxy(
    {},
    {
      get(_, prop) {
        return createDebugHook(prop as keyof WebSocketHooks);
      },
    },
  );
}