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

use std::sync::Arc;

use alloy_sol_macro::sol;
use alloy_sol_types::SolCall;
use brontes_types::traits::TracingProvider;
use futures::TryFutureExt;
use reth_rpc_types::{request::TransactionInput, TransactionRequest};

use super::UniswapV2Pool;
use crate::errors::AmmError;

sol!(
    IGetUniswapV2PoolDataBatchRequest,
    "./src/protocols/uniswap_v2/batch_request/GetUniswapV2PoolDataBatchRequestABI.json"
);

sol!(
    struct PoolData {
        address tokenA;
        uint8 tokenADecimals;
        address tokenB;
        uint8 tokenBDecimals;
        uint112 reserve0;
        uint112 reserve1;
    }

    function data_constructor(address[] memory pools) returns(PoolData[]);
);

fn populate_pool_data_from_tokens(mut pool: UniswapV2Pool, pool_data: PoolData) -> UniswapV2Pool {
    pool.token_a = pool_data.tokenA;
    pool.token_a_decimals = pool_data.tokenADecimals;
    pool.token_b = pool_data.tokenB;
    pool.token_b_decimals = pool_data.tokenBDecimals;
    pool.reserve_0 = pool_data.reserve0;
    pool.reserve_1 = pool_data.reserve1;

    pool
}

pub async fn get_v2_pool_data<M: TracingProvider>(
    pool: &mut UniswapV2Pool,
    block: Option<u64>,
    middleware: Arc<M>,
) -> Result<(), AmmError> {
    let mut bytecode = IGetUniswapV2PoolDataBatchRequest::BYTECODE.to_vec();
    data_constructorCall::new((vec![pool.address],)).abi_encode_raw(&mut bytecode);

    let req = TransactionRequest {
        to: None,
        input: TransactionInput::new(bytecode.into()),
        ..Default::default()
    };

    let res = middleware
        .eth_call_light(req, block.unwrap().into())
        .map_err(|e| eyre::eyre!("v2 state call failed, err={}", e))
        .await?;

    let mut return_data = data_constructorCall::abi_decode_returns(&res, false)?;
    *pool = populate_pool_data_from_tokens(pool.to_owned(), return_data._0.remove(0));
    Ok(())
}