@nmtjs/client 0.15.0-beta.1 → 0.15.0-beta.11

package/dist/clients/runtime.d.ts

@@ -0,0 +1,20 @@
+import type { TAnyRouterContract } from '@nmtjs/contract';
+import type { BaseClientOptions } from '../core.ts';
+import type { ClientTransportFactory } from '../transport.ts';
+import type { ClientCallers, RuntimeInputContractTypeProvider, RuntimeOutputContractTypeProvider } from '../types.ts';
+import { BaseClient } from '../core.ts';
+export declare class RuntimeContractTransformer {
+    #private;
+    constructor(router: TAnyRouterContract);
+    encode(_procedure: string, payload: any): unknown;
+    decode(_procedure: string, payload: any): unknown;
+}
+export declare class RuntimeClient<Transport extends ClientTransportFactory<any, any> = ClientTransportFactory<any, any>, RouterContract extends TAnyRouterContract = TAnyRouterContract, SafeCall extends boolean = false> extends BaseClient<Transport, RouterContract, SafeCall, RuntimeInputContractTypeProvider, RuntimeOutputContractTypeProvider> {
+    #private;
+    protected readonly transformer: RuntimeContractTransformer;
+    constructor(options: BaseClientOptions<RouterContract, SafeCall>, transport: Transport, transportOptions: Transport extends ClientTransportFactory<any, infer Options> ? Options : never);
+    get call(): ClientCallers<this["_"]["routes"], SafeCall, false>;
+    get stream(): ClientCallers<this["_"]["routes"], SafeCall, true>;
+    protected resolveProcedures(router: TAnyRouterContract, path?: string[]): void;
+    protected buildCallers(): ClientCallers<this['_']['routes'], SafeCall, boolean>;
+}

package/dist/clients/runtime.js

@@ -0,0 +1,81 @@
+import { IsProcedureContract, IsRouterContract } from '@nmtjs/contract';
+import { BaseClient } from "../core.js";
+export class RuntimeContractTransformer {
+    #procedures = new Map();
+    constructor(router) {
+        const registerProcedures = (r, path = []) => {
+            if (IsRouterContract(r)) {
+                for (const [key, route] of Object.entries(r.routes)) {
+                    registerProcedures(route, [...path, key]);
+                }
+            }
+            else if (IsProcedureContract(r)) {
+                const fullName = [...path].join('/');
+                this.#procedures.set(fullName, r);
+            }
+        };
+        registerProcedures(router);
+    }
+    encode(_procedure, payload) {
+        const procedure = this.#procedures.get(_procedure);
+        if (!procedure)
+            throw new Error(`Procedure not found: ${_procedure}`);
+        return procedure.input.encode(payload);
+    }
+    decode(_procedure, payload) {
+        const procedure = this.#procedures.get(_procedure);
+        if (!procedure)
+            throw new Error(`Procedure not found: ${_procedure}`);
+        return procedure.output.decode(payload);
+    }
+}
+export class RuntimeClient extends BaseClient {
+    transformer;
+    #procedures = new Map();
+    #callers;
+    constructor(options, transport, transportOptions) {
+        super(options, transport, transportOptions);
+        this.resolveProcedures(this.options.contract);
+        this.transformer = new RuntimeContractTransformer(this.options.contract);
+        this.#callers = this.buildCallers();
+    }
+    get call() {
+        return this.#callers;
+    }
+    get stream() {
+        return this.#callers;
+    }
+    resolveProcedures(router, path = []) {
+        for (const [key, route] of Object.entries(router.routes)) {
+            if (IsRouterContract(route)) {
+                this.resolveProcedures(route, [...path, key]);
+            }
+            else if (IsProcedureContract(route)) {
+                const fullName = [...path, key].join('/');
+                this.#procedures.set(fullName, route);
+            }
+        }
+    }
+    buildCallers() {
+        const callers = Object.create(null);
+        for (const [name, { stream }] of this.#procedures) {
+            const parts = name.split('/');
+            let current = callers;
+            for (let i = 0; i < parts.length; i++) {
+                const part = parts[i];
+                if (i === parts.length - 1) {
+                    current[part] = (payload, options) => this._call(name, payload, {
+                        ...options,
+                        _stream_response: !!stream,
+                    });
+                }
+                else {
+                    current[part] = current[part] ?? Object.create(null);
+                    current = current[part];
+                }
+            }
+        }
+        return callers;
+    }
+}
+//# sourceMappingURL=runtime.js.map

