From d093debfd598e4d3b2055be6d31fb9d8efa8df12 Mon Sep 17 00:00:00 2001
From: Leo <klesti.selimay@icloud.com>
Date: Thu, 17 Oct 2024 20:17:42 +0200
Subject: [PATCH] some new stuff

---
 src/key.rs | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100644 src/key.rs

diff --git a/src/key.rs b/src/key.rs
new file mode 100644
index 0000000..29ba932
--- /dev/null
+++ b/src/key.rs
@@ -0,0 +1,83 @@
+use std::io::{self, Read};
+
+#[derive(Debug, Clone, PartialEq)]
+pub enum KeyKind {
+    Enter,
+    Escape,
+    Up,
+    Down,
+    Left,
+    Right,
+    Char(String),
+}
+
+#[derive(Debug, Clone)]
+pub struct Key {
+    pub kind: KeyKind,
+    pub raw: String,
+    pub ctrl: bool,
+    pub shift: bool,
+}
+
+impl Key {
+    pub fn new(k: String) -> Key {
+        let mut ctrl = false;
+        let mut shift = false;
+        let kind = match k.as_str() {
+            "\r" => KeyKind::Enter,
+            "\x1b" => KeyKind::Escape,
+            "\x1b[A" => KeyKind::Up,
+            "\x1b[B" => KeyKind::Down,
+            "\x1b[C" => KeyKind::Right,
+            "\x1b[D" => KeyKind::Left,
+
+            // Ctrl keys
+            "\u{17}" => {
+                ctrl = true;
+                KeyKind::Char("w".to_string())
+            }
+
+            "\u{13}" => {
+                ctrl = true;
+                KeyKind::Char("s".to_string())
+            }
+
+            "\u{1}" => {
+                ctrl = true;
+                KeyKind::Char("a".to_string())
+            }
+
+            "\u{4}" => {
+                ctrl = true;
+                KeyKind::Char("d".to_string())
+            }
+
+            "\u{11}" => {
+                ctrl = true;
+                KeyKind::Char("q".to_string())
+            }
+
+            _ => {
+                shift = k.to_uppercase() == k;
+                KeyKind::Char(k.clone())
+            }
+        };
+        Key {
+            raw: k,
+            kind,
+            ctrl,
+            shift,
+        }
+    }
+}
+
+pub fn read_key() -> Key {
+    let mut buffer = vec![0; 3];
+    io::stdin().read(&mut buffer).unwrap();
+    Key::new(
+        String::from_utf8(buffer)
+            .unwrap()
+            .trim_matches('\0')
+            .to_string(),
+    )
+}