Reset after end, and also, add GO_ONLINE and GO_OFFLINE commands

Bill13579 · Apr 27, 2020 · 6346bca · 6346bca
1 parent 221c2cc
commit 6346bca
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -70,6 +70,8 @@ impl Handler {
                 "S" => State::Trigger(Requirement::new(data), timeout),
                 "IMG" => State::DisplayImage(data, timeout),
                 "INPUT" => State::Trigger(Requirement::new(data), timeout),
+                "GO_OFFLINE" => State::GoOffline(timeout),
+                "GO_ONLINE" => State::GoOnline(timeout),
                 &_ => {
                     eprintln!("Invalid command: {}", cmd_type);
                     process::exit(1);
@@ -192,7 +194,16 @@ impl EventHandler for Handler {
                                     }
                                     timeout = *t;
                                 },
+                                State::GoOffline(t) => {
+                                    ctx.invisible();
+                                    timeout = *t;
+                                },
+                                State::GoOnline(t) => {
+                                    ctx.online();
+                                    timeout = *t;
+                                },
                                 State::End => {
+                                    *entry = None;
                                     break;
                                 }
                             }
@@ -235,6 +246,8 @@ pub enum State {
     Trigger(Requirement, f32),
     Display(String, f32),
     DisplayImage(String, f32),
+    GoOffline(f32),
+    GoOnline(f32),
     End,
 }