bundle2: add a ui argument to readbundle

The bundle2 unbundler needs a ui argument. We are now passing this information
to `readbundle`.

hgext/shelve.py

@@ -71,7 +71,7 class shelvedfile(object):
     def applybundle(self):
         fp = self.opener()
         try:
-            gen = exchange.readbundle(fp, self.fname, self.vfs)
+            gen = exchange.readbundle(self.repo.ui, fp, self.fname, self.vfs)
             changegroup.addchangegroup(self.repo, gen, 'unshelve',
                                        'bundle:' + self.vfs.join(self.fname))
         finally:

mercurial/bundlerepo.py

@@ -202,7 +202,7 class bundlerepository(localrepo.localre
 
         self.tempfile = None
         f = util.posixfile(bundlename, "rb")
-        self.bundle = exchange.readbundle(f, bundlename)
+        self.bundle = exchange.readbundle(ui, f, bundlename)
         if self.bundle.compressed():
             fdtemp, temp = self.vfs.mkstemp(prefix="hg-bundle-",
                                             suffix=".hg10un")
@@ -220,7 +220,7 class bundlerepository(localrepo.localre
                 fptemp.close()
 
             f = self.vfs.open(self.tempfile, mode="rb")
-            self.bundle = exchange.readbundle(f, bundlename, self.vfs)
+            self.bundle = exchange.readbundle(ui, f, bundlename, self.vfs)
 
         # dict with the mapping 'filename' -> position in the bundle
         self.bundlefilespos = {}

mercurial/commands.py

@@ -1736,7 +1736,7 def debugbundle(ui, bundlepath, all=None
     """lists the contents of a bundle"""
     f = hg.openpath(ui, bundlepath)
     try:
-        gen = exchange.readbundle(f, bundlepath)
+        gen = exchange.readbundle(ui, f, bundlepath)
         if all:
             ui.write(("format: id, p1, p2, cset, delta base, len(delta)\n"))
 
@@ -5807,7 +5807,7 def unbundle(ui, repo, fname1, *fnames,
     try:
         for fname in fnames:
             f = hg.openpath(ui, fname)
-            gen = exchange.readbundle(f, fname)
+            gen = exchange.readbundle(ui, f, fname)
             modheads = changegroup.addchangegroup(repo, gen, 'unbundle',
                                                   'bundle:' + fname)
     finally:

mercurial/exchange.py

@@ -11,7 +11,7 import errno
 import util, scmutil, changegroup, base85
 import discovery, phases, obsolete, bookmarks, bundle2
 
-def readbundle(fh, fname, vfs=None):
+def readbundle(ui, fh, fname, vfs=None):
     header = changegroup.readexactly(fh, 6)
 
     if not fname:

mercurial/repair.py

@@ -147,7 +147,7 def strip(ui, repo, nodelist, backup="al
         if saveheads or savebases:
             ui.note(_("adding branch\n"))
             f = vfs.open(chgrpfile, "rb")
-            gen = exchange.readbundle(f, chgrpfile, vfs)
+            gen = exchange.readbundle(ui, f, chgrpfile, vfs)
             if not repo.ui.verbose:
                 # silence internal shuffling chatter
                 repo.ui.pushbuffer()

mercurial/wireproto.py

@@ -766,7 +766,7 def unbundle(repo, proto, heads):
         try:
             proto.getfile(fp)
             fp.seek(0)
-            gen = exchange.readbundle(fp, None)
+            gen = exchange.readbundle(repo.ui, fp, None)
             r = exchange.unbundle(repo, gen, their_heads, 'serve',
                                   proto._client())
             return pushres(r)