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

use std::sync::{Arc, Mutex};

use axiom_circuit::{
    axiom_codec::HiLo,
    axiom_eth::halo2_base::{AssignedValue, Context},
    subquery::{caller::SubqueryCaller, types::AssignedStorageSubquery},
};
use ethers::providers::Http;

use crate::Fr;

/// Storage subquery builder
pub struct Storage<'a> {
    pub block_number: AssignedValue<Fr>,
    pub addr: AssignedValue<Fr>,
    ctx: &'a mut Context<Fr>,
    caller: Arc<Mutex<SubqueryCaller<Http, Fr>>>,
}

pub(crate) fn get_storage(
    ctx: &mut Context<Fr>,
    caller: Arc<Mutex<SubqueryCaller<Http, Fr>>>,
    block_number: AssignedValue<Fr>,
    addr: AssignedValue<Fr>,
) -> Storage {
    Storage {
        block_number,
        addr,
        ctx,
        caller,
    }
}

impl<'a> Storage<'a> {
    /// Fetches the storage subquery and returns the HiLo<AssignedValue<Fr>> result
    ///
    /// * `slot` - The storage slot to fetch
    pub fn slot(self, slot: HiLo<AssignedValue<Fr>>) -> HiLo<AssignedValue<Fr>> {
        let mut subquery_caller = self.caller.lock().unwrap();
        let subquery = AssignedStorageSubquery {
            block_number: self.block_number,
            addr: self.addr,
            slot,
        };
        subquery_caller.call(self.ctx, subquery)
    }
}