Support count prefixes

matze · Aug 9, 2013 · 8f58d3b · 8f58d3b
1 parent ce0a2b4
commit 8f58d3b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/doc/move.txt b/doc/move.txt
@@ -23,6 +23,9 @@ To enable custom key maps you must disable the automatic key maps with >
 
     let g:move_map_keys = 0
 
+All mappings can be prefixed with a {count} and will move {count} steps
+instead of one.
+
 -------------------------------------------------------------------------------
 2.1 <Plug>MoveBlockDown
 

diff --git a/plugin/move.vim b/plugin/move.vim
@@ -24,6 +24,10 @@ endfunction
 function! s:MoveBlockDown() range
     let next_line = a:lastline + 1
 
+    if v:count > 0
+        let next_line = next_line + v:count - 1
+    endif
+
     if next_line > line('$')
         call s:ResetCursor()
         return
@@ -36,6 +40,10 @@ endfunction
 function! s:MoveBlockUp() range
     let prev_line = a:firstline - 2
 
+    if v:count > 0
+        let prev_line = prev_line - v:count + 1
+    endif
+
     if prev_line < 0
         call s:ResetCursor()
         return