pool-sync 3.0.0

A library for synchronizing and managing various types of liquidity pools across different blockchains
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

use alloy::primitives::FixedBytes;
use alloy::primitives::U256;
use alloy::rpc::types::Log;
use alloy::sol_types::SolEvent;
use alloy::{dyn_abi::DynSolValue, primitives::Address};
use serde::{Deserialize, Serialize};

use crate::pools::gen::Vault;
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct BalancerV2Pool {
    pub address: Address,
    pub pool_id: FixedBytes<32>,
    pub token0: Address,
    pub token1: Address,
    pub token0_name: String,
    pub token1_name: String,
    pub token0_decimals: u8,
    pub token1_decimals: u8,
    pub additional_tokens: Vec<Address>,
    pub additional_token_names: Vec<String>,
    pub additional_token_decimals: Vec<u8>,
    pub balances: Vec<U256>,
    pub weights: Vec<U256>,
    pub swap_fee: U256,
}

impl BalancerV2Pool {
    pub fn get_tokens(&self) -> Vec<Address> {
        let mut tokens = vec![self.token0, self.token1];
        tokens.extend(self.additional_tokens.iter());
        tokens
    }

    pub fn get_token_index(&self, token: &Address) -> Option<usize> {
        if *token == self.token0 {
            Some(0)
        } else if *token == self.token1 {
            Some(1)
        } else {
            self.additional_tokens
                .iter()
                .position(|&t| t == *token)
                .map(|pos| pos + 2)
        }
    }

    pub fn get_balance(&self, token: &Address) -> U256 {
        let index = self.get_token_index(token);
        if let Some(index) = index {
            self.balances[index]
        } else {
            U256::ZERO
        }
    }
}

pub fn process_balance_data(pool: &mut BalancerV2Pool, log: Log) {
    let event = Vault::Swap::decode_log(log.as_ref(), true).unwrap();

    let log_token_in_idx = pool.get_token_index(&event.tokenIn).unwrap();
    let log_token_out_idx = pool.get_token_index(&event.tokenOut).unwrap();

    pool.balances[log_token_in_idx] =
        pool.balances[log_token_in_idx].saturating_add(event.amountIn);
    pool.balances[log_token_out_idx] =
        pool.balances[log_token_out_idx].saturating_sub(event.amountOut);
}

impl From<&[DynSolValue]> for BalancerV2Pool {
    fn from(data: &[DynSolValue]) -> Self {
        let pool_address = data[0].as_address().unwrap();
        let pool_id = FixedBytes::from_slice(data[1].as_fixed_bytes().unwrap().0);
        let token0 = data[2].as_address().unwrap();
        let token1 = data[3].as_address().unwrap();
        let token0_decimals = data[4].as_uint().unwrap().0.to::<u8>();
        let token1_decimals = data[5].as_uint().unwrap().0.to::<u8>();
        let additional_tokens: Vec<Address> = data[6]
            .as_array()
            .unwrap()
            .iter()
            .map(|v| v.as_address().unwrap())
            .collect();
        let additional_token_decimals: Vec<u8> = data[7]
            .as_array()
            .unwrap()
            .iter()
            .map(|v| v.as_uint().unwrap().0.to::<u8>())
            .collect();
        let balances: Vec<U256> = data[8]
            .as_array()
            .unwrap()
            .iter()
            .map(|v| v.as_uint().unwrap().0)
            .collect();
        let weights: Vec<U256> = data[9]
            .as_array()
            .unwrap()
            .iter()
            .map(|v| v.as_uint().unwrap().0)
            .collect();
        let swap_fee = data[10].as_uint().unwrap().0;

        Self {
            address: pool_address,
            pool_id,
            token0,
            token1,
            token0_name: String::new(), // To be populated later
            token1_name: String::new(), // To be populated later
            token0_decimals,
            token1_decimals,
            additional_tokens,
            additional_token_names: Vec::new(), // To be populated later
            additional_token_decimals,
            balances,
            weights,
            swap_fee,
        }
    }
}