feat(peer): add peer.close() and peer.terminate() support

src/adapters/bun.ts

@@ -120,4 +120,12 @@ class BunPeer extends Peer<{
   unsubscribe(topic: string): void {
     this.ctx.bun.ws.unsubscribe(topic);
   }
+
+  close(code?: number, reason?: string) {
+    this.ctx.bun.ws.close(code, reason);
+  }
+
+  terminate() {
+    this.ctx.bun.ws.terminate();
+  }
 }

src/adapters/cloudflare.ts

@@ -108,4 +108,15 @@ class CloudflarePeer extends Peer<{
     this.ctx.cloudflare.server.send(toBufferLike(message));
     return 0;
   }
+
+  close(code?: number, reason?: string) {
+    this.ctx.cloudflare.client.close(code, reason);
+  }
+
+  /**
+   * Cloudflare WebSockets do not support termination. This calls the `close()` method.
+   */
+  terminate(): void {
+    this.close();
+  }
 }

src/adapters/deno.ts

@@ -94,4 +94,12 @@ class DenoPeer extends Peer<{
     this.ctx.deno.ws.send(toBufferLike(message));
     return 0;
   }
+
+  close(code?: number, reason?: string) {
+    this.ctx.deno.ws.close(code, reason);
+  }
+
+  terminate(): void {
+    this.ctx.deno.ws.terminate()
+  }
 }

src/adapters/node.ts

@@ -163,4 +163,12 @@ class NodePeer extends Peer<{
       }
     }
   }
+
+  close(code?: number, data?: string | Buffer) {
+    this.ctx.node.ws.close(code, data);
+  }
+
+  terminate() {
+    this.ctx.node.ws.terminate();
+  }
 }

src/adapters/uws.ts

@@ -6,6 +6,7 @@ import type {
   WebSocket,
   HttpRequest,
   HttpResponse,
+  RecognizedString,
 } from "uWebSockets.js";
 import { Peer } from "../peer";
 import { Message } from "../message";
@@ -194,6 +195,14 @@ class UWSPeer extends Peer<{
     this.ctx.uws.ws.publish(topic, message, options?.binary, options?.compress);
     return 0;
   }
+
+  close(code?: number, reason?: RecognizedString) {
+    this.ctx.uws.ws.end(code, reason);
+  }
+
+  terminate(): void {
+    this.ctx.uws.ws.close();
+  }
 }
 
 function _getHeaders(req: HttpRequest): HeadersInit {

src/peer.ts

@@ -68,4 +68,30 @@ export abstract class Peer<AdapterContext = any> implements WSRequest {
 
     return `${_id}${_addr}${_state}`;
   }
+
+  /**
+   * Closes the connection.
+   *
+   * Here is a list of close codes:
+   *
+   * - `1000` means "normal closure" (default)
+   * - `1009` means a message was too big and was rejected
+   * - `1011` means the server encountered an error
+   * - `1012` means the server is restarting
+   * - `1013` means the server is too busy or the client is rate-limited
+   * - `4000` through `4999` are reserved for applications (you can use it!)
+   *
+   * To close the connection abruptly, use `terminate()`.
+   *
+   * @param code The close code to send
+   * @param reason The close reason to send
+   */
+  abstract close(code?: number, reason?: string): void;
+
+  /**
+   * Abruptly close the connection.
+   *
+   * To gracefully close the connection, use `close()`.
+   */
+  abstract terminate(): void;
 }