@maci-protocol/contracts 0.0.0-ci.185b643

package/build/tasks/helpers/Prover.js

@@ -0,0 +1,236 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Prover = void 0;
+/* eslint-disable no-console, no-await-in-loop */
+const core_1 = require("@maci-protocol/core");
+const crypto_1 = require("@maci-protocol/crypto");
+const domainobjs_1 = require("@maci-protocol/domainobjs");
+const logger_1 = require("../../ts/logger");
+const utils_1 = require("../../ts/utils");
+/**
+ * Prover class is designed to prove message processing and tally proofs on-chain.
+ */
+class Prover {
+    /**
+     * Initialize class properties
+     *
+     * @param {IProverParams} params - constructor params
+     */
+    constructor({ pollContract, mpContract, maciContract, vkRegistryContract, verifierContract, tallyContract, }) {
+        this.pollContract = pollContract;
+        this.mpContract = mpContract;
+        this.maciContract = maciContract;
+        this.vkRegistryContract = vkRegistryContract;
+        this.verifierContract = verifierContract;
+        this.tallyContract = tallyContract;
+    }
+    /**
+     * Prove message processing on-chain
+     *
+     * @param proofs - proofs
+     */
+    async proveMessageProcessing(proofs) {
+        // retrieve the values we need from the smart contracts
+        const [treeDepths, messageBatchSize, numSignUpsAndMessages, numBatchesProcessed, stateTreeDepth, coordinatorPubKeyHash, mode, batchHashes, lastChainHash,] = await Promise.all([
+            this.pollContract.treeDepths(),
+            this.pollContract.messageBatchSize().then(Number),
+            this.pollContract.numSignUpsAndMessages(),
+            this.mpContract.numBatchesProcessed().then(Number),
+            this.maciContract.stateTreeDepth().then(Number),
+            this.pollContract.coordinatorPubKeyHash(),
+            this.mpContract.mode(),
+            this.pollContract.getBatchHashes().then((res) => [...res]),
+            this.pollContract.chainHash(),
+        ]);
+        const numMessages = Number(numSignUpsAndMessages[1]);
+        const totalMessageBatches = batchHashes.length;
+        let numberBatchesProcessed = numBatchesProcessed;
+        const onChainProcessVk = await this.vkRegistryContract.getProcessVk(stateTreeDepth, treeDepths.voteOptionTreeDepth, messageBatchSize, mode);
+        if (numberBatchesProcessed < totalMessageBatches) {
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Submitting proofs of message processing...") });
+        }
+        // process all batches left
+        for (let i = numberBatchesProcessed; i < totalMessageBatches; i += 1) {
+            let currentMessageBatchIndex;
+            if (numberBatchesProcessed === 0) {
+                const chainHash = lastChainHash;
+                if (numMessages % messageBatchSize !== 0) {
+                    batchHashes.push(chainHash);
+                }
+                currentMessageBatchIndex = batchHashes.length;
+                if (currentMessageBatchIndex > 0) {
+                    currentMessageBatchIndex -= 1;
+                }
+            }
+            else {
+                currentMessageBatchIndex = (totalMessageBatches - numberBatchesProcessed) * messageBatchSize;
+            }
+            if (numberBatchesProcessed > 0 && currentMessageBatchIndex > 0) {
+                currentMessageBatchIndex /= messageBatchSize;
+            }
+            const { proof, circuitInputs, publicInputs } = proofs[i];
+            const inputBatchHash = batchHashes[currentMessageBatchIndex - 1];
+            if (BigInt(circuitInputs.inputBatchHash).toString() !== inputBatchHash.toString()) {
+                throw new Error("input batch hash mismatch.");
+            }
+            const outputBatchHash = batchHashes[currentMessageBatchIndex];
+            if (BigInt(circuitInputs.outputBatchHash).toString() !== outputBatchHash.toString()) {
+                throw new Error("output batch hash mismatch.");
+            }
+            let currentSbCommitmentOnChain;
+            if (numberBatchesProcessed === 0) {
+                currentSbCommitmentOnChain = BigInt(await this.pollContract.currentSbCommitment());
+            }
+            else {
+                currentSbCommitmentOnChain = BigInt(await this.mpContract.sbCommitment());
+            }
+            this.validateCommitment(circuitInputs.currentSbCommitment, currentSbCommitmentOnChain);
+            if (circuitInputs.coordinatorPublicKeyHash.toString() !== coordinatorPubKeyHash.toString()) {
+                throw new Error("coordPubKey mismatch.");
+            }
+            const formattedProof = (0, utils_1.formatProofForVerifierContract)(proof);
+            const publicInputsOnChain = await this.mpContract
+                .getPublicCircuitInputs(currentMessageBatchIndex, (0, utils_1.asHex)(circuitInputs.newSbCommitment), outputBatchHash.toString())
+                .then((value) => [...value]);
+            this.validatePublicInput(publicInputs, publicInputsOnChain);
+            const vk = new domainobjs_1.VerifyingKey(new crypto_1.G1Point(onChainProcessVk.alpha1[0], onChainProcessVk.alpha1[1]), new crypto_1.G2Point(onChainProcessVk.beta2[0], onChainProcessVk.beta2[1]), new crypto_1.G2Point(onChainProcessVk.gamma2[0], onChainProcessVk.gamma2[1]), new crypto_1.G2Point(onChainProcessVk.delta2[0], onChainProcessVk.delta2[1]), onChainProcessVk.ic.map(([x, y]) => new crypto_1.G1Point(x, y)));
+            // verify the proof on chain using the verifier contract
+            const isValidOnChain = await this.verifierContract.verify(formattedProof, vk.asContractParam(), [...publicInputsOnChain]);
+            if (!isValidOnChain) {
+                throw new Error("The verifier contract found the proof invalid.");
+            }
+            try {
+                // validate process messaging proof and store the new state and ballot root commitment
+                const receipt = await this.mpContract
+                    .processMessages((0, utils_1.asHex)(circuitInputs.newSbCommitment), formattedProof)
+                    .then((tx) => tx.wait());
+                if (receipt?.status !== 1) {
+                    throw new Error("processMessages() failed.");
+                }
+                (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Process messages transaction hash: ${receipt.hash}`) });
+                // Wait for the node to catch up
+                numberBatchesProcessed = await this.mpContract.numBatchesProcessed().then(Number);
+                (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Progress: ${numberBatchesProcessed} / ${totalMessageBatches}`) });
+            }
+            catch (err) {
+                throw new Error(`processMessages() failed: ${err.message}`);
+            }
+        }
+        if (numberBatchesProcessed === totalMessageBatches) {
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)("All message processing proofs have been submitted.") });
+        }
+    }
+    /**
+     * Prove tally on-chain
+     *
+     * @param proofs tally proofs
+     */
+    async proveTally(proofs) {
+        const [treeDepths, numSignUpsAndMessages, tallyBatchNumber, mode, stateTreeDepth] = await Promise.all([
+            this.pollContract.treeDepths(),
+            this.pollContract.numSignUpsAndMessages(),
+            this.tallyContract.tallyBatchNum().then(Number),
+            this.tallyContract.mode(),
+            this.maciContract.stateTreeDepth().then(Number),
+        ]);
+        const onChainTallyVk = await this.vkRegistryContract.getTallyVk(stateTreeDepth, treeDepths.intStateTreeDepth, treeDepths.voteOptionTreeDepth, mode);
+        const numSignUps = Number(numSignUpsAndMessages[0]);
+        const tallyBatchSize = core_1.STATE_TREE_ARITY ** Number(treeDepths.intStateTreeDepth);
+        // vote tallying proofs
+        const totalTallyBatches = numSignUps % tallyBatchSize === 0
+            ? Math.floor(numSignUps / tallyBatchSize)
+            : Math.floor(numSignUps / tallyBatchSize) + 1;
+        let tallyBatchNum = tallyBatchNumber;
+        if (tallyBatchNum < totalTallyBatches) {
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Submitting proofs of vote tallying...") });
+        }
+        for (let i = tallyBatchNum; i < totalTallyBatches; i += 1) {
+            if (i === 0) {
+                await this.tallyContract.updateSbCommitment().then((tx) => tx.wait());
+            }
+            const batchStartIndex = i * tallyBatchSize;
+            const { proof, circuitInputs, publicInputs } = proofs[i];
+            const currentTallyCommitmentOnChain = await this.tallyContract.tallyCommitment();
+            this.validateCommitment(circuitInputs.currentTallyCommitment, currentTallyCommitmentOnChain);
+            const currentSbCommitmentOnChain = await this.mpContract.sbCommitment();
+            this.validateCommitment(circuitInputs.sbCommitment, currentSbCommitmentOnChain);
+            const publicInputsOnChain = await this.tallyContract.getPublicCircuitInputs(batchStartIndex, (0, utils_1.asHex)(circuitInputs.newTallyCommitment));
+            this.validatePublicInput(publicInputs, publicInputsOnChain);
+            // format the tally proof so it can be verified on chain
+            const formattedProof = (0, utils_1.formatProofForVerifierContract)(proof);
+            const vk = new domainobjs_1.VerifyingKey(new crypto_1.G1Point(onChainTallyVk.alpha1[0], onChainTallyVk.alpha1[1]), new crypto_1.G2Point(onChainTallyVk.beta2[0], onChainTallyVk.beta2[1]), new crypto_1.G2Point(onChainTallyVk.gamma2[0], onChainTallyVk.gamma2[1]), new crypto_1.G2Point(onChainTallyVk.delta2[0], onChainTallyVk.delta2[1]), onChainTallyVk.ic.map(([x, y]) => new crypto_1.G1Point(x, y)));
+            const isValidOnChain = await this.verifierContract.verify(formattedProof, vk.asContractParam(), [...publicInputsOnChain]);
+            if (!isValidOnChain) {
+                throw new Error("The verifier contract found the proof invalid.");
+            }
+            // verify the proof on chain
+            const receipt = await this.tallyContract
+                .tallyVotes((0, utils_1.asHex)(circuitInputs.newTallyCommitment), formattedProof)
+                .then((tx) => tx.wait());
+            if (receipt?.status !== 1) {
+                throw new Error("tallyVotes() failed");
+            }
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Progress: ${tallyBatchNum + 1} / ${totalTallyBatches}`) });
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Tally votes transaction hash: ${receipt.hash}`) });
+            tallyBatchNum = Number(await this.tallyContract.tallyBatchNum());
+        }
+        if (tallyBatchNum === totalTallyBatches) {
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)("All vote tallying proofs have been submitted.") });
+        }
+    }
+    /**
+     * Submit tally results on chain
+     *
+     * @param tallyData - tally data
+     * @param recipients - number of recipients
+     */
+    async submitResults(tallyData, recipients) {
+        (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Submitting results...") });
+        const tallyResults = tallyData.results.tally.map((t) => BigInt(t));
+        const resultLength = recipients ?? tallyResults.length;
+        // slice in case we are submitting partial results
+        const partialResults = tallyResults.slice(0, resultLength);
+        const [treeDepths] = await Promise.all([this.pollContract.treeDepths()]);
+        const tallyResultProofs = partialResults.map((_, index) => (0, crypto_1.genTreeProof)(index, tallyResults, Number(treeDepths.voteOptionTreeDepth)));
+        await this.tallyContract
+            .addTallyResults({
+            voteOptionIndices: partialResults.map((_, index) => index),
+            tallyResults: partialResults,
+            tallyResultProofs,
+            totalSpent: tallyData.totalSpentVoiceCredits.spent,
+            totalSpentSalt: tallyData.totalSpentVoiceCredits.salt,
+            tallyResultSalt: tallyData.results.salt,
+            newResultsCommitment: tallyData.results.commitment,
+            spentVoiceCreditsHash: tallyData.totalSpentVoiceCredits.commitment,
+            perVOSpentVoiceCreditsHash: tallyData.perVOSpentVoiceCredits?.commitment ?? 0n,
+        })
+            .then((tx) => tx.wait());
+        (0, logger_1.logGreen)({ text: (0, logger_1.success)("Results have been submitted.") });
+    }
+    /**
+     * Validate commitment
+     *
+     * @param commitment - off-chain commitment
+     * @param commitmentOnChain - on-chain commitment
+     * @throws error if commitments don't match
+     */
+    validateCommitment(value, onchainValue) {
+        if (onchainValue.toString() !== value.toString()) {
+            throw new Error("commitment mismatch");
+        }
+    }
+    /**
+     * Validate public input hash
+     *
+     * @param publicInputs - off-chain public input hash
+     * @param publicInputsOnChain - on-chain public input hash
+     * @throws error if public input hashes don't match
+     */
+    validatePublicInput(publicInputs, publicInputsOnChain) {
+        if (!publicInputsOnChain.every((value, index) => value.toString() === publicInputs[index].toString())) {
+            throw new Error("public input mismatch.");
+        }
+    }
+}
+exports.Prover = Prover;
+//# sourceMappingURL=Prover.js.map

