comparison mercurial/bundle2.py @ 35261:f392066d127c

bookmark: add pushkey hook compatiblity to the bundle2 part

Currently, pushing a bookmark update triggers a pushkey hooks. It is likely
that users in the wild use such hooks to control bookmark movement. Using a non
push-key mechanism to exchange bookmark means these hooks are no longer called,
possibly breaking existing users setup. So we add explicit call to the pushkey
hooks in the handling of the bundle2 part. This behavior can be disabled with a
new config knob: 'server.bookmarks-pushkey-compat'.

35260:af5507203d01

    This behavior is suitable for pushing. Semantic adjustment will be needed
    for pull.
    """
    changes = bookmarks.binarydecode(inpart)
    op.repo._bookmarks.applychanges(op.repo, op.gettransaction(), changes)

@parthandler('phase-heads')
def handlephases(op, inpart):
    """apply phases from bundle part to repo"""
    headsbyphase = phases.binarydecode(inpart)

35261:f392066d127c

    This behavior is suitable for pushing. Semantic adjustment will be needed
    for pull.
    """
    changes = bookmarks.binarydecode(inpart)

    tr = op.gettransaction()
    bookstore = op.repo._bookmarks

    pushkeycompat = op.repo.ui.configbool('server', 'bookmarks-pushkey-compat')
    if pushkeycompat:
        allhooks = []
        for book, node in changes:
            hookargs = tr.hookargs.copy()
            hookargs['pushkeycompat'] = '1'
            hookargs['namespace'] = 'bookmark'
            hookargs['key'] = book
            hookargs['old'] = nodemod.hex(bookstore.get(book, ''))
            hookargs['new'] = nodemod.hex(node if node is not None else '')
            allhooks.append(hookargs)
        for hookargs in allhooks:
            op.repo.hook('prepushkey', throw=True, **hookargs)

    bookstore.applychanges(op.repo, tr, changes)

    if pushkeycompat:
        def runhook():
            for hookargs in allhooks:
                op.repo.hook('prepushkey', **hookargs)
        op.repo._afterlock(runhook)

@parthandler('phase-heads')
def handlephases(op, inpart):
    """apply phases from bundle part to repo"""
    headsbyphase = phases.binarydecode(inpart)