@neo-one/node-consensus/Consensus.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const client_common_1 = require("@neo-one/client-common");
const logger_1 = require("@neo-one/logger");
const utils_1 = require("@neo-one/utils");
const asynciterablex_1 = require("@reactivex/ix-es2015-cjs/asynciterable/asynciterablex");
const scan_1 = require("@reactivex/ix-es2015-cjs/asynciterable/pipe/scan");
const common_1 = require("./common");
const ConsensusContext_1 = require("./ConsensusContext");
const ConsensusQueue_1 = require("./ConsensusQueue");
const handleConsensusPayload_1 = require("./handleConsensusPayload");
const handlePersistBlock_1 = require("./handlePersistBlock");
const handleTransactionReceived_1 = require("./handleTransactionReceived");
const runConsensus_1 = require("./runConsensus");
const logger = logger_1.createChild(logger_1.nodeLogger, { component: 'consensus' });
const MS_IN_SECOND = 1000;
class Consensus {
    constructor({ options, node }) {
        this.mutableQueue = new ConsensusQueue_1.ConsensusQueue();
        const privateKey = client_common_1.common.stringToPrivateKey(options.privateKey);
        const publicKey = client_common_1.crypto.privateKeyToPublicKey(privateKey);
        const feeAddress = client_common_1.crypto.publicKeyToScriptHash(publicKey);
        this.options = {
            privateKey,
            publicKey,
            feeAddress,
            privateNet: options.privateNet,
        };
        this.node = node;
        this.mutableConsensusContext = new ConsensusContext_1.ConsensusContext();
    }
    async start() {
        let disposable = utils_1.noopDisposable;
        try {
            await this.pause();
            this.doStart(this.options);
            disposable = utils_1.composeDisposables(disposable, async () => {
                await this.pause();
            });
            return disposable;
        }
        catch (err) {
            await disposable();
            throw err;
        }
    }
    onPersistBlock() {
        this.mutableQueue.write({ type: 'handlePersistBlock' });
    }
    onConsensusPayloadReceived(payload) {
        this.mutableQueue.write({
            type: 'handleConsensusPayload',
            payload,
        });
    }
    onTransactionReceived(transaction) {
        this.mutableQueue.write({
            type: 'handleTransactionReceived',
            transaction,
        });
    }
    async runConsensusNow() {
        if (this.options.privateNet) {
            await new Promise((resolve, reject) => {
                this.mutableQueue.write({ type: 'timer', promise: { resolve, reject } });
            });
        }
        else {
            throw new Error('Can only force consensus on a private network.');
        }
    }
    nowSeconds() {
        return this.mutableConsensusContext.nowSeconds();
    }
    async fastForwardOffset(seconds) {
        if (this.options.privateNet) {
            this.mutableConsensusContext.fastForwardOffset(seconds);
        }
        else {
            throw new Error('Can only fast forward on a private network.');
        }
    }
    async fastForwardToTime(seconds) {
        if (this.options.privateNet) {
            this.mutableConsensusContext.fastForwardToTime(seconds);
        }
        else {
            throw new Error('Can only fast forward on a private network.');
        }
    }
    async pause() {
        this.clearTimer();
        this.mutableQueue.done();
        this.mutableQueue = new ConsensusQueue_1.ConsensusQueue();
        if (this.mutableStartPromise !== undefined) {
            await this.mutableStartPromise;
        }
    }
    async reset() {
        this.mutableConsensusContext = new ConsensusContext_1.ConsensusContext();
    }
    async resume() {
        this.doStart(this.options);
    }
    doStart(options) {
        let completed = false;
        const mutableStartPromise = this.startInternal(options).then(() => {
            completed = true;
            this.mutableStartPromise = undefined;
        });
        if (!completed) {
            this.mutableStartPromise = mutableStartPromise;
        }
    }
    async startInternal(options) {
        logger.info({ name: 'neo_consensus_start' }, 'Consensus started.');
        const initialResult = await common_1.initializeNewConsensus({
            blockchain: this.node.blockchain,
            publicKey: options.publicKey,
            consensusContext: this.mutableConsensusContext,
        });
        await asynciterablex_1.AsyncIterableX.from(this.mutableQueue)
            .pipe(scan_1.scan(async (context, event) => {
            let result;
            switch (event.type) {
                case 'handlePersistBlock':
                    result = await handlePersistBlock_1.handlePersistBlock({
                        blockchain: this.node.blockchain,
                        publicKey: options.publicKey,
                        consensusContext: this.mutableConsensusContext,
                    });
                    break;
                case 'handleConsensusPayload':
                    result = await handleConsensusPayload_1.handleConsensusPayload({
                        context,
                        node: this.node,
                        privateKey: options.privateKey,
                        payload: event.payload,
                        consensusContext: this.mutableConsensusContext,
                    });
                    break;
                case 'handleTransactionReceived':
                    result = await handleTransactionReceived_1.handleTransactionReceived({
                        context,
                        node: this.node,
                        privateKey: options.privateKey,
                        transaction: event.transaction,
                        consensusContext: this.mutableConsensusContext,
                    });
                    break;
                case 'timer':
                    result = await runConsensus_1.runConsensus({
                        context,
                        node: this.node,
                        options,
                        consensusContext: this.mutableConsensusContext,
                    }).catch((err) => {
                        if (event.promise !== undefined) {
                            event.promise.reject(err);
                        }
                        throw err;
                    });
                    if (event.promise !== undefined) {
                        event.promise.resolve();
                    }
                    break;
                default:
                    utils_1.utils.assertNever(event);
                    throw new Error('For TS');
            }
            return this.handleResult(result);
        }, this.handleResult(initialResult)))
            .forEach(() => {
        });
        logger.info({ name: 'neo_consensus_stop' }, 'Consensus stopped.');
    }
    handleResult(result) {
        if (result.timerSeconds !== undefined) {
            this.handleTimer(result.timerSeconds);
        }
        return result.context;
    }
    handleTimer(mutableTimerSeconds) {
        this.clearTimer();
        this.mutableTimer = setTimeout(() => this.mutableQueue.write({ type: 'timer' }), mutableTimerSeconds * MS_IN_SECOND);
    }
    clearTimer() {
        if (this.mutableTimer !== undefined) {
            clearTimeout(this.mutableTimer);
            this.mutableTimer = undefined;
        }
    }
}
exports.Consensus = Consensus;

