From b08278807ee51e279c07dd6b7955713455a4c654 Mon Sep 17 00:00:00 2001
From: Blaž Hrastnik
Date: Sun, 13 Sep 2020 20:04:16 +0900
Subject: Add 'A', 'I' commands.

---
 helix-core/src/commands.rs | 78 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)

(limited to 'helix-core/src/commands.rs')

diff --git a/helix-core/src/commands.rs b/helix-core/src/commands.rs
index 836667a1..914ca0fe 100644
--- a/helix-core/src/commands.rs
+++ b/helix-core/src/commands.rs
@@ -60,8 +60,86 @@ pub fn append_mode(state: &mut State, _count: usize) {
     })
 }
 
+// TODO: I, A, o and O can share a lot of the primitives.
+
 // I inserts at the start of each line with a selection
+pub fn prepend_to_line(state: &mut State, _count: usize) {
+    state.mode = Mode::Insert;
+
+    // calculate line numbers for each selection range
+    let mut lines = state
+        .selection
+        .ranges()
+        .iter()
+        .map(|range| state.doc.char_to_line(range.head))
+        .collect::<Vec<_>>();
+
+    lines.sort();
+    lines.dedup();
+
+    let positions: Vec<_> = lines
+        .into_iter()
+        .map(|index| {
+            // adjust all positions to the end of the line.
+            let line_start = state.doc.line_to_char(index);
+            line_start
+        })
+        .collect();
+
+    let selection = Selection::new(
+        positions
+            .iter()
+            .copied()
+            .map(|pos| Range::new(pos, pos))
+            .collect(),
+        0,
+    );
+
+    let transaction = Transaction::new(state).with_selection(selection);
+
+    transaction.apply(state);
+    // TODO: need to store into history if successful
+}
+
 // A inserts at the end of each line with a selection
+pub fn append_to_line(state: &mut State, _count: usize) {
+    state.mode = Mode::Insert;
+
+    // calculate line numbers for each selection range
+    let mut lines = state
+        .selection
+        .ranges()
+        .iter()
+        .map(|range| state.doc.char_to_line(range.head))
+        .collect::<Vec<_>>();
+
+    lines.sort();
+    lines.dedup();
+
+    let positions: Vec<_> = lines
+        .into_iter()
+        .map(|index| {
+            // adjust all positions to the end of the line.
+            let line = state.doc.line(index);
+            let line_start = state.doc.line_to_char(index);
+            line_start + line.len_chars() - 1
+        })
+        .collect();
+
+    let selection = Selection::new(
+        positions
+            .iter()
+            .copied()
+            .map(|pos| Range::new(pos, pos))
+            .collect(),
+        0,
+    );
+
+    let transaction = Transaction::new(state).with_selection(selection);
+
+    transaction.apply(state);
+    // TODO: need to store into history if successful
+}
 
 // o inserts a new line after each line with a selection
 pub fn open_below(state: &mut State, _count: usize) {
-- 
cgit v1.2.3-70-g09d2