kabu_defi_pools/protocols/
uniswapv3.rs

1use alloy::primitives::{b256, Address, Bytes, B256};
2use alloy::sol_types::SolCall;
3use kabu_defi_abi::uniswap3::IUniswapV3Pool;
4use kabu_defi_address_book::FactoryAddress;
5
6use crate::protocols::helper::get_uniswap3pool_address;
7use crate::protocols::match_abi;
8use crate::protocols::protocol::Protocol;
9
10pub struct UniswapV3Protocol {}
11
12impl UniswapV3Protocol {
13    pub fn get_pool_address_for_tokens(token0: Address, token1: Address, fee: u32) -> Address {
14        let init_code: B256 = b256!("e34f199b19b2b4f47f68442619d555527d244f78a3297ea89325f843f87b8b54");
15
16        get_uniswap3pool_address(token0, token1, fee, FactoryAddress::UNISWAP_V3, init_code)
17    }
18
19    pub fn is_code(code: &Bytes) -> bool {
20        match_abi(code, vec![IUniswapV3Pool::swapCall::SELECTOR, IUniswapV3Pool::mintCall::SELECTOR, IUniswapV3Pool::collectCall::SELECTOR])
21    }
22}
23
24impl Protocol for UniswapV3Protocol {
25    fn get_pool_address_vec_for_tokens(token0: Address, token1: Address) -> Vec<Address> {
26        let init_code: B256 = "e34f199b19b2b4f47f68442619d555527d244f78a3297ea89325f843f87b8b54".parse().unwrap();
27
28        let pair_address0 = get_uniswap3pool_address(token0, token1, 100, FactoryAddress::UNISWAP_V3, init_code);
29        let pair_address1 = get_uniswap3pool_address(token0, token1, 500, FactoryAddress::UNISWAP_V3, init_code);
30        let pair_address2 = get_uniswap3pool_address(token0, token1, 3000, FactoryAddress::UNISWAP_V3, init_code);
31        let pair_address3 = get_uniswap3pool_address(token0, token1, 10000, FactoryAddress::UNISWAP_V3, init_code);
32
33        vec![pair_address0, pair_address1, pair_address2, pair_address3]
34    }
35}