package/build/tasks/helpers/Prover.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Prover.js","sourceRoot":"","sources":["../../../tasks/helpers/Prover.ts"],"names":[],"mappings":";;;AAAA,iDAAiD;AACjD,8CAAuD;AACvD,kDAAuE;AACvE,0DAAyD;AAMzD,4CAAsE;AACtE,0CAAuE;AAIvE;;GAEG;AACH,MAAa,MAAM;IA+BjB;;;;OAIG;IACH,YAAY,EACV,YAAY,EACZ,UAAU,EACV,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACC;QACd,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;QAC7C,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACrC,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,sBAAsB,CAAC,MAAe;QAC1C,uDAAuD;QACvD,MAAM,CACJ,UAAU,EACV,gBAAgB,EAChB,qBAAqB,EACrB,mBAAmB,EACnB,cAAc,EACd,qBAAqB,EACrB,IAAI,EACJ,WAAW,EACX,aAAa,EACd,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YACjD,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE;YACzC,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YAClD,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YAC/C,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE;YACzC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE;YACtB,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;YAC1D,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE;SAC9B,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,mBAAmB,GAAG,WAAW,CAAC,MAAM,CAAC;QAC/C,IAAI,sBAAsB,GAAG,mBAAmB,CAAC;QAEjD,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,YAAY,CACjE,cAAc,EACd,UAAU,CAAC,mBAAmB,EAC9B,gBAAgB,EAChB,IAAI,CACL,CAAC;QAEF,IAAI,sBAAsB,GAAG,mBAAmB,EAAE,CAAC;YACjD,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,4CAA4C,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,2BAA2B;QAC3B,KAAK,IAAI,CAAC,GAAG,sBAAsB,EAAE,CAAC,GAAG,mBAAmB,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACrE,IAAI,wBAAgC,CAAC;YAErC,IAAI,sBAAsB,KAAK,CAAC,EAAE,CAAC;gBACjC,MAAM,SAAS,GAAG,aAAa,CAAC;gBAEhC,IAAI,WAAW,GAAG,gBAAgB,KAAK,CAAC,EAAE,CAAC;oBACzC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC9B,CAAC;gBAED,wBAAwB,GAAG,WAAW,CAAC,MAAM,CAAC;gBAE9C,IAAI,wBAAwB,GAAG,CAAC,EAAE,CAAC;oBACjC,wBAAwB,IAAI,CAAC,CAAC;gBAChC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,wBAAwB,GAAG,CAAC,mBAAmB,GAAG,sBAAsB,CAAC,GAAG,gBAAgB,CAAC;YAC/F,CAAC;YAED,IAAI,sBAAsB,GAAG,CAAC,IAAI,wBAAwB,GAAG,CAAC,EAAE,CAAC;gBAC/D,wBAAwB,IAAI,gBAAgB,CAAC;YAC/C,CAAC;YAED,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAEzD,MAAM,cAAc,GAAG,WAAW,CAAC,wBAAwB,GAAG,CAAC,CAAC,CAAC;YAEjE,IAAI,MAAM,CAAC,aAAa,CAAC,cAA8B,CAAC,CAAC,QAAQ,EAAE,KAAK,cAAc,CAAC,QAAQ,EAAE,EAAE,CAAC;gBAClG,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAChD,CAAC;YAED,MAAM,eAAe,GAAG,WAAW,CAAC,wBAAwB,CAAC,CAAC;YAE9D,IAAI,MAAM,CAAC,aAAa,CAAC,eAA+B,CAAC,CAAC,QAAQ,EAAE,KAAK,eAAe,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACpG,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YAED,IAAI,0BAAkC,CAAC;YAEvC,IAAI,sBAAsB,KAAK,CAAC,EAAE,CAAC;gBACjC,0BAA0B,GAAG,MAAM,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,CAAC,CAAC;YACrF,CAAC;iBAAM,CAAC;gBACN,0BAA0B,GAAG,MAAM,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC,CAAC;YAC5E,CAAC;YAED,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,mBAAmC,EAAE,0BAA0B,CAAC,CAAC;YAEvG,IAAI,aAAa,CAAC,wBAAwB,CAAC,QAAQ,EAAE,KAAK,qBAAqB,CAAC,QAAQ,EAAE,EAAE,CAAC;gBAC3F,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,cAAc,GAAG,IAAA,sCAA8B,EAAC,KAAK,CAAC,CAAC;YAE7D,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,UAAU;iBAC9C,sBAAsB,CACrB,wBAAwB,EACxB,IAAA,aAAK,EAAC,aAAa,CAAC,eAA+B,CAAC,EACpD,eAAe,CAAC,QAAQ,EAAE,CAC3B;iBACA,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC;YAC/B,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,mBAAmB,CAAC,CAAC;YAE5D,MAAM,EAAE,GAAG,IAAI,yBAAY,CACzB,IAAI,gBAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EACnE,IAAI,gBAAO,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EACjE,IAAI,gBAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EACnE,IAAI,gBAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EACnE,gBAAgB,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,gBAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CACvD,CAAC;YAEF,wDAAwD;YACxD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CACvD,cAAc,EACd,EAAE,CAAC,eAAe,EAAyB,EAC3C,CAAC,GAAG,mBAAmB,CAAC,CACzB,CAAC;YAEF,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;YACpE,CAAC;YAED,IAAI,CAAC;gBACH,sFAAsF;gBACtF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU;qBAClC,eAAe,CAAC,IAAA,aAAK,EAAC,aAAa,CAAC,eAA+B,CAAC,EAAE,cAAc,CAAC;qBACrF,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;gBAE3B,IAAI,OAAO,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;gBAC/C,CAAC;gBAED,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,sCAAsC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;gBAElF,gCAAgC;gBAEhC,sBAAsB,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAElF,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,aAAa,sBAAsB,MAAM,mBAAmB,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7F,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,6BAA8B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;QAED,IAAI,sBAAsB,KAAK,mBAAmB,EAAE,CAAC;YACnD,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,oDAAoD,CAAC,EAAE,CAAC,CAAC;QACpF,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU,CAAC,MAAe;QAC9B,MAAM,CAAC,UAAU,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,IAAI,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpG,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE;YAC9B,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE;YACzC,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;YAC/C,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE;YACzB,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC;SAChD,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,CAC7D,cAAc,EACd,UAAU,CAAC,iBAAiB,EAC5B,UAAU,CAAC,mBAAmB,EAC9B,IAAI,CACL,CAAC;QAEF,MAAM,UAAU,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,cAAc,GAAG,uBAAgB,IAAI,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC;QAEhF,uBAAuB;QACvB,MAAM,iBAAiB,GACrB,UAAU,GAAG,cAAc,KAAK,CAAC;YAC/B,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,cAAc,CAAC;YACzC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;QAElD,IAAI,aAAa,GAAG,gBAAgB,CAAC;QAErC,IAAI,aAAa,GAAG,iBAAiB,EAAE,CAAC;YACtC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,uCAAuC,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,KAAK,IAAI,CAAC,GAAG,aAAa,EAAE,CAAC,GAAG,iBAAiB,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;gBACZ,MAAM,IAAI,CAAC,aAAa,CAAC,kBAAkB,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YACxE,CAAC;YAED,MAAM,eAAe,GAAG,CAAC,GAAG,cAAc,CAAC;YAC3C,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAEzD,MAAM,6BAA6B,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,CAAC;YACjF,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,sBAAsC,EAAE,6BAA6B,CAAC,CAAC;YAE7G,MAAM,0BAA0B,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC;YACxE,IAAI,CAAC,kBAAkB,CAAC,aAAa,CAAC,YAA4B,EAAE,0BAA0B,CAAC,CAAC;YAEhG,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,sBAAsB,CACzE,eAAe,EACf,IAAA,aAAK,EAAC,aAAa,CAAC,kBAAkC,CAAC,CACxD,CAAC;YACF,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,mBAAmB,CAAC,CAAC;YAE5D,wDAAwD;YACxD,MAAM,cAAc,GAAG,IAAA,sCAA8B,EAAC,KAAK,CAAC,CAAC;YAE7D,MAAM,EAAE,GAAG,IAAI,yBAAY,CACzB,IAAI,gBAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAC/D,IAAI,gBAAO,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAC7D,IAAI,gBAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAC/D,IAAI,gBAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAC/D,cAAc,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,gBAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CACrD,CAAC;YAEF,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CACvD,cAAc,EACd,EAAE,CAAC,eAAe,EAAyB,EAC3C,CAAC,GAAG,mBAAmB,CAAC,CACzB,CAAC;YAEF,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;YACpE,CAAC;YAED,4BAA4B;YAC5B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa;iBACrC,UAAU,CAAC,IAAA,aAAK,EAAC,aAAa,CAAC,kBAAkC,CAAC,EAAE,cAAc,CAAC;iBACnF,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAE3B,IAAI,OAAO,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YAED,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,aAAa,aAAa,GAAG,CAAC,MAAM,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;YACpF,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,iCAAiC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAE7E,aAAa,GAAG,MAAM,CAAC,MAAM,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,aAAa,KAAK,iBAAiB,EAAE,CAAC;YACxC,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,+CAA+C,CAAC,EAAE,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CAAC,SAAoB,EAAE,UAAmB;QAC3D,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;QAEpD,MAAM,YAAY,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,YAAY,GAAG,UAAU,IAAI,YAAY,CAAC,MAAM,CAAC;QAEvD,kDAAkD;QAClD,MAAM,cAAc,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;QAE3D,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QAEzE,MAAM,iBAAiB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CACxD,IAAA,qBAAY,EAAC,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAC1E,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa;aACrB,eAAe,CAAC;YACf,iBAAiB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC;YAC1D,YAAY,EAAE,cAAc;YAC5B,iBAAiB;YACjB,UAAU,EAAE,SAAS,CAAC,sBAAsB,CAAC,KAAK;YAClD,cAAc,EAAE,SAAS,CAAC,sBAAsB,CAAC,IAAI;YACrD,eAAe,EAAE,SAAS,CAAC,OAAO,CAAC,IAAI;YACvC,oBAAoB,EAAE,SAAS,CAAC,OAAO,CAAC,UAAU;YAClD,qBAAqB,EAAE,SAAS,CAAC,sBAAsB,CAAC,UAAU;YAClE,0BAA0B,EAAE,SAAS,CAAC,sBAAsB,EAAE,UAAU,IAAI,EAAE;SAC/E,CAAC;aACD,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;QAE3B,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,8BAA8B,CAAC,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED;;;;;;OAMG;IACK,kBAAkB,CAAC,KAAmB,EAAE,YAA0B;QACxE,IAAI,YAAY,CAAC,QAAQ,EAAE,KAAK,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,mBAAmB,CAAC,YAA4B,EAAE,mBAAmC;QAC3F,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,YAAY,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,CAAC;YACtG,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;CACF;AA7WD,wBA6WC"}

