comparison hgext/rebase.py @ 42975:43c84b816445

rebase: track new nodes when --keep is set

When --keep is passed with rebase, the new nodes created are not
accessible from templates.

This change enables accessing the newly-created nodes from nodechanges,
just as if --keep was not set.

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

42974:576fd1c8b20b

    for rev, newrev in sorted(state.items()):
        if newrev >= 0 and newrev != rev:
            oldnode = tonode(rev)
            newnode = collapsedas or tonode(newrev)
            moves[oldnode] = newnode
            if not keepf:
                succs = None
                if rev in skipped:
                    if stripcleanup or not repo[rev].obsolete():
                        succs = ()
                elif collapsedas:
                    collapsednodes.append(oldnode)
                else:
                    succs = (newnode,)
                if succs is not None:
                    replacements[(oldnode,)] = succs
    if collapsednodes:
        replacements[tuple(collapsednodes)] = (collapsedas,)
    scmutil.cleanupnodes(repo, replacements, 'rebase', moves, backup=backup)
    if fm:
        hf = fm.hexfunc
        fl = fm.formatlist
        fd = fm.formatdict
        changes = {}
        for oldns, newn in replacements.iteritems():
            for oldn in oldns:
                changes[hf(oldn)] = fl([hf(n) for n in newn], name='node')
        nodechanges = fd(changes, key="oldnode", value="newnodes")
        fm.data(nodechanges=nodechanges)

def pullrebase(orig, ui, repo, *args, **opts):
    'Call rebase after pull if the latter has been invoked with --rebase'
    if opts.get(r'rebase'):
        if ui.configbool('commands', 'rebase.requiredest'):

42975:43c84b816445

    for rev, newrev in sorted(state.items()):
        if newrev >= 0 and newrev != rev:
            oldnode = tonode(rev)
            newnode = collapsedas or tonode(newrev)
            moves[oldnode] = newnode
            succs = None
            if rev in skipped:
                if stripcleanup or not repo[rev].obsolete():
                    succs = ()
            elif collapsedas:
                collapsednodes.append(oldnode)
            else:
                succs = (newnode,)
            if succs is not None:
                replacements[(oldnode,)] = succs
    if collapsednodes:
        replacements[tuple(collapsednodes)] = (collapsedas,)
    if fm:
        hf = fm.hexfunc
        fl = fm.formatlist
        fd = fm.formatdict
        changes = {}
        for oldns, newn in replacements.iteritems():
            for oldn in oldns:
                changes[hf(oldn)] = fl([hf(n) for n in newn], name='node')
        nodechanges = fd(changes, key="oldnode", value="newnodes")
        fm.data(nodechanges=nodechanges)
    if keepf:
        replacements = {}
    scmutil.cleanupnodes(repo, replacements, 'rebase', moves, backup=backup)

def pullrebase(orig, ui, repo, *args, **opts):
    'Call rebase after pull if the latter has been invoked with --rebase'
    if opts.get(r'rebase'):
        if ui.configbool('commands', 'rebase.requiredest'):