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
use bee_common::packable::{Packable, Read, Write};
use bitflags::bitflags;
use serde::Serialize;
bitflags! {
#[derive(Default, Serialize)]
pub struct Flags: u8 {
const SOLID = 0b0000_0001;
const MILESTONE = 0b0000_0010;
const REFERENCED = 0b0000_0100;
const VALID = 0b0000_1000;
const REQUESTED = 0b0001_0000;
}
}
impl Flags {
pub fn is_solid(&self) -> bool {
self.contains(Flags::SOLID)
}
pub fn set_solid(&mut self, is_solid: bool) {
self.set(Flags::SOLID, is_solid);
}
pub fn is_milestone(&self) -> bool {
self.contains(Flags::MILESTONE)
}
pub fn set_milestone(&mut self, is_milestone: bool) {
self.set(Flags::MILESTONE, is_milestone);
}
pub fn is_referenced(&self) -> bool {
self.contains(Flags::REFERENCED)
}
pub fn set_referenced(&mut self, is_referenced: bool) {
self.set(Flags::REFERENCED, is_referenced);
}
pub fn is_valid(&self) -> bool {
self.contains(Flags::VALID)
}
pub fn set_valid(&mut self, is_valid: bool) {
self.set(Flags::VALID, is_valid);
}
pub fn was_requested(&self) -> bool {
self.contains(Flags::REQUESTED)
}
pub fn set_requested(&mut self, was_requested: bool) {
self.set(Flags::REQUESTED, was_requested);
}
}
impl Packable for Flags {
type Error = std::io::Error;
fn packed_len(&self) -> usize {
self.bits().packed_len()
}
fn pack<W: Write>(&self, writer: &mut W) -> Result<(), Self::Error> {
self.bits().pack(writer)
}
fn unpack_inner<R: Read + ?Sized, const CHECK: bool>(reader: &mut R) -> Result<Self, Self::Error> {
Ok(unsafe { Self::from_bits_unchecked(u8::unpack_inner::<R, CHECK>(reader)?) })
}
}