npm - @nmtjs/ws-client - Versions diffs - 0.1.8 → 0.2.0 - Mend

@nmtjs/ws-client 0.1.8 → 0.2.0

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

Files changed (4) hide show

package/dist/index.js CHANGED Viewed

@@ -1,336 +1,66 @@
-import { ClientTransport, ClientUpStream, Subscription, createClientDownStream, onAbort, once } from '@nmtjs/client';
-import { MessageType, MessageTypeName, TransportType, concat, decodeNumber, encodeNumber } from '@nmtjs/common';
-export class WsClientTransport extends ClientTransport {
+import { EventEmitter } from '@nmtjs/protocol/client';
+import { ServerMessageType, concat, decodeNumber, encodeNumber } from '@nmtjs/protocol/common';
+export class WebSocketClientTransport extends EventEmitter {
     options;
-    type;
-    calls;
-    subscriptions;
-    upStreams;
-    downStreams;
-    upStreamId;
-    queryParams;
-    ws;
-    isHealthy;
-    checkHealthAttempts;
-    autoreconnect;
-    isConnected;
-    wsFactory;
-    encodeRpcContext;
-    decodeRpcContext;
+    #webSocket;
+    #connecting;
     constructor(options){
-        super(), this.options = options, this.type = TransportType.WS, this.calls = new Map(), this.subscriptions = new Map(), this.upStreams = new Map(), this.downStreams = new Map(), this.upStreamId = 0, this.queryParams = {}, this.isHealthy = false, this.checkHealthAttempts = 0, this.isConnected = false;
-        this.autoreconnect = this.options.autoreconnect ?? true;
-        if (options.wsFactory) {
-            this.wsFactory = options.wsFactory;
-        } else {
-            this.wsFactory = (url)=>new WebSocket(url.toString());
-        }
-        this.on('close', (error)=>this.clear(error));
-        this.encodeRpcContext = {
-            addStream: (blob)=>{
-                const id = ++this.upStreamId;
-                const upstream = new ClientUpStream(id, blob);
-                this.upStreams.set(id, upstream);
-                return {
-                    id,
-                    metadata: blob.metadata
-                };
-            },
-            getStream: (id)=>this.upStreams.get(id)
-        };
-        this.decodeRpcContext = {
-            addStream: (id, metadata)=>{
-                const downstream = createClientDownStream(metadata, ()=>this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32')));
-                this.downStreams.set(id, downstream);
-                return downstream.blob;
-            },
-            getStream: (id)=>this.downStreams.get(id)
-        };
-    }
-    async connect() {
-        this.autoreconnect = this.options.autoreconnect ?? true;
-        await this.healthCheck();
-        this.ws = this.wsFactory(this.getURL('api', 'ws', {
-            ...this.queryParams,
-            accept: this.client.format.contentType,
-            'content-type': this.client.format.contentType,
-            services: this.client.services
-        }));
-        this.ws.binaryType = 'arraybuffer';
-        this.ws.onmessage = (event)=>{
-            const buffer = event.data;
+        super(), this.options = options, this.#webSocket = null, this.#connecting = null;
+    }
+    connect(auth = undefined, contentType) {
+        const wsUrl = new URL(this.options.origin);
+        wsUrl.protocol = wsUrl.protocol === 'https:' ? 'wss:' : 'ws:';
+        wsUrl.pathname = '/api';
+        wsUrl.searchParams.set('content-type', contentType);
+        wsUrl.searchParams.set('accept', contentType);
+        if (auth) wsUrl.searchParams.set('auth', auth);
+        const ws = this.options.wsFactory?.(wsUrl) ?? new WebSocket(wsUrl.toString());
+        ws.binaryType = 'arraybuffer';
+        ws.addEventListener('message', ({ data })=>{
+            const buffer = data;
             const type = decodeNumber(buffer, 'Uint8');
-            const handler = this[type];
-            if (handler) {
-                handler.call(this, buffer.slice(Uint8Array.BYTES_PER_ELEMENT), this.ws);
-            }
-        };
-        this.ws.onopen = (event)=>{
-            if (this.options.debug) console.log(`[WS] Connected to ${this.options.origin}`);
-            this.isConnected = true;
-            this.emit('open');
-            this.checkHealthAttempts = 0;
-        };
-        this.ws.onclose = (event)=>{
-            this.isConnected = false;
-            this.isHealthy = false;
-            this.emit('close', event.code === 1000 ? undefined : new Error(`Connection closed with code ${event.code}: ${event.reason}`));
-            if (this.autoreconnect) this.connect();
-        };
-        this.ws.onerror = (cause)=>{
-            this.isHealthy = false;
-            this.emit('close', new Error('Connection error', {
-                cause
-            }));
-        };
-        await once(this, 'open');
-        if (this.options.debug) console.log(`2[WS] Connected to ${this.options.origin}`);
-        this.emit('connect');
-    }
-    async disconnect() {
-        this.autoreconnect = false;
-        this.ws?.close();
-        await once(this, 'close');
-    }
-    async rpc(call) {
-        const { signal, callId, payload, procedure, service } = call;
-        const data = this.client.format.encodeRpc({
-            callId,
-            service,
-            procedure,
-            payload
-        }, this.encodeRpcContext);
-        onAbort(signal, ()=>{
-            const call = this.calls.get(callId);
-            if (call) {
-                const { reject } = call;
-                reject(new Error('Request aborted'));
-                this.calls.delete(callId);
-                this.send(MessageType.RpcAbort, encodeNumber(callId, 'Uint32'));
+            if (type in ServerMessageType) {
+                this.emit(`${type}`, buffer.slice(Uint8Array.BYTES_PER_ELEMENT));
             }
         });
-        if (!this.isConnected) await once(this, 'connect');
+        this.#webSocket = ws;
         return new Promise((resolve, reject)=>{
-            this.calls.set(callId, {
-                resolve,
-                reject
+            ws.addEventListener('open', ()=>{
+                this.emit('connect');
+                resolve();
+            }, {
+                once: true
             });
-            this.send(MessageType.Rpc, data);
-        });
-    }
-    setQueryParams(params) {
-        this.queryParams = params;
-    }
-    send(type, ...payload) {
-        this.ws?.send(concat(encodeNumber(type, 'Uint8'), ...payload));
-        if (this.options.debug) {
-            console.groupCollapsed(`[WS] Sent ${MessageTypeName[type]}`);
-            console.trace();
-            console.groupEnd();
-        }
-    }
-    async clear(error = new Error('Connection closed')) {
-        for (const call of this.calls.values()){
-            const { reject } = call;
-            reject(error);
-        }
-        for (const stream of this.upStreams.values()){
-            stream.reader.cancel(error);
-        }
-        for (const stream of this.downStreams.values()){
-            stream.writer.abort(error);
-        }
-        for (const subscription of this.subscriptions.values()){
-            subscription.unsubscribe();
-        }
-        this.calls.clear();
-        this.upStreams.clear();
-        this.downStreams.clear();
-        this.subscriptions.clear();
-    }
-    async healthCheck() {
-        while(!this.isHealthy){
-            try {
-                const signal = AbortSignal.timeout(10000);
-                const url = this.getURL('healthy', 'http', {
-                    auth: this.client.auth
-                });
-                const { ok } = await fetch(url, {
-                    signal
-                });
-                this.isHealthy = ok;
-            } catch (e) {}
-            if (!this.isHealthy) {
-                this.checkHealthAttempts++;
-                const seconds = Math.min(this.checkHealthAttempts, 15);
-                await new Promise((r)=>setTimeout(r, seconds * 1000));
-            }
-        }
-        this.emit('healthy');
-    }
-    getURL(path = '', protocol, params = {}) {
-        const base = new URL(this.options.origin);
-        const secure = base.protocol === 'https:';
-        const url = new URL(`${secure ? protocol + 's' : protocol}://${base.host}/${path}`);
-        for (let [key, values] of Object.entries(params)){
-            if (!Array.isArray(values)) values = [
-                values
-            ];
-            for (const value of values){
-                url.searchParams.append(key, value);
-            }
-        }
-        return url;
-    }
-    resolveRpc(callId, value) {
-        const call = this.calls.get(callId);
-        if (call) {
-            const { resolve } = call;
-            this.calls.delete(callId);
-            resolve(value);
-        }
-    }
-    async [MessageType.Event](buffer) {
-        const [service, event, payload] = this.client.format.decode(buffer);
-        if (this.options.debug) {
-            console.groupCollapsed(`[WS] Received "Event" ${service}/${event}`);
-            console.log(payload);
-            console.groupEnd();
-        }
-        this.emit('event', service, event, payload);
-    }
-    async [MessageType.Rpc](buffer) {
-        const { callId, error, payload } = this.client.format.decodeRpc(buffer, this.decodeRpcContext);
-        if (this.calls.has(callId)) {
-            this.resolveRpc(callId, error ? {
-                success: false,
-                error
-            } : {
-                success: true,
-                value: payload
+            ws.addEventListener('error', (event)=>reject(new Error('WebSocket error', {
+                    cause: event
+                })), {
+                once: true
             });
-        }
-    }
-    async [MessageType.UpStreamPull](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        const size = decodeNumber(buffer, 'Uint32', Uint32Array.BYTES_PER_ELEMENT);
-        if (this.options.debug) {
-            console.log(`[WS] Received "UpStreamPull" ${id}`);
-        }
-        const stream = this.upStreams.get(id);
-        if (stream) {
-            const buf = new Uint8Array(new ArrayBuffer(size));
-            const { done, value } = await stream.reader.read(buf);
-            if (done) {
-                this.send(MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'));
-            } else {
-                this.send(MessageType.UpStreamPush, concat(encodeNumber(id, 'Uint32'), value.buffer.slice(0, value.byteLength)));
-            }
-        }
-    }
-    async [MessageType.UpStreamEnd](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        if (this.options.debug) {
-            console.log(`[WS] Received "UpStreamEnd" ${id}`);
-        }
-        const stream = this.upStreams.get(id);
-        if (stream) {
-            stream.reader.cancel();
-            this.upStreams.delete(id);
-        }
-    }
-    [MessageType.UpStreamAbort](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        if (this.options.debug) {
-            console.log(`[WS] Received "UpStreamAbort" ${id}`);
-        }
-        const stream = this.upStreams.get(id);
-        if (stream) {
-            try {
-                stream.reader.cancel(new Error('Aborted by server'));
-            } finally{
-                this.upStreams.delete(id);
-            }
-        }
-    }
-    async [MessageType.DownStreamPush](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        if (this.options.debug) {
-            console.log(`[WS] Received "DownStreamPush" ${id}`);
-        }
-        const stream = this.downStreams.get(id);
-        if (stream) {
-            try {
-                await stream.writer.ready;
-                const chunk = buffer.slice(Uint32Array.BYTES_PER_ELEMENT);
-                await stream.writer.write(new Uint8Array(chunk));
-                await stream.writer.ready;
-                this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32'));
-            } catch (e) {
-                this.send(MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'));
-                this.downStreams.delete(id);
-            }
-        }
-    }
-    async [MessageType.DownStreamEnd](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        if (this.options.debug) {
-            console.log(`[WS] Received "DownStreamEnd" ${id}`);
-        }
-        const stream = this.downStreams.get(id);
-        if (stream) {
-            this.downStreams.delete(id);
-            stream.writer.close().catch(()=>{});
-        }
-    }
-    async [MessageType.DownStreamAbort](buffer) {
-        const id = decodeNumber(buffer, 'Uint32');
-        if (this.options.debug) {
-            console.log(`[WS] Received "DownStreamAbort" ${id}`);
-        }
-        const stream = this.downStreams.get(id);
-        if (stream) {
-            this.downStreams.delete(id);
-            stream.writer.abort(new Error('Aborted by server')).catch(()=>{});
-        }
-    }
-    [MessageType.RpcSubscription](buffer) {
-        const { callId, payload: [key, payload] } = this.client.format.decodeRpc(buffer, this.decodeRpcContext);
-        if (this.calls.has(callId)) {
-            const subscription = new Subscription(key, ()=>{
-                subscription.emit('end');
-                this.subscriptions.delete(key);
-                this.send(MessageType.ClientUnsubscribe, this.client.format.encode([
-                    key
-                ]));
-            });
-            this.subscriptions.set(key, subscription);
-            this.resolveRpc(callId, {
-                success: true,
-                value: {
-                    payload,
-                    subscription
+            ws.addEventListener('close', (event)=>{
+                this.emit('disconnect');
+                this.#webSocket = null;
+                if (this.options.autoreconnect === true) {
+                    setTimeout(()=>this.connect(auth, contentType), 1000);
                 }
+            }, {
+                once: true
             });
-        }
+        });
     }
-    [MessageType.ServerSubscriptionEvent](buffer) {
-        const [key, event, payload] = this.client.format.decode(buffer);
-        if (this.options.debug) {
-            console.groupCollapsed(`[WS] Received "ServerSubscriptionEvent" ${key}/${event}`);
-            console.log(payload);
-            console.groupEnd();
-        }
-        const subscription = this.subscriptions.get(key);
-        if (subscription) subscription.emit(event, payload);
+    async disconnect() {
+        if (this.#webSocket === null) return;
+        this.#webSocket.close();
+        return _once(this.#webSocket, 'close');
     }
-    [MessageType.ServerUnsubscribe](buffer) {
-        const [key] = this.client.format.decode(buffer);
-        if (this.options.debug) {
-            console.log(`[WS] Received "ServerUnsubscribe" ${key}`);
-        }
-        const subscription = this.subscriptions.get(key);
-        subscription?.emit('end');
-        this.subscriptions.delete(key);
+    async send(messageType, buffer) {
+        if (this.#connecting) await this.#connecting;
+        this.#webSocket.send(concat(encodeNumber(messageType, 'Uint8'), buffer));
     }
 }
+function _once(target, event) {
+    return new Promise((resolve)=>{
+        target.addEventListener(event, ()=>resolve(), {
+            once: true
+        });
+    });
+}

package/dist/index.js.map CHANGED Viewed

	@@ -1 +1 @@
1	- {"version":3,"sources":["../../index.ts"],"sourcesContent":["import {\n type ClientDownStreamWrapper,\n ClientTransport,\n type ClientTransportRpcCall,\n type ClientTransportRpcResult,\n ClientUpStream,\n Subscription,\n createClientDownStream,\n onAbort,\n once,\n} from '@nmtjs/client'\nimport {\n type DecodeRpcContext,\n type EncodeRpcContext,\n MessageType,\n MessageTypeName,\n TransportType,\n concat,\n decodeNumber,\n encodeNumber,\n} from '@nmtjs/common'\n\nexport type WsClientTransportOptions = {\n /*\n The origin of the server\n * @example 'http://localhost:3000'\n /\n origin: string\n /\n Whether to autoreconnect on close\n * @default true\n /\n autoreconnect?: boolean\n /\n Custom WebSocket class\n * @default globalThis.WebSocket\n /\n wsFactory?: (url: URL) => WebSocket\n\n debug?: boolean\n}\n\ntype WsCall = {\n resolve: (value: ClientTransportRpcResult) => void\n reject: (error: Error) => void\n}\n\nexport class WsClientTransport extends ClientTransport<{\n open: []\n close: [Error?]\n connect: []\n healthy: []\n}> {\n readonly type = TransportType.WS\n\n protected readonly calls = new Map<number, WsCall>()\n protected readonly subscriptions = new Map<number, Subscription>()\n protected readonly upStreams = new Map<number, ClientUpStream>()\n protected readonly downStreams = new Map<number, ClientDownStreamWrapper>()\n protected upStreamId = 0\n protected queryParams = {}\n protected ws?: WebSocket\n protected isHealthy = false\n protected checkHealthAttempts = 0\n protected autoreconnect: boolean\n protected isConnected = false\n\n private wsFactory!: (url: URL) => WebSocket\n private encodeRpcContext: EncodeRpcContext\n private decodeRpcContext: DecodeRpcContext\n\n constructor(private readonly options: WsClientTransportOptions) {\n super()\n\n this.autoreconnect = this.options.autoreconnect ?? true\n\n if (options.wsFactory) {\n this.wsFactory = options.wsFactory\n } else {\n this.wsFactory = (url: URL) => new WebSocket(url.toString())\n }\n\n // TODO: wtf is this for?\n this.on('close', (error) => this.clear(error))\n\n this.encodeRpcContext = {\n addStream: (blob) => {\n const id = ++this.upStreamId\n const upstream = new ClientUpStream(id, blob)\n this.upStreams.set(id, upstream)\n return { id, metadata: blob.metadata }\n },\n getStream: (id) => this.upStreams.get(id),\n }\n\n this.decodeRpcContext = {\n addStream: (id, metadata) => {\n const downstream = createClientDownStream(metadata, () =>\n this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32')),\n )\n this.downStreams.set(id, downstream)\n return downstream.blob\n },\n getStream: (id) => this.downStreams.get(id),\n }\n }\n\n async connect() {\n // reset default autoreconnect value\n this.autoreconnect = this.options.autoreconnect ?? true\n await this.healthCheck()\n\n this.ws = this.wsFactory(\n this.getURL('api', 'ws', {\n ...this.queryParams,\n accept: this.client.format.contentType,\n 'content-type': this.client.format.contentType,\n services: this.client.services,\n }),\n )\n\n this.ws.binaryType = 'arraybuffer'\n\n this.ws.onmessage = (event) => {\n const buffer: ArrayBuffer = event.data\n const type = decodeNumber(buffer, 'Uint8')\n const handler = this[type]\n if (handler) {\n handler.call(this, buffer.slice(Uint8Array.BYTES_PER_ELEMENT), this.ws)\n }\n }\n this.ws.onopen = (event) => {\n if (this.options.debug)\n console.log(`[WS] Connected to ${this.options.origin}`)\n this.isConnected = true\n this.emit('open')\n this.checkHealthAttempts = 0\n }\n this.ws.onclose = (event) => {\n this.isConnected = false\n this.isHealthy = false\n this.emit(\n 'close',\n event.code === 1000\n ? undefined\n : new Error(\n `Connection closed with code ${event.code}: ${event.reason}`,\n ),\n )\n // FIXME: cleanup calls, streams, subscriptions\n if (this.autoreconnect) this.connect()\n }\n this.ws.onerror = (cause) => {\n this.isHealthy = false\n this.emit('close', new Error('Connection error', { cause }))\n }\n await once(this, 'open')\n if (this.options.debug)\n console.log(`2[WS] Connected to ${this.options.origin}`)\n this.emit('connect')\n }\n\n async disconnect(): Promise<void> {\n this.autoreconnect = false\n this.ws?.close()\n await once(this, 'close')\n }\n\n async rpc(call: ClientTransportRpcCall): Promise<ClientTransportRpcResult> {\n const { signal, callId, payload, procedure, service } = call\n\n const data = this.client.format.encodeRpc(\n {\n callId,\n service,\n procedure,\n payload,\n },\n this.encodeRpcContext,\n )\n\n onAbort(signal, () => {\n const call = this.calls.get(callId)\n if (call) {\n const { reject } = call\n reject(new Error('Request aborted'))\n this.calls.delete(callId)\n this.send(MessageType.RpcAbort, encodeNumber(callId, 'Uint32'))\n }\n })\n\n if (!this.isConnected) await once(this, 'connect')\n\n return new Promise((resolve, reject) => {\n this.calls.set(callId, { resolve, reject })\n this.send(MessageType.Rpc, data)\n })\n }\n\n setQueryParams(params: Record<string, any>) {\n this.queryParams = params\n }\n\n protected send(type: MessageType, ...payload: ArrayBuffer[]) {\n this.ws?.send(concat(encodeNumber(type, 'Uint8'), ...payload))\n if (this.options.debug) {\n console.groupCollapsed(`[WS] Sent ${MessageTypeName[type]}`)\n console.trace()\n console.groupEnd()\n }\n }\n\n protected async clear(error: Error = new Error('Connection closed')) {\n for (const call of this.calls.values()) {\n const { reject } = call\n reject(error)\n }\n\n for (const stream of this.upStreams.values()) {\n stream.reader.cancel(error)\n }\n\n for (const stream of this.downStreams.values()) {\n stream.writer.abort(error)\n }\n\n for (const subscription of this.subscriptions.values()) {\n subscription.unsubscribe()\n }\n\n this.calls.clear()\n this.upStreams.clear()\n this.downStreams.clear()\n this.subscriptions.clear()\n }\n\n protected async healthCheck() {\n while (!this.isHealthy) {\n try {\n const signal = AbortSignal.timeout(10000)\n const url = this.getURL('healthy', 'http', {\n auth: this.client.auth,\n })\n const { ok } = await fetch(url, { signal })\n this.isHealthy = ok\n } catch (e) {}\n\n if (!this.isHealthy) {\n this.checkHealthAttempts++\n const seconds = Math.min(this.checkHealthAttempts, 15)\n await new Promise((r) => setTimeout(r, seconds 1000))\n }\n }\n this.emit('healthy')\n }\n\n private getURL(\n path = '',\n protocol: 'ws' \| 'http',\n params: Record<string, any> = {},\n ) {\n // TODO: add custom path support?\n const base = new URL(this.options.origin)\n const secure = base.protocol === 'https:'\n const url = new URL(\n `${secure ? protocol + 's' : protocol}://${base.host}/${path}`,\n )\n for (let [key, values] of Object.entries(params)) {\n if (!Array.isArray(values)) values = [values]\n for (const value of values) {\n url.searchParams.append(key, value)\n }\n }\n return url\n }\n\n protected resolveRpc(callId: number, value: ClientTransportRpcResult) {\n const call = this.calls.get(callId)\n if (call) {\n const { resolve } = call\n this.calls.delete(callId)\n resolve(value)\n }\n }\n\n protected async [MessageType.Event](buffer: ArrayBuffer) {\n const [service, event, payload] = this.client.format.decode(buffer)\n if (this.options.debug) {\n console.groupCollapsed(`[WS] Received \"Event\" ${service}/${event}`)\n console.log(payload)\n console.groupEnd()\n }\n this.emit('event', service, event, payload)\n }\n\n protected async [MessageType.Rpc](buffer: ArrayBuffer) {\n const { callId, error, payload } = this.client.format.decodeRpc(\n buffer,\n this.decodeRpcContext,\n )\n\n if (this.calls.has(callId)) {\n this.resolveRpc(\n callId,\n error ? { success: false, error } : { success: true, value: payload },\n )\n }\n }\n\n protected async [MessageType.UpStreamPull](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n const size = decodeNumber(buffer, 'Uint32', Uint32Array.BYTES_PER_ELEMENT)\n\n if (this.options.debug) {\n console.log(`[WS] Received \"UpStreamPull\" ${id}`)\n }\n\n const stream = this.upStreams.get(id)\n if (stream) {\n const buf = new Uint8Array(new ArrayBuffer(size))\n const { done, value } = await stream.reader.read(buf)\n if (done) {\n this.send(MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))\n } else {\n this.send(\n MessageType.UpStreamPush,\n concat(\n encodeNumber(id, 'Uint32'),\n value.buffer.slice(0, value.byteLength),\n ),\n )\n }\n }\n }\n\n protected async [MessageType.UpStreamEnd](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n if (this.options.debug) {\n console.log(`[WS] Received \"UpStreamEnd\" ${id}`)\n }\n const stream = this.upStreams.get(id)\n if (stream) {\n stream.reader.cancel()\n this.upStreams.delete(id)\n }\n }\n\n protected [MessageType.UpStreamAbort](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n if (this.options.debug) {\n console.log(`[WS] Received \"UpStreamAbort\" ${id}`)\n }\n const stream = this.upStreams.get(id)\n if (stream) {\n try {\n stream.reader.cancel(new Error('Aborted by server'))\n } finally {\n this.upStreams.delete(id)\n }\n }\n }\n\n protected async [MessageType.DownStreamPush](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n if (this.options.debug) {\n console.log(`[WS] Received \"DownStreamPush\" ${id}`)\n }\n const stream = this.downStreams.get(id)\n if (stream) {\n try {\n await stream.writer.ready\n const chunk = buffer.slice(Uint32Array.BYTES_PER_ELEMENT)\n await stream.writer.write(new Uint8Array(chunk))\n await stream.writer.ready\n this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32'))\n } catch (e) {\n this.send(MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))\n this.downStreams.delete(id)\n }\n }\n }\n\n protected async [MessageType.DownStreamEnd](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n if (this.options.debug) {\n console.log(`[WS] Received \"DownStreamEnd\" ${id}`)\n }\n const stream = this.downStreams.get(id)\n if (stream) {\n this.downStreams.delete(id)\n stream.writer.close().catch(() => {})\n }\n }\n\n protected async [MessageType.DownStreamAbort](buffer: ArrayBuffer) {\n const id = decodeNumber(buffer, 'Uint32')\n if (this.options.debug) {\n console.log(`[WS] Received \"DownStreamAbort\" ${id}`)\n }\n const stream = this.downStreams.get(id)\n if (stream) {\n this.downStreams.delete(id)\n stream.writer.abort(new Error('Aborted by server')).catch(() => {})\n }\n }\n\n protected [MessageType.RpcSubscription](buffer: ArrayBuffer) {\n const {\n callId,\n payload: [key, payload],\n } = this.client.format.decodeRpc(buffer, this.decodeRpcContext)\n if (this.calls.has(callId)) {\n const subscription = new Subscription(key, () => {\n subscription.emit('end')\n this.subscriptions.delete(key)\n this.send(\n MessageType.ClientUnsubscribe,\n this.client.format.encode([key]),\n )\n })\n this.subscriptions.set(key, subscription)\n this.resolveRpc(callId, {\n success: true,\n value: { payload, subscription },\n })\n }\n }\n\n protected [MessageType.ServerSubscriptionEvent](buffer: ArrayBuffer) {\n const [key, event, payload] = this.client.format.decode(buffer)\n if (this.options.debug) {\n console.groupCollapsed(\n `[WS] Received \"ServerSubscriptionEvent\" ${key}/${event}`,\n )\n console.log(payload)\n console.groupEnd()\n }\n const subscription = this.subscriptions.get(key)\n if (subscription) subscription.emit(event, payload)\n }\n\n protected [MessageType.ServerUnsubscribe](buffer: ArrayBuffer) {\n const [key] = this.client.format.decode(buffer)\n if (this.options.debug) {\n console.log(`[WS] Received \"ServerUnsubscribe\" ${key}`)\n }\n const subscription = this.subscriptions.get(key)\n subscription?.emit('end')\n this.subscriptions.delete(key)\n }\n}\n"],"names":["ClientTransport","ClientUpStream","Subscription","createClientDownStream","onAbort","once","MessageType","MessageTypeName","TransportType","concat","decodeNumber","encodeNumber","WsClientTransport","type","calls","subscriptions","upStreams","downStreams","upStreamId","queryParams","ws","isHealthy","checkHealthAttempts","autoreconnect","isConnected","wsFactory","encodeRpcContext","decodeRpcContext","constructor","options","WS","Map","url","WebSocket","toString","on","error","clear","addStream","blob","id","upstream","set","metadata","getStream","get","downstream","send","DownStreamPull","connect","healthCheck","getURL","accept","client","format","contentType","services","binaryType","onmessage","event","buffer","data","handler","call","slice","Uint8Array","BYTES_PER_ELEMENT","onopen","debug","console","log","origin","emit","onclose","code","undefined","Error","reason","onerror","cause","disconnect","close","rpc","signal","callId","payload","procedure","service","encodeRpc","reject","delete","RpcAbort","Promise","resolve","Rpc","setQueryParams","params","groupCollapsed","trace","groupEnd","values","stream","reader","cancel","writer","abort","subscription","unsubscribe","AbortSignal","timeout","auth","ok","fetch","e","seconds","Math","min","r","setTimeout","path","protocol","base","URL","secure","host","key","Object","entries","Array","isArray","value","searchParams","append","resolveRpc","Event","decode","decodeRpc","has","success","UpStreamPull","size","Uint32Array","buf","ArrayBuffer","done","read","UpStreamEnd","UpStreamPush","byteLength","UpStreamAbort","DownStreamPush","ready","chunk","write","DownStreamAbort","DownStreamEnd","catch","RpcSubscription","ClientUnsubscribe","encode","ServerSubscriptionEvent","ServerUnsubscribe"],"mappings":"AAAA,SAEEA,eAAe,EAGfC,cAAc,EACdC,YAAY,EACZC,sBAAsB,EACtBC,OAAO,EACPC,IAAI,QACC,gBAAe;AACtB,SAGEC,WAAW,EACXC,eAAe,EACfC,aAAa,EACbC,MAAM,EACNC,YAAY,EACZC,YAAY,QACP,gBAAe;AA2BtB,OAAO,MAAMC,0BAA0BZ;;IAM5Ba,KAAuB;IAEbC,MAAiC;IACjCC,cAA+C;IAC/CC,UAA6C;IAC7CC,YAAwD;IACjEC,WAAc;IACdC,YAAgB;IAChBC,GAAc;IACdC,UAAiB;IACjBC,oBAAuB;IACvBC,cAAsB;IACtBC,YAAmB;IAErBC,UAAmC;IACnCC,iBAAkC;IAClCC,iBAAkC;IAE1CC,YAAY,AAAiBC,OAAiC,CAAE;QAC9D,KAAK,SADsBA,UAAAA,cAlBpBhB,OAAOL,cAAcsB,EAAE,OAEbhB,QAAQ,IAAIiB,YACZhB,gBAAgB,IAAIgB,YACpBf,YAAY,IAAIe,YAChBd,cAAc,IAAIc,YAC3Bb,aAAa,QACbC,cAAc,CAAC,QAEfE,YAAY,YACZC,sBAAsB,QAEtBE,cAAc;QAStB,IAAI,CAACD,aAAa,GAAG,IAAI,CAACM,OAAO,CAACN,aAAa,IAAI;QAEnD,IAAIM,QAAQJ,SAAS,EAAE;YACrB,IAAI,CAACA,SAAS,GAAGI,QAAQJ,SAAS;QACpC,OAAO;YACL,IAAI,CAACA,SAAS,GAAG,CAACO,MAAa,IAAIC,UAAUD,IAAIE,QAAQ;QAC3D;QAGA,IAAI,CAACC,EAAE,CAAC,SAAS,CAACC,QAAU,IAAI,CAACC,KAAK,CAACD;QAEvC,IAAI,CAACV,gBAAgB,GAAG;YACtBY,WAAW,CAACC;gBACV,MAAMC,KAAK,EAAE,IAAI,CAACtB,UAAU;gBAC5B,MAAMuB,WAAW,IAAIxC,eAAeuC,IAAID;gBACxC,IAAI,CAACvB,SAAS,CAAC0B,GAAG,CAACF,IAAIC;gBACvB,OAAO;oBAAED;oBAAIG,UAAUJ,KAAKI,QAAQ;gBAAC;YACvC;YACAC,WAAW,CAACJ,KAAO,IAAI,CAACxB,SAAS,CAAC6B,GAAG,CAACL;QACxC;QAEA,IAAI,CAACb,gBAAgB,GAAG;YACtBW,WAAW,CAACE,IAAIG;gBACd,MAAMG,aAAa3C,uBAAuBwC,UAAU,IAClD,IAAI,CAACI,IAAI,CAACzC,YAAY0C,cAAc,EAAErC,aAAa6B,IAAI;gBAEzD,IAAI,CAACvB,WAAW,CAACyB,GAAG,CAACF,IAAIM;gBACzB,OAAOA,WAAWP,IAAI;YACxB;YACAK,WAAW,CAACJ,KAAO,IAAI,CAACvB,WAAW,CAAC4B,GAAG,CAACL;QAC1C;IACF;IAEA,MAAMS,UAAU;QAEd,IAAI,CAAC1B,aAAa,GAAG,IAAI,CAACM,OAAO,CAACN,aAAa,IAAI;QACnD,MAAM,IAAI,CAAC2B,WAAW;QAEtB,IAAI,CAAC9B,EAAE,GAAG,IAAI,CAACK,SAAS,CACtB,IAAI,CAAC0B,MAAM,CAAC,OAAO,MAAM;YACvB,GAAG,IAAI,CAAChC,WAAW;YACnBiC,QAAQ,IAAI,CAACC,MAAM,CAACC,MAAM,CAACC,WAAW;YACtC,gBAAgB,IAAI,CAACF,MAAM,CAACC,MAAM,CAACC,WAAW;YAC9CC,UAAU,IAAI,CAACH,MAAM,CAACG,QAAQ;QAChC;QAGF,IAAI,CAACpC,EAAE,CAACqC,UAAU,GAAG;QAErB,IAAI,CAACrC,EAAE,CAACsC,SAAS,GAAG,CAACC;YACnB,MAAMC,SAAsBD,MAAME,IAAI;YACtC,MAAMhD,OAAOH,aAAakD,QAAQ;YAClC,MAAME,UAAU,IAAI,CAACjD,KAAK;YAC1B,IAAIiD,SAAS;gBACXA,QAAQC,IAAI,CAAC,IAAI,EAAEH,OAAOI,KAAK,CAACC,WAAWC,iBAAiB,GAAG,IAAI,CAAC9C,EAAE;YACxE;QACF;QACA,IAAI,CAACA,EAAE,CAAC+C,MAAM,GAAG,CAACR;YAChB,IAAI,IAAI,CAAC9B,OAAO,CAACuC,KAAK,EACpBC,QAAQC,GAAG,CAAC,CAAC,kBAAkB,EAAE,IAAI,CAACzC,OAAO,CAAC0C,MAAM,CAAC,CAAC;YACxD,IAAI,CAAC/C,WAAW,GAAG;YACnB,IAAI,CAACgD,IAAI,CAAC;YACV,IAAI,CAAClD,mBAAmB,GAAG;QAC7B;QACA,IAAI,CAACF,EAAE,CAACqD,OAAO,GAAG,CAACd;YACjB,IAAI,CAACnC,WAAW,GAAG;YACnB,IAAI,CAACH,SAAS,GAAG;YACjB,IAAI,CAACmD,IAAI,CACP,SACAb,MAAMe,IAAI,KAAK,OACXC,YACA,IAAIC,MACF,CAAC,4BAA4B,EAAEjB,MAAMe,IAAI,CAAC,EAAE,EAAEf,MAAMkB,MAAM,CAAC,CAAC;YAIpE,IAAI,IAAI,CAACtD,aAAa,EAAE,IAAI,CAAC0B,OAAO;QACtC;QACA,IAAI,CAAC7B,EAAE,CAAC0D,OAAO,GAAG,CAACC;YACjB,IAAI,CAAC1D,SAAS,GAAG;YACjB,IAAI,CAACmD,IAAI,CAAC,SAAS,IAAII,MAAM,oBAAoB;gBAAEG;YAAM;QAC3D;QACA,MAAM1E,KAAK,IAAI,EAAE;QACjB,IAAI,IAAI,CAACwB,OAAO,CAACuC,KAAK,EACpBC,QAAQC,GAAG,CAAC,CAAC,mBAAmB,EAAE,IAAI,CAACzC,OAAO,CAAC0C,MAAM,CAAC,CAAC;QACzD,IAAI,CAACC,IAAI,CAAC;IACZ;IAEA,MAAMQ,aAA4B;QAChC,IAAI,CAACzD,aAAa,GAAG;QACrB,IAAI,CAACH,EAAE,EAAE6D;QACT,MAAM5E,KAAK,IAAI,EAAE;IACnB;IAEA,MAAM6E,IAAInB,IAA4B,EAAqC;QACzE,MAAM,EAAEoB,MAAM,EAAEC,MAAM,EAAEC,OAAO,EAAEC,SAAS,EAAEC,OAAO,EAAE,GAAGxB;QAExD,MAAMF,OAAO,IAAI,CAACR,MAAM,CAACC,MAAM,CAACkC,SAAS,CACvC;YACEJ;YACAG;YACAD;YACAD;QACF,GACA,IAAI,CAAC3D,gBAAgB;QAGvBtB,QAAQ+E,QAAQ;YACd,MAAMpB,OAAO,IAAI,CAACjD,KAAK,CAAC+B,GAAG,CAACuC;YAC5B,IAAIrB,MAAM;gBACR,MAAM,EAAE0B,MAAM,EAAE,GAAG1B;gBACnB0B,OAAO,IAAIb,MAAM;gBACjB,IAAI,CAAC9D,KAAK,CAAC4E,MAAM,CAACN;gBAClB,IAAI,CAACrC,IAAI,CAACzC,YAAYqF,QAAQ,EAAEhF,aAAayE,QAAQ;YACvD;QACF;QAEA,IAAI,CAAC,IAAI,CAAC5D,WAAW,EAAE,MAAMnB,KAAK,IAAI,EAAE;QAExC,OAAO,IAAIuF,QAAQ,CAACC,SAASJ;YAC3B,IAAI,CAAC3E,KAAK,CAAC4B,GAAG,CAAC0C,QAAQ;gBAAES;gBAASJ;YAAO;YACzC,IAAI,CAAC1C,IAAI,CAACzC,YAAYwF,GAAG,EAAEjC;QAC7B;IACF;IAEAkC,eAAeC,MAA2B,EAAE;QAC1C,IAAI,CAAC7E,WAAW,GAAG6E;IACrB;IAEUjD,KAAKlC,IAAiB,EAAE,GAAGwE,OAAsB,EAAE;QAC3D,IAAI,CAACjE,EAAE,EAAE2B,KAAKtC,OAAOE,aAAaE,MAAM,aAAawE;QACrD,IAAI,IAAI,CAACxD,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQ4B,cAAc,CAAC,CAAC,UAAU,EAAE1F,eAAe,CAACM,KAAK,CAAC,CAAC;YAC3DwD,QAAQ6B,KAAK;YACb7B,QAAQ8B,QAAQ;QAClB;IACF;IAEA,MAAgB9D,MAAMD,QAAe,IAAIwC,MAAM,oBAAoB,EAAE;QACnE,KAAK,MAAMb,QAAQ,IAAI,CAACjD,KAAK,CAACsF,MAAM,GAAI;YACtC,MAAM,EAAEX,MAAM,EAAE,GAAG1B;YACnB0B,OAAOrD;QACT;QAEA,KAAK,MAAMiE,UAAU,IAAI,CAACrF,SAAS,CAACoF,MAAM,GAAI;YAC5CC,OAAOC,MAAM,CAACC,MAAM,CAACnE;QACvB;QAEA,KAAK,MAAMiE,UAAU,IAAI,CAACpF,WAAW,CAACmF,MAAM,GAAI;YAC9CC,OAAOG,MAAM,CAACC,KAAK,CAACrE;QACtB;QAEA,KAAK,MAAMsE,gBAAgB,IAAI,CAAC3F,aAAa,CAACqF,MAAM,GAAI;YACtDM,aAAaC,WAAW;QAC1B;QAEA,IAAI,CAAC7F,KAAK,CAACuB,KAAK;QAChB,IAAI,CAACrB,SAAS,CAACqB,KAAK;QACpB,IAAI,CAACpB,WAAW,CAACoB,KAAK;QACtB,IAAI,CAACtB,aAAa,CAACsB,KAAK;IAC1B;IAEA,MAAgBa,cAAc;QAC5B,MAAO,CAAC,IAAI,CAAC7B,SAAS,CAAE;YACtB,IAAI;gBACF,MAAM8D,SAASyB,YAAYC,OAAO,CAAC;gBACnC,MAAM7E,MAAM,IAAI,CAACmB,MAAM,CAAC,WAAW,QAAQ;oBACzC2D,MAAM,IAAI,CAACzD,MAAM,CAACyD,IAAI;gBACxB;gBACA,MAAM,EAAEC,EAAE,EAAE,GAAG,MAAMC,MAAMhF,KAAK;oBAAEmD;gBAAO;gBACzC,IAAI,CAAC9D,SAAS,GAAG0F;YACnB,EAAE,OAAOE,GAAG,CAAC;YAEb,IAAI,CAAC,IAAI,CAAC5F,SAAS,EAAE;gBACnB,IAAI,CAACC,mBAAmB;gBACxB,MAAM4F,UAAUC,KAAKC,GAAG,CAAC,IAAI,CAAC9F,mBAAmB,EAAE;gBACnD,MAAM,IAAIsE,QAAQ,CAACyB,IAAMC,WAAWD,GAAGH,UAAU;YACnD;QACF;QACA,IAAI,CAAC1C,IAAI,CAAC;IACZ;IAEQrB,OACNoE,OAAO,EAAE,EACTC,QAAuB,EACvBxB,SAA8B,CAAC,CAAC,EAChC;QAEA,MAAMyB,OAAO,IAAIC,IAAI,IAAI,CAAC7F,OAAO,CAAC0C,MAAM;QACxC,MAAMoD,SAASF,KAAKD,QAAQ,KAAK;QACjC,MAAMxF,MAAM,IAAI0F,IACd,CAAC,EAAEC,SAASH,WAAW,MAAMA,SAAS,GAAG,EAAEC,KAAKG,IAAI,CAAC,CAAC,EAAEL,KAAK,CAAC;QAEhE,KAAK,IAAI,CAACM,KAAKzB,OAAO,IAAI0B,OAAOC,OAAO,CAAC/B,QAAS;YAChD,IAAI,CAACgC,MAAMC,OAAO,CAAC7B,SAASA,SAAS;gBAACA;aAAO;YAC7C,KAAK,MAAM8B,SAAS9B,OAAQ;gBAC1BpE,IAAImG,YAAY,CAACC,MAAM,CAACP,KAAKK;YAC/B;QACF;QACA,OAAOlG;IACT;IAEUqG,WAAWjD,MAAc,EAAE8C,KAA+B,EAAE;QACpE,MAAMnE,OAAO,IAAI,CAACjD,KAAK,CAAC+B,GAAG,CAACuC;QAC5B,IAAIrB,MAAM;YACR,MAAM,EAAE8B,OAAO,EAAE,GAAG9B;YACpB,IAAI,CAACjD,KAAK,CAAC4E,MAAM,CAACN;YAClBS,QAAQqC;QACV;IACF;IAEA,MAAgB,CAAC5H,YAAYgI,KAAK,CAAC,CAAC1E,MAAmB,EAAE;QACvD,MAAM,CAAC2B,SAAS5B,OAAO0B,QAAQ,GAAG,IAAI,CAAChC,MAAM,CAACC,MAAM,CAACiF,MAAM,CAAC3E;QAC5D,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQ4B,cAAc,CAAC,CAAC,sBAAsB,EAAEV,QAAQ,CAAC,EAAE5B,MAAM,CAAC;YAClEU,QAAQC,GAAG,CAACe;YACZhB,QAAQ8B,QAAQ;QAClB;QACA,IAAI,CAAC3B,IAAI,CAAC,SAASe,SAAS5B,OAAO0B;IACrC;IAEA,MAAgB,CAAC/E,YAAYwF,GAAG,CAAC,CAAClC,MAAmB,EAAE;QACrD,MAAM,EAAEwB,MAAM,EAAEhD,KAAK,EAAEiD,OAAO,EAAE,GAAG,IAAI,CAAChC,MAAM,CAACC,MAAM,CAACkF,SAAS,CAC7D5E,QACA,IAAI,CAACjC,gBAAgB;QAGvB,IAAI,IAAI,CAACb,KAAK,CAAC2H,GAAG,CAACrD,SAAS;YAC1B,IAAI,CAACiD,UAAU,CACbjD,QACAhD,QAAQ;gBAAEsG,SAAS;gBAAOtG;YAAM,IAAI;gBAAEsG,SAAS;gBAAMR,OAAO7C;YAAQ;QAExE;IACF;IAEA,MAAgB,CAAC/E,YAAYqI,YAAY,CAAC,CAAC/E,MAAmB,EAAE;QAC9D,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,MAAMgF,OAAOlI,aAAakD,QAAQ,UAAUiF,YAAY3E,iBAAiB;QAEzE,IAAI,IAAI,CAACrC,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,6BAA6B,EAAE9B,GAAG,CAAC;QAClD;QAEA,MAAM6D,SAAS,IAAI,CAACrF,SAAS,CAAC6B,GAAG,CAACL;QAClC,IAAI6D,QAAQ;YACV,MAAMyC,MAAM,IAAI7E,WAAW,IAAI8E,YAAYH;YAC3C,MAAM,EAAEI,IAAI,EAAEd,KAAK,EAAE,GAAG,MAAM7B,OAAOC,MAAM,CAAC2C,IAAI,CAACH;YACjD,IAAIE,MAAM;gBACR,IAAI,CAACjG,IAAI,CAACzC,YAAY4I,WAAW,EAAEvI,aAAa6B,IAAI;YACtD,OAAO;gBACL,IAAI,CAACO,IAAI,CACPzC,YAAY6I,YAAY,EACxB1I,OACEE,aAAa6B,IAAI,WACjB0F,MAAMtE,MAAM,CAACI,KAAK,CAAC,GAAGkE,MAAMkB,UAAU;YAG5C;QACF;IACF;IAEA,MAAgB,CAAC9I,YAAY4I,WAAW,CAAC,CAACtF,MAAmB,EAAE;QAC7D,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,4BAA4B,EAAE9B,GAAG,CAAC;QACjD;QACA,MAAM6D,SAAS,IAAI,CAACrF,SAAS,CAAC6B,GAAG,CAACL;QAClC,IAAI6D,QAAQ;YACVA,OAAOC,MAAM,CAACC,MAAM;YACpB,IAAI,CAACvF,SAAS,CAAC0E,MAAM,CAAClD;QACxB;IACF;IAEU,CAAClC,YAAY+I,aAAa,CAAC,CAACzF,MAAmB,EAAE;QACzD,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,8BAA8B,EAAE9B,GAAG,CAAC;QACnD;QACA,MAAM6D,SAAS,IAAI,CAACrF,SAAS,CAAC6B,GAAG,CAACL;QAClC,IAAI6D,QAAQ;YACV,IAAI;gBACFA,OAAOC,MAAM,CAACC,MAAM,CAAC,IAAI3B,MAAM;YACjC,SAAU;gBACR,IAAI,CAAC5D,SAAS,CAAC0E,MAAM,CAAClD;YACxB;QACF;IACF;IAEA,MAAgB,CAAClC,YAAYgJ,cAAc,CAAC,CAAC1F,MAAmB,EAAE;QAChE,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,+BAA+B,EAAE9B,GAAG,CAAC;QACpD;QACA,MAAM6D,SAAS,IAAI,CAACpF,WAAW,CAAC4B,GAAG,CAACL;QACpC,IAAI6D,QAAQ;YACV,IAAI;gBACF,MAAMA,OAAOG,MAAM,CAAC+C,KAAK;gBACzB,MAAMC,QAAQ5F,OAAOI,KAAK,CAAC6E,YAAY3E,iBAAiB;gBACxD,MAAMmC,OAAOG,MAAM,CAACiD,KAAK,CAAC,IAAIxF,WAAWuF;gBACzC,MAAMnD,OAAOG,MAAM,CAAC+C,KAAK;gBACzB,IAAI,CAACxG,IAAI,CAACzC,YAAY0C,cAAc,EAAErC,aAAa6B,IAAI;YACzD,EAAE,OAAOyE,GAAG;gBACV,IAAI,CAAClE,IAAI,CAACzC,YAAYoJ,eAAe,EAAE/I,aAAa6B,IAAI;gBACxD,IAAI,CAACvB,WAAW,CAACyE,MAAM,CAAClD;YAC1B;QACF;IACF;IAEA,MAAgB,CAAClC,YAAYqJ,aAAa,CAAC,CAAC/F,MAAmB,EAAE;QAC/D,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,8BAA8B,EAAE9B,GAAG,CAAC;QACnD;QACA,MAAM6D,SAAS,IAAI,CAACpF,WAAW,CAAC4B,GAAG,CAACL;QACpC,IAAI6D,QAAQ;YACV,IAAI,CAACpF,WAAW,CAACyE,MAAM,CAAClD;YACxB6D,OAAOG,MAAM,CAACvB,KAAK,GAAG2E,KAAK,CAAC,KAAO;QACrC;IACF;IAEA,MAAgB,CAACtJ,YAAYoJ,eAAe,CAAC,CAAC9F,MAAmB,EAAE;QACjE,MAAMpB,KAAK9B,aAAakD,QAAQ;QAChC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,gCAAgC,EAAE9B,GAAG,CAAC;QACrD;QACA,MAAM6D,SAAS,IAAI,CAACpF,WAAW,CAAC4B,GAAG,CAACL;QACpC,IAAI6D,QAAQ;YACV,IAAI,CAACpF,WAAW,CAACyE,MAAM,CAAClD;YACxB6D,OAAOG,MAAM,CAACC,KAAK,CAAC,IAAI7B,MAAM,sBAAsBgF,KAAK,CAAC,KAAO;QACnE;IACF;IAEU,CAACtJ,YAAYuJ,eAAe,CAAC,CAACjG,MAAmB,EAAE;QAC3D,MAAM,EACJwB,MAAM,EACNC,SAAS,CAACwC,KAAKxC,QAAQ,EACxB,GAAG,IAAI,CAAChC,MAAM,CAACC,MAAM,CAACkF,SAAS,CAAC5E,QAAQ,IAAI,CAACjC,gBAAgB;QAC9D,IAAI,IAAI,CAACb,KAAK,CAAC2H,GAAG,CAACrD,SAAS;YAC1B,MAAMsB,eAAe,IAAIxG,aAAa2H,KAAK;gBACzCnB,aAAalC,IAAI,CAAC;gBAClB,IAAI,CAACzD,aAAa,CAAC2E,MAAM,CAACmC;gBAC1B,IAAI,CAAC9E,IAAI,CACPzC,YAAYwJ,iBAAiB,EAC7B,IAAI,CAACzG,MAAM,CAACC,MAAM,CAACyG,MAAM,CAAC;oBAAClC;iBAAI;YAEnC;YACA,IAAI,CAAC9G,aAAa,CAAC2B,GAAG,CAACmF,KAAKnB;YAC5B,IAAI,CAAC2B,UAAU,CAACjD,QAAQ;gBACtBsD,SAAS;gBACTR,OAAO;oBAAE7C;oBAASqB;gBAAa;YACjC;QACF;IACF;IAEU,CAACpG,YAAY0J,uBAAuB,CAAC,CAACpG,MAAmB,EAAE;QACnE,MAAM,CAACiE,KAAKlE,OAAO0B,QAAQ,GAAG,IAAI,CAAChC,MAAM,CAACC,MAAM,CAACiF,MAAM,CAAC3E;QACxD,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQ4B,cAAc,CACpB,CAAC,wCAAwC,EAAE4B,IAAI,CAAC,EAAElE,MAAM,CAAC;YAE3DU,QAAQC,GAAG,CAACe;YACZhB,QAAQ8B,QAAQ;QAClB;QACA,MAAMO,eAAe,IAAI,CAAC3F,aAAa,CAAC8B,GAAG,CAACgF;QAC5C,IAAInB,cAAcA,aAAalC,IAAI,CAACb,OAAO0B;IAC7C;IAEU,CAAC/E,YAAY2J,iBAAiB,CAAC,CAACrG,MAAmB,EAAE;QAC7D,MAAM,CAACiE,IAAI,GAAG,IAAI,CAACxE,MAAM,CAACC,MAAM,CAACiF,MAAM,CAAC3E;QACxC,IAAI,IAAI,CAAC/B,OAAO,CAACuC,KAAK,EAAE;YACtBC,QAAQC,GAAG,CAAC,CAAC,kCAAkC,EAAEuD,IAAI,CAAC;QACxD;QACA,MAAMnB,eAAe,IAAI,CAAC3F,aAAa,CAAC8B,GAAG,CAACgF;QAC5CnB,cAAclC,KAAK;QACnB,IAAI,CAACzD,aAAa,CAAC2E,MAAM,CAACmC;IAC5B;AACF"}
1	+ {"version":3,"sources":["../../index.ts"],"sourcesContent":["import {\n type BaseClientFormat,\n EventEmitter,\n type ProtocolTransport,\n} from '@nmtjs/protocol/client'\nimport {\n type ClientMessageType,\n ServerMessageType,\n concat,\n decodeNumber,\n encodeNumber,\n} from '@nmtjs/protocol/common'\n\nexport type WebSocketClientTransportOptions = {\n /*\n The origin of the server\n * @example 'http://localhost:3000'\n /\n origin: string\n /\n Whether to autoreconnect on close\n * @default true\n /\n autoreconnect?: boolean\n /\n Custom WebSocket class\n * @default globalThis.WebSocket\n */\n wsFactory?: (url: URL) => WebSocket\n\n debug?: boolean\n}\n\nexport class WebSocketClientTransport\n extends EventEmitter\n implements ProtocolTransport\n{\n #webSocket: WebSocket \| null = null\n #connecting: Promise<void> \| null = null\n\n constructor(private readonly options: WebSocketClientTransportOptions) {\n super()\n }\n\n connect(\n auth: string \| undefined = undefined,\n contentType: BaseClientFormat['contentType'],\n ): Promise<void> {\n const wsUrl = new URL(this.options.origin)\n wsUrl.protocol = wsUrl.protocol === 'https:' ? 'wss:' : 'ws:'\n wsUrl.pathname = '/api'\n wsUrl.searchParams.set('content-type', contentType)\n wsUrl.searchParams.set('accept', contentType)\n if (auth) wsUrl.searchParams.set('auth', auth)\n\n const ws =\n this.options.wsFactory?.(wsUrl) ?? new WebSocket(wsUrl.toString())\n\n ws.binaryType = 'arraybuffer'\n\n ws.addEventListener('message', ({ data }) => {\n const buffer: ArrayBuffer = data\n const type = decodeNumber(buffer, 'Uint8')\n if (type in ServerMessageType) {\n this.emit(`${type}`, buffer.slice(Uint8Array.BYTES_PER_ELEMENT))\n }\n })\n\n this.#webSocket = ws\n\n return new Promise((resolve, reject) => {\n ws.addEventListener(\n 'open',\n () => {\n this.emit('connect')\n resolve()\n },\n { once: true },\n )\n\n ws.addEventListener(\n 'error',\n (event) => reject(new Error('WebSocket error', { cause: event })),\n { once: true },\n )\n\n ws.addEventListener(\n 'close',\n (event) => {\n this.emit('disconnect')\n this.#webSocket = null\n if (this.options.autoreconnect === true) {\n setTimeout(() => this.connect(auth, contentType), 1000)\n }\n },\n { once: true },\n )\n })\n }\n\n async disconnect(): Promise<void> {\n if (this.#webSocket === null) return\n this.#webSocket!.close()\n return _once(this.#webSocket, 'close')\n }\n\n async send(\n messageType: ClientMessageType,\n buffer: ArrayBuffer,\n ): Promise<void> {\n if (this.#connecting) await this.#connecting\n this.#webSocket!.send(concat(encodeNumber(messageType, 'Uint8'), buffer))\n }\n}\n\nfunction _once(target: EventTarget, event: string) {\n return new Promise<void>((resolve) => {\n target.addEventListener(event, () => resolve(), { once: true })\n })\n}\n"],"names":["EventEmitter","ServerMessageType","concat","decodeNumber","encodeNumber","WebSocketClientTransport","constructor","options","connect","auth","undefined","contentType","wsUrl","URL","origin","protocol","pathname","searchParams","set","ws","wsFactory","WebSocket","toString","binaryType","addEventListener","data","buffer","type","emit","slice","Uint8Array","BYTES_PER_ELEMENT","Promise","resolve","reject","once","event","Error","cause","autoreconnect","setTimeout","disconnect","close","_once","send","messageType","target"],"mappings":"AAAA,SAEEA,YAAY,QAEP,yBAAwB;AAC/B,SAEEC,iBAAiB,EACjBC,MAAM,EACNC,YAAY,EACZC,YAAY,QACP,yBAAwB;AAsB/B,OAAO,MAAMC,iCACHL;;IAGR,CAAA,SAAU,CAAyB;IACnC,CAAA,UAAW,CAA6B;IAExCM,YAAY,AAAiBC,OAAwC,CAAE;QACrE,KAAK,SADsBA,UAAAA,cAH7B,CAAA,SAAU,GAAqB,WAC/B,CAAA,UAAW,GAAyB;IAIpC;IAEAC,QACEC,OAA2BC,SAAS,EACpCC,WAA4C,EAC7B;QACf,MAAMC,QAAQ,IAAIC,IAAI,IAAI,CAACN,OAAO,CAACO,MAAM;QACzCF,MAAMG,QAAQ,GAAGH,MAAMG,QAAQ,KAAK,WAAW,SAAS;QACxDH,MAAMI,QAAQ,GAAG;QACjBJ,MAAMK,YAAY,CAACC,GAAG,CAAC,gBAAgBP;QACvCC,MAAMK,YAAY,CAACC,GAAG,CAAC,UAAUP;QACjC,IAAIF,MAAMG,MAAMK,YAAY,CAACC,GAAG,CAAC,QAAQT;QAEzC,MAAMU,KACJ,IAAI,CAACZ,OAAO,CAACa,SAAS,GAAGR,UAAU,IAAIS,UAAUT,MAAMU,QAAQ;QAEjEH,GAAGI,UAAU,GAAG;QAEhBJ,GAAGK,gBAAgB,CAAC,WAAW,CAAC,EAAEC,IAAI,EAAE;YACtC,MAAMC,SAAsBD;YAC5B,MAAME,OAAOxB,aAAauB,QAAQ;YAClC,IAAIC,QAAQ1B,mBAAmB;gBAC7B,IAAI,CAAC2B,IAAI,CAAC,CAAC,EAAED,KAAK,CAAC,EAAED,OAAOG,KAAK,CAACC,WAAWC,iBAAiB;YAChE;QACF;QAEA,IAAI,CAAC,CAAA,SAAU,GAAGZ;QAElB,OAAO,IAAIa,QAAQ,CAACC,SAASC;YAC3Bf,GAAGK,gBAAgB,CACjB,QACA;gBACE,IAAI,CAACI,IAAI,CAAC;gBACVK;YACF,GACA;gBAAEE,MAAM;YAAK;YAGfhB,GAAGK,gBAAgB,CACjB,SACA,CAACY,QAAUF,OAAO,IAAIG,MAAM,mBAAmB;oBAAEC,OAAOF;gBAAM,KAC9D;gBAAED,MAAM;YAAK;YAGfhB,GAAGK,gBAAgB,CACjB,SACA,CAACY;gBACC,IAAI,CAACR,IAAI,CAAC;gBACV,IAAI,CAAC,CAAA,SAAU,GAAG;gBAClB,IAAI,IAAI,CAACrB,OAAO,CAACgC,aAAa,KAAK,MAAM;oBACvCC,WAAW,IAAM,IAAI,CAAChC,OAAO,CAACC,MAAME,cAAc;gBACpD;YACF,GACA;gBAAEwB,MAAM;YAAK;QAEjB;IACF;IAEA,MAAMM,aAA4B;QAChC,IAAI,IAAI,CAAC,CAAA,SAAU,KAAK,MAAM;QAC9B,IAAI,CAAC,CAAA,SAAU,CAAEC,KAAK;QACtB,OAAOC,MAAM,IAAI,CAAC,CAAA,SAAU,EAAE;IAChC;IAEA,MAAMC,KACJC,WAA8B,EAC9BnB,MAAmB,EACJ;QACf,IAAI,IAAI,CAAC,CAAA,UAAW,EAAE,MAAM,IAAI,CAAC,CAAA,UAAW;QAC5C,IAAI,CAAC,CAAA,SAAU,CAAEkB,IAAI,CAAC1C,OAAOE,aAAayC,aAAa,UAAUnB;IACnE;AACF;AAEA,SAASiB,MAAMG,MAAmB,EAAEV,KAAa;IAC/C,OAAO,IAAIJ,QAAc,CAACC;QACxBa,OAAOtB,gBAAgB,CAACY,OAAO,IAAMH,WAAW;YAAEE,MAAM;QAAK;IAC/D;AACF"}

package/index.ts CHANGED Viewed

@@ -1,26 +1,17 @@
 import {
-  type ClientDownStreamWrapper,
-  ClientTransport,
-  type ClientTransportRpcCall,
-  type ClientTransportRpcResult,
-  ClientUpStream,
-  Subscription,
-  createClientDownStream,
-  onAbort,
-  once,
-} from '@nmtjs/client'
+  type BaseClientFormat,
+  EventEmitter,
+  type ProtocolTransport,
+} from '@nmtjs/protocol/client'
 import {
-  type DecodeRpcContext,
-  type EncodeRpcContext,
-  MessageType,
-  MessageTypeName,
-  TransportType,
+  type ClientMessageType,
+  ServerMessageType,
   concat,
   decodeNumber,
   encodeNumber,
-} from '@nmtjs/common'
+} from '@nmtjs/protocol/common'
-export type WsClientTransportOptions = {
+export type WebSocketClientTransportOptions = {
   /**
    * The origin of the server
    * @example 'http://localhost:3000'
@@ -40,412 +31,90 @@ export type WsClientTransportOptions = {
   debug?: boolean
 }
-type WsCall = {
-  resolve: (value: ClientTransportRpcResult) => void
-  reject: (error: Error) => void
-}
-export class WsClientTransport extends ClientTransport<{
-  open: []
-  close: [Error?]
-  connect: []
-  healthy: []
-}> {
-  readonly type = TransportType.WS
-  protected readonly calls = new Map<number, WsCall>()
-  protected readonly subscriptions = new Map<number, Subscription>()
-  protected readonly upStreams = new Map<number, ClientUpStream>()
-  protected readonly downStreams = new Map<number, ClientDownStreamWrapper>()
-  protected upStreamId = 0
-  protected queryParams = {}
-  protected ws?: WebSocket
-  protected isHealthy = false
-  protected checkHealthAttempts = 0
-  protected autoreconnect: boolean
-  protected isConnected = false
+export class WebSocketClientTransport
+  extends EventEmitter
+  implements ProtocolTransport
+{
+  #webSocket: WebSocket | null = null
+  #connecting: Promise<void> | null = null
-  private wsFactory!: (url: URL) => WebSocket
-  private encodeRpcContext: EncodeRpcContext
-  private decodeRpcContext: DecodeRpcContext
-  constructor(private readonly options: WsClientTransportOptions) {
+  constructor(private readonly options: WebSocketClientTransportOptions) {
     super()
-    this.autoreconnect = this.options.autoreconnect ?? true
-    if (options.wsFactory) {
-      this.wsFactory = options.wsFactory
-    } else {
-      this.wsFactory = (url: URL) => new WebSocket(url.toString())
-    }
-    // TODO: wtf is this for?
-    this.on('close', (error) => this.clear(error))
-    this.encodeRpcContext = {
-      addStream: (blob) => {
-        const id = ++this.upStreamId
-        const upstream = new ClientUpStream(id, blob)
-        this.upStreams.set(id, upstream)
-        return { id, metadata: blob.metadata }
-      },
-      getStream: (id) => this.upStreams.get(id),
-    }
-    this.decodeRpcContext = {
-      addStream: (id, metadata) => {
-        const downstream = createClientDownStream(metadata, () =>
-          this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32')),
-        )
-        this.downStreams.set(id, downstream)
-        return downstream.blob
-      },
-      getStream: (id) => this.downStreams.get(id),
-    }
   }
-  async connect() {
-    // reset default autoreconnect value
-    this.autoreconnect = this.options.autoreconnect ?? true
-    await this.healthCheck()
+  connect(
+    auth: string | undefined = undefined,
+    contentType: BaseClientFormat['contentType'],
+  ): Promise<void> {
+    const wsUrl = new URL(this.options.origin)
+    wsUrl.protocol = wsUrl.protocol === 'https:' ? 'wss:' : 'ws:'
+    wsUrl.pathname = '/api'
+    wsUrl.searchParams.set('content-type', contentType)
+    wsUrl.searchParams.set('accept', contentType)
+    if (auth) wsUrl.searchParams.set('auth', auth)
-    this.ws = this.wsFactory(
-      this.getURL('api', 'ws', {
-        ...this.queryParams,
-        accept: this.client.format.contentType,
-        'content-type': this.client.format.contentType,
-        services: this.client.services,
-      }),
-    )
+    const ws =
+      this.options.wsFactory?.(wsUrl) ?? new WebSocket(wsUrl.toString())
-    this.ws.binaryType = 'arraybuffer'
+    ws.binaryType = 'arraybuffer'
-    this.ws.onmessage = (event) => {
-      const buffer: ArrayBuffer = event.data
+    ws.addEventListener('message', ({ data }) => {
+      const buffer: ArrayBuffer = data
       const type = decodeNumber(buffer, 'Uint8')
-      const handler = this[type]
-      if (handler) {
-        handler.call(this, buffer.slice(Uint8Array.BYTES_PER_ELEMENT), this.ws)
-      }
-    }
-    this.ws.onopen = (event) => {
-      if (this.options.debug)
-        console.log(`[WS] Connected to ${this.options.origin}`)
-      this.isConnected = true
-      this.emit('open')
-      this.checkHealthAttempts = 0
-    }
-    this.ws.onclose = (event) => {
-      this.isConnected = false
-      this.isHealthy = false
-      this.emit(
-        'close',
-        event.code === 1000
-          ? undefined
-          : new Error(
-              `Connection closed with code ${event.code}: ${event.reason}`,
-            ),
-      )
-      // FIXME: cleanup calls, streams, subscriptions
-      if (this.autoreconnect) this.connect()
-    }
-    this.ws.onerror = (cause) => {
-      this.isHealthy = false
-      this.emit('close', new Error('Connection error', { cause }))
-    }
-    await once(this, 'open')
-    if (this.options.debug)
-      console.log(`2[WS] Connected to ${this.options.origin}`)
-    this.emit('connect')
-  }
-  async disconnect(): Promise<void> {
-    this.autoreconnect = false
-    this.ws?.close()
-    await once(this, 'close')
-  }
-  async rpc(call: ClientTransportRpcCall): Promise<ClientTransportRpcResult> {
-    const { signal, callId, payload, procedure, service } = call
-    const data = this.client.format.encodeRpc(
-      {
-        callId,
-        service,
-        procedure,
-        payload,
-      },
-      this.encodeRpcContext,
-    )
-    onAbort(signal, () => {
-      const call = this.calls.get(callId)
-      if (call) {
-        const { reject } = call
-        reject(new Error('Request aborted'))
-        this.calls.delete(callId)
-        this.send(MessageType.RpcAbort, encodeNumber(callId, 'Uint32'))
+      if (type in ServerMessageType) {
+        this.emit(`${type}`, buffer.slice(Uint8Array.BYTES_PER_ELEMENT))
       }
     })
-    if (!this.isConnected) await once(this, 'connect')
+    this.#webSocket = ws
     return new Promise((resolve, reject) => {
-      this.calls.set(callId, { resolve, reject })
-      this.send(MessageType.Rpc, data)
-    })
-  }
-  setQueryParams(params: Record<string, any>) {
-    this.queryParams = params
-  }
-  protected send(type: MessageType, ...payload: ArrayBuffer[]) {
-    this.ws?.send(concat(encodeNumber(type, 'Uint8'), ...payload))
-    if (this.options.debug) {
-      console.groupCollapsed(`[WS] Sent ${MessageTypeName[type]}`)
-      console.trace()
-      console.groupEnd()
-    }
-  }
-  protected async clear(error: Error = new Error('Connection closed')) {
-    for (const call of this.calls.values()) {
-      const { reject } = call
-      reject(error)
-    }
-    for (const stream of this.upStreams.values()) {
-      stream.reader.cancel(error)
-    }
-    for (const stream of this.downStreams.values()) {
-      stream.writer.abort(error)
-    }
-    for (const subscription of this.subscriptions.values()) {
-      subscription.unsubscribe()
-    }
-    this.calls.clear()
-    this.upStreams.clear()
-    this.downStreams.clear()
-    this.subscriptions.clear()
-  }
-  protected async healthCheck() {
-    while (!this.isHealthy) {
-      try {
-        const signal = AbortSignal.timeout(10000)
-        const url = this.getURL('healthy', 'http', {
-          auth: this.client.auth,
-        })
-        const { ok } = await fetch(url, { signal })
-        this.isHealthy = ok
-      } catch (e) {}
-      if (!this.isHealthy) {
-        this.checkHealthAttempts++
-        const seconds = Math.min(this.checkHealthAttempts, 15)
-        await new Promise((r) => setTimeout(r, seconds * 1000))
-      }
-    }
-    this.emit('healthy')
-  }
-  private getURL(
-    path = '',
-    protocol: 'ws' | 'http',
-    params: Record<string, any> = {},
-  ) {
-    // TODO: add custom path support?
-    const base = new URL(this.options.origin)
-    const secure = base.protocol === 'https:'
-    const url = new URL(
-      `${secure ? protocol + 's' : protocol}://${base.host}/${path}`,
-    )
-    for (let [key, values] of Object.entries(params)) {
-      if (!Array.isArray(values)) values = [values]
-      for (const value of values) {
-        url.searchParams.append(key, value)
-      }
-    }
-    return url
-  }
-  protected resolveRpc(callId: number, value: ClientTransportRpcResult) {
-    const call = this.calls.get(callId)
-    if (call) {
-      const { resolve } = call
-      this.calls.delete(callId)
-      resolve(value)
-    }
-  }
-  protected async [MessageType.Event](buffer: ArrayBuffer) {
-    const [service, event, payload] = this.client.format.decode(buffer)
-    if (this.options.debug) {
-      console.groupCollapsed(`[WS] Received "Event" ${service}/${event}`)
-      console.log(payload)
-      console.groupEnd()
-    }
-    this.emit('event', service, event, payload)
-  }
-  protected async [MessageType.Rpc](buffer: ArrayBuffer) {
-    const { callId, error, payload } = this.client.format.decodeRpc(
-      buffer,
-      this.decodeRpcContext,
-    )
-    if (this.calls.has(callId)) {
-      this.resolveRpc(
-        callId,
-        error ? { success: false, error } : { success: true, value: payload },
+      ws.addEventListener(
+        'open',
+        () => {
+          this.emit('connect')
+          resolve()
+        },
+        { once: true },
       )
-    }
-  }
-  protected async [MessageType.UpStreamPull](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    const size = decodeNumber(buffer, 'Uint32', Uint32Array.BYTES_PER_ELEMENT)
-    if (this.options.debug) {
-      console.log(`[WS] Received "UpStreamPull" ${id}`)
-    }
-    const stream = this.upStreams.get(id)
-    if (stream) {
-      const buf = new Uint8Array(new ArrayBuffer(size))
-      const { done, value } = await stream.reader.read(buf)
-      if (done) {
-        this.send(MessageType.UpStreamEnd, encodeNumber(id, 'Uint32'))
-      } else {
-        this.send(
-          MessageType.UpStreamPush,
-          concat(
-            encodeNumber(id, 'Uint32'),
-            value.buffer.slice(0, value.byteLength),
-          ),
-        )
-      }
-    }
-  }
-  protected async [MessageType.UpStreamEnd](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    if (this.options.debug) {
-      console.log(`[WS] Received "UpStreamEnd" ${id}`)
-    }
-    const stream = this.upStreams.get(id)
-    if (stream) {
-      stream.reader.cancel()
-      this.upStreams.delete(id)
-    }
-  }
-  protected [MessageType.UpStreamAbort](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    if (this.options.debug) {
-      console.log(`[WS] Received "UpStreamAbort" ${id}`)
-    }
-    const stream = this.upStreams.get(id)
-    if (stream) {
-      try {
-        stream.reader.cancel(new Error('Aborted by server'))
-      } finally {
-        this.upStreams.delete(id)
-      }
-    }
-  }
-  protected async [MessageType.DownStreamPush](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    if (this.options.debug) {
-      console.log(`[WS] Received "DownStreamPush" ${id}`)
-    }
-    const stream = this.downStreams.get(id)
-    if (stream) {
-      try {
-        await stream.writer.ready
-        const chunk = buffer.slice(Uint32Array.BYTES_PER_ELEMENT)
-        await stream.writer.write(new Uint8Array(chunk))
-        await stream.writer.ready
-        this.send(MessageType.DownStreamPull, encodeNumber(id, 'Uint32'))
-      } catch (e) {
-        this.send(MessageType.DownStreamAbort, encodeNumber(id, 'Uint32'))
-        this.downStreams.delete(id)
-      }
-    }
-  }
-  protected async [MessageType.DownStreamEnd](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    if (this.options.debug) {
-      console.log(`[WS] Received "DownStreamEnd" ${id}`)
-    }
-    const stream = this.downStreams.get(id)
-    if (stream) {
-      this.downStreams.delete(id)
-      stream.writer.close().catch(() => {})
-    }
-  }
+      ws.addEventListener(
+        'error',
+        (event) => reject(new Error('WebSocket error', { cause: event })),
+        { once: true },
+      )
-  protected async [MessageType.DownStreamAbort](buffer: ArrayBuffer) {
-    const id = decodeNumber(buffer, 'Uint32')
-    if (this.options.debug) {
-      console.log(`[WS] Received "DownStreamAbort" ${id}`)
-    }
-    const stream = this.downStreams.get(id)
-    if (stream) {
-      this.downStreams.delete(id)
-      stream.writer.abort(new Error('Aborted by server')).catch(() => {})
-    }
+      ws.addEventListener(
+        'close',
+        (event) => {
+          this.emit('disconnect')
+          this.#webSocket = null
+          if (this.options.autoreconnect === true) {
+            setTimeout(() => this.connect(auth, contentType), 1000)
+          }
+        },
+        { once: true },
+      )
+    })
   }
-  protected [MessageType.RpcSubscription](buffer: ArrayBuffer) {
-    const {
-      callId,
-      payload: [key, payload],
-    } = this.client.format.decodeRpc(buffer, this.decodeRpcContext)
-    if (this.calls.has(callId)) {
-      const subscription = new Subscription(key, () => {
-        subscription.emit('end')
-        this.subscriptions.delete(key)
-        this.send(
-          MessageType.ClientUnsubscribe,
-          this.client.format.encode([key]),
-        )
-      })
-      this.subscriptions.set(key, subscription)
-      this.resolveRpc(callId, {
-        success: true,
-        value: { payload, subscription },
-      })
-    }
+  async disconnect(): Promise<void> {
+    if (this.#webSocket === null) return
+    this.#webSocket!.close()
+    return _once(this.#webSocket, 'close')
   }
-  protected [MessageType.ServerSubscriptionEvent](buffer: ArrayBuffer) {
-    const [key, event, payload] = this.client.format.decode(buffer)
-    if (this.options.debug) {
-      console.groupCollapsed(
-        `[WS] Received "ServerSubscriptionEvent" ${key}/${event}`,
-      )
-      console.log(payload)
-      console.groupEnd()
-    }
-    const subscription = this.subscriptions.get(key)
-    if (subscription) subscription.emit(event, payload)
+  async send(
+    messageType: ClientMessageType,
+    buffer: ArrayBuffer,
+  ): Promise<void> {
+    if (this.#connecting) await this.#connecting
+    this.#webSocket!.send(concat(encodeNumber(messageType, 'Uint8'), buffer))
   }
+}
-  protected [MessageType.ServerUnsubscribe](buffer: ArrayBuffer) {
-    const [key] = this.client.format.decode(buffer)
-    if (this.options.debug) {
-      console.log(`[WS] Received "ServerUnsubscribe" ${key}`)
-    }
-    const subscription = this.subscriptions.get(key)
-    subscription?.emit('end')
-    this.subscriptions.delete(key)
-  }
+function _once(target: EventTarget, event: string) {
+  return new Promise<void>((resolve) => {
+    target.addEventListener(event, () => resolve(), { once: true })
+  })
 }

package/package.json CHANGED Viewed

@@ -7,13 +7,10 @@
       "types": "./index.ts"
     }
   },
-  "peerDependencies": {
-    "@nmtjs/client": "^0.4.2",
-    "@nmtjs/common": "^0.4.2"
-  },
-  "devDependencies": {
-    "@nmtjs/client": "^0.4.2",
-    "@nmtjs/common": "^0.4.2"
+  "dependencies": {
+    "@nmtjs/client": "^0.6.1",
+    "@nmtjs/common": "^0.6.1",
+    "@nmtjs/protocol": "^0.6.1"
   },
   "files": [
     "index.ts",
@@ -22,7 +19,7 @@
     "LICENSE.md",
     "README.md"
   ],
-  "version": "0.1.8",
+  "version": "0.2.0",
   "scripts": {
     "build": "neemata-build -p neutral ./index.ts",
     "type-check": "tsc --noEmit"