From 8787162d7c973fad5a737172be81d7cafe9348ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= <gustav@sornas.net>
Date: Wed, 28 Apr 2021 23:36:05 +0200
Subject: move main loop to state

---
 src/state/mod.rs | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'src/state/mod.rs')

diff --git a/src/state/mod.rs b/src/state/mod.rs
index 5313b4f..7b498f2 100644
--- a/src/state/mod.rs
+++ b/src/state/mod.rs
@@ -1,7 +1,29 @@
 mod threads;
 
+use std::io::{Stdin, Write};
+use termion::event::Key;
+use termion::input::TermRead;
+
 pub use threads::Threads;
 
 pub enum State {
     Threads(Threads),
 }
+
+impl State {
+    pub fn run<W: Write>(mut self, mut screen: W, stdin: Stdin) {
+        for c in stdin.keys() {
+            let c = c.unwrap();
+            // Global keybinds
+            match c {
+                Key::Char('q') => break,
+                _ => ()
+            }
+
+            // Pass to current state
+            self = match self {
+                State::Threads(s) => s.tick(&mut screen, c),
+            };
+        }
+    }
+}
-- 
cgit v1.2.1