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
|
use std::time::SystemTime;
use modules::types::ModuleType;
#[derive(Serialize, Deserialize, Debug, Clone, PartialEq)]
pub enum ConstructionStatus {
None,
Constructing,
Constructed,
}
#[derive(Serialize, Deserialize, Debug, Clone)]
pub struct Construction {
pub status : ConstructionStatus,
construction : Option<ModuleType>,
time : u64,
start : Option<SystemTime>,
}
impl Construction {
pub fn new() -> Construction {
Construction {
status : ConstructionStatus::None,
construction : None,
time : 5,
start : None,
}
}
pub fn process(&mut self) {
if let Some(timer) = self.start.clone() {
if timer.elapsed().unwrap().as_secs() > self.time {
self.start = Some(SystemTime::now());
self.status = ConstructionStatus::Constructed;
}
}
}
pub fn toggle(&mut self) {
match self.status {
ConstructionStatus::None => self.on(),
_ => self.off(),
};
}
pub fn on(&mut self) {
self.start = Some(SystemTime::now());
self.status = ConstructionStatus::Constructing;
}
pub fn off(&mut self) {
self.start = None;
self.status = ConstructionStatus::None;
}
pub fn take(&mut self) {
self.off()
}
}
|