@jmealo/fastify-uws 1.0.4

package/dist/response.js

@@ -0,0 +1,131 @@
+import { STATUS_CODES as u } from "node:http";
+import { Writable as a } from "streamx";
+import { ERR_HEAD_SET as o, ERR_STREAM_DESTROYED as h } from "./errors.js";
+import { kHeaders as n, kHead as d } from "./symbols.js";
+class c {
+  isMultiValue;
+  name;
+  value;
+  constructor(t, e) {
+    this.isMultiValue = Array.isArray(e), this.name = t, this.value = this.isMultiValue ? e : String(e);
+  }
+}
+const y = Buffer.alloc(0);
+function f(i, t = !1) {
+  return i ? {
+    chunk: i,
+    empty: !1,
+    end: t,
+    byteLength: Buffer.isBuffer(i) ? i.byteLength : Buffer.byteLength(i)
+  } : { chunk: y, empty: !0, end: t, byteLength: 0 };
+}
+const p = () => {
+}, l = {
+  byteLength(i) {
+    return i.byteLength;
+  }
+};
+class k extends a {
+  socket;
+  statusCode = 200;
+  statusMessage;
+  headersSent = !1;
+  chunked = !1;
+  contentLength = null;
+  writableEnded = !1;
+  firstChunk = !0;
+  _boundEmitDrain;
+  [n];
+  constructor(t) {
+    super(l), this.socket = t, this._boundEmitDrain = this._emitDrain.bind(this), this[n] = /* @__PURE__ */ new Map();
+    const e = this.destroy.bind(this);
+    this.once("error", p), t.once("error", e), t.once("close", e), t.once("aborted", () => {
+      this.emit("aborted");
+    });
+  }
+  get aborted() {
+    return this.socket.aborted;
+  }
+  get finished() {
+    return this.socket.writableEnded && !this.socket.aborted;
+  }
+  get status() {
+    return `${this.statusCode} ${this.statusMessage || u[this.statusCode]}`;
+  }
+  get bytesWritten() {
+    return this.socket.bytesWritten;
+  }
+  hasHeader(t) {
+    return this[n].has(t.toLowerCase());
+  }
+  getHeader(t) {
+    return this[n].get(t.toLowerCase())?.value;
+  }
+  getHeaders() {
+    const t = {};
+    return this[n].forEach((e, s) => {
+      t[s] = e.value;
+    }), t;
+  }
+  setHeader(t, e) {
+    if (this.headersSent) throw new o();
+    t = t.replace(/[\r\n]/g, "");
+    const s = t.toLowerCase();
+    return s === "content-length" ? (this.contentLength = Number(e), this) : s === "transfer-encoding" ? (this.chunked = String(e).includes("chunked"), this) : (typeof e == "string" ? e = e.replace(/[\r\n]/g, "") : Array.isArray(e) && (e = e.map((r) => typeof r == "string" ? r.replace(/[\r\n]/g, "") : r)), this[n].set(s, new c(t, e)), this);
+  }
+  removeHeader(t) {
+    if (this.headersSent) throw new o();
+    this[n].delete(t.toLowerCase());
+  }
+  writeHead(t, e, s) {
+    if (this.headersSent) throw new o();
+    if (this.statusCode = t, typeof e == "object" ? s = e : e && (this.statusMessage = typeof e == "string" ? e.replace(/[\r\n]/g, "") : e), s)
+      for (const r of Object.keys(s))
+        this.setHeader(r, s[r]);
+    return this;
+  }
+  end(t, e, s) {
+    if (typeof t == "function" ? (s = t, t = void 0) : typeof e == "function" && (s = e), this.aborted)
+      return s && process.nextTick(s), this;
+    if (this.destroyed) throw new h();
+    return this.writableEnded = !0, s && this.once("finish", s), super.end(f(t, !0)), this;
+  }
+  addTrailers() {
+  }
+  destroy(t) {
+    return this.destroyed || this.destroying || this.aborted ? this : (this.socket.destroy(t), this);
+  }
+  write(t, e, s) {
+    if (typeof e == "function" && (s = e), this.aborted)
+      return s && process.nextTick(s), !1;
+    if (this.destroyed) throw new h();
+    const r = f(t);
+    return this.firstChunk && this.contentLength !== null && this.contentLength === r.byteLength ? (r.end = !0, this.writableEnded = !0, super.end(r), s && process.nextTick(s), !0) : (this.firstChunk = !1, this.headersSent || (this.headersSent = !0, this.socket[d] = {
+      headers: this[n],
+      status: this.status
+    }), this.socket.write(r, null, () => {
+      this._boundEmitDrain(), s && s();
+    }), !this.socket.writableNeedDrain);
+  }
+  _emitDrain() {
+    this.emit("drain");
+  }
+  _write(t, e) {
+    if (this.aborted) return e();
+    if (this.headersSent || (this.headersSent = !0, this.socket[d] = {
+      headers: this[n],
+      status: this.status
+    }), t.end) {
+      this.socket.end(t, null, e);
+      return;
+    }
+    this.socket.write(t, null, e);
+  }
+  _destroy(t) {
+    if (this.socket.destroyed) return t();
+    this.socket.once("close", t);
+  }
+}
+export {
+  k as Response
+};

