# HG changeset patch
# User David Schleimer <dschleimer@fb.com>
# Date 1348507108 25200
# Node ID a9f315eae67ccb836710079efc4192b94f50a6d0
# Parent  dfb3afa6c6193ce6f0438f1af346f37de863fdc4
push: use native rebase instead of our rebase wrapper

Our rebase wrapper doesn't quite do the right thing with repect to a
just-pushed revision.  In particular, it will try to rebase the
just-pushed revision on top of the version of that revision we just
pulled down from svn.  This will sometimes result in a local revision
with an identical commit message as the revision from svn, but no file
changes.

This changes the rebase portion of the push command to instead use the
native rebase with a revset that excludes the revision we just pushed
to svn from the set to be rebased.  It also moves to a single strip
operation that removes all of the revisions based on a pre-push or
partially pushed revision.  This moves to a separate rebase and strip
operation since we now need to strip revisions we are not rebasing.

diff --git a/hgsubversion/wrappers.py b/hgsubversion/wrappers.py
--- a/hgsubversion/wrappers.py
+++ b/hgsubversion/wrappers.py
@@ -12,6 +12,7 @@ from mercurial import util as hgutil
 from mercurial import node
 from mercurial import i18n
 from mercurial import extensions
+from mercurial import repair
 
 import replay
 import pushmod
@@ -193,6 +194,7 @@ def push(repo, dest, force, revs):
         ui.status('searching for changes\n')
         hashes = meta.revmap.hashes()
         outgoing = util.outgoing_revisions(repo, hashes, workingrev.node())
+        to_strip=[]
         if not (outgoing and len(outgoing)):
             ui.status('no changes found\n')
             return 1 # so we get a sane exit status, see hg's commands.push
@@ -248,38 +250,44 @@ def push(repo, dest, force, revs):
 
             # 5. Rebase all children of the currently-pushing rev to the
             # new head
-            heads = repo.heads(old_ctx.node())
-            for needs_transplant in heads:
-                def extrafn(ctx, extra):
-                    if ctx.node() == oldest:
-                        return
-                    extra['branch'] = ctx.branch()
-                # TODO: can we avoid calling our own rebase wrapper here?
-                # Tweaking the encoding is fine for internal
-                # manipulations, but it can lead to various breakage
-                # when starting to operate with the working directory
-                # and the dirstate.
-                util.swap_out_encoding(old_encoding)
-                try:
-                    rebase(hgrebase.rebase, ui, repo, svn=True,
-                           svnextrafn=extrafn, svnsourcerev=needs_transplant)
-                finally:
-                    util.swap_out_encoding()
-                # Reload the repo after the rebase. Do not reuse
-                # contexts across this.
-                newtip = newtipctx.node()
-                repo = getlocalpeer(ui, {}, meta.path)
-                newtipctx = repo[newtip]
-                # Rewrite the node ids in outgoing to their rebased versions.
-                rebasemap = dict()
-                for child in newtipctx.descendants():
-                    rebasesrc = child.extra().get('rebase_source')
-                    if rebasesrc:
-                        rebasemap[node.bin(rebasesrc)] = child.node()
-                outgoing = [rebasemap.get(n) or n for n in outgoing]
-            # TODO: stop constantly creating the SVNMeta instances.
+            #
+            # there may be commits descended from the one we just
+            # pushed to svn that we aren't going to push to svn in
+            # this operation
+            oldhex = node.hex(old_ctx.node())
+            needs_rebase_set = "%s:: and not(%s)" % (oldhex, oldhex)
+            def extrafn(ctx, extra):
+                extra['branch'] = ctx.branch()
+
+            util.swap_out_encoding(old_encoding)
+            try:
+                hgrebase.rebase(ui, repo, dest=node.hex(newtipctx.node()),
+                                rev=[needs_rebase_set],
+                                extrafn=extrafn,
+                                # We actually want to strip one more rev than
+                                # we're rebasing
+                                keep=True)
+            finally:
+                util.swap_out_encoding()
+
+            to_strip.append(old_ctx.node())
+            # don't trust the pre-rebase repo.  Do not reuse
+            # contexts across this.
+            newtip = newtipctx.node()
+            repo = getlocalpeer(ui, {}, meta.path)
+            newtipctx = repo[newtip]
+
+            rebasemap = dict()
+            for child in newtipctx.descendants():
+                rebasesrc = child.extra().get('rebase_source')
+                if rebasesrc:
+                    rebasemap[node.bin(rebasesrc)] = child.node()
+            outgoing = [rebasemap.get(n) or n for n in outgoing]
+
             meta = repo.svnmeta(svn.uuid, svn.subdir)
             hashes = meta.revmap.hashes()
+        hg.update(repo, repo['tip'].node())
+        repair.strip(ui, repo, to_strip, "all")
     finally:
         util.swap_out_encoding(old_encoding)
     return 1 # so we get a sane exit status, see hg's commands.push