# HG changeset patch
# User Matt Mackall <mpm@selenic.com>
# Date 1375396389 18000
# Node ID fe78eb7bcca046143e9f6e78da3621cbd2a19354
# Parent  14c91b18d79877641925f6d3d437083e1e0f8c77
rebase: don't clobber wd on --abort when we've updated away (issue4009)

diff -r 14c91b18d798 -r fe78eb7bcca0 hgext/rebase.py
--- a/hgext/rebase.py	Fri Jul 26 13:34:51 2013 -0700
+++ b/hgext/rebase.py	Thu Aug 01 17:33:09 2013 -0500
@@ -592,6 +592,18 @@
             raise
         raise util.Abort(_('no rebase in progress'))
 
+def inrebase(repo, originalwd, state):
+    '''check whether the workdir is in an interrupted rebase'''
+    parents = [p.rev() for p in repo.parents()]
+    if originalwd in parents:
+        return True
+
+    for newrev in state.itervalues():
+        if newrev in parents:
+            return True
+
+    return False
+
 def abort(repo, originalwd, target, state):
     'Restore the repository to its original state'
     dstates = [s for s in state.values() if s != nullrev]
@@ -609,8 +621,11 @@
                        "can't abort\n"))
         return -1
     else:
+        # Update away from the rebase if necessary
+        if inrebase(repo, originalwd, state):
+            merge.update(repo, repo[originalwd].rev(), False, True, False)
+
         # Strip from the first rebased revision
-        merge.update(repo, repo[originalwd].rev(), False, True, False)
         rebased = filter(lambda x: x > -1 and x != target, state.values())
         if rebased:
             strippoints = [c.node()  for c in repo.set('roots(%ld)', rebased)]