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

use crate::cfg_client;

cfg_client! {
    use solana_program::address_lookup_table::state::AddressLookupTable;
    use solana_program::address_lookup_table::AddressLookupTableAccount;
    use solana_client::nonblocking::rpc_client::RpcClient;
    use crate::OnDemandError;
    use solana_sdk::pubkey::Pubkey;

    const LUT_SIGNER_SEED: &[u8] = b"LutSigner";

    pub fn find_lut_signer(k: &Pubkey) -> Pubkey {
        Pubkey::find_program_address(
            &[LUT_SIGNER_SEED, k.as_ref()],
            &solana_address_lookup_table_program::id()).0
    }

    pub fn find_lut_of(k: &Pubkey, lut_slot: u64) -> Pubkey {
         Pubkey::find_program_address(&[
            find_lut_signer(k).as_ref(),
            lut_slot.to_le_bytes().as_ref(),
        ], &solana_address_lookup_table_program::id()).0
    }

    pub async fn fetch(client: &RpcClient, address: &Pubkey) -> Result<AddressLookupTableAccount, OnDemandError> {
        let account = client.get_account_data(address)
            .await
            .map_err(|_| OnDemandError::AddressLookupTableFetchError)?;
        let lut = AddressLookupTable::deserialize(&account)
            .map_err(|_| OnDemandError::AddressLookupTableDeserializeError)?;
        let out = AddressLookupTableAccount {
            key: address.clone(),
            addresses: lut.addresses.iter().cloned().collect(),
        };
        Ok(out)
    }
}