# Copyright 2011 Peter Arrenbrecht <peter.arrenbrecht@gmail.com>

#                Logilab SA        <contact@logilab.fr>

#                Pierre-Yves David <pierre-yves.david@ens-lyon.org>

#                Patrick Mezard <patrick@mezard.eu>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

"""logic related to hg evolve command"""

import collections

import re

from mercurial import (

    bookmarks as bookmarksmod,

    cmdutil,

    commands,

    context,

    copies,

    error,

    encoding,

    hg,

    merge,

    mergeutil,

    node as nodemod,

    obsolete,

    obsutil,

    phases,

    pycompat,

    repair,

    scmutil,

    simplemerge,

    util,

)

from mercurial.i18n import _

from . import (

    cmdrewrite,

    compat,

    exthelper,

    rewriteutil,

    state,

    utility,

)

TROUBLES = compat.TROUBLES

shorttemplate = utility.shorttemplate

stacktemplate = utility.stacktemplate

_bookmarksupdater = rewriteutil.bookmarksupdater

sha1re = re.compile(br'\b[0-9a-f]{6,40}\b')

eh = exthelper.exthelper()

mergetoolopts = commands.mergetoolopts

abortmessage = _(b"see `hg help evolve.interrupted`\n")

def _solveone(ui, repo, ctx, evolvestate, dryrun, confirm,

              progresscb, category, lastsolved=None, stacktmplt=False):

    """Resolve the troubles affecting one revision

    returns a tuple (bool, newnode) where,

        bool: a boolean value indicating whether the instability was solved

        newnode: if bool is True, then the newnode of the resultant commit

                 formed. newnode can be node, when resolution led to no new

                 commit. If bool is False, this is ".".

    """

    tr = repo.currenttransaction()

    assert tr is not None

    displayer = None

    if stacktmplt:

        displayer = compat.changesetdisplayer(ui, repo,

                                              {b'template': stacktemplate})

    else:

        displayer = compat.changesetdisplayer(ui, repo,

                                              {b'template': shorttemplate})

    if b'orphan' == category:

        result = _solveunstable(ui, repo, ctx, evolvestate, displayer,

                                dryrun, confirm, progresscb,

                                lastsolved=lastsolved)

    elif b'phasedivergent' == category:

        result = _solvephasedivergence(ui, repo, ctx, evolvestate,

                                       displayer, dryrun, confirm,

                                       progresscb)

    elif b'contentdivergent' == category:

        result = _solvedivergent(ui, repo, ctx, evolvestate, displayer,

                                 dryrun, confirm, progresscb)

    else:

        assert False, b"unknown trouble category: %s" % (category)

    return result

def _solveunstable(ui, repo, orig, evolvestate, displayer, dryrun=False,

                   confirm=False, progresscb=None, lastsolved=None):

    """ Tries to stabilize the changeset orig which is orphan.

    returns a tuple (bool, newnode) where,

        bool: a boolean value indicating whether the instability was solved

        newnode: if bool is True, then the newnode of the resultant commit

                 formed. newnode can be node, when resolution led to no new

                 commit. If bool is False, this is ".".

    """

    pctx = orig.p1()

    keepbranch = orig.p1().branch() != orig.branch()

    if len(orig.parents()) == 2:

        p1obs = orig.p1().obsolete()

        p2obs = orig.p2().obsolete()

        if not p1obs and p2obs:

            pctx = orig.p2()  # second parent is obsolete ?

            keepbranch = orig.p2().branch() != orig.branch()

        elif not p2obs and p1obs:

            pass

        else:

            # store that we are resolving an orphan merge with both parents

            # obsolete and proceed with first parent

            evolvestate[b'orphanmerge'] = True

            # we should process the second parent first, so that in case of

            # no-conflicts the first parent is processed later and preserved as

            # first parent

            pctx = orig.p2()

            keepbranch = orig.p2().branch() != orig.branch()

    if not pctx.obsolete():

        ui.warn(_(b"cannot solve instability of %s, skipping\n") % orig)

        return (False, b".")

    obs = pctx

    newer = obsutil.successorssets(repo, obs.node())

    # search of a parent which is not killed, but also isn't the orig

    while not newer or newer == [()] or newer[0][0] == orig.node():

        ui.debug(b"stabilize target %s is plain dead,"

                 b" trying to stabilize on its parent\n" %

                 obs)

        obs = obs.parents()[0]

        newer = obsutil.successorssets(repo, obs.node())

    if len(newer) > 1:

        msg = _(b"skipping %s: divergent rewriting. can't choose "

                b"destination\n") % obs

        ui.write_err(msg)

        return (False, b".")

    targets = newer[0]

    assert targets

    if len(targets) > 1:

        # split target, figure out which one to pick, are they all in line?

        targetrevs = [repo[r].rev() for r in targets]

        roots = repo.revs(b'roots(%ld)', targetrevs)

        heads = repo.revs(b'heads(%ld)', targetrevs)

        if len(roots) > 1 or len(heads) > 1:

            cheader = _(b"ancestor '%s' split over multiple topological"

                        b" branches.\nchoose an evolve destination:") % orig

            selectedrev = utility.revselectionprompt(ui, repo, list(heads),

                                                     cheader)

            if selectedrev is None:

                msg = _(b"could not solve instability, "

                        b"ambiguous destination: "

                        b"parent split across two branches\n")

                ui.write_err(msg)

                return (False, b".")

            target = repo[selectedrev]

        else:

            target = repo[heads.first()]

    else:

        target = targets[0]

    target = repo[target]

    if not ui.quiet or confirm:

        repo.ui.write(_(b'move:'), label=b'evolve.operation')

        displayer.show(orig)

        if lastsolved is None or target != repo[lastsolved]:

            repo.ui.write(_(b'atop:'))

            displayer.show(target)

    if confirm and ui.prompt(b'perform evolve? [Ny]', b'n') != b'y':

        raise error.Abort(_(b'evolve aborted by user'))

    todo = b'hg rebase -r %s -d %s\n' % (orig, target)

    if dryrun:

        if progresscb:

            progresscb()

        repo.ui.write(todo)

        return (False, b".")

    else:

        repo.ui.note(todo)

        if progresscb:

            progresscb()

        with state.saver(evolvestate, {b'current': orig.node()}):

            newid = relocate(repo, orig, target, evolvestate, pctx,

                             keepbranch, b'orphan')

            return (True, newid)