package/dist/clients/runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../src/clients/runtime.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAA;AAUvE,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAEvC,MAAM,OAAO,0BAA0B;IACrC,WAAW,GAAG,IAAI,GAAG,EAAiC,CAAA;IAEtD,YAAY,MAA0B;QACpC,MAAM,kBAAkB,GAAG,CAAC,CAAiB,EAAE,OAAiB,EAAE,EAAE,EAAE;YACpE,IAAI,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;oBACpD,kBAAkB,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,CAAC,CAAC,CAAA;gBAC3C,CAAC;YACH,CAAC;iBAAM,IAAI,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;gBACpC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAA;YACnC,CAAC;QACH,CAAC,CAAA;QACD,kBAAkB,CAAC,MAAM,CAAC,CAAA;IAC5B,CAAC;IAED,MAAM,CAAC,UAAkB,EAAE,OAAY;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAA;QAClD,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,EAAE,CAAC,CAAA;QACrE,OAAO,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IACxC,CAAC;IAED,MAAM,CAAC,UAAkB,EAAE,OAAY;QACrC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAA;QAClD,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,EAAE,CAAC,CAAA;QACrE,OAAO,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IACzC,CAAC;CACF;AAED,MAAM,OAAO,aAOX,SAAQ,UAMT;IACoB,WAAW,CAA4B;IAEjD,WAAW,GAAG,IAAI,GAAG,EAAiC,CAAA;IACtD,QAAQ,CAAwD;IAEzE,YACE,OAAoD,EACpD,SAAoB,EACpB,gBAKS;QAET,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAA;QAE3C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QAC7C,IAAI,CAAC,WAAW,GAAG,IAAI,0BAA0B,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;QACxE,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,CAAA;IACrC,CAAC;IAED,IAAa,IAAI;QACf,OAAO,IAAI,CAAC,QAA+D,CAAA;IAC7E,CAAC;IAED,IAAa,MAAM;QACjB,OAAO,IAAI,CAAC,QAA8D,CAAA;IAC5E,CAAC;IAES,iBAAiB,CAAC,MAA0B,EAAE,OAAiB,EAAE;QACzE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACzD,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,CAAC,CAAC,CAAA;YAC/C,CAAC;iBAAM,IAAI,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtC,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;gBACzC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;YACvC,CAAC;QACH,CAAC;IACH,CAAC;IAES,YAAY;QAKpB,MAAM,OAAO,GAAwB,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAExD,KAAK,MAAM,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;YAC7B,IAAI,OAAO,GAAG,OAAO,CAAA;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAA;gBACrB,IAAI,CAAC,KAAK,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3B,OAAO,CAAC,IAAI,CAAC,GAAG,CACd,OAAiB,EACjB,OAAoC,EACpC,EAAE,CACF,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE;wBACxB,GAAG,OAAO;wBACV,gBAAgB,EAAE,CAAC,CAAC,MAAM;qBAC3B,CAAC,CAAA;gBACN,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;oBACpD,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,OAAgE,CAAA;IACzE,CAAC;CACF"}

package/dist/clients/static.d.ts

