motiejus/dotfiles

commit 04c68837d3fe77b84e8dee7282d5d15ce484c8bf (tree)
parent a9572e9ea6ddc76ac364d5f18595b3366d9561ce
Author: Tim Pope <code@tpope.net>
Date:   Thu,  4 Jul 2019 08:40:23 -0400

Fix stacking untracked files

Diffstat:
M
autoload/fugitive.vim
 | 
22
++++++++++++++++++----

1 file changed, 18 insertions(+), 4 deletions(-)
diff --git a/autoload/fugitive.vim b/autoload/fugitive.vim
@@ -2605,6 +2605,15 @@ function! s:DoStageUnstagedHeading(heading) abort
   return s:DoToggleUnstagedHeading(a:heading)
 endfunction
 
+function! s:DoToggleUntrackedHeading(...) abort
+  call s:TreeChomp('add', '.')
+  return 1
+endfunction
+
+function! s:DoStageUntrackedHeading(heading) abort
+  return s:DoToggleUntrackedHeading(a:heading)
+endfunction
+
 function! s:DoToggleStaged(record) abort
   if a:record.patch
     return s:StageApply(a:record, 1, ['--cached'])
@@ -2614,10 +2623,6 @@ function! s:DoToggleStaged(record) abort
   endif
 endfunction
 
-function! s:DoStageStaged(record) abort
-  return -1
-endfunction
-
 function! s:DoUnstageStaged(record) abort
   return s:DoToggleStaged(a:record)
 endfunction
@@ -2644,6 +2649,15 @@ function! s:DoUnstageUnstaged(record) abort
   endif
 endfunction
 
+function! s:DoToggleUntracked(record) abort
+  call s:TreeChomp(['add', '--'] + a:record.paths)
+  return 1
+endfunction
+
+function! s:DoStageUntracked(record) abort
+  return s:DoToggleUntracked(a:record)
+endfunction
+
 function! s:StagePatch(lnum1,lnum2) abort
   let add = []
   let reset = []