def _solvephasedivergence(ui, repo, bumped, evolvestate, displayer,

                          dryrun=False, confirm=False, progresscb=None):

    """Stabilize a phase divergent changeset

    returns a tuple (bool, newnode) where,

        bool: a boolean value indicating whether the instability was solved

        newnode: if bool is True, then the newnode of the resultant commit

                 formed. newnode can be node, when resolution led to no new

                 commit. If bool is False, this is ".".

    """

    repo = repo.unfiltered()

    bumped = repo[bumped.rev()]

    # For now we deny bumped merge

    if len(bumped.parents()) > 1:

        msg = _(b'skipping %s : we do not handle merge yet\n') % bumped

        ui.write_err(msg)

        return (False, b".")

    prec = next(repo.set(b'last(allpredecessors(%d) and public())', bumped.rev()))

    # For now we deny target merge

    if len(prec.parents()) > 1:

        msg = _(b'skipping: %s: public version is a merge, '

                b'this is not handled yet\n') % prec

        ui.write_err(msg)

        return (False, b".")

    if not ui.quiet or confirm:

        repo.ui.write(_(b'recreate:'), label=b'evolve.operation')

        displayer.show(bumped)

        repo.ui.write(_(b'atop:'))

        displayer.show(prec)

    if confirm and ui.prompt(_(b'perform evolve? [Ny]'), b'n') != b'y':

        raise error.Abort(_(b'evolve aborted by user'))

    if dryrun:

        todo = b'hg rebase --rev %s --dest %s;\n' % (bumped, prec.p1())

        repo.ui.write(todo)

        repo.ui.write((b'hg update %s;\n' % prec))

        repo.ui.write((b'hg revert --all --rev %s;\n' % bumped))

        repo.ui.write((b'hg commit --msg "%s update to %s"\n' %

                       (TROUBLES['PHASEDIVERGENT'], bumped)))

        return (False, b".")

    if progresscb:

        progresscb()

    # Checking for whether the phase-divergent changeset has common parents as

    # it's precursors. Phase-divergent changeset and precursor having different

    # parents is a result of when the changeset is rebased, picked, histedit or

    # evolved or any other operation which can change parent. In such cases,

    # when parents are not same, we first rebase the divergent changeset onto

    # parent or precursor and then perform later steps

    if not list(repo.set(b'parents(%d) and parents(%d)', bumped.rev(), prec.rev())):

        # Need to rebase the changeset at the right place

        repo.ui.status(

            _(b'rebasing to destination parent: %s\n') % prec.p1())

        with state.saver(evolvestate, {b'current': bumped.hex(),

                                       b'precursor': prec.hex()}):

            newnode = relocate(repo, bumped, prec.p1(), evolvestate,

                               category=b'phasedivergent')

            if newnode is not None:

                new = repo[newnode]

                obsolete.createmarkers(repo, [(bumped, (new,))],

                                       operation=b'evolve')

                bumped = new

                evolvestate[b'temprevs'].append(newnode)

    return _resolvephasedivergent(ui, repo, prec, bumped)