@@ -0,0 +1,13 @@
+import type { TAnyRouterContract } from '@nmtjs/contract';
+import type { BaseClientOptions } from '../core.ts';
+import type { ClientTransportFactory } from '../transport.ts';
+import type { ClientCallers, StaticInputContractTypeProvider, StaticOutputContractTypeProvider } from '../types.ts';
+import { BaseClient } from '../core.ts';
+import { BaseClientTransformer } from '../transformers.ts';
+export declare class StaticClient<Transport extends ClientTransportFactory<any, any> = ClientTransportFactory<any, any>, RouterContract extends TAnyRouterContract = TAnyRouterContract, SafeCall extends boolean = false> extends BaseClient<Transport, RouterContract, SafeCall, StaticInputContractTypeProvider, StaticOutputContractTypeProvider> {
+    protected readonly transformer: BaseClientTransformer;
+    constructor(options: BaseClientOptions<RouterContract, SafeCall>, transport: Transport, transportOptions: Transport extends ClientTransportFactory<any, infer Options> ? Options : never);
+    get call(): ClientCallers<this["_"]["routes"], SafeCall, false>;
+    get stream(): ClientCallers<this["_"]["routes"], SafeCall, true>;
+    protected createProxy<T>(target: Record<string, unknown>, isStream: boolean, path?: string[]): T;
+}

package/dist/clients/static.js

@@ -0,0 +1,27 @@
+import { BaseClient } from "../core.js";
+import { BaseClientTransformer } from "../transformers.js";
+export class StaticClient extends BaseClient {
+    transformer;
+    constructor(options, transport, transportOptions) {
+        super(options, transport, transportOptions);
+        this.transformer = new BaseClientTransformer();
+    }
+    get call() {
+        return this.createProxy(Object.create(null), false);
+    }
+    get stream() {
+        return this.createProxy(Object.create(null), true);
+    }
+    createProxy(target, isStream, path = []) {
+        return new Proxy(target, {
+            get: (obj, prop) => {
+                if (prop === 'then')
+                    return obj;
+                const newPath = [...path, String(prop)];
+                const caller = (payload, options) => this._call(newPath.join('/'), payload, options);
+                return this.createProxy(caller, isStream, newPath);
+            },
+        });
+    }
+}
+//# sourceMappingURL=static.js.map

package/dist/clients/static.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"static.js","sourceRoot":"","sources":["../../src/clients/static.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAA;AAE1D,MAAM,OAAO,YAOX,SAAQ,UAMT;IACoB,WAAW,CAAuB;IAErD,YACE,OAAoD,EACpD,SAAoB,EACpB,gBAKS;QAET,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAA;QAC3C,IAAI,CAAC,WAAW,GAAG,IAAI,qBAAqB,EAAE,CAAA;IAChD,CAAC;IAED,IAAa,IAAI;QACf,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,CAIjD,CAAA;IACH,CAAC;IAED,IAAa,MAAM;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,CAIhD,CAAA;IACH,CAAC;IAES,WAAW,CACnB,MAA+B,EAC/B,QAAiB,EACjB,OAAiB,EAAE;QAEnB,OAAO,IAAI,KAAK,CAAC,MAAM,EAAE;YACvB,GAAG,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;gBACjB,IAAI,IAAI,KAAK,MAAM;oBAAE,OAAO,GAAG,CAAA;gBAC/B,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;gBACvC,MAAM,MAAM,GAAG,CACb,OAAiB,EACjB,OAAoC,EACpC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;gBACpD,OAAO,IAAI,CAAC,WAAW,CAAC,MAAa,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC3D,CAAC;SACF,CAAM,CAAA;IACT,CAAC;CACF"}

package/dist/core.d.ts

