catbuffer_rust/
account_address_restriction_transaction_builder.rs

1/*
2 * // Copyright (c) 2016-2019, Jaguar0625, gimre, BloodyRookie, Tech Bureau, Corp.
3 * // Copyright (c) 2020-present, Jaguar0625, gimre, BloodyRookie.
4 * // All rights reserved.
5 * //
6 * // This file is part of Catapult.
7 * //
8 * // Catapult is free software: you can redistribute it and/or modify
9 * // it under the terms of the GNU Lesser General Public License as published by
10 * // the Free Software Foundation, either version 3 of the License, or
11 * // (at your option) any later version.
12 * //
13 * // Catapult is distributed in the hope that it will be useful,
14 * // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * // GNU Lesser General Public License for more details.
17 * //
18 * // You should have received a copy of the GNU Lesser General Public License
19 * // along with Catapult. If not, see <http://www.gnu.org/licenses/>.
20 */
21
22use super::account_address_restriction_transaction_body_builder::*;
23use super::account_restriction_flags_dto::*;
24use super::amount_dto::*;
25use super::entity_type_dto::*;
26use super::generator_utils::*;
27use super::key_dto::*;
28use super::network_type_dto::*;
29use super::signature_dto::*;
30use super::timestamp_dto::*;
31use super::transaction_builder::*;
32use super::unresolved_address_dto::*;
33
34/// Binary layout for a non-embedded account address restriction transaction.
35#[derive(Debug, Clone)]
36pub struct AccountAddressRestrictionTransactionBuilder {
37    /// Transaction.
38    pub super_object: TransactionBuilder,
39    /// Account address restriction transaction body.
40    pub body: AccountAddressRestrictionTransactionBodyBuilder,
41}
42
43impl AccountAddressRestrictionTransactionBuilder {
44    const VERSION: u8 = 1;
45    const ENTITY_TYPE: u16 = 0x4150;
46
47
48    /// Creates an instance of AccountAddressRestrictionTransactionBuilder from binary payload.
49    /// payload: Byte payload to use to serialize the object.
50    /// # Returns
51    /// A AccountAddressRestrictionTransactionBuilder.
52    pub fn from_binary(payload: &[u8]) -> Self {
53        let mut _bytes = payload.to_vec();
54        let super_object = TransactionBuilder::from_binary(&_bytes);
55        assert_eq!(Self::VERSION, super_object.version, "Invalid entity version ({})", super_object.version);
56        assert_eq!(Self::ENTITY_TYPE, super_object._type.get_value(), "Invalid entity type ({:?})", super_object._type);
57        let mut _bytes = _bytes[super_object.get_size()..].to_vec();
58        let account_address_restriction_transaction_body = AccountAddressRestrictionTransactionBodyBuilder::from_binary(&_bytes); // kind:CUSTOM1
59        _bytes = _bytes[account_address_restriction_transaction_body.get_size()..].to_vec();
60        // create object and call.
61        AccountAddressRestrictionTransactionBuilder { super_object, body: account_address_restriction_transaction_body }  // Transaction
62    }
63
64
65    pub fn get_restriction_flags(&self) -> Vec<AccountRestrictionFlagsDto> {
66        self.body.restriction_flags.clone()
67    }
68    pub fn set_restriction_flags(&mut self, restriction_flags: Vec<AccountRestrictionFlagsDto>) {
69        self.body.restriction_flags = restriction_flags;   // MARKER1 AttributeKind.FLAGS
70    }
71
72
73    pub fn get_restriction_additions(&self) -> Vec<UnresolvedAddressDto> {
74        self.body.restriction_additions.clone()
75    }
76
77    pub fn get_restriction_deletions(&self) -> Vec<UnresolvedAddressDto> {
78        self.body.restriction_deletions.clone()
79    }
80    /// Gets the size of the type.
81    ///
82    /// Returns:
83    /// A size in bytes.
84    pub fn get_size(&self) -> usize {
85        let mut size = self.super_object.get_size();
86        size += self.body.get_size();
87        size
88    }
89
90    /// Serializes self to bytes.
91    ///
92    /// # Returns
93    /// A Serialized bytes.
94    pub fn serializer(&self) -> Vec<u8> {
95        let mut buf: Vec<u8> = vec![];
96        buf.append(&mut (self.get_size() as u32).to_le_bytes().to_vec());
97        buf.append(&mut self.super_object.serializer());
98        buf.append(&mut self.body.serializer()); // kind:CUSTOM TransactionBody
99        buf
100    }
101}
102