@nmtjs/ws-transport 0.0.1

package/LICENSE.md

@@ -0,0 +1,7 @@
+Copyright (c) 2024 Denis Ilchyshyn
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package/README.md

@@ -0,0 +1 @@
+## WebSockets transport for [Neemata](https://github.com/neematajs/neemata)

package/dist/index.js

@@ -0,0 +1,5 @@
+export * from "./lib/server.js";
+export * from "./lib/transport.js";
+export * from "./lib/types.js";
+export * from "./lib/utils.js";
+export * from "./lib/providers.js";

package/dist/index.js.map

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

package/dist/lib/providers.js

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

package/dist/lib/providers.js.map

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

package/dist/lib/server.js

@@ -0,0 +1,347 @@
+import { randomUUID } from 'node:crypto';
+import { ApiError, Scope, ServerDownStream, ServerUpStream, SubscriptionResponse, onAbort } from '@nmtjs/application';
+import { MessageType, TransportType, decodeNumber, encodeNumber } from '@nmtjs/common';
+import { App, SSLApp } from 'uWebSockets.js';
+import { connectionData } from "./providers.js";
+import { InternalError, getFormat, getRequestData, send, sendMessage, sendRpcMessage } from "./utils.js";
+export class WsTransportServer {
+    application;
+    options;
+    server;
+    transportType;
+    constructor(application, options){
+        this.application = application;
+        this.options = options;
+        this.transportType = TransportType.WS;
+        this.server = this.options.tls ? SSLApp(options.tls) : App();
+        this.server.get('/healthy', (res)=>{
+            res.writeHeader('Access-Control-Allow-Origin', '*');
+            res.writeHeader('Access-Control-Allow-Headers', 'Content-Type');
+            res.writeHeader('Access-Control-Allow-Methods', 'GET');
+            res.writeHeader('Content-Type', 'text/plain');
+            res.end('OK');
+        }).ws('/api', {
+            sendPingsAutomatically: true,
+            maxPayloadLength: this.options.maxPayloadLength,
+            upgrade: (res, req, context)=>{
+                const requestData = getRequestData(req);
+                const container = this.application.container.createScope(Scope.Connection);
+                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
+                    });
+                }
+                const data = {
+                    id: randomUUID(),
+                    format: getFormat(requestData, this.application.format),
+                    container,
+                    streams: {
+                        up: new Map(),
+                        down: new Map(),
+                        streamId: 0
+                    },
+                    abortControllers: new Map(),
+                    subscriptions: new Map(),
+                    services,
+                    data: {
+                        query: requestData.query,
+                        headers: requestData.headers,
+                        proxiedRemoteAddress: Buffer.from(res.getProxiedRemoteAddressAsText()).toString(),
+                        remoteAddress: Buffer.from(res.getRemoteAddressAsText()).toString()
+                    },
+                    backpressure: null,
+                    context: {}
+                };
+                res.upgrade(data, req.getHeader('sec-websocket-key'), req.getHeader('sec-websocket-protocol'), req.getHeader('sec-websocket-extensions'), context);
+            },
+            open: (ws)=>{
+                const data = ws.getUserData();
+                data.context.decode = this.createDecodeRpcContext(ws);
+                data.context.encode = this.createEncodeRpcContext(ws);
+                data.container.provide(connectionData, data.data);
+                this.application.connections.add({
+                    id: data.id,
+                    services: data.services,
+                    type: this.transportType,
+                    subscriptions: data.subscriptions,
+                    sendEvent: (service, event, payload)=>sendMessage(ws, MessageType.Event, [
+                            service,
+                            event,
+                            payload
+                        ])
+                });
+            },
+            message: async (ws, event)=>{
+                const buffer = event;
+                const messageType = decodeNumber(buffer, 'Uint8');
+                if (messageType in this === false) {
+                    ws.end(1011, 'Unknown message type');
+                } else {
+                    try {
+                        await this[messageType](ws, buffer.slice(Uint8Array.BYTES_PER_ELEMENT));
+                    } catch (error) {
+                        this.logError(error, 'Error while processing message');
+                    }
+                }
+            },
+            drain: (ws)=>{
+                const data = ws.getUserData();
+                data.backpressure?.resolve();
+                data.backpressure = null;
+            },
+            close: (ws)=>{
+                const data = ws.getUserData();
+                this.application.connections.remove(data.id);
+                const error = new Error('Connection closed');
+                for (const ac of data.abortControllers.values()){
+                    ac.abort(error);
+                }
+                data.abortControllers.clear();
+                for (const stream of data.streams.down.values()){
+                    stream.destroy(error);
+                }
+                data.streams.down.clear();
+                for (const stream of data.streams.up.values()){
+                    stream.destroy(error);
+                }
+                data.streams.up.clear();
+                for (const subscription of data.subscriptions.values()){
+                    subscription.destroy();
+                }
+                data.subscriptions.clear();
+                this.handleContainerDisposal(data.container);
+            }
+        });
+    }
+    async start() {
+        return new Promise((resolve, reject)=>{
+            const hostname = this.options.hostname ?? '127.0.0.1';
+            this.server.listen(hostname, this.options.port, (socket)=>{
+                if (socket) {
+                    this.logger.info('Server started on %s:%s', hostname, this.options.port);
+                    resolve();
+                } else {
+                    reject(new Error('Failed to start server'));
+                }
+            });
+        });
+    }
+    async stop() {
+        this.server.close();
+    }
+    get api() {
+        return this.application.api;
+    }
+    get logger() {
+        return this.application.logger;
+    }
+    async logError(cause, message = 'Unknown error while processing request') {
+        this.logger.error(new Error(message, {
+            cause
+        }));
+    }
+    handleContainerDisposal(container) {
+        container.dispose();
+    }
+    async handleRPC(options) {
+        return await this.api.call({
+            ...options,
+            transport: this.transportType
+        });
+    }
+    createEncodeRpcContext(ws) {
+        const data = ws.getUserData();
+        return {
+            addStream: (blob)=>{
+                const id = ++data.streams.streamId;
+                const downstream = new ServerDownStream(id, blob);
+                downstream.pause();
+                downstream.on('error', console.dir);
+                downstream.once('error', (err)=>{
+                    console.log({
+                        err
+                    });
+                    if (downstream.errored?.message !== 'Aborted by client') send(ws, MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'));
+                });
+                downstream.on('data', (chunk)=>{
+                    downstream.pause();
+                    send(ws, MessageType.DownStreamPush, encodeNumber(id, 'Uint32'), Buffer.from(chunk).buffer);
+                });
+                data.streams.down.set(id, downstream);
+                return {
+                    id,
+                    metadata: blob.metadata
+                };
+            },
+            getStream: (id)=>{
+                return data.streams.down.get(id);
+            }
+        };
+    }
+    createDecodeRpcContext(ws) {
+        const data = ws.getUserData();
+        return {
+            addStream (signal, id, metadata) {
+                const upstream = new ServerUpStream(metadata, {
+                    read: (size)=>{
+                        send(ws, MessageType.UpStreamPull, encodeNumber(id, 'Uint32'), encodeNumber(size, 'Uint32'));
+                    }
+                });
+                data.streams.up.set(id, upstream);
+                onAbort(signal, ()=>{
+                    upstream.destroy(new Error('Call aborted by client'));
+                });
+                upstream.once('error', (error)=>{
+                    if (error.message !== 'Aborted by server') send(ws, MessageType.UpStreamAbort, encodeNumber(id, 'Uint32'));
+                });
+                upstream.once('close', ()=>{
+                    if (upstream.errored?.message !== 'Aborted by server') send(ws, MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'));
+                });
+                return upstream;
+            },
+            getStream (id) {
+                return data.streams.down.get(id);
+            }
+        };
+    }
+    async [MessageType.Rpc](ws, buffer) {
+        const data = ws.getUserData();
+        const connection = this.application.connections.get(data.id);
+        if (!connection) return void ws.end(1011, 'Unknown connection');
+        const ac = new AbortController();
+        const rpc = data.format.decoder.decodeRpc(buffer, {
+            ...data.context.decode,
+            addStream: data.context.decode.addStream.bind(null, ac.signal)
+        });
+        data.abortControllers.set(rpc.callId, ac);
+        const container = data.container.createScope(Scope.Call);
+        try {
+            const { service, procedure } = this.api.find(rpc.service, rpc.procedure, this.transportType);
+            const response = await this.handleRPC({
+                connection,
+                service,
+                procedure,
+                container,
+                signal: ac.signal,
+                payload: rpc.payload
+            });
+            if (response instanceof SubscriptionResponse) {
+                sendRpcMessage(ws, MessageType.RpcSubscription, {
+                    callId: rpc.callId,
+                    error: null,
+                    payload: [
+                        response.subscription.key,
+                        response.payload
+                    ]
+                });
+                response.subscription.on('event', (event, payload)=>{
+                    sendMessage(ws, MessageType.ServerSubscriptionEvent, [
+                        response.subscription.key,
+                        event,
+                        payload
+                    ]);
+                });
+                response.subscription.once('end', ()=>{
+                    sendMessage(ws, MessageType.ServerUnsubscribe, [
+                        response.subscription.key
+                    ]);
+                });
+            } else {
+                sendRpcMessage(ws, MessageType.Rpc, {
+                    callId: rpc.callId,
+                    error: null,
+                    payload: response
+                });
+            }
+        } catch (error) {
+            if (error instanceof ApiError) {
+                this.logger.debug(new Error('Api error', {
+                    cause: error
+                }));
+                sendRpcMessage(ws, MessageType.Rpc, {
+                    callId: rpc.callId,
+                    error,
+                    payload: null
+                });
+            } else {
+                this.logger.error(new Error('Unexpected error', {
+                    cause: error
+                }));
+                sendRpcMessage(ws, MessageType.Rpc, {
+                    callId: rpc.callId,
+                    error: InternalError(),
+                    payload: null
+                });
+            }
+        } finally{
+            data.abortControllers.delete(rpc.callId);
+            this.handleContainerDisposal(container);
+        }
+    }
+    async [MessageType.UpStreamPush](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.up.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        stream.push(Buffer.from(buffer.slice(Uint32Array.BYTES_PER_ELEMENT)));
+    }
+    async [MessageType.UpStreamEnd](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.up.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        stream.push(null);
+        data.streams.up.delete(id);
+    }
+    async [MessageType.UpStreamAbort](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.up.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        stream.destroy(new Error('Aborted by client'));
+        data.streams.up.delete(id);
+    }
+    async [MessageType.DownStreamPull](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.down.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        await data.backpressure?.promise;
+        if (stream.readableEnded) send(ws, MessageType.DownStreamEnd, encodeNumber(id, 'Uint32'));
+        else stream.resume();
+    }
+    async [MessageType.DownStreamEnd](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.down.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        data.streams.down.delete(id);
+    }
+    async [MessageType.DownStreamAbort](ws, buffer) {
+        const data = ws.getUserData();
+        const id = decodeNumber(buffer, 'Uint32');
+        const stream = data.streams.down.get(id);
+        if (!stream) return ws.end(1011, 'Unknown stream');
+        stream.destroy(new Error('Aborted by client'));
+        data.streams.down.delete(id);
+    }
+    async [MessageType.ClientUnsubscribe](ws, buffer) {
+        const data = ws.getUserData();
+        const [key] = data.format.decoder.decode(buffer);
+        const subscription = data.subscriptions.get(key);
+        if (!subscription) return void ws.end();
+        subscription.destroy();
+        data.subscriptions.delete(key);
+    }
+    async [MessageType.RpcAbort](ws, buffer) {
+        const data = ws.getUserData();
+        const callId = decodeNumber(buffer, 'Uint32');
+        const ac = data.abortControllers.get(callId);
+        if (ac) ac.abort(new Error('Aborted by client'));
+    }
+}

package/dist/lib/server.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../lib/server.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto'\nimport {\n  type AnyProcedure,\n  ApiError,\n  type ApplicationContext,\n  type Connection,\n  type Container,\n  Scope,\n  ServerDownStream,\n  ServerUpStream,\n  type Service,\n  SubscriptionResponse,\n  onAbort,\n} from '@nmtjs/application'\nimport {\n  type ApiBlobMetadata,\n  type EncodeRpcContext,\n  MessageType,\n  TransportType,\n  decodeNumber,\n  encodeNumber,\n} from '@nmtjs/common'\nimport { App, SSLApp, type TemplatedApp } from 'uWebSockets.js'\nimport { connectionData } from './providers.ts'\nimport type {\n  WsTransportOptions,\n  WsTransportSocket,\n  WsUserData,\n} from './types.ts'\nimport {\n  InternalError,\n  getFormat,\n  getRequestData,\n  send,\n  sendMessage,\n  sendRpcMessage,\n} from './utils.ts'\n\nexport class WsTransportServer {\n  protected server!: TemplatedApp\n  protected readonly transportType = TransportType.WS\n\n  constructor(\n    protected readonly application: ApplicationContext,\n    protected readonly options: WsTransportOptions,\n  ) {\n    this.server = this.options.tls ? SSLApp(options.tls!) : App()\n\n    this.server\n      .get('/healthy', (res) => {\n        // cors\n        res.writeHeader('Access-Control-Allow-Origin', '*')\n        res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')\n        res.writeHeader('Access-Control-Allow-Methods', 'GET')\n        res.writeHeader('Content-Type', 'text/plain')\n        res.end('OK')\n      })\n      .ws<WsUserData>('/api', {\n        sendPingsAutomatically: true,\n        maxPayloadLength: this.options.maxPayloadLength,\n        upgrade: (res, req, context) => {\n          const requestData = getRequestData(req)\n          const container = this.application.container.createScope(\n            Scope.Connection,\n          )\n          const services = requestData.query.getAll('services')\n\n          for (const serviceName of services) {\n            const service = this.application.registry.services.get(serviceName)\n            if (!service)\n              throw new Response(`Service ${service} not found`, {\n                status: 400,\n              })\n            if (this.transportType in service.contract.transports === false)\n              throw new Response(`Service ${service} not supported`, {\n                status: 400,\n              })\n          }\n\n          const data: WsUserData = {\n            id: randomUUID(),\n            format: getFormat(requestData, this.application.format),\n            container,\n            streams: {\n              up: new Map(),\n              down: new Map(),\n              streamId: 0,\n            },\n            abortControllers: new Map(),\n            subscriptions: new Map(),\n            services,\n            data: {\n              query: requestData.query,\n              headers: requestData.headers,\n              proxiedRemoteAddress: Buffer.from(\n                res.getProxiedRemoteAddressAsText(),\n              ).toString(),\n              remoteAddress: Buffer.from(\n                res.getRemoteAddressAsText(),\n              ).toString(),\n            },\n            backpressure: null,\n            context: {} as any,\n          }\n\n          res.upgrade(\n            data,\n            req.getHeader('sec-websocket-key'),\n            req.getHeader('sec-websocket-protocol'),\n            req.getHeader('sec-websocket-extensions'),\n            context,\n          )\n        },\n        open: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.context.decode = this.createDecodeRpcContext(ws)\n          data.context.encode = this.createEncodeRpcContext(ws)\n          data.container.provide(connectionData, data.data)\n          this.application.connections.add({\n            id: data.id,\n            services: data.services,\n            type: this.transportType,\n            subscriptions: data.subscriptions,\n            sendEvent: (service, event, payload) =>\n              sendMessage(ws, MessageType.Event, [service, event, payload]),\n          })\n        },\n        message: async (ws: WsTransportSocket, event) => {\n          const buffer = event as unknown as ArrayBuffer\n          const messageType = decodeNumber(buffer, 'Uint8')\n          if (messageType in this === false) {\n            ws.end(1011, 'Unknown message type')\n          } else {\n            try {\n              await this[messageType](\n                ws,\n                buffer.slice(Uint8Array.BYTES_PER_ELEMENT),\n              )\n            } catch (error: any) {\n              this.logError(error, 'Error while processing message')\n            }\n          }\n        },\n        drain: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n          data.backpressure?.resolve()\n          data.backpressure = null\n        },\n        close: (ws: WsTransportSocket) => {\n          const data = ws.getUserData()\n\n          this.application.connections.remove(data.id)\n          const error = new Error('Connection closed')\n\n          for (const ac of data.abortControllers.values()) {\n            ac.abort(error)\n          }\n          data.abortControllers.clear()\n\n          for (const stream of data.streams.down.values()) {\n            stream.destroy(error)\n          }\n          data.streams.down.clear()\n\n          for (const stream of data.streams.up.values()) {\n            stream.destroy(error)\n          }\n          data.streams.up.clear()\n\n          for (const subscription of data.subscriptions.values()) {\n            subscription.destroy()\n          }\n          data.subscriptions.clear()\n\n          this.handleContainerDisposal(data.container)\n        },\n      })\n  }\n\n  async start() {\n    return new Promise<void>((resolve, reject) => {\n      const hostname = this.options.hostname ?? '127.0.0.1'\n      this.server.listen(hostname, this.options.port!, (socket) => {\n        if (socket) {\n          this.logger.info(\n            'Server started on %s:%s',\n            hostname,\n            this.options.port!,\n          )\n          resolve()\n        } else {\n          reject(new Error('Failed to start server'))\n        }\n      })\n    })\n  }\n\n  async stop() {\n    this.server.close()\n  }\n\n  protected get api() {\n    return this.application.api\n  }\n\n  protected get logger() {\n    return this.application.logger\n  }\n\n  protected async logError(\n    cause: Error,\n    message = 'Unknown error while processing request',\n  ) {\n    this.logger.error(new Error(message, { cause }))\n  }\n\n  protected handleContainerDisposal(container: Container) {\n    container.dispose()\n  }\n\n  protected async handleRPC(options: {\n    connection: Connection\n    service: Service\n    procedure: AnyProcedure\n    container: Container\n    signal: AbortSignal\n    payload: any\n  }) {\n    return await this.api.call({\n      ...options,\n      transport: this.transportType,\n    })\n  }\n\n  protected createEncodeRpcContext(ws: WsTransportSocket): EncodeRpcContext {\n    const data = ws.getUserData()\n    return {\n      addStream: (blob) => {\n        const id = ++data.streams.streamId\n        const downstream = new ServerDownStream(id, blob)\n        downstream.pause()\n        downstream.on('error', console.dir)\n        downstream.once('error', (err) => {\n          console.log({ err })\n          if (downstream.errored?.message !== 'Aborted by client')\n            send(ws, MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        downstream.on('data', (chunk) => {\n          downstream.pause()\n          send(\n            ws,\n            MessageType.DownStreamPush,\n            encodeNumber(id, 'Uint32'),\n            Buffer.from(chunk).buffer as ArrayBuffer,\n          )\n        })\n        data.streams.down.set(id, downstream)\n        return { id, metadata: blob.metadata }\n      },\n      getStream: (id) => {\n        return data.streams.down.get(id)!\n      },\n    }\n  }\n\n  protected createDecodeRpcContext(ws: WsTransportSocket) {\n    const data = ws.getUserData()\n    return {\n      addStream(signal: AbortSignal, id: number, metadata: ApiBlobMetadata) {\n        const upstream = new ServerUpStream(metadata, {\n          read: (size) => {\n            send(\n              ws,\n              MessageType.UpStreamPull,\n              encodeNumber(id, 'Uint32'),\n              encodeNumber(size, 'Uint32'),\n            )\n          },\n        })\n\n        data.streams.up.set(id, upstream)\n        onAbort(signal, () => {\n          upstream.destroy(new Error('Call aborted by client'))\n        })\n        upstream.once('error', (error) => {\n          if (error.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamAbort, encodeNumber(id, 'Uint32'))\n        })\n        upstream.once('close', () => {\n          if (upstream.errored?.message !== 'Aborted by server')\n            send(ws, MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))\n        })\n\n        return upstream\n      },\n      getStream(id) {\n        return data.streams.down.get(id)\n      },\n    }\n  }\n\n  protected async [MessageType.Rpc](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n\n    const connection = this.application.connections.get(data.id)\n    if (!connection) return void ws.end(1011, 'Unknown connection')\n\n    const ac = new AbortController()\n    const rpc = data.format.decoder.decodeRpc(buffer, {\n      ...data.context.decode,\n      addStream: data.context.decode.addStream.bind(null, ac.signal),\n    })\n\n    data.abortControllers.set(rpc.callId, ac)\n\n    const container = data.container.createScope(Scope.Call)\n\n    try {\n      const { service, procedure } = this.api.find(\n        rpc.service,\n        rpc.procedure,\n        this.transportType,\n      )\n\n      const response = await this.handleRPC({\n        connection,\n        service,\n        procedure,\n        container,\n        signal: ac.signal,\n        payload: rpc.payload,\n      })\n\n      if (response instanceof SubscriptionResponse) {\n        sendRpcMessage(ws, MessageType.RpcSubscription, {\n          callId: rpc.callId,\n          error: null,\n          payload: [response.subscription.key, response.payload],\n        })\n\n        response.subscription.on('event', (event, payload) => {\n          sendMessage(ws, MessageType.ServerSubscriptionEvent, [\n            response.subscription.key,\n            event,\n            payload,\n          ])\n        })\n        response.subscription.once('end', () => {\n          sendMessage(ws, MessageType.ServerUnsubscribe, [\n            response.subscription.key,\n          ])\n        })\n      } else {\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: null,\n          payload: response,\n        })\n      }\n    } catch (error) {\n      if (error instanceof ApiError) {\n        this.logger.debug(new Error('Api error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error,\n          payload: null,\n        })\n      } else {\n        this.logger.error(new Error('Unexpected error', { cause: error }))\n        sendRpcMessage(ws, MessageType.Rpc, {\n          callId: rpc.callId,\n          error: InternalError(),\n          payload: null,\n        })\n      }\n    } finally {\n      data.abortControllers.delete(rpc.callId)\n      this.handleContainerDisposal(container)\n    }\n  }\n\n  async [MessageType.UpStreamPush](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(Buffer.from(buffer.slice(Uint32Array.BYTES_PER_ELEMENT)))\n  }\n\n  async [MessageType.UpStreamEnd](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.push(null)\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.UpStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.up.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.up.delete(id)\n  }\n\n  async [MessageType.DownStreamPull](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    await data.backpressure?.promise\n    if (stream.readableEnded)\n      send(ws, MessageType.DownStreamEnd, encodeNumber(id, 'Uint32'))\n    else stream.resume()\n  }\n\n  async [MessageType.DownStreamEnd](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.DownStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const id = decodeNumber(buffer, 'Uint32')\n    const stream = data.streams.down.get(id)\n    if (!stream) return ws.end(1011, 'Unknown stream')\n    stream.destroy(new Error('Aborted by client'))\n    data.streams.down.delete(id)\n  }\n\n  async [MessageType.ClientUnsubscribe](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const data = ws.getUserData()\n    const [key] = data.format.decoder.decode(buffer)\n    const subscription = data.subscriptions.get(key)\n    if (!subscription) return void ws.end()\n    subscription.destroy()\n    data.subscriptions.delete(key)\n  }\n\n  async [MessageType.RpcAbort](ws: WsTransportSocket, buffer: ArrayBuffer) {\n    const data = ws.getUserData()\n    const callId = decodeNumber(buffer, 'Uint32')\n    const ac = data.abortControllers.get(callId)\n    if (ac) ac.abort(new Error('Aborted by client'))\n  }\n}\n"],"names":["randomUUID","ApiError","Scope","ServerDownStream","ServerUpStream","SubscriptionResponse","onAbort","MessageType","TransportType","decodeNumber","encodeNumber","App","SSLApp","connectionData","InternalError","getFormat","getRequestData","send","sendMessage","sendRpcMessage","WsTransportServer","server","transportType","constructor","application","options","WS","tls","get","res","writeHeader","end","ws","sendPingsAutomatically","maxPayloadLength","upgrade","req","context","requestData","container","createScope","Connection","services","query","getAll","serviceName","service","registry","Response","status","contract","transports","data","id","format","streams","up","Map","down","streamId","abortControllers","subscriptions","headers","proxiedRemoteAddress","Buffer","from","getProxiedRemoteAddressAsText","toString","remoteAddress","getRemoteAddressAsText","backpressure","getHeader","open","getUserData","decode","createDecodeRpcContext","encode","createEncodeRpcContext","provide","connections","add","type","sendEvent","event","payload","Event","message","buffer","messageType","slice","Uint8Array","BYTES_PER_ELEMENT","error","logError","drain","resolve","close","remove","Error","ac","values","abort","clear","stream","destroy","subscription","handleContainerDisposal","start","Promise","reject","hostname","listen","port","socket","logger","info","stop","api","cause","dispose","handleRPC","call","transport","addStream","blob","downstream","pause","on","console","dir","once","err","log","errored","DownStreamAbort","chunk","DownStreamPush","set","metadata","getStream","signal","upstream","read","size","UpStreamPull","UpStreamAbort","UpStreamEnd","Rpc","connection","AbortController","rpc","decoder","decodeRpc","bind","callId","Call","procedure","find","response","RpcSubscription","key","ServerSubscriptionEvent","ServerUnsubscribe","debug","delete","UpStreamPush","push","Uint32Array","DownStreamPull","promise","readableEnded","DownStreamEnd","resume","ClientUnsubscribe","RpcAbort"],"mappings":"AAAA,SAASA,UAAU,QAAQ,cAAa;AACxC,SAEEC,QAAQ,EAIRC,KAAK,EACLC,gBAAgB,EAChBC,cAAc,EAEdC,oBAAoB,EACpBC,OAAO,QACF,qBAAoB;AAC3B,SAGEC,WAAW,EACXC,aAAa,EACbC,YAAY,EACZC,YAAY,QACP,gBAAe;AACtB,SAASC,GAAG,EAAEC,MAAM,QAA2B,iBAAgB;AAC/D,SAASC,cAAc,QAAQ,iBAAgB;AAM/C,SACEC,aAAa,EACbC,SAAS,EACTC,cAAc,EACdC,IAAI,EACJC,WAAW,EACXC,cAAc,QACT,aAAY;AAEnB,OAAO,MAAMC;;;IACDC,OAAqB;IACZC,cAAgC;IAEnDC,YACE,AAAmBC,WAA+B,EAClD,AAAmBC,OAA2B,CAC9C;aAFmBD,cAAAA;aACAC,UAAAA;aAJFH,gBAAgBd,cAAckB,EAAE;QAMjD,IAAI,CAACL,MAAM,GAAG,IAAI,CAACI,OAAO,CAACE,GAAG,GAAGf,OAAOa,QAAQE,GAAG,IAAKhB;QAExD,IAAI,CAACU,MAAM,CACRO,GAAG,CAAC,YAAY,CAACC;YAEhBA,IAAIC,WAAW,CAAC,+BAA+B;YAC/CD,IAAIC,WAAW,CAAC,gCAAgC;YAChDD,IAAIC,WAAW,CAAC,gCAAgC;YAChDD,IAAIC,WAAW,CAAC,gBAAgB;YAChCD,IAAIE,GAAG,CAAC;QACV,GACCC,EAAE,CAAa,QAAQ;YACtBC,wBAAwB;YACxBC,kBAAkB,IAAI,CAACT,OAAO,CAACS,gBAAgB;YAC/CC,SAAS,CAACN,KAAKO,KAAKC;gBAClB,MAAMC,cAActB,eAAeoB;gBACnC,MAAMG,YAAY,IAAI,CAACf,WAAW,CAACe,SAAS,CAACC,WAAW,CACtDtC,MAAMuC,UAAU;gBAElB,MAAMC,WAAWJ,YAAYK,KAAK,CAACC,MAAM,CAAC;gBAE1C,KAAK,MAAMC,eAAeH,SAAU;oBAClC,MAAMI,UAAU,IAAI,CAACtB,WAAW,CAACuB,QAAQ,CAACL,QAAQ,CAACd,GAAG,CAACiB;oBACvD,IAAI,CAACC,SACH,MAAM,IAAIE,SAAS,CAAC,QAAQ,EAAEF,QAAQ,UAAU,CAAC,EAAE;wBACjDG,QAAQ;oBACV;oBACF,IAAI,IAAI,CAAC3B,aAAa,IAAIwB,QAAQI,QAAQ,CAACC,UAAU,KAAK,OACxD,MAAM,IAAIH,SAAS,CAAC,QAAQ,EAAEF,QAAQ,cAAc,CAAC,EAAE;wBACrDG,QAAQ;oBACV;gBACJ;gBAEA,MAAMG,OAAmB;oBACvBC,IAAIrD;oBACJsD,QAAQvC,UAAUuB,aAAa,IAAI,CAACd,WAAW,CAAC8B,MAAM;oBACtDf;oBACAgB,SAAS;wBACPC,IAAI,IAAIC;wBACRC,MAAM,IAAID;wBACVE,UAAU;oBACZ;oBACAC,kBAAkB,IAAIH;oBACtBI,eAAe,IAAIJ;oBACnBf;oBACAU,MAAM;wBACJT,OAAOL,YAAYK,KAAK;wBACxBmB,SAASxB,YAAYwB,OAAO;wBAC5BC,sBAAsBC,OAAOC,IAAI,CAC/BpC,IAAIqC,6BAA6B,IACjCC,QAAQ;wBACVC,eAAeJ,OAAOC,IAAI,CACxBpC,IAAIwC,sBAAsB,IAC1BF,QAAQ;oBACZ;oBACAG,cAAc;oBACdjC,SAAS,CAAC;gBACZ;gBAEAR,IAAIM,OAAO,CACTiB,MACAhB,IAAImC,SAAS,CAAC,sBACdnC,IAAImC,SAAS,CAAC,2BACdnC,IAAImC,SAAS,CAAC,6BACdlC;YAEJ;YACAmC,MAAM,CAACxC;gBACL,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAC3BrB,KAAKf,OAAO,CAACqC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC3C;gBAClDoB,KAAKf,OAAO,CAACuC,MAAM,GAAG,IAAI,CAACC,sBAAsB,CAAC7C;gBAClDoB,KAAKb,SAAS,CAACuC,OAAO,CAACjE,gBAAgBuC,KAAKA,IAAI;gBAChD,IAAI,CAAC5B,WAAW,CAACuD,WAAW,CAACC,GAAG,CAAC;oBAC/B3B,IAAID,KAAKC,EAAE;oBACXX,UAAUU,KAAKV,QAAQ;oBACvBuC,MAAM,IAAI,CAAC3D,aAAa;oBACxBuC,eAAeT,KAAKS,aAAa;oBACjCqB,WAAW,CAACpC,SAASqC,OAAOC,UAC1BlE,YAAYc,IAAIzB,YAAY8E,KAAK,EAAE;4BAACvC;4BAASqC;4BAAOC;yBAAQ;gBAChE;YACF;YACAE,SAAS,OAAOtD,IAAuBmD;gBACrC,MAAMI,SAASJ;gBACf,MAAMK,cAAc/E,aAAa8E,QAAQ;gBACzC,IAAIC,eAAe,IAAI,KAAK,OAAO;oBACjCxD,GAAGD,GAAG,CAAC,MAAM;gBACf,OAAO;oBACL,IAAI;wBACF,MAAM,IAAI,CAACyD,YAAY,CACrBxD,IACAuD,OAAOE,KAAK,CAACC,WAAWC,iBAAiB;oBAE7C,EAAE,OAAOC,OAAY;wBACnB,IAAI,CAACC,QAAQ,CAACD,OAAO;oBACvB;gBACF;YACF;YACAE,OAAO,CAAC9D;gBACN,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAC3BrB,KAAKkB,YAAY,EAAEyB;gBACnB3C,KAAKkB,YAAY,GAAG;YACtB;YACA0B,OAAO,CAAChE;gBACN,MAAMoB,OAAOpB,GAAGyC,WAAW;gBAE3B,IAAI,CAACjD,WAAW,CAACuD,WAAW,CAACkB,MAAM,CAAC7C,KAAKC,EAAE;gBAC3C,MAAMuC,QAAQ,IAAIM,MAAM;gBAExB,KAAK,MAAMC,MAAM/C,KAAKQ,gBAAgB,CAACwC,MAAM,GAAI;oBAC/CD,GAAGE,KAAK,CAACT;gBACX;gBACAxC,KAAKQ,gBAAgB,CAAC0C,KAAK;gBAE3B,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACG,IAAI,CAAC0C,MAAM,GAAI;oBAC/CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACG,IAAI,CAAC4C,KAAK;gBAEvB,KAAK,MAAMC,UAAUnD,KAAKG,OAAO,CAACC,EAAE,CAAC4C,MAAM,GAAI;oBAC7CG,OAAOC,OAAO,CAACZ;gBACjB;gBACAxC,KAAKG,OAAO,CAACC,EAAE,CAAC8C,KAAK;gBAErB,KAAK,MAAMG,gBAAgBrD,KAAKS,aAAa,CAACuC,MAAM,GAAI;oBACtDK,aAAaD,OAAO;gBACtB;gBACApD,KAAKS,aAAa,CAACyC,KAAK;gBAExB,IAAI,CAACI,uBAAuB,CAACtD,KAAKb,SAAS;YAC7C;QACF;IACJ;IAEA,MAAMoE,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACb,SAASc;YACjC,MAAMC,WAAW,IAAI,CAACrF,OAAO,CAACqF,QAAQ,IAAI;YAC1C,IAAI,CAACzF,MAAM,CAAC0F,MAAM,CAACD,UAAU,IAAI,CAACrF,OAAO,CAACuF,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAACC,MAAM,CAACC,IAAI,CACd,2BACAL,UACA,IAAI,CAACrF,OAAO,CAACuF,IAAI;oBAEnBjB;gBACF,OAAO;oBACLc,OAAO,IAAIX,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMkB,OAAO;QACX,IAAI,CAAC/F,MAAM,CAAC2E,KAAK;IACnB;IAEA,IAAcqB,MAAM;QAClB,OAAO,IAAI,CAAC7F,WAAW,CAAC6F,GAAG;IAC7B;IAEA,IAAcH,SAAS;QACrB,OAAO,IAAI,CAAC1F,WAAW,CAAC0F,MAAM;IAChC;IAEA,MAAgBrB,SACdyB,KAAY,EACZhC,UAAU,wCAAwC,EAClD;QACA,IAAI,CAAC4B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAMZ,SAAS;YAAEgC;QAAM;IAC/C;IAEUZ,wBAAwBnE,SAAoB,EAAE;QACtDA,UAAUgF,OAAO;IACnB;IAEA,MAAgBC,UAAU/F,OAOzB,EAAE;QACD,OAAO,MAAM,IAAI,CAAC4F,GAAG,CAACI,IAAI,CAAC;YACzB,GAAGhG,OAAO;YACViG,WAAW,IAAI,CAACpG,aAAa;QAC/B;IACF;IAEUuD,uBAAuB7C,EAAqB,EAAoB;QACxE,MAAMoB,OAAOpB,GAAGyC,WAAW;QAC3B,OAAO;YACLkD,WAAW,CAACC;gBACV,MAAMvE,KAAK,EAAED,KAAKG,OAAO,CAACI,QAAQ;gBAClC,MAAMkE,aAAa,IAAI1H,iBAAiBkD,IAAIuE;gBAC5CC,WAAWC,KAAK;gBAChBD,WAAWE,EAAE,CAAC,SAASC,QAAQC,GAAG;gBAClCJ,WAAWK,IAAI,CAAC,SAAS,CAACC;oBACxBH,QAAQI,GAAG,CAAC;wBAAED;oBAAI;oBAClB,IAAIN,WAAWQ,OAAO,EAAE/C,YAAY,qBAClCrE,KAAKe,IAAIzB,YAAY+H,eAAe,EAAE5H,aAAa2C,IAAI;gBAC3D;gBACAwE,WAAWE,EAAE,CAAC,QAAQ,CAACQ;oBACrBV,WAAWC,KAAK;oBAChB7G,KACEe,IACAzB,YAAYiI,cAAc,EAC1B9H,aAAa2C,IAAI,WACjBW,OAAOC,IAAI,CAACsE,OAAOhD,MAAM;gBAE7B;gBACAnC,KAAKG,OAAO,CAACG,IAAI,CAAC+E,GAAG,CAACpF,IAAIwE;gBAC1B,OAAO;oBAAExE;oBAAIqF,UAAUd,KAAKc,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAACtF;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;YAC/B;QACF;IACF;IAEUsB,uBAAuB3C,EAAqB,EAAE;QACtD,MAAMoB,OAAOpB,GAAGyC,WAAW;QAC3B,OAAO;YACLkD,WAAUiB,MAAmB,EAAEvF,EAAU,EAAEqF,QAAyB;gBAClE,MAAMG,WAAW,IAAIzI,eAAesI,UAAU;oBAC5CI,MAAM,CAACC;wBACL9H,KACEe,IACAzB,YAAYyI,YAAY,EACxBtI,aAAa2C,IAAI,WACjB3C,aAAaqI,MAAM;oBAEvB;gBACF;gBAEA3F,KAAKG,OAAO,CAACC,EAAE,CAACiF,GAAG,CAACpF,IAAIwF;gBACxBvI,QAAQsI,QAAQ;oBACdC,SAASrC,OAAO,CAAC,IAAIN,MAAM;gBAC7B;gBACA2C,SAASX,IAAI,CAAC,SAAS,CAACtC;oBACtB,IAAIA,MAAMN,OAAO,KAAK,qBACpBrE,KAAKe,IAAIzB,YAAY0I,aAAa,EAAEvI,aAAa2C,IAAI;gBACzD;gBACAwF,SAASX,IAAI,CAAC,SAAS;oBACrB,IAAIW,SAASR,OAAO,EAAE/C,YAAY,qBAChCrE,KAAKe,IAAIzB,YAAY2I,WAAW,EAAExI,aAAa2C,IAAI;gBACvD;gBAEA,OAAOwF;YACT;YACAF,WAAUtF,EAAE;gBACV,OAAOD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;YAC/B;QACF;IACF;IAEA,MAAgB,CAAC9C,YAAY4I,GAAG,CAAC,CAC/BnH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAE3B,MAAM2E,aAAa,IAAI,CAAC5H,WAAW,CAACuD,WAAW,CAACnD,GAAG,CAACwB,KAAKC,EAAE;QAC3D,IAAI,CAAC+F,YAAY,OAAO,KAAKpH,GAAGD,GAAG,CAAC,MAAM;QAE1C,MAAMoE,KAAK,IAAIkD;QACf,MAAMC,MAAMlG,KAAKE,MAAM,CAACiG,OAAO,CAACC,SAAS,CAACjE,QAAQ;YAChD,GAAGnC,KAAKf,OAAO,CAACqC,MAAM;YACtBiD,WAAWvE,KAAKf,OAAO,CAACqC,MAAM,CAACiD,SAAS,CAAC8B,IAAI,CAAC,MAAMtD,GAAGyC,MAAM;QAC/D;QAEAxF,KAAKQ,gBAAgB,CAAC6E,GAAG,CAACa,IAAII,MAAM,EAAEvD;QAEtC,MAAM5D,YAAYa,KAAKb,SAAS,CAACC,WAAW,CAACtC,MAAMyJ,IAAI;QAEvD,IAAI;YACF,MAAM,EAAE7G,OAAO,EAAE8G,SAAS,EAAE,GAAG,IAAI,CAACvC,GAAG,CAACwC,IAAI,CAC1CP,IAAIxG,OAAO,EACXwG,IAAIM,SAAS,EACb,IAAI,CAACtI,aAAa;YAGpB,MAAMwI,WAAW,MAAM,IAAI,CAACtC,SAAS,CAAC;gBACpC4B;gBACAtG;gBACA8G;gBACArH;gBACAqG,QAAQzC,GAAGyC,MAAM;gBACjBxD,SAASkE,IAAIlE,OAAO;YACtB;YAEA,IAAI0E,oBAAoBzJ,sBAAsB;gBAC5Cc,eAAea,IAAIzB,YAAYwJ,eAAe,EAAE;oBAC9CL,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS;wBAAC0E,SAASrD,YAAY,CAACuD,GAAG;wBAAEF,SAAS1E,OAAO;qBAAC;gBACxD;gBAEA0E,SAASrD,YAAY,CAACsB,EAAE,CAAC,SAAS,CAAC5C,OAAOC;oBACxClE,YAAYc,IAAIzB,YAAY0J,uBAAuB,EAAE;wBACnDH,SAASrD,YAAY,CAACuD,GAAG;wBACzB7E;wBACAC;qBACD;gBACH;gBACA0E,SAASrD,YAAY,CAACyB,IAAI,CAAC,OAAO;oBAChChH,YAAYc,IAAIzB,YAAY2J,iBAAiB,EAAE;wBAC7CJ,SAASrD,YAAY,CAACuD,GAAG;qBAC1B;gBACH;YACF,OAAO;gBACL7I,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO;oBACPR,SAAS0E;gBACX;YACF;QACF,EAAE,OAAOlE,OAAO;YACd,IAAIA,iBAAiB3F,UAAU;gBAC7B,IAAI,CAACiH,MAAM,CAACiD,KAAK,CAAC,IAAIjE,MAAM,aAAa;oBAAEoB,OAAO1B;gBAAM;gBACxDzE,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D;oBACAR,SAAS;gBACX;YACF,OAAO;gBACL,IAAI,CAAC8B,MAAM,CAACtB,KAAK,CAAC,IAAIM,MAAM,oBAAoB;oBAAEoB,OAAO1B;gBAAM;gBAC/DzE,eAAea,IAAIzB,YAAY4I,GAAG,EAAE;oBAClCO,QAAQJ,IAAII,MAAM;oBAClB9D,OAAO9E;oBACPsE,SAAS;gBACX;YACF;QACF,SAAU;YACRhC,KAAKQ,gBAAgB,CAACwG,MAAM,CAACd,IAAII,MAAM;YACvC,IAAI,CAAChD,uBAAuB,CAACnE;QAC/B;IACF;IAEA,MAAM,CAAChC,YAAY8J,YAAY,CAAC,CAACrI,EAAqB,EAAEuD,MAAmB,EAAE;QAC3E,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO+D,IAAI,CAACtG,OAAOC,IAAI,CAACsB,OAAOE,KAAK,CAAC8E,YAAY5E,iBAAiB;IACpE;IAEA,MAAM,CAACpF,YAAY2I,WAAW,CAAC,CAAClH,EAAqB,EAAEuD,MAAmB,EAAE;QAC1E,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAO+D,IAAI,CAAC;QACZlH,KAAKG,OAAO,CAACC,EAAE,CAAC4G,MAAM,CAAC/G;IACzB;IAEA,MAAM,CAAC9C,YAAY0I,aAAa,CAAC,CAC/BjH,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACC,EAAE,CAAC5B,GAAG,CAACyB;QACnC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACC,EAAE,CAAC4G,MAAM,CAAC/G;IACzB;IAEA,MAAM,CAAC9C,YAAYiK,cAAc,CAAC,CAChCxI,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjC,MAAMqB,KAAKkB,YAAY,EAAEmG;QACzB,IAAIlE,OAAOmE,aAAa,EACtBzJ,KAAKe,IAAIzB,YAAYoK,aAAa,EAAEjK,aAAa2C,IAAI;aAClDkD,OAAOqE,MAAM;IACpB;IAEA,MAAM,CAACrK,YAAYoK,aAAa,CAAC,CAC/B3I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCqB,KAAKG,OAAO,CAACG,IAAI,CAAC0G,MAAM,CAAC/G;IAC3B;IAEA,MAAM,CAAC9C,YAAY+H,eAAe,CAAC,CACjCtG,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMpB,KAAK5C,aAAa8E,QAAQ;QAChC,MAAMgB,SAASnD,KAAKG,OAAO,CAACG,IAAI,CAAC9B,GAAG,CAACyB;QACrC,IAAI,CAACkD,QAAQ,OAAOvE,GAAGD,GAAG,CAAC,MAAM;QACjCwE,OAAOC,OAAO,CAAC,IAAIN,MAAM;QACzB9C,KAAKG,OAAO,CAACG,IAAI,CAAC0G,MAAM,CAAC/G;IAC3B;IAEA,MAAM,CAAC9C,YAAYsK,iBAAiB,CAAC,CACnC7I,EAAqB,EACrBuD,MAAmB,EACnB;QACA,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAM,CAACuF,IAAI,GAAG5G,KAAKE,MAAM,CAACiG,OAAO,CAAC7E,MAAM,CAACa;QACzC,MAAMkB,eAAerD,KAAKS,aAAa,CAACjC,GAAG,CAACoI;QAC5C,IAAI,CAACvD,cAAc,OAAO,KAAKzE,GAAGD,GAAG;QACrC0E,aAAaD,OAAO;QACpBpD,KAAKS,aAAa,CAACuG,MAAM,CAACJ;IAC5B;IAEA,MAAM,CAACzJ,YAAYuK,QAAQ,CAAC,CAAC9I,EAAqB,EAAEuD,MAAmB,EAAE;QACvE,MAAMnC,OAAOpB,GAAGyC,WAAW;QAC3B,MAAMiF,SAASjJ,aAAa8E,QAAQ;QACpC,MAAMY,KAAK/C,KAAKQ,gBAAgB,CAAChC,GAAG,CAAC8H;QACrC,IAAIvD,IAAIA,GAAGE,KAAK,CAAC,IAAIH,MAAM;IAC7B;AACF"}

package/dist/lib/transport.js

@@ -0,0 +1,11 @@
+import { Hook, createPlugin } from '@nmtjs/application';
+import { WsTransportServer } from "./server.js";
+export const WsTransport = createPlugin('WsTransport', (app, options)=>{
+    const server = new WsTransportServer(app, options);
+    app.hooks.add(Hook.OnStartup, async ()=>{
+        await server.start();
+    });
+    app.hooks.add(Hook.OnShutdown, async ()=>{
+        await server.stop();
+    });
+});

package/dist/lib/transport.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../lib/transport.ts"],"sourcesContent":["import { Hook, createPlugin } from '@nmtjs/application'\nimport { WsTransportServer } from './server.ts'\nimport type { WsTransportOptions } from './types.ts'\n\nexport const WsTransport = createPlugin<WsTransportOptions>(\n  'WsTransport',\n  (app, options) => {\n    const server = new WsTransportServer(app, options)\n    app.hooks.add(Hook.OnStartup, async () => {\n      await server.start()\n    })\n    app.hooks.add(Hook.OnShutdown, async () => {\n      await server.stop()\n    })\n  },\n)\n"],"names":["Hook","createPlugin","WsTransportServer","WsTransport","app","options","server","hooks","add","OnStartup","start","OnShutdown","stop"],"mappings":"AAAA,SAASA,IAAI,EAAEC,YAAY,QAAQ,qBAAoB;AACvD,SAASC,iBAAiB,QAAQ,cAAa;AAG/C,OAAO,MAAMC,cAAcF,aACzB,eACA,CAACG,KAAKC;IACJ,MAAMC,SAAS,IAAIJ,kBAAkBE,KAAKC;IAC1CD,IAAIG,KAAK,CAACC,GAAG,CAACR,KAAKS,SAAS,EAAE;QAC5B,MAAMH,OAAOI,KAAK;IACpB;IACAN,IAAIG,KAAK,CAACC,GAAG,CAACR,KAAKW,UAAU,EAAE;QAC7B,MAAML,OAAOM,IAAI;IACnB;AACF,GACD"}

package/dist/lib/types.js

@@ -0,0 +1 @@
+export { };

package/dist/lib/types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../lib/types.ts"],"sourcesContent":["import type {\n  Callback,\n  Connection,\n  Container,\n  ServerDownStream,\n  ServerUpStream,\n  Subscription,\n} from '@nmtjs/application'\nimport type {\n  BaseServerDecoder,\n  BaseServerEncoder,\n  BaseServerFormat,\n  DecodeRpcContext,\n  EncodeRpcContext,\n} from '@nmtjs/common'\nimport type { AppOptions, WebSocket } from 'uWebSockets.js'\n\nexport type WsUserData = {\n  id: Connection['id']\n  services: string[]\n  backpressure: null | {\n    promise: Promise<void>\n    resolve: Callback\n  }\n  abortControllers: Map<number, AbortController>\n  streams: {\n    /**\n     * Client to server streams\n     */\n    up: Map<number, ServerUpStream>\n    /**\n     * Server to client streams\n     */\n    down: Map<number, ServerDownStream>\n    streamId: number\n  }\n  subscriptions: Map<string, Subscription>\n  container: Container\n  data: WsConnectionData\n  format: {\n    encoder: BaseServerEncoder\n    decoder: BaseServerDecoder\n  }\n  context: {\n    decode: Omit<DecodeRpcContext, 'addStream'> & {\n      addStream: (\n        signal: AbortSignal,\n        ...args: Parameters<DecodeRpcContext['addStream']>\n      ) => any\n    }\n    encode: EncodeRpcContext\n  }\n}\n\nexport type WsTransportSocket = WebSocket<WsUserData>\n\nexport type WsTransportOptions = {\n  port?: number\n  hostname?: string\n  unix?: string\n  tls?: AppOptions\n  maxPayloadLength?: number\n  maxStreamChunkLength?: number\n  // cors?: ServerOptions['cors']\n}\n\nexport type WsConnectionData = {\n  headers: Map<string, string>\n  query: URLSearchParams\n  remoteAddress: string\n  proxiedRemoteAddress: string\n}\n"],"names":[],"mappings":"AAkEA,WAKC"}

package/dist/lib/utils.js

@@ -0,0 +1,63 @@
+import { ApiError } from '@nmtjs/application';
+import { ErrorCode, concat, encodeNumber } from '@nmtjs/common';
+export const sendMessage = (ws, type, payload)=>{
+    return send(ws, type, ws.getUserData().format.encoder.encode(payload));
+};
+export const sendRpcMessage = (ws, type, rpc)=>{
+    const data = ws.getUserData();
+    return send(ws, type, data.format.encoder.encodeRpc(rpc, data.context.encode));
+};
+export const send = (ws, type, ...buffers)=>{
+    const data = ws.getUserData();
+    try {
+        const result = ws.send(concat(encodeNumber(type, 'Uint8'), ...buffers), true);
+        if (result === 0) {
+            data.backpressure = Promise.withResolvers();
+            return false;
+        }
+        if (result === 2) {
+            return null;
+        }
+        return true;
+    } catch (error) {
+        return null;
+    }
+};
+export const getFormat = ({ headers, query }, format)=>{
+    const contentType = headers.get('content-type') || query.get('content-type');
+    const acceptType = headers.get('accept') || query.get('accept');
+    const encoder = contentType ? format.supportsEncoder(contentType) : undefined;
+    if (!encoder) throw new Error('Unsupported content-type');
+    const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined;
+    if (!decoder) throw new Error('Unsupported accept');
+    return {
+        encoder,
+        decoder
+    };
+};
+export const toRecord = (input)=>{
+    const obj = {};
+    input.forEach((value, key)=>{
+        obj[key] = value;
+    });
+    return obj;
+};
+export const getRequestData = (req)=>{
+    const url = req.getUrl();
+    const method = req.getMethod();
+    const headers = new Map();
+    req.forEach((key, value)=>headers.set(key, value));
+    const query = new URLSearchParams(req.getQuery());
+    const origin = headers.has('origin') ? new URL(url, headers.get('origin')) : null;
+    return {
+        url,
+        origin,
+        method,
+        headers,
+        query
+    };
+};
+export const InternalError = (message = 'Internal Server Error')=>new ApiError(ErrorCode.InternalServerError, message);
+export const NotFoundError = (message = 'Not Found')=>new ApiError(ErrorCode.NotFound, message);
+export const ForbiddenError = (message = 'Forbidden')=>new ApiError(ErrorCode.Forbidden, message);
+export const RequestTimeoutError = (message = 'Request Timeout')=>new ApiError(ErrorCode.RequestTimeout, message);

package/dist/lib/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../lib/utils.ts"],"sourcesContent":["import { ApiError, type Format } from '@nmtjs/application'\nimport {\n  ErrorCode,\n  type RpcResponse,\n  concat,\n  encodeNumber,\n} from '@nmtjs/common'\nimport type { HttpRequest } from 'uWebSockets.js'\nimport type { WsTransportSocket } from './types.ts'\n\nexport const sendMessage = (\n  ws: WsTransportSocket,\n  type: number,\n  payload: any,\n) => {\n  return send(ws, type, ws.getUserData().format.encoder.encode(payload))\n}\n\nexport const sendRpcMessage = (\n  ws: WsTransportSocket,\n  type: number,\n  rpc: RpcResponse,\n) => {\n  const data = ws.getUserData()\n  return send(ws, type, data.format.encoder.encodeRpc(rpc, data.context.encode))\n}\n\nexport const send = (\n  ws: WsTransportSocket,\n  type: number,\n  ...buffers: ArrayBuffer[]\n): boolean | null => {\n  const data = ws.getUserData()\n  try {\n    const result = ws.send(\n      concat(encodeNumber(type, 'Uint8'), ...buffers),\n      true,\n    )\n    if (result === 0) {\n      data.backpressure = Promise.withResolvers()\n      return false\n    }\n    if (result === 2) {\n      return null\n    }\n    return true\n  } catch (error) {\n    return null\n  }\n}\n\nexport const getFormat = ({ headers, query }: RequestData, format: Format) => {\n  const contentType = headers.get('content-type') || query.get('content-type')\n  const acceptType = headers.get('accept') || query.get('accept')\n\n  const encoder = contentType ? format.supportsEncoder(contentType) : undefined\n  if (!encoder) throw new Error('Unsupported content-type')\n\n  const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined\n  if (!decoder) throw new Error('Unsupported accept')\n\n  return {\n    encoder,\n    decoder,\n  }\n}\n\nexport const toRecord = (input: {\n  forEach: (cb: (value, key) => void) => void\n}) => {\n  const obj: Record<string, string> = {}\n  input.forEach((value, key) => {\n    obj[key] = value\n  })\n  return obj\n}\n\ntype RequestData = {\n  url: string\n  origin: URL | null\n  method: string\n  headers: Map<string, string>\n  query: URLSearchParams\n}\n\nexport const getRequestData = (req: HttpRequest): RequestData => {\n  const url = req.getUrl()\n  const method = req.getMethod()\n  const headers = new Map()\n  req.forEach((key, value) => headers.set(key, value))\n  const query = new URLSearchParams(req.getQuery())\n  const origin = headers.has('origin')\n    ? new URL(url, headers.get('origin'))\n    : null\n\n  return {\n    url,\n    origin,\n    method,\n    headers,\n    query,\n  }\n}\n\nexport const InternalError = (message = 'Internal Server Error') =>\n  new ApiError(ErrorCode.InternalServerError, message)\n\nexport const NotFoundError = (message = 'Not Found') =>\n  new ApiError(ErrorCode.NotFound, message)\n\nexport const ForbiddenError = (message = 'Forbidden') =>\n  new ApiError(ErrorCode.Forbidden, message)\n\nexport const RequestTimeoutError = (message = 'Request Timeout') =>\n  new ApiError(ErrorCode.RequestTimeout, message)\n"],"names":["ApiError","ErrorCode","concat","encodeNumber","sendMessage","ws","type","payload","send","getUserData","format","encoder","encode","sendRpcMessage","rpc","data","encodeRpc","context","buffers","result","backpressure","Promise","withResolvers","error","getFormat","headers","query","contentType","get","acceptType","supportsEncoder","undefined","Error","decoder","supportsDecoder","toRecord","input","obj","forEach","value","key","getRequestData","req","url","getUrl","method","getMethod","Map","set","URLSearchParams","getQuery","origin","has","URL","InternalError","message","InternalServerError","NotFoundError","NotFound","ForbiddenError","Forbidden","RequestTimeoutError","RequestTimeout"],"mappings":"AAAA,SAASA,QAAQ,QAAqB,qBAAoB;AAC1D,SACEC,SAAS,EAETC,MAAM,EACNC,YAAY,QACP,gBAAe;AAItB,OAAO,MAAMC,cAAc,CACzBC,IACAC,MACAC;IAEA,OAAOC,KAAKH,IAAIC,MAAMD,GAAGI,WAAW,GAAGC,MAAM,CAACC,OAAO,CAACC,MAAM,CAACL;AAC/D,EAAC;AAED,OAAO,MAAMM,iBAAiB,CAC5BR,IACAC,MACAQ;IAEA,MAAMC,OAAOV,GAAGI,WAAW;IAC3B,OAAOD,KAAKH,IAAIC,MAAMS,KAAKL,MAAM,CAACC,OAAO,CAACK,SAAS,CAACF,KAAKC,KAAKE,OAAO,CAACL,MAAM;AAC9E,EAAC;AAED,OAAO,MAAMJ,OAAO,CAClBH,IACAC,MACA,GAAGY;IAEH,MAAMH,OAAOV,GAAGI,WAAW;IAC3B,IAAI;QACF,MAAMU,SAASd,GAAGG,IAAI,CACpBN,OAAOC,aAAaG,MAAM,aAAaY,UACvC;QAEF,IAAIC,WAAW,GAAG;YAChBJ,KAAKK,YAAY,GAAGC,QAAQC,aAAa;YACzC,OAAO;QACT;QACA,IAAIH,WAAW,GAAG;YAChB,OAAO;QACT;QACA,OAAO;IACT,EAAE,OAAOI,OAAO;QACd,OAAO;IACT;AACF,EAAC;AAED,OAAO,MAAMC,YAAY,CAAC,EAAEC,OAAO,EAAEC,KAAK,EAAe,EAAEhB;IACzD,MAAMiB,cAAcF,QAAQG,GAAG,CAAC,mBAAmBF,MAAME,GAAG,CAAC;IAC7D,MAAMC,aAAaJ,QAAQG,GAAG,CAAC,aAAaF,MAAME,GAAG,CAAC;IAEtD,MAAMjB,UAAUgB,cAAcjB,OAAOoB,eAAe,CAACH,eAAeI;IACpE,IAAI,CAACpB,SAAS,MAAM,IAAIqB,MAAM;IAE9B,MAAMC,UAAUJ,aAAanB,OAAOwB,eAAe,CAACL,cAAcE;IAClE,IAAI,CAACE,SAAS,MAAM,IAAID,MAAM;IAE9B,OAAO;QACLrB;QACAsB;IACF;AACF,EAAC;AAED,OAAO,MAAME,WAAW,CAACC;IAGvB,MAAMC,MAA8B,CAAC;IACrCD,MAAME,OAAO,CAAC,CAACC,OAAOC;QACpBH,GAAG,CAACG,IAAI,GAAGD;IACb;IACA,OAAOF;AACT,EAAC;AAUD,OAAO,MAAMI,iBAAiB,CAACC;IAC7B,MAAMC,MAAMD,IAAIE,MAAM;IACtB,MAAMC,SAASH,IAAII,SAAS;IAC5B,MAAMrB,UAAU,IAAIsB;IACpBL,IAAIJ,OAAO,CAAC,CAACE,KAAKD,QAAUd,QAAQuB,GAAG,CAACR,KAAKD;IAC7C,MAAMb,QAAQ,IAAIuB,gBAAgBP,IAAIQ,QAAQ;IAC9C,MAAMC,SAAS1B,QAAQ2B,GAAG,CAAC,YACvB,IAAIC,IAAIV,KAAKlB,QAAQG,GAAG,CAAC,aACzB;IAEJ,OAAO;QACLe;QACAQ;QACAN;QACApB;QACAC;IACF;AACF,EAAC;AAED,OAAO,MAAM4B,gBAAgB,CAACC,UAAU,uBAAuB,GAC7D,IAAIvD,SAASC,UAAUuD,mBAAmB,EAAED,SAAQ;AAEtD,OAAO,MAAME,gBAAgB,CAACF,UAAU,WAAW,GACjD,IAAIvD,SAASC,UAAUyD,QAAQ,EAAEH,SAAQ;AAE3C,OAAO,MAAMI,iBAAiB,CAACJ,UAAU,WAAW,GAClD,IAAIvD,SAASC,UAAU2D,SAAS,EAAEL,SAAQ;AAE5C,OAAO,MAAMM,sBAAsB,CAACN,UAAU,iBAAiB,GAC7D,IAAIvD,SAASC,UAAU6D,cAAc,EAAEP,SAAQ"}

package/index.ts

@@ -0,0 +1,5 @@
+export * from './lib/server.ts'
+export * from './lib/transport.ts'
+export * from './lib/types.ts'
+export * from './lib/utils.ts'
+export * from './lib/providers.ts'

package/lib/providers.ts

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

package/lib/server.ts

@@ -0,0 +1,469 @@
+import { randomUUID } from 'node:crypto'
+import {
+  type AnyProcedure,
+  ApiError,
+  type ApplicationContext,
+  type Connection,
+  type Container,
+  Scope,
+  ServerDownStream,
+  ServerUpStream,
+  type Service,
+  SubscriptionResponse,
+  onAbort,
+} from '@nmtjs/application'
+import {
+  type ApiBlobMetadata,
+  type EncodeRpcContext,
+  MessageType,
+  TransportType,
+  decodeNumber,
+  encodeNumber,
+} from '@nmtjs/common'
+import { App, SSLApp, type TemplatedApp } from 'uWebSockets.js'
+import { connectionData } from './providers.ts'
+import type {
+  WsTransportOptions,
+  WsTransportSocket,
+  WsUserData,
+} from './types.ts'
+import {
+  InternalError,
+  getFormat,
+  getRequestData,
+  send,
+  sendMessage,
+  sendRpcMessage,
+} from './utils.ts'
+
+export class WsTransportServer {
+  protected server!: TemplatedApp
+  protected readonly transportType = TransportType.WS
+
+  constructor(
+    protected readonly application: ApplicationContext,
+    protected readonly options: WsTransportOptions,
+  ) {
+    this.server = this.options.tls ? SSLApp(options.tls!) : App()
+
+    this.server
+      .get('/healthy', (res) => {
+        // cors
+        res.writeHeader('Access-Control-Allow-Origin', '*')
+        res.writeHeader('Access-Control-Allow-Headers', 'Content-Type')
+        res.writeHeader('Access-Control-Allow-Methods', 'GET')
+        res.writeHeader('Content-Type', 'text/plain')
+        res.end('OK')
+      })
+      .ws<WsUserData>('/api', {
+        sendPingsAutomatically: true,
+        maxPayloadLength: this.options.maxPayloadLength,
+        upgrade: (res, req, context) => {
+          const requestData = getRequestData(req)
+          const container = this.application.container.createScope(
+            Scope.Connection,
+          )
+          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,
+              })
+          }
+
+          const data: WsUserData = {
+            id: randomUUID(),
+            format: getFormat(requestData, this.application.format),
+            container,
+            streams: {
+              up: new Map(),
+              down: new Map(),
+              streamId: 0,
+            },
+            abortControllers: new Map(),
+            subscriptions: new Map(),
+            services,
+            data: {
+              query: requestData.query,
+              headers: requestData.headers,
+              proxiedRemoteAddress: Buffer.from(
+                res.getProxiedRemoteAddressAsText(),
+              ).toString(),
+              remoteAddress: Buffer.from(
+                res.getRemoteAddressAsText(),
+              ).toString(),
+            },
+            backpressure: null,
+            context: {} as any,
+          }
+
+          res.upgrade(
+            data,
+            req.getHeader('sec-websocket-key'),
+            req.getHeader('sec-websocket-protocol'),
+            req.getHeader('sec-websocket-extensions'),
+            context,
+          )
+        },
+        open: (ws: WsTransportSocket) => {
+          const data = ws.getUserData()
+          data.context.decode = this.createDecodeRpcContext(ws)
+          data.context.encode = this.createEncodeRpcContext(ws)
+          data.container.provide(connectionData, data.data)
+          this.application.connections.add({
+            id: data.id,
+            services: data.services,
+            type: this.transportType,
+            subscriptions: data.subscriptions,
+            sendEvent: (service, event, payload) =>
+              sendMessage(ws, MessageType.Event, [service, event, payload]),
+          })
+        },
+        message: async (ws: WsTransportSocket, event) => {
+          const buffer = event as unknown as ArrayBuffer
+          const messageType = decodeNumber(buffer, 'Uint8')
+          if (messageType in this === false) {
+            ws.end(1011, 'Unknown message type')
+          } else {
+            try {
+              await this[messageType](
+                ws,
+                buffer.slice(Uint8Array.BYTES_PER_ELEMENT),
+              )
+            } catch (error: any) {
+              this.logError(error, 'Error while processing message')
+            }
+          }
+        },
+        drain: (ws: WsTransportSocket) => {
+          const data = ws.getUserData()
+          data.backpressure?.resolve()
+          data.backpressure = null
+        },
+        close: (ws: WsTransportSocket) => {
+          const data = ws.getUserData()
+
+          this.application.connections.remove(data.id)
+          const error = new Error('Connection closed')
+
+          for (const ac of data.abortControllers.values()) {
+            ac.abort(error)
+          }
+          data.abortControllers.clear()
+
+          for (const stream of data.streams.down.values()) {
+            stream.destroy(error)
+          }
+          data.streams.down.clear()
+
+          for (const stream of data.streams.up.values()) {
+            stream.destroy(error)
+          }
+          data.streams.up.clear()
+
+          for (const subscription of data.subscriptions.values()) {
+            subscription.destroy()
+          }
+          data.subscriptions.clear()
+
+          this.handleContainerDisposal(data.container)
+        },
+      })
+  }
+
+  async start() {
+    return new Promise<void>((resolve, reject) => {
+      const hostname = this.options.hostname ?? '127.0.0.1'
+      this.server.listen(hostname, this.options.port!, (socket) => {
+        if (socket) {
+          this.logger.info(
+            'Server started on %s:%s',
+            hostname,
+            this.options.port!,
+          )
+          resolve()
+        } else {
+          reject(new Error('Failed to start server'))
+        }
+      })
+    })
+  }
+
+  async stop() {
+    this.server.close()
+  }
+
+  protected get api() {
+    return this.application.api
+  }
+
+  protected get logger() {
+    return this.application.logger
+  }
+
+  protected async logError(
+    cause: Error,
+    message = 'Unknown error while processing request',
+  ) {
+    this.logger.error(new Error(message, { cause }))
+  }
+
+  protected handleContainerDisposal(container: Container) {
+    container.dispose()
+  }
+
+  protected async handleRPC(options: {
+    connection: Connection
+    service: Service
+    procedure: AnyProcedure
+    container: Container
+    signal: AbortSignal
+    payload: any
+  }) {
+    return await this.api.call({
+      ...options,
+      transport: this.transportType,
+    })
+  }
+
+  protected createEncodeRpcContext(ws: WsTransportSocket): EncodeRpcContext {
+    const data = ws.getUserData()
+    return {
+      addStream: (blob) => {
+        const id = ++data.streams.streamId
+        const downstream = new ServerDownStream(id, blob)
+        downstream.pause()
+        downstream.on('error', console.dir)
+        downstream.once('error', (err) => {
+          console.log({ err })
+          if (downstream.errored?.message !== 'Aborted by client')
+            send(ws, MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))
+        })
+        downstream.on('data', (chunk) => {
+          downstream.pause()
+          send(
+            ws,
+            MessageType.DownStreamPush,
+            encodeNumber(id, 'Uint32'),
+            Buffer.from(chunk).buffer as ArrayBuffer,
+          )
+        })
+        data.streams.down.set(id, downstream)
+        return { id, metadata: blob.metadata }
+      },
+      getStream: (id) => {
+        return data.streams.down.get(id)!
+      },
+    }
+  }
+
+  protected createDecodeRpcContext(ws: WsTransportSocket) {
+    const data = ws.getUserData()
+    return {
+      addStream(signal: AbortSignal, id: number, metadata: ApiBlobMetadata) {
+        const upstream = new ServerUpStream(metadata, {
+          read: (size) => {
+            send(
+              ws,
+              MessageType.UpStreamPull,
+              encodeNumber(id, 'Uint32'),
+              encodeNumber(size, 'Uint32'),
+            )
+          },
+        })
+
+        data.streams.up.set(id, upstream)
+        onAbort(signal, () => {
+          upstream.destroy(new Error('Call aborted by client'))
+        })
+        upstream.once('error', (error) => {
+          if (error.message !== 'Aborted by server')
+            send(ws, MessageType.UpStreamAbort, encodeNumber(id, 'Uint32'))
+        })
+        upstream.once('close', () => {
+          if (upstream.errored?.message !== 'Aborted by server')
+            send(ws, MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))
+        })
+
+        return upstream
+      },
+      getStream(id) {
+        return data.streams.down.get(id)
+      },
+    }
+  }
+
+  protected async [MessageType.Rpc](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+
+    const connection = this.application.connections.get(data.id)
+    if (!connection) return void ws.end(1011, 'Unknown connection')
+
+    const ac = new AbortController()
+    const rpc = data.format.decoder.decodeRpc(buffer, {
+      ...data.context.decode,
+      addStream: data.context.decode.addStream.bind(null, ac.signal),
+    })
+
+    data.abortControllers.set(rpc.callId, ac)
+
+    const container = data.container.createScope(Scope.Call)
+
+    try {
+      const { service, procedure } = this.api.find(
+        rpc.service,
+        rpc.procedure,
+        this.transportType,
+      )
+
+      const response = await this.handleRPC({
+        connection,
+        service,
+        procedure,
+        container,
+        signal: ac.signal,
+        payload: rpc.payload,
+      })
+
+      if (response instanceof SubscriptionResponse) {
+        sendRpcMessage(ws, MessageType.RpcSubscription, {
+          callId: rpc.callId,
+          error: null,
+          payload: [response.subscription.key, response.payload],
+        })
+
+        response.subscription.on('event', (event, payload) => {
+          sendMessage(ws, MessageType.ServerSubscriptionEvent, [
+            response.subscription.key,
+            event,
+            payload,
+          ])
+        })
+        response.subscription.once('end', () => {
+          sendMessage(ws, MessageType.ServerUnsubscribe, [
+            response.subscription.key,
+          ])
+        })
+      } else {
+        sendRpcMessage(ws, MessageType.Rpc, {
+          callId: rpc.callId,
+          error: null,
+          payload: response,
+        })
+      }
+    } catch (error) {
+      if (error instanceof ApiError) {
+        this.logger.debug(new Error('Api error', { cause: error }))
+        sendRpcMessage(ws, MessageType.Rpc, {
+          callId: rpc.callId,
+          error,
+          payload: null,
+        })
+      } else {
+        this.logger.error(new Error('Unexpected error', { cause: error }))
+        sendRpcMessage(ws, MessageType.Rpc, {
+          callId: rpc.callId,
+          error: InternalError(),
+          payload: null,
+        })
+      }
+    } finally {
+      data.abortControllers.delete(rpc.callId)
+      this.handleContainerDisposal(container)
+    }
+  }
+
+  async [MessageType.UpStreamPush](ws: WsTransportSocket, buffer: ArrayBuffer) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.up.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    stream.push(Buffer.from(buffer.slice(Uint32Array.BYTES_PER_ELEMENT)))
+  }
+
+  async [MessageType.UpStreamEnd](ws: WsTransportSocket, buffer: ArrayBuffer) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.up.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    stream.push(null)
+    data.streams.up.delete(id)
+  }
+
+  async [MessageType.UpStreamAbort](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.up.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    stream.destroy(new Error('Aborted by client'))
+    data.streams.up.delete(id)
+  }
+
+  async [MessageType.DownStreamPull](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.down.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    await data.backpressure?.promise
+    if (stream.readableEnded)
+      send(ws, MessageType.DownStreamEnd, encodeNumber(id, 'Uint32'))
+    else stream.resume()
+  }
+
+  async [MessageType.DownStreamEnd](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.down.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    data.streams.down.delete(id)
+  }
+
+  async [MessageType.DownStreamAbort](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+    const id = decodeNumber(buffer, 'Uint32')
+    const stream = data.streams.down.get(id)
+    if (!stream) return ws.end(1011, 'Unknown stream')
+    stream.destroy(new Error('Aborted by client'))
+    data.streams.down.delete(id)
+  }
+
+  async [MessageType.ClientUnsubscribe](
+    ws: WsTransportSocket,
+    buffer: ArrayBuffer,
+  ) {
+    const data = ws.getUserData()
+    const [key] = data.format.decoder.decode(buffer)
+    const subscription = data.subscriptions.get(key)
+    if (!subscription) return void ws.end()
+    subscription.destroy()
+    data.subscriptions.delete(key)
+  }
+
+  async [MessageType.RpcAbort](ws: WsTransportSocket, buffer: ArrayBuffer) {
+    const data = ws.getUserData()
+    const callId = decodeNumber(buffer, 'Uint32')
+    const ac = data.abortControllers.get(callId)
+    if (ac) ac.abort(new Error('Aborted by client'))
+  }
+}

