1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
/*
* // Copyright (c) 2016-2019, Jaguar0625, gimre, BloodyRookie, Tech Bureau, Corp.
* // Copyright (c) 2020-present, Jaguar0625, gimre, BloodyRookie.
* // All rights reserved.
* //
* // This file is part of Catapult.
* //
* // Catapult is free software: you can redistribute it and/or modify
* // it under the terms of the GNU Lesser General Public License as published by
* // the Free Software Foundation, either version 3 of the License, or
* // (at your option) any later version.
* //
* // Catapult is distributed in the hope that it will be useful,
* // but WITHOUT ANY WARRANTY; without even the implied warranty of
* // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* // GNU Lesser General Public License for more details.
* //
* // You should have received a copy of the GNU Lesser General Public License
* // along with Catapult. If not, see <http://www.gnu.org/licenses/>.
*/
use num_derive::{FromPrimitive, ToPrimitive};
use num_traits::{FromPrimitive, ToPrimitive};
use strum::IntoEnumIterator;
use strum_macros::EnumIter;
use super::generator_utils::*;
/// Enumeration of account restriction flags.
#[allow(non_camel_case_types)]
#[repr(u16)]
#[derive(Debug, Clone, Copy, PartialEq, FromPrimitive, ToPrimitive, EnumIter)]
pub enum AccountRestrictionFlagsDto {
/// Restriction type is an address.
ADDRESS = 1,
/// Restriction type is a mosaic identifier.
MOSAIC_ID = 2,
/// Restriction type is a transaction type.
TRANSACTION_TYPE = 4,
/// Restriction is interpreted as outgoing.
OUTGOING = 16384,
/// Restriction is interpreted as blocking (instead of allowing) operation.
BLOCK = 32768,
}
impl AccountRestrictionFlagsDto {
pub const LENGTH: usize = std::mem::size_of::<Self>();
/// Gets the size of the type.
///
/// # Returns
///
/// A usize.
pub fn get_size(&self) -> usize {
Self::LENGTH
}
/// Gets the value of the enum.
///
/// # Returns
///
/// A u16
pub fn get_value(&self) -> u16 {
self.to_u16().unwrap()
}
/// Converts a bit representation to a vec of AccountRestrictionFlagsDto.
///
/// # Returns
///
/// A vec to AccountRestrictionFlagsDto flags representing the int value.
pub fn bytes_to_flags(bit_mask_value: &[u8]) -> Vec<AccountRestrictionFlagsDto> {
use std::convert::TryFrom;
let bit_mask_value = <[u8; Self::LENGTH]>::try_from(&bit_mask_value[..Self::LENGTH]).unwrap();
Self::int_to_flags(u16::from_le_bytes(bit_mask_value))
}
/// Converts a bit representation to a vec of AccountRestrictionFlagsDto.
///
/// # Returns
///
/// A vec to AccountRestrictionFlagsDto flags representing the int value.
pub fn int_to_flags(bit_mask_value: u16) -> Vec<AccountRestrictionFlagsDto> {
let mut results: Vec<AccountRestrictionFlagsDto> = vec![];
for flag in AccountRestrictionFlagsDto::iter() {
if 0 != flag.get_value() & bit_mask_value {
results.push(flag);
}
}
results
}
/// Converts a vec of AccountRestrictionFlagsDto to a bit representation.
///
/// # Returns
///
/// A u16 value of the vec of flags
pub fn flags_to_int(flags: Vec<AccountRestrictionFlagsDto>) -> u16 {
let mut result: u16 = 0;
for flag in AccountRestrictionFlagsDto::iter() {
if flags.iter().any(|&i| i == flag) {
result += flag.get_value();
}
}
result
}
/// Creates an `AccountRestrictionFlagsDto` from a slice.
///
/// # Returns
///
/// A `AccountRestrictionFlagsDto`.
pub fn from_binary(src: &[u8]) -> Self {
// assert_eq!(src.len(), Self::LENGTH);
let buf = fixed_bytes::<{ Self::LENGTH }>(src);
Self::from_u16(u16::from_le_bytes(buf)).unwrap()
}
/// Serializes an type to bytes.
///
/// # Returns
///
/// A Serialized bytes.
pub fn serializer(&self) -> Vec<u8> {
self.get_value().to_le_bytes().to_vec()
}
}