comparison hgext/obsolete.py @ 282:05ab164c6593

obsolete: handle rebase --collapse

Recording obsolete in concludenode() did not work for --collapse because
only the first revision was passed to the call. The new strategy is to
track the rebase state in defineparents() and concludenode() and to
create markers only after a successful non-abort call. In theory, this
should also fix --continue/--abort cases.

The change in test-stabilize-order.t comes from concludenode() no longer
creating obsolete marker. stabilize command was actually duplicating
markers, once in concludenode(), once explicitely.

281:258169d3428b

def buildstate(orig, repo, dest, rebaseset, *ags, **kws):
    """wrapper for rebase 's buildstate that exclude obsolete changeset"""
    rebaseset = repo.revs('%ld - extinct()', rebaseset)
    return orig(repo, dest, rebaseset, *ags, **kws)


def concludenode(orig, repo, rev, *args, **kwargs):
    """wrapper for rebase 's concludenode that set obsolete relation"""
    newrev = orig(repo, rev, *args, **kwargs)
    oldnode = repo[rev].node()
    if newrev is not None:
        newnode = repo[newrev].node()
    else:
        # Revision was emptied and removed, there is no successor.
        newnode = nullid
    repo.addobsolete(newnode, oldnode)
    return newrev

def cmdrebase(orig, ui, repo, *args, **kwargs):
    if kwargs.get('keep', False):
        raise util.Abort(_('rebase --keep option is unsupported with obsolete '
                           'extension'), hint=_("see 'hg help obsolete'"))
    kwargs = dict(kwargs)
    kwargs['keep'] = True
    return orig(ui, repo, *args, **kwargs)



def extsetup(ui):

    revset.symbols["obsolete"] = revsetobsolete

    try:
        rebase = extensions.find('rebase')
        if rebase:
            extensions.wrapfunction(rebase, 'buildstate', buildstate)
            extensions.wrapfunction(rebase, 'concludenode', concludenode)
            extensions.wrapcommand(rebase.cmdtable, "rebase", cmdrebase)
    except KeyError:
        pass # rebase not found

282:05ab164c6593

def buildstate(orig, repo, dest, rebaseset, *ags, **kws):
    """wrapper for rebase 's buildstate that exclude obsolete changeset"""
    rebaseset = repo.revs('%ld - extinct()', rebaseset)
    return orig(repo, dest, rebaseset, *ags, **kws)

def defineparents(orig, repo, rev, target, state, *args, **kwargs):
    rebasestate = getattr(repo, '_rebasestate', None)
    if rebasestate is not None:
        repo._rebasestate = dict(state)
        repo._rebasetarget = target
    return orig(repo, rev, target, state, *args, **kwargs)

def concludenode(orig, repo, rev, p1, *args, **kwargs):
    """wrapper for rebase 's concludenode that set obsolete relation"""
    newrev = orig(repo, rev, p1, *args, **kwargs)
    rebasestate = getattr(repo, '_rebasestate', None)
    if rebasestate is not None:
        if newrev is not None:
            nrev = repo[newrev].rev()
        else:
            nrev = p1
        repo._rebasestate[rev] = nrev
    return newrev

def cmdrebase(orig, ui, repo, *args, **kwargs):
    if kwargs.get('keep', False):
        raise util.Abort(_('rebase --keep option is unsupported with obsolete '
                           'extension'), hint=_("see 'hg help obsolete'"))
    kwargs = dict(kwargs)
    kwargs['keep'] = True

    # We want to mark rebased revision as obsolete and set their
    # replacements if any. Doing it in concludenode() prevents
    # aborting the rebase, and is not called with all relevant
    # revisions in --collapse case. Instead, we try to track the
    # rebase state structure by sampling/updating it in
    # defineparents() and concludenode(). The obsolete markers are
    # added from this state after a successful call.
    repo._rebasestate = {}
    repo._rebasetarget = None
    maxrev = len(repo) - 1
    try:
        res = orig(ui, repo, *args, **kwargs)
        if not res and not kwargs.get('abort') and repo._rebasetarget:
            # We have to tell rewritten revisions from removed
            # ones. When collapsing, removed revisions are considered
            # to be collapsed onto the final one, while in the normal
            # case their are marked obsolete without successor.
            emptynode = nullid
            if kwargs.get('collapse'):
                emptynode = repo[max(repo._rebasestate.values())].node()
            # Rebased revisions are assumed to be descendants of
            # targetrev. If a source revision is mapped to targetrev
            # or to another rebased revision, it must have been
            # removed.
            targetrev = repo[repo._rebasetarget].rev()
            newrevs = set([targetrev])
            for rev, newrev in sorted(repo._rebasestate.items()):
                oldnode = repo[rev].node()
                if newrev not in newrevs and newrev >= 0:
                    newnode = repo[newrev].node()
                    newrevs.add(newrev)
                else:
                    newnode = emptynode
                repo.addobsolete(newnode, oldnode)
        return res
    finally:
        delattr(repo, '_rebasestate')
        delattr(repo, '_rebasetarget')


def extsetup(ui):

    revset.symbols["obsolete"] = revsetobsolete

    try:
        rebase = extensions.find('rebase')
        if rebase:
            extensions.wrapfunction(rebase, 'buildstate', buildstate)
            extensions.wrapfunction(rebase, 'defineparents', defineparents)
            extensions.wrapfunction(rebase, 'concludenode', concludenode)
            extensions.wrapcommand(rebase.cmdtable, "rebase", cmdrebase)
    except KeyError:
        pass # rebase not found