package/dist/server.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const k=require("uWebSockets.js"),E=require("node:dns/promises"),S=require("node:fs"),g=require("node:http"),y=require("node:os"),L=require("node:path"),R=require("node:events"),q=require("ipaddr.js"),u=require("./errors.cjs"),w=require("./http-socket.cjs"),D=require("./request.cjs"),H=require("./response.cjs"),e=require("./symbols.cjs");function T(t,s,r){if(t!=null){if(Array.isArray(t)){if(t.length===0)return;if(t.length>1)throw new Error(`fastify-uws: multiple ${s} values are not supported by uWebSockets.js`);t=t[0]}if(typeof t=="string")return t;if(Buffer.isBuffer(t)){const i=L.join(y.tmpdir(),`fastify-uws-${s}-${process.pid}-${Date.now()}.pem`);return S.writeFileSync(i,t,{mode:384}),r.push(i),i}}}function C(t){if(t.https){const s=[],r=T(t.https.key,"key",s),i=T(t.https.cert,"cert",s),o=T(t.https.ca,"ca",s),c={};return r&&(c.key_file_name=r),i&&(c.cert_file_name=i),o&&(c.ca_file_name=o),t.https.passphrase&&(c.passphrase=t.https.passphrase),{app:k.SSLApp(c),tempFiles:s}}return{app:k.App(),tempFiles:[]}}const v=new Map(g.METHODS.map(t=>[t.toLowerCase(),t])),h={};class F extends R.EventEmitter{[e.kHandler];timeout=0;drainTimeout;maxHeadersCount=0;maxRequestsPerSocket=0;headersTimeout=0;keepAliveTimeout=0;requestTimeout=0;[e.kHttps];[e.kWs];[e.kAddress]=null;[e.kListenSocket]=null;[e.kApp];[e.kClosed]=!1;[e.kListenAll]=!1;[e.kListening]=!1;[e.kTempFiles];constructor(s,r={}){super();const{http2:i=!1,https:o=null,connectionTimeout:c=0,drainTimeout:a}=r;this[e.kHandler]=s,this.timeout=c,this.drainTimeout=a,this[e.kHttps]=o,this[e.kWs]=null;const{app:l,tempFiles:f}=C({https:o});this[e.kApp]=l,this[e.kTempFiles]=f}get encrypted(){return!!this[e.kHttps]}get listening(){return!!this[e.kListening]}setTimeout(s,r){return this.timeout=s,r&&this.once("timeout",r),this}address(){return this[e.kAddress]}listen(s,r){s?.signal&&s.signal.addEventListener("abort",()=>{this.close()},{once:!0}),this[e.kListen](s).then(()=>{this[e.kListening]=!0,r?.(),this.emit("listening")}).catch(i=>{this[e.kAddress]=null,process.nextTick(()=>this.emit("error",i))})}closeIdleConnections(){}closeAllConnections(){if(this[e.kWs])for(const s of this[e.kWs].connections)s.close()}close(s=()=>{}){if(this[e.kClosed])return s();const r=this[e.kAddress]?.port;r!==void 0&&h[r]===this&&delete h[r],this[e.kAddress]=null,this[e.kListening]=!1,this[e.kClosed]=!0,this[e.kListenSocket]&&(k.us_listen_socket_close(this[e.kListenSocket]),this[e.kListenSocket]=null),this.closeAllConnections();for(const i of this[e.kTempFiles])try{S.unlinkSync(i)}catch{}this[e.kTempFiles]=[],process.nextTick(()=>{this.emit("close"),s()})}ref(){}unref(){}async[e.kListen]({port:s,host:r}){if(this[e.kClosed])throw new u.ERR_SERVER_DESTROYED;if(s!=null&&Number.isNaN(Number(s)))throw new u.ERR_SOCKET_BAD_PORT(s);let i=s==null?0:Number(s);const o=await E.lookup(r);if(this[e.kAddress]={address:o.address,family:o.family===6?"IPv6":"IPv4",port:i},this[e.kAddress].address.startsWith("["))throw new u.ERR_ENOTFOUND(this[e.kAddress].address);const a=q.parse(this[e.kAddress].address).toNormalizedString(),l=this[e.kApp],f=(m,p)=>{const d=v.get(p.getMethod()),n=new w.HTTPSocket(this,m,d==="GET"||d==="HEAD");if(!d){n[e.kClientError]=!0,this.emit("clientError",new u.ERR_INVALID_METHOD,n);return}const A=new D.Request(p,n,d),_=new H.Response(n);A.headers.upgrade&&this.emit("upgrade",A,n),this[e.kHandler](A,_)};return l.any("/*",f),i!==0&&h[i]&&(this[e.kWs]=h[i][e.kWs]),this[e.kWs]&&this[e.kWs].addServer(this),new Promise((m,p)=>{const d=n=>{if(!n)return p(new u.ERR_ADDRINUSE(this[e.kAddress].address,i));this[e.kListenSocket]=n,i=this[e.kAddress].port=k.us_socket_local_port(n),h[i]||(h[i]=this),m()};this[e.kListenAll]=r==="localhost",this[e.kListenAll]?l.listen(i,d):l.listen(a,i,d)})}}exports.Server=F;

