@nmtjs/ws-transport 0.1.7 → 0.2.0

package/dist/lib/injectables.js

@@ -1,2 +1,5 @@
-import { injectables } from '@nmtjs/application';
-export const connectionData = injectables.connectionData;
+import { ProtocolInjectables } from '@nmtjs/protocol/server';
+const connectionData = ProtocolInjectables.connectionData;
+export const WsTransportInjectables = {
+    connectionData
+};

package/dist/lib/injectables.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/injectables.ts"],"sourcesContent":["import {\n  type LazyInjectable,\n  type Scope,\n  injectables,\n} from '@nmtjs/application'\nimport type { WsConnectionData } from './types.ts'\n\nexport const connectionData = injectables.connectionData as LazyInjectable<\n  WsConnectionData,\n  Scope.Connection\n>\n"],"names":["injectables","connectionData"],"mappings":"AAAA,SAGEA,WAAW,QACN,qBAAoB;AAG3B,OAAO,MAAMC,iBAAiBD,YAAYC,cAAc,CAGvD"}
+{"version":3,"sources":["../../../lib/injectables.ts"],"sourcesContent":["import type { LazyInjectable, Scope } from '@nmtjs/core'\nimport { ProtocolInjectables } from '@nmtjs/protocol/server'\nimport type { WsUserData } from './types.ts'\n\nconst connectionData =\n  ProtocolInjectables.connectionData as unknown as LazyInjectable<\n    WsUserData['request'],\n    Scope.Connection\n  >\n\nexport const WsTransportInjectables = {\n  connectionData,\n} as const\n"],"names":["ProtocolInjectables","connectionData","WsTransportInjectables"],"mappings":"AACA,SAASA,mBAAmB,QAAQ,yBAAwB;AAG5D,MAAMC,iBACJD,oBAAoBC,cAAc;AAKpC,OAAO,MAAMC,yBAAyB;IACpCD;AACF,EAAU"}

package/dist/lib/server.js

@@ -1,18 +1,18 @@
 import { randomUUID } from 'node:crypto';
-import { ApiError, Scope, ServerDownStream, ServerUpStream, SubscriptionResponse, builtin, onAbort } from '@nmtjs/application';
-import { MessageType, TransportType, decodeNumber, encodeNumber } from '@nmtjs/common';
 import { App, SSLApp } from 'uWebSockets.js';
