# merge.py - directory-level update/merge handling for Mercurial
#
# Copyright 2006 Matt Mackall <mpm@selenic.com>
#
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.

from node import *
from i18n import gettext as _
from demandload import *
demandload(globals(), "util os tempfile")

def merge3(repo, fn, my, other, p1, p2):
    """perform a 3-way merge in the working directory"""

    def temp(prefix, node):
        pre = "%s~%s." % (os.path.basename(fn), prefix)
        (fd, name) = tempfile.mkstemp(prefix=pre)
        f = os.fdopen(fd, "wb")
        repo.wwrite(fn, fl.read(node), f)
        f.close()
        return name

    fl = repo.file(fn)
    base = fl.ancestor(my, other)
    a = repo.wjoin(fn)
    b = temp("base", base)
    c = temp("other", other)

    repo.ui.note(_("resolving %s\n") % fn)
    repo.ui.debug(_("file %s: my %s other %s ancestor %s\n") %
                          (fn, short(my), short(other), short(base)))

    cmd = (os.environ.get("HGMERGE") or repo.ui.config("ui", "merge")
           or "hgmerge")
    r = util.system('%s "%s" "%s" "%s"' % (cmd, a, b, c), cwd=repo.root,
                    environ={'HG_FILE': fn,
                             'HG_MY_NODE': p1,
                             'HG_OTHER_NODE': p2,
                             'HG_FILE_MY_NODE': hex(my),
                             'HG_FILE_OTHER_NODE': hex(other),
                             'HG_FILE_BASE_NODE': hex(base)})
    if r:
        repo.ui.warn(_("merging %s failed!\n") % fn)

    os.unlink(b)
    os.unlink(c)
    return r