def _resolvephasedivergent(ui, repo, prec, bumped):

    """final step of a phase divergence resolution

    This will create a new changesets (or nothing when applicable), the two

    changesets needs to be on the same parents.

    """

    tr = repo.currenttransaction()

    assert tr is not None

    cl = repo.changelog

    prec_parent = cl.parentrevs(prec.rev())

    bump_parent = cl.parentrevs(bumped.rev())

    assert prec_parent == bump_parent, (prec_parent, bump_parent)

    bmupdate = _bookmarksupdater(repo, bumped.node(), tr)

    newid = None

    replacementnode = None

    # Create the new commit context. This is done by applying the changes from

    # the precursor to the bumped node onto the precursor. This is effectively

    # like reverting to the bumped node.

    wctx = context.overlayworkingctx(repo)

    wctx.setbase(prec)

    merge.update(repo, bumped.node(), ancestor=prec, mergeancestor=True,

                 branchmerge=True, force=False, wc=wctx)

    if not wctx.isempty():

        text = b'%s update to %s:\n\n' % (TROUBLES['PHASEDIVERGENT'], prec)

        text += bumped.description()

        memctx = wctx.tomemctx(text,

                               parents=(prec.node(), nodemod.nullid),

                               date=bumped.date(),

                               extra=bumped.extra(),

                               user=bumped.user())

        newid = repo.commitctx(memctx)

        replacementnode = newid

    if newid is None:

        repo.ui.status(_(b'no changes to commit\n'))

        obsolete.createmarkers(repo, [(bumped, ())], operation=b'evolve')

        newid = prec.node()

    else:

        repo.ui.status(_(b'committed as %s\n') % nodemod.short(newid))

        phases.retractboundary(repo, tr, bumped.phase(), [newid])

        obsolete.createmarkers(repo, [(bumped, (repo[newid],))],

                               flag=obsolete.bumpedfix, operation=b'evolve')

    bmupdate(newid)

    # reroute the working copy parent to the new changeset

    with repo.dirstate.parentchange():

        repo.dirstate.setparents(newid, nodemod.nullid)

    return (True, replacementnode)

def _solvedivergent(ui, repo, divergent, evolvestate, displayer, dryrun=False,

                    confirm=False, progresscb=None):

    """tries to solve content-divergence of a changeset

    returns a tuple (bool, newnode) where,

        bool: a boolean value indicating whether the instability was solved

        newnode: if bool is True, then the newnode of the resultant commit

                 formed. newnode can be node, when resolution led to no new

                 commit. If bool is False, this is ".".

    """

    repo = repo.unfiltered()

    divergent = repo[divergent.rev()]

    evolvestate[b'divergent'] = divergent.node()

    evolvestate[b'orig-divergent'] = divergent.node()

    # sometimes we will relocate a node in case of different parents and we can

    # encounter conflicts after relocation is done while solving

    # content-divergence and if the user calls `hg evolve --stop`, we need to

    # strip that relocated commit. However if `--all` is passed, we need to

    # reset this value for each content-divergence resolution which we are doing

    # below.

    evolvestate[b'relocated'] = None

    evolvestate[b'relocating'] = False

    # in case or relocation we get a new other node, we need to store the old

    # other for purposes like `--abort` or `--stop`

    evolvestate[b'old-other'] = None

    base, others = divergentdata(divergent)

    # we don't handle split in content-divergence yet

    if len(others) > 1:

        othersstr = b"[%s]" % (b','.join([bytes(i) for i in others]))

        msg = _(b"skipping %s: %s with a changeset that got split"

                b" into multiple ones:\n"

                b"|[%s]\n"

                b"| This is not handled by automatic evolution yet\n"

                b"| You have to fallback to manual handling with commands "

                b"such as:\n"

                b"| - hg touch -D\n"

                b"| - hg prune\n"

                b"| \n"

                b"| You should contact your local evolution Guru for help.\n"

                ) % (divergent, TROUBLES['CONTENTDIVERGENT'], othersstr)

        ui.write_err(msg)

        return (False, b".")

    other = others[0]

    evolvestate[b'other-divergent'] = other.node()

    evolvestate[b'base'] = base.node()

    def swapnodes(div, other):

        div, other = other, div

        evolvestate[b'divergent'] = div.node()

        evolvestate[b'other-divergent'] = other.node()

        return div, other

    # haspubdiv: to keep track if we are solving public content-divergence

    haspubdiv = False

    if not (divergent.mutable() and other.mutable()):

        haspubdiv = True

        # for simplicity, we keep public one to local side while merging