@maci-protocol/coordinator 0.0.0-ci.00107eb

package/build/ts/proof/proof.gateway.d.ts

@@ -0,0 +1,35 @@
+import type { Server } from "socket.io";
+import { GenerateProofDto } from "./dto";
+import { ProofGeneratorService } from "./proof.service";
+/**
+ * ProofGateway is responsible for websockets integration between client and ProofGeneratorService.
+ */
+export declare class ProofGateway {
+    private readonly proofGeneratorService;
+    /**
+     * Logger
+     */
+    private readonly logger;
+    /**
+     * Websocket server
+     */
+    server: Server;
+    /**
+     * Initialize ProofGateway
+     *
+     * @param proofGeneratorService - proof generator service
+     */
+    constructor(proofGeneratorService: ProofGeneratorService);
+    /**
+     * Generate proofs api method.
+     * Events:
+     * 1. EProofGenerationEvents.START - trigger method call
+     * 2. EProofGenerationEvents.PROGRESS - returns generated proofs with batch info
+     * 3. EProofGenerationEvents.FINISH - returns generated proofs and tally data when available
+     * 4. EProofGenerationEvents.ERROR - triggered when exception is thrown
+     *
+     * @param args - generate proof dto
+     */
+    generate(data: GenerateProofDto): Promise<void>;
+}
+//# sourceMappingURL=proof.gateway.d.ts.map

package/build/ts/proof/proof.gateway.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.gateway.d.ts","sourceRoot":"","sources":["../../../ts/proof/proof.gateway.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAIxC,OAAO,EAAE,gBAAgB,EAAE,MAAM,OAAO,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAGxD;;GAEG;AACH,qBAMa,YAAY;IAiBX,OAAO,CAAC,QAAQ,CAAC,qBAAqB;IAhBlD;;OAEG;IACH,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiC;IAExD;;OAEG;IAEH,MAAM,EAAG,MAAM,CAAC;IAEhB;;;;OAIG;gBAC0B,qBAAqB,EAAE,qBAAqB;IAEzE;;;;;;;;;OASG;IAUG,QAAQ,CAEZ,IAAI,EAAE,gBAAgB,GACrB,OAAO,CAAC,IAAI,CAAC;CAcjB"}

package/build/ts/proof/proof.gateway.js

@@ -0,0 +1,93 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var ProofGateway_1;
+import { Logger, UseGuards, UsePipes, ValidationPipe } from "@nestjs/common";
+import { MessageBody, SubscribeMessage, WebSocketGateway, WebSocketServer, WsException } from "@nestjs/websockets";
+import { AccountSignatureGuard } from "../auth/AccountSignatureGuard.service";
+import { GenerateProofDto } from "./dto";
+import { ProofGeneratorService } from "./proof.service";
+import { EProofGenerationEvents } from "./types";
+/**
+ * ProofGateway is responsible for websockets integration between client and ProofGeneratorService.
+ */
+let ProofGateway = ProofGateway_1 = class ProofGateway {
+    proofGeneratorService;
+    /**
+     * Logger
+     */
+    logger = new Logger(ProofGateway_1.name);
+    /**
+     * Websocket server
+     */
+    server;
+    /**
+     * Initialize ProofGateway
+     *
+     * @param proofGeneratorService - proof generator service
+     */
+    constructor(proofGeneratorService) {
+        this.proofGeneratorService = proofGeneratorService;
+    }
+    /**
+     * Generate proofs api method.
+     * Events:
+     * 1. EProofGenerationEvents.START - trigger method call
+     * 2. EProofGenerationEvents.PROGRESS - returns generated proofs with batch info
+     * 3. EProofGenerationEvents.FINISH - returns generated proofs and tally data when available
+     * 4. EProofGenerationEvents.ERROR - triggered when exception is thrown
+     *
+     * @param args - generate proof dto
+     */
+    async generate(data) {
+        await this.proofGeneratorService.generate(data, {
+            onBatchComplete: (result) => {
+                this.server.emit(EProofGenerationEvents.PROGRESS, result);
+            },
+            onComplete: (proofs, tallyData) => {
+                this.server.emit(EProofGenerationEvents.FINISH, { proofs, tallyData });
+            },
+            onFail: (error) => {
+                this.logger.error(`Error:`, error);
+                this.server.emit(EProofGenerationEvents.ERROR, { message: error.message });
+            },
+        });
+    }
+};
+__decorate([
+    WebSocketServer(),
+    __metadata("design:type", Function)
+], ProofGateway.prototype, "server", void 0);
+__decorate([
+    SubscribeMessage(EProofGenerationEvents.START),
+    UsePipes(new ValidationPipe({
+        transform: true,
+        exceptionFactory(validationErrors) {
+            return new WsException(validationErrors);
+        },
+    })),
+    __param(0, MessageBody()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [GenerateProofDto]),
+    __metadata("design:returntype", Promise)
+], ProofGateway.prototype, "generate", null);
+ProofGateway = ProofGateway_1 = __decorate([
+    WebSocketGateway({
+        cors: {
+            origin: process.env.COORDINATOR_ALLOWED_ORIGINS?.split(","),
+        },
+    }),
+    UseGuards(AccountSignatureGuard),
+    __metadata("design:paramtypes", [ProofGeneratorService])
+], ProofGateway);
+export { ProofGateway };
+//# sourceMappingURL=proof.gateway.js.map