package/build/tasks/helpers/TreeMerger.d.ts

@@ -0,0 +1,32 @@
+import type { ITreeMergeParams } from "./types";
+/**
+ * @notice Tree merger keeps merging simple for hardhat task.
+ * This class is using for merging signups and messages.
+ */
+export declare class TreeMerger {
+    /**
+     * Poll contract
+     */
+    private pollContract;
+    /**
+     * Ethers signer
+     */
+    private deployer;
+    /**
+     * Initialize class properties
+     *
+     * @param {ITreeMergeParams} params - contracts and signer
+     */
+    constructor({ deployer, pollContract }: ITreeMergeParams);
+    /**
+     * Check if voting period is over. Otherwise, throw an error.
+     */
+    checkPollDuration(): Promise<void>;
+    /**
+     * Merge user signup MACI state
+     *
+     * @param pollId - poll id
+     */
+    mergeSignups(): Promise<void>;
+}
+//# sourceMappingURL=TreeMerger.d.ts.map

package/build/tasks/helpers/TreeMerger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TreeMerger.d.ts","sourceRoot":"","sources":["../../../tasks/helpers/TreeMerger.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAMhD;;;GAGG;AACH,qBAAa,UAAU;IACrB;;OAEG;IACH,OAAO,CAAC,YAAY,CAAO;IAE3B;;OAEG;IACH,OAAO,CAAC,QAAQ,CAAsB;IAEtC;;;;OAIG;gBACS,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,gBAAgB;IAKxD;;OAEG;IACG,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAaxC;;;;OAIG;IACG,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;CAmBpC"}

