merkrs 0.3.0

Merkle tree library for Rust, compatible with OpenZeppelin's JavaScript implementation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

#![allow(
    clippy::print_stdout,
    clippy::indexing_slicing,
    unused_crate_dependencies,
    reason = "example"
)]
//! Serialization example.
//!
//! Demonstrates how to dump and load Merkle trees for persistence.
//!
//! Run: `cargo run --example serialization`

use merkrs::bytes::encode_hex;
use merkrs::{
    Bytes32, SimpleMerkleTree, SimpleMerkleTreeData, StandardMerkleTree, StandardMerkleTreeData,
    simple, standard,
};
use serde_json::json;

type Result<T> = std::result::Result<T, Box<dyn std::error::Error>>;

fn main() -> Result<()> {
    println!("=== SimpleMerkleTree Serialization ===\n");
    simple_tree_example()?;

    println!("\n=== StandardMerkleTree Serialization ===\n");
    standard_tree_example()?;

    Ok(())
}

fn simple_tree_example() -> Result<()> {
    let values: Vec<Bytes32> = vec![[1u8; 32], [2u8; 32], [3u8; 32], [4u8; 32]];
    let tree = SimpleMerkleTree::new(&values, simple::Options::default())?;

    println!("Original root: {}", encode_hex(tree.root()));

    let data = tree.to_data();
    let json_str = serde_json::to_string_pretty(&data)?;
    println!("\nSerialized JSON:\n{json_str}");

    let loaded_data: SimpleMerkleTreeData = serde_json::from_str(&json_str)?;
    let loaded_tree = SimpleMerkleTree::from_data(loaded_data, None)?;

    println!("\nLoaded root: {}", encode_hex(loaded_tree.root()));
    println!("Roots match: {}", tree.root() == loaded_tree.root());

    let proof = loaded_tree.proof(&values[0])?;
    let valid = loaded_tree.verify_proof(&values[0], &proof)?;
    println!("Proof verification after load: {valid}");

    Ok(())
}

fn standard_tree_example() -> Result<()> {
    let recipients = vec![
        vec![
            json!("0x1111111111111111111111111111111111111111"),
            json!(1000),
        ],
        vec![
            json!("0x2222222222222222222222222222222222222222"),
            json!(2000),
        ],
    ];

    let tree = StandardMerkleTree::new(
        recipients,
        vec!["address".into(), "uint256".into()],
        standard::Options::default(),
    )?;

    println!("Original root: {}", encode_hex(tree.root()));

    let data = tree.to_data();
    let json_str = serde_json::to_string_pretty(&data)?;
    println!("\nSerialized JSON:\n{json_str}");

    let loaded_data: StandardMerkleTreeData = serde_json::from_str(&json_str)?;
    let loaded_tree = StandardMerkleTree::from_data(loaded_data)?;

    println!("\nLoaded root: {}", encode_hex(loaded_tree.root()));
    println!("Roots match: {}", tree.root() == loaded_tree.root());

    println!("\nLoaded entries:");
    for (i, value) in loaded_tree.entries() {
        println!("  [{i}] {} -> {}", value[0], value[1]);
    }

    Ok(())
}