(draft, incomplete) WebSocket based-proxy service on Workers

wow.worker.js

const WOW_TOKEN = "W0w!"

async function handleRequest(request) {
  const upgradeHeader = request.headers.get('Upgrade');
  const token = request.headers.get('X-WOW-TOKEN');
  const addr = request.headers.get('X-WOW-ADDR');
  if (token !== WOW_TOKEN) {
    return new Response('Forbidden', { status: 403 });
  }
  if (!upgradeHeader || upgradeHeader !== 'websocket') {
    return new Response('Expected Upgrade: websocket', { status: 426 });
  }

  let remote;
  try {
    remote = connect(addr);

    const writer = remote.writable.getWriter()
    await writer.write(encoded);

  } catch (error) {
    return new Response("Socket connection failed: " + error, { status: 500 });
  }

  const webSocketPair = new WebSocketPair();
  const [client, server] = Object.values(webSocketPair);

  async function onWebSocketEvent(event) {
    switch (event.type) {
      case 'message':
        console.log(event.data);
        // server.send(event.data);
        const writer = remote.writable.getWriter()
        await writer.write(event.data);
        break;
      default:
        console.error(event);
    }
  }

  server.accept();
  ['message', 'error', 'close'].forEach((event) => server.addEventListener(event, onWebSocketEvent))

  async function processRemoteSocket() {
    const reader = remote.getReader()
    const data = await reader.read()
    await server.send(data)
  }
  processRemoteSocket()

  return new Response(null, {
    status: 101,
    webSocket: client,
  });
}