package/build/tasks/helpers/TreeMerger.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TreeMerger = void 0;
+const logger_1 = require("../../ts/logger");
+/**
+ * @notice Tree merger keeps merging simple for hardhat task.
+ * This class is using for merging signups and messages.
+ */
+class TreeMerger {
+    /**
+     * Initialize class properties
+     *
+     * @param {ITreeMergeParams} params - contracts and signer
+     */
+    constructor({ deployer, pollContract }) {
+        this.pollContract = pollContract;
+        this.deployer = deployer;
+    }
+    /**
+     * Check if voting period is over. Otherwise, throw an error.
+     */
+    async checkPollDuration() {
+        // check if it's time to merge the message AQ
+        const deadline = await this.pollContract.endDate();
+        const blockNum = await this.deployer.provider.getBlockNumber();
+        const block = await this.deployer.provider.getBlock(blockNum);
+        const now = Number(block?.timestamp);
+        if (now < deadline) {
+            throw new Error("Voting period is not over");
+        }
+    }
+    /**
+     * Merge user signup MACI state
+     *
+     * @param pollId - poll id
+     */
+    async mergeSignups() {
+        // check if the state tree has been fully merged
+        const isMerged = await this.pollContract.stateMerged();
+        if (!isMerged) {
+            // go and merge the state tree
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Merging subroots to a main state root...") });
+            const receipt = await this.pollContract.mergeState().then((tx) => tx.wait());
+            if (receipt?.status !== 1) {
+                throw new Error("Error merging signup state subroots");
+            }
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Merge state transaction hash: ${receipt.hash}`) });
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Gas used: ${receipt.gasUsed.toString()}`) });
+        }
+        else {
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("The state tree has already been merged.") });
+        }
+    }
+}
+exports.TreeMerger = TreeMerger;
+//# sourceMappingURL=TreeMerger.js.map

