Fully implemented vi-style editor commands

2025-05-19 16:08:21 -04:00
parent 4af31ba8ef
commit b30de3232b
6 changed files with 1076 additions and 324 deletions
--- a/src/prompt/readline/keys.rs
+++ b/src/prompt/readline/keys.rs
@@ -0,0 +1,111 @@
+// Credit to Rustyline for the design ideas in this module
+// https://github.com/kkawakam/rustyline
+#[derive(Clone,Debug)]
+pub struct KeyEvent(pub KeyCode, pub ModKeys);
+
+impl KeyEvent {
+	pub fn new(ch: char, mut mods: ModKeys) -> Self {
+		use {KeyCode as K, KeyEvent as E, ModKeys as M};
+
+		if !ch.is_control() {
+			if !mods.is_empty() {
+				mods.remove(M::SHIFT); // TODO Validate: no SHIFT even if
+															 // `c` is uppercase
+			}
+			return E(K::Char(ch), mods);
+		}
+		match ch {
+			'\x00' => E(K::Char('@'), mods | M::CTRL), // '\0'
+			'\x01' => E(K::Char('A'), mods | M::CTRL),
+			'\x02' => E(K::Char('B'), mods | M::CTRL),
+			'\x03' => E(K::Char('C'), mods | M::CTRL),
+			'\x04' => E(K::Char('D'), mods | M::CTRL),
+			'\x05' => E(K::Char('E'), mods | M::CTRL),
+			'\x06' => E(K::Char('F'), mods | M::CTRL),
+			'\x07' => E(K::Char('G'), mods | M::CTRL), // '\a'
+			'\x08' => E(K::Backspace, mods), // '\b'
+			'\x09' => {
+				// '\t'
+				if mods.contains(M::SHIFT) {
+					mods.remove(M::SHIFT);
+					E(K::BackTab, mods)
+				} else {
+					E(K::Tab, mods)
+				}
+			}
+			'\x0a' => E(K::Char('J'), mods | M::CTRL), // '\n' (10)
+			'\x0b' => E(K::Char('K'), mods | M::CTRL),
+			'\x0c' => E(K::Char('L'), mods | M::CTRL),
+			'\x0d' => E(K::Enter, mods), // '\r' (13)
+			'\x0e' => E(K::Char('N'), mods | M::CTRL),
+			'\x0f' => E(K::Char('O'), mods | M::CTRL),
+			'\x10' => E(K::Char('P'), mods | M::CTRL),
+			'\x11' => E(K::Char('Q'), mods | M::CTRL),
+			'\x12' => E(K::Char('R'), mods | M::CTRL),
+			'\x13' => E(K::Char('S'), mods | M::CTRL),
+			'\x14' => E(K::Char('T'), mods | M::CTRL),
+			'\x15' => E(K::Char('U'), mods | M::CTRL),
+			'\x16' => E(K::Char('V'), mods | M::CTRL),
+			'\x17' => E(K::Char('W'), mods | M::CTRL),
+			'\x18' => E(K::Char('X'), mods | M::CTRL),
+			'\x19' => E(K::Char('Y'), mods | M::CTRL),
+			'\x1a' => E(K::Char('Z'), mods | M::CTRL),
+			'\x1b' => E(K::Esc, mods), // Ctrl-[, '\e'
+			'\x1c' => E(K::Char('\\'), mods | M::CTRL),
+			'\x1d' => E(K::Char(']'), mods | M::CTRL),
+			'\x1e' => E(K::Char('^'), mods | M::CTRL),
+			'\x1f' => E(K::Char('_'), mods | M::CTRL),
+			'\x7f' => E(K::Backspace, mods), // Rubout, Ctrl-?
+			'\u{9b}' => E(K::Esc, mods | M::SHIFT),
+			_ => E(K::Null, mods),
+		}
+	}
+}
+
+#[derive(Clone,Debug)]
+pub enum KeyCode {
+    UnknownEscSeq,
+    Backspace,
+    BackTab,
+    BracketedPasteStart,
+    BracketedPasteEnd,
+    Char(char),
+    Delete,
+    Down,
+    End,
+    Enter,
+    Esc,
+    F(u8),
+    Home,
+    Insert,
+    Left,
+    Null,
+    PageDown,
+    PageUp,
+    Right,
+    Tab,
+    Up,
+}
+
+bitflags::bitflags! {
+	#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
+	pub struct ModKeys: u8 {
+		/// Control modifier
+		const CTRL  = 1<<3;
+		/// Escape or Alt modifier
+		const ALT  = 1<<2;
+		/// Shift modifier
+		const SHIFT = 1<<1;
+
+		/// No modifier
+		const NONE = 0;
+		/// Ctrl + Shift
+		const CTRL_SHIFT = Self::CTRL.bits() | Self::SHIFT.bits();
+		/// Alt + Shift
+		const ALT_SHIFT = Self::ALT.bits() | Self::SHIFT.bits();
+		/// Ctrl + Alt
+		const CTRL_ALT = Self::CTRL.bits() | Self::ALT.bits();
+		/// Ctrl + Alt + Shift
+		const CTRL_ALT_SHIFT = Self::CTRL.bits() | Self::ALT.bits() | Self::SHIFT.bits();
+	}
+}