npm - @nmtjs/ws-transport - Versions diffs - 0.1.3 → 0.1.4 - Mend

@nmtjs/ws-transport 0.1.3 → 0.1.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (4) hide show

package/dist/lib/server.js CHANGED Viewed

@@ -32,12 +32,8 @@ export class WsTransportServer {
                 const services = requestData.query.getAll('services');
                 for (const serviceName of services){
                     const service = this.application.registry.services.get(serviceName);
-                    if (!service) throw new Response(`Service ${service} not found`, {
-                        status: 400
-                    });
-                    if (this.transportType in service.contract.transports === false) throw new Response(`Service ${service} not supported`, {
-                        status: 400
-                    });
+                    if (!service) return void res.writeStatus('400 Bad Request').end(`Service ${service} not found`);
+                    if (this.transportType in service.contract.transports === false) return void res.writeStatus('400 Bad Request').end(`Service ${service} not supported`);
                 }
                 const data = {
                     id: randomUUID(),
@@ -64,6 +60,7 @@ export class WsTransportServer {
             },
             open: (ws)=>{
                 const data = ws.getUserData();
+                this.logger.debug('Connection %s opened', data.id);
                 data.context.decode = this.createDecodeRpcContext(ws);
                 data.context.encode = this.createEncodeRpcContext(ws);
                 data.container.provide(connectionData, data.data);
@@ -97,8 +94,9 @@ export class WsTransportServer {
                 data.backpressure?.resolve();
                 data.backpressure = null;
             },
-            close: (ws)=>{
+            close: (ws, code, message)=>{
                 const data = ws.getUserData();
+                this.logger.debug('Connection %s closed with code %s: %s', data.id, code, Buffer.from(message).toString());
                 this.application.connections.remove(data.id);
                 const error = new Error('Connection closed');
                 for (const ac of data.abortControllers.values()){

package/dist/lib/server.js.map CHANGED Viewed

	@@ -1 +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 {\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.writeHeader('Access-Control-Allow-Origin', origin)\n res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')\n res.writeHeader('Access-Control-Allow-Methods', 'GET, POST')\n res.writeHeader('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","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,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,IAAIM,WAAW,CAAC,+BAA+BwF;QAC/C9F,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,oCAAoC;IACtD;IAEU2E,wBAAwBnE,SAAoB,EAAE;QACtDA,UAAUiF,OAAO;IACnB;IAEA,MAAgBC,UAAUnG,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAAC+F,GAAG,CAACK,IAAI,CAAC;YACzB,GAAGpG,OAAO;YACVqG,WAAW,IAAI,CAACxG,aAAa;QAC/B;IACF;IAEU0D,uBAAuB5C,EAAqB,EAAoB;QACxE,MAAMmB,OAAOnB,GAAGwC,WAAW;QAC3B,OAAO;YACLmD,WAAW,CAACC;gBACV,MAAMxE,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMmE,aAAa,IAAI9H,iBAAiBqD,IAAIwE;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,EAAEhD,YAAY,qBAClCxE,KAAKmB,IAAI7B,YAAYmI,eAAe,EAAEhI,aAAa8C,IAAI;gBAC3D;gBACAyE,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChBjH,KACEmB,IACA7B,YAAYqI,cAAc,EAC1BlI,aAAa8C,IAAI,WACjBW,OAAOC,IAAI,CAACuE,OAAOjD,MAAM;gBAE7B;gBACAnC,KAAKG,OAAO,CAACG,IAAI,CAACgF,GAAG,CAACrF,IAAIyE;gBAC1B,OAAO;oBAAEzE;oBAAIsF,UAAUd,KAAKc,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;YACLmD,WAAUiB,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,CAACkF,GAAG,CAACrF,IAAIyF;gBACxB3I,QAAQ0I,QAAQ;oBACdC,SAAStC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA4C,SAASX,IAAI,CAAC,SAAS,CAACvC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBxE,KAAKmB,IAAI7B,YAAY8I,aAAa,EAAE3I,aAAa8C,IAAI;gBACzD;gBACAyF,SAASX,IAAI,CAAC,SAAS;oBACrB,IAAIW,SAASR,OAAO,EAAEhD,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;YACtBkD,WAAWxE,KAAKf,OAAO,CAACqC,MAAM,CAACkD,SAAS,CAAC8B,IAAI,CAAC,MAAMvD,GAAG0C,MAAM;QAC/D;QAEAzF,KAAKQ,gBAAgB,CAAC8E,GAAG,CAACa,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,CAACtC,SAAS,CAAC;gBACpC4B;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,CAACuB,EAAE,CAAC,SAAS,CAAC7C,OAAOC;oBACxCrE,YAAYkB,IAAI7B,YAAY8J,uBAAuB,EAAE;wBACnDH,SAAStD,YAAY,CAACwD,GAAG;wBACzB9E;wBACAC;qBACD;gBACH;gBACA2E,SAAStD,YAAY,CAAC0B,IAAI,CAAC,OAAO;oBAChCpH,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,YAAYmI,eAAe,CAAC,CACjCtG,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"}
1	+ {"version":3,"sources":["../../../lib/server.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto'\nimport {\n type AnyBaseProcedure,\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 return void res\n .writeStatus('400 Bad Request')\n .end(`Service ${service} not found`)\n if (this.transportType in service.contract.transports === false)\n return void res\n .writeStatus('400 Bad Request')\n .end(`Service ${service} not supported`)\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 this.logger.debug('Connection %s opened', data.id)\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, code, message) => {\n const data = ws.getUserData()\n\n this.logger.debug(\n 'Connection %s closed with code %s: %s',\n data.id,\n code,\n Buffer.from(message).toString(),\n )\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.writeHeader('Access-Control-Allow-Origin', origin)\n res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')\n res.writeHeader('Access-Control-Allow-Methods', 'GET, POST')\n res.writeHeader('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: AnyBaseProcedure\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","contract","transports","data","id","format","streams","up","Map","down","streamId","abortControllers","subscriptions","headers","proxiedRemoteAddress","Buffer","from","getProxiedRemoteAddressAsText","toString","remoteAddress","getRemoteAddressAsText","backpressure","getHeader","open","getUserData","logger","debug","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","code","remove","Error","ac","values","abort","clear","stream","destroy","subscription","handleContainerDisposal","start","Promise","reject","hostname","listen","port","socket","info","stop","api","cause","origin","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","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,OAAO,KAAKrB,IACTG,WAAW,CAAC,mBACZI,GAAG,CAAC,CAAC,QAAQ,EAAEc,QAAQ,UAAU,CAAC;oBACvC,IAAI,IAAI,CAAC3B,aAAa,IAAI2B,QAAQE,QAAQ,CAACC,UAAU,KAAK,OACxD,OAAO,KAAKxB,IACTG,WAAW,CAAC,mBACZI,GAAG,CAAC,CAAC,QAAQ,EAAEc,QAAQ,cAAc,CAAC;gBAC7C;gBAEA,MAAMI,OAAmB;oBACvBC,IAAItD;oBACJuD,QAAQxC,UAAU0B,aAAa,IAAI,CAACjB,WAAW,CAAC+B,MAAM;oBACtDb;oBACAc,SAAS;wBACPC,IAAI,IAAIC;wBACRC,MAAM,IAAID;wBACVE,UAAU;oBACZ;oBACAC,kBAAkB,IAAIH;oBACtBI,eAAe,IAAIJ;oBACnBb;oBACAQ,MAAM;wBACJP,OAAOL,YAAYK,KAAK;wBACxBiB,SAAStB,YAAYsB,OAAO;wBAC5BC,sBAAsBC,OAAOC,IAAI,CAC/BtC,IAAIuC,6BAA6B,IACjCC,QAAQ;wBACVC,eAAeJ,OAAOC,IAAI,CACxBtC,IAAI0C,sBAAsB,IAC1BF,QAAQ;oBACZ;oBACAG,cAAc;oBACd/B,SAAS,CAAC;gBACZ;gBAEAZ,IAAIW,OAAO,CACTc,MACAxB,IAAI2C,SAAS,CAAC,sBACd3C,IAAI2C,SAAS,CAAC,2BACd3C,IAAI2C,SAAS,CAAC,6BACdhC;YAEJ;YACAiC,MAAM,CAACrC;gBACL,MAAMiB,OAAOjB,GAAGsC,WAAW;gBAC3B,IAAI,CAACC,MAAM,CAACC,KAAK,CAAC,wBAAwBvB,KAAKC,EAAE;gBACjDD,KAAKb,OAAO,CAACqC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC1C;gBAClDiB,KAAKb,OAAO,CAACuC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC5C;gBAClDiB,KAAKX,SAAS,CAACuC,OAAO,CAACpE,gBAAgBwC,KAAKA,IAAI;gBAChD,IAAI,CAAC7B,WAAW,CAAC0D,WAAW,CAACC,GAAG,CAAC;oBAC/B7B,IAAID,KAAKC,EAAE;oBACXT,UAAUQ,KAAKR,QAAQ;oBACvBuC,MAAM,IAAI,CAAC9D,aAAa;oBACxBwC,eAAeT,KAAKS,aAAa;oBACjCuB,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,MAAMiB,OAAOjB,GAAGsC,WAAW;gBAC3BrB,KAAKkB,YAAY,EAAE2B;gBACnB7C,KAAKkB,YAAY,GAAG;YACtB;YACA4B,OAAO,CAAC/D,IAAuBgE,MAAMX;gBACnC,MAAMpC,OAAOjB,GAAGsC,WAAW;gBAE3B,IAAI,CAACC,MAAM,CAACC,KAAK,CACf,yCACAvB,KAAKC,EAAE,EACP8C,MACAnC,OAAOC,IAAI,CAACuB,SAASrB,QAAQ;gBAG/B,IAAI,CAAC5C,WAAW,CAAC0D,WAAW,CAACmB,MAAM,CAAChD,KAAKC,EAAE;gBAC3C,MAAMyC,QAAQ,IAAIO,MAAM;gBAExB,KAAK,MAAMC,MAAMlD,KAAKQ,gBAAgB,CAAC2C,MAAM,GAAI;oBAC/CD,GAAGE,KAAK,CAACV;gBACX;gBACA1C,KAAKQ,gBAAgB,CAAC6C,KAAK;gBAE3B,KAAK,MAAMC,UAAUtD,KAAKG,OAAO,CAACG,IAAI,CAAC6C,MAAM,GAAI;oBAC/CG,OAAOC,OAAO,CAACb;gBACjB;gBACA1C,KAAKG,OAAO,CAACG,IAAI,CAAC+C,KAAK;gBAEvB,KAAK,MAAMC,UAAUtD,KAAKG,OAAO,CAACC,EAAE,CAAC+C,MAAM,GAAI;oBAC7CG,OAAOC,OAAO,CAACb;gBACjB;gBACA1C,KAAKG,OAAO,CAACC,EAAE,CAACiD,KAAK;gBAErB,KAAK,MAAMG,gBAAgBxD,KAAKS,aAAa,CAAC0C,MAAM,GAAI;oBACtDK,aAAaD,OAAO;gBACtB;gBACAvD,KAAKS,aAAa,CAAC4C,KAAK;gBAExB,IAAI,CAACI,uBAAuB,CAACzD,KAAKX,SAAS;YAC7C;QACF;IACJ;IAEA,MAAMqE,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACd,SAASe;YACjC,MAAMC,WAAW,IAAI,CAACzF,OAAO,CAACyF,QAAQ,IAAI;YAC1C,IAAI,CAAC7F,MAAM,CAAC8F,MAAM,CAACD,UAAU,IAAI,CAACzF,OAAO,CAAC2F,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAAC1C,MAAM,CAAC2C,IAAI,CACd,2BACAJ,UACA,IAAI,CAACzF,OAAO,CAAC2F,IAAI;oBAEnBlB;gBACF,OAAO;oBACLe,OAAO,IAAIX,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMiB,OAAO;QACX,IAAI,CAAClG,MAAM,CAAC8E,KAAK;IACnB;IAEA,IAAcqB,MAAM;QAClB,OAAO,IAAI,CAAChG,WAAW,CAACgG,GAAG;IAC7B;IAEA,IAAc7C,SAAS;QACrB,OAAO,IAAI,CAACnD,WAAW,CAACmD,MAAM;IAChC;IAEA,MAAgBqB,SACdyB,KAAY,EACZhC,UAAU,wCAAwC,EAClD;QACA,IAAI,CAACd,MAAM,CAACoB,KAAK,CAAC,IAAIO,MAAMb,SAAS;YAAEgC;QAAM;IAC/C;IAEU3F,UAAUF,GAAiB,EAAEC,GAAgB,EAAE;QAEvD,MAAM6F,SAAS7F,IAAI2C,SAAS,CAAC;QAC7B,IAAI,CAACkD,QAAQ;QACb9F,IAAIM,WAAW,CAAC,+BAA+BwF;QAC/C9F,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,oCAAoC;IACtD;IAEU4E,wBAAwBpE,SAAoB,EAAE;QACtDA,UAAUiF,OAAO;IACnB;IAEA,MAAgBC,UAAUnG,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAAC+F,GAAG,CAACK,IAAI,CAAC;YACzB,GAAGpG,OAAO;YACVqG,WAAW,IAAI,CAACxG,aAAa;QAC/B;IACF;IAEU0D,uBAAuB5C,EAAqB,EAAoB;QACxE,MAAMiB,OAAOjB,GAAGsC,WAAW;QAC3B,OAAO;YACLqD,WAAW,CAACC;gBACV,MAAM1E,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMqE,aAAa,IAAI9H,iBAAiBmD,IAAI0E;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,EAAEhD,YAAY,qBAClCxE,KAAKmB,IAAI7B,YAAYmI,eAAe,EAAEhI,aAAa4C,IAAI;gBAC3D;gBACA2E,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChBjH,KACEmB,IACA7B,YAAYqI,cAAc,EAC1BlI,aAAa4C,IAAI,WACjBW,OAAOC,IAAI,CAACyE,OAAOjD,MAAM;gBAE7B;gBACArC,KAAKG,OAAO,CAACG,IAAI,CAACkF,GAAG,CAACvF,IAAI2E;gBAC1B,OAAO;oBAAE3E;oBAAIwF,UAAUd,KAAKc,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAACzF;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;YAC/B;QACF;IACF;IAEUwB,uBAAuB1C,EAAqB,EAAE;QACtD,MAAMiB,OAAOjB,GAAGsC,WAAW;QAC3B,OAAO;YACLqD,WAAUiB,MAAmB,EAAE1F,EAAU,EAAEwF,QAAyB;gBAClE,MAAMG,WAAW,IAAI7I,eAAe0I,UAAU;oBAC5CI,MAAM,CAACC;wBACLlI,KACEmB,IACA7B,YAAY6I,YAAY,EACxB1I,aAAa4C,IAAI,WACjB5C,aAAayI,MAAM;oBAEvB;gBACF;gBAEA9F,KAAKG,OAAO,CAACC,EAAE,CAACoF,GAAG,CAACvF,IAAI2F;gBACxB3I,QAAQ0I,QAAQ;oBACdC,SAASrC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA2C,SAASX,IAAI,CAAC,SAAS,CAACvC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBxE,KAAKmB,IAAI7B,YAAY8I,aAAa,EAAE3I,aAAa4C,IAAI;gBACzD;gBACA2F,SAASX,IAAI,CAAC,SAAS;oBACrB,IAAIW,SAASR,OAAO,EAAEhD,YAAY,qBAChCxE,KAAKmB,IAAI7B,YAAY+I,WAAW,EAAE5I,aAAa4C,IAAI;gBACvD;gBAEA,OAAO2F;YACT;YACAF,WAAUzF,EAAE;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;YAC/B;QACF;IACF;IAEA,MAAgB,CAAC/C,YAAYgJ,GAAG,CAAC,CAC/BnH,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAE3B,MAAM8E,aAAa,IAAI,CAAChI,WAAW,CAAC0D,WAAW,CAACjD,GAAG,CAACoB,KAAKC,EAAE;QAC3D,IAAI,CAACkG,YAAY,OAAO,KAAKpH,GAAGD,GAAG,CAAC,MAAM;QAE1C,MAAMoE,KAAK,IAAIkD;QACf,MAAMC,MAAMrG,KAAKE,MAAM,CAACoG,OAAO,CAACC,SAAS,CAAClE,QAAQ;YAChD,GAAGrC,KAAKb,OAAO,CAACqC,MAAM;YACtBkD,WAAW1E,KAAKb,OAAO,CAACqC,MAAM,CAACkD,SAAS,CAAC8B,IAAI,CAAC,MAAMtD,GAAGyC,MAAM;QAC/D;QAEA3F,KAAKQ,gBAAgB,CAACgF,GAAG,CAACa,IAAII,MAAM,EAAEvD;QAEtC,MAAM7D,YAAYW,KAAKX,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,CAACtC,SAAS,CAAC;gBACpC4B;gBACAvG;gBACA+G;gBACAtH;gBACAsG,QAAQzC,GAAGyC,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,SAASrD,YAAY,CAACuD,GAAG;wBAAEF,SAAS3E,OAAO;qBAAC;gBACxD;gBAEA2E,SAASrD,YAAY,CAACsB,EAAE,CAAC,SAAS,CAAC7C,OAAOC;oBACxCrE,YAAYkB,IAAI7B,YAAY8J,uBAAuB,EAAE;wBACnDH,SAASrD,YAAY,CAACuD,GAAG;wBACzB9E;wBACAC;qBACD;gBACH;gBACA2E,SAASrD,YAAY,CAACyB,IAAI,CAAC,OAAO;oBAChCpH,YAAYkB,IAAI7B,YAAY+J,iBAAiB,EAAE;wBAC7CJ,SAASrD,YAAY,CAACuD,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,CAAC0E,MAAM,CAACC,KAAK,CAAC,IAAI0B,MAAM,aAAa;oBAAEmB,OAAO1B;gBAAM;gBACxD5E,eAAeiB,IAAI7B,YAAYgJ,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB/D;oBACAR,SAAS;gBACX;YACF,OAAO;gBACL,IAAI,CAACZ,MAAM,CAACoB,KAAK,CAAC,IAAIO,MAAM,oBAAoB;oBAAEmB,OAAO1B;gBAAM;gBAC/D5E,eAAeiB,IAAI7B,YAAYgJ,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB/D,OAAOjF;oBACPyE,SAAS;gBACX;YACF;QACF,SAAU;YACRlC,KAAKQ,gBAAgB,CAAC0G,MAAM,CAACb,IAAII,MAAM;YACvC,IAAI,CAAChD,uBAAuB,CAACpE;QAC/B;IACF;IAEA,MAAM,CAACnC,YAAYiK,YAAY,CAAC,CAACpI,EAAqB,EAAEsD,MAAmB,EAAE;QAC3E,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO8D,IAAI,CAACxG,OAAOC,IAAI,CAACwB,OAAOE,KAAK,CAAC8E,YAAY5E,iBAAiB;IACpE;IAEA,MAAM,CAACvF,YAAY+I,WAAW,CAAC,CAAClH,EAAqB,EAAEsD,MAAmB,EAAE;QAC1E,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO8D,IAAI,CAAC;QACZpH,KAAKG,OAAO,CAACC,EAAE,CAAC8G,MAAM,CAACjH;IACzB;IAEA,MAAM,CAAC/C,YAAY8I,aAAa,CAAC,CAC/BjH,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzBjD,KAAKG,OAAO,CAACC,EAAE,CAAC8G,MAAM,CAACjH;IACzB;IAEA,MAAM,CAAC/C,YAAYoK,cAAc,CAAC,CAChCvI,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjC,MAAMkB,KAAKkB,YAAY,EAAEqG;QACzB,IAAIjE,OAAOkE,aAAa,EACtB5J,KAAKmB,IAAI7B,YAAYuK,aAAa,EAAEpK,aAAa4C,IAAI;aAClDqD,OAAOoE,MAAM;IACpB;IAEA,MAAM,CAACxK,YAAYuK,aAAa,CAAC,CAC/B1I,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCkB,KAAKG,OAAO,CAACG,IAAI,CAAC4G,MAAM,CAACjH;IAC3B;IAEA,MAAM,CAAC/C,YAAYmI,eAAe,CAAC,CACjCtG,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAaiF,QAAQ;QAChC,MAAMiB,SAAStD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACqD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzBjD,KAAKG,OAAO,CAACG,IAAI,CAAC4G,MAAM,CAACjH;IAC3B;IAEA,MAAM,CAAC/C,YAAYyK,iBAAiB,CAAC,CACnC5I,EAAqB,EACrBsD,MAAmB,EACnB;QACA,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAM,CAAC0F,IAAI,GAAG/G,KAAKE,MAAM,CAACoG,OAAO,CAAC9E,MAAM,CAACa;QACzC,MAAMmB,eAAexD,KAAKS,aAAa,CAAC7B,GAAG,CAACmI;QAC5C,IAAI,CAACvD,cAAc,OAAO,KAAKzE,GAAGD,GAAG;QACrC0E,aAAaD,OAAO;QACpBvD,KAAKS,aAAa,CAACyG,MAAM,CAACH;IAC5B;IAEA,MAAM,CAAC7J,YAAY0K,QAAQ,CAAC,CAAC7I,EAAqB,EAAEsD,MAAmB,EAAE;QACvE,MAAMrC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMoF,SAASrJ,aAAaiF,QAAQ;QACpC,MAAMa,KAAKlD,KAAKQ,gBAAgB,CAAC5B,GAAG,CAAC6H;QACrC,IAAIvD,IAAIA,GAAGE,KAAK,CAAC,IAAIH,MAAM;IAC7B;AACF"}

package/lib/server.ts CHANGED Viewed

@@ -1,6 +1,6 @@
 import { randomUUID } from 'node:crypto'
 import {
-  type AnyProcedure,
+  type AnyBaseProcedure,
   ApiError,
   type ApplicationContext,
   type Connection,
@@ -78,13 +78,13 @@ export class WsTransportServer {
           for (const serviceName of services) {
             const service = this.application.registry.services.get(serviceName)
             if (!service)
-              throw new Response(`Service ${service} not found`, {
-                status: 400,
-              })
+              return void res
+                .writeStatus('400 Bad Request')
+                .end(`Service ${service} not found`)
             if (this.transportType in service.contract.transports === false)
-              throw new Response(`Service ${service} not supported`, {
-                status: 400,
-              })
+              return void res
+                .writeStatus('400 Bad Request')
+                .end(`Service ${service} not supported`)
           }
           const data: WsUserData = {
@@ -123,6 +123,7 @@ export class WsTransportServer {
         },
         open: (ws: WsTransportSocket) => {
           const data = ws.getUserData()
+          this.logger.debug('Connection %s opened', data.id)
           data.context.decode = this.createDecodeRpcContext(ws)
           data.context.encode = this.createEncodeRpcContext(ws)
           data.container.provide(connectionData, data.data)
@@ -156,9 +157,16 @@ export class WsTransportServer {
           data.backpressure?.resolve()
           data.backpressure = null
         },
-        close: (ws: WsTransportSocket) => {
+        close: (ws: WsTransportSocket, code, message) => {
           const data = ws.getUserData()
+          this.logger.debug(
+            'Connection %s closed with code %s: %s',
+            data.id,
+            code,
+            Buffer.from(message).toString(),
+          )
           this.application.connections.remove(data.id)
           const error = new Error('Connection closed')
@@ -241,7 +249,7 @@ export class WsTransportServer {
   protected async handleRPC(options: {
     connection: Connection
     service: Service
-    procedure: AnyProcedure
+    procedure: AnyBaseProcedure
     container: Container
     signal: AbortSignal
     payload: any

package/package.json CHANGED Viewed

@@ -8,15 +8,15 @@
     }
   },
   "dependencies": {
-    "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.47.0"
+    "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.48.0"
   },
   "peerDependencies": {
-    "@nmtjs/application": "^0.2.1",
-    "@nmtjs/common": "^0.2.1"
+    "@nmtjs/application": "^0.3.8",
+    "@nmtjs/common": "^0.3.8"
   },
   "devDependencies": {
-    "@nmtjs/application": "^0.2.1",
-    "@nmtjs/common": "^0.2.1",
+    "@nmtjs/application": "^0.3.8",
+    "@nmtjs/common": "^0.3.8",
     "@types/node": "^18.0.0"
   },
   "engines": {
@@ -30,7 +30,7 @@
     "LICENSE.md",
     "README.md"
   ],
-  "version": "0.1.3",
+  "version": "0.1.4",
   "scripts": {
     "build": "neemata-build ./index.ts './lib/**/*.ts'",
     "type-check": "tsc --noEmit"