anvil_zksync_core/node/
keys.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
use zksync_types::{Address, StorageKey};

#[derive(Copy, Clone)]
pub enum StorageKeyLayout {
    ZkEra,
    ZkOs,
}

impl StorageKeyLayout {
    pub fn get_nonce_key(&self, account: &Address) -> StorageKey {
        match self {
            StorageKeyLayout::ZkEra => zksync_types::get_nonce_key(account),
            StorageKeyLayout::ZkOs => crate::node::zkos::zkos_get_nonce_key(account),
        }
    }

    pub fn get_storage_key_for_base_token(&self, address: &Address) -> StorageKey {
        match self {
            StorageKeyLayout::ZkEra => zksync_types::utils::storage_key_for_eth_balance(address),
            StorageKeyLayout::ZkOs => crate::node::zkos::zkos_storage_key_for_eth_balance(address),
        }
    }
}