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 (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;
    }
}