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

use crate::num_bigint::BigUint;
use multiversx_sc::{
    api::ESDT_NFT_CREATE_FUNC_NAME,
    codec::{top_encode_to_vec_u8, TopDecode},
};

use crate::{
    tx_mock::{BlockchainUpdate, TxCache, TxInput, TxLog, TxResult},
    world_mock::{EsdtInstance, EsdtInstanceMetadata},
};

use super::super::builtin_func_trait::BuiltinFunction;

pub struct ESDTNftCreate;

impl BuiltinFunction for ESDTNftCreate {
    fn name(&self) -> &str {
        ESDT_NFT_CREATE_FUNC_NAME
    }

    fn execute(&self, tx_input: TxInput, tx_cache: TxCache) -> (TxResult, BlockchainUpdate) {
        if tx_input.args.len() < 7 {
            let err_result = TxResult::from_vm_error("ESDTNFTCreate too few arguments");
            return (err_result, BlockchainUpdate::empty());
        }
        assert!(
            tx_input.to == tx_input.from,
            "ESDTNFTCreate expects that to == from"
        );

        let token_identifier = tx_input.args[0].as_slice();
        let amount = BigUint::from_bytes_be(tx_input.args[1].as_slice());
        let name = tx_input.args[2].clone();
        let royalties = u64::top_decode(tx_input.args[3].as_slice()).unwrap();
        let hash = tx_input.args[4].clone();
        let attributes = tx_input.args[5].clone();
        let uris = tx_input.args[6..].to_vec();

        let new_nonce = tx_cache.with_account_mut(&tx_input.to, |account| {
            let esdt_data = account
                .esdt
                .get_mut_by_identifier(token_identifier)
                .unwrap_or_else(|| panic!("ESDTNFTCreate unexpected token identifier"));
            esdt_data.last_nonce += 1;
            esdt_data.instances.push_instance(EsdtInstance {
                nonce: esdt_data.last_nonce,
                balance: amount.clone(),
                metadata: EsdtInstanceMetadata {
                    name,
                    creator: Some(tx_input.from.clone()),
                    royalties,
                    hash: Some(hash),
                    uri: uris,
                    attributes,
                },
            });

            esdt_data.last_nonce
        });

        let esdt_nft_create_log = TxLog {
            address: tx_input.from.clone(),
            endpoint: ESDT_NFT_CREATE_FUNC_NAME.into(),
            topics: vec![
                token_identifier.to_vec(),
                top_encode_to_vec_u8(&new_nonce).unwrap(),
                amount.to_bytes_be(),
                Vec::new(), // in actuality this should contain the fully marshalled ESDT data
            ],
            data: vec![],
        };

        let tx_result = TxResult {
            result_status: 0,
            result_values: vec![top_encode_to_vec_u8(&new_nonce).unwrap()],
            result_logs: vec![esdt_nft_create_log],
            ..Default::default()
        };

        (tx_result, tx_cache.into_blockchain_updates())
    }
}