package/build/tasks/helpers/TreeMerger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TreeMerger.js","sourceRoot":"","sources":["../../../tasks/helpers/TreeMerger.ts"],"names":[],"mappings":";;;AAKA,4CAAsE;AAEtE;;;GAGG;AACH,MAAa,UAAU;IAWrB;;;;OAIG;IACH,YAAY,EAAE,QAAQ,EAAE,YAAY,EAAoB;QACtD,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB;QACrB,6CAA6C;QAC7C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;QAC/D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC9D,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAErC,IAAI,GAAG,GAAG,QAAQ,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,YAAY;QAChB,gDAAgD;QAChD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC;QAEvD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,8BAA8B;YAC9B,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,0CAA0C,CAAC,EAAE,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YAE7E,IAAI,OAAO,EAAE,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,CAAC;YAED,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,iCAAiC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7E,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,aAAa,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACxE,CAAC;aAAM,CAAC;YACN,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,yCAAyC,CAAC,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;CACF;AA7DD,gCA6DC"}

package/build/tasks/helpers/abi.d.ts

@@ -0,0 +1,15 @@
+import type { MACILibraryAddresses } from "../../typechain-types/factories/contracts/MACI__factory";
+import type { MessageProcessorLibraryAddresses } from "../../typechain-types/factories/contracts/MessageProcessor__factory";
+import type { PollFactoryLibraryAddresses } from "../../typechain-types/factories/contracts/PollFactory__factory";
+import type { TallyLibraryAddresses } from "../../typechain-types/factories/contracts/Tally__factory";
+/**
+ * Create linked contract addresses object
+ *
+ * @param poseidonT3ContractAddress - PoseidonT3 contract address
+ * @param poseidonT4ContractAddress - PoseidonT4 contract address
+ * @param poseidonT5ContractAddress - PoseidonT5 contract address
+ * @param poseidonT6ContractAddress - PoseidonT6 contract address
+ * @returns
+ */
+export declare const linkPoseidonLibraries: (poseidonT3ContractAddress: string, poseidonT4ContractAddress: string, poseidonT5ContractAddress: string, poseidonT6ContractAddress: string) => TallyLibraryAddresses | MessageProcessorLibraryAddresses | MACILibraryAddresses | PollFactoryLibraryAddresses;
+//# sourceMappingURL=abi.d.ts.map

