chore: format code

Cargo.toml

@@ -22,7 +22,7 @@ tokio = { version = "1.39.3", features = ["macros", "rt-multi-thread"] }
 url = { version = "2.4.0", features = ["serde"] }
 
 [dev-dependencies]
-aptos-testcontainer = { git = "https://github.com/sota-zk-labs/aptos-testcontainer.git", tag = "v0.0.1", features = [
+aptos-testcontainer = { git = "https://github.com/sota-zk-labs/aptos-testcontainer.git", tag = "v0.0.3", features = [
   "testing",
 ] }
 

src/bin/main.rs

@@ -1,9 +1,12 @@
-use log::{error, info};
+use log::info;
 
 use verifier_onchain_services::config::{AppConfig, EnvConfig};
+use verifier_onchain_services::contracts_caller::memory_page_fact_registry::register_continuous_page_batch::register_continuous_page_batch;
 use verifier_onchain_services::contracts_caller::memory_page_fact_registry::sample_register_memory::sample_register_continuous_page_batch;
 use verifier_onchain_services::contracts_caller::verify_fri::sample_verify_fri_input::sample_verify_fri_input;
+use verifier_onchain_services::contracts_caller::verify_fri::verify_fri::verify_fri;
 use verifier_onchain_services::contracts_caller::verify_merkle::sample_verify_merkle_input::sample_verify_merkle_input;
+use verifier_onchain_services::contracts_caller::verify_merkle::verify_merkle::verify_merkle;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
@@ -17,18 +20,37 @@ async fn main() -> anyhow::Result<()> {
     config.account.set_sequence_number(sequence_number);
 
     for i in 1..4 {
-        sample_verify_merkle_input(&config, i).await?;
+        let (merkle_view, initial_merkle_queue, height, expected_root) =
+            sample_verify_merkle_input(i)?;
+        verify_merkle(
+            &config,
+            merkle_view,
+            initial_merkle_queue,
+            height,
+            expected_root,
+        )
+        .await?;
         info!("Verify Merkle {} success", i);
     }
 
     for i in 1..8 {
-        sample_verify_fri_input(&config, i).await?;
+        let (fri_verify_input, proof, fri_queue, evaluation_point, fri_step_size, expected_root) =
+            sample_verify_fri_input(i)?;
+        verify_fri(
+            &config,
+            fri_verify_input,
+            proof,
+            fri_queue,
+            evaluation_point,
+            fri_step_size,
+            expected_root,
+        )
+        .await?;
         info!("Verify FRI {} success", i);
     }
 
-    if !sample_register_continuous_page_batch(&config).await? {
-        error!("something went wrong!")
-    }
-
+    let register_continuous_page_batch_input = sample_register_continuous_page_batch()?;
+    register_continuous_page_batch(&config, register_continuous_page_batch_input).await?;
+    info!("Register continuous page batch success");
     Ok(())
 }

src/contracts_caller/memory_page_fact_registry/sample_register_memory.rs

@@ -1,42 +1,34 @@
 use std::fs::File;
 use std::io::BufReader;
 
-use crate::config::AppConfig;
-use crate::contracts_caller::memory_page_fact_registry::register_continuous_memorypage::register_continuous_memorypage;
-use crate::contracts_caller::memory_page_fact_registry::register_continuous_page_batch::register_continuous_page_batch;
 use crate::contracts_caller::memory_page_fact_registry::types::register_continuous_memorypage::ContinuousMemorypage;
 use crate::contracts_caller::memory_page_fact_registry::types::register_continuous_page_batch::MemoryPageEntries;
 
-pub async fn sample_register_continuous_page_batch(config: &AppConfig) -> anyhow::Result<bool> {
+pub fn sample_register_continuous_page_batch() -> anyhow::Result<MemoryPageEntries> {
     let file_path =
         "src/data_samples/memory_page_fact_registry/register_continuous_page_batch.json"
             .to_string();
     let input_file = File::open(file_path)?;
     let reader = BufReader::new(input_file);
     let memory_page_entries: MemoryPageEntries = serde_json::from_reader(reader)?;
-
-    register_continuous_page_batch(config, memory_page_entries).await
+    Ok(memory_page_entries)
 }
 
-pub async fn sample_register_continuous_page(config: &AppConfig) -> anyhow::Result<bool> {
+pub fn sample_register_continuous_page() -> anyhow::Result<ContinuousMemorypage> {
     let file_path =
         "src/data_samples/memory_page_fact_registry/register_memory_page.json".to_string();
     let input_file = File::open(file_path)?;
     let reader = BufReader::new(input_file);
     let continuous_memmory_page: ContinuousMemorypage = serde_json::from_reader(reader)?;
-
-    register_continuous_memorypage(config, continuous_memmory_page).await
+    Ok(continuous_memmory_page)
 }
 
-pub async fn sample_large_data_register_continuous_page_batch(
-    config: &AppConfig,
-) -> anyhow::Result<bool> {
+pub fn sample_large_data_register_continuous_page_batch() -> anyhow::Result<MemoryPageEntries> {
     let file_path =
         "src/data_samples/memory_page_fact_registry/large_data_register_continuous_page_batch.json"
             .to_string();
     let input_file = File::open(file_path)?;
     let reader = BufReader::new(input_file);
     let memory_page_entries: MemoryPageEntries = serde_json::from_reader(reader)?;
-
-    register_continuous_page_batch(config, memory_page_entries).await
+    Ok(memory_page_entries)
 }

src/contracts_caller/transaction_helper.rs

@@ -90,25 +90,3 @@ pub fn get_event_from_transaction(
     };
     event.ok_or(Error::new(CoreError::NotFound))
 }
-
-pub fn get_events_from_transaction(
-    transaction: &Transaction,
-    event_type: MoveType,
-) -> anyhow::Result<Vec<&Event>> {
-    let events: Vec<&Event> = match transaction {
-        Transaction::UserTransaction(txn) => {
-            txn.events.iter().filter(|s| s.typ == event_type).collect()
-        }
-        Transaction::BlockMetadataTransaction(_) => vec![],
-        Transaction::PendingTransaction(_) => vec![],
-        Transaction::GenesisTransaction(_) => vec![],
-        Transaction::StateCheckpointTransaction(_) => vec![],
-        Transaction::BlockEpilogueTransaction(_) => vec![],
-        Transaction::ValidatorTransaction(_) => vec![],
-    };
-    if events.is_empty() {
-        Err(Error::new(CoreError::NotFound))
-    } else {
-        Ok(events)
-    }
-}

src/contracts_caller/verify_fri/sample_verify_fri_input.rs

@@ -2,13 +2,21 @@ use std::fs::File;
 use std::io::BufReader;
 use std::str::FromStr;
 
-use crate::config::AppConfig;
-use crate::contracts_caller::verify_fri::types::fri_verify_input::FriVerifyInput;
-use crate::contracts_caller::verify_fri::verify_fri::verify_fri;
 use aptos_sdk::move_types::u256::U256;
 use aptos_sdk::move_types::value::MoveValue;
 
-pub async fn sample_verify_fri_input(config: &AppConfig, index: isize) -> anyhow::Result<()> {
+use crate::contracts_caller::verify_fri::types::fri_verify_input::FriVerifyInput;
+
+pub fn sample_verify_fri_input(
+    index: isize,
+) -> anyhow::Result<(
+    FriVerifyInput,
+    MoveValue,
+    MoveValue,
+    MoveValue,
+    MoveValue,
+    MoveValue,
+)> {
     let file_path = format!("./src/data_samples/fri_verify/fri_verify_{}.json", index);
     let input_file = File::open(file_path)?;
     let reader = BufReader::new(input_file);
@@ -36,15 +44,12 @@ pub async fn sample_verify_fri_input(config: &AppConfig, index: isize) -> anyhow
         MoveValue::U256(U256::from_str(&fri_verify_input.evaluation_point.clone())?);
     let fri_step_size = MoveValue::U256(U256::from_str(&fri_verify_input.fri_step_size.clone())?);
     let expected_root = MoveValue::U256(U256::from_str(&fri_verify_input.expected_root.clone())?);
-    verify_fri(
-        config,
+    Ok((
         fri_verify_input,
         proof,
         fri_queue,
         evaluation_point,
         fri_step_size,
         expected_root,
-    )
-    .await?;
-    Ok(())
+    ))
 }

src/contracts_caller/verify_merkle/sample_verify_merkle_input.rs

@@ -2,13 +2,14 @@ use std::fs::File;
 use std::io::BufReader;
 use std::str::FromStr;
 
-use crate::config::AppConfig;
-use crate::contracts_caller::verify_merkle::types::verify_merkle_input::MerkleVerifyInput;
-use crate::contracts_caller::verify_merkle::verify_merkle::verify_merkle;
 use aptos_sdk::move_types::u256::U256;
 use aptos_sdk::move_types::value::MoveValue;
 
-pub async fn sample_verify_merkle_input(config: &AppConfig, index: isize) -> anyhow::Result<()> {
+use crate::contracts_caller::verify_merkle::types::verify_merkle_input::MerkleVerifyInput;
+
+pub fn sample_verify_merkle_input(
+    index: isize,
+) -> anyhow::Result<(MoveValue, MoveValue, MoveValue, MoveValue)> {
     let file_path = format!(
         "./src/data_samples/merkle_verify/merkle_verify_{}.json",
         index
@@ -36,13 +37,5 @@ pub async fn sample_verify_merkle_input(config: &AppConfig, index: isize) -> any
     let height = MoveValue::U64(u64::from_str(&merkle_verify_input.height.clone())?);
     let expected_root =
         MoveValue::U256(U256::from_str(&merkle_verify_input.expected_root.clone())?);
-    verify_merkle(
-        config,
-        merkle_view,
-        initial_merkle_queue,
-        height,
-        expected_root,
-    )
-    .await?;
-    Ok(())
+    Ok((merkle_view, initial_merkle_queue, height, expected_root))
 }