call_value_init.rs

use crate::{
    api::{
        const_handles, use_raw_handle, CallValueApi, CallValueApiImpl, ErrorApi, ErrorApiImpl,
        ManagedBufferApiImpl, ManagedTypeApi,
    },
    err_msg,
    imports::ContractBase,
    types::{
        BigUint, EgldOrEsdtTokenIdentifier, EsdtTokenPayment, ManagedRef, ManagedType, ManagedVec,
    },
};

/// Called initially in the generated code whenever no payable annotation is provided.
pub fn not_payable<A>()
where
    A: CallValueApi,
{
    A::call_value_api_impl().check_not_payable();
}

/// Called initially in the generated code whenever `#[payable("*")]` annotation is provided.
pub fn payable_any<A>()
where
    A: CallValueApi,
{
}

/// Called initially in the generated code whenever `#[payable("EGLD")]` annotation is provided.
pub fn payable_egld<A>()
where
    A: CallValueApi + ErrorApi,
{
    if A::call_value_api_impl().esdt_num_transfers() > 0 {
        A::error_api_impl().signal_error(err_msg::NON_PAYABLE_FUNC_ESDT.as_bytes());
    }
}

/// Called initially in the generated code whenever `#[payable("<token identifier>")]` annotation is provided.
///
/// Was never really used, expected to be deprecated/removed.
pub fn payable_single_specific_token<A, O>(obj: &O, expected_tokend_identifier: &str)
where
    A: CallValueApi + ManagedTypeApi + ErrorApi,
    O: ContractBase<Api = A>,
{
    let transfers = obj.call_value().all_esdt_transfers();
    if transfers.len() != 1 {
        A::error_api_impl().signal_error(err_msg::SINGLE_ESDT_EXPECTED.as_bytes());
    }
    let expected_token_handle: A::ManagedBufferHandle =
        use_raw_handle(const_handles::MBUF_TEMPORARY_1);
    A::managed_type_impl().mb_overwrite(
        expected_token_handle.clone(),
        expected_tokend_identifier.as_bytes(),
    );
    let transfer = transfers.get(0);
    if !A::managed_type_impl().mb_eq(
        transfer.token_identifier.get_handle(),
        expected_token_handle,
    ) {
        A::error_api_impl().signal_error(err_msg::BAD_TOKEN_PROVIDED.as_bytes());
    }
}

/// Initializes an argument annotated with `#[payment_amount]` or `#[payment]`.
pub fn arg_payment_amount<A, O>(obj: &O) -> BigUint<A>
where
    A: CallValueApi + ManagedTypeApi,
    O: ContractBase<Api = A>,
{
    obj.call_value().egld_or_single_esdt().amount
}

/// Initializes an argument annotated with `#[payment_token]`.
pub fn arg_payment_token<A, O>(obj: &O) -> EgldOrEsdtTokenIdentifier<A>
where
    A: CallValueApi + ManagedTypeApi,
    O: ContractBase<Api = A>,
{
    obj.call_value().egld_or_single_esdt().token_identifier
}

/// Initializes an argument annotated with `#[payment_nonce]`.
pub fn arg_payment_nonce<A, O>(obj: &O) -> u64
where
    A: CallValueApi + ManagedTypeApi,
    O: ContractBase<Api = A>,
{
    obj.call_value().egld_or_single_esdt().token_nonce
}

/// Initializes an argument annotated with `#[payment_multi]`.
pub fn arg_payment_multi<A, O>(
    obj: &O,
) -> ManagedRef<'static, A, ManagedVec<A, EsdtTokenPayment<A>>>
where
    A: CallValueApi + ManagedTypeApi,
    O: ContractBase<Api = A>,
{
    obj.call_value().all_esdt_transfers()
}