package/build/ts/proof/proof.gateway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.gateway.js","sourceRoot":"","sources":["../../../ts/proof/proof.gateway.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAC7E,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAKnH,OAAO,EAAE,qBAAqB,EAAE,MAAM,uCAAuC,CAAC;AAE9E,OAAO,EAAE,gBAAgB,EAAE,MAAM,OAAO,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AACxD,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAEjD;;GAEG;AAOI,IAAM,YAAY,oBAAlB,MAAM,YAAY;IAiBM;IAhB7B;;OAEG;IACc,MAAM,GAAG,IAAI,MAAM,CAAC,cAAY,CAAC,IAAI,CAAC,CAAC;IAExD;;OAEG;IAEH,MAAM,CAAU;IAEhB;;;;OAIG;IACH,YAA6B,qBAA4C;QAA5C,0BAAqB,GAArB,qBAAqB,CAAuB;IAAG,CAAC;IAE7E;;;;;;;;;OASG;IAUG,AAAN,KAAK,CAAC,QAAQ,CAEZ,IAAsB;QAEtB,MAAM,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,IAAI,EAAE;YAC9C,eAAe,EAAE,CAAC,MAAgC,EAAE,EAAE;gBACpD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC5D,CAAC;YACD,UAAU,EAAE,CAAC,MAAgB,EAAE,SAAsB,EAAE,EAAE;gBACvD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACzE,CAAC;YACD,MAAM,EAAE,CAAC,KAAY,EAAE,EAAE;gBACvB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7E,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA7CC;IADC,eAAe,EAAE;;4CACF;AA4BV;IATL,gBAAgB,CAAC,sBAAsB,CAAC,KAAK,CAAC;IAC9C,QAAQ,CACP,IAAI,cAAc,CAAC;QACjB,SAAS,EAAE,IAAI;QACf,gBAAgB,CAAC,gBAAgB;YAC/B,OAAO,IAAI,WAAW,CAAC,gBAAgB,CAAC,CAAC;QAC3C,CAAC;KACF,CAAC,CACH;IAEE,WAAA,WAAW,EAAE,CAAA;;qCACR,gBAAgB;;4CAcvB;AAtDU,YAAY;IANxB,gBAAgB,CAAC;QAChB,IAAI,EAAE;YACJ,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,2BAA2B,EAAE,KAAK,CAAC,GAAG,CAAC;SAC5D;KACF,CAAC;IACD,SAAS,CAAC,qBAAqB,CAAC;qCAkBqB,qBAAqB;GAjB9D,YAAY,CAuDxB"}

