catbuffer_rust/
secret_proof_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::amount_dto::*;
23use super::entity_type_dto::*;
24use super::generator_utils::*;
25use super::hash256_dto::*;
26use super::key_dto::*;
27use super::lock_hash_algorithm_dto::*;
28use super::network_type_dto::*;
29use super::secret_proof_transaction_body_builder::*;
30use super::signature_dto::*;
31use super::timestamp_dto::*;
32use super::transaction_builder::*;
33use super::unresolved_address_dto::*;
34
35/// Binary layout for a non-embedded secret proof transaction.
36#[derive(Debug, Clone)]
37pub struct SecretProofTransactionBuilder {
38    /// Transaction.
39    pub super_object: TransactionBuilder,
40    /// Secret proof transaction body.
41    pub body: SecretProofTransactionBodyBuilder,
42}
43
44impl SecretProofTransactionBuilder {
45    const VERSION: u8 = 1;
46    const ENTITY_TYPE: u16 = 0x4252;
47
48
49    /// Creates an instance of SecretProofTransactionBuilder from binary payload.
50    /// payload: Byte payload to use to serialize the object.
51    /// # Returns
52    /// A SecretProofTransactionBuilder.
53    pub fn from_binary(payload: &[u8]) -> Self {
54        let mut _bytes = payload.to_vec();
55        let super_object = TransactionBuilder::from_binary(&_bytes);
56        assert_eq!(Self::VERSION, super_object.version, "Invalid entity version ({})", super_object.version);
57        assert_eq!(Self::ENTITY_TYPE, super_object._type.get_value(), "Invalid entity type ({:?})", super_object._type);
58        let mut _bytes = _bytes[super_object.get_size()..].to_vec();
59        let secret_proof_transaction_body = SecretProofTransactionBodyBuilder::from_binary(&_bytes); // kind:CUSTOM1
60        _bytes = _bytes[secret_proof_transaction_body.get_size()..].to_vec();
61        // create object and call.
62        SecretProofTransactionBuilder { super_object, body: secret_proof_transaction_body }  // Transaction
63    }
64
65
66    pub fn get_recipient_address(&self) -> UnresolvedAddressDto {
67        self.body.recipient_address.clone()
68    }
69    pub fn set_recipient_address(&mut self, recipient_address: UnresolvedAddressDto) {
70        self.body.recipient_address = recipient_address;   // MARKER1 AttributeKind.CUSTOM
71    }
72
73
74    pub fn get_secret(&self) -> Hash256Dto {
75        self.body.secret.clone()
76    }
77    pub fn set_secret(&mut self, secret: Hash256Dto) {
78        self.body.secret = secret;   // MARKER1 AttributeKind.CUSTOM
79    }
80
81
82    pub fn get_hash_algorithm(&self) -> LockHashAlgorithmDto {
83        self.body.hash_algorithm.clone()
84    }
85    pub fn set_hash_algorithm(&mut self, hash_algorithm: LockHashAlgorithmDto) {
86        self.body.hash_algorithm = hash_algorithm;   // MARKER1 AttributeKind.CUSTOM
87    }
88
89
90    pub fn get_proof(&self) -> Vec<u8> {
91        self.body.proof.clone()
92    }
93    pub fn set_proof(&mut self, proof: Vec<u8>) {
94        self.body.proof = proof;   // MARKER1 AttributeKind.BUFFER
95    }
96
97    /// Gets the size of the type.
98    ///
99    /// Returns:
100    /// A size in bytes.
101    pub fn get_size(&self) -> usize {
102        let mut size = self.super_object.get_size();
103        size += self.body.get_size();
104        size
105    }
106
107    /// Serializes self to bytes.
108    ///
109    /// # Returns
110    /// A Serialized bytes.
111    pub fn serializer(&self) -> Vec<u8> {
112        let mut buf: Vec<u8> = vec![];
113        buf.append(&mut (self.get_size() as u32).to_le_bytes().to_vec());
114        buf.append(&mut self.super_object.serializer());
115        buf.append(&mut self.body.serializer()); // kind:CUSTOM TransactionBody
116        buf
117    }
118}
119