merge.py
484 lines
| 17.2 KiB
| text/x-python
|
PythonLexer
/ mercurial / merge.py
|
r2775 | # merge.py - directory-level update/merge handling for Mercurial | ||
# | ||||
|
r4635 | # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com> | ||
|
r2775 | # | ||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
# GNU General Public License version 2, incorporated herein by reference. | ||||
|
r2775 | |||
|
r6518 | from node import nullid, nullrev, hex, bin | ||
|
r3891 | from i18n import _ | ||
|
r8312 | import util, filemerge, copies | ||
import errno, os, shutil | ||||
|
r6512 | |||
class mergestate(object): | ||||
'''track 3-way merge state of individual files''' | ||||
def __init__(self, repo): | ||||
self._repo = repo | ||||
|
r6518 | self._read() | ||
|
r7848 | def reset(self, node=None): | ||
|
r6512 | self._state = {} | ||
|
r7848 | if node: | ||
self._local = node | ||||
|
r6512 | shutil.rmtree(self._repo.join("merge"), True) | ||
|
r6518 | def _read(self): | ||
self._state = {} | ||||
try: | ||||
|
r6530 | localnode = None | ||
|
r6518 | f = self._repo.opener("merge/state") | ||
|
r6530 | for i, l in enumerate(f): | ||
if i == 0: | ||||
localnode = l[:-1] | ||||
else: | ||||
bits = l[:-1].split("\0") | ||||
self._state[bits[0]] = bits[1:] | ||||
self._local = bin(localnode) | ||||
|
r6518 | except IOError, err: | ||
if err.errno != errno.ENOENT: | ||||
raise | ||||
def _write(self): | ||||
f = self._repo.opener("merge/state", "w") | ||||
f.write(hex(self._local) + "\n") | ||||
|
r7622 | for d, v in self._state.iteritems(): | ||
|
r6518 | f.write("\0".join([d] + v) + "\n") | ||
|
r6512 | def add(self, fcl, fco, fca, fd, flags): | ||
|
r6517 | hash = util.sha1(fcl.path()).hexdigest() | ||
|
r6512 | self._repo.opener("merge/" + hash, "w").write(fcl.data()) | ||
|
r6518 | self._state[fd] = ['u', hash, fcl.path(), fca.path(), | ||
hex(fca.filenode()), fco.path(), flags] | ||||
self._write() | ||||
|
r6512 | def __contains__(self, dfile): | ||
return dfile in self._state | ||||
def __getitem__(self, dfile): | ||||
|
r6518 | return self._state[dfile][0] | ||
def __iter__(self): | ||||
l = self._state.keys() | ||||
l.sort() | ||||
for f in l: | ||||
yield f | ||||
|
r6512 | def mark(self, dfile, state): | ||
|
r6518 | self._state[dfile][0] = state | ||
self._write() | ||||
|
r6512 | def resolve(self, dfile, wctx, octx): | ||
if self[dfile] == 'r': | ||||
return 0 | ||||
|
r6518 | state, hash, lfile, afile, anode, ofile, flags = self._state[dfile] | ||
|
r6512 | f = self._repo.opener("merge/" + hash) | ||
self._repo.wwrite(dfile, f.read(), flags) | ||||
fcd = wctx[dfile] | ||||
fco = octx[ofile] | ||||
fca = self._repo.filectx(afile, fileid=anode) | ||||
r = filemerge.filemerge(self._repo, self._local, lfile, fcd, fco, fca) | ||||
if not r: | ||||
self.mark(dfile, 'r') | ||||
return r | ||||
|
r2775 | |||
|
r6269 | def _checkunknown(wctx, mctx): | ||
|
r3315 | "check for collisions between unknown files and files in mctx" | ||
|
r3218 | for f in wctx.unknown(): | ||
|
r6272 | if f in mctx and mctx[f].cmp(wctx[f].data()): | ||
raise util.Abort(_("untracked file in working directory differs" | ||||
" from file in requested revision: '%s'") % f) | ||||
|
r3107 | |||
|
r6269 | def _checkcollision(mctx): | ||
|
r3785 | "check for case folding collisions in the destination context" | ||
folded = {} | ||||
|
r6272 | for fn in mctx: | ||
|
r3785 | fold = fn.lower() | ||
if fold in folded: | ||||
raise util.Abort(_("case-folding collision between %s and %s") | ||||
% (fn, folded[fold])) | ||||
folded[fold] = fn | ||||
|
r6269 | def _forgetremoved(wctx, mctx, branchmerge): | ||
|
r3107 | """ | ||
Forget removed files | ||||
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. | ||||
|
r6242 | |||
If we're merging, and the other revision has removed a file | ||||
that is not present in the working directory, we need to mark it | ||||
as removed. | ||||
|
r3107 | """ | ||
action = [] | ||||
|
r6242 | state = branchmerge and 'r' or 'f' | ||
for f in wctx.deleted(): | ||||
|
r6272 | if f not in mctx: | ||
|
r6242 | action.append((f, state)) | ||
if not branchmerge: | ||||
for f in wctx.removed(): | ||||
|
r6272 | if f not in mctx: | ||
|
r6242 | action.append((f, "f")) | ||
|
r3107 | |||
return action | ||||
|
r3295 | def manifestmerge(repo, p1, p2, pa, overwrite, partial): | ||
|
r3105 | """ | ||
|
r3315 | Merge p1 and p2 with ancestor ma and generate merge action list | ||
overwrite = whether we clobber working files | ||||
partial = function to filter file lists | ||||
|
r3105 | """ | ||
|
r3314 | repo.ui.note(_("resolving manifests\n")) | ||
repo.ui.debug(_(" overwrite %s partial %s\n") % (overwrite, bool(partial))) | ||||
repo.ui.debug(_(" ancestor %s local %s remote %s\n") % (pa, p1, p2)) | ||||
|
r3295 | m1 = p1.manifest() | ||
m2 = p2.manifest() | ||||
backwards = (pa == p2) | ||||
|
r8734 | |||
|
r8742 | if overwrite: | ||
|
r8734 | ma = m1 | ||
|
r8742 | elif backwards: | ||
ma = p1.p1().manifest() | ||||
else: | ||||
ma = pa.manifest() | ||||
|
r8734 | |||
|
r3314 | action = [] | ||
|
r6274 | copy, copied, diverge = {}, {}, {} | ||
|
r3295 | |||
|
r8733 | def fmerge(f, f2, fa): | ||
|
r4007 | """merge flags""" | ||
|
r5708 | a, m, n = ma.flags(fa), m1.flags(f), m2.flags(f2) | ||
if m == n: # flags agree | ||||
return m # unchanged | ||||
|
r8733 | if m and n and not a: # flags set, don't agree, differ from parent | ||
r = repo.ui.prompt( | ||||
_(" conflicting flags for %s\n" | ||||
"(n)one, e(x)ec or sym(l)ink?") % f, | ||||
(_("&None"), _("E&xec"), _("Sym&link")), _("n")) | ||||
return r != _("n") and r or '' | ||||
|
r5708 | if m and m != a: # changed from a to m | ||
return m | ||||
if n and n != a: # changed from a to n | ||||
return n | ||||
return '' # flag was cleared | ||||
|
r3118 | |||
|
r3307 | def act(msg, m, f, *args): | ||
|
r3295 | repo.ui.debug(" %s: %s -> %s\n" % (f, msg, m)) | ||
|
r3121 | action.append((f, m) + args) | ||
|
r6275 | if pa and not (backwards or overwrite): | ||
|
r6425 | if repo.ui.configbool("merge", "followcopies", True): | ||
dirs = repo.ui.configbool("merge", "followdirs", True) | ||||
copy, diverge = copies.copies(repo, p1, p2, pa, dirs) | ||||
|
r8152 | copied = set(copy.values()) | ||
|
r7622 | for of, fl in diverge.iteritems(): | ||
|
r6274 | act("divergent renames", "dr", of, fl) | ||
|
r3295 | |||
|
r3105 | # Compare manifests | ||
for f, n in m1.iteritems(): | ||||
|
r3248 | if partial and not partial(f): | ||
continue | ||||
|
r3105 | if f in m2: | ||
|
r8735 | rflags = fmerge(f, f, f) | ||
|
r3105 | # are files different? | ||
if n != m2[f]: | ||||
a = ma.get(f, nullid) | ||||
# is remote's version newer? | ||||
|
r8747 | if m2[f] != a: | ||
# are both different from the ancestor? | ||||
if n != a: | ||||
act("versions differ", "m", f, f, f, rflags, False) | ||||
else: | ||||
act("remote is newer", "g", f, rflags) | ||||
continue | ||||
# contents don't need updating, check mode bits | ||||
if m1.flags(f) != rflags: | ||||
|
r5708 | act("update permissions", "e", f, rflags) | ||
|
r3730 | elif f in copied: | ||
continue | ||||
|
r3249 | elif f in copy: | ||
f2 = copy[f] | ||||
|
r3733 | if f2 not in m2: # directory rename | ||
act("remote renamed directory to " + f2, "d", | ||||
|
r4007 | f, None, f2, m1.flags(f)) | ||
|
r8748 | else: # case 2 A,B/B/B or case 4,21 A/B/B | ||
act("local copied/moved to " + f2, "m", | ||||
|
r3730 | f, f2, f, fmerge(f, f2, f2), False) | ||
|
r8744 | elif n[20:] == "a": # added, no remote | ||
act("remote deleted", "f", f) | ||||
elif f in ma: # clean, a different, no remote | ||||
|
r8739 | if n != ma[f]: | ||
|
r3295 | if repo.ui.prompt( | ||
|
r5670 | _(" local changed %s which remote deleted\n" | ||
"use (c)hanged version or (d)elete?") % f, | ||||
|
r8259 | (_("&Changed"), _("&Delete")), _("c")) == _("d"): | ||
|
r3307 | act("prompt delete", "r", f) | ||
|
r8736 | else: | ||
act("prompt keep", "a", f) | ||||
|
r8744 | elif n[20:] != "u": | ||
|
r3307 | act("other deleted", "r", f) | ||
|
r3105 | |||
for f, n in m2.iteritems(): | ||||
|
r3248 | if partial and not partial(f): | ||
continue | ||||
if f in m1: | ||||
continue | ||||
|
r3729 | if f in copied: | ||
continue | ||||
|
r3249 | if f in copy: | ||
f2 = copy[f] | ||||
|
r3733 | if f2 not in m1: # directory rename | ||
act("local renamed directory to " + f2, "d", | ||||
|
r4007 | None, f, f2, m2.flags(f)) | ||
|
r3733 | elif f2 in m2: # rename case 1, A/A,B/A | ||
|
r3730 | act("remote copied to " + f, "m", | ||
f2, f, f, fmerge(f2, f, f2), False) | ||||
else: # case 3,20 A/B/A | ||||
act("remote moved to " + f, "m", | ||||
f2, f, f, fmerge(f2, f, f2), True) | ||||
|
r8741 | elif f not in ma: | ||
|
r4007 | act("remote created", "g", f, m2.flags(f)) | ||
|
r8741 | elif n != ma[f]: | ||
if repo.ui.prompt( | ||||
_("remote changed %s which local deleted\n" | ||||
"use (c)hanged version or leave (d)eleted?") % f, | ||||
(_("&Changed"), _("&Deleted")), _("c")) == _("c"): | ||||
act("prompt recreating", "g", f, m2.flags(f)) | ||||
|
r3105 | |||
return action | ||||
|
r8366 | def actionkey(a): | ||
return a[1] == 'r' and -1 or 0, a | ||||
|
r6805 | |||
|
r3297 | def applyupdates(repo, action, wctx, mctx): | ||
|
r3315 | "apply the merge action list to the working directory" | ||
|
r3111 | updated, merged, removed, unresolved = 0, 0, 0, 0 | ||
|
r6512 | ms = mergestate(repo) | ||
ms.reset(wctx.parents()[0].node()) | ||||
moves = [] | ||||
|
r8366 | action.sort(key=actionkey) | ||
|
r6512 | |||
# prescan for merges | ||||
|
r5042 | for a in action: | ||
f, m = a[:2] | ||||
if m == 'm': # merge | ||||
f2, fd, flags, move = a[2:] | ||||
|
r6512 | repo.ui.debug(_("preserving %s for resolve of %s\n") % (f, fd)) | ||
fcl = wctx[f] | ||||
fco = mctx[f2] | ||||
fca = fcl.ancestor(fco) or repo.filectx(f, fileid=nullrev) | ||||
ms.add(fcl, fco, fca, fd, flags) | ||||
if f != fd and move: | ||||
moves.append(f) | ||||
# remove renamed files after safely stored | ||||
for f in moves: | ||||
if util.lexists(repo.wjoin(f)): | ||||
repo.ui.debug(_("removing %s\n") % f) | ||||
os.unlink(repo.wjoin(f)) | ||||
|
r5042 | |||
|
r5158 | audit_path = util.path_auditor(repo.root) | ||
|
r3111 | for a in action: | ||
f, m = a[:2] | ||||
|
r3733 | if f and f[0] == "/": | ||
|
r3111 | continue | ||
if m == "r": # remove | ||||
repo.ui.note(_("removing %s\n") % f) | ||||
|
r5158 | audit_path(f) | ||
|
r3111 | 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)) | ||||
|
r3673 | removed += 1 | ||
|
r3308 | elif m == "m": # merge | ||
|
r4007 | f2, fd, flags, move = a[2:] | ||
|
r6512 | r = ms.resolve(fd, wctx, mctx) | ||
|
r3400 | if r > 0: | ||
|
r3249 | unresolved += 1 | ||
|
r3309 | else: | ||
|
r3400 | if r is None: | ||
updated += 1 | ||||
else: | ||||
merged += 1 | ||||
|
r6877 | util.set_flags(repo.wjoin(fd), 'l' in flags, 'x' in flags) | ||
|
r5059 | if f != fd and move and util.lexists(repo.wjoin(f)): | ||
|
r5042 | repo.ui.debug(_("removing %s\n") % f) | ||
os.unlink(repo.wjoin(f)) | ||||
|
r3111 | elif m == "g": # get | ||
|
r4007 | flags = a[2] | ||
|
r3111 | repo.ui.note(_("getting %s\n") % f) | ||
|
r3303 | t = mctx.filectx(f).data() | ||
|
r4007 | repo.wwrite(f, t, flags) | ||
|
r3111 | updated += 1 | ||
|
r3733 | elif m == "d": # directory rename | ||
|
r4007 | f2, fd, flags = a[2:] | ||
|
r3733 | if f: | ||
repo.ui.note(_("moving %s to %s\n") % (f, fd)) | ||||
t = wctx.filectx(f).data() | ||||
|
r4007 | repo.wwrite(fd, t, flags) | ||
|
r3733 | util.unlink(repo.wjoin(f)) | ||
if f2: | ||||
repo.ui.note(_("getting %s to %s\n") % (f2, fd)) | ||||
t = mctx.filectx(f2).data() | ||||
|
r4007 | repo.wwrite(fd, t, flags) | ||
|
r3733 | updated += 1 | ||
|
r4674 | elif m == "dr": # divergent renames | ||
fl = a[2] | ||||
|
r6953 | repo.ui.warn(_("warning: detected divergent renames of %s to:\n") % f) | ||
|
r4674 | for nf in fl: | ||
repo.ui.warn(" %s\n" % nf) | ||||
|
r3111 | elif m == "e": # exec | ||
|
r4007 | flags = a[2] | ||
|
r6877 | util.set_flags(repo.wjoin(f), 'l' in flags, 'x' in flags) | ||
|
r3111 | |||
return updated, merged, removed, unresolved | ||||
|
r3372 | def recordupdates(repo, action, branchmerge): | ||
|
r3315 | "record merge actions to the dirstate" | ||
|
r3111 | for a in action: | ||
f, m = a[:2] | ||||
if m == "r": # remove | ||||
if branchmerge: | ||||
|
r4904 | repo.dirstate.remove(f) | ||
|
r3111 | else: | ||
|
r4904 | repo.dirstate.forget(f) | ||
|
r7768 | elif m == "a": # re-add | ||
if not branchmerge: | ||||
repo.dirstate.add(f) | ||||
|
r3111 | elif m == "f": # forget | ||
|
r4904 | repo.dirstate.forget(f) | ||
|
r7569 | elif m == "e": # exec change | ||
|
r7630 | repo.dirstate.normallookup(f) | ||
|
r7569 | elif m == "g": # get | ||
|
r3111 | if branchmerge: | ||
|
r4904 | repo.dirstate.normaldirty(f) | ||
|
r3111 | else: | ||
|
r4904 | repo.dirstate.normal(f) | ||
|
r3111 | elif m == "m": # merge | ||
|
r3303 | f2, fd, flag, move = a[2:] | ||
|
r3251 | if branchmerge: | ||
# We've done a branch merge, mark this file as merged | ||||
# so that we properly record the merger later | ||||
|
r4904 | repo.dirstate.merge(fd) | ||
|
r3372 | if f != f2: # copy/rename | ||
if move: | ||||
|
r4904 | repo.dirstate.remove(f) | ||
|
r3372 | if f != fd: | ||
repo.dirstate.copy(f, fd) | ||||
else: | ||||
repo.dirstate.copy(f2, fd) | ||||
|
r3251 | 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. | ||||
|
r5210 | repo.dirstate.normallookup(fd) | ||
|
r3308 | if move: | ||
|
r4904 | repo.dirstate.forget(f) | ||
|
r3733 | elif m == "d": # directory rename | ||
f2, fd, flag = a[2:] | ||||
|
r4819 | if not f2 and f not in repo.dirstate: | ||
# untracked file moved | ||||
continue | ||||
|
r3733 | if branchmerge: | ||
|
r4904 | repo.dirstate.add(fd) | ||
|
r3733 | if f: | ||
|
r4904 | repo.dirstate.remove(f) | ||
|
r3733 | repo.dirstate.copy(f, fd) | ||
if f2: | ||||
repo.dirstate.copy(f2, fd) | ||||
else: | ||||
|
r4904 | repo.dirstate.normal(fd) | ||
|
r3733 | if f: | ||
|
r4904 | repo.dirstate.forget(f) | ||
|
r3111 | |||
|
r4917 | def update(repo, node, branchmerge, force, partial): | ||
|
r3315 | """ | ||
Perform a merge between the working directory and the given node | ||||
branchmerge = whether to merge between branches | ||||
force = whether to force branch merging or file overwriting | ||||
partial = a function to filter file lists (dirstate not updated) | ||||
""" | ||||
|
r2815 | |||
|
r4917 | wlock = repo.wlock() | ||
|
r4915 | try: | ||
|
r6747 | wc = repo[None] | ||
|
r4915 | if node is None: | ||
# tip of current branch | ||||
try: | ||||
node = repo.branchtags()[wc.branch()] | ||||
except KeyError: | ||||
|
r5570 | if wc.branch() == "default": # no default branch! | ||
node = repo.lookup("tip") # update to tip | ||||
else: | ||||
raise util.Abort(_("branch %s not found") % wc.branch()) | ||||
|
r4915 | overwrite = force and not branchmerge | ||
pl = wc.parents() | ||||
|
r6747 | p1, p2 = pl[0], repo[node] | ||
|
r4915 | pa = p1.ancestor(p2) | ||
fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2) | ||||
fastforward = False | ||||
|
r3314 | |||
|
r4915 | ### check phase | ||
if not overwrite and len(pl) > 1: | ||||
raise util.Abort(_("outstanding uncommitted merges")) | ||||
|
r6375 | if branchmerge: | ||
if pa == p2: | ||||
raise util.Abort(_("can't merge with ancestor")) | ||||
elif pa == p1: | ||||
if p1.branch() != p2.branch(): | ||||
|
r4915 | fastforward = True | ||
else: | ||||
|
r6375 | raise util.Abort(_("nothing to merge (use 'hg update'" | ||
" or check 'hg heads')")) | ||||
if not force and (wc.files() or wc.deleted()): | ||||
|
r8545 | raise util.Abort(_("outstanding uncommitted changes " | ||
"(use 'hg status' to list changes)")) | ||||
|
r6375 | elif not overwrite: | ||
if pa == p1 or pa == p2: # linear | ||||
pass # all good | ||||
elif p1.branch() == p2.branch(): | ||||
if wc.files() or wc.deleted(): | ||||
raise util.Abort(_("crosses branches (use 'hg merge' or " | ||||
"'hg update -C' to discard changes)")) | ||||
|
r6381 | raise util.Abort(_("crosses branches (use 'hg merge' " | ||
|
r6375 | "or 'hg update -C')")) | ||
elif wc.files() or wc.deleted(): | ||||
raise util.Abort(_("crosses named branches (use " | ||||
"'hg update -C' to discard changes)")) | ||||
else: | ||||
# Allow jumping branches if there are no changes | ||||
overwrite = True | ||||
|
r2814 | |||
|
r4915 | ### calculate phase | ||
action = [] | ||||
if not force: | ||||
|
r6269 | _checkunknown(wc, p2) | ||
|
r6746 | if not util.checkcase(repo.path): | ||
|
r6269 | _checkcollision(p2) | ||
action += _forgetremoved(wc, p2, branchmerge) | ||||
|
r4915 | action += manifestmerge(repo, wc, p2, pa, overwrite, partial) | ||
|
r2775 | |||
|
r4915 | ### apply phase | ||
if not branchmerge: # just jump to the new rev | ||||
fp1, fp2, xp1, xp2 = fp2, nullid, xp2, '' | ||||
if not partial: | ||||
repo.hook('preupdate', throw=True, parent1=xp1, parent2=xp2) | ||||
|
r2775 | |||
|
r4915 | stats = applyupdates(repo, action, wc, p2) | ||
|
r2899 | |||
|
r4915 | if not partial: | ||
recordupdates(repo, action, branchmerge) | ||||
repo.dirstate.setparents(fp1, fp2) | ||||
if not branchmerge and not fastforward: | ||||
repo.dirstate.setbranch(p2.branch()) | ||||
repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3]) | ||||
|
r3314 | |||
|
r4915 | return stats | ||
finally: | ||||
|
r8109 | wlock.release() | ||