package/build/ts/proof/proof.module.d.ts

@@ -0,0 +1,3 @@
+export declare class ProofModule {
+}
+//# sourceMappingURL=proof.module.d.ts.map

package/build/ts/proof/proof.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.module.d.ts","sourceRoot":"","sources":["../../../ts/proof/proof.module.ts"],"names":[],"mappings":"AAUA,qBAMa,WAAW;CAAG"}

package/build/ts/proof/proof.module.js

@@ -0,0 +1,25 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+import { Module } from "@nestjs/common";
+import { CryptoModule } from "../crypto/crypto.module";
+import { FileModule } from "../file/file.module";
+import { SessionKeysModule } from "../sessionKeys/sessionKeys.module";
+import { ProofController } from "./proof.controller";
+import { ProofGateway } from "./proof.gateway";
+import { ProofGeneratorService } from "./proof.service";
+let ProofModule = class ProofModule {
+};
+ProofModule = __decorate([
+    Module({
+        imports: [FileModule, CryptoModule, SessionKeysModule],
+        controllers: [ProofController],
+        providers: [ProofGeneratorService, ProofGateway],
+        exports: [ProofGeneratorService],
+    })
+], ProofModule);
+export { ProofModule };
+//# sourceMappingURL=proof.module.js.map

package/build/ts/proof/proof.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.module.js","sourceRoot":"","sources":["../../../ts/proof/proof.module.ts"],"names":[],"mappings":";;;;;;AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAExC,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AACvD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AAEtE,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAQjD,IAAM,WAAW,GAAjB,MAAM,WAAW;CAAG,CAAA;AAAd,WAAW;IANvB,MAAM,CAAC;QACN,OAAO,EAAE,CAAC,UAAU,EAAE,YAAY,EAAE,iBAAiB,CAAC;QACtD,WAAW,EAAE,CAAC,eAAe,CAAC;QAC9B,SAAS,EAAE,CAAC,qBAAqB,EAAE,YAAY,CAAC;QAChD,OAAO,EAAE,CAAC,qBAAqB,CAAC;KACjC,CAAC;GACW,WAAW,CAAG"}

package/build/ts/proof/proof.service.d.ts

@@ -0,0 +1,52 @@
+import { type IGenerateProofsOptions } from "@maci-protocol/sdk";
+import { IProof, ITallyData } from "@maci-protocol/sdk";
+import type { IGenerateArgs, IGenerateData, IMergeArgs, ISubmitProofsArgs } from "./types";
+import { FileService } from "../file/file.service";
+import { SessionKeysService } from "../sessionKeys/sessionKeys.service";
+/**
+ * ProofGeneratorService is responsible for generating message processing and tally proofs.
+ */
+export declare class ProofGeneratorService {
+    private readonly fileService;
+    private readonly sessionKeysService;
+    /**
+     * Deployment helper
+     */
+    private readonly deployment;
+    /**
+     * Logger
+     */
+    private readonly logger;
+    /**
+     * Proof generator initialization
+     */
+    constructor(fileService: FileService, sessionKeysService: SessionKeysService);
+    /**
+     * Read and parse proofs
+     * @param folder - folder path to read proofs from
+     * @param type - type of proofs to read (tally or process)
+     * @returns proofs
+     */
+    readProofs(folder: string, type: "tally" | "process"): Promise<IProof[]>;
+    /**
+     * Generate proofs for message processing and tally
+     *
+     * @param args - generate proofs arguments
+     * @returns - generated proofs for message processing and tally
+     */
+    generate({ approval, sessionKeyAddress, chain, poll, maciContractAddress, mode, startBlock, endBlock, blocksPerBatch, useWasm, }: IGenerateArgs, options?: IGenerateProofsOptions): Promise<IGenerateData>;
+    /**
+     * Merge state and message trees
+     *
+     * @param args - merge arguments
+     * @returns whether the proofs were successfully merged
+     */
+    merge({ maciContractAddress, pollId, approval, sessionKeyAddress, chain }: IMergeArgs): Promise<boolean>;
+    /**
+     * Submit proofs on-chain
+     *
+     * @param args - submit proofs on-chain arguments
+     */
+    submit({ maciContractAddress, pollId, sessionKeyAddress, approval, chain, }: ISubmitProofsArgs): Promise<ITallyData>;
+}
+//# sourceMappingURL=proof.service.d.ts.map

