@nmtjs/ws-transport 0.0.1 → 0.1.1

package/dist/index.js

@@ -2,4 +2,4 @@ export * from "./lib/server.js";
 export * from "./lib/transport.js";
 export * from "./lib/types.js";
 export * from "./lib/utils.js";
-export * from "./lib/providers.js";
+export * from "./lib/injectables.js";

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../index.ts"],"sourcesContent":["export * from './lib/server.ts'\nexport * from './lib/transport.ts'\nexport * from './lib/types.ts'\nexport * from './lib/utils.ts'\nexport * from './lib/providers.ts'\n"],"names":[],"mappings":"AAAA,cAAc,kBAAiB;AAC/B,cAAc,qBAAoB;AAClC,cAAc,iBAAgB;AAC9B,cAAc,iBAAgB;AAC9B,cAAc,qBAAoB"}
+{"version":3,"sources":["../../index.ts"],"sourcesContent":["export * from './lib/server.ts'\nexport * from './lib/transport.ts'\nexport * from './lib/types.ts'\nexport * from './lib/utils.ts'\nexport * from './lib/injectables.ts'\n"],"names":[],"mappings":"AAAA,cAAc,kBAAiB;AAC/B,cAAc,qBAAoB;AAClC,cAAc,iBAAgB;AAC9B,cAAc,iBAAgB;AAC9B,cAAc,uBAAsB"}

package/dist/lib/injectables.js

@@ -0,0 +1,2 @@
+import { injectables } from '@nmtjs/application';
+export const connectionData = injectables.connectionData;

package/dist/lib/injectables.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../lib/injectables.ts"],"sourcesContent":["import {\n  type LazyInjectable,\n  type Scope,\n  injectables,\n} from '@nmtjs/application'\nimport type { WsConnectionData } from './types.ts'\n\nexport const connectionData = injectables.connectionData as LazyInjectable<\n  WsConnectionData,\n  Scope.Connection\n>\n"],"names":["injectables","connectionData"],"mappings":"AAAA,SAGEA,WAAW,QACN,qBAAoB;AAG3B,OAAO,MAAMC,iBAAiBD,YAAYC,cAAc,CAGvD"}

package/dist/lib/server.js

@@ -2,7 +2,7 @@ import { randomUUID } from 'node:crypto';
 import { ApiError, Scope, ServerDownStream, ServerUpStream, SubscriptionResponse, onAbort } from '@nmtjs/application';
 import { MessageType, TransportType, decodeNumber, encodeNumber } from '@nmtjs/common';
 import { App, SSLApp } from 'uWebSockets.js';
