feat(sdk): return processProofs, tallyProofs and tallyData in generat…

…eProofs (#2193)

packages/cli/ts/commands/genProofs.ts

@@ -102,7 +102,7 @@ export const genProofsCommand = async ({
     logError("Invalid MACI private key");
   }
 
-  const tallyData = await generateProofs({
+  const { tallyData } = await generateProofs({
     networkName: network?.name || "",
     chainId: network?.chainId.toString() || "0",
     maciContractAddress,

packages/sdk/ts/proof/generate.ts

@@ -3,9 +3,7 @@ import { Keypair, PrivKey } from "maci-domainobjs";
 
 import fs from "fs";
 
-import type { ITallyData } from "../tally";
-
-import { IGenerateProofsArgs } from "./types";
+import { IGenerateProofsArgs, IGenerateProofsData } from "./types";
 
 /**
  * Generate proofs for the message processing and tally calculations
@@ -35,7 +33,7 @@ export const generateProofs = async ({
   processWitgen,
   processWasm,
   tallyFile,
-}: IGenerateProofsArgs): Promise<ITallyData> => {
+}: IGenerateProofsArgs): Promise<IGenerateProofsData> => {
   // if we do not have the output directory just create it
   const isOutputDirExists = fs.existsSync(outputDir);
 
@@ -103,8 +101,8 @@ export const generateProofs = async ({
     useQuadraticVoting,
   });
 
-  await proofGenerator.generateMpProofs();
-  const { tallyData } = await proofGenerator.generateTallyProofs(networkName, chainId);
+  const processProofs = await proofGenerator.generateMpProofs();
+  const { proofs: tallyProofs, tallyData } = await proofGenerator.generateTallyProofs(networkName, chainId);
 
-  return tallyData;
+  return { processProofs, tallyProofs, tallyData };
 };

packages/sdk/ts/proof/index.ts

@@ -1,3 +1,3 @@
-export type { IGenerateProofsArgs, IProof, IProveOnChainArgs } from "./types";
+export type { IGenerateProofsArgs, IGenerateProofsData, IProof, IProveOnChainArgs } from "./types";
 export { generateProofs } from "./generate";
 export { proveOnChain } from "./prove";

packages/sdk/ts/proof/types.ts

@@ -3,6 +3,8 @@ import type { Groth16Proof, SnarkProof } from "maci-contracts";
 import type { CircuitInputs } from "maci-core";
 import type { PublicSignals } from "snarkjs";
 
+import { ITallyData } from "../tally";
+
 /**
  * Interface for the arguments to the proveOnChain command
  */
@@ -167,3 +169,12 @@ export interface IGenerateProofsArgs {
    */
   tallyFile: string;
 }
+
+/**
+ * Interface for the data returned by the generateProofs function
+ */
+export interface IGenerateProofsData {
+  processProofs: IProof[];
+  tallyProofs: IProof[];
+  tallyData: ITallyData;
+}