package/build/ts/proof/proof.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.service.d.ts","sourceRoot":"","sources":["../../../ts/proof/proof.service.ts"],"names":[],"mappings":"AACA,OAAO,EAIL,KAAK,sBAAsB,EAI5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,UAAU,EAAgC,MAAM,oBAAoB,CAAC;AAOtF,OAAO,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAI3F,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE;;GAEG;AACH,qBACa,qBAAqB;IAe9B,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IAfrC;;OAEG;IACH,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IAExC;;OAEG;IACH,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAEhC;;OAEG;gBAEgB,WAAW,EAAE,WAAW,EACxB,kBAAkB,EAAE,kBAAkB;IAOzD;;;;;OAKG;IACG,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAY9E;;;;;OAKG;IACG,QAAQ,CACZ,EACE,QAAQ,EACR,iBAAiB,EACjB,KAAK,EACL,IAAI,EACJ,mBAAmB,EACnB,IAAI,EACJ,UAAU,EACV,QAAQ,EACR,cAAc,EACd,OAAO,GACR,EAAE,aAAa,EAChB,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,aAAa,CAAC;IAuEzB;;;;;OAKG;IACG,KAAK,CAAC,EAAE,mBAAmB,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;IAY9G;;;;OAIG;IACG,MAAM,CAAC,EACX,mBAAmB,EACnB,MAAM,EACN,iBAAiB,EACjB,QAAQ,EACR,KAAK,GACN,EAAE,iBAAiB,GAAG,OAAO,CAAC,UAAU,CAAC;CAiB3C"}

package/build/ts/proof/proof.service.js

