Fix cursor up/down with count from prompt line

Shougo · Dec 3, 2016 · ca618c4 · ca618c4
1 parent 3469ced
commit ca618c4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/autoload/unite/mappings.vim b/autoload/unite/mappings.vim
@@ -738,7 +738,8 @@ function! unite#mappings#cursor_up(is_skip_not_matched) abort "{{{
     return repeat("\<Up>", cnt) .
         \ (unite#helper#is_prompt(line('.') - cnt) ? "\<End>" : "\<Home>")
   else
-    return cnt == 1 ? 'k' : cnt.'k'
+    let cnt += v:count == 0 ? 0 : (v:count - 1)
+    return "\<Esc>" . (cnt == 1 ? 'k' : cnt.'k')
   endif
 endfunction"}}}
 function! unite#mappings#cursor_down(is_skip_not_matched) abort "{{{
@@ -767,7 +768,8 @@ function! unite#mappings#cursor_down(is_skip_not_matched) abort "{{{
     return repeat("\<Down>", cnt) .
           \ (unite#helper#is_prompt(line('.') + cnt) ? "\<End>" : "\<Home>")
   else
-    return cnt == 1 ? 'j' : cnt.'j'
+    let cnt += v:count == 0 ? 0 : (v:count - 1)
+    return "\<Esc>" . (cnt == 1 ? 'j' : cnt.'j')
   endif
 endfunction"}}}
 function! s:smart_preview() abort "{{{