def _decodebundle2caps(bundlecaps):
b2caps = {}
for bcaps in bundlecaps:
if bcaps.startswith("bundle2="):
blob = util.urlreq.unquote(bcaps[len("bundle2=") :])
b2caps.update(bundle2.decodecaps(blob))
return b2caps
def _rebundle(bundlerepo, bundleroots, unknownhead, cgversion, bundlecaps):
"""
Bundle may include more revision then user requested. For example,
if user asks for revision but bundle also consists its descendants.
This function will filter out all revision that user is not requested.
"""
parts = []
outgoing = discovery.outgoing(bundlerepo,
commonheads=bundleroots,
missingheads=[unknownhead])
cgstream = changegroup.makestream(bundlerepo,
outgoing,
cgversion,
"pull",
bundlecaps=bundlecaps)
cgstream = util.chunkbuffer(cgstream).read()
cgpart = bundle2.bundlepart("changegroup", data=cgstream)
cgpart.addparam("version", cgversion)
parts.append(cgpart)
# This parsing should be refactored to be shared with
# exchange.getbundlechunks. But I'll do that in a separate diff.
if bundlecaps is None:
bundlecaps = set()
b2caps = {}
for bcaps in bundlecaps:
if bcaps.startswith("bundle2="):
blob = util.urlreq.unquote(bcaps[len("bundle2="):])
b2caps.update(bundle2.decodecaps(blob))
if constants.scratchmutationparttype in b2caps:
mutdata = mutation.bundle(bundlerepo, outgoing.missing)
parts.append(
bundle2.bundlepart(constants.scratchmutationparttype,
data=mutdata))
try:
treemod = extensions.find("treemanifest")
remotefilelog = extensions.find("remotefilelog")
except KeyError:
pass
else:
missing = outgoing.missing
if remotefilelog.shallowbundle.cansendtrees(bundlerepo,
missing,
source="pull",
bundlecaps=bundlecaps,
b2caps=b2caps):
try:
treepart = treemod.createtreepackpart(
bundlerepo, outgoing, treemod.TREEGROUP_PARTTYPE2)
parts.append(treepart)
except BaseException as ex:
parts.append(bundle2.createerrorpart(str(ex)))
return parts