package/dist/server.d.ts

@@ -0,0 +1,51 @@
+import { default as uws } from 'uWebSockets.js';
+import { ServerOptions } from 'node:https';
+import { AddressInfo } from 'node:net';
+import { EventEmitter } from 'node:events';
+import { FastifyServerFactoryHandler, FastifyServerOptions } from 'fastify';
+import { kAddress, kApp, kClosed, kHandler, kHttps, kListen, kListenAll, kListening, kListenSocket, kTempFiles, kWs } from './symbols';
+import { WebSocketServer } from './websocket-server';
+interface FastifyUwsOptions extends FastifyServerOptions {
+    http2?: boolean;
+    https?: ServerOptions | null;
+    drainTimeout?: number;
+}
+export declare class Server extends EventEmitter {
+    [kHandler]: FastifyServerFactoryHandler;
+    timeout: number;
+    drainTimeout?: number;
+    maxHeadersCount: number;
+    maxRequestsPerSocket: number;
+    headersTimeout: number;
+    keepAliveTimeout: number;
+    requestTimeout: number;
+    [kHttps]?: FastifyUwsOptions['https'];
+    [kWs]?: WebSocketServer | null;
+    [kAddress]: AddressInfo | null;
+    [kListenSocket]: uws.us_listen_socket | null;
+    [kApp]: uws.TemplatedApp;
+    [kClosed]: boolean;
+    [kListenAll]: boolean;
+    [kListening]: boolean;
+    [kTempFiles]: string[];
+    constructor(handler: FastifyServerFactoryHandler, opts?: FastifyUwsOptions);
+    get encrypted(): boolean;
+    get listening(): boolean;
+    setTimeout(timeout: number, cb?: () => void): this;
+    address(): AddressInfo | string | null;
+    listen(listenOptions: {
+        host: string;
+        port: number;
+        signal?: AbortSignal;
+    }, cb?: () => void): void;
+    closeIdleConnections(): void;
+    closeAllConnections(): void;
+    close(cb?: () => void): void;
+    ref(): void;
+    unref(): void;
+    [kListen]({ port: rawPort, host }: {
+        port?: number;
+        host: string;
+    }): Promise<void>;
+}
+export {};

package/dist/server.js

