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

use alloy_primitives::Address;
use brontes_macros::discovery_impl;
use brontes_pricing::Protocol;

discovery_impl!(
    PancakeSwapV3Discovery,
    crate::UniswapV3Factory::createPoolCall,
    0x0BFbCF9fa4f9C56B0F40a671Ad40E0805A091865,
    |deployed_address: Address, trace_index: u64, call_data: createPoolCall, _| async move {
        let mut token_a = call_data.tokenA;
        let mut token_b = call_data.tokenB;

        if token_a > token_b {
            std::mem::swap(&mut token_a, &mut token_b)
        }

        vec![NormalizedNewPool {
            pool_address: deployed_address,
            trace_index,
            protocol: Protocol::PancakeSwapV3,
            tokens: vec![token_a, token_b],
        }]
    }
);

discovery_impl!(
    PancakeSwapV2Discovery,
    crate::UniswapV2Factory::createPairCall,
    0x1097053Fd2ea711dad45caCcc45EfF7548fCB362,
    |deployed_address: Address, trace_index: u64, call_data: createPairCall, _| async move {
        let mut token_a = call_data.tokenA;
        let mut token_b = call_data.tokenB;

        if token_a > token_b {
            std::mem::swap(&mut token_a, &mut token_b)
        }

        vec![NormalizedNewPool {
            pool_address: deployed_address,
            trace_index,
            protocol: Protocol::PancakeSwapV2,
            tokens: vec![token_a, token_b],
        }]
    }
);

#[cfg(test)]
pub mod test {
    use alloy_primitives::{hex, Address, TxHash};
    use brontes_types::{
        db::token_info::TokenInfoWithAddress, normalized_actions::pool::NormalizedNewPool, Protocol,
    };

    use crate::test_utils::ClassifierTestUtils;

    #[brontes_macros::test]
    async fn test_pancake_v3_discovery() {
        let utils = ClassifierTestUtils::new().await;
        let tx =
            TxHash::new(hex!("2b16d7a3937375d50b29bbec621b3f33bee00c76d1f4c907ae483fa49f63e2f1"));

        let eq_create = NormalizedNewPool {
            trace_index:  1,
            protocol:     Protocol::PancakeSwapV3,
            pool_address: Address::new(hex!("Ed4D5317823Ff7BC8BB868C1612Bb270a8311179")),
            tokens:       vec![
                Address::new(hex!("186eF81fd8E77EEC8BfFC3039e7eC41D5FC0b457")),
                TokenInfoWithAddress::usdt().address,
            ],
        };

        utils
            .test_discovery_classification(
                tx,
                Address::new(hex!("Ed4D5317823Ff7BC8BB868C1612Bb270a8311179")),
                |mut pool| {
                    assert_eq!(pool.len(), 1);
                    let pool = pool.remove(0);
                    assert_eq!(pool.protocol, eq_create.protocol);
                    assert_eq!(pool.tokens, eq_create.tokens);
                },
            )
            .await
            .unwrap();
    }
}