changeset 31528:2b599f5468a4

histedit: inline cleanupnode Move "cleanupnode" (unsafe strip) into "safecleanupnode" so it's impossible to call the unsafe function directly. This helps reduce future programming errors.
author Jun Wu <quark@fb.com>
date Mon, 13 Mar 2017 22:30:07 -0700
parents 6f0b7475cf9a
children 61ff3852f6ed
files hgext/histedit.py
diffstat 1 files changed, 17 insertions(+), 23 deletions(-) [+]
line wrap: on
line diff
--- a/hgext/histedit.py	Mon Mar 13 22:22:18 2017 -0700
+++ b/hgext/histedit.py	Mon Mar 13 22:30:07 2017 -0700
@@ -1580,28 +1580,6 @@
         finally:
             release(tr, lock)
 
-def cleanupnode(ui, repo, name, nodes):
-    """strip a group of nodes from the repository
-
-    The set of node to strip may contains unknown nodes."""
-    ui.debug('should strip %s nodes %s\n' %
-             (name, ', '.join([node.short(n) for n in nodes])))
-    with repo.lock():
-        # do not let filtering get in the way of the cleanse
-        # we should probably get rid of obsolescence marker created during the
-        # histedit, but we currently do not have such information.
-        repo = repo.unfiltered()
-        # Find all nodes that need to be stripped
-        # (we use %lr instead of %ln to silently ignore unknown items)
-        nm = repo.changelog.nodemap
-        nodes = sorted(n for n in nodes if n in nm)
-        roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
-        for c in roots:
-            # We should process node in reverse order to strip tip most first.
-            # but this trigger a bug in changegroup hook.
-            # This would reduce bundle overhead
-            repair.strip(ui, repo, c)
-
 def safecleanupnode(ui, repo, name, nodes):
     """strip or obsolete nodes
 
@@ -1628,7 +1606,23 @@
         if markers:
             obsolete.createmarkers(repo, markers)
     else:
-        return cleanupnode(ui, repo, name, nodes)
+        ui.debug('should strip %s nodes %s\n' %
+                 (name, ', '.join([node.short(n) for n in nodes])))
+        with repo.lock():
+            # Do not let filtering get in the way of the cleanse we should
+            # probably get rid of obsolescence marker created during the
+            # histedit, but we currently do not have such information.
+            repo = repo.unfiltered()
+            # Find all nodes that need to be stripped
+            # (we use %lr instead of %ln to silently ignore unknown items)
+            nm = repo.changelog.nodemap
+            nodes = sorted(n for n in nodes if n in nm)
+            roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
+            for c in roots:
+                # We should process node in reverse order to strip tip most
+                # first, but this trigger a bug in changegroup hook. This
+                # would reduce bundle overhead
+                repair.strip(ui, repo, c)
 
 def stripwrapper(orig, ui, repo, nodelist, *args, **kwargs):
     if isinstance(nodelist, str):