@@ -0,0 +1,155 @@
+import E from "uWebSockets.js";
+import H from "node:dns/promises";
+import N from "node:fs";
+import { METHODS as O } from "node:http";
+import v from "node:os";
+import F from "node:path";
+import { EventEmitter as I } from "node:events";
+import x from "ipaddr.js";
+import { ERR_SERVER_DESTROYED as M, ERR_SOCKET_BAD_PORT as q, ERR_ENOTFOUND as B, ERR_INVALID_METHOD as $, ERR_ADDRINUSE as b } from "./errors.js";
+import { HTTPSocket as P } from "./http-socket.js";
+import { Request as V } from "./request.js";
+import { Response as W } from "./response.js";
+import { kHandler as y, kHttps as g, kWs as p, kAddress as r, kListenSocket as m, kApp as w, kClosed as a, kListenAll as S, kListening as d, kTempFiles as k, kListen as L, kClientError as j } from "./symbols.js";
+function D(e, t, i) {
+  if (e != null) {
+    if (Array.isArray(e)) {
+      if (e.length === 0) return;
+      if (e.length > 1)
+        throw new Error(`fastify-uws: multiple ${t} values are not supported by uWebSockets.js`);
+      e = e[0];
+    }
+    if (typeof e == "string") return e;
+    if (Buffer.isBuffer(e)) {
+      const s = F.join(v.tmpdir(), `fastify-uws-${t}-${process.pid}-${Date.now()}.pem`);
+      return N.writeFileSync(s, e, { mode: 384 }), i.push(s), s;
+    }
+  }
+}
+function U(e) {
+  if (e.https) {
+    const t = [], i = D(
+      e.https.key,
+      "key",
+      t
+    ), s = D(
+      e.https.cert,
+      "cert",
+      t
+    ), n = D(
+      e.https.ca,
+      "ca",
+      t
+    ), h = {};
+    return i && (h.key_file_name = i), s && (h.cert_file_name = s), n && (h.ca_file_name = n), e.https.passphrase && (h.passphrase = e.https.passphrase), { app: E.SSLApp(h), tempFiles: t };
+  }
+  return { app: E.App(), tempFiles: [] };
+}
+const z = new Map(O.map((e) => [e.toLowerCase(), e])), f = {};
+class nt extends I {
+  [y];
+  timeout = 0;
+  drainTimeout;
+  maxHeadersCount = 0;
+  maxRequestsPerSocket = 0;
+  headersTimeout = 0;
+  keepAliveTimeout = 0;
+  requestTimeout = 0;
+  [g];
+  [p];
+  [r] = null;
+  [m] = null;
+  [w];
+  [a] = !1;
+  [S] = !1;
+  [d] = !1;
+  [k];
+  constructor(t, i = {}) {
+    super();
+    const { http2: s = !1, https: n = null, connectionTimeout: h = 0, drainTimeout: _ } = i;
+    this[y] = t, this.timeout = h, this.drainTimeout = _, this[g] = n, this[p] = null;
+    const { app: l, tempFiles: T } = U({ https: n });
+    this[w] = l, this[k] = T;
+  }
+  get encrypted() {
+    return !!this[g];
+  }
+  get listening() {
+    return !!this[d];
+  }
+  setTimeout(t, i) {
+    return this.timeout = t, i && this.once("timeout", i), this;
+  }
+  address() {
+    return this[r];
+  }
+  listen(t, i) {
+    t?.signal && t.signal.addEventListener(
+      "abort",
+      () => {
+        this.close();
+      },
+      { once: !0 }
+    ), this[L](t).then(() => {
+      this[d] = !0, i?.(), this.emit("listening");
+    }).catch((s) => {
+      this[r] = null, process.nextTick(() => this.emit("error", s));
+    });
+  }
+  closeIdleConnections() {
+  }
+  closeAllConnections() {
+    if (this[p])
+      for (const t of this[p].connections)
+        t.close();
+  }
+  close(t = () => {
+  }) {
+    if (this[a]) return t();
+    const i = this[r]?.port;
+    i !== void 0 && f[i] === this && delete f[i], this[r] = null, this[d] = !1, this[a] = !0, this[m] && (E.us_listen_socket_close(this[m]), this[m] = null), this.closeAllConnections();
+    for (const s of this[k])
+      try {
+        N.unlinkSync(s);
+      } catch {
+      }
+    this[k] = [], process.nextTick(() => {
+      this.emit("close"), t();
+    });
+  }
+  ref() {
+  }
+  unref() {
+  }
+  async [L]({ port: t, host: i }) {
+    if (this[a]) throw new M();
+    if (t != null && Number.isNaN(Number(t)))
+      throw new q(t);
+    let s = t == null ? 0 : Number(t);
+    const n = await H.lookup(i);
+    if (this[r] = {
+      address: n.address,
+      family: n.family === 6 ? "IPv6" : "IPv4",
+      port: s
+    }, this[r].address.startsWith("[")) throw new B(this[r].address);
+    const _ = x.parse(this[r].address).toNormalizedString(), l = this[w], T = (A, u) => {
+      const c = z.get(u.getMethod()), o = new P(this, A, c === "GET" || c === "HEAD");
+      if (!c) {
+        o[j] = !0, this.emit("clientError", new $(), o);
+        return;
+      }
+      const R = new V(u, o, c), C = new W(o);
+      R.headers.upgrade && this.emit("upgrade", R, o), this[y](R, C);
+    };
+    return l.any("/*", T), s !== 0 && f[s] && (this[p] = f[s][p]), this[p] && this[p].addServer(this), new Promise((A, u) => {
+      const c = (o) => {
+        if (!o) return u(new b(this[r].address, s));
+        this[m] = o, s = this[r].port = E.us_socket_local_port(o), f[s] || (f[s] = this), A();
+      };
+      this[S] = i === "localhost", this[S] ? l.listen(s, c) : l.listen(_, s, c);
+    });
+  }
+}
+export {
+  nt as Server
+};

package/dist/symbols.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const s=Symbol("uws.https"),e=Symbol("uws.res"),o=Symbol("uws.server"),t=Symbol("uws.headers"),n=Symbol("uws.url"),l=Symbol("uws.address"),k=Symbol("uws.remoteAddress"),d=Symbol("uws.encoding"),r=Symbol("uws.timeoutRef"),c=Symbol("uws.ended"),m=Symbol("uws.readyState"),i=Symbol("uws.writeOnly"),S=Symbol("uws.handler"),u=Symbol("uws.listenSocket"),y=Symbol("uws.listen"),w=Symbol("uws.app"),b=Symbol("uws.closed"),a=Symbol("uws.ws"),p=Symbol("uws.topic"),A=Symbol("uws.uwsRemoteAddress"),R=Symbol("uws.head"),g=Symbol("uws.listenAll"),H=Symbol("uws.listening"),L=Symbol("uws.clientError"),E=Symbol("uws.tempFiles");exports.kAddress=l;exports.kApp=w;exports.kClientError=L;exports.kClosed=b;exports.kEncoding=d;exports.kEnded=c;exports.kHandler=S;exports.kHead=R;exports.kHeaders=t;exports.kHttps=s;exports.kListen=y;exports.kListenAll=g;exports.kListenSocket=u;exports.kListening=H;exports.kReadyState=m;exports.kRemoteAddress=k;exports.kRes=e;exports.kServer=o;exports.kTempFiles=E;exports.kTimeoutRef=r;exports.kTopic=p;exports.kUrl=n;exports.kUwsRemoteAddress=A;exports.kWriteOnly=i;exports.kWs=a;

