use async_trait::async_trait;
use hedera_proto::services;
use hedera_proto::services::token_service_client::TokenServiceClient;
use tonic::transport::Channel;
use crate::entity_id::AutoValidateChecksum;
use crate::protobuf::ToProtobuf;
use crate::transaction::{
AnyTransactionData,
ToTransactionDataProtobuf,
TransactionExecute,
};
use crate::{
AccountId,
Error,
LedgerId,
TokenId,
Transaction,
TransactionId,
};
pub type TokenPauseTransaction = Transaction<TokenPauseTransactionData>;
#[cfg_attr(feature = "ffi", serde_with::skip_serializing_none)]
#[derive(Debug, Clone, Default)]
#[cfg_attr(feature = "ffi", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "ffi", serde(rename_all = "camelCase"))]
pub struct TokenPauseTransactionData {
token_id: Option<TokenId>,
}
impl TokenPauseTransaction {
pub fn token_id(&mut self, token_id: impl Into<TokenId>) -> &mut Self {
self.body.data.token_id = Some(token_id.into());
self
}
}
#[async_trait]
impl TransactionExecute for TokenPauseTransactionData {
fn validate_checksums_for_ledger_id(&self, ledger_id: &LedgerId) -> Result<(), Error> {
self.token_id.validate_checksum_for_ledger_id(ledger_id)
}
async fn execute(
&self,
channel: Channel,
request: services::Transaction,
) -> Result<tonic::Response<services::TransactionResponse>, tonic::Status> {
TokenServiceClient::new(channel).pause_token(request).await
}
}
impl ToTransactionDataProtobuf for TokenPauseTransactionData {
fn to_transaction_data_protobuf(
&self,
_node_account_id: AccountId,
_transaction_id: &TransactionId,
) -> services::transaction_body::Data {
let token = self.token_id.to_protobuf();
services::transaction_body::Data::TokenPause(services::TokenPauseTransactionBody { token })
}
}
impl From<TokenPauseTransactionData> for AnyTransactionData {
fn from(transaction: TokenPauseTransactionData) -> Self {
Self::TokenPause(transaction)
}
}
#[cfg(test)]
mod tests {
#[cfg(feature = "ffi")]
mod ffi {
use assert_matches::assert_matches;
use crate::transaction::{
AnyTransaction,
AnyTransactionData,
};
use crate::{
TokenId,
TokenPauseTransaction,
};
const TOKEN_PAUSE_TRANSACTION_JSON: &str = r#"{
"$type": "tokenPause",
"tokenId": "0.0.1001"
}"#;
#[test]
fn it_should_serialize() -> anyhow::Result<()> {
let mut transaction = TokenPauseTransaction::new();
transaction.token_id(TokenId::from(1001));
let transaction_json = serde_json::to_string_pretty(&transaction)?;
assert_eq!(transaction_json, TOKEN_PAUSE_TRANSACTION_JSON);
Ok(())
}
#[test]
fn it_should_deserialize() -> anyhow::Result<()> {
let transaction: AnyTransaction = serde_json::from_str(TOKEN_PAUSE_TRANSACTION_JSON)?;
let data = assert_matches!(transaction.body.data, AnyTransactionData::TokenPause(transaction) => transaction);
assert_eq!(data.token_id.unwrap(), TokenId::from(1001));
Ok(())
}
}
}