@@ -0,0 +1,164 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var ProofGeneratorService_1;
+import { PublicKey } from "@maci-protocol/domainobjs";
+import { Deployment, EContracts, getPoll, mergeSignups, EMode, } from "@maci-protocol/sdk";
+import { generateProofs, proveOnChain } from "@maci-protocol/sdk";
+import { Logger, Injectable } from "@nestjs/common";
+import hre from "hardhat";
+import fs from "fs";
+import path from "path";
+import { ErrorCodes } from "../common";
+import { getCoordinatorKeypair } from "../common/coordinatorKeypair";
+import { FileService } from "../file/file.service";
+import { SessionKeysService } from "../sessionKeys/sessionKeys.service";
+/**
+ * ProofGeneratorService is responsible for generating message processing and tally proofs.
+ */
+let ProofGeneratorService = ProofGeneratorService_1 = class ProofGeneratorService {
+    fileService;
+    sessionKeysService;
+    /**
+     * Deployment helper
+     */
+    deployment;
+    /**
+     * Logger
+     */
+    logger;
+    /**
+     * Proof generator initialization
+     */
+    constructor(fileService, sessionKeysService) {
+        this.fileService = fileService;
+        this.sessionKeysService = sessionKeysService;
+        this.deployment = Deployment.getInstance({ hre });
+        this.deployment.setHre(hre);
+        this.logger = new Logger(ProofGeneratorService_1.name);
+    }
+    /**
+     * Read and parse proofs
+     * @param folder - folder path to read proofs from
+     * @param type - type of proofs to read (tally or process)
+     * @returns proofs
+     */
+    async readProofs(folder, type) {
+        const files = await fs.promises.readdir(folder);
+        return Promise.all(files
+            .filter((f) => f.startsWith(`${type}_`) && f.endsWith(".json"))
+            .sort()
+            .map(async (file) => fs.promises.readFile(`${folder}/${file}`, "utf8").then((result) => JSON.parse(result))));
+    }
+    /**
+     * Generate proofs for message processing and tally
+     *
+     * @param args - generate proofs arguments
+     * @returns - generated proofs for message processing and tally
+     */
+    async generate({ approval, sessionKeyAddress, chain, poll, maciContractAddress, mode, startBlock, endBlock, blocksPerBatch, useWasm, }, options) {
+        try {
+            const signer = await this.sessionKeysService.getCoordinatorSigner(chain, sessionKeyAddress, approval);
+            const pollData = await getPoll({
+                maciAddress: maciContractAddress,
+                pollId: poll,
+                signer,
+            });
+            const pollContract = await this.deployment.getContract({
+                name: EContracts.Poll,
+                address: pollData.address,
+            });
+            const publicKeyOnChain = await pollContract.coordinatorPublicKey();
+            const coordinatorPublicKeyOnChain = new PublicKey([
+                BigInt(publicKeyOnChain.x.toString()),
+                BigInt(publicKeyOnChain.y.toString()),
+            ]);
+            const coordinatorKeypair = getCoordinatorKeypair();
+            if (!coordinatorKeypair.publicKey.equals(coordinatorPublicKeyOnChain)) {
+                this.logger.error(`Error: ${ErrorCodes.PRIVATE_KEY_MISMATCH}, wrong private key`);
+                throw new Error(ErrorCodes.PRIVATE_KEY_MISMATCH.toString());
+            }
+            // There are only QV and Non-QV modes available for tally circuit
+            const tally = this.fileService.getZkeyFilePaths(process.env.COORDINATOR_TALLY_ZKEY_NAME, 
+            // if FULL use NON_QV because there are only VoteTallyQV and VoteTallyNonQV zkeys
+            mode === EMode.FULL ? EMode.NON_QV : mode);
+            const messageProcessor = this.fileService.getZkeyFilePaths(process.env.COORDINATOR_MESSAGE_PROCESS_ZKEY_NAME, mode);
+            const { processProofs, tallyProofs, tallyData } = await generateProofs({
+                outputDir: path.resolve("./proofs"),
+                coordinatorPrivateKey: coordinatorKeypair.privateKey.serialize(),
+                signer,
+                maciAddress: maciContractAddress,
+                pollId: BigInt(poll),
+                startBlock,
+                endBlock,
+                blocksPerBatch,
+                useWasm,
+                rapidsnark: process.env.COORDINATOR_RAPIDSNARK_EXE,
+                mode,
+                voteTallyZkey: tally.zkey,
+                voteTallyWitnessGenerator: tally.witnessGenerator,
+                voteTallyWasm: tally.wasm,
+                messageProcessorZkey: messageProcessor.zkey,
+                messageProcessorWitnessGenerator: messageProcessor.witnessGenerator,
+                messageProcessorWasm: messageProcessor.wasm,
+                tallyFile: path.resolve("./tally.json"),
+            });
+            return {
+                processProofs,
+                tallyProofs,
+                tallyData,
+            };
+        }
+        catch (error) {
+            options?.onFail?.(error);
+            throw error;
+        }
+    }
+    /**
+     * Merge state and message trees
+     *
+     * @param args - merge arguments
+     * @returns whether the proofs were successfully merged
+     */
+    async merge({ maciContractAddress, pollId, approval, sessionKeyAddress, chain }) {
+        const signer = await this.sessionKeysService.getCoordinatorSigner(chain, sessionKeyAddress, approval);
+        await mergeSignups({
+            maciAddress: maciContractAddress,
+            pollId: BigInt(pollId),
+            signer,
+        });
+        return true;
+    }
+    /**
+     * Submit proofs on-chain
+     *
+     * @param args - submit proofs on-chain arguments
+     */
+    async submit({ maciContractAddress, pollId, sessionKeyAddress, approval, chain, }) {
+        const signer = await this.sessionKeysService.getCoordinatorSigner(chain, sessionKeyAddress, approval);
+        const tallyData = await proveOnChain({
+            pollId: BigInt(pollId),
+            maciAddress: maciContractAddress,
+            proofDir: "./proofs",
+            tallyFile: "./tally.json",
+            signer,
+        });
+        if (!tallyData) {
+            throw new Error("Tally data is undefined");
+        }
+        return tallyData;
+    }
+};
+ProofGeneratorService = ProofGeneratorService_1 = __decorate([
+    Injectable(),
+    __metadata("design:paramtypes", [FileService,
+        SessionKeysService])
+], ProofGeneratorService);
+export { ProofGeneratorService };
+//# sourceMappingURL=proof.service.js.map