package/dist/symbols.d.ts

@@ -0,0 +1,27 @@
+export declare const kHttps: unique symbol;
+export declare const kRes: unique symbol;
+export declare const kServer: unique symbol;
+export declare const kHeaders: unique symbol;
+export declare const kUrl: unique symbol;
+export declare const kAddress: unique symbol;
+export declare const kRemoteAddress: unique symbol;
+export declare const kEncoding: unique symbol;
+export declare const kTimeoutRef: unique symbol;
+export declare const kEnded: unique symbol;
+export declare const kReadyState: unique symbol;
+export declare const kWriteOnly: unique symbol;
+export declare const kHandler: unique symbol;
+export declare const kListenSocket: unique symbol;
+export declare const kListen: unique symbol;
+export declare const kApp: unique symbol;
+export declare const kClosed: unique symbol;
+export declare const kWs: unique symbol;
+export declare const kTopic: unique symbol;
+export declare const kDestroyError: unique symbol;
+export declare const kUwsRemoteAddress: unique symbol;
+export declare const kHead: unique symbol;
+export declare const kWebSocketOptions: unique symbol;
+export declare const kListenAll: unique symbol;
+export declare const kListening: unique symbol;
+export declare const kClientError: unique symbol;
+export declare const kTempFiles: unique symbol;

package/dist/symbols.js

@@ -0,0 +1,28 @@
+const s = /* @__PURE__ */ Symbol("uws.https"), o = /* @__PURE__ */ Symbol("uws.res"), e = /* @__PURE__ */ Symbol("uws.server"), t = /* @__PURE__ */ Symbol("uws.headers"), n = /* @__PURE__ */ Symbol("uws.url"), l = /* @__PURE__ */ Symbol("uws.address"), c = /* @__PURE__ */ Symbol("uws.remoteAddress"), m = /* @__PURE__ */ Symbol("uws.encoding"), S = /* @__PURE__ */ Symbol("uws.timeoutRef"), u = /* @__PURE__ */ Symbol("uws.ended"), w = /* @__PURE__ */ Symbol("uws.readyState"), y = /* @__PURE__ */ Symbol("uws.writeOnly"), d = /* @__PURE__ */ Symbol("uws.handler"), r = /* @__PURE__ */ Symbol("uws.listenSocket"), k = /* @__PURE__ */ Symbol("uws.listen"), b = /* @__PURE__ */ Symbol("uws.app"), i = /* @__PURE__ */ Symbol("uws.closed"), a = /* @__PURE__ */ Symbol("uws.ws"), p = /* @__PURE__ */ Symbol("uws.topic"), A = /* @__PURE__ */ Symbol("uws.uwsRemoteAddress"), R = /* @__PURE__ */ Symbol("uws.head"), g = /* @__PURE__ */ Symbol("uws.listenAll"), h = /* @__PURE__ */ Symbol("uws.listening"), E = /* @__PURE__ */ Symbol("uws.clientError"), H = /* @__PURE__ */ Symbol("uws.tempFiles");
+export {
+  l as kAddress,
+  b as kApp,
+  E as kClientError,
+  i as kClosed,
+  m as kEncoding,
+  u as kEnded,
+  d as kHandler,
+  R as kHead,
+  t as kHeaders,
+  s as kHttps,
+  k as kListen,
+  g as kListenAll,
+  r as kListenSocket,
+  h as kListening,
+  w as kReadyState,
+  c as kRemoteAddress,
+  o as kRes,
+  e as kServer,
+  H as kTempFiles,
+  S as kTimeoutRef,
+  p as kTopic,
+  n as kUrl,
+  A as kUwsRemoteAddress,
+  y as kWriteOnly,
+  a as kWs
+};

package/dist/websocket-server.cjs

