From 2d49302d0a2a27c0f951c526e6cac23811fb64e1 Mon Sep 17 00:00:00 2001
From: CrispyPin <crispin@tasa.se>
Date: Sat, 4 Mar 2023 12:43:54 +0100
Subject: [PATCH] improve selection controls

---
 src/editor.rs | 83 +++++++++++++++++++++++++++++++++------------------
 1 file changed, 54 insertions(+), 29 deletions(-)

diff --git a/src/editor.rs b/src/editor.rs
index 23ce325..f86f576 100644
--- a/src/editor.rs
+++ b/src/editor.rs
@@ -1,6 +1,6 @@
 use crossterm::{
 	cursor::{self, MoveTo},
-	event::{self, Event, KeyCode, KeyModifiers},
+	event::{self, Event, KeyCode, KeyEvent, KeyModifiers},
 	queue,
 	style::{Color, Colors, ResetColor, SetColors},
 	terminal::{self, Clear, ClearType},
@@ -89,34 +89,62 @@ impl Editor {
 
 	fn input(&mut self) {
 		match event::read() {
-			Ok(Event::Key(event)) => match event.modifiers {
-				KeyModifiers::NONE => match event.code {
-					KeyCode::Esc => self.active = false,
-					KeyCode::Char(ch) => self.insert_char(ch),
-					KeyCode::Backspace => self.backspace(),
-					KeyCode::Delete => self.delete(),
-					KeyCode::Left => self.move_left(),
-					KeyCode::Right => self.move_right(),
-					KeyCode::Up => self.move_up(),
-					KeyCode::Down => self.move_down(),
-					KeyCode::Home => self.move_home(),
-					KeyCode::End => self.move_end(),
+			Ok(Event::Key(event)) => {
+				if self.input_movement(&event) {
+					return;
+				}
+				match event.modifiers {
+					KeyModifiers::NONE => match event.code {
+						KeyCode::Esc => self.active = false,
+						KeyCode::Char(ch) => self.insert_char(ch),
+						KeyCode::Enter => self.insert_char('\n'),
+						KeyCode::Backspace => self.backspace(),
+						KeyCode::Delete => self.delete(),
+						_ => (),
+					},
+					KeyModifiers::CONTROL => match event.code {
+						KeyCode::Char('s') => self.save(),
+						KeyCode::Char('c') => self.copy(),
+						KeyCode::Char('x') => self.cut(),
+						KeyCode::Char('v') => self.paste(),
+						_ => (),
+					},
 					_ => (),
-				},
-				KeyModifiers::CONTROL => match event.code {
-					KeyCode::Char('s') => self.save(),
-					KeyCode::Char('p') => self.toggle_marker(),
-					KeyCode::Char('c') => self.copy(),
-					KeyCode::Char('x') => self.cut(),
-					KeyCode::Char('v') => self.paste(),
-					_ => (),
-				},
-				_ => (),
-			},
+				}
+			}
 			_ => (),
 		}
 	}
 
+	/// Cursor movement logic, returns true if cursor moved (so consider the event consumed in that case)
+	fn input_movement(&mut self, event: &KeyEvent) -> bool {
+		if let KeyCode::Left
+		| KeyCode::Right
+		| KeyCode::Up
+		| KeyCode::Down
+		| KeyCode::Home
+		| KeyCode::End = event.code
+		{
+			if event.modifiers.contains(KeyModifiers::SHIFT) {
+				self.set_marker();
+			} else {
+				self.marker = None;
+			}
+			match event.code {
+				KeyCode::Left => self.move_left(),
+				KeyCode::Right => self.move_right(),
+				KeyCode::Up => self.move_up(),
+				KeyCode::Down => self.move_down(),
+				KeyCode::Home => self.move_home(),
+				KeyCode::End => self.move_end(),
+				_ => (),
+			}
+			true
+		} else {
+			false
+		}
+	}
+
 	fn draw(&self) {
 		queue!(stdout(), Clear(ClearType::All)).unwrap();
 
@@ -244,10 +272,8 @@ impl Editor {
 		}
 	}
 
-	fn toggle_marker(&mut self) {
-		if self.marker.is_some() {
-			self.marker = None;
-		} else {
+	fn set_marker(&mut self) {
+		if self.marker.is_none() {
 			self.marker = Some(self.char_index());
 		}
 	}
@@ -282,7 +308,6 @@ impl Editor {
 
 	fn insert_char(&mut self, ch: char) {
 		self.unsaved_changes = true;
-		// eprintln!("inserting {ch} at {}", self.index());
 		self.text.insert(self.char_index(), ch);
 		self.find_lines();
 		self.move_right();