-import { connectionData } from "./injectables.js";
-import { InternalError, getFormat, getRequestData, send, sendMessage, sendRpcMessage } from "./utils.js";
+import { createPromise } from '@nmtjs/common';
+import { ClientMessageType, decodeNumber } from '@nmtjs/protocol/common';
+import { ProtocolInjectables } from '@nmtjs/protocol/server';
+import { getRequestData, send } from "./utils.js";
 export class WsTransportServer {
-    application;
+    context;
     options;
     server;
-    transportType;
-    constructor(application, options){
-        this.application = application;
+    clients;
+    constructor(context, options){
+        this.context = context;
         this.options = options;
-        this.transportType = TransportType.WS;
+        this.clients = new Map();
         this.server = this.options.tls ? SSLApp(options.tls) : App();
         this.server.options('/*', (res, req)=>{
             this.applyCors(res, req);
@@ -27,62 +27,51 @@ export class WsTransportServer {
             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) return void res.writeStatus('400 Bad Request').end(`Service ${service} not found`);
-                    if (this.transportType in service.contract.transports === false) return void res.writeStatus('400 Bad Request').end(`Service ${service} not supported`);
-                }
+                const contentType = requestData.headers.get('content-type') || requestData.query.get('content-type');
+                const acceptType = requestData.headers.get('accept') || requestData.query.get('accept');
                 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: {
+                    request: {
                         query: requestData.query,
                         headers: requestData.headers,
                         proxiedRemoteAddress: Buffer.from(res.getProxiedRemoteAddressAsText()).toString(),
-                        remoteAddress: Buffer.from(res.getRemoteAddressAsText()).toString()
+                        remoteAddress: Buffer.from(res.getRemoteAddressAsText()).toString(),
+                        contentType,
+                        acceptType
                     },
+                    opening: createPromise(),
                     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();
-                this.logger.debug('Connection %s opened', data.id);
-                data.context.decode = this.createDecodeRpcContext(ws);
-                data.context.encode = this.createEncodeRpcContext(ws);
-                data.container.provide(connectionData, data.data);
-                const connection = 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
-                        ])
-                });
-                data.container.provide(builtin.connection, connection);
+            open: async (ws)=>{
+                const { id, context, request, opening } = ws.getUserData();
+                this.clients.set(id, ws);
+                this.logger.debug('Connection %s opened', id);
+                try {
+                    const { context: _context, connection } = await this.context.protocol.connections.add(this, {
+                        id,
+                        data: request
+                    }, {
+                        acceptType: request.acceptType,
+                        contentType: request.contentType
+                    });
+                    Object.assign(context, _context);
+                    context.container.provide(ProtocolInjectables.connection, connection);
+                    opening.resolve();
+                } catch (error) {
+                    opening.reject(error);
+                }
             },
-            message: async (ws, event)=>{
-                const buffer = event;
+            message: async (ws, buffer)=>{
+                const { opening } = ws.getUserData();
                 const messageType = decodeNumber(buffer, 'Uint8');
                 if (messageType in this === false) {
                     ws.end(1011, 'Unknown message type');
                 } else {
                     try {
+                        await opening.promise;
                         await this[messageType](ws, buffer.slice(Uint8Array.BYTES_PER_ELEMENT));
                     } catch (error) {
                         this.logError(error, 'Error while processing message');
@@ -94,31 +83,17 @@ export class WsTransportServer {
                 data.backpressure?.resolve();
                 data.backpressure = null;
             },
-            close: (ws, code, message)=>{
-                const data = ws.getUserData();
-                this.logger.debug('Connection %s closed with code %s: %s', data.id, code, Buffer.from(message).toString());
-                this.application.connections.remove(data.id);
-                const error = new Error('Connection closed');
-                for (const ac of data.abortControllers.values()){
-                    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);
+            close: async (ws, code, message)=>{
+                const { id } = ws.getUserData();
+                this.logger.debug('Connection %s closed with code %s: %s', id, code, Buffer.from(message).toString());
+                this.clients.delete(id);
+                await this.protocol.connections.remove(id);
             }
         });
     }
+    send(connection, messageType, buffer) {
+        send(connection.data.ws, messageType, buffer);
+    }
     async start() {
         return new Promise((resolve, reject)=>{
             const hostname = this.options.hostname ?? '127.0.0.1';
@@ -135,11 +110,11 @@ export class WsTransportServer {
     async stop() {
         this.server.close();
     }
-    get api() {
-        return this.application.api;
+    get protocol() {
+        return this.context.protocol;
     }
     get logger() {
-        return this.application.logger;
+        return this.context.logger;
     }
     async logError(cause, message = 'Unknown error while processing request') {
         this.logger.error(new Error(message, {
@@ -154,203 +129,41 @@ export class WsTransportServer {
         res.writeHeader('Access-Control-Allow-Methods', 'GET, POST');
         res.writeHeader('Access-Control-Allow-Credentials', 'true');
     }
-    async handleContainerDisposal(container) {
-        await 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'));
+    [ClientMessageType.Rpc](ws, buffer) {
+        const { id } = ws.getUserData();
+        this.protocol.rpcRaw(id, buffer);
+    }
+    [ClientMessageType.RpcAbort](ws, buffer) {
+        const { id } = ws.getUserData();
+        this.protocol.rpcAbortRaw(id, buffer);
+    }
+    [ClientMessageType.RpcStreamAbort](ws, buffer) {
+        const { id } = ws.getUserData();
+        this.protocol.rpcStreamAbortRaw(id, buffer);
+    }
+    [ClientMessageType.ClientStreamPush](ws, buffer) {
+        const { id } = ws.getUserData();
+        const streamId = decodeNumber(buffer, 'Uint32');
+        this.protocol.clientStreams.push(id, streamId, buffer.slice(Uint32Array.BYTES_PER_ELEMENT));
+    }
+    [ClientMessageType.ClientStreamEnd](ws, buffer) {
+        const { id } = ws.getUserData();
+        const streamId = decodeNumber(buffer, 'Uint32');
+        this.protocol.clientStreams.end(id, streamId);
+    }
+    [ClientMessageType.ClientStreamAbort](ws, buffer) {
+        const { id } = ws.getUserData();
+        const streamId = decodeNumber(buffer, 'Uint32');
+        this.protocol.clientStreams.abort(id, streamId);
+    }
+    [ClientMessageType.ServerStreamPull](ws, buffer) {
+        const { id } = ws.getUserData();
+        const streamId = decodeNumber(buffer, 'Uint32');
+        this.protocol.serverStreams.pull(id, streamId);
+    }
+    [ClientMessageType.ServerStreamAbort](ws, buffer) {
+        const { id } = ws.getUserData();
+        const streamId = decodeNumber(buffer, 'Uint32');
+        this.protocol.serverStreams.abort(id, streamId);
     }
 }

package/dist/lib/server.js.map

@@ -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  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"}
+{"version":3,"sources":["../../../lib/server.ts"],"sourcesContent":["import { randomUUID } from 'node:crypto'\nimport {\n  App,\n  type HttpRequest,\n  type HttpResponse,\n  SSLApp,\n  type TemplatedApp,\n} from 'uWebSockets.js'\n\nimport { createPromise } from '@nmtjs/common'\nimport {\n  ClientMessageType,\n  type ServerMessageType,\n  decodeNumber,\n  decodeText,\n} from '@nmtjs/protocol/common'\nimport {\n  type Connection,\n  ProtocolInjectables,\n  type Transport,\n  type TransportPluginContext,\n} from '@nmtjs/protocol/server'\nimport type {\n  WsTransportOptions,\n  WsTransportSocket,\n  WsUserData,\n} from './types.ts'\nimport { getRequestData, send } from './utils.ts'\n\nexport type WsConnectionData = {\n  ws: WsTransportSocket\n}\n\nexport class WsTransportServer implements Transport<WsConnectionData> {\n  protected server!: TemplatedApp\n  protected clients: Map<string, WsTransportSocket> = new Map()\n\n  constructor(\n    protected readonly context: TransportPluginContext,\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\n          const contentType =\n            requestData.headers.get('content-type') ||\n            requestData.query.get('content-type')\n\n          const acceptType =\n            requestData.headers.get('accept') || requestData.query.get('accept')\n\n          const data: WsUserData = {\n            id: randomUUID(),\n            request: {\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              contentType,\n              acceptType,\n            },\n            opening: createPromise(),\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: async (ws: WsTransportSocket) => {\n          const { id, context, request, opening } = ws.getUserData()\n          this.clients.set(id, ws)\n          this.logger.debug('Connection %s opened', id)\n          try {\n            const { context: _context, connection } =\n              await this.context.protocol.connections.add(\n                this,\n                { id, data: request },\n                {\n                  acceptType: request.acceptType,\n                  contentType: request.contentType,\n                },\n              )\n            Object.assign(context, _context)\n            context.container.provide(\n              ProtocolInjectables.connection,\n              connection,\n            )\n            opening.resolve()\n          } catch (error) {\n            opening.reject(error)\n          }\n        },\n        message: async (ws: WsTransportSocket, buffer) => {\n          const { opening } = ws.getUserData()\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 opening.promise\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: async (ws: WsTransportSocket, code, message) => {\n          const { id } = ws.getUserData()\n\n          this.logger.debug(\n            'Connection %s closed with code %s: %s',\n            id,\n            code,\n            Buffer.from(message).toString(),\n          )\n          this.clients.delete(id)\n          await this.protocol.connections.remove(id)\n        },\n      })\n  }\n\n  send(\n    connection: Connection<WsConnectionData>,\n    messageType: ServerMessageType,\n    buffer: ArrayBuffer,\n  ) {\n    send(connection.data.ws, messageType, buffer)\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 protocol() {\n    return this.context.protocol\n  }\n\n  protected get logger() {\n    return this.context.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 [ClientMessageType.Rpc](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    this.protocol.rpcRaw(id, buffer)\n  }\n\n  protected [ClientMessageType.RpcAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    this.protocol.rpcAbortRaw(id, buffer)\n  }\n\n  protected [ClientMessageType.RpcStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    this.protocol.rpcStreamAbortRaw(id, buffer)\n  }\n\n  protected [ClientMessageType.ClientStreamPush](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    const streamId = decodeNumber(buffer, 'Uint32')\n    this.protocol.clientStreams.push(\n      id,\n      streamId,\n      buffer.slice(Uint32Array.BYTES_PER_ELEMENT),\n    )\n  }\n\n  protected [ClientMessageType.ClientStreamEnd](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    const streamId = decodeNumber(buffer, 'Uint32')\n    this.protocol.clientStreams.end(id, streamId)\n  }\n\n  protected [ClientMessageType.ClientStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    const streamId = decodeNumber(buffer, 'Uint32')\n    this.protocol.clientStreams.abort(id, streamId)\n  }\n\n  protected [ClientMessageType.ServerStreamPull](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    const streamId = decodeNumber(buffer, 'Uint32')\n    this.protocol.serverStreams.pull(id, streamId)\n  }\n\n  protected [ClientMessageType.ServerStreamAbort](\n    ws: WsTransportSocket,\n    buffer: ArrayBuffer,\n  ) {\n    const { id } = ws.getUserData()\n    const streamId = decodeNumber(buffer, 'Uint32')\n    this.protocol.serverStreams.abort(id, streamId)\n  }\n}\n"],"names":["randomUUID","App","SSLApp","createPromise","ClientMessageType","decodeNumber","ProtocolInjectables","getRequestData","send","WsTransportServer","server","clients","constructor","context","options","Map","tls","res","req","applyCors","writeStatus","endWithoutBody","get","writeHeader","end","ws","sendPingsAutomatically","maxPayloadLength","upgrade","requestData","contentType","headers","query","acceptType","data","id","request","proxiedRemoteAddress","Buffer","from","getProxiedRemoteAddressAsText","toString","remoteAddress","getRemoteAddressAsText","opening","backpressure","getHeader","open","getUserData","set","logger","debug","_context","connection","protocol","connections","add","Object","assign","container","provide","resolve","error","reject","message","buffer","messageType","promise","slice","Uint8Array","BYTES_PER_ELEMENT","logError","drain","close","code","delete","remove","start","Promise","hostname","listen","port","socket","info","Error","stop","cause","origin","Rpc","rpcRaw","RpcAbort","rpcAbortRaw","RpcStreamAbort","rpcStreamAbortRaw","ClientStreamPush","streamId","clientStreams","push","Uint32Array","ClientStreamEnd","ClientStreamAbort","abort","ServerStreamPull","serverStreams","pull","ServerStreamAbort"],"mappings":"AAAA,SAASA,UAAU,QAAQ,cAAa;AACxC,SACEC,GAAG,EAGHC,MAAM,QAED,iBAAgB;AAEvB,SAASC,aAAa,QAAQ,gBAAe;AAC7C,SACEC,iBAAiB,EAEjBC,YAAY,QAEP,yBAAwB;AAC/B,SAEEC,mBAAmB,QAGd,yBAAwB;AAM/B,SAASC,cAAc,EAAEC,IAAI,QAAQ,aAAY;AAMjD,OAAO,MAAMC;;;IACDC,OAAqB;IACrBC,QAAmD;IAE7DC,YACE,AAAmBC,OAA+B,EAClD,AAAmBC,OAA2B,CAC9C;aAFmBD,UAAAA;aACAC,UAAAA;aAJXH,UAA0C,IAAII;QAMtD,IAAI,CAACL,MAAM,GAAG,IAAI,CAACI,OAAO,CAACE,GAAG,GAAGd,OAAOY,QAAQE,GAAG,IAAKf;QAExD,IAAI,CAACS,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,KAAKL;gBAClB,MAAMgB,cAActB,eAAeW;gBAEnC,MAAMY,cACJD,YAAYE,OAAO,CAACT,GAAG,CAAC,mBACxBO,YAAYG,KAAK,CAACV,GAAG,CAAC;gBAExB,MAAMW,aACJJ,YAAYE,OAAO,CAACT,GAAG,CAAC,aAAaO,YAAYG,KAAK,CAACV,GAAG,CAAC;gBAE7D,MAAMY,OAAmB;oBACvBC,IAAInC;oBACJoC,SAAS;wBACPJ,OAAOH,YAAYG,KAAK;wBACxBD,SAASF,YAAYE,OAAO;wBAC5BM,sBAAsBC,OAAOC,IAAI,CAC/BtB,IAAIuB,6BAA6B,IACjCC,QAAQ;wBACVC,eAAeJ,OAAOC,IAAI,CACxBtB,IAAI0B,sBAAsB,IAC1BF,QAAQ;wBACVX;wBACAG;oBACF;oBACAW,SAASzC;oBACT0C,cAAc;oBACdhC,SAAS,CAAC;gBACZ;gBAEAI,IAAIW,OAAO,CACTM,MACAhB,IAAI4B,SAAS,CAAC,sBACd5B,IAAI4B,SAAS,CAAC,2BACd5B,IAAI4B,SAAS,CAAC,6BACdjC;YAEJ;YACAkC,MAAM,OAAOtB;gBACX,MAAM,EAAEU,EAAE,EAAEtB,OAAO,EAAEuB,OAAO,EAAEQ,OAAO,EAAE,GAAGnB,GAAGuB,WAAW;gBACxD,IAAI,CAACrC,OAAO,CAACsC,GAAG,CAACd,IAAIV;gBACrB,IAAI,CAACyB,MAAM,CAACC,KAAK,CAAC,wBAAwBhB;gBAC1C,IAAI;oBACF,MAAM,EAAEtB,SAASuC,QAAQ,EAAEC,UAAU,EAAE,GACrC,MAAM,IAAI,CAACxC,OAAO,CAACyC,QAAQ,CAACC,WAAW,CAACC,GAAG,CACzC,IAAI,EACJ;wBAAErB;wBAAID,MAAME;oBAAQ,GACpB;wBACEH,YAAYG,QAAQH,UAAU;wBAC9BH,aAAaM,QAAQN,WAAW;oBAClC;oBAEJ2B,OAAOC,MAAM,CAAC7C,SAASuC;oBACvBvC,QAAQ8C,SAAS,CAACC,OAAO,CACvBtD,oBAAoB+C,UAAU,EAC9BA;oBAEFT,QAAQiB,OAAO;gBACjB,EAAE,OAAOC,OAAO;oBACdlB,QAAQmB,MAAM,CAACD;gBACjB;YACF;YACAE,SAAS,OAAOvC,IAAuBwC;gBACrC,MAAM,EAAErB,OAAO,EAAE,GAAGnB,GAAGuB,WAAW;gBAClC,MAAMkB,cAAc7D,aAAa4D,QAAQ;gBACzC,IAAIC,eAAe,IAAI,KAAK,OAAO;oBACjCzC,GAAGD,GAAG,CAAC,MAAM;gBACf,OAAO;oBACL,IAAI;wBACF,MAAMoB,QAAQuB,OAAO;wBACrB,MAAM,IAAI,CAACD,YAAY,CACrBzC,IACAwC,OAAOG,KAAK,CAACC,WAAWC,iBAAiB;oBAE7C,EAAE,OAAOR,OAAY;wBACnB,IAAI,CAACS,QAAQ,CAACT,OAAO;oBACvB;gBACF;YACF;YACAU,OAAO,CAAC/C;gBACN,MAAMS,OAAOT,GAAGuB,WAAW;gBAC3Bd,KAAKW,YAAY,EAAEgB;gBACnB3B,KAAKW,YAAY,GAAG;YACtB;YACA4B,OAAO,OAAOhD,IAAuBiD,MAAMV;gBACzC,MAAM,EAAE7B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;gBAE7B,IAAI,CAACE,MAAM,CAACC,KAAK,CACf,yCACAhB,IACAuC,MACApC,OAAOC,IAAI,CAACyB,SAASvB,QAAQ;gBAE/B,IAAI,CAAC9B,OAAO,CAACgE,MAAM,CAACxC;gBACpB,MAAM,IAAI,CAACmB,QAAQ,CAACC,WAAW,CAACqB,MAAM,CAACzC;YACzC;QACF;IACJ;IAEA3B,KACE6C,UAAwC,EACxCa,WAA8B,EAC9BD,MAAmB,EACnB;QACAzD,KAAK6C,WAAWnB,IAAI,CAACT,EAAE,EAAEyC,aAAaD;IACxC;IAEA,MAAMY,QAAQ;QACZ,OAAO,IAAIC,QAAc,CAACjB,SAASE;YACjC,MAAMgB,WAAW,IAAI,CAACjE,OAAO,CAACiE,QAAQ,IAAI;YAC1C,IAAI,CAACrE,MAAM,CAACsE,MAAM,CAACD,UAAU,IAAI,CAACjE,OAAO,CAACmE,IAAI,EAAG,CAACC;gBAChD,IAAIA,QAAQ;oBACV,IAAI,CAAChC,MAAM,CAACiC,IAAI,CACd,2BACAJ,UACA,IAAI,CAACjE,OAAO,CAACmE,IAAI;oBAEnBpB;gBACF,OAAO;oBACLE,OAAO,IAAIqB,MAAM;gBACnB;YACF;QACF;IACF;IAEA,MAAMC,OAAO;QACX,IAAI,CAAC3E,MAAM,CAAC+D,KAAK;IACnB;IAEA,IAAcnB,WAAW;QACvB,OAAO,IAAI,CAACzC,OAAO,CAACyC,QAAQ;IAC9B;IAEA,IAAcJ,SAAS;QACrB,OAAO,IAAI,CAACrC,OAAO,CAACqC,MAAM;IAC5B;IAEA,MAAgBqB,SACde,KAAY,EACZtB,UAAU,wCAAwC,EAClD;QACA,IAAI,CAACd,MAAM,CAACY,KAAK,CAAC,IAAIsB,MAAMpB,SAAS;YAAEsB;QAAM;IAC/C;IAEUnE,UAAUF,GAAiB,EAAEC,GAAgB,EAAE;QAEvD,MAAMqE,SAASrE,IAAI4B,SAAS,CAAC;QAC7B,IAAI,CAACyC,QAAQ;QACbtE,IAAIM,WAAW,CAAC,+BAA+BgE;QAC/CtE,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,gCAAgC;QAChDN,IAAIM,WAAW,CAAC,oCAAoC;IACtD;IAEU,CAACnB,kBAAkBoF,GAAG,CAAC,CAC/B/D,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,IAAI,CAACM,QAAQ,CAACmC,MAAM,CAACtD,IAAI8B;IAC3B;IAEU,CAAC7D,kBAAkBsF,QAAQ,CAAC,CACpCjE,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,IAAI,CAACM,QAAQ,CAACqC,WAAW,CAACxD,IAAI8B;IAChC;IAEU,CAAC7D,kBAAkBwF,cAAc,CAAC,CAC1CnE,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,IAAI,CAACM,QAAQ,CAACuC,iBAAiB,CAAC1D,IAAI8B;IACtC;IAEU,CAAC7D,kBAAkB0F,gBAAgB,CAAC,CAC5CrE,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,MAAM+C,WAAW1F,aAAa4D,QAAQ;QACtC,IAAI,CAACX,QAAQ,CAAC0C,aAAa,CAACC,IAAI,CAC9B9D,IACA4D,UACA9B,OAAOG,KAAK,CAAC8B,YAAY5B,iBAAiB;IAE9C;IAEU,CAAClE,kBAAkB+F,eAAe,CAAC,CAC3C1E,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,MAAM+C,WAAW1F,aAAa4D,QAAQ;QACtC,IAAI,CAACX,QAAQ,CAAC0C,aAAa,CAACxE,GAAG,CAACW,IAAI4D;IACtC;IAEU,CAAC3F,kBAAkBgG,iBAAiB,CAAC,CAC7C3E,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,MAAM+C,WAAW1F,aAAa4D,QAAQ;QACtC,IAAI,CAACX,QAAQ,CAAC0C,aAAa,CAACK,KAAK,CAAClE,IAAI4D;IACxC;IAEU,CAAC3F,kBAAkBkG,gBAAgB,CAAC,CAC5C7E,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,MAAM+C,WAAW1F,aAAa4D,QAAQ;QACtC,IAAI,CAACX,QAAQ,CAACiD,aAAa,CAACC,IAAI,CAACrE,IAAI4D;IACvC;IAEU,CAAC3F,kBAAkBqG,iBAAiB,CAAC,CAC7ChF,EAAqB,EACrBwC,MAAmB,EACnB;QACA,MAAM,EAAE9B,EAAE,EAAE,GAAGV,GAAGuB,WAAW;QAC7B,MAAM+C,WAAW1F,aAAa4D,QAAQ;QACtC,IAAI,CAACX,QAAQ,CAACiD,aAAa,CAACF,KAAK,CAAClE,IAAI4D;IACxC;AACF"}

package/dist/lib/transport.js

@@ -1,11 +1,5 @@
-import { Hook, createPlugin } from '@nmtjs/application';
+import { createTransport } from '@nmtjs/protocol/server';
 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();
-    });
+export const WsTransport = createTransport('WsTransport', (context, options)=>{
+    return new WsTransportServer(context, options);
 });

package/dist/lib/transport.js.map

@@ -1 +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"}
+{"version":3,"sources":["../../../lib/transport.ts"],"sourcesContent":["import { createTransport } from '@nmtjs/protocol/server'\nimport { type WsConnectionData, WsTransportServer } from './server.ts'\nimport type { WsTransportOptions } from './types.ts'\n\nexport const WsTransport = createTransport<\n  WsConnectionData,\n  WsTransportOptions\n>('WsTransport', (context, options) => {\n  return new WsTransportServer(context, options)\n})\n"],"names":["createTransport","WsTransportServer","WsTransport","context","options"],"mappings":"AAAA,SAASA,eAAe,QAAQ,yBAAwB;AACxD,SAAgCC,iBAAiB,QAAQ,cAAa;AAGtE,OAAO,MAAMC,cAAcF,gBAGzB,eAAe,CAACG,SAASC;IACzB,OAAO,IAAIH,kBAAkBE,SAASC;AACxC,GAAE"}

package/dist/lib/types.js.map

@@ -1 +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"}
+{"version":3,"sources":["../../../lib/types.ts"],"sourcesContent":["import type { InteractivePromise } from '@nmtjs/common'\nimport type { Connection, ConnectionContext } from '@nmtjs/protocol/server'\nimport type { AppOptions, WebSocket } from 'uWebSockets.js'\n\nexport type WsUserData = {\n  id: Connection['id']\n  opening: InteractivePromise<void>\n  backpressure: InteractivePromise<void> | null\n  request: {\n    headers: Map<string, string>\n    query: URLSearchParams\n    remoteAddress: string\n    proxiedRemoteAddress: string\n    acceptType: string | null\n    contentType: string | null\n  }\n  context: ConnectionContext\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}\n"],"names":[],"mappings":"AAqBA,WAOC"}

package/dist/lib/utils.js

@@ -1,18 +1,13 @@
-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));
-};
+import { createPromise } from '@nmtjs/common';
+import { ErrorCode, concat, encodeNumber } from '@nmtjs/protocol/common';
+import { ProtocolError } from '@nmtjs/protocol/server';
 export const send = (ws, type, ...buffers)=>{
     const data = ws.getUserData();
     try {
-        const result = ws.send(concat(encodeNumber(type, 'Uint8'), ...buffers), true);
+        const buffer = concat(encodeNumber(type, 'Uint8'), ...buffers);
+        const result = ws.send(buffer, true);
         if (result === 0) {
-            data.backpressure = Promise.withResolvers();
+            data.backpressure = createPromise();
             return false;
         }
         if (result === 2) {
@@ -23,18 +18,6 @@ export const send = (ws, type, ...buffers)=>{
         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)=>{
@@ -57,7 +40,7 @@ export const getRequestData = (req)=>{
         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);
+export const InternalError = (message = 'Internal Server Error')=>new ProtocolError(ErrorCode.InternalServerError, message);
+export const NotFoundError = (message = 'Not Found')=>new ProtocolError(ErrorCode.NotFound, message);
+export const ForbiddenError = (message = 'Forbidden')=>new ProtocolError(ErrorCode.Forbidden, message);
+export const RequestTimeoutError = (message = 'Request Timeout')=>new ProtocolError(ErrorCode.RequestTimeout, message);