starkli 0.1.3

Starkli (/ˈstɑːrklaɪ/), a blazing fast CLI tool for Starknet powered by starknet-rs
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

// Very temporary implementation of a hard-coded addres book

use std::cell::OnceCell;

use anyhow::Result;
use starknet::{
    core::{chain_id, types::FieldElement},
    macros::felt,
};

use crate::chain_id::ChainIdSource;

pub const HARDCODED_ADDRESS_BOOK: [AddressBookEntry; 4] = [
    AddressBookEntry {
        chain_id: chain_id::MAINNET,
        name: "eth",
        address: felt!("0x049d36570d4e46f48e99674bd3fcc84644ddd6b96f7c741b1562b82f9e004dc7"),
    },
    AddressBookEntry {
        chain_id: chain_id::TESTNET,
        name: "eth",
        address: felt!("0x049d36570d4e46f48e99674bd3fcc84644ddd6b96f7c741b1562b82f9e004dc7"),
    },
    AddressBookEntry {
        chain_id: chain_id::TESTNET2,
        name: "eth",
        address: felt!("0x049d36570d4e46f48e99674bd3fcc84644ddd6b96f7c741b1562b82f9e004dc7"),
    },
    AddressBookEntry {
        chain_id: chain_id::MAINNET,
        name: "zklend",
        address: felt!("0x04c0a5193d58f74fbace4b74dcf65481e734ed1714121bdc571da345540efa05"),
    },
];

pub struct AddressBookEntry {
    pub chain_id: FieldElement,
    pub name: &'static str,
    pub address: FieldElement,
}

/// A resolver that lazily fetches chain id to avoid unnecessary network calls.
pub struct AddressBookResolver<S> {
    chain_id_source: S,
    chain_id: OnceCell<FieldElement>,
}

impl<S> AddressBookResolver<S> {
    pub fn new(chain_id_source: S) -> Self {
        Self {
            chain_id_source,
            chain_id: OnceCell::new(),
        }
    }
}

impl<S> AddressBookResolver<S>
where
    S: ChainIdSource,
{
    pub async fn resolve_name(&self, name: &str) -> Result<Option<FieldElement>> {
        let chain_id_cell = &self.chain_id;

        let chain_id = match chain_id_cell.get() {
            Some(chain_id) => *chain_id,
            None => {
                let chain_id = self.chain_id_source.get_chain_id().await?;

                // It's OK if another thread set it first
                let _ = chain_id_cell.set(chain_id);

                chain_id
            }
        };

        Ok(HARDCODED_ADDRESS_BOOK.iter().find_map(|entry| {
            if entry.chain_id == chain_id && entry.name == name {
                Some(entry.address)
            } else {
                None
            }
        }))
    }
}