@nmtjs/ws-transport 0.1.6 → 0.1.7
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.
- package/dist/lib/server.js +3 -2
- package/dist/lib/server.js.map +1 -1
- package/lib/server.ts +5 -1
- package/package.json +5 -5
package/dist/lib/server.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { randomUUID } from 'node:crypto';
|
|
2
|
-
import { ApiError, Scope, ServerDownStream, ServerUpStream, SubscriptionResponse, onAbort } from '@nmtjs/application';
|
|
2
|
+
import { ApiError, Scope, ServerDownStream, ServerUpStream, SubscriptionResponse, builtin, onAbort } from '@nmtjs/application';
|
|
3
3
|
import { MessageType, TransportType, decodeNumber, encodeNumber } from '@nmtjs/common';
|
|
4
4
|
import { App, SSLApp } from 'uWebSockets.js';
|
|
5
5
|
import { connectionData } from "./injectables.js";
|
|
@@ -63,7 +63,7 @@ export class WsTransportServer {
|
|
|
63
63
|
data.context.decode = this.createDecodeRpcContext(ws);
|
|
64
64
|
data.context.encode = this.createEncodeRpcContext(ws);
|
|
65
65
|
data.container.provide(connectionData, data.data);
|
|
66
|
-
this.application.connections.add({
|
|
66
|
+
const connection = this.application.connections.add({
|
|
67
67
|
id: data.id,
|
|
68
68
|
services: data.services,
|
|
69
69
|
type: this.transportType,
|
|
@@ -74,6 +74,7 @@ export class WsTransportServer {
|
|
|
74
74
|
payload
|
|
75
75
|
])
|
|
76
76
|
});
|
|
77
|
+
data.container.provide(builtin.connection, connection);
|
|
77
78
|
},
|
|
78
79
|
message: async (ws, event)=>{
|
|
79
80
|
const buffer = event;
|
package/dist/lib/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
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 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 async handleContainerDisposal(container: Container) {\n await 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,MAAMC,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;IAEA,MAAgB4E,wBAAwBpE,SAAoB,EAAE;QAC5D,MAAMA,UAAUiF,OAAO;IACzB;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"}
|
|
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 builtin,\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 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\n const connection = 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 data.container.provide(builtin.connection, connection)\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 async handleContainerDisposal(container: Container) {\n await 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","builtin","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","connection","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","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,EACPC,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,MAAMC,cAAczB,eAAea;gBACnC,MAAMa,YAAY,IAAI,CAAClB,WAAW,CAACkB,SAAS,CAACC,WAAW,CACtD1C,MAAM2C,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,IAAIvD;oBACJwD,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;gBAEhD,MAAM6B,aAAa,IAAI,CAAC1D,WAAW,CAAC2D,WAAW,CAACC,GAAG,CAAC;oBAClD9B,IAAID,KAAKC,EAAE;oBACXT,UAAUQ,KAAKR,QAAQ;oBACvBwC,MAAM,IAAI,CAAC/D,aAAa;oBACxBwC,eAAeT,KAAKS,aAAa;oBACjCwB,WAAW,CAACrC,SAASsC,OAAOC,UAC1BtE,YAAYkB,IAAI7B,YAAYkF,KAAK,EAAE;4BAACxC;4BAASsC;4BAAOC;yBAAQ;gBAChE;gBAEAnC,KAAKX,SAAS,CAACuC,OAAO,CAAC5E,QAAQ6E,UAAU,EAAEA;YAC7C;YACAQ,SAAS,OAAOtD,IAAuBmD;gBACrC,MAAMI,SAASJ;gBACf,MAAMK,cAAcnF,aAAakF,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,MAAMiB,OAAOjB,GAAGsC,WAAW;gBAC3BrB,KAAKkB,YAAY,EAAE4B;gBACnB9C,KAAKkB,YAAY,GAAG;YACtB;YACA6B,OAAO,CAAChE,IAAuBiE,MAAMX;gBACnC,MAAMrC,OAAOjB,GAAGsC,WAAW;gBAE3B,IAAI,CAACC,MAAM,CAACC,KAAK,CACf,yCACAvB,KAAKC,EAAE,EACP+C,MACApC,OAAOC,IAAI,CAACwB,SAAStB,QAAQ;gBAG/B,IAAI,CAAC5C,WAAW,CAAC2D,WAAW,CAACmB,MAAM,CAACjD,KAAKC,EAAE;gBAC3C,MAAM0C,QAAQ,IAAIO,MAAM;gBAExB,KAAK,MAAMC,MAAMnD,KAAKQ,gBAAgB,CAAC4C,MAAM,GAAI;oBAC/CD,GAAGE,KAAK,CAACV;gBACX;gBACA3C,KAAKQ,gBAAgB,CAAC8C,KAAK;gBAE3B,KAAK,MAAMC,UAAUvD,KAAKG,OAAO,CAACG,IAAI,CAAC8C,MAAM,GAAI;oBAC/CG,OAAOC,OAAO,CAACb;gBACjB;gBACA3C,KAAKG,OAAO,CAACG,IAAI,CAACgD,KAAK;gBAEvB,KAAK,MAAMC,UAAUvD,KAAKG,OAAO,CAACC,EAAE,CAACgD,MAAM,GAAI;oBAC7CG,OAAOC,OAAO,CAACb;gBACjB;gBACA3C,KAAKG,OAAO,CAACC,EAAE,CAACkD,KAAK;gBAErB,KAAK,MAAMG,gBAAgBzD,KAAKS,aAAa,CAAC2C,MAAM,GAAI;oBACtDK,aAAaD,OAAO;gBACtB;gBACAxD,KAAKS,aAAa,CAAC6C,KAAK;gBAExB,IAAI,CAACI,uBAAuB,CAAC1D,KAAKX,SAAS;YAC7C;QACF;IACJ;IAEA,MAAMsE,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACd,SAASe;YACjC,MAAMC,WAAW,IAAI,CAAC1F,OAAO,CAAC0F,QAAQ,IAAI;YAC1C,IAAI,CAAC9F,MAAM,CAAC+F,MAAM,CAACD,UAAU,IAAI,CAAC1F,OAAO,CAAC4F,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAAC3C,MAAM,CAAC4C,IAAI,CACd,2BACAJ,UACA,IAAI,CAAC1F,OAAO,CAAC4F,IAAI;oBAEnBlB;gBACF,OAAO;oBACLe,OAAO,IAAIX,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMiB,OAAO;QACX,IAAI,CAACnG,MAAM,CAAC+E,KAAK;IACnB;IAEA,IAAcqB,MAAM;QAClB,OAAO,IAAI,CAACjG,WAAW,CAACiG,GAAG;IAC7B;IAEA,IAAc9C,SAAS;QACrB,OAAO,IAAI,CAACnD,WAAW,CAACmD,MAAM;IAChC;IAEA,MAAgBsB,SACdyB,KAAY,EACZhC,UAAU,wCAAwC,EAClD;QACA,IAAI,CAACf,MAAM,CAACqB,KAAK,CAAC,IAAIO,MAAMb,SAAS;YAAEgC;QAAM;IAC/C;IAEU5F,UAAUF,GAAiB,EAAEC,GAAgB,EAAE;QAEvD,MAAM8F,SAAS9F,IAAI2C,SAAS,CAAC;QAC7B,IAAI,CAACmD,QAAQ;QACb/F,IAAIM,WAAW,CAAC,+BAA+ByF;QAC/C/F,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,oCAAoC;IACtD;IAEA,MAAgB6E,wBAAwBrE,SAAoB,EAAE;QAC5D,MAAMA,UAAUkF,OAAO;IACzB;IAEA,MAAgBC,UAAUpG,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAACgG,GAAG,CAACK,IAAI,CAAC;YACzB,GAAGrG,OAAO;YACVsG,WAAW,IAAI,CAACzG,aAAa;QAC/B;IACF;IAEU0D,uBAAuB5C,EAAqB,EAAoB;QACxE,MAAMiB,OAAOjB,GAAGsC,WAAW;QAC3B,OAAO;YACLsD,WAAW,CAACC;gBACV,MAAM3E,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMsE,aAAa,IAAIhI,iBAAiBoD,IAAI2E;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,qBAClCzE,KAAKmB,IAAI7B,YAAYoI,eAAe,EAAEjI,aAAa4C,IAAI;gBAC3D;gBACA4E,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChBlH,KACEmB,IACA7B,YAAYsI,cAAc,EAC1BnI,aAAa4C,IAAI,WACjBW,OAAOC,IAAI,CAAC0E,OAAOjD,MAAM;gBAE7B;gBACAtC,KAAKG,OAAO,CAACG,IAAI,CAACmF,GAAG,CAACxF,IAAI4E;gBAC1B,OAAO;oBAAE5E;oBAAIyF,UAAUd,KAAKc,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAAC1F;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;YAC/B;QACF;IACF;IAEUwB,uBAAuB1C,EAAqB,EAAE;QACtD,MAAMiB,OAAOjB,GAAGsC,WAAW;QAC3B,OAAO;YACLsD,WAAUiB,MAAmB,EAAE3F,EAAU,EAAEyF,QAAyB;gBAClE,MAAMG,WAAW,IAAI/I,eAAe4I,UAAU;oBAC5CI,MAAM,CAACC;wBACLnI,KACEmB,IACA7B,YAAY8I,YAAY,EACxB3I,aAAa4C,IAAI,WACjB5C,aAAa0I,MAAM;oBAEvB;gBACF;gBAEA/F,KAAKG,OAAO,CAACC,EAAE,CAACqF,GAAG,CAACxF,IAAI4F;gBACxB5I,QAAQ2I,QAAQ;oBACdC,SAASrC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA2C,SAASX,IAAI,CAAC,SAAS,CAACvC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBzE,KAAKmB,IAAI7B,YAAY+I,aAAa,EAAE5I,aAAa4C,IAAI;gBACzD;gBACA4F,SAASX,IAAI,CAAC,SAAS;oBACrB,IAAIW,SAASR,OAAO,EAAEhD,YAAY,qBAChCzE,KAAKmB,IAAI7B,YAAYgJ,WAAW,EAAE7I,aAAa4C,IAAI;gBACvD;gBAEA,OAAO4F;YACT;YACAF,WAAU1F,EAAE;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;YAC/B;QACF;IACF;IAEA,MAAgB,CAAC/C,YAAYiJ,GAAG,CAAC,CAC/BpH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAE3B,MAAMQ,aAAa,IAAI,CAAC1D,WAAW,CAAC2D,WAAW,CAAClD,GAAG,CAACoB,KAAKC,EAAE;QAC3D,IAAI,CAAC4B,YAAY,OAAO,KAAK9C,GAAGD,GAAG,CAAC,MAAM;QAE1C,MAAMqE,KAAK,IAAIiD;QACf,MAAMC,MAAMrG,KAAKE,MAAM,CAACoG,OAAO,CAACC,SAAS,CAACjE,QAAQ;YAChD,GAAGtC,KAAKb,OAAO,CAACqC,MAAM;YACtBmD,WAAW3E,KAAKb,OAAO,CAACqC,MAAM,CAACmD,SAAS,CAAC6B,IAAI,CAAC,MAAMrD,GAAGyC,MAAM;QAC/D;QAEA5F,KAAKQ,gBAAgB,CAACiF,GAAG,CAACY,IAAII,MAAM,EAAEtD;QAEtC,MAAM9D,YAAYW,KAAKX,SAAS,CAACC,WAAW,CAAC1C,MAAM8J,IAAI;QAEvD,IAAI;YACF,MAAM,EAAE9G,OAAO,EAAE+G,SAAS,EAAE,GAAG,IAAI,CAACvC,GAAG,CAACwC,IAAI,CAC1CP,IAAIzG,OAAO,EACXyG,IAAIM,SAAS,EACb,IAAI,CAAC1I,aAAa;YAGpB,MAAM4I,WAAW,MAAM,IAAI,CAACrC,SAAS,CAAC;gBACpC3C;gBACAjC;gBACA+G;gBACAtH;gBACAuG,QAAQzC,GAAGyC,MAAM;gBACjBzD,SAASkE,IAAIlE,OAAO;YACtB;YAEA,IAAI0E,oBAAoB9J,sBAAsB;gBAC5Ce,eAAeiB,IAAI7B,YAAY4J,eAAe,EAAE;oBAC9CL,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS;wBAAC0E,SAASpD,YAAY,CAACsD,GAAG;wBAAEF,SAAS1E,OAAO;qBAAC;gBACxD;gBAEA0E,SAASpD,YAAY,CAACsB,EAAE,CAAC,SAAS,CAAC7C,OAAOC;oBACxCtE,YAAYkB,IAAI7B,YAAY8J,uBAAuB,EAAE;wBACnDH,SAASpD,YAAY,CAACsD,GAAG;wBACzB7E;wBACAC;qBACD;gBACH;gBACA0E,SAASpD,YAAY,CAACyB,IAAI,CAAC,OAAO;oBAChCrH,YAAYkB,IAAI7B,YAAY+J,iBAAiB,EAAE;wBAC7CJ,SAASpD,YAAY,CAACsD,GAAG;qBAC1B;gBACH;YACF,OAAO;gBACLjJ,eAAeiB,IAAI7B,YAAYiJ,GAAG,EAAE;oBAClCM,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS0E;gBACX;YACF;QACF,EAAE,OAAOlE,OAAO;YACd,IAAIA,iBAAiBhG,UAAU;gBAC7B,IAAI,CAAC2E,MAAM,CAACC,KAAK,CAAC,IAAI2B,MAAM,aAAa;oBAAEmB,OAAO1B;gBAAM;gBACxD7E,eAAeiB,IAAI7B,YAAYiJ,GAAG,EAAE;oBAClCM,QAAQJ,IAAII,MAAM;oBAClB9D;oBACAR,SAAS;gBACX;YACF,OAAO;gBACL,IAAI,CAACb,MAAM,CAACqB,KAAK,CAAC,IAAIO,MAAM,oBAAoB;oBAAEmB,OAAO1B;gBAAM;gBAC/D7E,eAAeiB,IAAI7B,YAAYiJ,GAAG,EAAE;oBAClCM,QAAQJ,IAAII,MAAM;oBAClB9D,OAAOlF;oBACP0E,SAAS;gBACX;YACF;QACF,SAAU;YACRnC,KAAKQ,gBAAgB,CAAC0G,MAAM,CAACb,IAAII,MAAM;YACvC,IAAI,CAAC/C,uBAAuB,CAACrE;QAC/B;IACF;IAEA,MAAM,CAACnC,YAAYiK,YAAY,CAAC,CAACpI,EAAqB,EAAEuD,MAAmB,EAAE;QAC3E,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjCyE,OAAO6D,IAAI,CAACxG,OAAOC,IAAI,CAACyB,OAAOE,KAAK,CAAC6E,YAAY3E,iBAAiB;IACpE;IAEA,MAAM,CAACxF,YAAYgJ,WAAW,CAAC,CAACnH,EAAqB,EAAEuD,MAAmB,EAAE;QAC1E,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjCyE,OAAO6D,IAAI,CAAC;QACZpH,KAAKG,OAAO,CAACC,EAAE,CAAC8G,MAAM,CAACjH;IACzB;IAEA,MAAM,CAAC/C,YAAY+I,aAAa,CAAC,CAC/BlH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACC,EAAE,CAACxB,GAAG,CAACqB;QACnC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjCyE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzBlD,KAAKG,OAAO,CAACC,EAAE,CAAC8G,MAAM,CAACjH;IACzB;IAEA,MAAM,CAAC/C,YAAYoK,cAAc,CAAC,CAChCvI,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjC,MAAMkB,KAAKkB,YAAY,EAAEqG;QACzB,IAAIhE,OAAOiE,aAAa,EACtB5J,KAAKmB,IAAI7B,YAAYuK,aAAa,EAAEpK,aAAa4C,IAAI;aAClDsD,OAAOmE,MAAM;IACpB;IAEA,MAAM,CAACxK,YAAYuK,aAAa,CAAC,CAC/B1I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjCkB,KAAKG,OAAO,CAACG,IAAI,CAAC4G,MAAM,CAACjH;IAC3B;IAEA,MAAM,CAAC/C,YAAYoI,eAAe,CAAC,CACjCvG,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMpB,KAAK7C,aAAakF,QAAQ;QAChC,MAAMiB,SAASvD,KAAKG,OAAO,CAACG,IAAI,CAAC1B,GAAG,CAACqB;QACrC,IAAI,CAACsD,QAAQ,OAAOxE,GAAGD,GAAG,CAAC,MAAM;QACjCyE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzBlD,KAAKG,OAAO,CAACG,IAAI,CAAC4G,MAAM,CAACjH;IAC3B;IAEA,MAAM,CAAC/C,YAAYyK,iBAAiB,CAAC,CACnC5I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAM,CAAC0F,IAAI,GAAG/G,KAAKE,MAAM,CAACoG,OAAO,CAAC9E,MAAM,CAACc;QACzC,MAAMmB,eAAezD,KAAKS,aAAa,CAAC7B,GAAG,CAACmI;QAC5C,IAAI,CAACtD,cAAc,OAAO,KAAK1E,GAAGD,GAAG;QACrC2E,aAAaD,OAAO;QACpBxD,KAAKS,aAAa,CAACyG,MAAM,CAACH;IAC5B;IAEA,MAAM,CAAC7J,YAAY0K,QAAQ,CAAC,CAAC7I,EAAqB,EAAEuD,MAAmB,EAAE;QACvE,MAAMtC,OAAOjB,GAAGsC,WAAW;QAC3B,MAAMoF,SAASrJ,aAAakF,QAAQ;QACpC,MAAMa,KAAKnD,KAAKQ,gBAAgB,CAAC5B,GAAG,CAAC6H;QACrC,IAAItD,IAAIA,GAAGE,KAAK,CAAC,IAAIH,MAAM;IAC7B;AACF"}
|
package/lib/server.ts
CHANGED
|
@@ -10,6 +10,7 @@ import {
|
|
|
10
10
|
ServerUpStream,
|
|
11
11
|
type Service,
|
|
12
12
|
SubscriptionResponse,
|
|
13
|
+
builtin,
|
|
13
14
|
onAbort,
|
|
14
15
|
} from '@nmtjs/application'
|
|
15
16
|
import {
|
|
@@ -126,7 +127,8 @@ export class WsTransportServer {
|
|
|
126
127
|
data.context.decode = this.createDecodeRpcContext(ws)
|
|
127
128
|
data.context.encode = this.createEncodeRpcContext(ws)
|
|
128
129
|
data.container.provide(connectionData, data.data)
|
|
129
|
-
|
|
130
|
+
|
|
131
|
+
const connection = this.application.connections.add({
|
|
130
132
|
id: data.id,
|
|
131
133
|
services: data.services,
|
|
132
134
|
type: this.transportType,
|
|
@@ -134,6 +136,8 @@ export class WsTransportServer {
|
|
|
134
136
|
sendEvent: (service, event, payload) =>
|
|
135
137
|
sendMessage(ws, MessageType.Event, [service, event, payload]),
|
|
136
138
|
})
|
|
139
|
+
|
|
140
|
+
data.container.provide(builtin.connection, connection)
|
|
137
141
|
},
|
|
138
142
|
message: async (ws: WsTransportSocket, event) => {
|
|
139
143
|
const buffer = event as unknown as ArrayBuffer
|
package/package.json
CHANGED
|
@@ -11,12 +11,12 @@
|
|
|
11
11
|
"uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.48.0"
|
|
12
12
|
},
|
|
13
13
|
"peerDependencies": {
|
|
14
|
-
"@nmtjs/application": "^0.3.
|
|
15
|
-
"@nmtjs/common": "^0.3.
|
|
14
|
+
"@nmtjs/application": "^0.3.9",
|
|
15
|
+
"@nmtjs/common": "^0.3.9"
|
|
16
16
|
},
|
|
17
17
|
"devDependencies": {
|
|
18
|
-
"@nmtjs/application": "^0.3.
|
|
19
|
-
"@nmtjs/common": "^0.3.
|
|
18
|
+
"@nmtjs/application": "^0.3.9",
|
|
19
|
+
"@nmtjs/common": "^0.3.9",
|
|
20
20
|
"@types/node": "^18.0.0"
|
|
21
21
|
},
|
|
22
22
|
"engines": {
|
|
@@ -30,7 +30,7 @@
|
|
|
30
30
|
"LICENSE.md",
|
|
31
31
|
"README.md"
|
|
32
32
|
],
|
|
33
|
-
"version": "0.1.
|
|
33
|
+
"version": "0.1.7",
|
|
34
34
|
"scripts": {
|
|
35
35
|
"build": "neemata-build ./index.ts './lib/**/*.ts'",
|
|
36
36
|
"type-check": "tsc --noEmit"
|