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

use crate::chains;

#[derive(Clone, Debug)]
pub struct Network {
    pub rpc_url: String,
    pub chain_id: u64,
}

impl TryFrom<&str> for Network {
    type Error = anyhow::Error;

    fn try_from(value: &str) -> Result<Self, Self::Error> {
        let chain = match value {
            "mainnet" => {
                return Ok(Network {
                    rpc_url: "https://api.bitski.com/v1/web3/mainnet".to_owned(),
                    chain_id: 1,
                })
            }
            // remap to the short name since 'goerli' isn't in the list
            "goerli" => chains::chain_from_str("gor"),
            _ => chains::chain_from_str(value),
        }?;

        Ok(Network {
            rpc_url: format!("https://api.bitski.com/v1/web3/{}", chain.chain_id),
            chain_id: chain.chain_id,
        })
    }
}

#[test]
fn test_chain_name_try_from() {
    let n = Network::try_from("goerli").expect("could not get goerli chain");
    assert_eq!(n.chain_id, 5);

    let n = Network::try_from("mainnet").expect("could not get mainnet chain");
    assert_eq!(n.chain_id, 1);

    let n = Network::try_from("polygon").expect("could not get polygon chain");
    assert_eq!(n.chain_id, 137);
}