multiversx-sc-snippets 0.65.1

MultiversX framework for building smart contract interaction snippets
Documentation 
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use std::collections::HashMap;

use crate::sdk::{data::transaction::Transaction, wallet::Wallet};
use multiversx_sc_scenario::multiversx_sc::types::Address;
use multiversx_sdk::data::account::Account;
use multiversx_sdk::data::esdt::EsdtBalance;
use multiversx_sdk::gateway::{
    GatewayAsyncService, GetAccountEsdtTokensRequest, GetAccountRequest, GetAccountStorageRequest,
};

use crate::InteractorBase;

/// A user account that can sign transactions (a pem is present).
pub struct Sender {
    pub address: Address,
    pub hrp: String,
    pub wallet: Wallet,
    pub current_nonce: Option<u64>,
}

impl<GatewayProxy> InteractorBase<GatewayProxy>
where
    GatewayProxy: GatewayAsyncService,
{
    pub async fn recall_nonce(&self, address: &Address) -> u64 {
        let account = self
            .proxy
            .request(GetAccountRequest::new(&address.to_bech32(self.get_hrp())))
            .await
            .expect("failed to retrieve account nonce");

        account.nonce
    }

    pub async fn get_account(&self, address: &Address) -> Account {
        self.proxy
            .request(GetAccountRequest::new(&address.to_bech32(self.get_hrp())))
            .await
            .expect("failed to retrieve account")
    }

    pub async fn get_account_storage(&self, address: &Address) -> HashMap<String, String> {
        self.proxy
            .request(GetAccountStorageRequest::new(
                &address.to_bech32(self.get_hrp()),
            ))
            .await
            .expect("failed to retrieve account")
    }

    pub async fn get_account_esdt(&self, address: &Address) -> HashMap<String, EsdtBalance> {
        self.proxy
            .request(GetAccountEsdtTokensRequest::new(
                &address.to_bech32(self.get_hrp()),
            ))
            .await
            .expect("failed to retrieve account")
    }

    /// Updates:
    /// - the transaction with the nonce read from the network
    /// - the sender's current_nonce
    pub(crate) async fn set_tx_nonce_update_sender(
        &mut self,
        sender_address: &Address,
        transaction: &mut Transaction,
    ) {
        // read
        let sender = self
            .sender_map
            .get(sender_address)
            .expect("the wallet that was supposed to sign is not registered");

        // recall
        let nonce = self.recall_nonce(&sender.address).await;
        println!("sender's recalled nonce: {nonce}");

        // set tx nonce
        transaction.nonce = nonce;
        println!("-- tx nonce: {}", transaction.nonce);

        // update
        let sender = self
            .sender_map
            .get_mut(sender_address)
            .expect("the wallet that was supposed to sign is not registered");
        sender.current_nonce = Some(nonce + 1);
    }

    pub(crate) fn sign_tx(&self, sender_address: &Address, transaction: &mut Transaction) {
        // read
        let sender = self
            .sender_map
            .get(sender_address)
            .expect("the wallet that was supposed to sign is not registered");

        // sign
        let signature = sender.wallet.sign_tx(transaction);
        transaction.signature = Some(hex::encode(signature));
    }
}