@@ -0,0 +1 @@
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});const m=require("uWebSockets.js"),f=require("eventemitter3"),p=require("streamx"),g=require("./http-socket.cjs"),y=require("./request.cjs"),E=require("./response.cjs"),r=require("./symbols.cjs"),S={compression:m.SHARED_COMPRESSOR,maxPayloadLength:16*1024*1024,idleTimeout:16},l="!",B=Buffer.from(l);class h extends f.EventEmitter{namespace;connection;topics;[r.kEnded];static allocTopic(e,t){if(t[r.kTopic])return t;const o=Buffer.concat([e,B,Buffer.isBuffer(t)?t:Buffer.from(t)]);return o[r.kTopic]=!0,o}constructor(e,t,o={}){super(),this.namespace=e,this.connection=t,t.websocket=this,this.topics=o,this[r.kEnded]=!1}get uws(){return!0}allocTopic(e){return this.topics[e]?this.topics[e]:h.allocTopic(this.namespace,e)}send(e,t,o){if(!this[r.kEnded])return this.connection.send(e,t,o)}publish(e,t,o,s){if(!this[r.kEnded])return this.connection.publish(this.allocTopic(e),t,o,s)}subscribe(e){if(!this[r.kEnded])return this.connection.subscribe(this.allocTopic(e))}unsubscribe(e){if(!this[r.kEnded])return this.connection.unsubscribe(this.allocTopic(e))}isSubscribed(e){return this[r.kEnded]?!1:this.connection.isSubscribed(this.allocTopic(e))}getTopics(){return this[r.kEnded]?[]:this.connection.getTopics().map(e=>e.slice(e.indexOf(l)+1))}close(){if(!this[r.kEnded])return this[r.kEnded]=!0,this.connection.close()}end(e,t){if(!this[r.kEnded])return this[r.kEnded]=!0,this.connection.end(e,t)}cork(e){if(!this[r.kEnded])return this.connection.cork(e)}getBufferedAmount(){return this[r.kEnded]?0:this.connection.getBufferedAmount()}ping(e){if(!this[r.kEnded])return this.connection.ping(e)}}class W extends p.Duplex{socket;constructor(e,t={}){const{compress:o=!1}=t;super({highWaterMark:t.highWaterMark,mapReadable:s=>t.mapReadable?t.mapReadable(s):s.data,byteLengthReadable:s=>t.byteLengthReadable?t.byteLengthReadable(s):s.isBinary?s.data.byteLength:1024,mapWritable:s=>t.mapWritable?t.mapWritable(s):{data:s,isBinary:Buffer.isBuffer(s),compress:o},byteLengthWritable:s=>t.byteLengthWritable?t.byteLengthWritable(s):s.isBinary?s.data.byteLength:1024}),this.socket=e,this._onMessage=this._onMessage.bind(this)}_open(e){this.socket.on("message",this._onMessage),e()}_close(e){this.socket.off("message",this._onMessage),this.socket.close(),e()}_onMessage(e,t){this.push({data:e,isBinary:t})}_write(e,t){const o=this.socket.send(e.data,e.isBinary,e.compress);if(o===1)return t();if(o===0){const s=()=>{this.socket.off("close",n),t()},n=()=>{this.socket.off("drain",s),t(new Error("WebSocket closed before drain"))};this.socket.once("drain",s),this.socket.once("close",n);return}if(o===2)return t(new Error("WebSocket send dropped due to backpressure limit"));t(new Error("WebSocket is closed"))}}class T extends f.EventEmitter{options;connections;constructor(e={}){super(),this.options={...S,...e},this.connections=new Set}addServer(e){const{options:t}=this,o=e[r.kApp],s=e[r.kHandler];o.ws("/*",{upgrade:(n,i,c)=>{const u=i.getMethod().toUpperCase(),a=new g.HTTPSocket(e,n,u==="GET"||u==="HEAD"),d=new y.Request(i,a,u),k=new E.Response(a);d[r.kWs]=c,e.emit("upgrade",d,a),s(d,k)},open:n=>{this.connections.add(n),n.handler(n),this.emit("open",n)},close:(n,i,c)=>{this.connections.delete(n);const u=n,a=u.websocket;a&&(a[r.kEnded]=!0),u.req.socket.destroy();const d=c instanceof ArrayBuffer?Buffer.copyBytesFrom(new Uint8Array(c)):c;a?.emit("close",i,d),this.emit("close",n,i,d)},drain:n=>{n.websocket?.emit("drain"),this.emit("drain",n)},message:(n,i,c)=>{const u=i instanceof ArrayBuffer?Buffer.copyBytesFrom(new Uint8Array(i)):i;n.websocket?.emit("message",u,c),this.emit("message",n,u,c)},ping:(n,i)=>{const c=i instanceof ArrayBuffer?Buffer.copyBytesFrom(new Uint8Array(i)):i;n.websocket?.emit("ping",c),this.emit("ping",n,c)},pong:(n,i)=>{const c=i instanceof ArrayBuffer?Buffer.copyBytesFrom(new Uint8Array(i)):i;n.websocket?.emit("pong",c),this.emit("pong",n,c)},...t})}}exports.WebSocket=h;exports.WebSocketServer=T;exports.WebSocketStream=W;

package/dist/websocket-server.d.ts

