comparison hgext/uncommit.py @ 41342:fe83040400b7

uncommit: set dirstateparents from within _fixdirstate()

It's now done in the same way for uncommit and unamend, so we can
share the code.

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

41341:19c590ce8661

def _fixdirstate(repo, oldctx, newctx, match=None):
    """ fix the dirstate after switching the working directory from oldctx to
    newctx which can be result of either unamend or uncommit.
    """
    ds = repo.dirstate
    copies = dict(ds.copies())
    s = newctx.status(oldctx, match=match)
    for f in s.modified:
        if ds[f] == 'r':
            # modified + removed -> removed

            else:
                # Fully removed the old commit
                mapping[old.node()] = ()

            with repo.dirstate.parentchange():
                repo.dirstate.setparents(newid, node.nullid)
                _fixdirstate(repo, old, repo[newid], match)

            scmutil.cleanupnodes(repo, mapping, 'uncommit', fixphase=True)

def predecessormarkers(ctx):

        newprednode = repo.commitctx(newctx)
        newpredctx = repo[newprednode]
        dirstate = repo.dirstate

        with dirstate.parentchange():
            dirstate.setparents(newprednode, node.nullid)
            _fixdirstate(repo, curctx, newpredctx)

        mapping = {curctx.node(): (newprednode,)}
        scmutil.cleanupnodes(repo, mapping, 'unamend', fixphase=True)

41342:fe83040400b7

def _fixdirstate(repo, oldctx, newctx, match=None):
    """ fix the dirstate after switching the working directory from oldctx to
    newctx which can be result of either unamend or uncommit.
    """
    ds = repo.dirstate
    ds.setparents(newctx.node(), node.nullid)
    copies = dict(ds.copies())
    s = newctx.status(oldctx, match=match)
    for f in s.modified:
        if ds[f] == 'r':
            # modified + removed -> removed

            else:
                # Fully removed the old commit
                mapping[old.node()] = ()

            with repo.dirstate.parentchange():
                _fixdirstate(repo, old, repo[newid], match)

            scmutil.cleanupnodes(repo, mapping, 'uncommit', fixphase=True)

def predecessormarkers(ctx):

        newprednode = repo.commitctx(newctx)
        newpredctx = repo[newprednode]
        dirstate = repo.dirstate

        with dirstate.parentchange():
            _fixdirstate(repo, curctx, newpredctx)

        mapping = {curctx.node(): (newprednode,)}
        scmutil.cleanupnodes(repo, mapping, 'unamend', fixphase=True)