From 1629f59ff44b517fd0e738aca41d5657d083d842 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Gustav=20H=C3=A5llberg?= <gustav@virtutech.com>
Date: Fri, 14 Aug 2009 10:54:03 +0200
Subject: [PATCH] stgit.el: Allow revert on index and work tree
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Gustav HÃ¥llberg <gustav@virtutech.com>
---
 contrib/stgit.el | 42 ++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 40 insertions(+), 2 deletions(-)

diff --git a/contrib/stgit.el b/contrib/stgit.el
index 006c082..e91bd9b 100644
--- a/contrib/stgit.el
+++ b/contrib/stgit.el
@@ -168,6 +168,10 @@ Returns nil if there was no output."
   "Returns non-nil if the index contains no changes from HEAD."
   (zerop (stgit-run-git-silent "diff-index" "--cached" "--quiet" "HEAD")))
 
+(defun stgit-work-tree-empty-p ()
+  "Returns non-nil if the work tree contains no changes from index."
+  (zerop (stgit-run-git-silent "diff-files" "--quiet")))
+
 (defvar stgit-index-node)
 (defvar stgit-worktree-node)
 
@@ -817,7 +821,7 @@ file for (applied) copies and renames."
             ("c" .        stgit-new-and-refresh)
             ("\C-c\C-c" . stgit-commit)
             ("\C-c\C-u" . stgit-uncommit)
-            ("U" .        stgit-revert-file)
+            ("U" .        stgit-revert)
             ("R" .        stgit-resolve-file)
             ("\r" .       stgit-select)
             ("+" .        stgit-expand)
@@ -883,6 +887,8 @@ Commands for patches:
 \\[stgit-edit]	Edit patch description
 \\[stgit-delete]	Delete patch(es)
 
+\\[stgit-revert]	Revert all changes in index or work tree
+
 \\[stgit-push-next]	Push next patch onto stack
 \\[stgit-pop-next]	Pop current patch from stack
 \\[stgit-push-or-pop]	Push or pop patch at point
@@ -900,7 +906,7 @@ Commands for files:
 \\[stgit-diff]	Show the file's diff
 
 \\[stgit-file-toggle-index]	Toggle change between index and work tree
-\\[stgit-revert-file]	Revert changes to file
+\\[stgit-revert]	Revert changes to file
 
 Display commands:
 \\[stgit-toggle-worktree]	Toggle showing index and work tree
@@ -1219,6 +1225,38 @@ working tree."
         (stgit-reload)
         (stgit-goto-patch patch-name next-file)))))
 
+(defun stgit-revert ()
+  "Revert the change at point, which must be the index, the work
+tree, or a single change in either."
+  (interactive)
+  (let ((patched-file (stgit-patched-file-at-point)))
+    (if patched-file
+        (stgit-revert-file)
+      (let* ((patch-name (or (stgit-patch-name-at-point)
+                             (error "No patch or file at point")))
+             (patch-desc (case patch-name
+                           (:index "index")
+                           (:work  "work tree")
+                           (t (error (substitute-command-keys
+                                      "Use \\[stgit-delete] to delete a patch"))))))
+        (when (if (eq patch-name :work)
+                  (stgit-work-tree-empty-p)
+                (stgit-index-empty-p))
+          (error (format "There are no changes in the %s to revert"
+                         patch-desc)))
+        (and (eq patch-name :index)
+             (not (stgit-work-tree-empty-p))
+             (error "Cannot revert index as work tree contains unstaged changes"))
+
+        (when (yes-or-no-p (format "Revert all changes in the %s? "
+                                   patch-desc))
+          (if (eq patch-name :index)
+              (stgit-run-git-silent "reset" "--hard" "-q")
+            (stgit-run-git-silent "checkout" "--" "."))
+          (stgit-refresh-index)
+          (stgit-refresh-worktree)
+          (stgit-goto-patch patch-name))))))
+
 (defun stgit-resolve-file ()
   "Resolve conflict in the file at point."
   (interactive)
-- 
2.11.0