-import { connectionData } from "./providers.js";
+import { connectionData } from "./injectables.js";
 import { InternalError, getFormat, getRequestData, send, sendMessage, sendRpcMessage } from "./utils.js";
 export class WsTransportServer {
     application;
@@ -14,16 +14,19 @@ export class WsTransportServer {
         this.options = options;
         this.transportType = TransportType.WS;
         this.server = this.options.tls ? SSLApp(options.tls) : App();
-        this.server.get('/healthy', (res)=>{
-            res.writeHeader('Access-Control-Allow-Origin', '*');
-            res.writeHeader('Access-Control-Allow-Headers', 'Content-Type');
-            res.writeHeader('Access-Control-Allow-Methods', 'GET');
+        this.server.options('/*', (res, req)=>{
+            this.applyCors(res, req);
+            res.writeStatus('200 OK');
+            res.endWithoutBody();
+        }).get('/healthy', (res, req)=>{
+            this.applyCors(res, req);
             res.writeHeader('Content-Type', 'text/plain');
             res.end('OK');
         }).ws('/api', {
             sendPingsAutomatically: true,
             maxPayloadLength: this.options.maxPayloadLength,
             upgrade: (res, req, context)=>{
+                this.applyCors(res, req);
                 const requestData = getRequestData(req);
                 const container = this.application.container.createScope(Scope.Connection);
                 const services = requestData.query.getAll('services');
@@ -145,6 +148,14 @@ export class WsTransportServer {
             cause
         }));
     }
+    applyCors(res, req) {
+        const origin = req.getHeader('origin');
+        if (!origin) return;
+        res.headers.set('Access-Control-Allow-Origin', origin);
+        res.headers.set('Access-Control-Allow-Headers', 'Content-Type');
+        res.headers.set('Access-Control-Allow-Methods', 'GET, POST');
+        res.headers.set('Access-Control-Allow-Credentials', 'true');
+    }
     handleContainerDisposal(container) {
         container.dispose();
     }

package/dist/lib/server.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto'\nimport {\n  type AnyProcedure,\n  ApiError,\n  type ApplicationContext,\n  type Connection,\n  type Container,\n  Scope,\n  ServerDownStream,\n  ServerUpStream,\n  type Service,\n  SubscriptionResponse,\n  onAbort,\n} from '@nmtjs/application'\nimport {\n  type ApiBlobMetadata,\n  type EncodeRpcContext,\n  MessageType,\n  TransportType,\n  decodeNumber,\n  encodeNumber,\n} from '@nmtjs/common'\nimport { App, SSLApp, type TemplatedApp } from 'uWebSockets.js'\nimport { connectionData } from './providers.ts'\nimport type {\n  WsTransportOptions,\n  WsTransportSocket,\n  WsUserData,\n} from './types.ts'\nimport {\n  InternalError,\n  getFormat,\n  getRequestData,\n  send,\n  sendMessage,\n  sendRpcMessage,\n} from './utils.ts'\n\nexport class WsTransportServer {\n  protected server!: TemplatedApp\n  protected readonly transportType = TransportType.WS\n\n  constructor(\n    protected readonly application: ApplicationContext,\n    protected readonly options: WsTransportOptions,\n  ) {\n    this.server = this.options.tls ? SSLApp(options.tls!) : App()\n\n    this.server\n      .get('/healthy', (res) => {\n        // cors\n        res.writeHeader('Access-Control-Allow-Origin', '*')\n        res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')\n        res.writeHeader('Access-Control-Allow-Methods', 'GET')\n        res.writeHeader('Content-Type', 'text/plain')\n        res.end('OK')\n      })\n      .ws<WsUserData>('/api', {\n        sendPingsAutomatically: true,\n        maxPayloadLength: this.options.maxPayloadLength,\n        upgrade: (res, req, context) => {\n          const requestData = getRequestData(req)\n          const container = this.application.container.createScope(\n            Scope.Connection,\n          )\n          const services = requestData.query.getAll('services')\n\n          for (const serviceName of services) {\n            const service = this.application.registry.services.get(serviceName)\n            if (!service)\n              throw new Response(`Service ${service} not found`, {\n                status: 400,\n              })\n            if (this.transportType in service.contract.transports === false)\n              throw new Response(`Service ${service} not supported`, {\n                status: 400,\n              })\n          }\n\n          const data: WsUserData = {\n            id: randomUUID(),\n            format: getFormat(requestData, this.application.format),\n            container,\n            streams: {\n              up: new Map(),\n              down: new Map(),\n              streamId: 0,\n            },\n            abortControllers: new Map(),\n            subscriptions: new Map(),\n            services,\n            data: {\n              query: requestData.query,\n              headers: requestData.headers,\n              proxiedRemoteAddress: Buffer.from(\n                res.getProxiedRemoteAddressAsText(),\n              ).toString(),\n              remoteAddress: Buffer.from(\n                res.getRemoteAddressAsText(),\n              ).toString(),\n            },\n            backpressure: null,\n            context: {} as any,\n          }\n\n          res.upgrade(\n            data,\n            req.getHeader('sec-websocket-key'),\n            req.getHeader('sec-websocket-protocol'),\n            req.getHeader('sec-websocket-extensions'),\n            context,\n          )\n        },\n        open: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.context.decode = this.createDecodeRpcContext(ws)\n          data.context.encode = this.createEncodeRpcContext(ws)\n          data.container.provide(connectionData, data.data)\n          this.application.connections.add({\n            id: data.id,\n            services: data.services,\n            type: this.transportType,\n            subscriptions: data.subscriptions,\n            sendEvent: (service, event, payload) =>\n              sendMessage(ws, MessageType.Event, [service, event, payload]),\n          })\n        },\n        message: async (ws: WsTransportSocket, event) => {\n          const buffer = event as unknown as ArrayBuffer\n          const messageType = decodeNumber(buffer, 'Uint8')\n          if (messageType in this === false) {\n            ws.end(1011, 'Unknown message type')\n          } else {\n            try {\n              await this[messageType](\n                ws,\n                buffer.slice(Uint8Array.BYTES_PER_ELEMENT),\n              )\n            } catch (error: any) {\n              this.logError(error, 'Error while processing message')\n            }\n          }\n        },\n        drain: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.backpressure?.resolve()\n          data.backpressure = null\n        },\n        close: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n\n          this.application.connections.remove(data.id)\n          const error = new Error('Connection closed')\n\n          for (const ac of data.abortControllers.values()) {\n            ac.abort(error)\n          }\n          data.abortControllers.clear()\n\n          for (const stream of data.streams.down.values()) {\n            stream.destroy(error)\n          }\n          data.streams.down.clear()\n\n          for (const stream of data.streams.up.values()) {\n            stream.destroy(error)\n          }\n          data.streams.up.clear()\n\n          for (const subscription of data.subscriptions.values()) {\n            subscription.destroy()\n          }\n          data.subscriptions.clear()\n\n          this.handleContainerDisposal(data.container)\n        },\n      })\n  }\n\n  async start() {\n    return new Promise<void>((resolve, reject) => {\n      const hostname = this.options.hostname ?? '127.0.0.1'\n      this.server.listen(hostname, this.options.port!, (socket) => {\n        if (socket) {\n          this.logger.info(\n            'Server started on %s:%s',\n            hostname,\n            this.options.port!,\n          )\n          resolve()\n        } else {\n          reject(new Error('Failed to start server'))\n        }\n      })\n    })\n  }\n\n  async stop() {\n    this.server.close()\n  }\n\n  protected get api() {\n    return this.application.api\n  }\n\n  protected get logger() {\n    return this.application.logger\n  }\n\n  protected async logError(\n    cause: Error,\n    message = 'Unknown error while processing request',\n  ) {\n    this.logger.error(new Error(message, { cause }))\n  }\n\n  protected handleContainerDisposal(container: Container) {\n    container.dispose()\n  }\n\n  protected async handleRPC(options: {\n    connection: Connection\n    service: Service\n    procedure: AnyProcedure\n    container: Container\n    signal: AbortSignal\n    payload: any\n  }) {\n    return await this.api.call({\n      ...options,\n      transport: this.transportType,\n    })\n  }\n\n  protected createEncodeRpcContext(ws: WsTransportSocket): EncodeRpcContext {\n    const data = ws.getUserData()\n    return {\n      addStream: (blob) => {\n        const id = ++data.streams.streamId\n        const downstream = new ServerDownStream(id, blob)\n        downstream.pause()\n        downstream.on('error', console.dir)\n        downstream.once('error', (err) => {\n          console.log({ err })\n          if (downstream.errored?.message !== 'Aborted by client')\n            send(ws, MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        downstream.on('data', (chunk) => {\n          downstream.pause()\n          send(\n            ws,\n            MessageType.DownStreamPush,\n            encodeNumber(id, 'Uint32'),\n            Buffer.from(chunk).buffer as ArrayBuffer,\n          )\n        })\n        data.streams.down.set(id, downstream)\n        return { id, metadata: blob.metadata }\n      },\n      getStream: (id) => {\n        return data.streams.down.get(id)!\n      },\n    }\n  }\n\n  protected createDecodeRpcContext(ws: WsTransportSocket) {\n    const data = ws.getUserData()\n    return {\n      addStream(signal: AbortSignal, id: number, metadata: ApiBlobMetadata) {\n        const upstream = new ServerUpStream(metadata, {\n          read: (size) => {\n            send(\n              ws,\n              MessageType.UpStreamPull,\n              encodeNumber(id, 'Uint32'),\n              encodeNumber(size, 'Uint32'),\n            )\n          },\n        })\n\n        data.streams.up.set(id, upstream)\n        onAbort(signal, () => {\n          upstream.destroy(new Error('Call aborted by client'))\n        })\n        upstream.once('error', (error) => {\n          if (error.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        upstream.once('close', () => {\n          if (upstream.errored?.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))\n        })\n\n        return upstream\n      },\n      getStream(id) {\n        return data.streams.down.get(id)\n      },\n    }\n  }\n\n  protected async [MessageType.Rpc](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n\n    const connection = this.application.connections.get(data.id)\n    if (!connection) return void ws.end(1011, 'Unknown connection')\n\n    const ac = new AbortController()\n    const rpc = data.format.decoder.decodeRpc(buffer, {\n      ...data.context.decode,\n      addStream: data.context.decode.addStream.bind(null, ac.signal),\n    })\n\n    data.abortControllers.set(rpc.callId, ac)\n\n    const container = data.container.createScope(Scope.Call)\n\n    try {\n      const { service, procedure } = this.api.find(\n        rpc.service,\n        rpc.procedure,\n        this.transportType,\n      )\n\n      const response = await this.handleRPC({\n        connection,\n        service,\n        procedure,\n        container,\n        signal: ac.signal,\n        payload: rpc.payload,\n      })\n\n      if (response instanceof SubscriptionResponse) {\n        sendRpcMessage(ws, MessageType.RpcSubscription, {\n          callId: rpc.callId,\n          error: null,\n          payload: [response.subscription.key, response.payload],\n        })\n\n        response.subscription.on('event', (event, payload) => {\n          sendMessage(ws, MessageType.ServerSubscriptionEvent, [\n            response.subscription.key,\n            event,\n            payload,\n          ])\n        })\n        response.subscription.once('end', () => {\n          sendMessage(ws, MessageType.ServerUnsubscribe, [\n            response.subscription.key,\n          ])\n        })\n      } else {\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: null,\n          payload: response,\n        })\n      }\n    } catch (error) {\n      if (error instanceof ApiError) {\n        this.logger.debug(new Error('Api error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error,\n          payload: null,\n        })\n      } else {\n        this.logger.error(new Error('Unexpected error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: InternalError(),\n          payload: null,\n        })\n      }\n    } finally {\n      data.abortControllers.delete(rpc.callId)\n      this.handleContainerDisposal(container)\n    }\n  }\n\n  async [MessageType.UpStreamPush](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(Buffer.from(buffer.slice(Uint32Array.BYTES_PER_ELEMENT)))\n  }\n\n  async [MessageType.UpStreamEnd](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(null)\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.UpStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.DownStreamPull](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    await data.backpressure?.promise\n    if (stream.readableEnded)\n      send(ws, MessageType.DownStreamEnd, encodeNumber(id, 'Uint32'))\n    else stream.resume()\n  }\n\n  async [MessageType.DownStreamEnd](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.DownStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.ClientUnsubscribe](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const [key] = data.format.decoder.decode(buffer)\n    const subscription = data.subscriptions.get(key)\n    if (!subscription) return void ws.end()\n    subscription.destroy()\n    data.subscriptions.delete(key)\n  }\n\n  async [MessageType.RpcAbort](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const callId = decodeNumber(buffer, 'Uint32')\n    const ac = data.abortControllers.get(callId)\n    if (ac) ac.abort(new Error('Aborted by client'))\n  }\n}\n"],"names":["randomUUID","ApiError","Scope","ServerDownStream","ServerUpStream","SubscriptionResponse","onAbort","MessageType","TransportType","decodeNumber","encodeNumber","App","SSLApp","connectionData","InternalError","getFormat","getRequestData","send","sendMessage","sendRpcMessage","WsTransportServer","server","transportType","constructor","application","options","WS","tls","get","res","writeHeader","end","ws","sendPingsAutomatically","maxPayloadLength","upgrade","req","context","requestData","container","createScope","Connection","services","query","getAll","serviceName","service","registry","Response","status","contract","transports","data","id","format","streams","up","Map","down","streamId","abortControllers","subscriptions","headers","proxiedRemoteAddress","Buffer","from","getProxiedRemoteAddressAsText","toString","remoteAddress","getRemoteAddressAsText","backpressure","getHeader","open","getUserData","decode","createDecodeRpcContext","encode","createEncodeRpcContext","provide","connections","add","type","sendEvent","event","payload","Event","message","buffer","messageType","slice","Uint8Array","BYTES_PER_ELEMENT","error","logError","drain","resolve","close","remove","Error","ac","values","abort","clear","stream","destroy","subscription","handleContainerDisposal","start","Promise","reject","hostname","listen","port","socket","logger","info","stop","api","cause","dispose","handleRPC","call","transport","addStream","blob","downstream","pause","on","console","dir","once","err","log","errored","DownStreamAbort","chunk","DownStreamPush","set","metadata","getStream","signal","upstream","read","size","UpStreamPull","UpStreamAbort","UpStreamEnd","Rpc","connection","AbortController","rpc","decoder","decodeRpc","bind","callId","Call","procedure","find","response","RpcSubscription","key","ServerSubscriptionEvent","ServerUnsubscribe","debug","delete","UpStreamPush","push","Uint32Array","DownStreamPull","promise","readableEnded","DownStreamEnd","resume","ClientUnsubscribe","RpcAbort"],"mappings":"AAAA,SAASA,UAAU,QAAQ,cAAa;AACxC,SAEEC,QAAQ,EAIRC,KAAK,EACLC,gBAAgB,EAChBC,cAAc,EAEdC,oBAAoB,EACpBC,OAAO,QACF,qBAAoB;AAC3B,SAGEC,WAAW,EACXC,aAAa,EACbC,YAAY,EACZC,YAAY,QACP,gBAAe;AACtB,SAASC,GAAG,EAAEC,MAAM,QAA2B,iBAAgB;AAC/D,SAASC,cAAc,QAAQ,iBAAgB;AAM/C,SACEC,aAAa,EACbC,SAAS,EACTC,cAAc,EACdC,IAAI,EACJC,WAAW,EACXC,cAAc,QACT,aAAY;AAEnB,OAAO,MAAMC;;;IACDC,OAAqB;IACZC,cAAgC;IAEnDC,YACE,AAAmBC,WAA+B,EAClD,AAAmBC,OAA2B,CAC9C;aAFmBD,cAAAA;aACAC,UAAAA;aAJFH,gBAAgBd,cAAckB,EAAE;QAMjD,IAAI,CAACL,MAAM,GAAG,IAAI,CAACI,OAAO,CAACE,GAAG,GAAGf,OAAOa,QAAQE,GAAG,IAAKhB;QAExD,IAAI,CAACU,MAAM,CACRO,GAAG,CAAC,YAAY,CAACC;YAEhBA,IAAIC,WAAW,CAAC,+BAA+B;YAC/CD,IAAIC,WAAW,CAAC,gCAAgC;YAChDD,IAAIC,WAAW,CAAC,gCAAgC;YAChDD,IAAIC,WAAW,CAAC,gBAAgB;YAChCD,IAAIE,GAAG,CAAC;QACV,GACCC,EAAE,CAAa,QAAQ;YACtBC,wBAAwB;YACxBC,kBAAkB,IAAI,CAACT,OAAO,CAACS,gBAAgB;YAC/CC,SAAS,CAACN,KAAKO,KAAKC;gBAClB,MAAMC,cAActB,eAAeoB;gBACnC,MAAMG,YAAY,IAAI,CAACf,WAAW,CAACe,SAAS,CAACC,WAAW,CACtDtC,MAAMuC,UAAU;gBAElB,MAAMC,WAAWJ,YAAYK,KAAK,CAACC,MAAM,CAAC;gBAE1C,KAAK,MAAMC,eAAeH,SAAU;oBAClC,MAAMI,UAAU,IAAI,CAACtB,WAAW,CAACuB,QAAQ,CAACL,QAAQ,CAACd,GAAG,CAACiB;oBACvD,IAAI,CAACC,SACH,MAAM,IAAIE,SAAS,CAAC,QAAQ,EAAEF,QAAQ,UAAU,CAAC,EAAE;wBACjDG,QAAQ;oBACV;oBACF,IAAI,IAAI,CAAC3B,aAAa,IAAIwB,QAAQI,QAAQ,CAACC,UAAU,KAAK,OACxD,MAAM,IAAIH,SAAS,CAAC,QAAQ,EAAEF,QAAQ,cAAc,CAAC,EAAE;wBACrDG,QAAQ;oBACV;gBACJ;gBAEA,MAAMG,OAAmB;oBACvBC,IAAIrD;oBACJsD,QAAQvC,UAAUuB,aAAa,IAAI,CAACd,WAAW,CAAC8B,MAAM;oBACtDf;oBACAgB,SAAS;wBACPC,IAAI,IAAIC;wBACRC,MAAM,IAAID;wBACVE,UAAU;oBACZ;oBACAC,kBAAkB,IAAIH;oBACtBI,eAAe,IAAIJ;oBACnBf;oBACAU,MAAM;wBACJT,OAAOL,YAAYK,KAAK;wBACxBmB,SAASxB,YAAYwB,OAAO;wBAC5BC,sBAAsBC,OAAOC,IAAI,CAC/BpC,IAAIqC,6BAA6B,IACjCC,QAAQ;wBACVC,eAAeJ,OAAOC,IAAI,CACxBpC,IAAIwC,sBAAsB,IAC1BF,QAAQ;oBACZ;oBACAG,cAAc;oBACdjC,SAAS,CAAC;gBACZ;gBAEAR,IAAIM,OAAO,CACTiB,MACAhB,IAAImC,SAAS,CAAC,sBACdnC,IAAImC,SAAS,CAAC,2BACdnC,IAAImC,SAAS,CAAC,6BACdlC;YAEJ;YACAmC,MAAM,CAACxC;gBACL,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAC3BrB,KAAKf,OAAO,CAACqC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC3C;gBAClDoB,KAAKf,OAAO,CAACuC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC7C;gBAClDoB,KAAKb,SAAS,CAACuC,OAAO,CAACjE,gBAAgBuC,KAAKA,IAAI;gBAChD,IAAI,CAAC5B,WAAW,CAACuD,WAAW,CAACC,GAAG,CAAC;oBAC/B3B,IAAID,KAAKC,EAAE;oBACXX,UAAUU,KAAKV,QAAQ;oBACvBuC,MAAM,IAAI,CAAC3D,aAAa;oBACxBuC,eAAeT,KAAKS,aAAa;oBACjCqB,WAAW,CAACpC,SAASqC,OAAOC,UAC1BlE,YAAYc,IAAIzB,YAAY8E,KAAK,EAAE;4BAACvC;4BAASqC;4BAAOC;yBAAQ;gBAChE;YACF;YACAE,SAAS,OAAOtD,IAAuBmD;gBACrC,MAAMI,SAASJ;gBACf,MAAMK,cAAc/E,aAAa8E,QAAQ;gBACzC,IAAIC,eAAe,IAAI,KAAK,OAAO;oBACjCxD,GAAGD,GAAG,CAAC,MAAM;gBACf,OAAO;oBACL,IAAI;wBACF,MAAM,IAAI,CAACyD,YAAY,CACrBxD,IACAuD,OAAOE,KAAK,CAACC,WAAWC,iBAAiB;oBAE7C,EAAE,OAAOC,OAAY;wBACnB,IAAI,CAACC,QAAQ,CAACD,OAAO;oBACvB;gBACF;YACF;YACAE,OAAO,CAAC9D;gBACN,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAC3BrB,KAAKkB,YAAY,EAAEyB;gBACnB3C,KAAKkB,YAAY,GAAG;YACtB;YACA0B,OAAO,CAAChE;gBACN,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAE3B,IAAI,CAACjD,WAAW,CAACuD,WAAW,CAACkB,MAAM,CAAC7C,KAAKC,EAAE;gBAC3C,MAAMuC,QAAQ,IAAIM,MAAM;gBAExB,KAAK,MAAMC,MAAM/C,KAAKQ,gBAAgB,CAACwC,MAAM,GAAI;oBAC/CD,GAAGE,KAAK,CAACT;gBACX;gBACAxC,KAAKQ,gBAAgB,CAAC0C,KAAK;gBAE3B,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACG,IAAI,CAAC0C,MAAM,GAAI;oBAC/CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACG,IAAI,CAAC4C,KAAK;gBAEvB,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACC,EAAE,CAAC4C,MAAM,GAAI;oBAC7CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACC,EAAE,CAAC8C,KAAK;gBAErB,KAAK,MAAMG,gBAAgBrD,KAAKS,aAAa,CAACuC,MAAM,GAAI;oBACtDK,aAAaD,OAAO;gBACtB;gBACApD,KAAKS,aAAa,CAACyC,KAAK;gBAExB,IAAI,CAACI,uBAAuB,CAACtD,KAAKb,SAAS;YAC7C;QACF;IACJ;IAEA,MAAMoE,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACb,SAASc;YACjC,MAAMC,WAAW,IAAI,CAACrF,OAAO,CAACqF,QAAQ,IAAI;YAC1C,IAAI,CAACzF,MAAM,CAAC0F,MAAM,CAACD,UAAU,IAAI,CAACrF,OAAO,CAACuF,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAACC,MAAM,CAACC,IAAI,CACd,2BACAL,UACA,IAAI,CAACrF,OAAO,CAACuF,IAAI;oBAEnBjB;gBACF,OAAO;oBACLc,OAAO,IAAIX,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMkB,OAAO;QACX,IAAI,CAAC/F,MAAM,CAAC2E,KAAK;IACnB;IAEA,IAAcqB,MAAM;QAClB,OAAO,IAAI,CAAC7F,WAAW,CAAC6F,GAAG;IAC7B;IAEA,IAAcH,SAAS;QACrB,OAAO,IAAI,CAAC1F,WAAW,CAAC0F,MAAM;IAChC;IAEA,MAAgBrB,SACdyB,KAAY,EACZhC,UAAU,wCAAwC,EAClD;QACA,IAAI,CAAC4B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAMZ,SAAS;YAAEgC;QAAM;IAC/C;IAEUZ,wBAAwBnE,SAAoB,EAAE;QACtDA,UAAUgF,OAAO;IACnB;IAEA,MAAgBC,UAAU/F,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAAC4F,GAAG,CAACI,IAAI,CAAC;YACzB,GAAGhG,OAAO;YACViG,WAAW,IAAI,CAACpG,aAAa;QAC/B;IACF;IAEUuD,uBAAuB7C,EAAqB,EAAoB;QACxE,MAAMoB,OAAOpB,GAAGyC,WAAW;QAC3B,OAAO;YACLkD,WAAW,CAACC;gBACV,MAAMvE,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMkE,aAAa,IAAI1H,iBAAiBkD,IAAIuE;gBAC5CC,WAAWC,KAAK;gBAChBD,WAAWE,EAAE,CAAC,SAASC,QAAQC,GAAG;gBAClCJ,WAAWK,IAAI,CAAC,SAAS,CAACC;oBACxBH,QAAQI,GAAG,CAAC;wBAAED;oBAAI;oBAClB,IAAIN,WAAWQ,OAAO,EAAE/C,YAAY,qBAClCrE,KAAKe,IAAIzB,YAAY+H,eAAe,EAAE5H,aAAa2C,IAAI;gBAC3D;gBACAwE,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChB7G,KACEe,IACAzB,YAAYiI,cAAc,EAC1B9H,aAAa2C,IAAI,WACjBW,OAAOC,IAAI,CAACsE,OAAOhD,MAAM;gBAE7B;gBACAnC,KAAKG,OAAO,CAACG,IAAI,CAAC+E,GAAG,CAACpF,IAAIwE;gBAC1B,OAAO;oBAAExE;oBAAIqF,UAAUd,KAAKc,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAACtF;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;YAC/B;QACF;IACF;IAEUsB,uBAAuB3C,EAAqB,EAAE;QACtD,MAAMoB,OAAOpB,GAAGyC,WAAW;QAC3B,OAAO;YACLkD,WAAUiB,MAAmB,EAAEvF,EAAU,EAAEqF,QAAyB;gBAClE,MAAMG,WAAW,IAAIzI,eAAesI,UAAU;oBAC5CI,MAAM,CAACC;wBACL9H,KACEe,IACAzB,YAAYyI,YAAY,EACxBtI,aAAa2C,IAAI,WACjB3C,aAAaqI,MAAM;oBAEvB;gBACF;gBAEA3F,KAAKG,OAAO,CAACC,EAAE,CAACiF,GAAG,CAACpF,IAAIwF;gBACxBvI,QAAQsI,QAAQ;oBACdC,SAASrC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA2C,SAASX,IAAI,CAAC,SAAS,CAACtC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBrE,KAAKe,IAAIzB,YAAY0I,aAAa,EAAEvI,aAAa2C,IAAI;gBACzD;gBACAwF,SAASX,IAAI,CAAC,SAAS;oBACrB,IAAIW,SAASR,OAAO,EAAE/C,YAAY,qBAChCrE,KAAKe,IAAIzB,YAAY2I,WAAW,EAAExI,aAAa2C,IAAI;gBACvD;gBAEA,OAAOwF;YACT;YACAF,WAAUtF,EAAE;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;YAC/B;QACF;IACF;IAEA,MAAgB,CAAC9C,YAAY4I,GAAG,CAAC,CAC/BnH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAE3B,MAAM2E,aAAa,IAAI,CAAC5H,WAAW,CAACuD,WAAW,CAACnD,GAAG,CAACwB,KAAKC,EAAE;QAC3D,IAAI,CAAC+F,YAAY,OAAO,KAAKpH,GAAGD,GAAG,CAAC,MAAM;QAE1C,MAAMoE,KAAK,IAAIkD;QACf,MAAMC,MAAMlG,KAAKE,MAAM,CAACiG,OAAO,CAACC,SAAS,CAACjE,QAAQ;YAChD,GAAGnC,KAAKf,OAAO,CAACqC,MAAM;YACtBiD,WAAWvE,KAAKf,OAAO,CAACqC,MAAM,CAACiD,SAAS,CAAC8B,IAAI,CAAC,MAAMtD,GAAGyC,MAAM;QAC/D;QAEAxF,KAAKQ,gBAAgB,CAAC6E,GAAG,CAACa,IAAII,MAAM,EAAEvD;QAEtC,MAAM5D,YAAYa,KAAKb,SAAS,CAACC,WAAW,CAACtC,MAAMyJ,IAAI;QAEvD,IAAI;YACF,MAAM,EAAE7G,OAAO,EAAE8G,SAAS,EAAE,GAAG,IAAI,CAACvC,GAAG,CAACwC,IAAI,CAC1CP,IAAIxG,OAAO,EACXwG,IAAIM,SAAS,EACb,IAAI,CAACtI,aAAa;YAGpB,MAAMwI,WAAW,MAAM,IAAI,CAACtC,SAAS,CAAC;gBACpC4B;gBACAtG;gBACA8G;gBACArH;gBACAqG,QAAQzC,GAAGyC,MAAM;gBACjBxD,SAASkE,IAAIlE,OAAO;YACtB;YAEA,IAAI0E,oBAAoBzJ,sBAAsB;gBAC5Cc,eAAea,IAAIzB,YAAYwJ,eAAe,EAAE;oBAC9CL,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS;wBAAC0E,SAASrD,YAAY,CAACuD,GAAG;wBAAEF,SAAS1E,OAAO;qBAAC;gBACxD;gBAEA0E,SAASrD,YAAY,CAACsB,EAAE,CAAC,SAAS,CAAC5C,OAAOC;oBACxClE,YAAYc,IAAIzB,YAAY0J,uBAAuB,EAAE;wBACnDH,SAASrD,YAAY,CAACuD,GAAG;wBACzB7E;wBACAC;qBACD;gBACH;gBACA0E,SAASrD,YAAY,CAACyB,IAAI,CAAC,OAAO;oBAChChH,YAAYc,IAAIzB,YAAY2J,iBAAiB,EAAE;wBAC7CJ,SAASrD,YAAY,CAACuD,GAAG;qBAC1B;gBACH;YACF,OAAO;gBACL7I,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS0E;gBACX;YACF;QACF,EAAE,OAAOlE,OAAO;YACd,IAAIA,iBAAiB3F,UAAU;gBAC7B,IAAI,CAACiH,MAAM,CAACiD,KAAK,CAAC,IAAIjE,MAAM,aAAa;oBAAEoB,OAAO1B;gBAAM;gBACxDzE,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D;oBACAR,SAAS;gBACX;YACF,OAAO;gBACL,IAAI,CAAC8B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAM,oBAAoB;oBAAEoB,OAAO1B;gBAAM;gBAC/DzE,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO9E;oBACPsE,SAAS;gBACX;YACF;QACF,SAAU;YACRhC,KAAKQ,gBAAgB,CAACwG,MAAM,CAACd,IAAII,MAAM;YACvC,IAAI,CAAChD,uBAAuB,CAACnE;QAC/B;IACF;IAEA,MAAM,CAAChC,YAAY8J,YAAY,CAAC,CAACrI,EAAqB,EAAEuD,MAAmB,EAAE;QAC3E,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO+D,IAAI,CAACtG,OAAOC,IAAI,CAACsB,OAAOE,KAAK,CAAC8E,YAAY5E,iBAAiB;IACpE;IAEA,MAAM,CAACpF,YAAY2I,WAAW,CAAC,CAAClH,EAAqB,EAAEuD,MAAmB,EAAE;QAC1E,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO+D,IAAI,CAAC;QACZlH,KAAKG,OAAO,CAACC,EAAE,CAAC4G,MAAM,CAAC/G;IACzB;IAEA,MAAM,CAAC9C,YAAY0I,aAAa,CAAC,CAC/BjH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACC,EAAE,CAAC4G,MAAM,CAAC/G;IACzB;IAEA,MAAM,CAAC9C,YAAYiK,cAAc,CAAC,CAChCxI,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjC,MAAMqB,KAAKkB,YAAY,EAAEmG;QACzB,IAAIlE,OAAOmE,aAAa,EACtBzJ,KAAKe,IAAIzB,YAAYoK,aAAa,EAAEjK,aAAa2C,IAAI;aAClDkD,OAAOqE,MAAM;IACpB;IAEA,MAAM,CAACrK,YAAYoK,aAAa,CAAC,CAC/B3I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCqB,KAAKG,OAAO,CAACG,IAAI,CAAC0G,MAAM,CAAC/G;IAC3B;IAEA,MAAM,CAAC9C,YAAY+H,eAAe,CAAC,CACjCtG,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACG,IAAI,CAAC0G,MAAM,CAAC/G;IAC3B;IAEA,MAAM,CAAC9C,YAAYsK,iBAAiB,CAAC,CACnC7I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAM,CAACuF,IAAI,GAAG5G,KAAKE,MAAM,CAACiG,OAAO,CAAC7E,MAAM,CAACa;QACzC,MAAMkB,eAAerD,KAAKS,aAAa,CAACjC,GAAG,CAACoI;QAC5C,IAAI,CAACvD,cAAc,OAAO,KAAKzE,GAAGD,GAAG;QACrC0E,aAAaD,OAAO;QACpBpD,KAAKS,aAAa,CAACuG,MAAM,CAACJ;IAC5B;IAEA,MAAM,CAACzJ,YAAYuK,QAAQ,CAAC,CAAC9I,EAAqB,EAAEuD,MAAmB,EAAE;QACvE,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMiF,SAASjJ,aAAa8E,QAAQ;QACpC,MAAMY,KAAK/C,KAAKQ,gBAAgB,CAAChC,GAAG,CAAC8H;QACrC,IAAIvD,IAAIA,GAAGE,KAAK,CAAC,IAAIH,MAAM;IAC7B;AACF"}
+{"version":3,"sources":["../../../lib/server.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto'\nimport {\n  type AnyProcedure,\n  ApiError,\n  type ApplicationContext,\n  type Connection,\n  type Container,\n  Scope,\n  ServerDownStream,\n  ServerUpStream,\n  type Service,\n  SubscriptionResponse,\n  onAbort,\n} from '@nmtjs/application'\nimport {\n  type ApiBlobMetadata,\n  type EncodeRpcContext,\n  MessageType,\n  TransportType,\n  decodeNumber,\n  encodeNumber,\n} from '@nmtjs/common'\nimport {\n  App,\n  type HttpRequest,\n  type HttpResponse,\n  SSLApp,\n  type TemplatedApp,\n} from 'uWebSockets.js'\n\nimport { connectionData } from './injectables.ts'\nimport type {\n  WsTransportOptions,\n  WsTransportSocket,\n  WsUserData,\n} from './types.ts'\nimport {\n  InternalError,\n  getFormat,\n  getRequestData,\n  send,\n  sendMessage,\n  sendRpcMessage,\n} from './utils.ts'\n\nexport class WsTransportServer {\n  protected server!: TemplatedApp\n  protected readonly transportType = TransportType.WS\n\n  constructor(\n    protected readonly application: ApplicationContext,\n    protected readonly options: WsTransportOptions,\n  ) {\n    this.server = this.options.tls ? SSLApp(options.tls!) : App()\n\n    this.server\n      .options('/*', (res, req) => {\n        this.applyCors(res, req)\n        res.writeStatus('200 OK')\n        res.endWithoutBody()\n      })\n      .get('/healthy', (res, req) => {\n        this.applyCors(res, req)\n        res.writeHeader('Content-Type', 'text/plain')\n        res.end('OK')\n      })\n      .ws<WsUserData>('/api', {\n        sendPingsAutomatically: true,\n        maxPayloadLength: this.options.maxPayloadLength,\n        upgrade: (res, req, context) => {\n          this.applyCors(res, req)\n          const requestData = getRequestData(req)\n          const container = this.application.container.createScope(\n            Scope.Connection,\n          )\n          const services = requestData.query.getAll('services')\n\n          for (const serviceName of services) {\n            const service = this.application.registry.services.get(serviceName)\n            if (!service)\n              throw new Response(`Service ${service} not found`, {\n                status: 400,\n              })\n            if (this.transportType in service.contract.transports === false)\n              throw new Response(`Service ${service} not supported`, {\n                status: 400,\n              })\n          }\n\n          const data: WsUserData = {\n            id: randomUUID(),\n            format: getFormat(requestData, this.application.format),\n            container,\n            streams: {\n              up: new Map(),\n              down: new Map(),\n              streamId: 0,\n            },\n            abortControllers: new Map(),\n            subscriptions: new Map(),\n            services,\n            data: {\n              query: requestData.query,\n              headers: requestData.headers,\n              proxiedRemoteAddress: Buffer.from(\n                res.getProxiedRemoteAddressAsText(),\n              ).toString(),\n              remoteAddress: Buffer.from(\n                res.getRemoteAddressAsText(),\n              ).toString(),\n            },\n            backpressure: null,\n            context: {} as any,\n          }\n\n          res.upgrade(\n            data,\n            req.getHeader('sec-websocket-key'),\n            req.getHeader('sec-websocket-protocol'),\n            req.getHeader('sec-websocket-extensions'),\n            context,\n          )\n        },\n        open: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.context.decode = this.createDecodeRpcContext(ws)\n          data.context.encode = this.createEncodeRpcContext(ws)\n          data.container.provide(connectionData, data.data)\n          this.application.connections.add({\n            id: data.id,\n            services: data.services,\n            type: this.transportType,\n            subscriptions: data.subscriptions,\n            sendEvent: (service, event, payload) =>\n              sendMessage(ws, MessageType.Event, [service, event, payload]),\n          })\n        },\n        message: async (ws: WsTransportSocket, event) => {\n          const buffer = event as unknown as ArrayBuffer\n          const messageType = decodeNumber(buffer, 'Uint8')\n          if (messageType in this === false) {\n            ws.end(1011, 'Unknown message type')\n          } else {\n            try {\n              await this[messageType](\n                ws,\n                buffer.slice(Uint8Array.BYTES_PER_ELEMENT),\n              )\n            } catch (error: any) {\n              this.logError(error, 'Error while processing message')\n            }\n          }\n        },\n        drain: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.backpressure?.resolve()\n          data.backpressure = null\n        },\n        close: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n\n          this.application.connections.remove(data.id)\n          const error = new Error('Connection closed')\n\n          for (const ac of data.abortControllers.values()) {\n            ac.abort(error)\n          }\n          data.abortControllers.clear()\n\n          for (const stream of data.streams.down.values()) {\n            stream.destroy(error)\n          }\n          data.streams.down.clear()\n\n          for (const stream of data.streams.up.values()) {\n            stream.destroy(error)\n          }\n          data.streams.up.clear()\n\n          for (const subscription of data.subscriptions.values()) {\n            subscription.destroy()\n          }\n          data.subscriptions.clear()\n\n          this.handleContainerDisposal(data.container)\n        },\n      })\n  }\n\n  async start() {\n    return new Promise<void>((resolve, reject) => {\n      const hostname = this.options.hostname ?? '127.0.0.1'\n      this.server.listen(hostname, this.options.port!, (socket) => {\n        if (socket) {\n          this.logger.info(\n            'Server started on %s:%s',\n            hostname,\n            this.options.port!,\n          )\n          resolve()\n        } else {\n          reject(new Error('Failed to start server'))\n        }\n      })\n    })\n  }\n\n  async stop() {\n    this.server.close()\n  }\n\n  protected get api() {\n    return this.application.api\n  }\n\n  protected get logger() {\n    return this.application.logger\n  }\n\n  protected async logError(\n    cause: Error,\n    message = 'Unknown error while processing request',\n  ) {\n    this.logger.error(new Error(message, { cause }))\n  }\n\n  protected applyCors(res: HttpResponse, req: HttpRequest) {\n    // TODO: this should be configurable\n    const origin = req.getHeader('origin')\n    if (!origin) return\n    res.headers.set('Access-Control-Allow-Origin', origin)\n    res.headers.set('Access-Control-Allow-Headers', 'Content-Type')\n    res.headers.set('Access-Control-Allow-Methods', 'GET, POST')\n    res.headers.set('Access-Control-Allow-Credentials', 'true')\n  }\n\n  protected handleContainerDisposal(container: Container) {\n    container.dispose()\n  }\n\n  protected async handleRPC(options: {\n    connection: Connection\n    service: Service\n    procedure: AnyProcedure\n    container: Container\n    signal: AbortSignal\n    payload: any\n  }) {\n    return await this.api.call({\n      ...options,\n      transport: this.transportType,\n    })\n  }\n\n  protected createEncodeRpcContext(ws: WsTransportSocket): EncodeRpcContext {\n    const data = ws.getUserData()\n    return {\n      addStream: (blob) => {\n        const id = ++data.streams.streamId\n        const downstream = new ServerDownStream(id, blob)\n        downstream.pause()\n        downstream.on('error', console.dir)\n        downstream.once('error', (err) => {\n          console.log({ err })\n          if (downstream.errored?.message !== 'Aborted by client')\n            send(ws, MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        downstream.on('data', (chunk) => {\n          downstream.pause()\n          send(\n            ws,\n            MessageType.DownStreamPush,\n            encodeNumber(id, 'Uint32'),\n            Buffer.from(chunk).buffer as ArrayBuffer,\n          )\n        })\n        data.streams.down.set(id, downstream)\n        return { id, metadata: blob.metadata }\n      },\n      getStream: (id) => {\n        return data.streams.down.get(id)!\n      },\n    }\n  }\n\n  protected createDecodeRpcContext(ws: WsTransportSocket) {\n    const data = ws.getUserData()\n    return {\n      addStream(signal: AbortSignal, id: number, metadata: ApiBlobMetadata) {\n        const upstream = new ServerUpStream(metadata, {\n          read: (size) => {\n            send(\n              ws,\n              MessageType.UpStreamPull,\n              encodeNumber(id, 'Uint32'),\n              encodeNumber(size, 'Uint32'),\n            )\n          },\n        })\n\n        data.streams.up.set(id, upstream)\n        onAbort(signal, () => {\n          upstream.destroy(new Error('Call aborted by client'))\n        })\n        upstream.once('error', (error) => {\n          if (error.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        upstream.once('close', () => {\n          if (upstream.errored?.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))\n        })\n\n        return upstream\n      },\n      getStream(id) {\n        return data.streams.down.get(id)\n      },\n    }\n  }\n\n  protected async [MessageType.Rpc](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n\n    const connection = this.application.connections.get(data.id)\n    if (!connection) return void ws.end(1011, 'Unknown connection')\n\n    const ac = new AbortController()\n    const rpc = data.format.decoder.decodeRpc(buffer, {\n      ...data.context.decode,\n      addStream: data.context.decode.addStream.bind(null, ac.signal),\n    })\n\n    data.abortControllers.set(rpc.callId, ac)\n\n    const container = data.container.createScope(Scope.Call)\n\n    try {\n      const { service, procedure } = this.api.find(\n        rpc.service,\n        rpc.procedure,\n        this.transportType,\n      )\n\n      const response = await this.handleRPC({\n        connection,\n        service,\n        procedure,\n        container,\n        signal: ac.signal,\n        payload: rpc.payload,\n      })\n\n      if (response instanceof SubscriptionResponse) {\n        sendRpcMessage(ws, MessageType.RpcSubscription, {\n          callId: rpc.callId,\n          error: null,\n          payload: [response.subscription.key, response.payload],\n        })\n\n        response.subscription.on('event', (event, payload) => {\n          sendMessage(ws, MessageType.ServerSubscriptionEvent, [\n            response.subscription.key,\n            event,\n            payload,\n          ])\n        })\n        response.subscription.once('end', () => {\n          sendMessage(ws, MessageType.ServerUnsubscribe, [\n            response.subscription.key,\n          ])\n        })\n      } else {\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: null,\n          payload: response,\n        })\n      }\n    } catch (error) {\n      if (error instanceof ApiError) {\n        this.logger.debug(new Error('Api error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error,\n          payload: null,\n        })\n      } else {\n        this.logger.error(new Error('Unexpected error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: InternalError(),\n          payload: null,\n        })\n      }\n    } finally {\n      data.abortControllers.delete(rpc.callId)\n      this.handleContainerDisposal(container)\n    }\n  }\n\n  async [MessageType.UpStreamPush](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(Buffer.from(buffer.slice(Uint32Array.BYTES_PER_ELEMENT)))\n  }\n\n  async [MessageType.UpStreamEnd](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(null)\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.UpStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.DownStreamPull](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    await data.backpressure?.promise\n    if (stream.readableEnded)\n      send(ws, MessageType.DownStreamEnd, encodeNumber(id, 'Uint32'))\n    else stream.resume()\n  }\n\n  async [MessageType.DownStreamEnd](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.DownStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.ClientUnsubscribe](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const [key] = data.format.decoder.decode(buffer)\n    const subscription = data.subscriptions.get(key)\n    if (!subscription) return void ws.end()\n    subscription.destroy()\n    data.subscriptions.delete(key)\n  }\n\n  async [MessageType.RpcAbort](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const callId = decodeNumber(buffer, 'Uint32')\n    const ac = data.abortControllers.get(callId)\n    if (ac) ac.abort(new Error('Aborted by client'))\n  }\n}\n"],"names":["randomUUID","ApiError","Scope","ServerDownStream","ServerUpStream","SubscriptionResponse","onAbort","MessageType","TransportType","decodeNumber","encodeNumber","App","SSLApp","connectionData","InternalError","getFormat","getRequestData","send","sendMessage","sendRpcMessage","WsTransportServer","server","transportType","constructor","application","options","WS","tls","res","req","applyCors","writeStatus","endWithoutBody","get","writeHeader","end","ws","sendPingsAutomatically","maxPayloadLength","upgrade","context","requestData","container","createScope","Connection","services","query","getAll","serviceName","service","registry","Response","status","contract","transports","data","id","format","streams","up","Map","down","streamId","abortControllers","subscriptions","headers","proxiedRemoteAddress","Buffer","from","getProxiedRemoteAddressAsText","toString","remoteAddress","getRemoteAddressAsText","backpressure","getHeader","open","getUserData","decode","createDecodeRpcContext","encode","createEncodeRpcContext","provide","connections","add","type","sendEvent","event","payload","Event","message","buffer","messageType","slice","Uint8Array","BYTES_PER_ELEMENT","error","logError","drain","resolve","close","remove","Error","ac","values","abort","clear","stream","destroy","subscription","handleContainerDisposal","start","Promise","reject","hostname","listen","port","socket","logger","info","stop","api","cause","origin","set","dispose","handleRPC","call","transport","addStream","blob","downstream","pause","on","console","dir","once","err","log","errored","DownStreamAbort","chunk","DownStreamPush","metadata","getStream","signal","upstream","read","size","UpStreamPull","UpStreamAbort","UpStreamEnd","Rpc","connection","AbortController","rpc","decoder","decodeRpc","bind","callId","Call","procedure","find","response","RpcSubscription","key","ServerSubscriptionEvent","ServerUnsubscribe","debug","delete","UpStreamPush","push","Uint32Array","DownStreamPull","promise","readableEnded","DownStreamEnd","resume","ClientUnsubscribe","RpcAbort"],"mappings":"AAAA,SAASA,UAAU,QAAQ,cAAa;AACxC,SAEEC,QAAQ,EAIRC,KAAK,EACLC,gBAAgB,EAChBC,cAAc,EAEdC,oBAAoB,EACpBC,OAAO,QACF,qBAAoB;AAC3B,SAGEC,WAAW,EACXC,aAAa,EACbC,YAAY,EACZC,YAAY,QACP,gBAAe;AACtB,SACEC,GAAG,EAGHC,MAAM,QAED,iBAAgB;AAEvB,SAASC,cAAc,QAAQ,mBAAkB;AAMjD,SACEC,aAAa,EACbC,SAAS,EACTC,cAAc,EACdC,IAAI,EACJC,WAAW,EACXC,cAAc,QACT,aAAY;AAEnB,OAAO,MAAMC;;;IACDC,OAAqB;IACZC,cAAgC;IAEnDC,YACE,AAAmBC,WAA+B,EAClD,AAAmBC,OAA2B,CAC9C;aAFmBD,cAAAA;aACAC,UAAAA;aAJFH,gBAAgBd,cAAckB,EAAE;QAMjD,IAAI,CAACL,MAAM,GAAG,IAAI,CAACI,OAAO,CAACE,GAAG,GAAGf,OAAOa,QAAQE,GAAG,IAAKhB;QAExD,IAAI,CAACU,MAAM,CACRI,OAAO,CAAC,MAAM,CAACG,KAAKC;YACnB,IAAI,CAACC,SAAS,CAACF,KAAKC;YACpBD,IAAIG,WAAW,CAAC;YAChBH,IAAII,cAAc;QACpB,GACCC,GAAG,CAAC,YAAY,CAACL,KAAKC;YACrB,IAAI,CAACC,SAAS,CAACF,KAAKC;YACpBD,IAAIM,WAAW,CAAC,gBAAgB;YAChCN,IAAIO,GAAG,CAAC;QACV,GACCC,EAAE,CAAa,QAAQ;YACtBC,wBAAwB;YACxBC,kBAAkB,IAAI,CAACb,OAAO,CAACa,gBAAgB;YAC/CC,SAAS,CAACX,KAAKC,KAAKW;gBAClB,IAAI,CAACV,SAAS,CAACF,KAAKC;gBACpB,MAAMY,cAAczB,eAAea;gBACnC,MAAMa,YAAY,IAAI,CAAClB,WAAW,CAACkB,SAAS,CAACC,WAAW,CACtDzC,MAAM0C,UAAU;gBAElB,MAAMC,WAAWJ,YAAYK,KAAK,CAACC,MAAM,CAAC;gBAE1C,KAAK,MAAMC,eAAeH,SAAU;oBAClC,MAAMI,UAAU,IAAI,CAACzB,WAAW,CAAC0B,QAAQ,CAACL,QAAQ,CAACZ,GAAG,CAACe;oBACvD,IAAI,CAACC,SACH,MAAM,IAAIE,SAAS,CAAC,QAAQ,EAAEF,QAAQ,UAAU,CAAC,EAAE;wBACjDG,QAAQ;oBACV;oBACF,IAAI,IAAI,CAAC9B,aAAa,IAAI2B,QAAQI,QAAQ,CAACC,UAAU,KAAK,OACxD,MAAM,IAAIH,SAAS,CAAC,QAAQ,EAAEF,QAAQ,cAAc,CAAC,EAAE;wBACrDG,QAAQ;oBACV;gBACJ;gBAEA,MAAMG,OAAmB;oBACvBC,IAAIxD;oBACJyD,QAAQ1C,UAAU0B,aAAa,IAAI,CAACjB,WAAW,CAACiC,MAAM;oBACtDf;oBACAgB,SAAS;wBACPC,IAAI,IAAIC;wBACRC,MAAM,IAAID;wBACVE,UAAU;oBACZ;oBACAC,kBAAkB,IAAIH;oBACtBI,eAAe,IAAIJ;oBACnBf;oBACAU,MAAM;wBACJT,OAAOL,YAAYK,KAAK;wBACxBmB,SAASxB,YAAYwB,OAAO;wBAC5BC,sBAAsBC,OAAOC,IAAI,CAC/BxC,IAAIyC,6BAA6B,IACjCC,QAAQ;wBACVC,eAAeJ,OAAOC,IAAI,CACxBxC,IAAI4C,sBAAsB,IAC1BF,QAAQ;oBACZ;oBACAG,cAAc;oBACdjC,SAAS,CAAC;gBACZ;gBAEAZ,IAAIW,OAAO,CACTgB,MACA1B,IAAI6C,SAAS,CAAC,sBACd7C,IAAI6C,SAAS,CAAC,2BACd7C,IAAI6C,SAAS,CAAC,6BACdlC;YAEJ;YACAmC,MAAM,CAACvC;gBACL,MAAMmB,OAAOnB,GAAGwC,WAAW;gBAC3BrB,KAAKf,OAAO,CAACqC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC1C;gBAClDmB,KAAKf,OAAO,CAACuC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC5C;gBAClDmB,KAAKb,SAAS,CAACuC,OAAO,CAACpE,gBAAgB0C,KAAKA,IAAI;gBAChD,IAAI,CAAC/B,WAAW,CAAC0D,WAAW,CAACC,GAAG,CAAC;oBAC/B3B,IAAID,KAAKC,EAAE;oBACXX,UAAUU,KAAKV,QAAQ;oBACvBuC,MAAM,IAAI,CAAC9D,aAAa;oBACxB0C,eAAeT,KAAKS,aAAa;oBACjCqB,WAAW,CAACpC,SAASqC,OAAOC,UAC1BrE,YAAYkB,IAAI7B,YAAYiF,KAAK,EAAE;4BAACvC;4BAASqC;4BAAOC;yBAAQ;gBAChE;YACF;YACAE,SAAS,OAAOrD,IAAuBkD;gBACrC,MAAMI,SAASJ;gBACf,MAAMK,cAAclF,aAAaiF,QAAQ;gBACzC,IAAIC,eAAe,IAAI,KAAK,OAAO;oBACjCvD,GAAGD,GAAG,CAAC,MAAM;gBACf,OAAO;oBACL,IAAI;wBACF,MAAM,IAAI,CAACwD,YAAY,CACrBvD,IACAsD,OAAOE,KAAK,CAACC,WAAWC,iBAAiB;oBAE7C,EAAE,OAAOC,OAAY;wBACnB,IAAI,CAACC,QAAQ,CAACD,OAAO;oBACvB;gBACF;YACF;YACAE,OAAO,CAAC7D;gBACN,MAAMmB,OAAOnB,GAAGwC,WAAW;gBAC3BrB,KAAKkB,YAAY,EAAEyB;gBACnB3C,KAAKkB,YAAY,GAAG;YACtB;YACA0B,OAAO,CAAC/D;gBACN,MAAMmB,OAAOnB,GAAGwC,WAAW;gBAE3B,IAAI,CAACpD,WAAW,CAAC0D,WAAW,CAACkB,MAAM,CAAC7C,KAAKC,EAAE;gBAC3C,MAAMuC,QAAQ,IAAIM,MAAM;gBAExB,KAAK,MAAMC,MAAM/C,KAAKQ,gBAAgB,CAACwC,MAAM,GAAI;oBAC/CD,GAAGE,KAAK,CAACT;gBACX;gBACAxC,KAAKQ,gBAAgB,CAAC0C,KAAK;gBAE3B,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACG,IAAI,CAAC0C,MAAM,GAAI;oBAC/CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACG,IAAI,CAAC4C,KAAK;gBAEvB,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACC,EAAE,CAAC4C,MAAM,GAAI;oBAC7CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACC,EAAE,CAAC8C,KAAK;gBAErB,KAAK,MAAMG,gBAAgBrD,KAAKS,aAAa,CAACuC,MAAM,GAAI;oBACtDK,aAAaD,OAAO;gBACtB;gBACApD,KAAKS,aAAa,CAACyC,KAAK;gBAExB,IAAI,CAACI,uBAAuB,CAACtD,KAAKb,SAAS;YAC7C;QACF;IACJ;IAEA,MAAMoE,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACb,SAASc;YACjC,MAAMC,WAAW,IAAI,CAACxF,OAAO,CAACwF,QAAQ,IAAI;YAC1C,IAAI,CAAC5F,MAAM,CAAC6F,MAAM,CAACD,UAAU,IAAI,CAACxF,OAAO,CAAC0F,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAACC,MAAM,CAACC,IAAI,CACd,2BACAL,UACA,IAAI,CAACxF,OAAO,CAAC0F,IAAI;oBAEnBjB;gBACF,OAAO;oBACLc,OAAO,IAAIX,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMkB,OAAO;QACX,IAAI,CAAClG,MAAM,CAAC8E,KAAK;IACnB;IAEA,IAAcqB,MAAM;QAClB,OAAO,IAAI,CAAChG,WAAW,CAACgG,GAAG;IAC7B;IAEA,IAAcH,SAAS;QACrB,OAAO,IAAI,CAAC7F,WAAW,CAAC6F,MAAM;IAChC;IAEA,MAAgBrB,SACdyB,KAAY,EACZhC,UAAU,wCAAwC,EAClD;QACA,IAAI,CAAC4B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAMZ,SAAS;YAAEgC;QAAM;IAC/C;IAEU3F,UAAUF,GAAiB,EAAEC,GAAgB,EAAE;QAEvD,MAAM6F,SAAS7F,IAAI6C,SAAS,CAAC;QAC7B,IAAI,CAACgD,QAAQ;QACb9F,IAAIqC,OAAO,CAAC0D,GAAG,CAAC,+BAA+BD;QAC/C9F,IAAIqC,OAAO,CAAC0D,GAAG,CAAC,gCAAgC;QAChD/F,IAAIqC,OAAO,CAAC0D,GAAG,CAAC,gCAAgC;QAChD/F,IAAIqC,OAAO,CAAC0D,GAAG,CAAC,oCAAoC;IACtD;IAEUd,wBAAwBnE,SAAoB,EAAE;QACtDA,UAAUkF,OAAO;IACnB;IAEA,MAAgBC,UAAUpG,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAAC+F,GAAG,CAACM,IAAI,CAAC;YACzB,GAAGrG,OAAO;YACVsG,WAAW,IAAI,CAACzG,aAAa;QAC/B;IACF;IAEU0D,uBAAuB5C,EAAqB,EAAoB;QACxE,MAAMmB,OAAOnB,GAAGwC,WAAW;QAC3B,OAAO;YACLoD,WAAW,CAACC;gBACV,MAAMzE,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMoE,aAAa,IAAI/H,iBAAiBqD,IAAIyE;gBAC5CC,WAAWC,KAAK;gBAChBD,WAAWE,EAAE,CAAC,SAASC,QAAQC,GAAG;gBAClCJ,WAAWK,IAAI,CAAC,SAAS,CAACC;oBACxBH,QAAQI,GAAG,CAAC;wBAAED;oBAAI;oBAClB,IAAIN,WAAWQ,OAAO,EAAEjD,YAAY,qBAClCxE,KAAKmB,IAAI7B,YAAYoI,eAAe,EAAEjI,aAAa8C,IAAI;gBAC3D;gBACA0E,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChBlH,KACEmB,IACA7B,YAAYsI,cAAc,EAC1BnI,aAAa8C,IAAI,WACjBW,OAAOC,IAAI,CAACwE,OAAOlD,MAAM;gBAE7B;gBACAnC,KAAKG,OAAO,CAACG,IAAI,CAAC8D,GAAG,CAACnE,IAAI0E;gBAC1B,OAAO;oBAAE1E;oBAAIsF,UAAUb,KAAKa,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAACvF;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC5B,GAAG,CAACuB;YAC/B;QACF;IACF;IAEUsB,uBAAuB1C,EAAqB,EAAE;QACtD,MAAMmB,OAAOnB,GAAGwC,WAAW;QAC3B,OAAO;YACLoD,WAAUgB,MAAmB,EAAExF,EAAU,EAAEsF,QAAyB;gBAClE,MAAMG,WAAW,IAAI7I,eAAe0I,UAAU;oBAC5CI,MAAM,CAACC;wBACLlI,KACEmB,IACA7B,YAAY6I,YAAY,EACxB1I,aAAa8C,IAAI,WACjB9C,aAAayI,MAAM;oBAEvB;gBACF;gBAEA5F,KAAKG,OAAO,CAACC,EAAE,CAACgE,GAAG,CAACnE,IAAIyF;gBACxB3I,QAAQ0I,QAAQ;oBACdC,SAAStC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA4C,SAASV,IAAI,CAAC,SAAS,CAACxC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBxE,KAAKmB,IAAI7B,YAAY8I,aAAa,EAAE3I,aAAa8C,IAAI;gBACzD;gBACAyF,SAASV,IAAI,CAAC,SAAS;oBACrB,IAAIU,SAASP,OAAO,EAAEjD,YAAY,qBAChCxE,KAAKmB,IAAI7B,YAAY+I,WAAW,EAAE5I,aAAa8C,IAAI;gBACvD;gBAEA,OAAOyF;YACT;YACAF,WAAUvF,EAAE;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC5B,GAAG,CAACuB;YAC/B;QACF;IACF;IAEA,MAAgB,CAACjD,YAAYgJ,GAAG,CAAC,CAC/BnH,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAE3B,MAAM4E,aAAa,IAAI,CAAChI,WAAW,CAAC0D,WAAW,CAACjD,GAAG,CAACsB,KAAKC,EAAE;QAC3D,IAAI,CAACgG,YAAY,OAAO,KAAKpH,GAAGD,GAAG,CAAC,MAAM;QAE1C,MAAMmE,KAAK,IAAImD;QACf,MAAMC,MAAMnG,KAAKE,MAAM,CAACkG,OAAO,CAACC,SAAS,CAAClE,QAAQ;YAChD,GAAGnC,KAAKf,OAAO,CAACqC,MAAM;YACtBmD,WAAWzE,KAAKf,OAAO,CAACqC,MAAM,CAACmD,SAAS,CAAC6B,IAAI,CAAC,MAAMvD,GAAG0C,MAAM;QAC/D;QAEAzF,KAAKQ,gBAAgB,CAAC4D,GAAG,CAAC+B,IAAII,MAAM,EAAExD;QAEtC,MAAM5D,YAAYa,KAAKb,SAAS,CAACC,WAAW,CAACzC,MAAM6J,IAAI;QAEvD,IAAI;YACF,MAAM,EAAE9G,OAAO,EAAE+G,SAAS,EAAE,GAAG,IAAI,CAACxC,GAAG,CAACyC,IAAI,CAC1CP,IAAIzG,OAAO,EACXyG,IAAIM,SAAS,EACb,IAAI,CAAC1I,aAAa;YAGpB,MAAM4I,WAAW,MAAM,IAAI,CAACrC,SAAS,CAAC;gBACpC2B;gBACAvG;gBACA+G;gBACAtH;gBACAsG,QAAQ1C,GAAG0C,MAAM;gBACjBzD,SAASmE,IAAInE,OAAO;YACtB;YAEA,IAAI2E,oBAAoB7J,sBAAsB;gBAC5Cc,eAAeiB,IAAI7B,YAAY4J,eAAe,EAAE;oBAC9CL,QAAQJ,IAAII,MAAM;oBAClB/D,OAAO;oBACPR,SAAS;wBAAC2E,SAAStD,YAAY,CAACwD,GAAG;wBAAEF,SAAS3E,OAAO;qBAAC;gBACxD;gBAEA2E,SAAStD,YAAY,CAACwB,EAAE,CAAC,SAAS,CAAC9C,OAAOC;oBACxCrE,YAAYkB,IAAI7B,YAAY8J,uBAAuB,EAAE;wBACnDH,SAAStD,YAAY,CAACwD,GAAG;wBACzB9E;wBACAC;qBACD;gBACH;gBACA2E,SAAStD,YAAY,CAAC2B,IAAI,CAAC,OAAO;oBAChCrH,YAAYkB,IAAI7B,YAAY+J,iBAAiB,EAAE;wBAC7CJ,SAAStD,YAAY,CAACwD,GAAG;qBAC1B;gBACH;YACF,OAAO;gBACLjJ,eAAeiB,IAAI7B,YAAYgJ,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB/D,OAAO;oBACPR,SAAS2E;gBACX;YACF;QACF,EAAE,OAAOnE,OAAO;YACd,IAAIA,iBAAiB9F,UAAU;gBAC7B,IAAI,CAACoH,MAAM,CAACkD,KAAK,CAAC,IAAIlE,MAAM,aAAa;oBAAEoB,OAAO1B;gBAAM;gBACxD5E,eAAeiB,IAAI7B,YAAYgJ,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB/D;oBACAR,SAAS;gBACX;YACF,OAAO;gBACL,IAAI,CAAC8B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAM,oBAAoB;oBAAEoB,OAAO1B;gBAAM;gBAC/D5E,eAAeiB,IAAI7B,YAAYgJ,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB/D,OAAOjF;oBACPyE,SAAS;gBACX;YACF;QACF,SAAU;YACRhC,KAAKQ,gBAAgB,CAACyG,MAAM,CAACd,IAAII,MAAM;YACvC,IAAI,CAACjD,uBAAuB,CAACnE;QAC/B;IACF;IAEA,MAAM,CAACnC,YAAYkK,YAAY,CAAC,CAACrI,EAAqB,EAAEsD,MAAmB,EAAE;QAC3E,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC1B,GAAG,CAACuB;QACnC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjCuE,OAAOgE,IAAI,CAACvG,OAAOC,IAAI,CAACsB,OAAOE,KAAK,CAAC+E,YAAY7E,iBAAiB;IACpE;IAEA,MAAM,CAACvF,YAAY+I,WAAW,CAAC,CAAClH,EAAqB,EAAEsD,MAAmB,EAAE;QAC1E,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC1B,GAAG,CAACuB;QACnC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjCuE,OAAOgE,IAAI,CAAC;QACZnH,KAAKG,OAAO,CAACC,EAAE,CAAC6G,MAAM,CAAChH;IACzB;IAEA,MAAM,CAACjD,YAAY8I,aAAa,CAAC,CAC/BjH,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC1B,GAAG,CAACuB;QACnC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjCuE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACC,EAAE,CAAC6G,MAAM,CAAChH;IACzB;IAEA,MAAM,CAACjD,YAAYqK,cAAc,CAAC,CAChCxI,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC5B,GAAG,CAACuB;QACrC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjC,MAAMoB,KAAKkB,YAAY,EAAEoG;QACzB,IAAInE,OAAOoE,aAAa,EACtB7J,KAAKmB,IAAI7B,YAAYwK,aAAa,EAAErK,aAAa8C,IAAI;aAClDkD,OAAOsE,MAAM;IACpB;IAEA,MAAM,CAACzK,YAAYwK,aAAa,CAAC,CAC/B3I,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC5B,GAAG,CAACuB;QACrC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjCoB,KAAKG,OAAO,CAACG,IAAI,CAAC2G,MAAM,CAAChH;IAC3B;IAEA,MAAM,CAACjD,YAAYoI,eAAe,CAAC,CACjCvG,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMpB,KAAK/C,aAAaiF,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC5B,GAAG,CAACuB;QACrC,IAAI,CAACkD,QAAQ,OAAOtE,GAAGD,GAAG,CAAC,MAAM;QACjCuE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACG,IAAI,CAAC2G,MAAM,CAAChH;IAC3B;IAEA,MAAM,CAACjD,YAAY0K,iBAAiB,CAAC,CACnC7I,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAM,CAACwF,IAAI,GAAG7G,KAAKE,MAAM,CAACkG,OAAO,CAAC9E,MAAM,CAACa;QACzC,MAAMkB,eAAerD,KAAKS,aAAa,CAAC/B,GAAG,CAACmI;QAC5C,IAAI,CAACxD,cAAc,OAAO,KAAKxE,GAAGD,GAAG;QACrCyE,aAAaD,OAAO;QACpBpD,KAAKS,aAAa,CAACwG,MAAM,CAACJ;IAC5B;IAEA,MAAM,CAAC7J,YAAY2K,QAAQ,CAAC,CAAC9I,EAAqB,EAAEsD,MAAmB,EAAE;QACvE,MAAMnC,OAAOnB,GAAGwC,WAAW;QAC3B,MAAMkF,SAASrJ,aAAaiF,QAAQ;QACpC,MAAMY,KAAK/C,KAAKQ,gBAAgB,CAAC9B,GAAG,CAAC6H;QACrC,IAAIxD,IAAIA,GAAGE,KAAK,CAAC,IAAIH,MAAM;IAC7B;AACF"}

package/index.ts

@@ -2,4 +2,4 @@ export * from './lib/server.ts'
 export * from './lib/transport.ts'
 export * from './lib/types.ts'
 export * from './lib/utils.ts'
-export * from './lib/providers.ts'
+export * from './lib/injectables.ts'

package/lib/injectables.ts

@@ -0,0 +1,11 @@
+import {
+  type LazyInjectable,
+  type Scope,
+  injectables,
+} from '@nmtjs/application'
+import type { WsConnectionData } from './types.ts'
+
+export const connectionData = injectables.connectionData as LazyInjectable<
+  WsConnectionData,
+  Scope.Connection
+>

package/lib/server.ts

@@ -20,8 +20,15 @@ import {
   decodeNumber,
   encodeNumber,
 } from '@nmtjs/common'
-import { App, SSLApp, type TemplatedApp } from 'uWebSockets.js'
-import { connectionData } from './providers.ts'
+import {
+  App,
+  type HttpRequest,
+  type HttpResponse,
+  SSLApp,
+  type TemplatedApp,
+} from 'uWebSockets.js'
+
+import { connectionData } from './injectables.ts'
 import type {
   WsTransportOptions,
   WsTransportSocket,
@@ -47,11 +54,13 @@ export class WsTransportServer {
     this.server = this.options.tls ? SSLApp(options.tls!) : App()
 
     this.server
-      .get('/healthy', (res) => {
-        // cors
-        res.writeHeader('Access-Control-Allow-Origin', '*')
-        res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')
-        res.writeHeader('Access-Control-Allow-Methods', 'GET')
+      .options('/*', (res, req) => {
+        this.applyCors(res, req)
+        res.writeStatus('200 OK')
+        res.endWithoutBody()
+      })
+      .get('/healthy', (res, req) => {
+        this.applyCors(res, req)
         res.writeHeader('Content-Type', 'text/plain')
         res.end('OK')
       })
@@ -59,6 +68,7 @@ export class WsTransportServer {
         sendPingsAutomatically: true,
         maxPayloadLength: this.options.maxPayloadLength,
         upgrade: (res, req, context) => {
+          this.applyCors(res, req)
           const requestData = getRequestData(req)
           const container = this.application.container.createScope(
             Scope.Connection,
@@ -214,6 +224,16 @@ export class WsTransportServer {
     this.logger.error(new Error(message, { cause }))
   }
 
+  protected applyCors(res: HttpResponse, req: HttpRequest) {
+    // TODO: this should be configurable
+    const origin = req.getHeader('origin')
+    if (!origin) return
+    res.headers.set('Access-Control-Allow-Origin', origin)
+    res.headers.set('Access-Control-Allow-Headers', 'Content-Type')
+    res.headers.set('Access-Control-Allow-Methods', 'GET, POST')
+    res.headers.set('Access-Control-Allow-Credentials', 'true')
+  }
+
   protected handleContainerDisposal(container: Container) {
     container.dispose()
   }

package/package.json

@@ -11,12 +11,13 @@
     "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.47.0"
   },
   "peerDependencies": {
-    "@nmtjs/application": "^0.0.1",
-    "@nmtjs/common": "^0.0.1"
+    "@nmtjs/application": "^0.2.1",
+    "@nmtjs/common": "^0.2.1"
   },
   "devDependencies": {
-    "@nmtjs/application": "^0.0.1",
-    "@nmtjs/common": "^0.0.1"
+    "@nmtjs/application": "^0.2.1",
+    "@nmtjs/common": "^0.2.1",
+    "@types/node": "^18.0.0"
   },
   "engines": {
     "node": ">=18.9.0"
@@ -29,7 +30,7 @@
     "LICENSE.md",
     "README.md"
   ],
-  "version": "0.0.1",
+  "version": "0.1.1",
   "scripts": {
     "build": "neemata-build ./index.ts './lib/**/*.ts'",
     "type-check": "tsc --noEmit"

package/dist/lib/providers.js

@@ -1,2 +0,0 @@
-import { providers } from '@nmtjs/application';
-export const connectionData = providers.connectionData.$withType();

package/dist/lib/providers.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../../lib/providers.ts"],"sourcesContent":["import { providers } from '@nmtjs/application'\nimport type { WsConnectionData } from './types.ts'\n\nexport const connectionData =\n  providers.connectionData.$withType<WsConnectionData>()\n"],"names":["providers","connectionData","$withType"],"mappings":"AAAA,SAASA,SAAS,QAAQ,qBAAoB;AAG9C,OAAO,MAAMC,iBACXD,UAAUC,cAAc,CAACC,SAAS,GAAoB"}

package/lib/providers.ts

@@ -1,5 +0,0 @@
-import { providers } from '@nmtjs/application'
-import type { WsConnectionData } from './types.ts'
-
-export const connectionData =
-  providers.connectionData.$withType<WsConnectionData>()