package/build/ts/proof/proof.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proof.service.js","sourceRoot":"","sources":["../../../ts/proof/proof.service.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EACL,UAAU,EACV,UAAU,EAGV,OAAO,EACP,YAAY,EACZ,KAAK,GACN,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAsB,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACtF,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,GAAG,MAAM,SAAS,CAAC;AAE1B,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAIxB,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAExE;;GAEG;AAEI,IAAM,qBAAqB,6BAA3B,MAAM,qBAAqB;IAeb;IACA;IAfnB;;OAEG;IACc,UAAU,CAAa;IAExC;;OAEG;IACc,MAAM,CAAS;IAEhC;;OAEG;IACH,YACmB,WAAwB,EACxB,kBAAsC;QADtC,gBAAW,GAAX,WAAW,CAAa;QACxB,uBAAkB,GAAlB,kBAAkB,CAAoB;QAEvD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,WAAW,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,uBAAqB,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,IAAyB;QACxD,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChD,OAAO,OAAO,CAAC,GAAG,CAChB,KAAK;aACF,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;aAC9D,IAAI,EAAE;aACN,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAClB,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,MAAM,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAW,CAAC,CACjG,CACJ,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,QAAQ,CACZ,EACE,QAAQ,EACR,iBAAiB,EACjB,KAAK,EACL,IAAI,EACJ,mBAAmB,EACnB,IAAI,EACJ,UAAU,EACV,QAAQ,EACR,cAAc,EACd,OAAO,GACO,EAChB,OAAgC;QAEhC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;YAEtG,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC;gBAC7B,WAAW,EAAE,mBAAmB;gBAChC,MAAM,EAAE,IAAI;gBACZ,MAAM;aACP,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAO;gBAC3D,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;YAEH,MAAM,gBAAgB,GAAG,MAAM,YAAY,CAAC,oBAAoB,EAAE,CAAC;YACnE,MAAM,2BAA2B,GAAG,IAAI,SAAS,CAAC;gBAChD,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;gBACrC,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;aACtC,CAAC,CAAC;YAEH,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;YAEnD,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,MAAM,CAAC,2BAA2B,CAAC,EAAE,CAAC;gBACtE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,UAAU,CAAC,oBAAoB,qBAAqB,CAAC,CAAC;gBAClF,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC9D,CAAC;YAED,iEAAiE;YACjE,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAC7C,OAAO,CAAC,GAAG,CAAC,2BAA4B;YACxC,iFAAiF;YACjF,IAAI,KAAK,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAC1C,CAAC;YACF,MAAM,gBAAgB,GAAG,IAAI,CAAC,WAAW,CAAC,gBAAgB,CACxD,OAAO,CAAC,GAAG,CAAC,qCAAsC,EAClD,IAAI,CACL,CAAC;YAEF,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,MAAM,cAAc,CAAC;gBACrE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBACnC,qBAAqB,EAAE,kBAAkB,CAAC,UAAU,CAAC,SAAS,EAAE;gBAChE,MAAM;gBACN,WAAW,EAAE,mBAAmB;gBAChC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC;gBACpB,UAAU;gBACV,QAAQ;gBACR,cAAc;gBACd,OAAO;gBACP,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B;gBAClD,IAAI;gBACJ,aAAa,EAAE,KAAK,CAAC,IAAI;gBACzB,yBAAyB,EAAE,KAAK,CAAC,gBAAgB;gBACjD,aAAa,EAAE,KAAK,CAAC,IAAI;gBACzB,oBAAoB,EAAE,gBAAgB,CAAC,IAAI;gBAC3C,gCAAgC,EAAE,gBAAgB,CAAC,gBAAgB;gBACnE,oBAAoB,EAAE,gBAAgB,CAAC,IAAI;gBAC3C,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC;aACxC,CAAC,CAAC;YAEH,OAAO;gBACL,aAAa;gBACb,WAAW;gBACX,SAAS;aACV,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,MAAM,EAAE,CAAC,KAAc,CAAC,CAAC;YAClC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CAAC,EAAE,mBAAmB,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,KAAK,EAAc;QACzF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;QAEtG,MAAM,YAAY,CAAC;YACjB,WAAW,EAAE,mBAAmB;YAChC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACtB,MAAM;SACP,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,MAAM,CAAC,EACX,mBAAmB,EACnB,MAAM,EACN,iBAAiB,EACjB,QAAQ,EACR,KAAK,GACa;QAClB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC,CAAC;QAEtG,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC;YACnC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC;YACtB,WAAW,EAAE,mBAAmB;YAChC,QAAQ,EAAE,UAAU;YACpB,SAAS,EAAE,cAAc;YACzB,MAAM;SACP,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CACF,CAAA;AAlLY,qBAAqB;IADjC,UAAU,EAAE;qCAgBqB,WAAW;QACJ,kBAAkB;GAhB9C,qBAAqB,CAkLjC"}

