@maci-protocol/contracts 0.0.0-ci.2653bc0

package/build/tasks/helpers/ProofGenerator.js

@@ -0,0 +1,305 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProofGenerator = void 0;
+/* eslint-disable no-console */
+const core_1 = require("@maci-protocol/core");
+const crypto_1 = require("@maci-protocol/crypto");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const logger_1 = require("../../ts/logger");
+const proofs_1 = require("../../ts/proofs");
+const utils_1 = require("../../ts/utils");
+/**
+ * Proof generator class for message processing and tally.
+ */
+class ProofGenerator {
+    /**
+     * Get maci state from local file or from contract
+     *
+     * @param {IPrepareStateParams} params - params to prepare maci state
+     * @returns {MaciState} maci state
+     */
+    static async prepareState({ maciContract, pollContract, pollId, maciPrivateKey, coordinatorKeypair, signer, ipfsMessageBackupFiles, outputDir, options: { transactionHash, stateFile, startBlock, endBlock, blocksPerBatch }, }) {
+        const isOutputDirExists = fs_1.default.existsSync(path_1.default.resolve(outputDir));
+        if (!isOutputDirExists) {
+            await fs_1.default.promises.mkdir(path_1.default.resolve(outputDir));
+        }
+        if (stateFile) {
+            const content = await fs_1.default.promises
+                .readFile(stateFile)
+                .then((res) => JSON.parse(res.toString()));
+            const serializedPrivateKey = maciPrivateKey.serialize();
+            const maciState = core_1.MaciState.fromJSON(content);
+            maciState.polls.forEach((poll) => {
+                poll.setCoordinatorKeypair(serializedPrivateKey);
+            });
+            return maciState;
+        }
+        // build an off-chain representation of the MACI contract using data in the contract storage
+        const [defaultStartBlockSignup, defaultStartBlockPoll, stateRoot, numSignups] = await Promise.all([
+            maciContract.queryFilter(maciContract.filters.SignUp(), startBlock).then((events) => events[0]?.blockNumber ?? 0),
+            maciContract
+                .queryFilter(maciContract.filters.DeployPoll(), startBlock)
+                .then((events) => events[0]?.blockNumber ?? 0),
+            maciContract.getStateTreeRoot(),
+            maciContract.numSignUps(),
+        ]);
+        const defaultStartBlock = Math.min(defaultStartBlockPoll, defaultStartBlockSignup);
+        let fromBlock = startBlock ? Number(startBlock) : defaultStartBlock;
+        const defaultEndBlock = await Promise.all([
+            pollContract
+                .queryFilter(pollContract.filters.MergeState(stateRoot, numSignups), fromBlock)
+                .then((events) => events[events.length - 1]?.blockNumber),
+        ]).then((blocks) => Math.max(...blocks));
+        if (transactionHash) {
+            const tx = await signer.provider.getTransaction(transactionHash);
+            fromBlock = tx?.blockNumber ?? defaultStartBlock;
+        }
+        (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Starting to fetch logs from block ${fromBlock}`) });
+        const maciContractAddress = await maciContract.getAddress();
+        return Promise.resolve().then(() => __importStar(require("../../ts/genMaciState"))).then(({ genMaciStateFromContract }) => genMaciStateFromContract({
+            provider: signer.provider,
+            address: maciContractAddress,
+            coordinatorKeypair,
+            pollId: BigInt(pollId),
+            fromBlock,
+            blocksPerRequest: blocksPerBatch,
+            endBlock: endBlock || defaultEndBlock,
+            ipfsMessageBackupFiles,
+        }));
+    }
+    /**
+     * Initialize class properties
+     *
+     * @param {IProofGeneratorParams} params - initialization params
+     */
+    constructor({ poll, mp, tally, rapidsnark, maciContractAddress, tallyContractAddress, outputDir, tallyOutputFile, useQuadraticVoting, }) {
+        this.poll = poll;
+        this.maciContractAddress = maciContractAddress;
+        this.tallyContractAddress = tallyContractAddress;
+        this.outputDir = outputDir;
+        this.tallyOutputFile = tallyOutputFile;
+        this.mp = mp;
+        this.tally = tally;
+        this.rapidsnark = rapidsnark;
+        this.useQuadraticVoting = useQuadraticVoting;
+    }
+    /**
+     * Generate message processing proofs
+     *
+     * @returns message processing proofs
+     */
+    async generateMpProofs(options) {
+        (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Generating proofs of message processing...`) });
+        performance.mark("mp-proofs-start");
+        const { messageBatchSize } = this.poll.batchSizes;
+        const numMessages = this.poll.messages.length;
+        let totalMessageBatches = numMessages <= messageBatchSize ? 1 : Math.floor(numMessages / messageBatchSize);
+        if (numMessages > messageBatchSize && numMessages % messageBatchSize > 0) {
+            totalMessageBatches += 1;
+        }
+        try {
+            const inputs = [];
+            // while we have unprocessed messages, process them
+            while (this.poll.hasUnprocessedMessages()) {
+                // process messages in batches
+                const circuitInputs = this.poll.processMessages(BigInt(this.poll.pollId), this.useQuadraticVoting);
+                // generate the proof for this batch
+                inputs.push(circuitInputs);
+                (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Progress: ${this.poll.numBatchesProcessed} / ${totalMessageBatches}`) });
+            }
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Wait until proof generation is finished") });
+            const processZkey = await (0, proofs_1.extractVk)(this.mp.zkey, false);
+            const proofs = await Promise.all(inputs.map((circuitInputs, index) => this.generateProofs(circuitInputs, this.mp, `process_${index}.json`, processZkey).then((data) => {
+                options?.onBatchComplete?.({ current: index, total: totalMessageBatches, proofs: data });
+                return data;
+            }))).then((data) => data.reduce((acc, x) => acc.concat(x), []));
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)("Proof generation is finished") });
+            // cleanup threads
+            await (0, utils_1.cleanThreads)();
+            performance.mark("mp-proofs-end");
+            performance.measure("Generate message processor proofs", "mp-proofs-start", "mp-proofs-end");
+            options?.onComplete?.(proofs);
+            return proofs;
+        }
+        catch (error) {
+            options?.onFail?.(error);
+            throw error;
+        }
+    }
+    /**
+     * Generate tally proofs
+     *
+     * @param networkName - current network name
+     * @param chainId - current chain id
+     * @returns tally proofs
+     */
+    async generateTallyProofs(networkName, chainId, options) {
+        (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Generating proofs of vote tallying...`) });
+        performance.mark("tally-proofs-start");
+        const { tallyBatchSize } = this.poll.batchSizes;
+        const numStateLeaves = this.poll.pollStateLeaves.length;
+        let totalTallyBatches = numStateLeaves <= tallyBatchSize ? 1 : Math.floor(numStateLeaves / tallyBatchSize);
+        if (numStateLeaves > tallyBatchSize && numStateLeaves % tallyBatchSize > 0) {
+            totalTallyBatches += 1;
+        }
+        try {
+            let tallyCircuitInputs;
+            const inputs = [];
+            while (this.poll.hasUntalliedBallots()) {
+                tallyCircuitInputs = (this.useQuadraticVoting
+                    ? this.poll.tallyVotes()
+                    : this.poll.tallyVotesNonQv());
+                inputs.push(tallyCircuitInputs);
+                (0, logger_1.logMagenta)({ text: (0, logger_1.info)(`Progress: ${this.poll.numBatchesTallied} / ${totalTallyBatches}`) });
+            }
+            (0, logger_1.logMagenta)({ text: (0, logger_1.info)("Wait until proof generation is finished") });
+            const tallyVk = await (0, proofs_1.extractVk)(this.tally.zkey, false);
+            const proofs = await Promise.all(inputs.map((circuitInputs, index) => this.generateProofs(circuitInputs, this.tally, `tally_${index}.json`, tallyVk).then((data) => {
+                options?.onBatchComplete?.({ current: index, total: totalTallyBatches, proofs: data });
+                return data;
+            }))).then((data) => data.reduce((acc, x) => acc.concat(x), []));
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)("Proof generation is finished") });
+            // cleanup threads
+            await (0, utils_1.cleanThreads)();
+            // verify the results
+            // Compute newResultsCommitment
+            const newResultsCommitment = (0, crypto_1.genTreeCommitment)(this.poll.tallyResult, BigInt((0, utils_1.asHex)(tallyCircuitInputs.newResultsRootSalt)), this.poll.treeDepths.voteOptionTreeDepth);
+            // compute newSpentVoiceCreditsCommitment
+            const newSpentVoiceCreditsCommitment = (0, crypto_1.hashLeftRight)(this.poll.totalSpentVoiceCredits, BigInt((0, utils_1.asHex)(tallyCircuitInputs.newSpentVoiceCreditSubtotalSalt)));
+            let newPerVOSpentVoiceCreditsCommitment;
+            let newTallyCommitment;
+            // create the tally file data to store for verification later
+            const tallyFileData = {
+                maci: this.maciContractAddress,
+                pollId: this.poll.pollId.toString(),
+                network: networkName,
+                chainId,
+                isQuadratic: Boolean(this.useQuadraticVoting),
+                tallyAddress: this.tallyContractAddress,
+                newTallyCommitment: (0, utils_1.asHex)(tallyCircuitInputs.newTallyCommitment),
+                results: {
+                    tally: this.poll.tallyResult.map((x) => x.toString()),
+                    salt: (0, utils_1.asHex)(tallyCircuitInputs.newResultsRootSalt),
+                    commitment: (0, utils_1.asHex)(newResultsCommitment),
+                },
+                totalSpentVoiceCredits: {
+                    spent: this.poll.totalSpentVoiceCredits.toString(),
+                    salt: (0, utils_1.asHex)(tallyCircuitInputs.newSpentVoiceCreditSubtotalSalt),
+                    commitment: (0, utils_1.asHex)(newSpentVoiceCreditsCommitment),
+                },
+            };
+            if (this.useQuadraticVoting) {
+                // Compute newPerVOSpentVoiceCreditsCommitment
+                newPerVOSpentVoiceCreditsCommitment = (0, crypto_1.genTreeCommitment)(this.poll.perVOSpentVoiceCredits, BigInt((0, utils_1.asHex)(tallyCircuitInputs.newPerVOSpentVoiceCreditsRootSalt)), this.poll.treeDepths.voteOptionTreeDepth);
+                // Compute newTallyCommitment
+                newTallyCommitment = (0, crypto_1.hash3)([
+                    newResultsCommitment,
+                    newSpentVoiceCreditsCommitment,
+                    newPerVOSpentVoiceCreditsCommitment,
+                ]);
+                // update perVOSpentVoiceCredits in the tally file data
+                tallyFileData.perVOSpentVoiceCredits = {
+                    tally: this.poll.perVOSpentVoiceCredits.map((x) => x.toString()),
+                    salt: (0, utils_1.asHex)(tallyCircuitInputs.newPerVOSpentVoiceCreditsRootSalt),
+                    commitment: (0, utils_1.asHex)(newPerVOSpentVoiceCreditsCommitment),
+                };
+            }
+            else {
+                newTallyCommitment = (0, crypto_1.hashLeftRight)(newResultsCommitment, newSpentVoiceCreditsCommitment);
+            }
+            await fs_1.default.promises.writeFile(this.tallyOutputFile, JSON.stringify(tallyFileData, null, 4));
+            (0, logger_1.logGreen)({ text: (0, logger_1.success)(`Tally file:\n${JSON.stringify(tallyFileData, null, 4)}\n`) });
+            // compare the commitments
+            if ((0, utils_1.asHex)(newTallyCommitment) === tallyFileData.newTallyCommitment) {
+                (0, logger_1.logGreen)({ text: (0, logger_1.success)("The tally commitment is correct") });
+            }
+            else {
+                throw new Error("Error: the newTallyCommitment is invalid.");
+            }
+            performance.mark("tally-proofs-end");
+            performance.measure("Generate tally proofs", "tally-proofs-start", "tally-proofs-end");
+            options?.onComplete?.(proofs, tallyFileData);
+            return { proofs, tallyData: tallyFileData };
+        }
+        catch (error) {
+            options?.onFail?.(error);
+            throw error;
+        }
+    }
+    /**
+     * Generic function for proofs generation
+     *
+     * @param {CircuitInputs} circuitInputs - circuit inputs
+     * @param {ICircuitFiles} circuitFiles - circuit files (zkey, witgen, wasm)
+     * @param outputFile - output file
+     * @returns proofs
+     */
+    async generateProofs(circuitInputs, circuitFiles, outputFile, vk) {
+        const proofs = [];
+        const { proof, publicSignals } = circuitFiles.wasm
+            ? await (0, proofs_1.genProofSnarkjs)({
+                inputs: circuitInputs,
+                zkeyPath: circuitFiles.zkey,
+                silent: true,
+                wasmPath: circuitFiles.wasm,
+            })
+            : await (0, proofs_1.genProofRapidSnark)({
+                inputs: circuitInputs,
+                zkeyPath: circuitFiles.zkey,
+                rapidsnarkExePath: this.rapidsnark,
+                witnessExePath: circuitFiles.witgen,
+            });
+        // verify it
+        // eslint-disable-next-line no-await-in-loop
+        const isValid = await (0, proofs_1.verifyProof)(publicSignals, proof, vk, false);
+        if (!isValid) {
+            throw new Error("Error: generated an invalid proof");
+        }
+        proofs.push({
+            circuitInputs,
+            proof,
+            publicInputs: publicSignals,
+        });
+        await fs_1.default.promises.writeFile(path_1.default.resolve(this.outputDir, outputFile), JSON.stringify(proofs[proofs.length - 1], null, 4));
+        return proofs;
+    }
+}
+exports.ProofGenerator = ProofGenerator;
+//# sourceMappingURL=ProofGenerator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ProofGenerator.js","sourceRoot":"","sources":["../../../tasks/helpers/ProofGenerator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+BAA+B;AAC/B,8CAAqF;AACrF,kDAAgF;AAEhF,4CAAoB;AACpB,gDAAwB;AAaxB,4CAAsE;AACtE,4CAA8F;AAC9F,0CAAqD;AAErD;;GAEG;AACH,MAAa,cAAc;IA8CzB;;;;;OAKG;IACH,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EACxB,YAAY,EACZ,YAAY,EACZ,MAAM,EACN,cAAc,EACd,kBAAkB,EAClB,MAAM,EACN,sBAAsB,EACtB,SAAS,EACT,OAAO,EAAE,EAAE,eAAe,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,cAAc,EAAE,GACzD;QACpB,MAAM,iBAAiB,GAAG,YAAE,CAAC,UAAU,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;QAEjE,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,MAAM,YAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,MAAM,YAAE,CAAC,QAAQ;iBAC9B,QAAQ,CAAC,SAAS,CAAC;iBACnB,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAA8B,CAAC,CAAC;YAC1E,MAAM,oBAAoB,GAAG,cAAc,CAAC,SAAS,EAAE,CAAC;YAExD,MAAM,SAAS,GAAG,gBAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAE9C,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC/B,IAAI,CAAC,qBAAqB,CAAC,oBAAoB,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YAEH,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,4FAA4F;QAC5F,MAAM,CAAC,uBAAuB,EAAE,qBAAqB,EAAE,SAAS,EAAE,UAAU,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAChG,YAAY,CAAC,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC,CAAC;YACjH,YAAY;iBACT,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,UAAU,CAAC;iBAC1D,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC,CAAC;YAChD,YAAY,CAAC,gBAAgB,EAAE;YAC/B,YAAY,CAAC,UAAU,EAAE;SAC1B,CAAC,CAAC;QACH,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,CAAC;QACnF,IAAI,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC;QAEpE,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACxC,YAAY;iBACT,WAAW,CAAC,YAAY,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,EAAE,SAAS,CAAC;iBAC9E,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,WAAW,CAAC;SAC5D,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC;QAEzC,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,QAAS,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YAClE,SAAS,GAAG,EAAE,EAAE,WAAW,IAAI,iBAAiB,CAAC;QACnD,CAAC;QAED,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,qCAAqC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;QAE7E,MAAM,mBAAmB,GAAG,MAAM,YAAY,CAAC,UAAU,EAAE,CAAC;QAE5D,OAAO,kDAAO,uBAAuB,IAAE,IAAI,CAAC,CAAC,EAAE,wBAAwB,EAAE,EAAE,EAAE,CAC3E,wBAAwB,CAAC;YACvB,QAAQ,EAAE,MAAM,CAAC,QAAS;YAC1B,OAAO,EAAE,mBAAmB;YAC5B,kBAAkB;YAClB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACtB,SAAS;YACT,gBAAgB,EAAE,cAAc;YAChC,QAAQ,EAAE,QAAQ,IAAI,eAAe;YACrC,sBAAsB;SACvB,CAAC,CACH,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,YAAY,EACV,IAAI,EACJ,EAAE,EACF,KAAK,EACL,UAAU,EACV,mBAAmB,EACnB,oBAAoB,EACpB,SAAS,EACT,eAAe,EACf,kBAAkB,GACI;QACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,oBAAoB,GAAG,oBAAoB,CAAC;QACjD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IAC/C,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,gBAAgB,CAAC,OAAgC;QACrD,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,4CAA4C,CAAC,EAAE,CAAC,CAAC;QACzE,WAAW,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAEpC,MAAM,EAAE,gBAAgB,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;QAClD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAC9C,IAAI,mBAAmB,GAAG,WAAW,IAAI,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,gBAAgB,CAAC,CAAC;QAE3G,IAAI,WAAW,GAAG,gBAAgB,IAAI,WAAW,GAAG,gBAAgB,GAAG,CAAC,EAAE,CAAC;YACzE,mBAAmB,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAoB,EAAE,CAAC;YAEnC,mDAAmD;YACnD,OAAO,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,CAAC;gBAC1C,8BAA8B;gBAC9B,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,CAC7C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EACxB,IAAI,CAAC,kBAAkB,CACI,CAAC;gBAE9B,oCAAoC;gBACpC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAE3B,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,aAAa,IAAI,CAAC,IAAI,CAAC,mBAAmB,MAAM,mBAAmB,EAAE,CAAC,EAAE,CAAC,CAAC;YACpG,CAAC;YAED,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,yCAAyC,CAAC,EAAE,CAAC,CAAC;YAEtE,MAAM,WAAW,GAAG,MAAM,IAAA,kBAAS,EAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,MAAM,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,KAAK,EAAE,EAAE,CAClC,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE,WAAW,KAAK,OAAO,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC9F,OAAO,EAAE,eAAe,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,mBAAmB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzF,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CACH,CACF,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAE7D,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,8BAA8B,CAAC,EAAE,CAAC,CAAC;YAE5D,kBAAkB;YAClB,MAAM,IAAA,oBAAY,GAAE,CAAC;YAErB,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAClC,WAAW,CAAC,OAAO,CAAC,mCAAmC,EAAE,iBAAiB,EAAE,eAAe,CAAC,CAAC;YAE7F,OAAO,EAAE,UAAU,EAAE,CAAC,MAAM,CAAC,CAAC;YAE9B,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,MAAM,EAAE,CAAC,KAAc,CAAC,CAAC;YAElC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,mBAAmB,CACvB,WAAmB,EACnB,OAAgB,EAChB,OAAgC;QAEhC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,uCAAuC,CAAC,EAAE,CAAC,CAAC;QACpE,WAAW,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAEvC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC;QAChD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC;QACxD,IAAI,iBAAiB,GAAG,cAAc,IAAI,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,cAAc,CAAC,CAAC;QAC3G,IAAI,cAAc,GAAG,cAAc,IAAI,cAAc,GAAG,cAAc,GAAG,CAAC,EAAE,CAAC;YAC3E,iBAAiB,IAAI,CAAC,CAAC;QACzB,CAAC;QAED,IAAI,CAAC;YACH,IAAI,kBAAiC,CAAC;YACtC,MAAM,MAAM,GAAoB,EAAE,CAAC;YAEnC,OAAO,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;gBACvC,kBAAkB,GAAG,CAAC,IAAI,CAAC,kBAAkB;oBAC3C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;oBACxB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAA6B,CAAC;gBAE7D,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;gBAEhC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,aAAa,IAAI,CAAC,IAAI,CAAC,iBAAiB,MAAM,iBAAiB,EAAE,CAAC,EAAE,CAAC,CAAC;YAChG,CAAC;YAED,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,IAAA,aAAI,EAAC,yCAAyC,CAAC,EAAE,CAAC,CAAC;YAEtE,MAAM,OAAO,GAAG,MAAM,IAAA,kBAAS,EAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAExD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9B,MAAM,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,KAAK,EAAE,EAAE,CAClC,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,IAAI,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC3F,OAAO,EAAE,eAAe,EAAE,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvF,OAAO,IAAI,CAAC;YACd,CAAC,CAAC,CACH,CACF,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAE7D,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,8BAA8B,CAAC,EAAE,CAAC,CAAC;YAE5D,kBAAkB;YAClB,MAAM,IAAA,oBAAY,GAAE,CAAC;YAErB,qBAAqB;YACrB,+BAA+B;YAC/B,MAAM,oBAAoB,GAAG,IAAA,0BAAiB,EAC5C,IAAI,CAAC,IAAI,CAAC,WAAW,EACrB,MAAM,CAAC,IAAA,aAAK,EAAC,kBAAmB,CAAC,kBAAkC,CAAC,CAAC,EACrE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,mBAAmB,CACzC,CAAC;YAEF,yCAAyC;YACzC,MAAM,8BAA8B,GAAG,IAAA,sBAAa,EAClD,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAChC,MAAM,CAAC,IAAA,aAAK,EAAC,kBAAmB,CAAC,+BAA+C,CAAC,CAAC,CACnF,CAAC;YAEF,IAAI,mCAAuD,CAAC;YAC5D,IAAI,kBAA0B,CAAC;YAE/B,6DAA6D;YAC7D,MAAM,aAAa,GAAc;gBAC/B,IAAI,EAAE,IAAI,CAAC,mBAAmB;gBAC9B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;gBACnC,OAAO,EAAE,WAAW;gBACpB,OAAO;gBACP,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC;gBAC7C,YAAY,EAAE,IAAI,CAAC,oBAAoB;gBACvC,kBAAkB,EAAE,IAAA,aAAK,EAAC,kBAAmB,CAAC,kBAAkC,CAAC;gBACjF,OAAO,EAAE;oBACP,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;oBACrD,IAAI,EAAE,IAAA,aAAK,EAAC,kBAAmB,CAAC,kBAAkC,CAAC;oBACnE,UAAU,EAAE,IAAA,aAAK,EAAC,oBAAoB,CAAC;iBACxC;gBACD,sBAAsB,EAAE;oBACtB,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE;oBAClD,IAAI,EAAE,IAAA,aAAK,EAAC,kBAAmB,CAAC,+BAA+C,CAAC;oBAChF,UAAU,EAAE,IAAA,aAAK,EAAC,8BAA8B,CAAC;iBAClD;aACF,CAAC;YAEF,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,8CAA8C;gBAC9C,mCAAmC,GAAG,IAAA,0BAAiB,EACrD,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAChC,MAAM,CAAC,IAAA,aAAK,EAAC,kBAAmB,CAAC,iCAAiD,CAAC,CAAC,EACpF,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,mBAAmB,CACzC,CAAC;gBAEF,6BAA6B;gBAC7B,kBAAkB,GAAG,IAAA,cAAK,EAAC;oBACzB,oBAAoB;oBACpB,8BAA8B;oBAC9B,mCAAmC;iBACpC,CAAC,CAAC;gBAEH,uDAAuD;gBACvD,aAAa,CAAC,sBAAsB,GAAG;oBACrC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;oBAChE,IAAI,EAAE,IAAA,aAAK,EAAC,kBAAmB,CAAC,iCAAiD,CAAC;oBAClF,UAAU,EAAE,IAAA,aAAK,EAAC,mCAAmC,CAAC;iBACvD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,kBAAkB,GAAG,IAAA,sBAAa,EAAC,oBAAoB,EAAE,8BAA8B,CAAC,CAAC;YAC3F,CAAC;YAED,MAAM,YAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAE1F,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,gBAAgB,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAExF,0BAA0B;YAC1B,IAAI,IAAA,aAAK,EAAC,kBAAkB,CAAC,KAAK,aAAa,CAAC,kBAAkB,EAAE,CAAC;gBACnE,IAAA,iBAAQ,EAAC,EAAE,IAAI,EAAE,IAAA,gBAAO,EAAC,iCAAiC,CAAC,EAAE,CAAC,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,CAAC;YAED,WAAW,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACrC,WAAW,CAAC,OAAO,CAAC,uBAAuB,EAAE,oBAAoB,EAAE,kBAAkB,CAAC,CAAC;YAEvF,OAAO,EAAE,UAAU,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAE7C,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,MAAM,EAAE,CAAC,KAAc,CAAC,CAAC;YAElC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,cAAc,CAC1B,aAA4B,EAC5B,YAA2B,EAC3B,UAAkB,EAClB,EAAmB;QAEnB,MAAM,MAAM,GAAY,EAAE,CAAC;QAE3B,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,YAAY,CAAC,IAAI;YAChD,CAAC,CAAC,MAAM,IAAA,wBAAe,EAAC;gBACpB,MAAM,EAAE,aAAa;gBACrB,QAAQ,EAAE,YAAY,CAAC,IAAI;gBAC3B,MAAM,EAAE,IAAI;gBACZ,QAAQ,EAAE,YAAY,CAAC,IAAI;aAC5B,CAAC;YACJ,CAAC,CAAC,MAAM,IAAA,2BAAkB,EAAC;gBACvB,MAAM,EAAE,aAAa;gBACrB,QAAQ,EAAE,YAAY,CAAC,IAAI;gBAC3B,iBAAiB,EAAE,IAAI,CAAC,UAAU;gBAClC,cAAc,EAAE,YAAY,CAAC,MAAM;aACpC,CAAC,CAAC;QAEP,YAAY;QACZ,4CAA4C;QAC5C,MAAM,OAAO,GAAG,MAAM,IAAA,oBAAW,EAAC,aAAa,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAEnE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,CAAC,IAAI,CAAC;YACV,aAAa;YACb,KAAK;YACL,YAAY,EAAE,aAAa;SAC5B,CAAC,CAAC;QAEH,MAAM,YAAE,CAAC,QAAQ,CAAC,SAAS,CACzB,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,EACxC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CACnD,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAzZD,wCAyZC"}

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

