*Add dist
This commit is contained in:
parent
31eca18b6e
commit
9beb8f14b5
GPG Key ID:
C997C339BE476FF2
|
|
@ -0,0 +1,44 @@
|
||||||
|
/// <reference types="node" />
|
||||||
|
import EventEmitter from "events";
|
||||||
|
export default class ProtomuxRPC extends EventEmitter {
|
||||||
|
private _id;
|
||||||
|
private _ending;
|
||||||
|
private _error?;
|
||||||
|
private _responding;
|
||||||
|
private _requests;
|
||||||
|
private _defaultValueEncoding;
|
||||||
|
private _responders;
|
||||||
|
private _channel;
|
||||||
|
private _request;
|
||||||
|
private _response;
|
||||||
|
constructor(stream: any, options?: any);
|
||||||
|
private _mux;
|
||||||
|
get mux(): any;
|
||||||
|
private _ready;
|
||||||
|
get ready(): Promise<void>;
|
||||||
|
get closed(): any;
|
||||||
|
get stream(): any;
|
||||||
|
private _init;
|
||||||
|
_onopen(handshake: any): void;
|
||||||
|
_onclose(): void;
|
||||||
|
_ondestroy(): void;
|
||||||
|
_onrequest({ id, method, value, }: {
|
||||||
|
id: number;
|
||||||
|
method: string;
|
||||||
|
value: any;
|
||||||
|
}): Promise<void>;
|
||||||
|
_onresponse({ id, error, value }: {
|
||||||
|
id: number;
|
||||||
|
error: string;
|
||||||
|
value: any;
|
||||||
|
}): void;
|
||||||
|
respond(method: string, options: Function | any, handler: Function): this;
|
||||||
|
unrespond(method: string): this;
|
||||||
|
request(method: string, value: any, options?: any): Promise<unknown>;
|
||||||
|
event(method: string, value: any, options?: any): void;
|
||||||
|
cork(): void;
|
||||||
|
uncork(): void;
|
||||||
|
end(): Promise<void>;
|
||||||
|
_endMaybe(): void;
|
||||||
|
destroy(err: Error): void;
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,225 @@
|
||||||
|
import EventEmitter from "events";
|
||||||
|
// @ts-ignore
|
||||||
|
import Protomux from "protomux";
|
||||||
|
// @ts-ignore
|
||||||
|
import c from "compact-encoding";
|
||||||
|
// @ts-ignore
|
||||||
|
import bitfield from "compact-encoding-bitfield";
|
||||||
|
// @ts-ignore
|
||||||
|
import bits from "bits-to-bytes";
|
||||||
|
export default class ProtomuxRPC extends EventEmitter {
|
||||||
|
_id;
|
||||||
|
_ending;
|
||||||
|
_error;
|
||||||
|
_responding;
|
||||||
|
_requests;
|
||||||
|
_defaultValueEncoding;
|
||||||
|
_responders;
|
||||||
|
_channel;
|
||||||
|
_request;
|
||||||
|
_response;
|
||||||
|
constructor(stream, options = {}) {
|
||||||
|
super();
|
||||||
|
this._mux = Protomux.from(stream);
|
||||||
|
this._defaultValueEncoding = options?.valueEncoding;
|
||||||
|
this._id = 1;
|
||||||
|
this._ending = false;
|
||||||
|
this._responding = 0;
|
||||||
|
this._requests = new Map();
|
||||||
|
this._responders = new Map();
|
||||||
|
this._ready = this._init(options);
|
||||||
|
}
|
||||||
|
_mux;
|
||||||
|
get mux() {
|
||||||
|
return this._mux;
|
||||||
|
}
|
||||||
|
_ready;
|
||||||
|
get ready() {
|
||||||
|
return this._ready;
|
||||||
|
}
|
||||||
|
get closed() {
|
||||||
|
return this._channel.closed;
|
||||||
|
}
|
||||||
|
get stream() {
|
||||||
|
return this._mux.stream;
|
||||||
|
}
|
||||||
|
async _init(options) {
|
||||||
|
this._channel = await this._mux.createChannel({
|
||||||
|
protocol: "protomux-rpc",
|
||||||
|
id: options?.id,
|
||||||
|
handshake: options?.handshake
|
||||||
|
? options?.handshakeEncoding || c.raw
|
||||||
|
: null,
|
||||||
|
onopen: this._onopen.bind(this),
|
||||||
|
onclose: this._onclose.bind(this),
|
||||||
|
ondestroy: this._ondestroy.bind(this),
|
||||||
|
});
|
||||||
|
if (this._channel === null)
|
||||||
|
throw new Error("duplicate channel");
|
||||||
|
this._request = await this._channel.addMessage({
|
||||||
|
encoding: request,
|
||||||
|
onmessage: this._onrequest.bind(this),
|
||||||
|
});
|
||||||
|
this._response = await this._channel.addMessage({
|
||||||
|
encoding: response,
|
||||||
|
onmessage: this._onresponse.bind(this),
|
||||||
|
});
|
||||||
|
this._channel.open(options?.handshake);
|
||||||
|
await this._channel.ready;
|
||||||
|
}
|
||||||
|
_onopen(handshake) {
|
||||||
|
this.emit("open", handshake);
|
||||||
|
}
|
||||||
|
_onclose() {
|
||||||
|
const err = this._error || new Error("channel closed");
|
||||||
|
for (const request of this._requests.values()) {
|
||||||
|
request.reject(err);
|
||||||
|
}
|
||||||
|
this._requests.clear();
|
||||||
|
this._responders.clear();
|
||||||
|
this.emit("close");
|
||||||
|
}
|
||||||
|
_ondestroy() {
|
||||||
|
this.emit("destroy");
|
||||||
|
}
|
||||||
|
async _onrequest({ id, method, value, }) {
|
||||||
|
let error = null;
|
||||||
|
const responder = this._responders.get(method);
|
||||||
|
if (responder === undefined)
|
||||||
|
error = `unknown method '${method}'`;
|
||||||
|
else {
|
||||||
|
const { valueEncoding = this._defaultValueEncoding, requestEncoding = valueEncoding, responseEncoding = valueEncoding, } = responder.options;
|
||||||
|
if (requestEncoding)
|
||||||
|
value = c.decode(requestEncoding, value);
|
||||||
|
this._responding++;
|
||||||
|
try {
|
||||||
|
value = await responder.handler(value);
|
||||||
|
}
|
||||||
|
catch (err) {
|
||||||
|
error = err.message;
|
||||||
|
}
|
||||||
|
this._responding--;
|
||||||
|
if (!error && responseEncoding && id) {
|
||||||
|
value = c.encode(responseEncoding, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (id)
|
||||||
|
this._response.send({ id, error, value });
|
||||||
|
this._endMaybe();
|
||||||
|
}
|
||||||
|
_onresponse({ id, error, value }) {
|
||||||
|
if (id === 0)
|
||||||
|
return;
|
||||||
|
const request = this._requests.get(id);
|
||||||
|
if (request === undefined)
|
||||||
|
return;
|
||||||
|
this._requests.delete(id);
|
||||||
|
if (error)
|
||||||
|
request.reject(new Error(error));
|
||||||
|
else {
|
||||||
|
const { valueEncoding = this._defaultValueEncoding, responseEncoding = valueEncoding, } = request.options;
|
||||||
|
if (responseEncoding)
|
||||||
|
value = c.decode(responseEncoding, value);
|
||||||
|
request.resolve(value);
|
||||||
|
}
|
||||||
|
this._endMaybe();
|
||||||
|
}
|
||||||
|
respond(method, options, handler) {
|
||||||
|
if (typeof options === "function") {
|
||||||
|
handler = options;
|
||||||
|
options = {};
|
||||||
|
}
|
||||||
|
this._responders.set(method, { options, handler });
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
unrespond(method) {
|
||||||
|
this._responders.delete(method);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
async request(method, value, options = {}) {
|
||||||
|
if (this.closed)
|
||||||
|
throw new Error("channel closed");
|
||||||
|
const { valueEncoding = this._defaultValueEncoding, requestEncoding = valueEncoding, } = options;
|
||||||
|
if (requestEncoding)
|
||||||
|
value = c.encode(requestEncoding, value);
|
||||||
|
const id = this._id++;
|
||||||
|
this._request.send({ id, method, value });
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
this._requests.set(id, { options, resolve, reject });
|
||||||
|
});
|
||||||
|
}
|
||||||
|
event(method, value, options = {}) {
|
||||||
|
if (this.closed)
|
||||||
|
throw new Error("channel closed");
|
||||||
|
const { valueEncoding = this._defaultValueEncoding, requestEncoding = valueEncoding, } = options;
|
||||||
|
if (requestEncoding)
|
||||||
|
value = c.encode(requestEncoding, value);
|
||||||
|
this._request.send({ id: 0, method, value });
|
||||||
|
}
|
||||||
|
cork() {
|
||||||
|
this._channel.cork();
|
||||||
|
}
|
||||||
|
uncork() {
|
||||||
|
this._channel.uncork();
|
||||||
|
}
|
||||||
|
async end() {
|
||||||
|
this._ending = true;
|
||||||
|
this._endMaybe();
|
||||||
|
await EventEmitter.once(this, "close");
|
||||||
|
}
|
||||||
|
_endMaybe() {
|
||||||
|
if (this._ending && this._responding === 0 && this._requests.size === 0) {
|
||||||
|
this._channel.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
destroy(err) {
|
||||||
|
this._error = err || new Error("channel destroyed");
|
||||||
|
this._channel.close();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
const request = {
|
||||||
|
preencode(state, m) {
|
||||||
|
c.uint.preencode(state, m.id);
|
||||||
|
c.string.preencode(state, m.method);
|
||||||
|
c.raw.preencode(state, m.value);
|
||||||
|
},
|
||||||
|
encode(state, m) {
|
||||||
|
c.uint.encode(state, m.id);
|
||||||
|
c.string.encode(state, m.method);
|
||||||
|
c.raw.encode(state, m.value);
|
||||||
|
},
|
||||||
|
decode(state) {
|
||||||
|
return {
|
||||||
|
id: c.uint.decode(state),
|
||||||
|
method: c.string.decode(state),
|
||||||
|
value: c.raw.decode(state),
|
||||||
|
};
|
||||||
|
},
|
||||||
|
};
|
||||||
|
const flags = bitfield(1);
|
||||||
|
const response = {
|
||||||
|
preencode(state, m) {
|
||||||
|
flags.preencode(state);
|
||||||
|
c.uint.preencode(state, m.id);
|
||||||
|
if (m.error)
|
||||||
|
c.string.preencode(state, m.error);
|
||||||
|
else
|
||||||
|
c.raw.preencode(state, m.value);
|
||||||
|
},
|
||||||
|
encode(state, m) {
|
||||||
|
flags.encode(state, bits.of(m.error));
|
||||||
|
c.uint.encode(state, m.id);
|
||||||
|
if (m.error)
|
||||||
|
c.string.encode(state, m.error);
|
||||||
|
else
|
||||||
|
c.raw.encode(state, m.value);
|
||||||
|
},
|
||||||
|
decode(state) {
|
||||||
|
const [error] = bits.iterator(flags.decode(state));
|
||||||
|
return {
|
||||||
|
id: c.uint.decode(state),
|
||||||
|
error: error ? c.string.decode(state) : null,
|
||||||
|
value: !error ? c.raw.decode(state) : null,
|
||||||
|
};
|
||||||
|
},
|
||||||
|
};
|
||||||
Loading…
Reference in New Issue