comparison mercurial/exchange.py @ 21063:7ca4f2049d3b

bundle2: move `readbundle` into the `exchange` module

The `readbundle` function is going to understand the bundle2 header. We move the
function to a more suitable place before making any other changes.

21062:e7c0a65a5c9c

from i18n import _
from node import hex, nullid
import errno
import util, scmutil, changegroup, base85
import discovery, phases, obsolete, bookmarks, bundle2


class pushoperation(object):
    """A object that represent a single push operation

21063:7ca4f2049d3b

from i18n import _
from node import hex, nullid
import errno
import util, scmutil, changegroup, base85
import discovery, phases, obsolete, bookmarks, bundle2

def readbundle(fh, fname, vfs=None):
    header = changegroup.readexactly(fh, 6)

    if not fname:
        fname = "stream"
        if not header.startswith('HG') and header.startswith('\0'):
            fh = changegroup.headerlessfixup(fh, header)
            header = "HG10UN"
    elif vfs:
        fname = vfs.join(fname)

    magic, version, alg = header[0:2], header[2:4], header[4:6]

    if magic != 'HG':
        raise util.Abort(_('%s: not a Mercurial bundle') % fname)
    if version != '10':
        raise util.Abort(_('%s: unknown bundle version %s') % (fname, version))
    return changegroup.unbundle10(fh, alg)


class pushoperation(object):
    """A object that represent a single push operation