@@ -0,0 +1,71 @@
+import type { Future, TypeProvider } from '@nmtjs/common';
+import type { TAnyRouterContract } from '@nmtjs/contract';
+import type { ProtocolBlobMetadata, ProtocolVersion } from '@nmtjs/protocol';
+import type { BaseClientFormat, MessageContext, ProtocolVersionInterface } from '@nmtjs/protocol/client';
+import { ProtocolBlob } from '@nmtjs/protocol';
+import { ProtocolError } from '@nmtjs/protocol/client';
+import type { BaseClientTransformer } from './transformers.ts';
+import type { ClientTransportFactory } from './transport.ts';
+import type { ClientCallers, ClientCallOptions, ResolveAPIRouterRoutes } from './types.ts';
+import { EventEmitter } from './events.ts';
+import { ClientStreams, ServerStreams } from './streams.ts';
+export { ErrorCode, ProtocolBlob, type ProtocolBlobMetadata, } from '@nmtjs/protocol';
+export * from './types.ts';
+export declare class ClientError extends ProtocolError {
+}
+export type ProtocolClientCall = Future<any> & {
+    procedure: string;
+    signal?: AbortSignal;
+};
+export interface BaseClientOptions<RouterContract extends TAnyRouterContract = TAnyRouterContract, SafeCall extends boolean = false> {
+    contract: RouterContract;
+    protocol: ProtocolVersion;
+    format: BaseClientFormat;
+    application?: string;
+    timeout?: number;
+    autoreconnect?: boolean;
+    safe?: SafeCall;
+}
+/**
+ * @todo Add error logging in ClientStreamPull rejection handler for easier debugging
+ * @todo Consider edge case where callId/streamId overflow at MAX_UINT32 with existing entries
+ */
+export declare abstract class BaseClient<TransportFactory extends ClientTransportFactory<any, any> = ClientTransportFactory<any, any>, RouterContract extends TAnyRouterContract = TAnyRouterContract, SafeCall extends boolean = false, InputTypeProvider extends TypeProvider = TypeProvider, OutputTypeProvider extends TypeProvider = TypeProvider> extends EventEmitter<{
+    connected: [];
+    disconnected: [reason: 'server' | 'client' | (string & {})];
+}> {
+    #private;
+    readonly options: BaseClientOptions<RouterContract, SafeCall>;
+    readonly transportFactory: TransportFactory;
+    readonly transportOptions: TransportFactory extends ClientTransportFactory<any, infer U> ? U : never;
+    _: {
+        routes: ResolveAPIRouterRoutes<RouterContract, InputTypeProvider, OutputTypeProvider>;
+        safe: SafeCall;
+    };
+    protected abstract readonly transformer: BaseClientTransformer;
+    abstract call: ClientCallers<this['_']['routes'], SafeCall, false>;
+    abstract stream: ClientCallers<this['_']['routes'], SafeCall, true>;
+    protected calls: Map<number, ProtocolClientCall>;
+    protected transport: TransportFactory extends ClientTransportFactory<any, any, infer T> ? T : never;
+    protected protocol: ProtocolVersionInterface;
+    protected messageContext: MessageContext | null;
+    protected clientStreams: ClientStreams;
+    protected serverStreams: ServerStreams<import("@nmtjs/protocol/client").ProtocolServerStreamInterface<unknown>>;
+    protected rpcStreams: ServerStreams<import("@nmtjs/protocol/client").ProtocolServerStreamInterface<unknown>>;
+    protected callId: number;
+    protected streamId: number;
+    protected cab: AbortController | null;
+    protected reconnectTimeout: number;
+    protected connecting: Promise<void> | null;
+    protected state: 'connected' | 'disconnected';
+    constructor(options: BaseClientOptions<RouterContract, SafeCall>, transportFactory: TransportFactory, transportOptions: TransportFactory extends ClientTransportFactory<any, infer U> ? U : never);
+    get auth(): any;
+    set auth(value: any);
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    blob(source: Blob | ReadableStream | string | AsyncIterable<Uint8Array>, metadata?: ProtocolBlobMetadata): ProtocolBlob;
+    protected _call(procedure: string, payload: any, options?: ClientCallOptions): Promise<any>;
+    protected onConnect(): Promise<void>;
+    protected onDisconnect(reason: 'client' | 'server' | (string & {})): Promise<void>;
+    protected onMessage(buffer: ArrayBufferView): Promise<void>;
+}

