@cloudpss/ubrpc 0.4.10

package/README.md

@@ -0,0 +1,3 @@
+# @cloudpss/ubrpc
+
+Rpc server/client build on websocket and ubjson.

package/dist/auth.d.ts

@@ -0,0 +1,5 @@
+/// <reference types="ws" />
+import type WebSocket from 'isomorphic-ws';
+import type { ConnectionID, RpcMetadata } from './types/payload.js';
+/** 接受认证消息 */
+export declare function waitAuth(socket: WebSocket, seq?: number, id?: ConnectionID): Promise<[number, ConnectionID, RpcMetadata]>;

package/dist/auth.js

@@ -0,0 +1,52 @@
+import { decodePayload, deserializeError } from './utils/serialize.js';
+import { VERSION } from './version.js';
+const AUTH_TIMEOUT = 5000;
+/** 接受认证消息 */
+export function waitAuth(socket, seq, id) {
+    return new Promise((resolve, reject) => {
+        const isResponse = seq != null && id != null;
+        const timeout = setTimeout(() => {
+            reject(new Error(`Failed to authenticate, no ${isResponse ? 'response' : 'request'} in ${AUTH_TIMEOUT}ms`));
+            finalize();
+        }, AUTH_TIMEOUT);
+        const handleMessage = (ev) => {
+            try {
+                const payload = decodePayload(ev.data);
+                if (!payload || payload.type !== 'auth') {
+                    if (isResponse)
+                        return reject(new Error(`Failed to authenticate, bad response from server`));
+                    else
+                        return reject(new Error(`Failed to authenticate, bad request from client`));
+                }
+                if (payload.version !== VERSION) {
+                    return reject(new Error(`Failed to authenticate, version not match, expected ${VERSION}, got ${payload.version}`));
+                }
+                if (isResponse) {
+                    if (payload.seq !== seq) {
+                        return reject(new Error(`Failed to authenticate, bad response seq from server`));
+                    }
+                    if (payload.id !== id) {
+                        return reject(new Error(`Failed to authenticate, bad response id from server`));
+                    }
+                }
+                if (payload.error) {
+                    return reject(deserializeError(payload.error));
+                }
+                const metadata = typeof payload.metadata != 'object' ? {} : payload.metadata ?? {};
+                resolve([payload.seq, payload.id, metadata]);
+            }
+            catch (ex) {
+                reject(ex);
+            }
+            finally {
+                finalize();
+            }
+        };
+        const finalize = () => {
+            clearTimeout(timeout);
+            socket.removeListener('message', handleMessage);
+        };
+        socket.addEventListener('message', handleMessage, { once: true });
+    });
+}
+//# sourceMappingURL=auth.js.map

package/dist/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../src/auth.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACvE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,MAAM,YAAY,GAAG,IAAI,CAAC;AAE1B,aAAa;AACb,MAAM,UAAU,QAAQ,CACpB,MAAiB,EACjB,GAAY,EACZ,EAAiB;IAEjB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACnC,MAAM,UAAU,GAAG,GAAG,IAAI,IAAI,IAAI,EAAE,IAAI,IAAI,CAAC;QAC7C,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,8BAA8B,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,OAAO,YAAY,IAAI,CAAC,CAAC,CAAC;YAC5G,QAAQ,EAAE,CAAC;QACf,CAAC,EAAE,YAAY,CAAC,CAAC;QACjB,MAAM,aAAa,GAAG,CAAC,EAA0B,EAAQ,EAAE;YACvD,IAAI;gBACA,MAAM,OAAO,GAAG,aAAa,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;gBACvC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE;oBACrC,IAAI,UAAU;wBAAE,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC,CAAC;;wBACxF,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC,CAAC;iBACpF;gBACD,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO,EAAE;oBAC7B,OAAO,MAAM,CACT,IAAI,KAAK,CACL,uDAAuD,OAAO,SAAS,OAAO,CAAC,OAAO,EAAE,CAC3F,CACJ,CAAC;iBACL;gBACD,IAAI,UAAU,EAAE;oBACZ,IAAI,OAAO,CAAC,GAAG,KAAK,GAAG,EAAE;wBACrB,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC,CAAC;qBACpF;oBACD,IAAI,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE;wBACnB,OAAO,MAAM,CAAC,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC,CAAC;qBACnF;iBACJ;gBACD,IAAI,OAAO,CAAC,KAAK,EAAE;oBACf,OAAO,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;iBAClD;gBACD,MAAM,QAAQ,GAAG,OAAO,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;gBACnF,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC;aAChD;YAAC,OAAO,EAAE,EAAE;gBACT,MAAM,CAAC,EAAE,CAAC,CAAC;aACd;oBAAS;gBACN,QAAQ,EAAE,CAAC;aACd;QACL,CAAC,CAAC;QACF,MAAM,QAAQ,GAAG,GAAS,EAAE;YACxB,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;QACpD,CAAC,CAAC;QACF,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACP,CAAC"}

