# HG changeset patch
# User Augie Fackler <augie@google.com>
# Date 1528817480 14400
# Node ID 1ad873c3e4a4ef8f1ac0f56068b09659cf3811d9
# Parent  a1d5951efce7207168b441a6b75847ef4c60974d
narrowbundle2: when we handle a widen, mark the operation as unsafe

We have to manually drive the context manager here since the logic is
awkwardly split between a couple of places.

Differential Revision: https://phab.mercurial-scm.org/D3719

diff -r a1d5951efce7 -r 1ad873c3e4a4 hgext/narrow/narrowbundle2.py
--- a/hgext/narrow/narrowbundle2.py	Mon Jun 11 17:06:29 2018 -0400
+++ b/hgext/narrow/narrowbundle2.py	Tue Jun 12 11:31:20 2018 -0400
@@ -408,6 +408,8 @@
                                  topic='widen')
         repo._bookmarks = bmstore
         if chgrpfile:
+            op._widen_uninterr = repo.ui.uninterruptable()
+            op._widen_uninterr.__enter__()
             # presence of _widen_bundle attribute activates widen handler later
             op._widen_bundle = chgrpfile
     # Set the new narrowspec if we're widening. The setnewnarrowpats() method
@@ -455,6 +457,7 @@
                         (undovfs.join(undofile), stringutil.forcebytestr(e)))
 
     # Remove partial backup only if there were no exceptions
+    op._widen_uninterr.__exit__(None, None, None)
     vfs.unlink(chgrpfile)
 
 def setup():