comparison mercurial/bundle2.py @ 21146:4676135ac555

bundle2: move all parts into a `bx2` namespace

All currently core parts are moved to a `bx2` namespace (for "bundle 2
experimental"). This should avoid conflicts between the final stable
format and the one about to be released.

21145:0c5088be66af

                handler(op, part)
            finally:
                if output is not None:
                    output = op.ui.popbuffer()
            if output:
                outpart = bundlepart('output',
                                     advisoryparams=[('in-reply-to',
                                                      str(part.id))],
                                     data=output)
                op.reply.addpart(outpart)
            part.read()

        if size is None or len(data) < size:
            self.consumed = True
        return data


@parthandler('changegroup')
def handlechangegroup(op, inpart):
    """apply a changegroup part on the repo

    This is a very early implementation that will massive rework before being
    inflicted to any end-user.

    ret = changegroup.addchangegroup(op.repo, cg, 'bundle2', 'bundle2')
    op.records.add('changegroup', {'return': ret})
    if op.reply is not None:
        # This is definitly not the final form of this
        # return. But one need to start somewhere.
        part = bundlepart('reply:changegroup', (),
                           [('in-reply-to', str(inpart.id)),
                            ('return', '%i' % ret)])
        op.reply.addpart(part)
    assert not inpart.read()

@parthandler('reply:changegroup')
def handlechangegroup(op, inpart):
    p = dict(inpart.advisoryparams)
    ret = int(p['return'])
    op.records.add('changegroup', {'return': ret}, int(p['in-reply-to']))

@parthandler('check:heads')
def handlechangegroup(op, inpart):
    """check that head of the repo did not change

    This is used to detect a push race when using unbundle.
    This replaces the "heads" argument of unbundle."""

        h = inpart.read(20)
    assert not h
    if heads != op.repo.heads():
        raise exchange.PushRaced()

@parthandler('output')
def handleoutput(op, inpart):
    """forward output captured on the server to the client"""
    for line in inpart.read().splitlines():
        op.ui.write(('remote: %s\n' % line))

@parthandler('replycaps')
def handlereplycaps(op, inpart):
    """Notify that a reply bundle should be created

    The payload contains the capabilities information for the reply"""
    caps = decodecaps(inpart.read())

21146:4676135ac555

                handler(op, part)
            finally:
                if output is not None:
                    output = op.ui.popbuffer()
            if output:
                outpart = bundlepart('b2x:output',
                                     advisoryparams=[('in-reply-to',
                                                      str(part.id))],
                                     data=output)
                op.reply.addpart(outpart)
            part.read()

        if size is None or len(data) < size:
            self.consumed = True
        return data


@parthandler('b2x:changegroup')
def handlechangegroup(op, inpart):
    """apply a changegroup part on the repo

    This is a very early implementation that will massive rework before being
    inflicted to any end-user.

    ret = changegroup.addchangegroup(op.repo, cg, 'bundle2', 'bundle2')
    op.records.add('changegroup', {'return': ret})
    if op.reply is not None:
        # This is definitly not the final form of this
        # return. But one need to start somewhere.
        part = bundlepart('b2x:reply:changegroup', (),
                           [('in-reply-to', str(inpart.id)),
                            ('return', '%i' % ret)])
        op.reply.addpart(part)
    assert not inpart.read()

@parthandler('b2x:reply:changegroup')
def handlechangegroup(op, inpart):
    p = dict(inpart.advisoryparams)
    ret = int(p['return'])
    op.records.add('changegroup', {'return': ret}, int(p['in-reply-to']))

@parthandler('b2x:check:heads')
def handlechangegroup(op, inpart):
    """check that head of the repo did not change

    This is used to detect a push race when using unbundle.
    This replaces the "heads" argument of unbundle."""

        h = inpart.read(20)
    assert not h
    if heads != op.repo.heads():
        raise exchange.PushRaced()

@parthandler('b2x:output')
def handleoutput(op, inpart):
    """forward output captured on the server to the client"""
    for line in inpart.read().splitlines():
        op.ui.write(('remote: %s\n' % line))

@parthandler('b2x:replycaps')
def handlereplycaps(op, inpart):
    """Notify that a reply bundle should be created

    The payload contains the capabilities information for the reply"""
    caps = decodecaps(inpart.read())