1 files changed, 0 insertions, 177 deletions
diff --git a/zkvms/zkm/sdk/src/local/prover.rs b/zkvms/zkm/sdk/src/local/prover.rs
deleted file mode 100644
index a62d3ca..0000000
--- a/zkvms/zkm/sdk/src/local/prover.rs
+++ /dev/null
@@ -1,177 +0,0 @@
-use crate::prover::{Prover, ProverInput, ProverResult};
-use anyhow::bail;
-use async_trait::async_trait;
-use std::collections::HashMap;
-use std::fs;
-use std::path::Path;
-use std::sync::{Arc, Mutex};
-use std::thread;
-use std::time::Duration;
-use std::time::Instant;
-use tokio::time::sleep;
-
-pub struct ProverTask {
-    proof_id: String,
-    input: ProverInput,
-    result: Option<ProverResult>,
-    is_done: bool,
-    vk_path: String,
-}
-
-impl ProverTask {
-    fn new(proof_id: &str, vk_path: &str, input: &ProverInput) -> ProverTask {
-        ProverTask {
-            proof_id: proof_id.to_string(),
-            input: input.clone(),
-            result: None,
-            is_done: false,
-            vk_path: vk_path.to_string(),
-        }
-    }
-
-    fn run(&mut self) {
-        let mut result = ProverResult::default();
-        let vk_path = self.vk_path.to_owned();
-        let inputdir = format!("/tmp/{}/input", self.proof_id);
-        let outputdir = format!("/tmp/{}/output", self.proof_id);
-        fs::create_dir_all(&inputdir).unwrap();
-        fs::create_dir_all(&outputdir).unwrap();
-        let (should_agg, receipt, elf_id) =
-            crate::local::stark::prove_stark(&self.input, &inputdir, &mut result).unwrap();
-        if self.input.execute_only {
-            result.proof_with_public_inputs = vec![];
-            result.stark_proof = vec![];
-            result.solidity_verifier = vec![];
-        } else if !should_agg {
-            log::info!(
-                "There is only one segment with segment size {}, will skip the aggregation!",
-                self.input.seg_size
-            );
-        } else if !self.input.precompile {
-            match crate::local::snark::prove_snark(&vk_path, &inputdir, &outputdir) {
-                Ok(()) => {
-                    result.stark_proof =
-                        std::fs::read(format!("{}/proof_with_public_inputs.json", inputdir))
-                            .unwrap();
-                    result.proof_with_public_inputs =
-                        std::fs::read(format!("{}/snark_proof_with_public_inputs.json", outputdir))
-                            .unwrap();
-                    //result.solidity_verifier =
-                    //    std::fs::read(format!("{}/verifier.sol", outputdir)).unwrap();
-                    result.public_values =
-                        std::fs::read(format!("{}/public_values.json", inputdir)).unwrap();
-                }
-                Err(e) => {
-                    log::error!("prove_snark error : {}", e);
-                }
-            }
-        }
-        if let Some(receipt) = receipt {
-            result.receipt.clone_from(&receipt);
-            result.elf_id.clone_from(&elf_id.unwrap());
-        }
-        self.result = Some(result);
-        self.is_done = true;
-    }
-
-    fn is_done(&self) -> bool {
-        self.is_done
-    }
-}
-
-pub struct LocalProver {
-    tasks: Arc<Mutex<HashMap<String, Arc<Mutex<ProverTask>>>>>,
-    vk_path: String,
-}
-
-impl LocalProver {
-    pub fn new(vk_path: &str) -> LocalProver {
-        LocalProver {
-            tasks: Arc::new(Mutex::new(HashMap::new())),
-            vk_path: vk_path.to_string(),
-        }
-    }
-}
-
-#[async_trait]
-impl Prover for LocalProver {
-    async fn request_proof<'a>(&self, input: &'a ProverInput) -> anyhow::Result<String> {
-        let proof_id: String = uuid::Uuid::new_v4().to_string();
-        let task: Arc<Mutex<ProverTask>> =
-            Arc::new(Mutex::new(ProverTask::new(&proof_id, &self.vk_path, input)));
-        self.tasks
-            .lock()
-            .unwrap()
-            .insert(proof_id.clone(), task.clone());
-        thread::spawn(move || {
-            task.lock().unwrap().run();
-        });
-        Ok(proof_id)
-    }
-
-    async fn wait_proof<'a>(
-        &self,
-        proof_id: &'a str,
-        timeout: Option<Duration>,
-    ) -> anyhow::Result<Option<ProverResult>> {
-        let task = self.tasks.lock().unwrap().get(proof_id).unwrap().clone();
-        let start_time = Instant::now();
-        loop {
-            if let Some(timeout) = timeout {
-                if start_time.elapsed() > timeout {
-                    return Err(anyhow::anyhow!("Proof generation timed out."));
-                }
-            }
-            if task.lock().unwrap().is_done() {
-                self.tasks.lock().unwrap().remove(proof_id);
-                return Ok(task.lock().unwrap().result.clone());
-            }
-            log::info!("Waiting the proof result.");
-            sleep(Duration::from_secs(30)).await;
-        }
-    }
-
-    async fn setup_and_generate_sol_verifier<'a>(
-        &self,
-        vk_path: &'a str,
-        input: &'a ProverInput,
-        _timeout: Option<Duration>,
-    ) -> anyhow::Result<()> {
-        let mut result = ProverResult::default();
-        let path = Path::new(vk_path);
-
-        if path.is_dir() {
-            fs::remove_dir_all(vk_path).unwrap();
-        }
-        fs::create_dir_all(vk_path).unwrap();
-
-        let (should_agg, _, _) =
-            crate::local::stark::prove_stark(input, vk_path, &mut result).unwrap();
-        if !should_agg {
-            log::info!("Setup: generating the stark proof false, please check the SEG_SIZE or other parameters.");
-            bail!("Setup: generating the stark proof false, please check the SEG_SIZE or other parameters!");
-        }
-
-        match crate::local::snark::setup_and_generate_sol_verifier(vk_path) {
-            Ok(()) => {
-                log::info!("setup_and_generate_sol_verifier successfully, the verify key and verifier contract are in the {}", vk_path);
-                Ok(())
-            }
-            Err(e) => {
-                log::error!("setup_and_generate_sol_verifier error : {}", e);
-                bail!("setup_and_generate_sol_verifier error");
-            }
-        }
-    }
-
-    async fn prove<'a>(
-        &self,
-        input: &'a ProverInput,
-        timeout: Option<Duration>,
-    ) -> anyhow::Result<Option<ProverResult>> {
-        log::info!("Calling request_proof.");
-        let proof_id = self.request_proof(input).await?;
-        log::info!("Calling wait_proof, proof_id={}", proof_id);
-        self.wait_proof(&proof_id, timeout).await
-    }
-}