@vgai/p2p-colyseus 0.1.0

package/src/webrtc.ts

@@ -0,0 +1,78 @@
+import type { PacketChannel } from './channels';
+import type { Envelope } from './protocol';
+
+interface DataChannelLike {
+  readonly readyState: string;
+  readonly bufferedAmount?: number;
+  send(data: string): void;
+  close(): void;
+  addEventListener(type: 'message', handler: (event: { data: unknown }) => void): void;
+  addEventListener(type: 'close', handler: () => void): void;
+}
+
+type EnvelopeHandler = (envelope: Envelope) => void;
+type CloseHandler = (reason?: string) => void;
+
+export class WebRTCDataChannelPacketChannel implements PacketChannel {
+  readonly mode = 'webrtc' as const;
+  private readonly envelopeHandlers = new Set<EnvelopeHandler>();
+  private readonly closeHandlers = new Set<CloseHandler>();
+
+  constructor(
+    readonly peerId: string,
+    private readonly dataChannel: DataChannelLike,
+  ) {
+    dataChannel.addEventListener('message', (event) => {
+      if (typeof event.data !== 'string') return;
+      const envelope = JSON.parse(event.data) as Envelope;
+      for (const handler of this.envelopeHandlers) handler(envelope);
+    });
+    dataChannel.addEventListener('close', () => {
+      for (const handler of this.closeHandlers) handler('closed');
+    });
+  }
+
+  send(envelope: Envelope): void {
+    if (this.dataChannel.readyState !== 'open') {
+      throw new Error(`WebRTC data channel for ${this.peerId} is not open`);
+    }
+    this.dataChannel.send(JSON.stringify(envelope));
+  }
+
+  onEnvelope(handler: EnvelopeHandler): () => void {
+    this.envelopeHandlers.add(handler);
+    return () => this.envelopeHandlers.delete(handler);
+  }
+
+  onClose(handler: CloseHandler): () => void {
+    this.closeHandlers.add(handler);
+    return () => this.closeHandlers.delete(handler);
+  }
+
+  close(): void {
+    this.dataChannel.close();
+  }
+
+  closeAfterFlush(delayMs = 100): void {
+    const started = Date.now();
+    const minDelayMs = delayMs;
+    const maxDelayMs = Math.max(1_000, delayMs);
+    const tick = () => {
+      const elapsed = Date.now() - started;
+      const flushed = (this.dataChannel.bufferedAmount ?? 0) === 0;
+      if ((flushed && elapsed >= minDelayMs) || elapsed >= maxDelayMs) {
+        this.close();
+        return;
+      }
+      setTimeout(tick, 10);
+    };
+    tick();
+  }
+}
+
+export function createWebRTCDataChannelPacketChannel(
+  peerId: string,
+  dataChannel: RTCDataChannel,
+): PacketChannel {
+  return new WebRTCDataChannelPacketChannel(peerId, dataChannel);
+}

package/src/websocket.ts

@@ -0,0 +1,59 @@
+import type { PacketChannel } from './channels';
+import type { Envelope } from './protocol';
+
+interface WebSocketLike {
+  readonly readyState: number;
+  send(data: string): void;
+  close(code?: number, reason?: string): void;
+  addEventListener(type: 'message', handler: (event: { data: unknown }) => void): void;
+  addEventListener(type: 'close', handler: (event?: { reason?: string }) => void): void;
+}
+
+type EnvelopeHandler = (envelope: Envelope) => void;
+type CloseHandler = (reason?: string) => void;
+
+const OPEN = 1;
+
+export class WebSocketPacketChannel implements PacketChannel {
+  readonly mode = 'websocket' as const;
+  private readonly envelopeHandlers = new Set<EnvelopeHandler>();
+  private readonly closeHandlers = new Set<CloseHandler>();
+
+  constructor(
+    readonly peerId: string,
+    private readonly socket: WebSocketLike,
+  ) {
+    socket.addEventListener('message', (event) => {
+      if (typeof event.data !== 'string') return;
+      const envelope = JSON.parse(event.data) as Envelope;
+      for (const handler of this.envelopeHandlers) handler(envelope);
+    });
+    socket.addEventListener('close', (event) => {
+      for (const handler of this.closeHandlers) handler(event?.reason);
+    });
+  }
+
+  send(envelope: Envelope): void {
+    if (this.socket.readyState !== OPEN)
+      throw new Error(`WebSocket for ${this.peerId} is not open`);
+    this.socket.send(JSON.stringify(envelope));
+  }
+
+  onEnvelope(handler: EnvelopeHandler): () => void {
+    this.envelopeHandlers.add(handler);
+    return () => this.envelopeHandlers.delete(handler);
+  }
+
+  onClose(handler: CloseHandler): () => void {
+    this.closeHandlers.add(handler);
+    return () => this.closeHandlers.delete(handler);
+  }
+
+  close(reason?: string): void {
+    this.socket.close(1000, reason);
+  }
+}
+
+export function createWebSocketPacketChannel(peerId: string, socket: WebSocket): PacketChannel {
+  return new WebSocketPacketChannel(peerId, socket);
+}