package/build/tasks/helpers/abi.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"abi.d.ts","sourceRoot":"","sources":["../../../tasks/helpers/abi.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,yDAAyD,CAAC;AACpG,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,qEAAqE,CAAC;AAC5H,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,gEAAgE,CAAC;AAClH,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,0DAA0D,CAAC;AAEtG;;;;;;;;GAQG;AACH,eAAO,MAAM,qBAAqB,GAChC,2BAA2B,MAAM,EACjC,2BAA2B,MAAM,EACjC,2BAA2B,MAAM,EACjC,2BAA2B,MAAM,KAChC,qBAAqB,GAAG,gCAAgC,GAAG,oBAAoB,GAAG,2BAKnF,CAAC"}

package/build/tasks/helpers/abi.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.linkPoseidonLibraries = void 0;
+/**
+ * Create linked contract addresses object
+ *
+ * @param poseidonT3ContractAddress - PoseidonT3 contract address
+ * @param poseidonT4ContractAddress - PoseidonT4 contract address
+ * @param poseidonT5ContractAddress - PoseidonT5 contract address
+ * @param poseidonT6ContractAddress - PoseidonT6 contract address
+ * @returns
+ */
+const linkPoseidonLibraries = (poseidonT3ContractAddress, poseidonT4ContractAddress, poseidonT5ContractAddress, poseidonT6ContractAddress) => ({
+    "contracts/crypto/PoseidonT3.sol:PoseidonT3": poseidonT3ContractAddress,
+    "contracts/crypto/PoseidonT4.sol:PoseidonT4": poseidonT4ContractAddress,
+    "contracts/crypto/PoseidonT5.sol:PoseidonT5": poseidonT5ContractAddress,
+    "contracts/crypto/PoseidonT6.sol:PoseidonT6": poseidonT6ContractAddress,
+});
+exports.linkPoseidonLibraries = linkPoseidonLibraries;
+//# sourceMappingURL=abi.js.map

package/build/tasks/helpers/abi.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"abi.js","sourceRoot":"","sources":["../../../tasks/helpers/abi.ts"],"names":[],"mappings":";;;AAKA;;;;;;;;GAQG;AACI,MAAM,qBAAqB,GAAG,CACnC,yBAAiC,EACjC,yBAAiC,EACjC,yBAAiC,EACjC,yBAAiC,EAC8E,EAAE,CAAC,CAAC;IACnH,4CAA4C,EAAE,yBAAyB;IACvE,4CAA4C,EAAE,yBAAyB;IACvE,4CAA4C,EAAE,yBAAyB;IACvE,4CAA4C,EAAE,yBAAyB;CACxE,CAAC,CAAC;AAVU,QAAA,qBAAqB,yBAU/B"}

package/build/tasks/helpers/benchmarks.d.ts

@@ -0,0 +1,13 @@
+import { Message, Keypair } from "@maci-protocol/domainobjs";
+import { Deployment } from "./Deployment";
+/**
+ * Utility function to send a batch of messages to a poll contract
+ * @dev This function will attempt to send a batch of messages to a poll contract
+ * and if it fails, it will attempt to send a smaller batch size
+ * @param deployment - The deployment instance
+ * @param message - The message to send
+ * @param keypair - The keypair to sign the message with
+ * @param batchSize - The number of messages to send
+ */
+export declare function publishBatch(deployment: Deployment, message: Message, keypair: Keypair, batchSize: number): Promise<void>;
+//# sourceMappingURL=benchmarks.d.ts.map