package/build/ts/proof/types.d.ts

@@ -0,0 +1,144 @@
+import type { ITallyData, IProof, EMode, ESupportedChains } from "@maci-protocol/sdk";
+import type { Hex } from "viem";
+/**
+ * WS events for proof generation
+ */
+export declare enum EProofGenerationEvents {
+    START = "start-generation",
+    PROGRESS = "progress-generation",
+    FINISH = "finish-generation",
+    ERROR = "exception"
+}
+export interface IReadProofsResults {
+    /**
+     * Process proofs list
+     */
+    process: IProof[];
+    /**
+     * Tally proofs list
+     */
+    tally: IProof[];
+}
+/**
+ * Interface that represents generate proofs arguments
+ */
+export interface IGenerateArgs {
+    /**
+     * Approval for the session key
+     */
+    approval?: string;
+    /**
+     * Session key address
+     */
+    sessionKeyAddress?: Hex;
+    /**
+     * Chain
+     */
+    chain: ESupportedChains;
+    /**
+     * Poll id
+     */
+    poll: number;
+    /**
+     * Maci contract address
+     */
+    maciContractAddress: string;
+    /**
+     * Whether to use Qv or NonQv
+     */
+    mode: EMode;
+    /**
+     * Start block for event processing
+     */
+    startBlock?: number;
+    /**
+     * End block for event processing
+     */
+    endBlock?: number;
+    /**
+     * Blocks per batch for event processing
+     */
+    blocksPerBatch?: number;
+    useWasm?: boolean;
+}
+/**
+ * Interface that represents generated proofs data
+ */
+export interface IGenerateData {
+    /**
+     * Message processing proofs
+     */
+    processProofs: IProof[];
+    /**
+     * Tally proofs
+     */
+    tallyProofs: IProof[];
+    /**
+     * TallyData
+     */
+    tallyData: ITallyData;
+}
+/**
+ * Interface that represents zkey filepaths
+ */
+export interface IGetZkeyFilesData {
+    /**
+     * Zkey filepath
+     */
+    zkey: string;
+    /**
+     * Wasm filepath
+     */
+    wasm: string;
+}
+/**
+ * Merge arguments
+ */
+export interface IMergeArgs {
+    /**
+     * MACI contract address
+     */
+    maciContractAddress: string;
+    /**
+     * Poll ID
+     */
+    pollId: number;
+    /**
+     * Approval for the session key
+     */
+    approval?: string;
+    /**
+     * Session key address
+     */
+    sessionKeyAddress?: Hex;
+    /**
+     * Chain
+     */
+    chain: ESupportedChains;
+}
+/**
+ * Submit proofs on-chain arguments
+ */
+export interface ISubmitProofsArgs {
+    /**
+     * MACI contract address
+     */
+    maciContractAddress: string;
+    /**
+     * Poll ID
+     */
+    pollId: number;
+    /**
+     * Approval for the session key
+     */
+    approval?: string;
+    /**
+     * Session key address
+     */
+    sessionKeyAddress?: Hex;
+    /**
+     * Chain
+     */
+    chain: ESupportedChains;
+}
+//# sourceMappingURL=types.d.ts.map

