use hiero_sdk_proto::services;
use crate::hooks::{
NftHookCall,
NftHookType,
};
use crate::protobuf::FromProtobuf;
use crate::{
AccountId,
TokenId,
};
#[derive(Debug, Clone, Eq, PartialEq)]
#[non_exhaustive]
pub struct TokenNftTransfer {
pub token_id: TokenId,
pub sender: AccountId,
pub receiver: AccountId,
pub serial: u64,
pub is_approved: bool,
pub sender_hook_call: Option<NftHookCall>,
pub receiver_hook_call: Option<NftHookCall>,
}
impl TokenNftTransfer {
pub(crate) fn from_protobuf(
pb: services::NftTransfer,
token_id: TokenId,
) -> crate::Result<Self> {
let sender_hook_call = match pb.sender_allowance_hook_call {
Some(services::nft_transfer::SenderAllowanceHookCall::PreTxSenderAllowanceHook(
hook,
)) => Some(NftHookCall::from_protobuf_with_type(hook, NftHookType::PreHookSender)?),
Some(
services::nft_transfer::SenderAllowanceHookCall::PrePostTxSenderAllowanceHook(hook),
) => Some(NftHookCall::from_protobuf_with_type(hook, NftHookType::PrePostHookSender)?),
None => None,
};
let receiver_hook_call = match pb.receiver_allowance_hook_call {
Some(
services::nft_transfer::ReceiverAllowanceHookCall::PreTxReceiverAllowanceHook(hook),
) => Some(NftHookCall::from_protobuf_with_type(hook, NftHookType::PreHookReceiver)?),
Some(
services::nft_transfer::ReceiverAllowanceHookCall::PrePostTxReceiverAllowanceHook(
hook,
),
) => {
Some(NftHookCall::from_protobuf_with_type(hook, NftHookType::PrePostHookReceiver)?)
}
None => None,
};
Ok(Self {
token_id,
sender: AccountId::from_protobuf(pb_getf!(pb, sender_account_id)?)?,
receiver: AccountId::from_protobuf(pb_getf!(pb, receiver_account_id)?)?,
serial: pb.serial_number as u64,
is_approved: pb.is_approval,
sender_hook_call,
receiver_hook_call,
})
}
}