package/build/tasks/helpers/benchmarks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"benchmarks.d.ts","sourceRoot":"","sources":["../../../tasks/helpers/benchmarks.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,2BAA2B,CAAC;AAK7D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG1C;;;;;;;;GAQG;AACH,wBAAsB,YAAY,CAChC,UAAU,EAAE,UAAU,EACtB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,IAAI,CAAC,CAyCf"}

package/build/tasks/helpers/benchmarks.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.publishBatch = publishBatch;
+const logger_1 = require("../../ts/logger");
+const types_1 = require("./types");
+/**
+ * Utility function to send a batch of messages to a poll contract
+ * @dev This function will attempt to send a batch of messages to a poll contract
+ * and if it fails, it will attempt to send a smaller batch size
+ * @param deployment - The deployment instance
+ * @param message - The message to send
+ * @param keypair - The keypair to sign the message with
+ * @param batchSize - The number of messages to send
+ */
+async function publishBatch(deployment, message, keypair, batchSize) {
+    // then try and publish a large batch of votes
+    const maciContract = await deployment.getContract({
+        name: types_1.EContracts.MACI,
+    });
+    const lastPollId = (await maciContract.nextPollId()) - 1n;
+    const pollContract = await deployment.getContract({
+        name: types_1.EContracts.Poll,
+        key: `poll-${lastPollId}`,
+    });
+    const messageBatch = Array.from({ length: batchSize }, () => message.asContractParam());
+    const pubKeyBatch = Array.from({ length: batchSize }, () => keypair.pubKey.asContractParam());
+    let optimalBatchSize = batchSize;
+    while (optimalBatchSize > 0) {
+        const finalMessageBatch = messageBatch.slice(0, optimalBatchSize);
+        const finalPubKeyBatch = pubKeyBatch.slice(0, optimalBatchSize);
+        try {
+            // eslint-disable-next-line no-await-in-loop
+            const tx = await pollContract.publishMessageBatch(finalMessageBatch, finalPubKeyBatch);
+            // eslint-disable-next-line no-await-in-loop
+            const receipt = await tx.wait();
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Successfully published batch of ${optimalBatchSize} messages`) });
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Gas used: ${receipt?.gasUsed.toString()} wei`) });
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Tx: ${tx.hash}`) });
+            break; // If successful, we've found the largest working batch size
+        }
+        catch (err) {
+            // If this size doesn't work, reduce by 1 and try again
+            optimalBatchSize -= 1;
+        }
+    }
+    if (optimalBatchSize > 0) {
+        (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Found optimal batch size: ${optimalBatchSize}`) });
+    }
+    else {
+        (0, logger_1.logRed)({ text: (0, logger_1.error)("Unable to publish even a single message") });
+    }
+}
+//# sourceMappingURL=benchmarks.js.map

package/build/tasks/helpers/benchmarks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"benchmarks.js","sourceRoot":"","sources":["../../../tasks/helpers/benchmarks.ts"],"names":[],"mappings":";;AAkBA,oCA8CC;AA7DD,4CAAmE;AAInE,mCAAqC;AAErC;;;;;;;;GAQG;AACI,KAAK,UAAU,YAAY,CAChC,UAAsB,EACtB,OAAgB,EAChB,OAAgB,EAChB,SAAiB;IAEjB,8CAA8C;IAC9C,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,WAAW,CAAO;QACtD,IAAI,EAAE,kBAAU,CAAC,IAAI;KACtB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,CAAC,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC,GAAG,EAAE,CAAC;IAE1D,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,WAAW,CAAO;QACtD,IAAI,EAAE,kBAAU,CAAC,IAAI;QACrB,GAAG,EAAE,QAAQ,UAAU,EAAE;KAC1B,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;IACxF,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC;IAE9F,IAAI,gBAAgB,GAAG,SAAS,CAAC;IAEjC,OAAO,gBAAgB,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,iBAAiB,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;QAClE,MAAM,gBAAgB,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;QAEhE,IAAI,CAAC;YACH,4CAA4C;YAC5C,MAAM,EAAE,GAAG,MAAM,YAAY,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,gBAAgB,CAAC,CAAC;YACvF,4CAA4C;YAC5C,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;YAChC,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,mCAAmC,gBAAgB,WAAW,CAAC,EAAE,CAAC,CAAC;YAC5F,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,aAAa,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;YAC5E,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,OAAO,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,MAAM,CAAC,4DAA4D;QACrE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,uDAAuD;YACvD,gBAAgB,IAAI,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;QACzB,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,6BAA6B,gBAAgB,EAAE,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC;SAAM,CAAC;QACN,IAAA,eAAM,EAAC,EAAE,IAAI,EAAE,IAAA,cAAK,EAAC,yCAAyC,CAAC,EAAE,CAAC,CAAC;IACrE,CAAC;AACH,CAAC"}

package/build/tasks/helpers/constants.d.ts

