use crate::{
crypto::PublicKey,
proto::{self, ToProto, Transaction::TransactionBody_oneof_data},
transaction::Transaction,
Client, ErrorKind,
};
use chrono::{DateTime, Utc};
use failure::Error;
use query_interface::{interfaces, vtable_for};
use std::any::Any;
pub struct TransactionFileCreate {
expiration_time: Option<DateTime<Utc>>,
key: Option<PublicKey>,
bytes: Vec<u8>,
}
interfaces!(
TransactionFileCreate: dyn Any,
dyn ToProto<TransactionBody_oneof_data>
);
impl TransactionFileCreate {
pub fn new(client: &Client) -> Transaction<Self> {
Transaction::new(
client,
Self {
expiration_time: None,
key: None,
bytes: Vec::new(),
},
)
}
}
impl Transaction<TransactionFileCreate> {
#[inline]
pub fn expires_at(&mut self, expiration: DateTime<Utc>) -> &mut Self {
self.inner().expiration_time = Some(expiration);
self
}
#[inline]
pub fn key(&mut self, key: PublicKey) -> &mut Self {
self.inner().key = Some(key);
self
}
#[inline]
pub fn contents(&mut self, bytes: Vec<u8>) -> &mut Self {
self.inner().bytes = bytes;
self
}
}
impl ToProto<TransactionBody_oneof_data> for TransactionFileCreate {
fn to_proto(&self) -> Result<TransactionBody_oneof_data, Error> {
let mut data = proto::FileCreate::FileCreateTransactionBody::new();
let mut shard = proto::BasicTypes::ShardID::new();
shard.set_shardNum(0);
data.set_shardID(shard);
let mut realm = proto::BasicTypes::RealmID::new();
realm.set_realmNum(0);
data.set_realmID(realm);
if let Some(expiration_time) = &self.expiration_time.as_ref() {
data.set_expirationTime(expiration_time.to_proto()?);
}
let key = match self.key.as_ref() {
Some(key) => key,
None => Err(ErrorKind::MissingField("key"))?,
};
let mut key_list = proto::BasicTypes::KeyList::new();
key_list.keys.push(key.to_proto()?);
data.set_keys(key_list);
data.set_contents(self.bytes.clone());
Ok(TransactionBody_oneof_data::fileCreate(data))
}
}