123 lines
4.0 KiB
JavaScript
123 lines
4.0 KiB
JavaScript
import { clearTimeout } from "timers";
|
|
import b4a from "b4a";
|
|
import { flatten, validateResponse, validateTimestampedResponse, } from "../util.js";
|
|
import { blake2b } from "libskynet";
|
|
import { ERR_INVALID_SIGNATURE, ERR_NO_RELAYS } from "../error.js";
|
|
import RpcQueryBase from "./base.js";
|
|
import { getActiveRelay, setupRelay } from "../sharedRelay.js";
|
|
function flatHash(data) {
|
|
const flattenedData = flatten(data).sort();
|
|
return Buffer.from(blake2b(Buffer.from(JSON.stringify(flattenedData)))).toString("hex");
|
|
}
|
|
export default class WisdomRpcQuery extends RpcQueryBase {
|
|
get result() {
|
|
return this._promise;
|
|
}
|
|
async _run() {
|
|
await setupRelay(this._network);
|
|
await this.queryRelay();
|
|
await this.checkResponse();
|
|
}
|
|
resolve(data, timeout = false) {
|
|
clearTimeout(this._timeoutTimer);
|
|
this._timeout = timeout;
|
|
this._completed = true;
|
|
if (timeout) {
|
|
data = {
|
|
error: "timeout",
|
|
};
|
|
}
|
|
this._promiseResolve?.(data);
|
|
}
|
|
async queryRelay() {
|
|
let activeRelay = getActiveRelay();
|
|
let relays = this.getRelays();
|
|
if (!relays.length) {
|
|
throw new Error(ERR_NO_RELAYS);
|
|
}
|
|
if (this._query.bypassCache) {
|
|
delete this._query.bypassCache;
|
|
const clearCacheQuery = this._network.factory.clearCache({
|
|
relays,
|
|
query: this._query,
|
|
});
|
|
await clearCacheQuery.result;
|
|
}
|
|
if ("bypassCache" in this._query) {
|
|
delete this._query.bypassCache;
|
|
}
|
|
return this.queryRpc(activeRelay, {
|
|
module: "rpc",
|
|
method: "broadcast_request",
|
|
data: {
|
|
request: this._query,
|
|
relays,
|
|
},
|
|
});
|
|
}
|
|
async checkResponse() {
|
|
if (this._error) {
|
|
this.resolve({ error: this._error });
|
|
return;
|
|
}
|
|
if (!validateResponse(
|
|
// @ts-ignore
|
|
getActiveRelay().stream.remotePublicKey, this._response)) {
|
|
this.resolve({ error: ERR_INVALID_SIGNATURE });
|
|
return;
|
|
}
|
|
let relays = [];
|
|
for (const relay in this._response?.relays) {
|
|
const resp = this._response?.relays[relay];
|
|
if (resp?.error) {
|
|
continue;
|
|
}
|
|
if (validateTimestampedResponse(b4a.from(relay, "hex"), resp)) {
|
|
relays.push(resp);
|
|
}
|
|
}
|
|
if (!relays.length) {
|
|
this.resolve({ error: ERR_NO_RELAYS });
|
|
return;
|
|
}
|
|
const responseObjects = relays.reduce((output, item) => {
|
|
const field = item.signedField || "data";
|
|
// @ts-ignore
|
|
const hash = flatHash(item[field]);
|
|
output[hash] = item?.data;
|
|
return output;
|
|
}, {});
|
|
const responses = relays.reduce((output, item) => {
|
|
const field = item.signedField || "data";
|
|
// @ts-ignore
|
|
const hash = flatHash(item[field]);
|
|
output[hash] = output[hash] ?? 0;
|
|
output[hash]++;
|
|
return output;
|
|
}, {});
|
|
for (const responseHash in responses) {
|
|
if (responses[responseHash] / relays.length >=
|
|
this._network.majorityThreshold) {
|
|
let response = responseObjects[responseHash];
|
|
response = { data: response };
|
|
this.resolve(response);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
getRelays() {
|
|
if (this._network.maxRelays === 0 ||
|
|
this._network.relays.length <= this._network.maxRelays) {
|
|
return this._network.relays;
|
|
}
|
|
const list = [];
|
|
let available = this._network.relays;
|
|
while (list.length <= this._network.maxRelays) {
|
|
const item = Math.floor(Math.random() * available.length);
|
|
list.push(available[item]);
|
|
available.splice(item, 1);
|
|
}
|
|
return list;
|
|
}
|
|
}
|