comparison mercurial/exchange.py @ 37165:6c7a6b04b274

bundlespec: move computing the bundle contentops in parsebundlespec

We will introduce a new bundlespec for stream bundle which will influence the
contentops.

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

37164:b229fd9adeae

                         'v2': '02',
                         'packed1': 's1',
                         'bundle2': '02', #legacy
                        }

# Compression engines allowed in version 1. THIS SHOULD NEVER CHANGE.
_bundlespecv1compengines = {'gzip', 'bzip2', 'none'}

@attr.s
class bundlespec(object):
    compression = attr.ib()
    version = attr.ib()
    params = attr.ib()

def parsebundlespec(repo, spec, strict=True, externalnames=False):
    """Parse a bundle string specification into parts.

    Bundle specifications denote a well-defined bundle/exchange format.

        if missingreqs:
            raise error.UnsupportedBundleSpecification(
                    _('missing support for repository features: %s') %
                      ', '.join(sorted(missingreqs)))

    if not externalnames:
        engine = util.compengines.forbundlename(compression)
        compression = engine.bundletype()[1]
        version = _bundlespeccgversions[version]

    return bundlespec(compression, version, params)

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

    alg = None

37165:6c7a6b04b274

                         'v2': '02',
                         'packed1': 's1',
                         'bundle2': '02', #legacy
                        }

# Maps bundle version with content opts to choose which part to bundle
_bundlespeccontentopts = {
    'v1': {
        'changegroup': True,
        'cg.version': '01',
        'obsolescence': False,
        'phases': False,
        'tagsfnodescache': False,
        'revbranchcache': False
    },
    'v2': {
        'changegroup': True,
        'cg.version': '02',
        'obsolescence': False,
        'phases': False,
        'tagsfnodescache': True,
        'revbranchcache': True
    },
    'packed1' : {
        'cg.version': 's1'
    }
}
_bundlespeccontentopts['bundle2'] = _bundlespeccontentopts['v2']

# Compression engines allowed in version 1. THIS SHOULD NEVER CHANGE.
_bundlespecv1compengines = {'gzip', 'bzip2', 'none'}

@attr.s
class bundlespec(object):
    compression = attr.ib()
    version = attr.ib()
    params = attr.ib()
    contentopts = attr.ib()

def parsebundlespec(repo, spec, strict=True, externalnames=False):
    """Parse a bundle string specification into parts.

    Bundle specifications denote a well-defined bundle/exchange format.

        if missingreqs:
            raise error.UnsupportedBundleSpecification(
                    _('missing support for repository features: %s') %
                      ', '.join(sorted(missingreqs)))

    # Compute contentopts based on the version
    contentopts = _bundlespeccontentopts.get(version, {}).copy()

    if not externalnames:
        engine = util.compengines.forbundlename(compression)
        compression = engine.bundletype()[1]
        version = _bundlespeccgversions[version]

    return bundlespec(compression, version, params, contentopts)

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

    alg = None