package/dist/core.js

@@ -0,0 +1,426 @@
+import { anyAbortSignal, createFuture, MAX_UINT32, noopFn } from '@nmtjs/common';
+import { ClientMessageType, ConnectionType, ErrorCode, ProtocolBlob, ServerMessageType, } from '@nmtjs/protocol';
+import { ProtocolError, ProtocolServerBlobStream, ProtocolServerRPCStream, ProtocolServerStream, versions, } from '@nmtjs/protocol/client';
+import { EventEmitter } from "./events.js";
+import { ClientStreams, ServerStreams } from "./streams.js";
+export { ErrorCode, ProtocolBlob, } from '@nmtjs/protocol';
+export * from "./types.js";
+export class ClientError extends ProtocolError {
+}
+const DEFAULT_RECONNECT_TIMEOUT = 1000;
+const DEFAULT_MAX_RECONNECT_TIMEOUT = 60000;
+/**
+ * @todo Add error logging in ClientStreamPull rejection handler for easier debugging
+ * @todo Consider edge case where callId/streamId overflow at MAX_UINT32 with existing entries
+ */
+export class BaseClient extends EventEmitter {
+    options;
+    transportFactory;
+    transportOptions;
+    _;
+    calls = new Map();
+    transport;
+    protocol;
+    messageContext;
+    clientStreams = new ClientStreams();
+    serverStreams = new ServerStreams();
+    rpcStreams = new ServerStreams();
+    callId = 0;
+    streamId = 0;
+    cab = null;
+    reconnectTimeout = DEFAULT_RECONNECT_TIMEOUT;
+    connecting = null;
+    state = 'disconnected';
+    #auth;
+    constructor(options, transportFactory, transportOptions) {
+        super();
+        this.options = options;
+        this.transportFactory = transportFactory;
+        this.transportOptions = transportOptions;
+        this.protocol = versions[options.protocol];
+        const { format, protocol } = this.options;
+        this.transport = this.transportFactory({ protocol, format }, this.transportOptions);
+        if (this.transport.type === ConnectionType.Bidirectional &&
+            this.options.autoreconnect) {
+            this.on('disconnected', async (reason) => {
+                while (this.state === 'disconnected') {
+                    const timeout = new Promise((resolve) => setTimeout(resolve, this.reconnectTimeout));
+                    this.reconnectTimeout = Math.min(this.reconnectTimeout * 2, DEFAULT_MAX_RECONNECT_TIMEOUT);
+                    await timeout;
+                    await this.connect().catch(noopFn);
+                }
+            });
+            this.on('connected', () => {
+                this.reconnectTimeout = DEFAULT_RECONNECT_TIMEOUT;
+            });
+            if (globalThis.window) {
+                globalThis.window.addEventListener('pageshow', () => {
+                    if (this.state === 'disconnected')
+                        this.connect();
+                });
+            }
+        }
+    }
+    get auth() {
+        return this.#auth;
+    }
+    set auth(value) {
+        this.#auth = value;
+    }
+    connect() {
+        if (this.state === 'connected')
+            return Promise.resolve();
+        if (this.connecting)
+            return this.connecting;
+        const _connect = async () => {
+            if (this.transport.type === ConnectionType.Bidirectional) {
+                this.cab = new AbortController();
+                const protocol = this.protocol;
+                const serverStreams = this.serverStreams;
+                const transport = {
+                    send: (buffer) => {
+                        this.#send(buffer).catch(noopFn);
+                    },
+                };
+                this.messageContext = {
+                    transport,
+                    encoder: this.options.format,
+                    decoder: this.options.format,
+                    addClientStream: (blob) => {
+                        const streamId = this.#getStreamId();
+                        return this.clientStreams.add(blob.source, streamId, blob.metadata);
+                    },
+                    addServerStream(streamId, metadata) {
+                        const stream = new ProtocolServerBlobStream(metadata, {
+                            pull: (controller) => {
+                                transport.send(protocol.encodeMessage(this, ClientMessageType.ServerStreamPull, { streamId, size: 65535 /* 64kb by default */ }));
+                            },
+                            close: () => {
+                                serverStreams.remove(streamId);
+                            },
+                            readableStrategy: { highWaterMark: 0 },
+                        });
+                        serverStreams.add(streamId, stream);
+                        return ({ signal } = {}) => {
+                            if (signal)
+                                signal.addEventListener('abort', () => {
+                                    transport.send(protocol.encodeMessage(this, ClientMessageType.ServerStreamAbort, { streamId }));
+                                    serverStreams.abort(streamId);
+                                }, { once: true });
+                            return stream;
+                        };
+                    },
+                    streamId: this.#getStreamId.bind(this),
+                };
+                return this.transport.connect({
+                    auth: this.auth,
+                    application: this.options.application,
+                    onMessage: this.onMessage.bind(this),
+                    onConnect: this.onConnect.bind(this),
+                    onDisconnect: this.onDisconnect.bind(this),
+                });
+            }
+        };
+        this.connecting = _connect()
+            .then(() => {
+            this.state = 'connected';
+        })
+            .finally(() => {
+            this.connecting = null;
+        });
+        return this.connecting;
+    }
+    async disconnect() {
+        if (this.transport.type === ConnectionType.Bidirectional) {
+            this.cab.abort();
+            await this.transport.disconnect();
+            this.messageContext = null;
+            this.cab = null;
+        }
+    }
+    blob(source, metadata) {
+        return ProtocolBlob.from(source, metadata);
+    }
+    async _call(procedure, payload, options = {}) {
+        const timeout = options.timeout ?? this.options.timeout;
+        const controller = new AbortController();
+        // attach all abort signals
+        const signals = [controller.signal];
+        if (timeout)
+            signals.push(AbortSignal.timeout(timeout));
+        if (options.signal)
+            signals.push(options.signal);
+        if (this.cab?.signal)
+            signals.push(this.cab.signal);
+        const signal = signals.length ? anyAbortSignal(...signals) : undefined;
+        const callId = this.#getCallId();
+        const call = createFuture();
+        call.procedure = procedure;
+        call.signal = signal;
+        this.calls.set(callId, call);
+        // Check if signal is already aborted before proceeding
+        if (signal?.aborted) {
+            this.calls.delete(callId);
+            const error = new ProtocolError(ErrorCode.ClientRequestError, signal.reason);
+            call.reject(error);
+        }
+        else {
+            if (signal) {
+                signal.addEventListener('abort', () => {
+                    call.reject(new ProtocolError(ErrorCode.ClientRequestError, signal.reason));
+                    if (this.transport.type === ConnectionType.Bidirectional &&
+                        this.messageContext) {
+                        const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.RpcAbort, { callId });
+                        this.#send(buffer).catch(noopFn);
+                    }
+                }, { once: true });
+            }
+            try {
+                const transformedPayload = this.transformer.encode(procedure, payload);
+                if (this.transport.type === ConnectionType.Bidirectional) {
+                    const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.Rpc, { callId, procedure, payload: transformedPayload });
+                    await this.#send(buffer, signal);
+                }
+                else {
+                    const response = await this.transport.call({
+                        application: this.options.application,
+                        format: this.options.format,
+                        auth: this.auth,
+                    }, { callId, procedure, payload: transformedPayload }, { signal, _stream_response: options._stream_response });
+                    this.#handleCallResponse(callId, response);
+                }
+            }
+            catch (error) {
+                call.reject(error);
+            }
+        }
+        const result = call.promise.then((value) => {
+            if (value instanceof ProtocolServerRPCStream) {
+                return value.createAsyncIterable(() => {
+                    controller.abort();
+                });
+            }
+            controller.abort();
+            return value;
+        }, (err) => {
+            controller.abort();
+            throw err;
+        });
+        if (this.options.safe) {
+            return await result
+                .then((result) => ({ result }))
+                .catch((error) => ({ error }))
+                .finally(() => {
+                this.calls.delete(callId);
+            });
+        }
+        else {
+            return await result.finally(() => {
+                this.calls.delete(callId);
+            });
+        }
+    }
+    async onConnect() {
+        this.state = 'connected';
+        this.emit('connected');
+    }
+    async onDisconnect(reason) {
+        this.state = 'disconnected';
+        this.emit('disconnected', reason);
+        this.clientStreams.clear(reason);
+        this.serverStreams.clear(reason);
+        this.rpcStreams.clear(reason);
+    }
+    async onMessage(buffer) {
+        if (!this.messageContext)
+            return;
+        const message = this.protocol.decodeMessage(this.messageContext, buffer);
+        switch (message.type) {
+            case ServerMessageType.RpcResponse:
+                this.#handleRPCResponseMessage(message);
+                break;
+            case ServerMessageType.RpcStreamResponse:
+                this.#handleRPCStreamResponseMessage(message);
+                break;
+            case ServerMessageType.RpcStreamChunk:
+                this.rpcStreams.push(message.callId, message.chunk);
+                break;
+            case ServerMessageType.RpcStreamEnd:
+                this.rpcStreams.end(message.callId);
+                this.calls.delete(message.callId);
+                break;
+            case ServerMessageType.RpcStreamAbort:
+                this.rpcStreams.abort(message.callId);
+                this.calls.delete(message.callId);
+                break;
+            case ServerMessageType.ServerStreamPush:
+                this.serverStreams.push(message.streamId, message.chunk);
+                break;
+            case ServerMessageType.ServerStreamEnd:
+                this.serverStreams.end(message.streamId);
+                break;
+            case ServerMessageType.ServerStreamAbort:
+                this.serverStreams.abort(message.streamId);
+                break;
+            case ServerMessageType.ClientStreamPull:
+                this.clientStreams.pull(message.streamId, message.size).then((chunk) => {
+                    if (chunk) {
+                        const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.ClientStreamPush, { streamId: message.streamId, chunk });
+                        this.#send(buffer).catch(noopFn);
+                    }
+                    else {
+                        const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.ClientStreamEnd, { streamId: message.streamId });
+                        this.#send(buffer).catch(noopFn);
+                        this.clientStreams.end(message.streamId);
+                    }
+                }, () => {
+                    const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.ClientStreamAbort, { streamId: message.streamId });
+                    this.#send(buffer).catch(noopFn);
+                    this.clientStreams.remove(message.streamId);
+                });
+                break;
+            case ServerMessageType.ClientStreamAbort:
+                this.clientStreams.abort(message.streamId);
+                break;
+        }
+    }
+    #handleRPCResponseMessage(message) {
+        const { callId, result, error } = message;
+        const call = this.calls.get(callId);
+        if (!call)
+            return;
+        if (error) {
+            call.reject(new ProtocolError(error.code, error.message, error.data));
+        }
+        else {
+            try {
+                const transformed = this.transformer.decode(call.procedure, result);
+                call.resolve(transformed);
+            }
+            catch (error) {
+                call.reject(new ProtocolError(ErrorCode.ClientRequestError, 'Unable to decode response', error));
+            }
+        }
+    }
+    #handleRPCStreamResponseMessage(message) {
+        const call = this.calls.get(message.callId);
+        if (message.error) {
+            if (!call)
+                return;
+            call.reject(new ProtocolError(message.error.code, message.error.message, message.error.data));
+        }
+        else {
+            if (call) {
+                const { procedure, signal } = call;
+                const stream = new ProtocolServerRPCStream({
+                    start: (controller) => {
+                        if (signal) {
+                            if (signal.aborted)
+                                controller.error(signal.reason);
+                            else
+                                signal.addEventListener('abort', () => {
+                                    controller.error(signal.reason);
+                                    if (this.rpcStreams.has(message.callId)) {
+                                        this.rpcStreams.remove(message.callId);
+                                        this.calls.delete(message.callId);
+                                        if (this.messageContext) {
+                                            const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.RpcAbort, { callId: message.callId, reason: signal.reason });
+                                            this.#send(buffer).catch(noopFn);
+                                        }
+                                    }
+                                }, { once: true });
+                        }
+                    },
+                    transform: (chunk) => {
+                        return this.transformer.decode(procedure, this.options.format.decode(chunk));
+                    },
+                    pull: () => {
+                        const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.RpcPull, { callId: message.callId });
+                        this.#send(buffer).catch(noopFn);
+                    },
+                    readableStrategy: { highWaterMark: 0 },
+                });
+                this.rpcStreams.add(message.callId, stream);
+                call.resolve(stream);
+            }
+            else {
+                // Call not found, but stream response received
+                // This can happen if the call was aborted or timed out
+                // Need to send an abort for the stream to avoid resource leaks from server side
+                if (this.messageContext) {
+                    const buffer = this.protocol.encodeMessage(this.messageContext, ClientMessageType.RpcAbort, { callId: message.callId });
+                    this.#send(buffer).catch(noopFn);
+                }
+            }
+        }
+    }
+    #handleCallResponse(callId, response) {
+        const call = this.calls.get(callId);
+        if (response.type === 'rpc_stream') {
+            if (call) {
+                const stream = new ProtocolServerStream({
+                    transform: (chunk) => {
+                        return this.transformer.decode(call.procedure, this.options.format.decode(chunk));
+                    },
+                });
+                this.rpcStreams.add(callId, stream);
+                call.resolve(({ signal }) => {
+                    response.stream.pipeTo(stream.writable, { signal }).catch(noopFn);
+                    return stream;
+                });
+            }
+            else {
+                // Call not found, but stream response received
+                // This can happen if the call was aborted or timed out
+                // Need to cancel the stream to avoid resource leaks from server side
+                response.stream.cancel().catch(noopFn);
+            }
+        }
+        else if (response.type === 'blob') {
+            if (call) {
+                const { metadata, source } = response;
+                const stream = new ProtocolServerBlobStream(metadata);
+                this.serverStreams.add(this.#getStreamId(), stream);
+                call.resolve(({ signal }) => {
+                    source.pipeTo(stream.writable, { signal }).catch(noopFn);
+                    return stream;
+                });
+            }
+            else {
+                // Call not found, but blob response received
+                // This can happen if the call was aborted or timed out
+                // Need to cancel the stream to avoid resource leaks from server side
+                response.source.cancel().catch(noopFn);
+            }
+        }
+        else if (response.type === 'rpc') {
+            if (!call)
+                return;
+            try {
+                const transformed = this.transformer.decode(call.procedure, response.result);
+                call.resolve(transformed);
+            }
+            catch (error) {
+                call.reject(new ProtocolError(ErrorCode.ClientRequestError, 'Unable to decode response', error));
+            }
+        }
+    }
+    #send(buffer, signal) {
+        if (this.transport.type === ConnectionType.Unidirectional)
+            throw new Error('Invalid transport type for send');
+        return this.transport.send(buffer, { signal });
+    }
+    #getStreamId() {
+        if (this.streamId >= MAX_UINT32) {
+            this.streamId = 0;
+        }
+        return this.streamId++;
+    }
+    #getCallId() {
+        if (this.callId >= MAX_UINT32) {
+            this.callId = 0;
+        }
+        return this.callId++;
+    }
+}
+//# sourceMappingURL=core.js.map