use crate::{
proto::{self, ToProto},
AccountId, Client, Transaction,
};
use failure::Error;
use protobuf::RepeatedField;
pub struct TransactionCryptoTransfer {
transfers: Vec<(AccountId, i64)>,
}
impl Transaction<TransactionCryptoTransfer> {
pub fn crypto_transfer(client: &Client) -> Self {
Self::new(
client,
TransactionCryptoTransfer {
transfers: Vec::new(),
},
)
}
pub fn transfer(&mut self, id: AccountId, amount: i64) -> &mut Self {
self.inner.transfers.push((id, amount));
self
}
}
impl ToProto<proto::Transaction::TransactionBody_oneof_data> for TransactionCryptoTransfer {
fn to_proto(&self) -> Result<proto::Transaction::TransactionBody_oneof_data, Error> {
let amounts: Result<Vec<proto::CryptoTransfer::AccountAmount>, Error> = self
.transfers
.iter()
.map(|(id, amount)| {
let mut pb = proto::CryptoTransfer::AccountAmount::new();
pb.set_accountID(id.to_proto()?);
pb.set_amount(*amount);
Ok(pb)
})
.collect();
let mut transfers = proto::CryptoTransfer::TransferList::new();
transfers.set_accountAmounts(RepeatedField::from_vec(amounts?));
let mut data = proto::CryptoTransfer::CryptoTransferTransactionBody::new();
data.set_transfers(transfers);
Ok(proto::Transaction::TransactionBody_oneof_data::cryptoTransfer(data))
}
}