comparison mercurial/streamclone.py @ 48597:8475a1364909

stream-clone: factor computation of requirement of a stream clone

This gather code duplicated in multiple place and will make it easier to modify
it safely in the future.

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

48596:739f2ca3aa3f

    """
    requirements = set(default_requirements)
    requirements -= supported_formats
    requirements.update(streamed_requirements)
    return requirements


def canperformstreamclone(pullop, bundle2=False):
    """Whether it is possible to perform a streaming clone as part of pull.

    Returns a tuple of (requirements, data generator).
    """
    if compression != b'UN':
        raise ValueError(b'we do not support the compression argument yet')

    requirements = repo.requirements & repo.supportedformats
    requires = b','.join(sorted(requirements))

    def gen():
        yield b'HGS1'
        yield compression

48597:8475a1364909

    """
    requirements = set(default_requirements)
    requirements -= supported_formats
    requirements.update(streamed_requirements)
    return requirements


def streamed_requirements(repo):
    """the set of requirement the new clone will have to support

    This is used for advertising the stream options and to generate the actual
    stream content."""
    requiredformats = repo.requirements & repo.supportedformats
    return requiredformats


def canperformstreamclone(pullop, bundle2=False):
    """Whether it is possible to perform a streaming clone as part of pull.

    Returns a tuple of (requirements, data generator).
    """
    if compression != b'UN':
        raise ValueError(b'we do not support the compression argument yet')

    requirements = streamed_requirements(repo)
    requires = b','.join(sorted(requirements))

    def gen():
        yield b'HGS1'
        yield compression