package/lib/transport.ts

@@ -0,0 +1,16 @@
+import { Hook, createPlugin } from '@nmtjs/application'
+import { WsTransportServer } from './server.ts'
+import type { WsTransportOptions } from './types.ts'
+
+export const WsTransport = createPlugin<WsTransportOptions>(
+  'WsTransport',
+  (app, options) => {
+    const server = new WsTransportServer(app, options)
+    app.hooks.add(Hook.OnStartup, async () => {
+      await server.start()
+    })
+    app.hooks.add(Hook.OnShutdown, async () => {
+      await server.stop()
+    })
+  },
+)

package/lib/types.ts

@@ -0,0 +1,72 @@
+import type {
+  Callback,
+  Connection,
+  Container,
+  ServerDownStream,
+  ServerUpStream,
+  Subscription,
+} from '@nmtjs/application'
+import type {
+  BaseServerDecoder,
+  BaseServerEncoder,
+  BaseServerFormat,
+  DecodeRpcContext,
+  EncodeRpcContext,
+} from '@nmtjs/common'
+import type { AppOptions, WebSocket } from 'uWebSockets.js'
+
+export type WsUserData = {
+  id: Connection['id']
+  services: string[]
+  backpressure: null | {
+    promise: Promise<void>
+    resolve: Callback
+  }
+  abortControllers: Map<number, AbortController>
+  streams: {
+    /**
+     * Client to server streams
+     */
+    up: Map<number, ServerUpStream>
+    /**
+     * Server to client streams
+     */
+    down: Map<number, ServerDownStream>
+    streamId: number
+  }
+  subscriptions: Map<string, Subscription>
+  container: Container
+  data: WsConnectionData
+  format: {
+    encoder: BaseServerEncoder
+    decoder: BaseServerDecoder
+  }
+  context: {
+    decode: Omit<DecodeRpcContext, 'addStream'> & {
+      addStream: (
+        signal: AbortSignal,
+        ...args: Parameters<DecodeRpcContext['addStream']>
+      ) => any
+    }
+    encode: EncodeRpcContext
+  }
+}
+
+export type WsTransportSocket = WebSocket<WsUserData>
+
+export type WsTransportOptions = {
+  port?: number
+  hostname?: string
+  unix?: string
+  tls?: AppOptions
+  maxPayloadLength?: number
+  maxStreamChunkLength?: number
+  // cors?: ServerOptions['cors']
+}
+
+export type WsConnectionData = {
+  headers: Map<string, string>
+  query: URLSearchParams
+  remoteAddress: string
+  proxiedRemoteAddress: string
+}

