From a702af0aeb4562cbcdf8e2bc4008b8ce95da9a56 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bla=C5=BE=20Hrastnik?= <blaz@mxxn.io>
Date: Fri, 22 Jan 2021 16:31:49 +0900
Subject: [PATCH] commands: add W and B (extend selection by word).

---
 helix-term/src/commands.rs | 18 ++++++++++++++++++
 helix-term/src/keymap.rs   |  3 +++
 2 files changed, 21 insertions(+)

diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs
index 8e39972c..4a461239 100644
--- a/helix-term/src/commands.rs
+++ b/helix-term/src/commands.rs
@@ -161,6 +161,24 @@ pub fn move_file_end(cx: &mut Context) {
     cx.doc().mode = Mode::Normal;
 }
 
+pub fn extend_next_word_start(cx: &mut Context) {
+    let count = cx.count;
+    let selection = cx
+        .doc()
+        .state
+        .extend_selection(Direction::Forward, Granularity::Word, count);
+    cx.doc().set_selection(selection);
+}
+
+pub fn extend_prev_word_start(cx: &mut Context) {
+    let count = cx.count;
+    let selection = cx
+        .doc()
+        .state
+        .extend_selection(Direction::Backward, Granularity::Word, count);
+    cx.doc().set_selection(selection);
+}
+
 pub fn check_cursor_in_view(view: &View) -> bool {
     let doc = &view.doc;
     let cursor = doc.selection().cursor();
diff --git a/helix-term/src/keymap.rs b/helix-term/src/keymap.rs
index 132c81d0..440a3efb 100644
--- a/helix-term/src/keymap.rs
+++ b/helix-term/src/keymap.rs
@@ -146,8 +146,11 @@ pub fn default() -> Keymaps {
                 vec![shift!('K')] => commands::extend_line_up,
                 vec![shift!('L')] => commands::extend_char_right,
                 vec![key!('w')] => commands::move_next_word_start,
+                vec![shift!('W')] => commands::extend_next_word_start,
                 vec![key!('b')] => commands::move_prev_word_start,
+                vec![shift!('B')] => commands::extend_prev_word_start,
                 vec![key!('e')] => commands::move_next_word_end,
+                // TODO: E
                 vec![key!('g')] => commands::goto_mode,
                 vec![key!('i')] => commands::insert_mode,
                 vec![shift!('I')] => commands::prepend_to_line,