//# sourceMappingURL=data:application/json;charset=utf8;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIkNvbnNlbnN1cy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOztBQUNBLDBEQUFzRjtBQUN0Riw0Q0FBMEQ7QUFFMUQsMENBQXNHO0FBQ3RHLDBGQUF1RjtBQUN2RiwyRUFBd0U7QUFDeEUscUNBQWtEO0FBQ2xELHlEQUFzRDtBQUN0RCxxREFBa0Q7QUFFbEQscUVBQWtFO0FBQ2xFLDZEQUEwRDtBQUMxRCwyRUFBd0U7QUFDeEUsaURBQThDO0FBRzlDLE1BQU0sTUFBTSxHQUFHLG9CQUFXLENBQUMsbUJBQVUsRUFBRSxFQUFFLFNBQVMsRUFBRSxXQUFXLEVBQUUsQ0FBQyxDQUFDO0FBYW5FLE1BQU0sWUFBWSxHQUFHLElBQUksQ0FBQztBQUUxQixNQUFhLFNBQVM7SUFRcEIsWUFBbUIsRUFBRSxPQUFPLEVBQUUsSUFBSSxFQUFzRDtRQUN0RixJQUFJLENBQUMsWUFBWSxHQUFHLElBQUksK0JBQWMsRUFBRSxDQUFDO1FBRXpDLE1BQU0sVUFBVSxHQUFHLHNCQUFNLENBQUMsa0JBQWtCLENBQUMsT0FBTyxDQUFDLFVBQVUsQ0FBQyxDQUFDO1FBQ2pFLE1BQU0sU0FBUyxHQUFHLHNCQUFNLENBQUMscUJBQXFCLENBQUMsVUFBVSxDQUFDLENBQUM7UUFDM0QsTUFBTSxVQUFVLEdBQUcsc0JBQU0sQ0FBQyxxQkFBcUIsQ0FBQyxTQUFTLENBQUMsQ0FBQztRQUMzRCxJQUFJLENBQUMsT0FBTyxHQUFHO1lBQ2IsVUFBVTtZQUNWLFNBQVM7WUFDVCxVQUFVO1lBQ1YsVUFBVSxFQUFFLE9BQU8sQ0FBQyxVQUFVO1NBQy9CLENBQUM7UUFFRixJQUFJLENBQUMsSUFBSSxHQUFHLElBQUksQ0FBQztRQUNqQixJQUFJLENBQUMsdUJBQXVCLEdBQUcsSUFBSSxtQ0FBZ0IsRUFBRSxDQUFDO0lBQ3hELENBQUM7SUFFTSxLQUFLLENBQUMsS0FBSztRQUNoQixJQUFJLFVBQVUsR0FBRyxzQkFBYyxDQUFDO1FBQ2hDLElBQUk7WUFDRixNQUFNLElBQUksQ0FBQyxLQUFLLEVBQUUsQ0FBQztZQUNuQixJQUFJLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQztZQUUzQixVQUFVLEdBQUcsMEJBQWtCLENBQUMsVUFBVSxFQUFFLEtBQUssSUFBSSxFQUFFO2dCQUNyRCxNQUFNLElBQUksQ0FBQyxLQUFLLEVBQUUsQ0FBQztZQUNyQixDQUFDLENBQUMsQ0FBQztZQUVILE9BQU8sVUFBVSxDQUFDO1NBQ25CO1FBQUMsT0FBTyxHQUFHLEVBQUU7WUFDWixNQUFNLFVBQVUsRUFBRSxDQUFDO1lBQ25CLE1BQU0sR0FBRyxDQUFDO1NBQ1g7SUFDSCxDQUFDO0lBRU0sY0FBYztRQUNuQixJQUFJLENBQUMsWUFBWSxDQUFDLEtBQUssQ0FBQyxFQUFFLElBQUksRUFBRSxvQkFBb0IsRUFBRSxDQUFDLENBQUM7SUFDMUQsQ0FBQztJQUVNLDBCQUEwQixDQUFDLE9BQXlCO1FBQ3pELElBQUksQ0FBQyxZQUFZLENBQUMsS0FBSyxDQUFDO1lBQ3RCLElBQUksRUFBRSx3QkFBd0I7WUFDOUIsT0FBTztTQUNSLENBQUMsQ0FBQztJQUNMLENBQUM7SUFFTSxxQkFBcUIsQ0FBQyxXQUF3QjtRQUNuRCxJQUFJLENBQUMsWUFBWSxDQUFDLEtBQUssQ0FBQztZQUN0QixJQUFJLEVBQUUsMkJBQTJCO1lBQ2pDLFdBQVc7U0FDWixDQUFDLENBQUM7SUFDTCxDQUFDO0lBRU0sS0FBSyxDQUFDLGVBQWU7UUFDMUIsSUFBSSxJQUFJLENBQUMsT0FBTyxDQUFDLFVBQVUsRUFBRTtZQUUzQixNQUFNLElBQUksT0FBTyxDQUFDLENBQUMsT0FBTyxFQUFFLE1BQU0sRUFBRSxFQUFFO2dCQUNwQyxJQUFJLENBQUMsWUFBWSxDQUFDLEtBQUssQ0FBQyxFQUFFLElBQUksRUFBRSxPQUFPLEVBQUUsT0FBTyxFQUFFLEVBQUUsT0FBTyxFQUFFLE1BQU0sRUFBRSxFQUFFLENBQUMsQ0FBQztZQUMzRSxDQUFDLENBQUMsQ0FBQztTQUNKO2FBQU07WUFDTCxNQUFNLElBQUksS0FBSyxDQUFDLGdEQUFnRCxDQUFDLENBQUM7U0FDbkU7SUFDSCxDQUFDO0lBRU0sVUFBVTtRQUNmLE9BQU8sSUFBSSxDQUFDLHVCQUF1QixDQUFDLFVBQVUsRUFBRSxDQUFDO0lBQ25ELENBQUM7SUFFTSxLQUFLLENBQUMsaUJBQWlCLENBQUMsT0FBZTtRQUM1QyxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUMsVUFBVSxFQUFFO1lBQzNCLElBQUksQ0FBQyx1QkFBdUIsQ0FBQyxpQkFBaUIsQ0FBQyxPQUFPLENBQUMsQ0FBQztTQUN6RDthQUFNO1lBQ0wsTUFBTSxJQUFJLEtBQUssQ0FBQyw2Q0FBNkMsQ0FBQyxDQUFDO1NBQ2hFO0lBQ0gsQ0FBQztJQUVNLEtBQUssQ0FBQyxpQkFBaUIsQ0FBQyxPQUFlO1FBQzVDLElBQUksSUFBSSxDQUFDLE9BQU8sQ0FBQyxVQUFVLEVBQUU7WUFDM0IsSUFBSSxDQUFDLHVCQUF1QixDQUFDLGlCQUFpQixDQUFDLE9BQU8sQ0FBQyxDQUFDO1NBQ3pEO2FBQU07WUFDTCxNQUFNLElBQUksS0FBSyxDQUFDLDZDQUE2QyxDQUFDLENBQUM7U0FDaEU7SUFDSCxDQUFDO0lBRU0sS0FBSyxDQUFDLEtBQUs7UUFDaEIsSUFBSSxDQUFDLFVBQVUsRUFBRSxDQUFDO1FBQ2xCLElBQUksQ0FBQyxZQUFZLENBQUMsSUFBSSxFQUFFLENBQUM7UUFDekIsSUFBSSxDQUFDLFlBQVksR0FBRyxJQUFJLCtCQUFjLEVBQUUsQ0FBQztRQUN6QyxJQUFJLElBQUksQ0FBQyxtQkFBbUIsS0FBSyxTQUFTLEVBQUU7WUFDMUMsTUFBTSxJQUFJLENBQUMsbUJBQW1CLENBQUM7U0FDaEM7SUFDSCxDQUFDO0lBRU0sS0FBSyxDQUFDLEtBQUs7UUFDaEIsSUFBSSxDQUFDLHVCQUF1QixHQUFHLElBQUksbUNBQWdCLEVBQUUsQ0FBQztJQUN4RCxDQUFDO0lBRU0sS0FBSyxDQUFDLE1BQU07UUFFakIsSUFBSSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLENBQUM7SUFDN0IsQ0FBQztJQUVPLE9BQU8sQ0FBQyxPQUF3QjtRQUN0QyxJQUFJLFNBQVMsR0FBRyxLQUFLLENBQUM7UUFDdEIsTUFBTSxtQkFBbUIsR0FBRyxJQUFJLENBQUMsYUFBYSxDQUFDLE9BQU8sQ0FBQyxDQUFDLElBQUksQ0FBQyxHQUFHLEVBQUU7WUFDaEUsU0FBUyxHQUFHLElBQUksQ0FBQztZQUNqQixJQUFJLENBQUMsbUJBQW1CLEdBQUcsU0FBUyxDQUFDO1FBQ3ZDLENBQUMsQ0FBQyxDQUFDO1FBQ0gsSUFBSSxDQUFDLFNBQVMsRUFBRTtZQUNkLElBQUksQ0FBQyxtQkFBbUIsR0FBRyxtQkFBbUIsQ0FBQztTQUNoRDtJQUNILENBQUM7SUFFTyxLQUFLLENBQUMsYUFBYSxDQUFDLE9BQXdCO1FBQ2xELE1BQU0sQ0FBQyxJQUFJLENBQUMsRUFBRSxJQUFJLEVBQUUscUJBQXFCLEVBQUUsRUFBRSxvQkFBb0IsQ0FBQyxDQUFDO1FBRW5FLE1BQU0sYUFBYSxHQUFHLE1BQU0sK0JBQXNCLENBQUM7WUFDakQsVUFBVSxFQUFFLElBQUksQ0FBQyxJQUFJLENBQUMsVUFBVTtZQUNoQyxTQUFTLEVBQUUsT0FBTyxDQUFDLFNBQVM7WUFDNUIsZ0JBQWdCLEVBQUUsSUFBSSxDQUFDLHVCQUF1QjtTQUMvQyxDQUFDLENBQUM7UUFFSCxNQUFNLCtCQUFjLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxZQUFZLENBQUM7YUFDekMsSUFBSSxDQUNILFdBQUksQ0FBQyxLQUFLLEVBQUUsT0FBZ0IsRUFBRSxLQUFZLEVBQUUsRUFBRTtZQUM1QyxJQUFJLE1BQU0sQ0FBQztZQUNYLFFBQVEsS0FBSyxDQUFDLElBQUksRUFBRTtnQkFDbEIsS0FBSyxvQkFBb0I7b0JBQ3ZCLE1BQU0sR0FBRyxNQUFNLHVDQUFrQixDQUFDO3dCQUNoQyxVQUFVLEVBQUUsSUFBSSxDQUFDLElBQUksQ0FBQyxVQUFVO3dCQUNoQyxTQUFTLEVBQUUsT0FBTyxDQUFDLFNBQVM7d0JBQzVCLGdCQUFnQixFQUFFLElBQUksQ0FBQyx1QkFBdUI7cUJBQy9DLENBQUMsQ0FBQztvQkFFSCxNQUFNO2dCQUNSLEtBQUssd0JBQXdCO29CQUMzQixNQUFNLEdBQUcsTUFBTSwrQ0FBc0IsQ0FBQzt3QkFDcEMsT0FBTzt3QkFDUCxJQUFJLEVBQUUsSUFBSSxDQUFDLElBQUk7d0JBQ2YsVUFBVSxFQUFFLE9BQU8sQ0FBQyxVQUFVO3dCQUM5QixPQUFPLEVBQUUsS0FBSyxDQUFDLE9BQU87d0JBQ3RCLGdCQUFnQixFQUFFLElBQUksQ0FBQyx1QkFBdUI7cUJBQy9DLENBQUMsQ0FBQztvQkFFSCxNQUFNO2dCQUNSLEtBQUssMkJBQTJCO29CQUM5QixNQUFNLEdBQUcsTUFBTSxxREFBeUIsQ0FBQzt3QkFDdkMsT0FBTzt3QkFDUCxJQUFJLEVBQUUsSUFBSSxDQUFDLElBQUk7d0JBQ2YsVUFBVSxFQUFFLE9BQU8sQ0FBQyxVQUFVO3dCQUM5QixXQUFXLEVBQUUsS0FBSyxDQUFDLFdBQVc7d0JBQzlCLGdCQUFnQixFQUFFLElBQUksQ0FBQyx1QkFBdUI7cUJBQy9DLENBQUMsQ0FBQztvQkFFSCxNQUFNO2dCQUNSLEtBQUssT0FBTztvQkFDVixNQUFNLEdBQUcsTUFBTSwyQkFBWSxDQUFDO3dCQUMxQixPQUFPO3dCQUNQLElBQUksRUFBRSxJQUFJLENBQUMsSUFBSTt3QkFDZixPQUFPO3dCQUNQLGdCQUFnQixFQUFFLElBQUksQ0FBQyx1QkFBdUI7cUJBQy9DLENBQUMsQ0FBQyxLQUFLLENBQUMsQ0FBQyxHQUFHLEVBQUUsRUFBRTt3QkFDZixJQUFJLEtBQUssQ0FBQyxPQUFPLEtBQUssU0FBUyxFQUFFOzRCQUMvQixLQUFLLENBQUMsT0FBTyxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUMsQ0FBQzt5QkFDM0I7d0JBQ0QsTUFBTSxHQUFHLENBQUM7b0JBQ1osQ0FBQyxDQUFDLENBQUM7b0JBQ0gsSUFBSSxLQUFLLENBQUMsT0FBTyxLQUFLLFNBQVMsRUFBRTt3QkFDL0IsS0FBSyxDQUFDLE9BQU8sQ0FBQyxPQUFPLEVBQUUsQ0FBQztxQkFDekI7b0JBQ0QsTUFBTTtnQkFDUjtvQkFDRSxhQUFXLENBQUMsV0FBVyxDQUFDLEtBQUssQ0FBQyxDQUFDO29CQUMvQixNQUFNLElBQUksS0FBSyxDQUFDLFFBQVEsQ0FBQyxDQUFDO2FBQzdCO1lBRUQsT0FBTyxJQUFJLENBQUMsWUFBWSxDQUFDLE1BQU0sQ0FBQyxDQUFDO1FBQ25DLENBQUMsRUFBRSxJQUFJLENBQUMsWUFBWSxDQUFDLGFBQWEsQ0FBQyxDQUFDLENBQ3JDO2FBQ0EsT0FBTyxDQUFDLEdBQUcsRUFBRTtRQUVkLENBQUMsQ0FBQyxDQUFDO1FBRUwsTUFBTSxDQUFDLElBQUksQ0FBQyxFQUFFLElBQUksRUFBRSxvQkFBb0IsRUFBRSxFQUFFLG9CQUFvQixDQUFDLENBQUM7SUFDcEUsQ0FBQztJQUVPLFlBQVksQ0FBQyxNQUF1QjtRQUMxQyxJQUFJLE1BQU0sQ0FBQyxZQUFZLEtBQUssU0FBUyxFQUFFO1lBQ3JDLElBQUksQ0FBQyxXQUFXLENBQUMsTUFBTSxDQUFDLFlBQVksQ0FBQyxDQUFDO1NBQ3ZDO1FBRUQsT0FBTyxNQUFNLENBQUMsT0FBTyxDQUFDO0lBQ3hCLENBQUM7SUFFTyxXQUFXLENBQUMsbUJBQTJCO1FBQzdDLElBQUksQ0FBQyxVQUFVLEVBQUUsQ0FBQztRQUNsQixJQUFJLENBQUMsWUFBWSxHQUFHLFVBQVUsQ0FDNUIsR0FBRyxFQUFFLENBQUMsSUFBSSxDQUFDLFlBQVksQ0FBQyxLQUFLLENBQUMsRUFBRSxJQUFJLEVBQUUsT0FBTyxFQUFFLENBQUMsRUFDaEQsbUJBQW1CLEdBQUcsWUFBWSxDQUU1QixDQUFDO0lBQ1gsQ0FBQztJQUVPLFVBQVU7UUFDaEIsSUFBSSxJQUFJLENBQUMsWUFBWSxLQUFLLFNBQVMsRUFBRTtZQUNuQyxZQUFZLENBQUMsSUFBSSxDQUFDLFlBQVksQ0FBQyxDQUFDO1lBQ2hDLElBQUksQ0FBQyxZQUFZLEdBQUcsU0FBUyxDQUFDO1NBQy9CO0lBQ0gsQ0FBQztDQUNGO0FBeE5ELDhCQXdOQyIsImZpbGUiOiJuZW8tb25lLW5vZGUtY29uc2Vuc3VzL3NyYy9Db25zZW5zdXMuanMiLCJzb3VyY2VzQ29udGVudCI6WyIvLy8gPHJlZmVyZW5jZSB0eXBlcz1cIkByZWFjdGl2ZXgvaXgtZXMyMDE1LWNqc1wiIC8+XG5pbXBvcnQgeyBjb21tb24sIGNyeXB0bywgRUNQb2ludCwgUHJpdmF0ZUtleSwgVUludDE2MCB9IGZyb20gJ0BuZW8tb25lL2NsaWVudC1jb21tb24nO1xuaW1wb3J0IHsgY3JlYXRlQ2hpbGQsIG5vZGVMb2dnZXIgfSBmcm9tICdAbmVvLW9uZS9sb2dnZXInO1xuaW1wb3J0IHsgQ29uc2Vuc3VzUGF5bG9hZCwgTm9kZSwgVHJhbnNhY3Rpb24gfSBmcm9tICdAbmVvLW9uZS9ub2RlLWNvcmUnO1xuaW1wb3J0IHsgY29tcG9zZURpc3Bvc2FibGVzLCBEaXNwb3NhYmxlLCBub29wRGlzcG9zYWJsZSwgdXRpbHMgYXMgY29tbW9uVXRpbHMgfSBmcm9tICdAbmVvLW9uZS91dGlscyc7XG5pbXBvcnQgeyBBc3luY0l0ZXJhYmxlWCB9IGZyb20gJ0ByZWFjdGl2ZXgvaXgtZXMyMDE1LWNqcy9hc3luY2l0ZXJhYmxlL2FzeW5jaXRlcmFibGV4JztcbmltcG9ydCB7IHNjYW4gfSBmcm9tICdAcmVhY3RpdmV4L2l4LWVzMjAxNS1janMvYXN5bmNpdGVyYWJsZS9waXBlL3NjYW4nO1xuaW1wb3J0IHsgaW5pdGlhbGl6ZU5ld0NvbnNlbnN1cyB9IGZyb20gJy4vY29tbW9uJztcbmltcG9ydCB7IENvbnNlbnN1c0NvbnRleHQgfSBmcm9tICcuL0NvbnNlbnN1c0NvbnRleHQnO1xuaW1wb3J0IHsgQ29uc2Vuc3VzUXVldWUgfSBmcm9tICcuL0NvbnNlbnN1c1F1ZXVlJztcbmltcG9ydCB7IENvbnRleHQgfSBmcm9tICcuL2NvbnRleHQnO1xuaW1wb3J0IHsgaGFuZGxlQ29uc2Vuc3VzUGF5bG9hZCB9IGZyb20gJy4vaGFuZGxlQ29uc2Vuc3VzUGF5bG9hZCc7XG5pbXBvcnQgeyBoYW5kbGVQZXJzaXN0QmxvY2sgfSBmcm9tICcuL2hhbmRsZVBlcnNpc3RCbG9jayc7XG5pbXBvcnQgeyBoYW5kbGVUcmFuc2FjdGlvblJlY2VpdmVkIH0gZnJvbSAnLi9oYW5kbGVUcmFuc2FjdGlvblJlY2VpdmVkJztcbmltcG9ydCB7IHJ1bkNvbnNlbnN1cyB9IGZyb20gJy4vcnVuQ29uc2Vuc3VzJztcbmltcG9ydCB7IEV2ZW50LCBSZXN1bHQgfSBmcm9tICcuL3R5cGVzJztcblxuY29uc3QgbG9nZ2VyID0gY3JlYXRlQ2hpbGQobm9kZUxvZ2dlciwgeyBjb21wb25lbnQ6ICdjb25zZW5zdXMnIH0pO1xuXG5leHBvcnQgaW50ZXJmYWNlIE9wdGlvbnMge1xuICByZWFkb25seSBwcml2YXRlS2V5OiBzdHJpbmc7XG4gIHJlYWRvbmx5IHByaXZhdGVOZXQ6IGJvb2xlYW47XG59XG5leHBvcnQgaW50ZXJmYWNlIEludGVybmFsT3B0aW9ucyB7XG4gIHJlYWRvbmx5IHByaXZhdGVLZXk6IFByaXZhdGVLZXk7XG4gIHJlYWRvbmx5IHB1YmxpY0tleTogRUNQb2ludDtcbiAgcmVhZG9ubHkgZmVlQWRkcmVzczogVUludDE2MDtcbiAgcmVhZG9ubHkgcHJpdmF0ZU5ldDogYm9vbGVhbjtcbn1cblxuY29uc3QgTVNfSU5fU0VDT05EID0gMTAwMDtcblxuZXhwb3J0IGNsYXNzIENvbnNlbnN1cyB7XG4gIHByaXZhdGUgbXV0YWJsZVF1ZXVlOiBDb25zZW5zdXNRdWV1ZTtcbiAgcHJpdmF0ZSBtdXRhYmxlVGltZXI6IG51bWJlciB8IHVuZGVmaW5lZDtcbiAgcHJpdmF0ZSByZWFkb25seSBvcHRpb25zOiBJbnRlcm5hbE9wdGlvbnM7XG4gIHByaXZhdGUgcmVhZG9ubHkgbm9kZTogTm9kZTtcbiAgcHJpdmF0ZSBtdXRhYmxlQ29uc2Vuc3VzQ29udGV4dDogQ29uc2Vuc3VzQ29udGV4dDtcbiAgcHJpdmF0ZSBtdXRhYmxlU3RhcnRQcm9taXNlOiBQcm9taXNlPHZvaWQ+IHwgdW5kZWZpbmVkO1xuXG4gIHB1YmxpYyBjb25zdHJ1Y3Rvcih7IG9wdGlvbnMsIG5vZGUgfTogeyByZWFkb25seSBvcHRpb25zOiBPcHRpb25zOyByZWFkb25seSBub2RlOiBOb2RlIH0pIHtcbiAgICB0aGlzLm11dGFibGVRdWV1ZSA9IG5ldyBDb25zZW5zdXNRdWV1ZSgpO1xuXG4gICAgY29uc3QgcHJpdmF0ZUtleSA9IGNvbW1vbi5zdHJpbmdUb1ByaXZhdGVLZXkob3B0aW9ucy5wcml2YXRlS2V5KTtcbiAgICBjb25zdCBwdWJsaWNLZXkgPSBjcnlwdG8ucHJpdmF0ZUtleVRvUHVibGljS2V5KHByaXZhdGVLZXkpO1xuICAgIGNvbnN0IGZlZUFkZHJlc3MgPSBjcnlwdG8ucHVibGljS2V5VG9TY3JpcHRIYXNoKHB1YmxpY0tleSk7XG4gICAgdGhpcy5vcHRpb25zID0ge1xuICAgICAgcHJpdmF0ZUtleSxcbiAgICAgIHB1YmxpY0tleSxcbiAgICAgIGZlZUFkZHJlc3MsXG4gICAgICBwcml2YXRlTmV0OiBvcHRpb25zLnByaXZhdGVOZXQsXG4gICAgfTtcblxuICAgIHRoaXMubm9kZSA9IG5vZGU7XG4gICAgdGhpcy5tdXRhYmxlQ29uc2Vuc3VzQ29udGV4dCA9IG5ldyBDb25zZW5zdXNDb250ZXh0KCk7XG4gIH1cblxuICBwdWJsaWMgYXN5bmMgc3RhcnQoKTogUHJvbWlzZTxEaXNwb3NhYmxlPiB7XG4gICAgbGV0IGRpc3Bvc2FibGUgPSBub29wRGlzcG9zYWJsZTtcbiAgICB0cnkge1xuICAgICAgYXdhaXQgdGhpcy5wYXVzZSgpO1xuICAgICAgdGhpcy5kb1N0YXJ0KHRoaXMub3B0aW9ucyk7XG5cbiAgICAgIGRpc3Bvc2FibGUgPSBjb21wb3NlRGlzcG9zYWJsZXMoZGlzcG9zYWJsZSwgYXN5bmMgKCkgPT4ge1xuICAgICAgICBhd2FpdCB0aGlzLnBhdXNlKCk7XG4gICAgICB9KTtcblxuICAgICAgcmV0dXJuIGRpc3Bvc2FibGU7XG4gICAgfSBjYXRjaCAoZXJyKSB7XG4gICAgICBhd2FpdCBkaXNwb3NhYmxlKCk7XG4gICAgICB0aHJvdyBlcnI7XG4gICAgfVxuICB9XG5cbiAgcHVibGljIG9uUGVyc2lzdEJsb2NrKCk6IHZvaWQge1xuICAgIHRoaXMubXV0YWJsZVF1ZXVlLndyaXRlKHsgdHlwZTogJ2hhbmRsZVBlcnNpc3RCbG9jaycgfSk7XG4gIH1cblxuICBwdWJsaWMgb25Db25zZW5zdXNQYXlsb2FkUmVjZWl2ZWQocGF5bG9hZDogQ29uc2Vuc3VzUGF5bG9hZCk6IHZvaWQge1xuICAgIHRoaXMubXV0YWJsZVF1ZXVlLndyaXRlKHtcbiAgICAgIHR5cGU6ICdoYW5kbGVDb25zZW5zdXNQYXlsb2FkJyxcbiAgICAgIHBheWxvYWQsXG4gICAgfSk7XG4gIH1cblxuICBwdWJsaWMgb25UcmFuc2FjdGlvblJlY2VpdmVkKHRyYW5zYWN0aW9uOiBUcmFuc2FjdGlvbik6IHZvaWQge1xuICAgIHRoaXMubXV0YWJsZVF1ZXVlLndyaXRlKHtcbiAgICAgIHR5cGU6ICdoYW5kbGVUcmFuc2FjdGlvblJlY2VpdmVkJyxcbiAgICAgIHRyYW5zYWN0aW9uLFxuICAgIH0pO1xuICB9XG5cbiAgcHVibGljIGFzeW5jIHJ1bkNvbnNlbnN1c05vdygpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICBpZiAodGhpcy5vcHRpb25zLnByaXZhdGVOZXQpIHtcbiAgICAgIC8vIHRzbGludDpkaXNhYmxlLW5leHQtbGluZSBwcm9taXNlLW11c3QtY29tcGxldGVcbiAgICAgIGF3YWl0IG5ldyBQcm9taXNlKChyZXNvbHZlLCByZWplY3QpID0+IHtcbiAgICAgICAgdGhpcy5tdXRhYmxlUXVldWUud3JpdGUoeyB0eXBlOiAndGltZXInLCBwcm9taXNlOiB7IHJlc29sdmUsIHJlamVjdCB9IH0pO1xuICAgICAgfSk7XG4gICAgfSBlbHNlIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcignQ2FuIG9ubHkgZm9yY2UgY29uc2Vuc3VzIG9uIGEgcHJpdmF0ZSBuZXR3b3JrLicpO1xuICAgIH1cbiAgfVxuXG4gIHB1YmxpYyBub3dTZWNvbmRzKCk6IG51bWJlciB7XG4gICAgcmV0dXJuIHRoaXMubXV0YWJsZUNvbnNlbnN1c0NvbnRleHQubm93U2Vjb25kcygpO1xuICB9XG5cbiAgcHVibGljIGFzeW5jIGZhc3RGb3J3YXJkT2Zmc2V0KHNlY29uZHM6IG51bWJlcik6IFByb21pc2U8dm9pZD4ge1xuICAgIGlmICh0aGlzLm9wdGlvbnMucHJpdmF0ZU5ldCkge1xuICAgICAgdGhpcy5tdXRhYmxlQ29uc2Vuc3VzQ29udGV4dC5mYXN0Rm9yd2FyZE9mZnNldChzZWNvbmRzKTtcbiAgICB9IGVsc2Uge1xuICAgICAgdGhyb3cgbmV3IEVycm9yKCdDYW4gb25seSBmYXN0IGZvcndhcmQgb24gYSBwcml2YXRlIG5ldHdvcmsuJyk7XG4gICAgfVxuICB9XG5cbiAgcHVibGljIGFzeW5jIGZhc3RGb3J3YXJkVG9UaW1lKHNlY29uZHM6IG51bWJlcik6IFByb21pc2U8dm9pZD4ge1xuICAgIGlmICh0aGlzLm9wdGlvbnMucHJpdmF0ZU5ldCkge1xuICAgICAgdGhpcy5tdXRhYmxlQ29uc2Vuc3VzQ29udGV4dC5mYXN0Rm9yd2FyZFRvVGltZShzZWNvbmRzKTtcbiAgICB9IGVsc2Uge1xuICAgICAgdGhyb3cgbmV3IEVycm9yKCdDYW4gb25seSBmYXN0IGZvcndhcmQgb24gYSBwcml2YXRlIG5ldHdvcmsuJyk7XG4gICAgfVxuICB9XG5cbiAgcHVibGljIGFzeW5jIHBhdXNlKCk6IFByb21pc2U8dm9pZD4ge1xuICAgIHRoaXMuY2xlYXJUaW1lcigpO1xuICAgIHRoaXMubXV0YWJsZVF1ZXVlLmRvbmUoKTtcbiAgICB0aGlzLm11dGFibGVRdWV1ZSA9IG5ldyBDb25zZW5zdXNRdWV1ZSgpO1xuICAgIGlmICh0aGlzLm11dGFibGVTdGFydFByb21pc2UgIT09IHVuZGVmaW5lZCkge1xuICAgICAgYXdhaXQgdGhpcy5tdXRhYmxlU3RhcnRQcm9taXNlO1xuICAgIH1cbiAgfVxuXG4gIHB1YmxpYyBhc3luYyByZXNldCgpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICB0aGlzLm11dGFibGVDb25zZW5zdXNDb250ZXh0ID0gbmV3IENvbnNlbnN1c0NvbnRleHQoKTtcbiAgfVxuXG4gIHB1YmxpYyBhc3luYyByZXN1bWUoKTogUHJvbWlzZTx2b2lkPiB7XG4gICAgLy8gdHNsaW50OmRpc2FibGUtbmV4dC1saW5lIG5vLWZsb2F0aW5nLXByb21pc2VzXG4gICAgdGhpcy5kb1N0YXJ0KHRoaXMub3B0aW9ucyk7XG4gIH1cblxuICBwcml2YXRlIGRvU3RhcnQob3B0aW9uczogSW50ZXJuYWxPcHRpb25zKTogdm9pZCB7XG4gICAgbGV0IGNvbXBsZXRlZCA9IGZhbHNlO1xuICAgIGNvbnN0IG11dGFibGVTdGFydFByb21pc2UgPSB0aGlzLnN0YXJ0SW50ZXJuYWwob3B0aW9ucykudGhlbigoKSA9PiB7XG4gICAgICBjb21wbGV0ZWQgPSB0cnVlO1xuICAgICAgdGhpcy5tdXRhYmxlU3RhcnRQcm9taXNlID0gdW5kZWZpbmVkO1xuICAgIH0pO1xuICAgIGlmICghY29tcGxldGVkKSB7XG4gICAgICB0aGlzLm11dGFibGVTdGFydFByb21pc2UgPSBtdXRhYmxlU3RhcnRQcm9taXNlO1xuICAgIH1cbiAgfVxuXG4gIHByaXZhdGUgYXN5bmMgc3RhcnRJbnRlcm5hbChvcHRpb25zOiBJbnRlcm5hbE9wdGlvbnMpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICBsb2dnZXIuaW5mbyh7IG5hbWU6ICduZW9fY29uc2Vuc3VzX3N0YXJ0JyB9LCAnQ29uc2Vuc3VzIHN0YXJ0ZWQuJyk7XG5cbiAgICBjb25zdCBpbml0aWFsUmVzdWx0ID0gYXdhaXQgaW5pdGlhbGl6ZU5ld0NvbnNlbnN1cyh7XG4gICAgICBibG9ja2NoYWluOiB0aGlzLm5vZGUuYmxvY2tjaGFpbixcbiAgICAgIHB1YmxpY0tleTogb3B0aW9ucy5wdWJsaWNLZXksXG4gICAgICBjb25zZW5zdXNDb250ZXh0OiB0aGlzLm11dGFibGVDb25zZW5zdXNDb250ZXh0LFxuICAgIH0pO1xuXG4gICAgYXdhaXQgQXN5bmNJdGVyYWJsZVguZnJvbSh0aGlzLm11dGFibGVRdWV1ZSlcbiAgICAgIC5waXBlKFxuICAgICAgICBzY2FuKGFzeW5jIChjb250ZXh0OiBDb250ZXh0LCBldmVudDogRXZlbnQpID0+IHtcbiAgICAgICAgICBsZXQgcmVzdWx0O1xuICAgICAgICAgIHN3aXRjaCAoZXZlbnQudHlwZSkge1xuICAgICAgICAgICAgY2FzZSAnaGFuZGxlUGVyc2lzdEJsb2NrJzpcbiAgICAgICAgICAgICAgcmVzdWx0ID0gYXdhaXQgaGFuZGxlUGVyc2lzdEJsb2NrKHtcbiAgICAgICAgICAgICAgICBibG9ja2NoYWluOiB0aGlzLm5vZGUuYmxvY2tjaGFpbixcbiAgICAgICAgICAgICAgICBwdWJsaWNLZXk6IG9wdGlvbnMucHVibGljS2V5LFxuICAgICAgICAgICAgICAgIGNvbnNlbnN1c0NvbnRleHQ6IHRoaXMubXV0YWJsZUNvbnNlbnN1c0NvbnRleHQsXG4gICAgICAgICAgICAgIH0pO1xuXG4gICAgICAgICAgICAgIGJyZWFrO1xuICAgICAgICAgICAgY2FzZSAnaGFuZGxlQ29uc2Vuc3VzUGF5bG9hZCc6XG4gICAgICAgICAgICAgIHJlc3VsdCA9IGF3YWl0IGhhbmRsZUNvbnNlbnN1c1BheWxvYWQoe1xuICAgICAgICAgICAgICAgIGNvbnRleHQsXG4gICAgICAgICAgICAgICAgbm9kZTogdGhpcy5ub2RlLFxuICAgICAgICAgICAgICAgIHByaXZhdGVLZXk6IG9wdGlvbnMucHJpdmF0ZUtleSxcbiAgICAgICAgICAgICAgICBwYXlsb2FkOiBldmVudC5wYXlsb2FkLFxuICAgICAgICAgICAgICAgIGNvbnNlbnN1c0NvbnRleHQ6IHRoaXMubXV0YWJsZUNvbnNlbnN1c0NvbnRleHQsXG4gICAgICAgICAgICAgIH0pO1xuXG4gICAgICAgICAgICAgIGJyZWFrO1xuICAgICAgICAgICAgY2FzZSAnaGFuZGxlVHJhbnNhY3Rpb25SZWNlaXZlZCc6XG4gICAgICAgICAgICAgIHJlc3VsdCA9IGF3YWl0IGhhbmRsZVRyYW5zYWN0aW9uUmVjZWl2ZWQoe1xuICAgICAgICAgICAgICAgIGNvbnRleHQsXG4gICAgICAgICAgICAgICAgbm9kZTogdGhpcy5ub2RlLFxuICAgICAgICAgICAgICAgIHByaXZhdGVLZXk6IG9wdGlvbnMucHJpdmF0ZUtleSxcbiAgICAgICAgICAgICAgICB0cmFuc2FjdGlvbjogZXZlbnQudHJhbnNhY3Rpb24sXG4gICAgICAgICAgICAgICAgY29uc2Vuc3VzQ29udGV4dDogdGhpcy5tdXRhYmxlQ29uc2Vuc3VzQ29udGV4dCxcbiAgICAgICAgICAgICAgfSk7XG5cbiAgICAgICAgICAgICAgYnJlYWs7XG4gICAgICAgICAgICBjYXNlICd0aW1lcic6XG4gICAgICAgICAgICAgIHJlc3VsdCA9IGF3YWl0IHJ1bkNvbnNlbnN1cyh7XG4gICAgICAgICAgICAgICAgY29udGV4dCxcbiAgICAgICAgICAgICAgICBub2RlOiB0aGlzLm5vZGUsXG4gICAgICAgICAgICAgICAgb3B0aW9ucyxcbiAgICAgICAgICAgICAgICBjb25zZW5zdXNDb250ZXh0OiB0aGlzLm11dGFibGVDb25zZW5zdXNDb250ZXh0LFxuICAgICAgICAgICAgICB9KS5jYXRjaCgoZXJyKSA9PiB7XG4gICAgICAgICAgICAgICAgaWYgKGV2ZW50LnByb21pc2UgIT09IHVuZGVmaW5lZCkge1xuICAgICAgICAgICAgICAgICAgZXZlbnQucHJvbWlzZS5yZWplY3QoZXJyKTtcbiAgICAgICAgICAgICAgICB9XG4gICAgICAgICAgICAgICAgdGhyb3cgZXJyO1xuICAgICAgICAgICAgICB9KTtcbiAgICAgICAgICAgICAgaWYgKGV2ZW50LnByb21pc2UgIT09IHVuZGVmaW5lZCkge1xuICAgICAgICAgICAgICAgIGV2ZW50LnByb21pc2UucmVzb2x2ZSgpO1xuICAgICAgICAgICAgICB9XG4gICAgICAgICAgICAgIGJyZWFrO1xuICAgICAgICAgICAgZGVmYXVsdDpcbiAgICAgICAgICAgICAgY29tbW9uVXRpbHMuYXNzZXJ0TmV2ZXIoZXZlbnQpO1xuICAgICAgICAgICAgICB0aHJvdyBuZXcgRXJyb3IoJ0ZvciBUUycpO1xuICAgICAgICAgIH1cblxuICAgICAgICAgIHJldHVybiB0aGlzLmhhbmRsZVJlc3VsdChyZXN1bHQpO1xuICAgICAgICB9LCB0aGlzLmhhbmRsZVJlc3VsdChpbml0aWFsUmVzdWx0KSksXG4gICAgICApXG4gICAgICAuZm9yRWFjaCgoKSA9PiB7XG4gICAgICAgIC8vIGRvIG5vdGhpbmdcbiAgICAgIH0pO1xuXG4gICAgbG9nZ2VyLmluZm8oeyBuYW1lOiAnbmVvX2NvbnNlbnN1c19zdG9wJyB9LCAnQ29uc2Vuc3VzIHN0b3BwZWQuJyk7XG4gIH1cblxuICBwcml2YXRlIGhhbmRsZVJlc3VsdChyZXN1bHQ6IFJlc3VsdDxDb250ZXh0Pik6IENvbnRleHQge1xuICAgIGlmIChyZXN1bHQudGltZXJTZWNvbmRzICE9PSB1bmRlZmluZWQpIHtcbiAgICAgIHRoaXMuaGFuZGxlVGltZXIocmVzdWx0LnRpbWVyU2Vjb25kcyk7XG4gICAgfVxuXG4gICAgcmV0dXJuIHJlc3VsdC5jb250ZXh0O1xuICB9XG5cbiAgcHJpdmF0ZSBoYW5kbGVUaW1lcihtdXRhYmxlVGltZXJTZWNvbmRzOiBudW1iZXIpOiB2b2lkIHtcbiAgICB0aGlzLmNsZWFyVGltZXIoKTtcbiAgICB0aGlzLm11dGFibGVUaW1lciA9IHNldFRpbWVvdXQoXG4gICAgICAoKSA9PiB0aGlzLm11dGFibGVRdWV1ZS53cml0ZSh7IHR5cGU6ICd0aW1lcicgfSksXG4gICAgICBtdXRhYmxlVGltZXJTZWNvbmRzICogTVNfSU5fU0VDT05ELFxuICAgICAgLy8gdHNsaW50OmRpc2FibGUtbmV4dC1saW5lIG5vLWFueVxuICAgICkgYXMgYW55O1xuICB9XG5cbiAgcHJpdmF0ZSBjbGVhclRpbWVyKCk6IHZvaWQge1xuICAgIGlmICh0aGlzLm11dGFibGVUaW1lciAhPT0gdW5kZWZpbmVkKSB7XG4gICAgICBjbGVhclRpbWVvdXQodGhpcy5tdXRhYmxlVGltZXIpO1xuICAgICAgdGhpcy5tdXRhYmxlVGltZXIgPSB1bmRlZmluZWQ7XG4gICAgfVxuICB9XG59XG4iXX0=