package/build/ts/proof/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../ts/proof/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtF,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAEhC;;GAEG;AACH,oBAAY,sBAAsB;IAChC,KAAK,qBAAqB;IAC1B,QAAQ,wBAAwB;IAChC,MAAM,sBAAsB;IAC5B,KAAK,cAAc;CACpB;AAKD,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,OAAO,EAAE,MAAM,EAAE,CAAC;IAElB;;OAEG;IACH,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,iBAAiB,CAAC,EAAE,GAAG,CAAC;IAExB;;OAEG;IACH,KAAK,EAAE,gBAAgB,CAAC;IAExB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,mBAAmB,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,IAAI,EAAE,KAAK,CAAC;IAEZ;;OAEG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,cAAc,CAAC,EAAE,MAAM,CAAC;IAKxB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B;;OAEG;IACH,aAAa,EAAE,MAAM,EAAE,CAAC;IAExB;;OAEG;IACH,WAAW,EAAE,MAAM,EAAE,CAAC;IAEtB;;OAEG;IACH,SAAS,EAAE,UAAU,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB;;OAEG;IACH,mBAAmB,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,iBAAiB,CAAC,EAAE,GAAG,CAAC;IAExB;;OAEG;IACH,KAAK,EAAE,gBAAgB,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,mBAAmB,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;IAEf;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;OAEG;IACH,iBAAiB,CAAC,EAAE,GAAG,CAAC;IAExB;;OAEG;IACH,KAAK,EAAE,gBAAgB,CAAC;CACzB"}

package/build/ts/proof/types.js

@@ -0,0 +1,11 @@
+/**
+ * WS events for proof generation
+ */
+export var EProofGenerationEvents;
+(function (EProofGenerationEvents) {
+    EProofGenerationEvents["START"] = "start-generation";
+    EProofGenerationEvents["PROGRESS"] = "progress-generation";
+    EProofGenerationEvents["FINISH"] = "finish-generation";
+    EProofGenerationEvents["ERROR"] = "exception";
+})(EProofGenerationEvents || (EProofGenerationEvents = {}));
+//# sourceMappingURL=types.js.map

package/build/ts/proof/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../ts/proof/types.ts"],"names":[],"mappings":"AAGA;;GAEG;AACH,MAAM,CAAN,IAAY,sBAKX;AALD,WAAY,sBAAsB;IAChC,oDAA0B,CAAA;IAC1B,0DAAgC,CAAA;IAChC,sDAA4B,CAAA;IAC5B,6CAAmB,CAAA;AACrB,CAAC,EALW,sBAAsB,KAAtB,sBAAsB,QAKjC"}

package/build/ts/redis/__tests__/redis.service.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=redis.service.test.d.ts.map

package/build/ts/redis/__tests__/redis.service.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis.service.test.d.ts","sourceRoot":"","sources":["../../../../ts/redis/__tests__/redis.service.test.ts"],"names":[],"mappings":""}