diff options
Diffstat (limited to 'src/modules/refinery.rs')
-rw-r--r-- | src/modules/refinery.rs | 21 |
1 files changed, 9 insertions, 12 deletions
diff --git a/src/modules/refinery.rs b/src/modules/refinery.rs index 177f2c5..cdb5de0 100644 --- a/src/modules/refinery.rs +++ b/src/modules/refinery.rs @@ -33,25 +33,22 @@ impl Refinery { } _ => (), } - if self.status == RefineryStatus::None { - self.start = None; - } } pub fn toggle(&mut self) { - self.status = match self.status { - RefineryStatus::None => { - self.start = Some(SystemTime::now()); - RefineryStatus::Refining - }, - _ => { - self.start = None; - RefineryStatus::None - } + match self.status { + RefineryStatus::None => self.on(), + _ => self.off(), }; } + pub fn on(&mut self) { + self.start = Some(SystemTime::now()); + self.status = RefineryStatus::Refining; + } + pub fn off(&mut self) { + self.start = None; self.status = RefineryStatus::None; } |