@@ -0,0 +1,66 @@
+import { EContracts, EPolicies } from "./types";
+/**
+ * Deploy steps
+ */
+export declare enum EDeploySteps {
+    Policies = "full:deploy-policies",
+    Verifier = "full:deploy-verifier",
+    Poseidon = "full:deploy-poseidon",
+    PollFactory = "full:deploy-poll-factory",
+    MessageProcessorFactory = "full:deploy-message-processor-factory",
+    TallyFactory = "full:deploy-tally-factory",
+    Maci = "full:deploy-maci",
+    VkRegistry = "full:deploy-vk-registry",
+    InitialVoiceCreditProxy = "poll:deploy-initial-voice-credit-proxy",
+    PollPolicy = "poll:deploy-policy",
+    Poll = "poll:deploy-poll"
+}
+/**
+ * Supported networks for deployment and task running
+ */
+export declare enum ESupportedChains {
+    Sepolia = "sepolia",
+    Optimism = "optimism",
+    OptimismSepolia = "optimism_sepolia",
+    Scroll = "scroll",
+    ScrollSepolia = "scroll_sepolia",
+    Arbitrum = "arbitrum",
+    ArbitrumSepolia = "arbitrum_sepolia",
+    Base = "base",
+    BaseSepolia = "base_sepolia",
+    Gnosis = "gnosis",
+    GnosisChiado = "gnosis_chiado",
+    Polygon = "polygon",
+    PolygonAmoy = "polygon_amoy",
+    Coverage = "coverage",
+    Hardhat = "hardhat"
+}
+/**
+ * Supported network chain ids for deployment and task running
+ */
+export declare enum EChainId {
+    Hardhat = 31337,
+    Optimism = 10,
+    OptimismSepolia = 11155420,
+    Sepolia = 11155111,
+    Scroll = 534352,
+    ScrollSepolia = 534351,
+    Arbitrum = 42161,
+    ArbitrumSepolia = 421614,
+    Base = 8453,
+    BaseSepolia = 84532,
+    Gnosis = 100,
+    GnosisChiado = 10200,
+    Polygon = 137,
+    PolygonAmoy = 80002,
+    Coverage = 1337
+}
+/**
+ * Get network rpc urls object
+ *
+ * @returns {Record<ESupportedChains, string>} rpc urls for supported networks
+ */
+export declare const getNetworkRpcUrls: () => Record<ESupportedChains, string>;
+export declare const getEtherscanApiKeys: () => Record<ESupportedChains, string | undefined>;
+export declare const FULL_POLICY_NAMES: Record<EContracts.FreeForAllPolicy | EContracts.TokenPolicy | EContracts.EASPolicy | EContracts.GitcoinPassportPolicy | EContracts.HatsPolicy | EContracts.ZupassPolicy | EContracts.SemaphorePolicy | EContracts.MerkleProofPolicy | EContracts.AnonAadhaarPolicy | EContracts.ERC20VotesPolicy, EPolicies>;
+//# sourceMappingURL=constants.d.ts.map

package/build/tasks/helpers/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../../tasks/helpers/constants.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEhD;;GAEG;AACH,oBAAY,YAAY;IACtB,QAAQ,yBAAyB;IACjC,QAAQ,yBAAyB;IACjC,QAAQ,yBAAyB;IACjC,WAAW,6BAA6B;IACxC,uBAAuB,0CAA0C;IACjE,YAAY,8BAA8B;IAC1C,IAAI,qBAAqB;IACzB,UAAU,4BAA4B;IACtC,uBAAuB,2CAA2C;IAClE,UAAU,uBAAuB;IACjC,IAAI,qBAAqB;CAC1B;AAED;;GAEG;AACH,oBAAY,gBAAgB;IAC1B,OAAO,YAAY;IACnB,QAAQ,aAAa;IACrB,eAAe,qBAAqB;IACpC,MAAM,WAAW;IACjB,aAAa,mBAAmB;IAChC,QAAQ,aAAa;IACrB,eAAe,qBAAqB;IACpC,IAAI,SAAS;IACb,WAAW,iBAAiB;IAC5B,MAAM,WAAW;IACjB,YAAY,kBAAkB;IAC9B,OAAO,YAAY;IACnB,WAAW,iBAAiB;IAC5B,QAAQ,aAAa;IACrB,OAAO,YAAY;CACpB;AAED;;GAEG;AACH,oBAAY,QAAQ;IAClB,OAAO,QAAQ;IACf,QAAQ,KAAK;IACb,eAAe,WAAW;IAC1B,OAAO,WAAW;IAClB,MAAM,SAAS;IACf,aAAa,SAAS;IACtB,QAAQ,QAAQ;IAChB,eAAe,SAAS;IACxB,IAAI,OAAO;IACX,WAAW,QAAQ;IACnB,MAAM,MAAM;IACZ,YAAY,QAAQ;IACpB,OAAO,MAAM;IACb,WAAW,QAAQ;IACnB,QAAQ,OAAO;CAChB;AAED;;;;GAIG;AACH,eAAO,MAAM,iBAAiB,QAAO,MAAM,CAAC,gBAAgB,EAAE,MAAM,CAgCnE,CAAC;AAEF,eAAO,MAAM,mBAAmB,QAAO,MAAM,CAAC,gBAAgB,EAAE,MAAM,GAAG,SAAS,CAgBhF,CAAC;AAEH,eAAO,MAAM,iBAAiB,EAAE,MAAM,CAClC,UAAU,CAAC,gBAAgB,GAC3B,UAAU,CAAC,WAAW,GACtB,UAAU,CAAC,SAAS,GACpB,UAAU,CAAC,qBAAqB,GAChC,UAAU,CAAC,UAAU,GACrB,UAAU,CAAC,YAAY,GACvB,UAAU,CAAC,eAAe,GAC1B,UAAU,CAAC,iBAAiB,GAC5B,UAAU,CAAC,iBAAiB,GAC5B,UAAU,CAAC,gBAAgB,EAC7B,SAAS,CAYV,CAAC"}