def update(repo, node, branchmerge=False, force=False, partial=None,
           forcemerge=False, wlock=None, show_stats=True, remind=True):
    pl = repo.dirstate.parents()
    if not force and pl[1] != nullid:
        raise util.Abort(_("outstanding uncommitted merges"))

    err = False

    p1, p2 = pl[0], node
    pa = repo.changelog.ancestor(p1, p2)
    m1n = repo.changelog.read(p1)[0]
    m2n = repo.changelog.read(p2)[0]
    man = repo.manifest.ancestor(m1n, m2n)
    m1 = repo.manifest.read(m1n)
    mf1 = repo.manifest.readflags(m1n)
    m2 = repo.manifest.read(m2n).copy()
    mf2 = repo.manifest.readflags(m2n)
    ma = repo.manifest.read(man)
    mfa = repo.manifest.readflags(man)

    modified, added, removed, deleted, unknown = repo.changes()

    # is this a jump, or a merge?  i.e. is there a linear path
    # from p1 to p2?
    linear_path = (pa == p1 or pa == p2)

    if branchmerge and linear_path:
        raise util.Abort(_("there is nothing to merge, just use "
                           "'hg update' or look at 'hg heads'"))
    if branchmerge and not forcemerge:
        if modified or added or removed:
            raise util.Abort(_("outstanding uncommitted changes"))

    if not forcemerge and not force:
        for f in unknown:
            if f in m2:
                t1 = repo.wread(f)
                t2 = repo.file(f).read(m2[f])
                if cmp(t1, t2) != 0:
                    raise util.Abort(_("'%s' already exists in the working"
                                       " dir and differs from remote") % f)

    # resolve the manifest to determine which files
    # we care about merging
    repo.ui.note(_("resolving manifests\n"))
    repo.ui.debug(_(" force %s branchmerge %s partial %s linear %s\n") %
                  (force, branchmerge, partial and True or False, linear_path))
    repo.ui.debug(_(" ancestor %s local %s remote %s\n") %
                  (short(man), short(m1n), short(m2n)))

    merge = {}
    get = {}
    remove = []

    # construct a working dir manifest
    mw = m1.copy()
    mfw = mf1.copy()
    umap = dict.fromkeys(unknown)

    for f in added + modified + unknown:
        mw[f] = ""
        mfw[f] = util.is_exec(repo.wjoin(f), mfw.get(f, False))

    if not partial and not wlock: wlock = repo.wlock()

    for f in deleted + removed:
        if f in mw:
            del mw[f]

        # If we're jumping between revisions (as opposed to merging),
        # and if neither the working directory nor the target rev has
        # the file, then we need to remove it from the dirstate, to
        # prevent the dirstate from listing the file when it is no
        # longer in the manifest.
        if not partial and linear_path and f not in m2:
            repo.dirstate.forget((f,))

    # Compare manifests
    for f, n in mw.iteritems():
        if partial and not partial(f):
            continue
        if f in m2:
            s = 0

            # is the wfile new since m1, and match m2?
            if f not in m1:
                t1 = repo.wread(f)
                t2 = repo.file(f).read(m2[f])
                if cmp(t1, t2) == 0:
                    n = m2[f]
                del t1, t2

            # are files different?
            if n != m2[f]:
                a = ma.get(f, nullid)
                # are both different from the ancestor?
                if n != a and m2[f] != a:
                    repo.ui.debug(_(" %s versions differ, resolve\n") % f)
                    # merge executable bits
                    # "if we changed or they changed, change in merge"
                    a, b, c = mfa.get(f, 0), mfw[f], mf2[f]
                    mode = ((a^b) | (a^c)) ^ a
                    merge[f] = (m1.get(f, nullid), m2[f], mode)
                    s = 1
                # are we clobbering?
                # is remote's version newer?
                # or are we going back in time?
                elif force or m2[f] != a or (p2 == pa and mw[f] == m1[f]):
                    repo.ui.debug(_(" remote %s is newer, get\n") % f)
                    get[f] = m2[f]
                    s = 1
            elif f in umap or f in added:
                # this unknown file is the same as the checkout
                # we need to reset the dirstate if the file was added
                get[f] = m2[f]

            if not s and mfw[f] != mf2[f]:
                if force:
                    repo.ui.debug(_(" updating permissions for %s\n") % f)
                    util.set_exec(repo.wjoin(f), mf2[f])
                else:
                    a, b, c = mfa.get(f, 0), mfw[f], mf2[f]
                    mode = ((a^b) | (a^c)) ^ a
                    if mode != b:
                        repo.ui.debug(_(" updating permissions for %s\n")
                                      % f)
                        util.set_exec(repo.wjoin(f), mode)
            del m2[f]
        elif f in ma:
            if n != ma[f]:
                r = _("d")
                if not force and (linear_path or branchmerge):
                    r = repo.ui.prompt(
                        (_(" local changed %s which remote deleted\n") % f) +
                         _("(k)eep or (d)elete?"), _("[kd]"), _("k"))
                if r == _("d"):
                    remove.append(f)
            else:
                repo.ui.debug(_("other deleted %s\n") % f)
                remove.append(f) # other deleted it
        else:
            # file is created on branch or in working directory
            if force and f not in umap:
                repo.ui.debug(_("remote deleted %s, clobbering\n") % f)
                remove.append(f)
            elif n == m1.get(f, nullid): # same as parent
                if p2 == pa: # going backwards?
                    repo.ui.debug(_("remote deleted %s\n") % f)
                    remove.append(f)
                else:
                    repo.ui.debug(_("local modified %s, keeping\n") % f)
            else:
                repo.ui.debug(_("working dir created %s, keeping\n") % f)

    for f, n in m2.iteritems():
        if partial and not partial(f):
            continue
        if f[0] == "/":
            continue
        if f in ma and n != ma[f]:
            r = _("k")
            if not force and (linear_path or branchmerge):
                r = repo.ui.prompt(
                    (_("remote changed %s which local deleted\n") % f) +
                     _("(k)eep or (d)elete?"), _("[kd]"), _("k"))
            if r == _("k"):
                get[f] = n
        elif f not in ma:
            repo.ui.debug(_("remote created %s\n") % f)
            get[f] = n
        else:
            if force or p2 == pa: # going backwards?
                repo.ui.debug(_("local deleted %s, recreating\n") % f)
                get[f] = n
            else:
                repo.ui.debug(_("local deleted %s\n") % f)

    del mw, m1, m2, ma

    if force:
        for f in merge:
            get[f] = merge[f][1]
        merge = {}

    if linear_path or force:
        # we don't need to do any magic, just jump to the new rev
        p1, p2 = p2, nullid
    else:
        if not branchmerge:
            repo.ui.status(_("this update spans a branch"
                             " affecting the following files:\n"))
            fl = merge.keys() + get.keys()
            fl.sort()
            for f in fl:
                cf = ""
                if f in merge:
                    cf = _(" (resolve)")
                repo.ui.status(" %s%s\n" % (f, cf))
            repo.ui.warn(_("aborting update spanning branches!\n"))
            repo.ui.status(_("(use 'hg merge' to merge across branches"
                             " or 'hg update -C' to lose changes)\n"))
            return 1

    xp1 = hex(p1)
    xp2 = hex(p2)
    if p2 == nullid: xxp2 = ''
    else: xxp2 = xp2

    repo.hook('preupdate', throw=True, parent1=xp1, parent2=xxp2)

    # get the files we don't need to change
    files = get.keys()
    files.sort()
    for f in files:
        if f[0] == "/":
            continue
        repo.ui.note(_("getting %s\n") % f)
        t = repo.file(f).read(get[f])
        repo.wwrite(f, t)
        util.set_exec(repo.wjoin(f), mf2[f])
        if not partial:
            if branchmerge:
                repo.dirstate.update([f], 'n', st_mtime=-1)
            else:
                repo.dirstate.update([f], 'n')

    # merge the tricky bits
    failedmerge = []
    files = merge.keys()
    files.sort()
    for f in files:
        repo.ui.status(_("merging %s\n") % f)
        my, other, flag = merge[f]
        ret = merge3(repo, f, my, other, xp1, xp2)
        if ret:
            err = True
            failedmerge.append(f)
        util.set_exec(repo.wjoin(f), flag)
        if not partial:
            if branchmerge:
                # We've done a branch merge, mark this file as merged
                # so that we properly record the merger later
                repo.dirstate.update([f], 'm')
            else:
                # We've update-merged a locally modified file, so
                # we set the dirstate to emulate a normal checkout
                # of that file some time in the past. Thus our
                # merge will appear as a normal local file
                # modification.
                f_len = len(repo.file(f).read(other))
                repo.dirstate.update([f], 'n', st_size=f_len, st_mtime=-1)

    remove.sort()
    for f in remove:
        repo.ui.note(_("removing %s\n") % f)
        util.audit_path(f)
        try:
            util.unlink(repo.wjoin(f))
        except OSError, inst:
            if inst.errno != errno.ENOENT:
                repo.ui.warn(_("update failed to remove %s: %s!\n") %
                             (f, inst.strerror))
    if not partial:
        if branchmerge:
            repo.dirstate.update(remove, 'r')
        else:
            repo.dirstate.forget(remove)

    if not partial:
        repo.dirstate.setparents(p1, p2)

    if show_stats:
        stats = ((len(get), _("updated")),
                 (len(merge) - len(failedmerge), _("merged")),
                 (len(remove), _("removed")),
                 (len(failedmerge), _("unresolved")))
        note = ", ".join([_("%d files %s") % s for s in stats])
        repo.ui.status("%s\n" % note)
    if not partial:
        if branchmerge:
            if failedmerge:
                repo.ui.status(_("There are unresolved merges,"
                                " you can redo the full merge using:\n"
                                "  hg update -C %s\n"
                                "  hg merge %s\n"
                                % (repo.changelog.rev(p1),
                                    repo.changelog.rev(p2))))
            elif remind:
                repo.ui.status(_("(branch merge, don't forget to commit)\n"))
        elif failedmerge:
            repo.ui.status(_("There are unresolved merges with"
                             " locally modified files.\n"))

    repo.hook('update', parent1=xp1, parent2=xxp2, error=int(err))
    return err