@@ -0,0 +1,82 @@
+import { default as uws } from 'uWebSockets.js';
+import { EventEmitter } from 'eventemitter3';
+import { Duplex } from 'streamx';
+import { Request } from './request';
+import { Server } from './server';
+import { kEnded } from './symbols';
+declare const defaultWebSocketConfig: {
+    compression: number;
+    maxPayloadLength: number;
+    idleTimeout: number;
+};
+export interface WsUserData {
+    req: Request;
+    handler: (ws: uws.WebSocket<WsUserData>) => void;
+    websocket?: WebSocket;
+}
+export declare class WebSocket extends EventEmitter {
+    namespace: Buffer;
+    connection: uws.WebSocket<WsUserData>;
+    topics: Record<string, Buffer>;
+    [kEnded]: boolean;
+    static allocTopic(namespace: Buffer, topic: Buffer | string): string | Buffer<ArrayBufferLike>;
+    constructor(namespace: Buffer, connection: uws.WebSocket<WsUserData>, topics?: Record<string, Buffer>);
+    get uws(): boolean;
+    allocTopic(topic: Buffer | string): string | Buffer<ArrayBufferLike>;
+    send(message: uws.RecognizedString, isBinary: boolean, compress: boolean): number | undefined;
+    publish(topic: Buffer | string, message: uws.RecognizedString, isBinary: boolean, compress: boolean): boolean | undefined;
+    subscribe(topic: Buffer | string): boolean | undefined;
+    unsubscribe(topic: Buffer | string): boolean | undefined;
+    isSubscribed(topic: Buffer | string): boolean;
+    getTopics(): string[];
+    close(): void;
+    end(code: number, shortMessage: uws.RecognizedString): void;
+    cork(cb: () => void): uws.WebSocket<WsUserData> | undefined;
+    getBufferedAmount(): number;
+    ping(message: uws.RecognizedString): number | undefined;
+}
+export declare class WebSocketStream extends Duplex {
+    socket: WebSocket;
+    constructor(socket: WebSocket, opts?: {
+        compress?: boolean | false;
+        highWaterMark?: number | 16384;
+        mapReadable?: (packet: {
+            data: any;
+            isBinary: boolean;
+        }) => any;
+        byteLengthReadable?: (packet: {
+            data: any;
+            isBinary: boolean;
+        }) => number | 1024;
+        mapWritable?: (data: any) => {
+            data: any;
+            isBinary: boolean;
+            compress: boolean;
+        };
+        byteLengthWritable?: (packet: {
+            data: any;
+            isBinary: boolean;
+            compress: boolean;
+        }) => number | 1024;
+    });
+    _open(cb: () => void): void;
+    _close(cb: () => void): void;
+    _onMessage(data: any, isBinary: boolean): void;
+    _write(packet: any, cb: (err?: Error | null) => void): void;
+}
+type WSOptions = {
+    closeOnBackpressureLimit?: boolean;
+    compression?: number;
+    idleTimeout?: number;
+    maxBackpressure?: number;
+    maxLifetime?: number;
+    maxPayloadLength?: number;
+    sendPingsAutomatically?: boolean;
+};
+export declare class WebSocketServer extends EventEmitter {
+    options: WSOptions & typeof defaultWebSocketConfig;
+    connections: Set<uws.WebSocket<WsUserData>>;
+    constructor(options?: WSOptions);
+    addServer(server: Server): void;
+}
+export {};

package/dist/websocket-server.js

