Visual mode mappings

AndrewRadev · Nov 24, 2012 · 0411017 · 0411017
1 parent c154bcf
commit 0411017
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/autoload/whitespaste.vim b/autoload/whitespaste.vim
@@ -1,4 +1,8 @@
-function! whitespaste#Paste(normal_command)
+function! whitespaste#Paste(normal_command, visual)
+  if a:visual
+    normal! gv
+  endif
+
   if getregtype() == 'V'
     call whitespaste#PasteLinewise(a:normal_command)
   elseif getregtype() == 'v'

diff --git a/plugin/whitespaste.vim b/plugin/whitespaste.vim
@@ -44,11 +44,13 @@ autocmd FileType vim let b:whitespaste_linewise_definitions = {
       \   ]
       \ }
 
-command WhitespasteBefore :call whitespaste#Paste('P')
+command! -count=0 WhitespasteBefore :call whitespaste#Paste('P', <count>)
-command WhitespasteAfter  :call whitespaste#Paste('p')
+command! -count=0 WhitespasteAfter  :call whitespaste#Paste('p', <count>)
 
 nmap <Plug>WhitespasteBefore :WhitespasteBefore<cr>
 nmap <Plug>WhitespasteAfter  :WhitespasteAfter<cr>
+xmap <Plug>WhitespasteBefore :WhitespasteBefore<cr>
+xmap <Plug>WhitespasteAfter  :WhitespasteAfter<cr>
 
 let &cpo = s:keepcpo
 unlet s:keepcpo