feat: prove feature to not run prover locally

Author: MarcosNicolau

aggregation-mode/Cargo.toml

@@ -3,6 +3,7 @@ name = "proof_aggregator"
 version = "0.1.0"
 edition = "2021"
 
+
 [dependencies]
 sp1-sdk = "4.1.3"
 sp1_aggregator = { path = "./zkvm/sp1/" }
@@ -12,8 +13,12 @@ tracing = { version = "0.1", features = ["log"] }
 tracing-subscriber = { version = "0.3.0", features = ["env-filter"] }
 
 [build-dependencies]
-sp1-build = "4.1.3" 
+sp1-build = { version = "4.1.3" } 
 
 [profile.release]
 opt-level = 3
 lto = true
+
+[features]
+prove = []
+

aggregation-mode/src/aggregator/sp1.rs

@@ -1,18 +1,31 @@
 use sp1_aggregator::SP1CompressedProof;
-use sp1_sdk::{ProverClient, SP1ProofWithPublicValues, SP1Stdin, SP1VerifyingKey};
+use sp1_sdk::{ProverClient, SP1ProofMode, SP1ProofWithPublicValues, SP1Stdin, SP1VerifyingKey};
 
 use super::interface::{AggregatedProof, AggregatedVerificationError, ProgramOutput};
 
 const PROGRAM_ELF: &[u8] = include_bytes!("../../zkvm/sp1/elf/sp1_aggregator_program");
 
 pub struct SP1AggregatedProof {
-    proof: SP1ProofWithPublicValues,
-    vk: SP1VerifyingKey,
+    pub proof: SP1ProofWithPublicValues,
+    pub vk: SP1VerifyingKey,
 }
 
 pub(crate) fn aggregate_proofs(
     proofs: Vec<SP1CompressedProof>,
 ) -> Result<ProgramOutput, AggregatedVerificationError> {
+    #[cfg(feature = "prove")]
+    {
+        prove(proofs)
+    }
+    // If not in prove mode, execute the program and create a mock proof
+    #[cfg(not(feature = "prove"))]
+    {
+        mock_prove(proofs)
+    }
+}
+
+#[cfg(feature = "prove")]
+fn prove(proofs: Vec<SP1CompressedProof>) -> Result<ProgramOutput, AggregatedVerificationError> {
     let mut stdin = SP1Stdin::new();
     stdin.write(&proofs);
 
@@ -33,3 +46,29 @@ pub(crate) fn aggregate_proofs(
 
     Ok(output)
 }
+
+fn mock_prove(
+    proofs: Vec<SP1CompressedProof>,
+) -> Result<ProgramOutput, AggregatedVerificationError> {
+    let mut stdin = SP1Stdin::new();
+    stdin.write(&proofs);
+
+    let client = ProverClient::from_env();
+    let (pk, vk) = client.setup(PROGRAM_ELF);
+    let (public_vales, _) = client
+        .execute(PROGRAM_ELF, &stdin)
+        .run()
+        .map_err(|_| AggregatedVerificationError::SP1Proving)?;
+
+    let output = ProgramOutput::new(AggregatedProof::SP1(SP1AggregatedProof {
+        proof: SP1ProofWithPublicValues::create_mock_proof(
+            &pk,
+            public_vales,
+            SP1ProofMode::Groth16,
+            "",
+        ),
+        vk,
+    }));
+
+    Ok(output)
+}