@@ -0,0 +1,162 @@
+import g from "uWebSockets.js";
+import { EventEmitter as l } from "eventemitter3";
+import { Duplex as k } from "streamx";
+import { HTTPSocket as y } from "./http-socket.js";
+import { Request as B } from "./request.js";
+import { Response as S } from "./response.js";
+import { kApp as W, kHandler as T, kEnded as o, kWs as w, kTopic as h } from "./symbols.js";
+const A = {
+  compression: g.SHARED_COMPRESSOR,
+  maxPayloadLength: 16 * 1024 * 1024,
+  idleTimeout: 16
+}, m = "!", E = Buffer.from(m);
+class d extends l {
+  namespace;
+  connection;
+  topics;
+  [o];
+  static allocTopic(e, t) {
+    if (t[h]) return t;
+    const i = Buffer.concat([
+      e,
+      E,
+      Buffer.isBuffer(t) ? t : Buffer.from(t)
+    ]);
+    return i[h] = !0, i;
+  }
+  constructor(e, t, i = {}) {
+    super(), this.namespace = e, this.connection = t, t.websocket = this, this.topics = i, this[o] = !1;
+  }
+  get uws() {
+    return !0;
+  }
+  allocTopic(e) {
+    return this.topics[e] ? this.topics[e] : d.allocTopic(this.namespace, e);
+  }
+  send(e, t, i) {
+    if (!this[o])
+      return this.connection.send(e, t, i);
+  }
+  publish(e, t, i, r) {
+    if (!this[o])
+      return this.connection.publish(this.allocTopic(e), t, i, r);
+  }
+  subscribe(e) {
+    if (!this[o])
+      return this.connection.subscribe(this.allocTopic(e));
+  }
+  unsubscribe(e) {
+    if (!this[o])
+      return this.connection.unsubscribe(this.allocTopic(e));
+  }
+  isSubscribed(e) {
+    return this[o] ? !1 : this.connection.isSubscribed(this.allocTopic(e));
+  }
+  getTopics() {
+    return this[o] ? [] : this.connection.getTopics().map((e) => e.slice(e.indexOf(m) + 1));
+  }
+  close() {
+    if (!this[o])
+      return this[o] = !0, this.connection.close();
+  }
+  end(e, t) {
+    if (!this[o])
+      return this[o] = !0, this.connection.end(e, t);
+  }
+  cork(e) {
+    if (!this[o])
+      return this.connection.cork(e);
+  }
+  getBufferedAmount() {
+    return this[o] ? 0 : this.connection.getBufferedAmount();
+  }
+  ping(e) {
+    if (!this[o])
+      return this.connection.ping(e);
+  }
+}
+class D extends k {
+  socket;
+  constructor(e, t = {}) {
+    const { compress: i = !1 } = t;
+    super({
+      highWaterMark: t.highWaterMark,
+      mapReadable: (r) => t.mapReadable ? t.mapReadable(r) : r.data,
+      byteLengthReadable: (r) => t.byteLengthReadable ? t.byteLengthReadable(r) : r.isBinary ? r.data.byteLength : 1024,
+      mapWritable: (r) => t.mapWritable ? t.mapWritable(r) : { data: r, isBinary: Buffer.isBuffer(r), compress: i },
+      byteLengthWritable: (r) => t.byteLengthWritable ? t.byteLengthWritable(r) : r.isBinary ? r.data.byteLength : 1024
+    }), this.socket = e, this._onMessage = this._onMessage.bind(this);
+  }
+  _open(e) {
+    this.socket.on("message", this._onMessage), e();
+  }
+  _close(e) {
+    this.socket.off("message", this._onMessage), this.socket.close(), e();
+  }
+  _onMessage(e, t) {
+    this.push({ data: e, isBinary: t });
+  }
+  _write(e, t) {
+    const i = this.socket.send(e.data, e.isBinary, e.compress);
+    if (i === 1) return t();
+    if (i === 0) {
+      const r = () => {
+        this.socket.off("close", n), t();
+      }, n = () => {
+        this.socket.off("drain", r), t(new Error("WebSocket closed before drain"));
+      };
+      this.socket.once("drain", r), this.socket.once("close", n);
+      return;
+    }
+    if (i === 2)
+      return t(new Error("WebSocket send dropped due to backpressure limit"));
+    t(new Error("WebSocket is closed"));
+  }
+}
+class H extends l {
+  options;
+  connections;
+  constructor(e = {}) {
+    super(), this.options = { ...A, ...e }, this.connections = /* @__PURE__ */ new Set();
+  }
+  addServer(e) {
+    const { options: t } = this, i = e[W], r = e[T];
+    i.ws("/*", {
+      upgrade: (n, s, c) => {
+        const a = s.getMethod().toUpperCase(), u = new y(e, n, a === "GET" || a === "HEAD"), f = new B(s, u, a), b = new S(u);
+        f[w] = c, e.emit("upgrade", f, u), r(f, b);
+      },
+      open: (n) => {
+        this.connections.add(n), n.handler(n), this.emit("open", n);
+      },
+      close: (n, s, c) => {
+        this.connections.delete(n);
+        const a = n, u = a.websocket;
+        u && (u[o] = !0), a.req.socket.destroy();
+        const f = c instanceof ArrayBuffer ? Buffer.copyBytesFrom(new Uint8Array(c)) : c;
+        u?.emit("close", s, f), this.emit("close", n, s, f);
+      },
+      drain: (n) => {
+        n.websocket?.emit("drain"), this.emit("drain", n);
+      },
+      message: (n, s, c) => {
+        const a = s instanceof ArrayBuffer ? Buffer.copyBytesFrom(new Uint8Array(s)) : s;
+        n.websocket?.emit("message", a, c), this.emit("message", n, a, c);
+      },
+      ping: (n, s) => {
+        const c = s instanceof ArrayBuffer ? Buffer.copyBytesFrom(new Uint8Array(s)) : s;
+        n.websocket?.emit("ping", c), this.emit("ping", n, c);
+      },
+      pong: (n, s) => {
+        const c = s instanceof ArrayBuffer ? Buffer.copyBytesFrom(new Uint8Array(s)) : s;
+        n.websocket?.emit("pong", c), this.emit("pong", n, c);
+      },
+      ...t
+    });
+  }
+}
+export {
+  d as WebSocket,
+  H as WebSocketServer,
+  D as WebSocketStream
+};