@@ -0,0 +1,73 @@
+import type { Proof } from "../../ts/types";
+import { IProverParams, TallyData } from "./types";
+/**
+ * Prover class is designed to prove message processing and tally proofs on-chain.
+ */
+export declare class Prover {
+    /**
+     * Poll contract typechain wrapper
+     */
+    private pollContract;
+    /**
+     * MessageProcessor contract typechain wrapper
+     */
+    private mpContract;
+    /**
+     * MACI contract typechain wrapper
+     */
+    private maciContract;
+    /**
+     * VkRegistry contract typechain wrapper
+     */
+    private vkRegistryContract;
+    /**
+     * Verifier contract typechain wrapper
+     */
+    private verifierContract;
+    /**
+     * Tally contract typechain wrapper
+     */
+    private tallyContract;
+    /**
+     * Initialize class properties
+     *
+     * @param {IProverParams} params - constructor params
+     */
+    constructor({ pollContract, mpContract, maciContract, vkRegistryContract, verifierContract, tallyContract, }: IProverParams);
+    /**
+     * Prove message processing on-chain
+     *
+     * @param proofs - proofs
+     */
+    proveMessageProcessing(proofs: Proof[]): Promise<void>;
+    /**
+     * Prove tally on-chain
+     *
+     * @param proofs tally proofs
+     */
+    proveTally(proofs: Proof[]): Promise<void>;
+    /**
+     * Submit tally results on chain
+     *
+     * @param tallyData - tally data
+     * @param recipients - number of recipients
+     */
+    submitResults(tallyData: TallyData, recipients?: number): Promise<void>;
+    /**
+     * Validate commitment
+     *
+     * @param commitment - off-chain commitment
+     * @param commitmentOnChain - on-chain commitment
+     * @throws error if commitments don't match
+     */
+    private validateCommitment;
+    /**
+     * 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
+     */
+    private validatePublicInput;
+}
+//# sourceMappingURL=Prover.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"Prover.d.ts","sourceRoot":"","sources":["../../../tasks/helpers/Prover.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAuB,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAOjE,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEnD;;GAEG;AACH,qBAAa,MAAM;IACjB;;OAEG;IACH,OAAO,CAAC,YAAY,CAAO;IAE3B;;OAEG;IACH,OAAO,CAAC,UAAU,CAAmB;IAErC;;OAEG;IACH,OAAO,CAAC,YAAY,CAAO;IAE3B;;OAEG;IACH,OAAO,CAAC,kBAAkB,CAAa;IAEvC;;OAEG;IACH,OAAO,CAAC,gBAAgB,CAAW;IAEnC;;OAEG;IACH,OAAO,CAAC,aAAa,CAAQ;IAE7B;;;;OAIG;gBACS,EACV,YAAY,EACZ,UAAU,EACV,YAAY,EACZ,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACd,EAAE,aAAa;IAShB;;;;OAIG;IACG,sBAAsB,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoJ5D;;;;OAIG;IACG,UAAU,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA4FhD;;;;;OAKG;IACG,aAAa,CAAC,SAAS,EAAE,SAAS,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAgC7E;;;;;;OAMG;IACH,OAAO,CAAC,kBAAkB;IAM1B;;;;;;OAMG;IACH,OAAO,CAAC,mBAAmB;CAK5B"}

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(currentSbCommitment, currentSbCommitmentOnChain) {
+        if (currentSbCommitmentOnChain.toString() !== currentSbCommitment.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