package/lib/utils.ts

@@ -0,0 +1,115 @@
+import { ApiError, type Format } from '@nmtjs/application'
+import {
+  ErrorCode,
+  type RpcResponse,
+  concat,
+  encodeNumber,
+} from '@nmtjs/common'
+import type { HttpRequest } from 'uWebSockets.js'
+import type { WsTransportSocket } from './types.ts'
+
+export const sendMessage = (
+  ws: WsTransportSocket,
+  type: number,
+  payload: any,
+) => {
+  return send(ws, type, ws.getUserData().format.encoder.encode(payload))
+}
+
+export const sendRpcMessage = (
+  ws: WsTransportSocket,
+  type: number,
+  rpc: RpcResponse,
+) => {
+  const data = ws.getUserData()
+  return send(ws, type, data.format.encoder.encodeRpc(rpc, data.context.encode))
+}
+
+export const send = (
+  ws: WsTransportSocket,
+  type: number,
+  ...buffers: ArrayBuffer[]
+): boolean | null => {
+  const data = ws.getUserData()
+  try {
+    const result = ws.send(
+      concat(encodeNumber(type, 'Uint8'), ...buffers),
+      true,
+    )
+    if (result === 0) {
+      data.backpressure = Promise.withResolvers()
+      return false
+    }
+    if (result === 2) {
+      return null
+    }
+    return true
+  } catch (error) {
+    return null
+  }
+}
+
+export const getFormat = ({ headers, query }: RequestData, format: Format) => {
+  const contentType = headers.get('content-type') || query.get('content-type')
+  const acceptType = headers.get('accept') || query.get('accept')
+
+  const encoder = contentType ? format.supportsEncoder(contentType) : undefined
+  if (!encoder) throw new Error('Unsupported content-type')
+
+  const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined
+  if (!decoder) throw new Error('Unsupported accept')
+
+  return {
+    encoder,
+    decoder,
+  }
+}
+
+export const toRecord = (input: {
+  forEach: (cb: (value, key) => void) => void
+}) => {
+  const obj: Record<string, string> = {}
+  input.forEach((value, key) => {
+    obj[key] = value
+  })
+  return obj
+}
+
+type RequestData = {
+  url: string
+  origin: URL | null
+  method: string
+  headers: Map<string, string>
+  query: URLSearchParams
+}
+
+export const getRequestData = (req: HttpRequest): RequestData => {
+  const url = req.getUrl()
+  const method = req.getMethod()
+  const headers = new Map()
+  req.forEach((key, value) => headers.set(key, value))
+  const query = new URLSearchParams(req.getQuery())
+  const origin = headers.has('origin')
+    ? new URL(url, headers.get('origin'))
+    : null
+
+  return {
+    url,
+    origin,
+    method,
+    headers,
+    query,
+  }
+}
+
+export const InternalError = (message = 'Internal Server Error') =>
+  new ApiError(ErrorCode.InternalServerError, message)
+
+export const NotFoundError = (message = 'Not Found') =>
+  new ApiError(ErrorCode.NotFound, message)
+
+export const ForbiddenError = (message = 'Forbidden') =>
+  new ApiError(ErrorCode.Forbidden, message)
+
+export const RequestTimeoutError = (message = 'Request Timeout') =>
+  new ApiError(ErrorCode.RequestTimeout, message)

package/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@nmtjs/ws-transport",
+  "type": "module",
+  "exports": {
+    ".": {
+      "default": "./dist/index.js",
+      "types": "./index.ts"
+    }
+  },
+  "dependencies": {
+    "uWebSockets.js": "github:uNetworking/uWebSockets.js#v20.47.0"
+  },
+  "peerDependencies": {
+    "@nmtjs/application": "^0.0.1",
+    "@nmtjs/common": "^0.0.1"
+  },
+  "devDependencies": {
+    "@nmtjs/application": "^0.0.1",
+    "@nmtjs/common": "^0.0.1"
+  },
+  "engines": {
+    "node": ">=18.9.0"
+  },
+  "files": [
+    "index.ts",
+    "lib",
+    "dist",
+    "tsconfig.json",
+    "LICENSE.md",
+    "README.md"
+  ],
+  "version": "0.0.1",
+  "scripts": {
+    "build": "neemata-build ./index.ts './lib/**/*.ts'",
+    "type-check": "tsc --noEmit"
+  }
+}

package/tsconfig.json

@@ -0,0 +1,3 @@
+{
+  "extends": "../../tsconfig.json"
+}