organized

src/app/mode.rs

@@ -0,0 +1,139 @@
+use std::fmt::Display;
+
+use ratatui::{
+    layout::Rect,
+    prelude,
+    widgets::{Paragraph, Widget},
+};
+
+use crate::app::app::App;
+
+pub enum Mode {
+    Insert(Editor),
+    Chord(Chord),
+    Normal,
+    Command(Chord),
+    Visual((usize, usize)),
+}
+
+impl Widget for &Mode {
+    fn render(self, area: Rect, buf: &mut prelude::Buffer) {
+        Paragraph::new(self.to_string()).render(area, buf);
+    }
+}
+
+impl Display for Mode {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Mode::Normal => write!(f, "-- NORMAL --"),
+            Mode::Insert(_) => write!(f, "-- INSERT --"),
+            Mode::Chord(_) => write!(f, "-- CHORD --"),
+            Mode::Command(_) => write!(f, "-- COMMAND --"),
+            Mode::Visual(_) => write!(f, "-- VISUAL --"),
+        }
+    }
+}
+
+impl Mode {
+    pub fn process_key(app: &mut App, key: char) {
+        match key {
+            // <
+            'h' => {
+                app.grid.selected_cell.0 = app.grid.selected_cell.0.saturating_sub(1);
+                return;
+            }
+            // v
+            'j' => {
+                app.grid.selected_cell.1 = app.grid.selected_cell.1.saturating_add(1);
+                return;
+            }
+            // ^
+            'k' => {
+                app.grid.selected_cell.1 = app.grid.selected_cell.1.saturating_sub(1);
+                return;
+            }
+            // >
+            'l' => {
+                app.grid.selected_cell.0 = app.grid.selected_cell.0.saturating_add(1);
+                return;
+            }
+            _ => {}
+        }
+
+        if let Mode::Normal = app.mode {
+            match key {
+                // edit cell
+                'i' | 'a' => {
+                    let (x, y) = app.grid.selected_cell;
+
+                    let val = app.grid.get_cell_raw(x, y).as_ref().map(|f| f.as_raw_string()).unwrap_or(String::new());
+
+                    app.mode = Mode::Insert(Editor::new(val, (x, y)));
+                }
+                'I' => { /* insert col before */ }
+                'A' => { /* insert col after */ }
+                'o' => { /* insert row below */ }
+                'O' => { /* insert row above */ }
+                'v' => app.mode = Mode::Visual(app.grid.selected_cell),
+                ':' => app.mode = Mode::Command(Chord::new(':')),
+                // loose chars will put you into chord mode
+                c => app.mode = Mode::Chord(Chord::new(c)),
+            }
+        }
+    }
+}
+
+pub struct Editor {
+    pub buf: String,
+    cursor: usize,
+    pub location: (usize, usize),
+}
+impl Editor {
+    fn new(value: String, loc: (usize, usize)) -> Self {
+        Self {
+            buf: value.to_string(),
+            cursor: value.len(),
+            location: loc,
+        }
+    }
+}
+
+impl Widget for &Editor {
+    fn render(self, area: prelude::Rect, buf: &mut prelude::Buffer) {
+        // TODO add visual cursor
+        Paragraph::new(self.buf.clone()).render(area, buf);
+    }
+}
+
+pub struct Chord {
+    buf: Vec<char>,
+}
+
+impl Chord {
+    pub fn new(inital: char) -> Self {
+        let mut buf = Vec::new();
+        buf.push(inital);
+
+        Self {
+            buf,
+        }
+    }
+
+    pub fn backspace(&mut self) {
+        self.buf.pop();
+    }
+
+    pub fn add_char(&mut self, c: char) {
+        self.buf.push(c)
+    }
+
+    pub fn as_string(&self) -> String {
+        self.buf.iter().collect()
+    }
+}
+
+impl Widget for &Chord {
+    fn render(self, area: prelude::Rect, buf: &mut prelude::Buffer) {
+        Paragraph::new(self.buf.iter().collect::<String>()).render(area, buf);
+    }
+}