From b3f91a149b40d0e02635d6578ccee133993c8303 Mon Sep 17 00:00:00 2001
From: Steven Walter <stevenrwalter@gmail.com>
Date: Mon, 30 Jun 2008 15:32:07 -0400
Subject: [PATCH] Fixup unedit

Rename to revert, add -a option
---
 README     |  6 ++++++
 yap/yap.py | 14 ++++++++++----
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/README b/README
index 5c61f6e..a3cc1b1 100644
--- a/README
+++ b/README
@@ -29,9 +29,15 @@
             * -f to remove leaf branch
         * switch
         * point
+
+TODO:
+    * Commands
         * push
         * fetch
         * merge
         * resolved
         * remote (list/create)
             * -d to remove
+    * Use pager for diff
+    * Support multiple files as args
+    * Support revert -a
diff --git a/yap/yap.py b/yap/yap.py
index 7329bb6..31ea917 100644
--- a/yap/yap.py
+++ b/yap/yap.py
@@ -163,10 +163,16 @@ class Yap(object):
         if not files:
             print "\t(none)"
 
-    def cmd_unedit(self, file):
-        "<file>"
-        self._assert_file_exists(file)
-        os.system("git checkout-index -f '%s'" % file)
+    @takes_options("a")
+    def cmd_revert(self, file=None, **flags):
+        "(-a | <file>)"
+        if '-a' in flags:
+            os.system("git checkout-index -f -a")
+        elif file is not None:
+            self._assert_file_exists(file)
+            os.system("git checkout-index -f '%s'" % file)
+        else:
+            raise TypeError
         self.cmd_status()
 
     @takes_options("ad")
-- 
2.11.4.GIT