package/dist/client.d.ts

@@ -0,0 +1,14 @@
+/// <reference types="ws" />
+import WebSocket from 'isomorphic-ws';
+import { RpcSocket } from './socket.js';
+import type { RpcMetadata } from './types/payload.js';
+import type { RpcObject } from './types/utils.js';
+/** 由 WS Client 建立的 RPC 连接 */
+export declare class RpcClientSocket<TRemote extends {}, TLocal extends {}> extends RpcSocket<TRemote, TLocal> {
+    constructor(url: string | URL, metadata?: RpcMetadata, local?: RpcObject<TLocal>);
+    readonly url: string;
+    /** @inheritdoc */
+    protected authSocket(): Promise<RpcMetadata>;
+    /** @inheritdoc */
+    protected onClose(ev: WebSocket.CloseEvent): void;
+}

package/dist/client.js

@@ -0,0 +1,39 @@
+import WebSocket from 'isomorphic-ws';
+import { v4 } from 'uuid';
+import { waitAuth } from './auth.js';
+import { RpcSocket } from './socket.js';
+import { send } from './utils/messaging.js';
+import { VERSION } from './version.js';
+/** 由 WS Client 建立的 RPC 连接 */
+export class RpcClientSocket extends RpcSocket {
+    constructor(url, metadata, local) {
+        super(v4(), local);
+        this._localMetadata = metadata;
+        this.url = typeof url == 'string' ? url : url.href;
+        this.seq = 2;
+        this.socket = new WebSocket(this.url);
+    }
+    /** @inheritdoc */
+    async authSocket() {
+        const socket = this.socket;
+        const seq = this.nextSeq();
+        send(socket, 'auth', {
+            seq,
+            id: this.id,
+            version: VERSION,
+            metadata: this.localMetadata ?? {},
+        });
+        const [, , metadata] = await waitAuth(socket, seq, this.id);
+        return metadata;
+    }
+    /** @inheritdoc */
+    onClose(ev) {
+        super.onClose(ev);
+        if (ev.code === 1000)
+            return;
+        setTimeout(() => {
+            this.socket = new WebSocket(this.url);
+        }, 1000);
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,6BAA6B;AAC7B,MAAM,OAAO,eAAuD,SAAQ,SAA0B;IAClG,YAAY,GAAiB,EAAE,QAAsB,EAAE,KAAyB;QAC5E,KAAK,CAAC,EAAE,EAAkB,EAAE,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;QAC/B,IAAI,CAAC,GAAG,GAAG,OAAO,GAAG,IAAI,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;QACnD,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC1C,CAAC;IAED,kBAAkB;IACC,KAAK,CAAC,UAAU;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3B,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE;YACjB,GAAG;YACH,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,OAAO,EAAE,OAAO;YAChB,QAAQ,EAAE,IAAI,CAAC,aAAa,IAAI,EAAE;SACrC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,AAAD,EAAG,QAAQ,CAAC,GAAG,MAAM,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5D,OAAO,QAAQ,CAAC;IACpB,CAAC;IACD,kBAAkB;IACC,OAAO,CAAC,EAAwB;QAC/C,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClB,IAAI,EAAE,CAAC,IAAI,KAAK,IAAI;YAAE,OAAO;QAE7B,UAAU,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1C,CAAC,EAAE,IAAI,CAAC,CAAC;IACb,CAAC;CACJ"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { RpcClientSocket } from './client.js';
+export { RpcSocket } from './socket.js';
+export { RpcServer, RpcServerSocket } from './server.js';

package/dist/index.js

@@ -0,0 +1,4 @@
+export { RpcClientSocket } from './client.js';
+export { RpcSocket } from './socket.js';
+export { RpcServer, RpcServerSocket } from './server.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC"}

package/dist/server.d.ts

@@ -0,0 +1,44 @@
+/// <reference types="ws" />
+import type WebSocket from 'isomorphic-ws';
+import { RpcSocket } from './socket.js';
+import type { ConnectionID, RpcMetadata } from './types/payload.js';
+import type { RpcObject } from './types/utils.js';
+declare const kOnClose: unique symbol;
+declare const kMetadata: unique symbol;
+declare const kReplaceSocket: unique symbol;
+/** 认证过的 socket */
+type WebSocketWithMetadata = WebSocket & {
+    [kMetadata]: RpcMetadata;
+};
+/** 由 WS Server 建立的 RPC 连接组 */
+export declare class RpcServer<TRemote extends {}, TLocal extends {}> {
+    /** 认证客户端 */
+    readonly authenticator: (metadata: RpcMetadata) => RpcMetadata | Promise<RpcMetadata>;
+    constructor(local?: RpcObject<TLocal> | ((socket: RpcServerSocket<TRemote, TLocal>) => RpcObject<TLocal>), 
+    /** 认证客户端 */
+    authenticator?: (metadata: RpcMetadata) => RpcMetadata | Promise<RpcMetadata>);
+    /** 用于响应调用的本地对象 */
+    readonly local?: (socket: RpcServerSocket<TRemote, TLocal>) => RpcObject<TLocal>;
+    readonly sockets: Map<string, RpcServerSocket<TRemote, TLocal>>;
+    /** WebSocket 连接后调用此方法建立 RPC 连接 */
+    connect(socket: WebSocket): Promise<RpcServerSocket<TRemote, TLocal>>;
+    /** 认证客户端 */
+    protected authSocket(socket: WebSocket): Promise<[ConnectionID, RpcMetadata]>;
+    private readonly disconnectingSockets;
+    /** WebSocket 断开时调用 */
+    [kOnClose](socket: RpcServerSocket<TRemote, TLocal>, ev: WebSocket.CloseEvent): void;
+}
+/** 由 WS Server 建立的 RPC 连接 */
+export declare class RpcServerSocket<TRemote extends object, TLocal extends object> extends RpcSocket<TRemote, TLocal> {
+    readonly server: RpcServer<TRemote, TLocal>;
+    constructor(id: ConnectionID, server: RpcServer<TRemote, TLocal>, socket: WebSocketWithMetadata);
+    /** @inheritdoc */
+    protected authSocket(): Promise<RpcMetadata>;
+    /** @inheritdoc */
+    protected get local(): RpcObject<TLocal> | undefined;
+    /** @inheritdoc */
+    protected onClose(ev: WebSocket.CloseEvent): void;
+    /** 替换 socket */
+    [kReplaceSocket](newSocket: WebSocketWithMetadata): void;
+}
+export {};

package/dist/server.js

@@ -0,0 +1,119 @@
+import { waitAuth } from './auth.js';
+import { RpcSocket } from './socket.js';
+import { send } from './utils/messaging.js';
+import { serializeError } from './utils/serialize.js';
+import { VERSION } from './version.js';
+const kOnClose = Symbol('kOnClose');
+const kMetadata = Symbol('kMetadata');
+const kReplaceSocket = Symbol('kReplaceSocket');
+/** 由 WS Server 建立的 RPC 连接组 */
+export class RpcServer {
+    constructor(local, 
+    /** 认证客户端 */
+    authenticator = () => ({})) {
+        this.authenticator = authenticator;
+        this.sockets = new Map();
+        this.disconnectingSockets = new Map();
+        if (local == null) {
+            this.local = undefined;
+        }
+        else if (typeof local == 'function') {
+            this.local = local;
+        }
+        else {
+            this.local = () => local;
+        }
+    }
+    /** WebSocket 连接后调用此方法建立 RPC 连接 */
+    async connect(socket) {
+        const s = socket;
+        const [id, metadata] = await this.authSocket(socket);
+        s[kMetadata] = metadata;
+        const exist = this.sockets.get(id);
+        if (!exist) {
+            const client = new RpcServerSocket(id, this, s);
+            this.sockets.set(id, client);
+            return client;
+        }
+        else {
+            const tid = this.disconnectingSockets.get(id);
+            if (tid != null) {
+                this.disconnectingSockets.delete(id);
+                clearTimeout(tid);
+            }
+            exist[kReplaceSocket](s);
+            return exist;
+        }
+    }
+    /** 认证客户端 */
+    async authSocket(socket) {
+        let seq, id, remoteMetadata;
+        try {
+            [seq, id, remoteMetadata] = await waitAuth(socket);
+            const localMetadata = await this.authenticator(remoteMetadata);
+            send(socket, 'auth', {
+                seq,
+                id,
+                version: VERSION,
+                metadata: localMetadata,
+            });
+            return [id, remoteMetadata];
+        }
+        catch (ex) {
+            if (seq != null && id != null) {
+                send(socket, 'auth', {
+                    seq,
+                    id,
+                    version: VERSION,
+                    metadata: {},
+                    error: serializeError(ex),
+                });
+            }
+            else {
+                send(socket, 'error', {
+                    seq: 1,
+                    error: serializeError(ex),
+                });
+            }
+            socket.close(3401, 'Auth error');
+            throw ex;
+        }
+    }
+    /** WebSocket 断开时调用 */
+    [kOnClose](socket, ev) {
+        const delay = ev.code === 1000 ? 0 : 5000;
+        const tid = setTimeout(() => {
+            socket.destroy();
+            this.sockets.delete(socket.id);
+            this.disconnectingSockets.delete(socket.id);
+        }, delay);
+        this.disconnectingSockets.set(socket.id, tid);
+    }
+}
+/** 由 WS Server 建立的 RPC 连接 */
+export class RpcServerSocket extends RpcSocket {
+    constructor(id, server, socket) {
+        super(id);
+        this.server = server;
+        this.seq = 1;
+        this.socket = socket;
+    }
+    /** @inheritdoc */
+    authSocket() {
+        return Promise.resolve(this.socket[kMetadata]);
+    }
+    /** @inheritdoc */
+    get local() {
+        return this.server.local?.(this);
+    }
+    /** @inheritdoc */
+    onClose(ev) {
+        super.onClose(ev);
+        this.server[kOnClose](this, ev);
+    }
+    /** 替换 socket */
+    [kReplaceSocket](newSocket) {
+        this.socket = newSocket;
+    }
+}
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC,OAAO,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,MAAM,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;AACpC,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;AACtC,MAAM,cAAc,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAKhD,8BAA8B;AAC9B,MAAM,OAAO,SAAS;IAClB,YACI,KAA6F;IAC7F,YAAY;IACH,gBAA+E,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC;QAAzF,kBAAa,GAAb,aAAa,CAA4E;QAa7F,YAAO,GAAG,IAAI,GAAG,EAA4C,CAAC;QAuDtD,yBAAoB,GAAG,IAAI,GAAG,EAAyC,CAAC;QAlErF,IAAI,KAAK,IAAI,IAAI,EAAE;YACf,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;SAC1B;aAAM,IAAI,OAAO,KAAK,IAAI,UAAU,EAAE;YACnC,IAAI,CAAC,KAAK,GAAG,KAAwE,CAAC;SACzF;aAAM;YACH,IAAI,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC;SAC5B;IACL,CAAC;IAMD,kCAAkC;IAClC,KAAK,CAAC,OAAO,CAAC,MAAiB;QAC3B,MAAM,CAAC,GAAG,MAA+B,CAAC;QAC1C,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC,CAAC,SAAS,CAAC,GAAG,QAAQ,CAAC;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,EAAE;YACR,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAChD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC7B,OAAO,MAAM,CAAC;SACjB;aAAM;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,GAAG,IAAI,IAAI,EAAE;gBACb,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACrC,YAAY,CAAC,GAAG,CAAC,CAAC;aACrB;YACD,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;YACzB,OAAO,KAAK,CAAC;SAChB;IACL,CAAC;IACD,YAAY;IACF,KAAK,CAAC,UAAU,CAAC,MAAiB;QACxC,IAAI,GAAG,EAAE,EAAE,EAAE,cAAc,CAAC;QAC5B,IAAI;YACA,CAAC,GAAG,EAAE,EAAE,EAAE,cAAc,CAAC,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE;gBACjB,GAAG;gBACH,EAAE;gBACF,OAAO,EAAE,OAAO;gBAChB,QAAQ,EAAE,aAAa;aAC1B,CAAC,CAAC;YACH,OAAO,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;SAC/B;QAAC,OAAO,EAAE,EAAE;YACT,IAAI,GAAG,IAAI,IAAI,IAAI,EAAE,IAAI,IAAI,EAAE;gBAC3B,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE;oBACjB,GAAG;oBACH,EAAE;oBACF,OAAO,EAAE,OAAO;oBAChB,QAAQ,EAAE,EAAE;oBACZ,KAAK,EAAE,cAAc,CAAC,EAAE,CAAC;iBAC5B,CAAC,CAAC;aACN;iBAAM;gBACH,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE;oBAClB,GAAG,EAAE,CAAC;oBACN,KAAK,EAAE,cAAc,CAAC,EAAE,CAAC;iBAC5B,CAAC,CAAC;aACN;YACD,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YACjC,MAAM,EAAE,CAAC;SACZ;IACL,CAAC;IAGD,sBAAsB;IACtB,CAAC,QAAQ,CAAC,CAAC,MAAwC,EAAE,EAAwB;QACzE,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAC1C,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,EAAE;YACxB,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC,EAAE,KAAK,CAAC,CAAC;QACV,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;IAClD,CAAC;CACJ;AAED,6BAA6B;AAC7B,MAAM,OAAO,eAA+D,SAAQ,SAA0B;IAC1G,YAAY,EAAgB,EAAW,MAAkC,EAAE,MAA6B;QACpG,KAAK,CAAC,EAAE,CAAC,CAAC;QADyB,WAAM,GAAN,MAAM,CAA4B;QAErE,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC;QACb,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IACD,kBAAkB;IACR,UAAU;QAChB,OAAO,OAAO,CAAC,OAAO,CAAE,IAAI,CAAC,MAAgC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9E,CAAC;IACD,kBAAkB;IAClB,IAAuB,KAAK;QACxB,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC;IACD,kBAAkB;IACC,OAAO,CAAC,EAAwB;QAC/C,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClB,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACpC,CAAC;IACD,gBAAgB;IAChB,CAAC,cAAc,CAAC,CAAC,SAAgC;QAC7C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;IAC5B,CAAC;CACJ"}

package/dist/socket.d.ts

@@ -0,0 +1,65 @@
+/// <reference types="ws" />
+import type WebSocket from 'isomorphic-ws';
+import { Observable } from 'rxjs';
+import type { ConnectionID, RpcCallPayload, RpcMetadata, RpcNotifyPayload, RpcPayload, RpcSubscribePayload } from './types/payload.js';
+import type { Methods, Notifications, RpcObject, RpcParameters, RpcReturns, Subjects } from './types/utils.js';
+/** RPC 连接 */
+export declare abstract class RpcSocket<TRemote extends {}, TLocal extends {}> {
+    readonly id: ConnectionID;
+    constructor(id: ConnectionID, local?: RpcObject<TLocal>);
+    protected _localMetadata?: RpcMetadata;
+    protected _remoteMetadata?: RpcMetadata;
+    /** 本地认证信息 */
+    get localMetadata(): RpcMetadata | undefined;
+    /** 远程认证信息 */
+    get remoteMetadata(): RpcMetadata | undefined;
+    /** 连接是否已认证 */
+    get authenticated(): boolean;
+    private readonly __local?;
+    /** 用于响应调用的本地对象 */
+    protected get local(): RpcObject<TLocal> | undefined;
+    private __socket?;
+    /** 作为底层传输的 WebSocket */
+    get socket(): WebSocket;
+    protected set socket(value: WebSocket);
+    private __readyCallbacks;
+    protected ready: Promise<void>;
+    private readonly __handlers;
+    /** 初始化 WebSocket */
+    protected initSocket(oldValue: WebSocket | undefined, newValue: WebSocket): Promise<void>;
+    /** 认证 WebSocket */
+    protected abstract authSocket(): Promise<RpcMetadata>;
+    /** 响应 WebSocket error */
+    protected onError(_ev: WebSocket.ErrorEvent): void;
+    /** 响应 WebSocket open */
+    protected onOpen(_ev: WebSocket.Event): void;
+    /** 响应 WebSocket close */
+    protected onClose(ev: WebSocket.CloseEvent): void;
+    /** 响应 WebSocket message */
+    protected onMessage(ev: WebSocket.MessageEvent): void;
+    /** 响应 Rpc 消息 */
+    protected onPayload(payload: RpcPayload): boolean;
+    /** 调用本地方法 */
+    protected localCall(payload: RpcCallPayload | RpcNotifyPayload): Promise<void>;
+    private readonly localSubscription;
+    /** 调用本地方法 */
+    protected localSubscribe(payload: RpcSubscribePayload): Promise<void>;
+    /** 发送数据 */
+    protected sendPayload<T extends RpcPayload['type']>(type: T, info: Omit<RpcPayload & {
+        type: T;
+    }, 'type'>): Promise<void>;
+    /** 序列号 */
+    protected seq: number;
+    /** 获取下一个序列号 */
+    protected nextSeq(): number;
+    private readonly pendingCalls;
+    /** 调用远程方法 */
+    call<TMethod extends Methods<TRemote>>(method: TMethod, ...args: RpcParameters<TRemote[TMethod]>): Promise<RpcReturns<TRemote[TMethod]>>;
+    /** 调用远程方法，放弃返回值 */
+    notify<TNotification extends Notifications<TRemote>>(method: TNotification, ...args: RpcParameters<TRemote[TNotification]>): void;
+    private readonly pendingSubscriptions;
+    /** 调用远程订阅 */
+    subscribe<TSubject extends Subjects<TRemote>>(method: TSubject, ...args: RpcParameters<TRemote[TSubject]>): Observable<RpcReturns<TRemote[TSubject]>>;
+    /** 结束 */
+    destroy(): void;
+}

package/dist/socket.js

@@ -0,0 +1,276 @@
+import { from, Observable } from 'rxjs';
+import { send } from './utils/messaging.js';
+import { decodePayload, deserializeError, serializeError } from './utils/serialize.js';
+import { ready } from './utils/websocket.js';
+/** RPC 连接 */
+export class RpcSocket {
+    constructor(id, local) {
+        this.id = id;
+        this.__handlers = Object.freeze({
+            open: (ev) => this.onOpen(ev),
+            close: (ev) => this.onClose(ev),
+            error: (ev) => this.onError(ev),
+            message: (ev) => this.onMessage(ev),
+        });
+        this.localSubscription = new Map();
+        /** 序列号 */
+        this.seq = 0;
+        this.pendingCalls = new Map();
+        this.pendingSubscriptions = new Map();
+        this.__local = local;
+        this.ready = new Promise((...args) => (this.__readyCallbacks = args));
+    }
+    /** 本地认证信息 */
+    get localMetadata() {
+        return this._localMetadata;
+    }
+    /** 远程认证信息 */
+    get remoteMetadata() {
+        return this._remoteMetadata;
+    }
+    /** 连接是否已认证 */
+    get authenticated() {
+        return this._remoteMetadata != null;
+    }
+    /** 用于响应调用的本地对象 */
+    get local() {
+        return this.__local;
+    }
+    /** 作为底层传输的 WebSocket */
+    get socket() {
+        if (!this.__socket)
+            throw new Error(`Socket not initialized`);
+        return this.__socket;
+    }
+    set socket(value) {
+        if (this.__socket === value)
+            return;
+        this.__socket = value;
+        void this.initSocket(this.__socket, value).then(...this.__readyCallbacks);
+    }
+    /** 初始化 WebSocket */
+    async initSocket(oldValue, newValue) {
+        try {
+            if (oldValue) {
+                oldValue.removeEventListener('open', this.__handlers.open);
+                oldValue.removeEventListener('close', this.__handlers.close);
+                oldValue.removeEventListener('error', this.__handlers.error);
+                oldValue.removeEventListener('message', this.__handlers.message);
+            }
+            await ready(newValue);
+            const info = await this.authSocket();
+            if (this.__socket === newValue) {
+                newValue.addEventListener('open', this.__handlers.open);
+                newValue.addEventListener('close', this.__handlers.close);
+                newValue.addEventListener('error', this.__handlers.error);
+                newValue.addEventListener('message', this.__handlers.message);
+                this._remoteMetadata = info ?? {};
+            }
+        }
+        catch (ex) {
+            this._remoteMetadata = undefined;
+            throw ex;
+        }
+    }
+    /** 响应 WebSocket error */
+    onError(_ev) {
+        this._remoteMetadata = undefined;
+    }
+    /** 响应 WebSocket open */
+    onOpen(_ev) {
+        //
+    }
+    /** 响应 WebSocket close */
+    onClose(ev) {
+        this._remoteMetadata = undefined;
+        if (ev.code === 1000) {
+            this.destroy();
+        }
+        else {
+            this.ready = new Promise((...args) => (this.__readyCallbacks = args));
+        }
+    }
+    /** 响应 WebSocket message */
+    onMessage(ev) {
+        const payload = decodePayload(ev.data);
+        let error;
+        if (payload) {
+            const handled = this.onPayload(payload);
+            if (!handled)
+                error = [payload.seq, `Unrecognized message, not handled.`];
+        }
+        else {
+            error = [this.nextSeq(), `Invalid message, unknown format.`];
+        }
+        if (error) {
+            void this.sendPayload('error', {
+                seq: error[0],
+                error: serializeError(new SyntaxError(error[1])),
+            });
+        }
+    }
+    /** 响应 Rpc 消息 */
+    onPayload(payload) {
+        switch (payload.type) {
+            case 'call':
+            case 'notify':
+                void this.localCall(payload);
+                return true;
+            case 'return': {
+                const pending = this.pendingCalls.get(payload.seq);
+                // 即使不存在等待的请求，也认为响应是有效的
+                if (!pending)
+                    return true;
+                this.pendingCalls.delete(payload.seq);
+                if (payload.error) {
+                    pending[1](deserializeError(payload.error));
+                }
+                else {
+                    pending[0](payload.result);
+                }
+                return true;
+            }
+            case 'subscribe':
+                void this.localSubscribe(payload);
+                return true;
+            case 'unsubscribe': {
+                const subscription = this.localSubscription.get(payload.seq);
+                // 即使不存在对应订阅，也认为响应是有效的
+                if (!subscription)
+                    return true;
+                subscription.unsubscribe();
+                this.localSubscription.delete(payload.seq);
+                return true;
+            }
+            case 'publish': {
+                const subscriber = this.pendingSubscriptions.get(payload.seq);
+                // 即使不存在对应订阅，也认为响应是有效的
+                if (!subscriber) {
+                    void this.sendPayload('unsubscribe', { seq: payload.seq });
+                    return true;
+                }
+                if (payload.error) {
+                    subscriber.error(deserializeError(payload.error));
+                }
+                else if (payload.complete) {
+                    subscriber.complete();
+                }
+                else {
+                    subscriber.next(payload.next);
+                }
+                return true;
+            }
+            case 'error':
+                return true;
+            default:
+                return false;
+        }
+    }
+    /** 调用本地方法 */
+    async localCall(payload) {
+        const noReturn = payload.type === 'notify';
+        const method = this.local ? this.local[payload.method] : undefined;
+        const seq = payload.seq;
+        if (typeof method != 'function') {
+            if (noReturn)
+                return;
+            return this.sendPayload('return', {
+                seq,
+                error: serializeError(new TypeError(`${payload.method} is not a function`)),
+            });
+        }
+        try {
+            const result = await Reflect.apply(method, this.local, payload.args);
+            if (noReturn)
+                return;
+            return this.sendPayload('return', { seq, result });
+        }
+        catch (ex) {
+            if (noReturn)
+                return;
+            return this.sendPayload('return', { seq, error: serializeError(ex) });
+        }
+    }
+    /** 调用本地方法 */
+    async localSubscribe(payload) {
+        const method = this.local ? this.local[payload.method] : undefined;
+        const seq = payload.seq;
+        if (typeof method != 'function') {
+            return this.sendPayload('publish', {
+                seq,
+                error: serializeError(new TypeError(`${payload.method} is not a function`)),
+            });
+        }
+        try {
+            const result = from((await Reflect.apply(method, this.local, payload.args)));
+            const subscription = result.subscribe({
+                next: (value) => {
+                    void this.sendPayload('publish', { seq, next: value });
+                },
+                error: (err) => {
+                    this.localSubscription.delete(payload.seq);
+                    void this.sendPayload('publish', { seq, error: serializeError(err) });
+                },
+                complete: () => {
+                    this.localSubscription.delete(payload.seq);
+                    void this.sendPayload('publish', { seq, complete: true });
+                },
+            });
+            this.localSubscription.set(seq, subscription);
+        }
+        catch (ex) {
+            return this.sendPayload('publish', { seq, error: serializeError(ex) });
+        }
+    }
+    /** 发送数据 */
+    async sendPayload(type, info) {
+        await this.ready;
+        send(this.socket, type, info);
+    }
+    /** 获取下一个序列号 */
+    nextSeq() {
+        const seq = this.seq;
+        this.seq += 2;
+        return seq;
+    }
+    /** 调用远程方法 */
+    call(method, ...args) {
+        return new Promise((resolve, reject) => {
+            const seq = this.nextSeq();
+            void this.sendPayload('call', { seq, method, args });
+            this.pendingCalls.set(seq, [resolve, reject]);
+        });
+    }
+    /** 调用远程方法，放弃返回值 */
+    notify(method, ...args) {
+        const seq = this.nextSeq();
+        void this.sendPayload('notify', { seq, method, args });
+    }
+    /** 调用远程订阅 */
+    subscribe(method, ...args) {
+        return new Observable((subscriber) => {
+            const seq = this.nextSeq();
+            void this.sendPayload('subscribe', { seq, method, args });
+            this.pendingSubscriptions.set(seq, subscriber);
+            return () => {
+                this.pendingSubscriptions.delete(seq);
+                void this.sendPayload('unsubscribe', { seq });
+            };
+        });
+    }
+    /** 结束 */
+    destroy() {
+        if (this.__socket &&
+            (this.__socket.readyState === this.__socket.CONNECTING || this.__socket.readyState === this.__socket.OPEN)) {
+            this.__socket.close(1000);
+            this.__socket = undefined;
+        }
+        this.localSubscription.forEach((s) => s.unsubscribe());
+        this.localSubscription.clear();
+        this.pendingCalls.forEach(([, reject]) => reject(new Error(`RPC Socket closed.`)));
+        this.pendingCalls.clear();
+        this.pendingSubscriptions.forEach((s) => s.error(new Error(`RPC Socket closed.`)));
+        this.pendingSubscriptions.clear();
+    }
+}
+//# sourceMappingURL=socket.js.map