From 04c68837d3fe77b84e8dee7282d5d15ce484c8bf Mon Sep 17 00:00:00 2001
From: Tim Pope <code@tpope.net>
Date: Thu, 4 Jul 2019 08:40:23 -0400
Subject: [PATCH] Fix stacking untracked files

---
 autoload/fugitive.vim | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/autoload/fugitive.vim b/autoload/fugitive.vim
index 9b66f68..2bc9046 100644
--- 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 = []