filemerge.py
838 lines
| 30.0 KiB
| text/x-python
|
PythonLexer
/ mercurial / filemerge.py
Matt Mackall
|
r6003 | # filemerge.py - file-level merge handling for Mercurial | ||
# | ||||
# Copyright 2006, 2007, 2008 Matt Mackall <mpm@selenic.com> | ||||
# | ||||
Martin Geisler
|
r8225 | # This software may be used and distributed according to the terms of the | ||
Matt Mackall
|
r10263 | # GNU General Public License version 2 or any later version. | ||
Matt Mackall
|
r6003 | |||
Gregory Szorc
|
r25949 | from __future__ import absolute_import | ||
import os | ||||
import re | ||||
import tempfile | ||||
from .i18n import _ | ||||
Siddharth Agarwal
|
r26979 | from .node import nullid, short | ||
Gregory Szorc
|
r25949 | |||
from . import ( | ||||
Pulkit Goyal
|
r30636 | encoding, | ||
Gregory Szorc
|
r25949 | error, | ||
Yuya Nishihara
|
r28955 | formatter, | ||
Gregory Szorc
|
r25949 | match, | ||
Pulkit Goyal
|
r30073 | pycompat, | ||
FUJIWARA Katsunori
|
r33663 | registrar, | ||
Siddharth Agarwal
|
r27651 | scmutil, | ||
Gregory Szorc
|
r25949 | simplemerge, | ||
tagmerge, | ||||
templatekw, | ||||
templater, | ||||
util, | ||||
) | ||||
Matt Mackall
|
r6004 | |||
Boris Feld
|
r34827 | def _toolstr(ui, tool, part, *args): | ||
return ui.config("merge-tools", tool + "." + part, *args) | ||||
Matt Mackall
|
r6004 | |||
Boris Feld
|
r34827 | def _toolbool(ui, tool, part,*args): | ||
return ui.configbool("merge-tools", tool + "." + part, *args) | ||||
Matt Mackall
|
r6004 | |||
Boris Feld
|
r34827 | def _toollist(ui, tool, part): | ||
return ui.configlist("merge-tools", tool + "." + part) | ||||
David Champion
|
r11148 | |||
FUJIWARA Katsunori
|
r16126 | internals = {} | ||
Gregory Szorc
|
r24099 | # Merge tools to document. | ||
internalsdoc = {} | ||||
FUJIWARA Katsunori
|
r16125 | |||
FUJIWARA Katsunori
|
r33663 | internaltool = registrar.internalmerge() | ||
Siddharth Agarwal
|
r26525 | # internal tool merge types | ||
FUJIWARA Katsunori
|
r33663 | nomerge = internaltool.nomerge | ||
mergeonly = internaltool.mergeonly # just the full merge, no premerge | ||||
fullmerge = internaltool.fullmerge # both premerge and merge | ||||
Siddharth Agarwal
|
r26525 | |||
Stanislau Hlebik
|
r32318 | _localchangedotherdeletedmsg = _( | ||
"local%(l)s changed %(fd)s which other%(o)s deleted\n" | ||||
"use (c)hanged version, (d)elete, or leave (u)nresolved?" | ||||
"$$ &Changed $$ &Delete $$ &Unresolved") | ||||
_otherchangedlocaldeletedmsg = _( | ||||
"other%(o)s changed %(fd)s which local%(l)s deleted\n" | ||||
"use (c)hanged version, leave (d)eleted, or " | ||||
"leave (u)nresolved?" | ||||
"$$ &Changed $$ &Deleted $$ &Unresolved") | ||||
Siddharth Agarwal
|
r26979 | class absentfilectx(object): | ||
"""Represents a file that's ostensibly in a context but is actually not | ||||
present in it. | ||||
This is here because it's very specific to the filemerge code for now -- | ||||
other code is likely going to break with the values this returns.""" | ||||
def __init__(self, ctx, f): | ||||
self._ctx = ctx | ||||
self._f = f | ||||
def path(self): | ||||
return self._f | ||||
def size(self): | ||||
return None | ||||
def data(self): | ||||
return None | ||||
def filenode(self): | ||||
return nullid | ||||
_customcmp = True | ||||
def cmp(self, fctx): | ||||
"""compare with other file context | ||||
returns True if different from fctx. | ||||
""" | ||||
return not (fctx.isabsent() and | ||||
fctx.ctx() == self.ctx() and | ||||
fctx.path() == self.path()) | ||||
def flags(self): | ||||
return '' | ||||
def changectx(self): | ||||
return self._ctx | ||||
def isbinary(self): | ||||
return False | ||||
def isabsent(self): | ||||
return True | ||||
Matt Mackall
|
r6004 | def _findtool(ui, tool): | ||
FUJIWARA Katsunori
|
r16126 | if tool in internals: | ||
Dov Feldstern
|
r6522 | return tool | ||
Matt Harbison
|
r23148 | return findexternaltool(ui, tool) | ||
def findexternaltool(ui, tool): | ||||
Steve Borho
|
r13565 | for kn in ("regkey", "regkeyalt"): | ||
k = _toolstr(ui, tool, kn) | ||||
if not k: | ||||
continue | ||||
Adrian Buehlmann
|
r14230 | p = util.lookupreg(k, _toolstr(ui, tool, "regname")) | ||
Matt Mackall
|
r6006 | if p: | ||
Kostia Balytskyi
|
r34947 | p = util.findexe(p + _toolstr(ui, tool, "regappend", "")) | ||
Matt Mackall
|
r6006 | if p: | ||
return p | ||||
Greg Ward
|
r15264 | exe = _toolstr(ui, tool, "executable", tool) | ||
return util.findexe(util.expandpath(exe)) | ||||
Matt Mackall
|
r6004 | |||
Siddharth Agarwal
|
r27039 | def _picktool(repo, ui, path, binary, symlink, changedelete): | ||
def supportscd(tool): | ||||
return tool in internals and internals[tool].mergetype == nomerge | ||||
def check(tool, pat, symlink, binary, changedelete): | ||||
Matt Mackall
|
r6004 | tmsg = tool | ||
if pat: | ||||
FUJIWARA Katsunori
|
r32254 | tmsg = _("%s (for pattern %s)") % (tool, pat) | ||
Mads Kiilerich
|
r7397 | if not _findtool(ui, tool): | ||
if pat: # explicitly requested tool deserves a warning | ||||
ui.warn(_("couldn't find merge tool %s\n") % tmsg) | ||||
else: # configured but non-existing tools are more silent | ||||
ui.note(_("couldn't find merge tool %s\n") % tmsg) | ||||
Matt Mackall
|
r6004 | elif symlink and not _toolbool(ui, tool, "symlink"): | ||
ui.warn(_("tool %s can't handle symlinks\n") % tmsg) | ||||
elif binary and not _toolbool(ui, tool, "binary"): | ||||
ui.warn(_("tool %s can't handle binary\n") % tmsg) | ||||
Siddharth Agarwal
|
r27039 | elif changedelete and not supportscd(tool): | ||
# the nomerge tools are the only tools that support change/delete | ||||
# conflicts | ||||
pass | ||||
Matt Mackall
|
r6007 | elif not util.gui() and _toolbool(ui, tool, "gui"): | ||
ui.warn(_("tool %s requires a GUI\n") % tmsg) | ||||
Matt Mackall
|
r6004 | else: | ||
return True | ||||
return False | ||||
Matt Mackall
|
r25835 | # internal config: ui.forcemerge | ||
Steve Borho
|
r12788 | # forcemerge comes from command line arguments, highest priority | ||
force = ui.config('ui', 'forcemerge') | ||||
if force: | ||||
toolpath = _findtool(ui, force) | ||||
Siddharth Agarwal
|
r27039 | if changedelete and not supportscd(toolpath): | ||
return ":prompt", None | ||||
Steve Borho
|
r12788 | else: | ||
Siddharth Agarwal
|
r27039 | if toolpath: | ||
return (force, util.shellquote(toolpath)) | ||||
else: | ||||
# mimic HGMERGE if given tool not found | ||||
return (force, force) | ||||
Steve Borho
|
r12788 | |||
# HGMERGE takes next precedence | ||||
Pulkit Goyal
|
r30636 | hgmerge = encoding.environ.get("HGMERGE") | ||
Steve Borho
|
r6025 | if hgmerge: | ||
Siddharth Agarwal
|
r27039 | if changedelete and not supportscd(hgmerge): | ||
return ":prompt", None | ||||
else: | ||||
return (hgmerge, hgmerge) | ||||
Matt Mackall
|
r6004 | |||
# then patterns | ||||
dhruva
|
r6016 | for pat, tool in ui.configitems("merge-patterns"): | ||
Matt Mackall
|
r8567 | mf = match.match(repo.root, '', [pat]) | ||
Siddharth Agarwal
|
r27039 | if mf(path) and check(tool, pat, symlink, False, changedelete): | ||
Benoit Boissinot
|
r10339 | toolpath = _findtool(ui, tool) | ||
Keegan Carruthers-Smith
|
r17885 | return (tool, util.shellquote(toolpath)) | ||
Matt Mackall
|
r6004 | |||
# then merge tools | ||||
tools = {} | ||||
Augie Fackler
|
r26730 | disabled = set() | ||
Matt Mackall
|
r10282 | for k, v in ui.configitems("merge-tools"): | ||
Matt Mackall
|
r6004 | t = k.split('.')[0] | ||
if t not in tools: | ||||
Boris Feld
|
r34827 | tools[t] = int(_toolstr(ui, t, "priority")) | ||
if _toolbool(ui, t, "disabled"): | ||||
Augie Fackler
|
r26730 | disabled.add(t) | ||
Steve Borho
|
r6076 | names = tools.keys() | ||
Augie Fackler
|
r30388 | tools = sorted([(-p, tool) for tool, p in tools.items() | ||
if tool not in disabled]) | ||||
Steve Borho
|
r6076 | uimerge = ui.config("ui", "merge") | ||
if uimerge: | ||||
Siddharth Agarwal
|
r27039 | # external tools defined in uimerge won't be able to handle | ||
# change/delete conflicts | ||||
if uimerge not in names and not changedelete: | ||||
Steve Borho
|
r6076 | return (uimerge, uimerge) | ||
tools.insert(0, (None, uimerge)) # highest priority | ||||
Matt Mackall
|
r6004 | tools.append((None, "hgmerge")) # the old default, if found | ||
Matt Mackall
|
r10282 | for p, t in tools: | ||
Siddharth Agarwal
|
r27039 | if check(t, None, symlink, binary, changedelete): | ||
Mads Kiilerich
|
r7397 | toolpath = _findtool(ui, t) | ||
Keegan Carruthers-Smith
|
r17885 | return (t, util.shellquote(toolpath)) | ||
Matt Mackall
|
r16254 | |||
# internal merge or prompt as last resort | ||||
Siddharth Agarwal
|
r27039 | if symlink or binary or changedelete: | ||
FUJIWARA Katsunori
|
r32253 | if not changedelete and len(tools): | ||
# any tool is rejected by capability for symlink or binary | ||||
ui.warn(_("no tool found to merge %s\n") % path) | ||||
Mads Kiilerich
|
r22707 | return ":prompt", None | ||
return ":merge", None | ||||
Matt Mackall
|
r6003 | |||
Matt Mackall
|
r6005 | def _eoltype(data): | ||
"Guess the EOL type of a file" | ||||
if '\0' in data: # binary | ||||
return None | ||||
if '\r\n' in data: # Windows | ||||
return '\r\n' | ||||
if '\r' in data: # Old Mac | ||||
return '\r' | ||||
if '\n' in data: # UNIX | ||||
return '\n' | ||||
return None # unknown | ||||
Phil Cohen
|
r34783 | def _matcheol(file, back): | ||
Matt Mackall
|
r6005 | "Convert EOL markers in a file to match origfile" | ||
Phil Cohen
|
r34783 | tostyle = _eoltype(back.data()) # No repo.wread filters? | ||
Matt Mackall
|
r6005 | if tostyle: | ||
Dan Villiom Podlaski Christiansen
|
r14168 | data = util.readfile(file) | ||
Matt Mackall
|
r6005 | style = _eoltype(data) | ||
if style: | ||||
newdata = data.replace(style, tostyle) | ||||
if newdata != data: | ||||
Dan Villiom Podlaski Christiansen
|
r14168 | util.writefile(file, newdata) | ||
Matt Mackall
|
r6005 | |||
Siddharth Agarwal
|
r26526 | @internaltool('prompt', nomerge) | ||
Simon Farnsworth
|
r29774 | def _iprompt(repo, mynode, orig, fcd, fco, fca, toolconf, labels=None): | ||
timeless
|
r28640 | """Asks the user which of the local `p1()` or the other `p2()` version to | ||
keep as the merged version.""" | ||||
FUJIWARA Katsunori
|
r16125 | ui = repo.ui | ||
fd = fcd.path() | ||||
Phil Cohen
|
r35283 | # Avoid prompting during an in-memory merge since it doesn't support merge | ||
# conflicts. | ||||
if fcd.changectx().isinmemory(): | ||||
raise error.InMemoryMergeConflictsError('in-memory merge does not ' | ||||
'support file conflicts') | ||||
Simon Farnsworth
|
r29774 | prompts = partextras(labels) | ||
prompts['fd'] = fd | ||||
Siddharth Agarwal
|
r26898 | try: | ||
Siddharth Agarwal
|
r27038 | if fco.isabsent(): | ||
index = ui.promptchoice( | ||||
Stanislau Hlebik
|
r32318 | _localchangedotherdeletedmsg % prompts, 2) | ||
Siddharth Agarwal
|
r27163 | choice = ['local', 'other', 'unresolved'][index] | ||
Siddharth Agarwal
|
r27038 | elif fcd.isabsent(): | ||
index = ui.promptchoice( | ||||
Stanislau Hlebik
|
r32318 | _otherchangedlocaldeletedmsg % prompts, 2) | ||
Siddharth Agarwal
|
r27163 | choice = ['other', 'local', 'unresolved'][index] | ||
Siddharth Agarwal
|
r27038 | else: | ||
Siddharth Agarwal
|
r27162 | index = ui.promptchoice( | ||
FUJIWARA Katsunori
|
r32253 | _("keep (l)ocal%(l)s, take (o)ther%(o)s, or leave (u)nresolved" | ||
" for %(fd)s?" | ||||
Simon Farnsworth
|
r29774 | "$$ &Local $$ &Other $$ &Unresolved") % prompts, 2) | ||
Siddharth Agarwal
|
r27162 | choice = ['local', 'other', 'unresolved'][index] | ||
Siddharth Agarwal
|
r26851 | |||
Siddharth Agarwal
|
r26898 | if choice == 'other': | ||
Simon Farnsworth
|
r29774 | return _iother(repo, mynode, orig, fcd, fco, fca, toolconf, | ||
labels) | ||||
Siddharth Agarwal
|
r27162 | elif choice == 'local': | ||
Simon Farnsworth
|
r29774 | return _ilocal(repo, mynode, orig, fcd, fco, fca, toolconf, | ||
labels) | ||||
Siddharth Agarwal
|
r27162 | elif choice == 'unresolved': | ||
Simon Farnsworth
|
r29774 | return _ifail(repo, mynode, orig, fcd, fco, fca, toolconf, | ||
labels) | ||||
Siddharth Agarwal
|
r26898 | except error.ResponseExpected: | ||
ui.write("\n") | ||||
Simon Farnsworth
|
r29774 | return _ifail(repo, mynode, orig, fcd, fco, fca, toolconf, | ||
labels) | ||||
FUJIWARA Katsunori
|
r16125 | |||
Siddharth Agarwal
|
r26526 | @internaltool('local', nomerge) | ||
Simon Farnsworth
|
r29774 | def _ilocal(repo, mynode, orig, fcd, fco, fca, toolconf, labels=None): | ||
timeless
|
r28640 | """Uses the local `p1()` version of files as the merged version.""" | ||
Siddharth Agarwal
|
r27036 | return 0, fcd.isabsent() | ||
FUJIWARA Katsunori
|
r16125 | |||
Siddharth Agarwal
|
r26526 | @internaltool('other', nomerge) | ||
Simon Farnsworth
|
r29774 | def _iother(repo, mynode, orig, fcd, fco, fca, toolconf, labels=None): | ||
timeless
|
r28640 | """Uses the other `p2()` version of files as the merged version.""" | ||
Siddharth Agarwal
|
r27037 | if fco.isabsent(): | ||
# local changed, remote deleted -- 'deleted' picked | ||||
Phil Cohen
|
r33152 | _underlyingfctxifabsent(fcd).remove() | ||
Siddharth Agarwal
|
r27037 | deleted = True | ||
else: | ||||
Phil Cohen
|
r33152 | _underlyingfctxifabsent(fcd).write(fco.data(), fco.flags()) | ||
Siddharth Agarwal
|
r27037 | deleted = False | ||
return 0, deleted | ||||
FUJIWARA Katsunori
|
r16125 | |||
Siddharth Agarwal
|
r26526 | @internaltool('fail', nomerge) | ||
Simon Farnsworth
|
r29774 | def _ifail(repo, mynode, orig, fcd, fco, fca, toolconf, labels=None): | ||
Matt Mackall
|
r16127 | """ | ||
FUJIWARA Katsunori
|
r16126 | Rather than attempting to merge files that were modified on both | ||
branches, it marks them as unresolved. The resolve command must be | ||||
used to resolve these conflicts.""" | ||||
Siddharth Agarwal
|
r27123 | # for change/delete conflicts write out the changed version, then fail | ||
if fcd.isabsent(): | ||||
Phil Cohen
|
r33152 | _underlyingfctxifabsent(fcd).write(fco.data(), fco.flags()) | ||
Siddharth Agarwal
|
r27032 | return 1, False | ||
FUJIWARA Katsunori
|
r16125 | |||
Phil Cohen
|
r33152 | def _underlyingfctxifabsent(filectx): | ||
"""Sometimes when resolving, our fcd is actually an absentfilectx, but | ||||
we want to write to it (to do the resolve). This helper returns the | ||||
underyling workingfilectx in that case. | ||||
""" | ||||
if filectx.isabsent(): | ||||
return filectx.changectx()[filectx.path()] | ||||
else: | ||||
return filectx | ||||
Siddharth Agarwal
|
r27041 | def _premerge(repo, fcd, fco, fca, toolconf, files, labels=None): | ||
FUJIWARA Katsunori
|
r16125 | tool, toolpath, binary, symlink = toolconf | ||
Siddharth Agarwal
|
r27041 | if symlink or fcd.isabsent() or fco.isabsent(): | ||
Mads Kiilerich
|
r18257 | return 1 | ||
Phil Cohen
|
r34034 | unused, unused, unused, back = files | ||
FUJIWARA Katsunori
|
r16125 | |||
ui = repo.ui | ||||
Pierre-Yves David
|
r22032 | validkeep = ['keep', 'keep-merge3'] | ||
Pierre-Yves David
|
r22031 | |||
FUJIWARA Katsunori
|
r16125 | # do we attempt to simplemerge first? | ||
try: | ||||
Mads Kiilerich
|
r18257 | premerge = _toolbool(ui, tool, "premerge", not binary) | ||
FUJIWARA Katsunori
|
r16125 | except error.ConfigError: | ||
Boris Feld
|
r34827 | premerge = _toolstr(ui, tool, "premerge", "").lower() | ||
Pierre-Yves David
|
r22031 | if premerge not in validkeep: | ||
_valid = ', '.join(["'" + v + "'" for v in validkeep]) | ||||
FUJIWARA Katsunori
|
r16125 | raise error.ConfigError(_("%s.premerge not valid " | ||
"('%s' is neither boolean nor %s)") % | ||||
(tool, premerge, _valid)) | ||||
if premerge: | ||||
Pierre-Yves David
|
r22032 | if premerge == 'keep-merge3': | ||
if not labels: | ||||
labels = _defaultconflictlabels | ||||
if len(labels) < 3: | ||||
labels.append('base') | ||||
Phil Cohen
|
r34051 | r = simplemerge.simplemerge(ui, fcd, fca, fco, quiet=True, label=labels) | ||
FUJIWARA Katsunori
|
r16125 | if not r: | ||
ui.debug(" premerge successful\n") | ||||
return 0 | ||||
Pierre-Yves David
|
r22031 | if premerge not in validkeep: | ||
Phil Cohen
|
r34034 | # restore from backup and try again | ||
Phil Cohen
|
r34038 | _restorebackup(fcd, back) | ||
FUJIWARA Katsunori
|
r16125 | return 1 # continue merging | ||
Siddharth Agarwal
|
r26948 | def _mergecheck(repo, mynode, orig, fcd, fco, fca, toolconf): | ||
Siddharth Agarwal
|
r26515 | tool, toolpath, binary, symlink = toolconf | ||
if symlink: | ||||
Siddharth Agarwal
|
r26518 | repo.ui.warn(_('warning: internal %s cannot merge symlinks ' | ||
'for %s\n') % (tool, fcd.path())) | ||||
Siddharth Agarwal
|
r26515 | return False | ||
Siddharth Agarwal
|
r27040 | if fcd.isabsent() or fco.isabsent(): | ||
repo.ui.warn(_('warning: internal %s cannot merge change/delete ' | ||||
'conflict for %s\n') % (tool, fcd.path())) | ||||
return False | ||||
Siddharth Agarwal
|
r26515 | return True | ||
Erik Huelsmann
|
r26070 | def _merge(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels, mode): | ||
Matt Mackall
|
r16127 | """ | ||
FUJIWARA Katsunori
|
r16126 | Uses the internal non-interactive simple merge algorithm for merging | ||
files. It will fail if there are any conflicts and leave markers in | ||||
Pierre-Yves David
|
r22027 | the partially merged file. Markers will have two sections, one for each side | ||
Erik Huelsmann
|
r26070 | of merge, unless mode equals 'union' which suppresses the markers.""" | ||
Siddharth Agarwal
|
r26572 | ui = repo.ui | ||
FUJIWARA Katsunori
|
r16125 | |||
Phil Cohen
|
r34051 | r = simplemerge.simplemerge(ui, fcd, fca, fco, label=labels, mode=mode) | ||
Siddharth Agarwal
|
r27033 | return True, r, False | ||
FUJIWARA Katsunori
|
r16125 | |||
Siddharth Agarwal
|
r26526 | @internaltool('union', fullmerge, | ||
Siddharth Agarwal
|
r26614 | _("warning: conflicts while merging %s! " | ||
"(edit, then use 'hg resolve --mark')\n"), | ||||
Siddharth Agarwal
|
r26948 | precheck=_mergecheck) | ||
Erik Huelsmann
|
r26071 | def _iunion(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
""" | ||||
Uses the internal non-interactive simple merge algorithm for merging | ||||
files. It will use both left and right sides for conflict regions. | ||||
No markers are inserted.""" | ||||
return _merge(repo, mynode, orig, fcd, fco, fca, toolconf, | ||||
files, labels, 'union') | ||||
Siddharth Agarwal
|
r26526 | @internaltool('merge', fullmerge, | ||
Siddharth Agarwal
|
r26614 | _("warning: conflicts while merging %s! " | ||
"(edit, then use 'hg resolve --mark')\n"), | ||||
Siddharth Agarwal
|
r26948 | precheck=_mergecheck) | ||
Erik Huelsmann
|
r26070 | def _imerge(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
""" | ||||
Uses the internal non-interactive simple merge algorithm for merging | ||||
files. It will fail if there are any conflicts and leave markers in | ||||
the partially merged file. Markers will have two sections, one for each side | ||||
of merge.""" | ||||
return _merge(repo, mynode, orig, fcd, fco, fca, toolconf, | ||||
files, labels, 'merge') | ||||
Siddharth Agarwal
|
r26526 | @internaltool('merge3', fullmerge, | ||
Siddharth Agarwal
|
r26614 | _("warning: conflicts while merging %s! " | ||
"(edit, then use 'hg resolve --mark')\n"), | ||||
Siddharth Agarwal
|
r26948 | precheck=_mergecheck) | ||
Pierre-Yves David
|
r22028 | def _imerge3(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
""" | ||||
Uses the internal non-interactive simple merge algorithm for merging | ||||
files. It will fail if there are any conflicts and leave markers in | ||||
the partially merged file. Marker will have three sections, one from each | ||||
side of the merge and one for the base content.""" | ||||
if not labels: | ||||
labels = _defaultconflictlabels | ||||
if len(labels) < 3: | ||||
labels.append('base') | ||||
return _imerge(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels) | ||||
Jordi Gutiérrez Hermoso
|
r26224 | def _imergeauto(repo, mynode, orig, fcd, fco, fca, toolconf, files, | ||
labels=None, localorother=None): | ||||
""" | ||||
Generic driver for _imergelocal and _imergeother | ||||
""" | ||||
assert localorother is not None | ||||
tool, toolpath, binary, symlink = toolconf | ||||
Phil Cohen
|
r34051 | r = simplemerge.simplemerge(repo.ui, fcd, fca, fco, label=labels, | ||
localorother=localorother) | ||||
Jordi Gutiérrez Hermoso
|
r26224 | return True, r | ||
Siddharth Agarwal
|
r26948 | @internaltool('merge-local', mergeonly, precheck=_mergecheck) | ||
Jordi Gutiérrez Hermoso
|
r26224 | def _imergelocal(*args, **kwargs): | ||
""" | ||||
Like :merge, but resolve all conflicts non-interactively in favor | ||||
timeless
|
r28640 | of the local `p1()` changes.""" | ||
Jordi Gutiérrez Hermoso
|
r26224 | success, status = _imergeauto(localorother='local', *args, **kwargs) | ||
Siddharth Agarwal
|
r27033 | return success, status, False | ||
Jordi Gutiérrez Hermoso
|
r26224 | |||
Siddharth Agarwal
|
r26948 | @internaltool('merge-other', mergeonly, precheck=_mergecheck) | ||
Jordi Gutiérrez Hermoso
|
r26224 | def _imergeother(*args, **kwargs): | ||
""" | ||||
Like :merge, but resolve all conflicts non-interactively in favor | ||||
timeless
|
r28640 | of the other `p2()` changes.""" | ||
Jordi Gutiérrez Hermoso
|
r26224 | success, status = _imergeauto(localorother='other', *args, **kwargs) | ||
Siddharth Agarwal
|
r27033 | return success, status, False | ||
Jordi Gutiérrez Hermoso
|
r26224 | |||
Siddharth Agarwal
|
r26526 | @internaltool('tagmerge', mergeonly, | ||
Angel Ezquerra
|
r21922 | _("automatic tag merging of %s failed! " | ||
Mads Kiilerich
|
r22707 | "(use 'hg resolve --tool :merge' or another merge " | ||
Angel Ezquerra
|
r21922 | "tool of your choice)\n")) | ||
def _itagmerge(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||||
""" | ||||
Uses the internal tag merge algorithm (experimental). | ||||
""" | ||||
Siddharth Agarwal
|
r27033 | success, status = tagmerge.merge(repo, fcd, fco, fca) | ||
return success, status, False | ||||
Angel Ezquerra
|
r21922 | |||
Siddharth Agarwal
|
r26526 | @internaltool('dump', fullmerge) | ||
Durham Goode
|
r21273 | def _idump(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
Matt Mackall
|
r16127 | """ | ||
FUJIWARA Katsunori
|
r16126 | Creates three versions of the files to merge, containing the | ||
contents of local, other and base. These files can then be used to | ||||
perform a merge manually. If the file to be merged is named | ||||
``a.txt``, these files will accordingly be named ``a.txt.local``, | ||||
``a.txt.other`` and ``a.txt.base`` and they will be placed in the | ||||
FUJIWARA Katsunori
|
r32255 | same directory as ``a.txt``. | ||
Joe Blaylock
|
r34916 | This implies premerge. Therefore, files aren't dumped, if premerge | ||
FUJIWARA Katsunori
|
r32255 | runs successfully. Use :forcedump to forcibly write files out. | ||
""" | ||||
Phil Cohen
|
r34036 | a = _workingpath(repo, fcd) | ||
Siddharth Agarwal
|
r26573 | fd = fcd.path() | ||
FUJIWARA Katsunori
|
r16125 | |||
Phil Cohen
|
r34786 | from . import context | ||
if isinstance(fcd, context.overlayworkingfilectx): | ||||
Phil Cohen
|
r35283 | raise error.InMemoryMergeConflictsError('in-memory merge does not ' | ||
'support the :dump tool.') | ||||
Phil Cohen
|
r34786 | |||
Phil Cohen
|
r34078 | util.writefile(a + ".local", fcd.decodeddata()) | ||
Siddharth Agarwal
|
r26573 | repo.wwrite(fd + ".other", fco.data(), fco.flags()) | ||
repo.wwrite(fd + ".base", fca.data(), fca.flags()) | ||||
Siddharth Agarwal
|
r27033 | return False, 1, False | ||
FUJIWARA Katsunori
|
r16125 | |||
FUJIWARA Katsunori
|
r32255 | @internaltool('forcedump', mergeonly) | ||
def _forcedump(repo, mynode, orig, fcd, fco, fca, toolconf, files, | ||||
labels=None): | ||||
""" | ||||
Creates three versions of the files as same as :dump, but omits premerge. | ||||
""" | ||||
return _idump(repo, mynode, orig, fcd, fco, fca, toolconf, files, | ||||
labels=labels) | ||||
Phil Cohen
|
r35479 | def _xmergeimm(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
# In-memory merge simply raises an exception on all external merge tools, | ||||
# for now. | ||||
# | ||||
# It would be possible to run most tools with temporary files, but this | ||||
# raises the question of what to do if the user only partially resolves the | ||||
# file -- we can't leave a merge state. (Copy to somewhere in the .hg/ | ||||
# directory and tell the user how to get it is my best idea, but it's | ||||
# clunky.) | ||||
raise error.InMemoryMergeConflictsError('in-memory merge does not support ' | ||||
'external merge tools') | ||||
Durham Goode
|
r21273 | def _xmerge(repo, mynode, orig, fcd, fco, fca, toolconf, files, labels=None): | ||
Siddharth Agarwal
|
r26574 | tool, toolpath, binary, symlink = toolconf | ||
Siddharth Agarwal
|
r27042 | if fcd.isabsent() or fco.isabsent(): | ||
repo.ui.warn(_('warning: %s cannot merge change/delete conflict ' | ||||
'for %s\n') % (tool, fcd.path())) | ||||
return False, 1, None | ||||
Phil Cohen
|
r34037 | unused, unused, unused, back = files | ||
Phil Cohen
|
r34036 | a = _workingpath(repo, fcd) | ||
Phil Cohen
|
r34037 | b, c = _maketempfiles(repo, fco, fca) | ||
try: | ||||
out = "" | ||||
env = {'HG_FILE': fcd.path(), | ||||
'HG_MY_NODE': short(mynode), | ||||
'HG_OTHER_NODE': str(fco.changectx()), | ||||
'HG_BASE_NODE': str(fca.changectx()), | ||||
'HG_MY_ISLINK': 'l' in fcd.flags(), | ||||
'HG_OTHER_ISLINK': 'l' in fco.flags(), | ||||
'HG_BASE_ISLINK': 'l' in fca.flags(), | ||||
} | ||||
ui = repo.ui | ||||
FUJIWARA Katsunori
|
r16125 | |||
Boris Feld
|
r34827 | args = _toolstr(ui, tool, "args") | ||
Phil Cohen
|
r34037 | if "$output" in args: | ||
Phil Cohen
|
r34783 | # read input from backup, write to original | ||
out = a | ||||
a = repo.wvfs.join(back.path()) | ||||
Phil Cohen
|
r34037 | replace = {'local': a, 'base': b, 'other': c, 'output': out} | ||
args = util.interpolate(r'\$', replace, args, | ||||
lambda s: util.shellquote(util.localpath(s))) | ||||
cmd = toolpath + ' ' + args | ||||
if _toolbool(ui, tool, "gui"): | ||||
repo.ui.status(_('running merge tool %s for file %s\n') % | ||||
(tool, fcd.path())) | ||||
repo.ui.debug('launching merge tool: %s\n' % cmd) | ||||
r = ui.system(cmd, cwd=repo.root, environ=env, blockedtag='mergetool') | ||||
Pulkit Goyal
|
r34507 | repo.ui.debug('merge tool returned: %d\n' % r) | ||
Phil Cohen
|
r34037 | return True, r, False | ||
finally: | ||||
util.unlink(b) | ||||
util.unlink(c) | ||||
FUJIWARA Katsunori
|
r16125 | |||
Yuya Nishihara
|
r35498 | def _formatconflictmarker(ctx, template, label, pad): | ||
Durham Goode
|
r21519 | """Applies the given template to the ctx, prefixed by the label. | ||
Pad is the minimum width of the label prefix, so that multiple markers | ||||
can have aligned templated parts. | ||||
""" | ||||
if ctx.node() is None: | ||||
ctx = ctx.p1() | ||||
props = templatekw.keywords.copy() | ||||
props['ctx'] = ctx | ||||
Yuya Nishihara
|
r32873 | templateresult = template.render(props) | ||
Durham Goode
|
r21519 | |||
label = ('%s:' % label).ljust(pad + 1) | ||||
Yuya Nishihara
|
r32873 | mark = '%s %s' % (label, templateresult) | ||
Durham Goode
|
r21519 | |||
FUJIWARA Katsunori
|
r21864 | if mark: | ||
mark = mark.splitlines()[0] # split for safety | ||||
FUJIWARA Katsunori
|
r21865 | # 8 for the prefix of conflict marker lines (e.g. '<<<<<<< ') | ||
return util.ellipsis(mark, 80 - 8) | ||||
Durham Goode
|
r21519 | |||
Durham Goode
|
r21524 | _defaultconflictlabels = ['local', 'other'] | ||
Pierre-Yves David
|
r22026 | def _formatlabels(repo, fcd, fco, fca, labels): | ||
Durham Goode
|
r21519 | """Formats the given labels using the conflict marker template. | ||
Returns a list of formatted labels. | ||||
""" | ||||
cd = fcd.changectx() | ||||
co = fco.changectx() | ||||
Pierre-Yves David
|
r22026 | ca = fca.changectx() | ||
Durham Goode
|
r21519 | |||
ui = repo.ui | ||||
Boris Feld
|
r33523 | template = ui.config('ui', 'mergemarkertemplate') | ||
Yuya Nishihara
|
r32047 | template = templater.unquotestring(template) | ||
Yuya Nishihara
|
r35485 | tres = formatter.templateresources(ui, repo) | ||
tmpl = formatter.maketemplater(ui, template, resources=tres) | ||||
Durham Goode
|
r21519 | |||
Pierre-Yves David
|
r22026 | pad = max(len(l) for l in labels) | ||
Durham Goode
|
r21519 | |||
Yuya Nishihara
|
r35498 | newlabels = [_formatconflictmarker(cd, tmpl, labels[0], pad), | ||
_formatconflictmarker(co, tmpl, labels[1], pad)] | ||||
Pierre-Yves David
|
r22026 | if len(labels) > 2: | ||
Yuya Nishihara
|
r35498 | newlabels.append(_formatconflictmarker(ca, tmpl, labels[2], pad)) | ||
Pierre-Yves David
|
r22026 | return newlabels | ||
Durham Goode
|
r21519 | |||
Simon Farnsworth
|
r29774 | def partextras(labels): | ||
"""Return a dictionary of extra labels for use in prompts to the user | ||||
Intended use is in strings of the form "(l)ocal%(l)s". | ||||
""" | ||||
if labels is None: | ||||
return { | ||||
"l": "", | ||||
"o": "", | ||||
} | ||||
return { | ||||
"l": " [%s]" % labels[0], | ||||
"o": " [%s]" % labels[1], | ||||
} | ||||
Phil Cohen
|
r34038 | def _restorebackup(fcd, back): | ||
# TODO: Add a workingfilectx.write(otherfilectx) path so we can use | ||||
# util.copy here instead. | ||||
Phil Cohen
|
r34783 | fcd.write(back.data(), fcd.flags()) | ||
Phil Cohen
|
r34038 | |||
Phil Cohen
|
r34783 | def _makebackup(repo, ui, wctx, fcd, premerge): | ||
"""Makes and returns a filectx-like object for ``fcd``'s backup file. | ||||
Phil Cohen
|
r34033 | |||
In addition to preserving the user's pre-existing modifications to `fcd` | ||||
(if any), the backup is used to undo certain premerges, confirm whether a | ||||
merge changed anything, and determine what line endings the new file should | ||||
have. | ||||
""" | ||||
if fcd.isabsent(): | ||||
return None | ||||
Phil Cohen
|
r34783 | # TODO: Break this import cycle somehow. (filectx -> ctx -> fileset -> | ||
# merge -> filemerge). (I suspect the fileset import is the weakest link) | ||||
from . import context | ||||
Phil Cohen
|
r34036 | a = _workingpath(repo, fcd) | ||
Phil Cohen
|
r34033 | back = scmutil.origpath(ui, repo, a) | ||
Phil Cohen
|
r34785 | inworkingdir = (back.startswith(repo.wvfs.base) and not | ||
back.startswith(repo.vfs.base)) | ||||
if isinstance(fcd, context.overlayworkingfilectx) and inworkingdir: | ||||
# If the backup file is to be in the working directory, and we're | ||||
# merging in-memory, we must redirect the backup to the memory context | ||||
# so we don't disturb the working directory. | ||||
relpath = back[len(repo.wvfs.base) + 1:] | ||||
wctx[relpath].write(fcd.data(), fcd.flags()) | ||||
return wctx[relpath] | ||||
else: | ||||
# Otherwise, write to wherever the user specified the backups should go. | ||||
# | ||||
# A arbitraryfilectx is returned, so we can run the same functions on | ||||
# the backup context regardless of where it lives. | ||||
if premerge: | ||||
util.copyfile(a, back) | ||||
return context.arbitraryfilectx(back, repo=repo) | ||||
Phil Cohen
|
r34033 | |||
Phil Cohen
|
r34037 | def _maketempfiles(repo, fco, fca): | ||
Phil Cohen
|
r34033 | """Writes out `fco` and `fca` as temporary files, so an external merge | ||
tool may use them. | ||||
""" | ||||
def temp(prefix, ctx): | ||||
fullbase, ext = os.path.splitext(ctx.path()) | ||||
pre = "%s~%s." % (os.path.basename(fullbase), prefix) | ||||
(fd, name) = tempfile.mkstemp(prefix=pre, suffix=ext) | ||||
data = repo.wwritedata(ctx.path(), ctx.data()) | ||||
f = os.fdopen(fd, pycompat.sysstr("wb")) | ||||
f.write(data) | ||||
f.close() | ||||
return name | ||||
b = temp("base", fca) | ||||
c = temp("other", fco) | ||||
Phil Cohen
|
r34037 | return b, c | ||
Phil Cohen
|
r34033 | |||
Phil Cohen
|
r34124 | def _filemerge(premerge, repo, wctx, mynode, orig, fcd, fco, fca, labels=None): | ||
Matt Mackall
|
r6003 | """perform a 3-way merge in the working directory | ||
Siddharth Agarwal
|
r26607 | premerge = whether this is a premerge | ||
Matt Mackall
|
r6512 | mynode = parent node before merge | ||
orig = original local filename before merge | ||||
fco = other file context | ||||
fca = ancestor file context | ||||
fcd = local file context for current/destination file | ||||
Siddharth Agarwal
|
r26606 | |||
Siddharth Agarwal
|
r27034 | Returns whether the merge is complete, the return value of the merge, and | ||
a boolean indicating whether the file was deleted from disk.""" | ||||
Matt Mackall
|
r6003 | |||
Siddharth Agarwal
|
r26608 | if not fco.cmp(fcd): # files identical? | ||
Siddharth Agarwal
|
r27034 | return True, None, False | ||
Siddharth Agarwal
|
r26512 | |||
Siddharth Agarwal
|
r26608 | ui = repo.ui | ||
fd = fcd.path() | ||||
binary = fcd.isbinary() or fco.isbinary() or fca.isbinary() | ||||
symlink = 'l' in fcd.flags() + fco.flags() | ||||
Siddharth Agarwal
|
r27039 | changedelete = fcd.isabsent() or fco.isabsent() | ||
tool, toolpath = _picktool(repo, ui, fd, binary, symlink, changedelete) | ||||
Siddharth Agarwal
|
r26608 | if tool in internals and tool.startswith('internal:'): | ||
# normalize to new-style names (':merge' etc) | ||||
tool = tool[len('internal'):] | ||||
Siddharth Agarwal
|
r27161 | ui.debug("picked tool '%s' for %s (binary %s symlink %s changedelete %s)\n" | ||
Pulkit Goyal
|
r32753 | % (tool, fd, pycompat.bytestr(binary), pycompat.bytestr(symlink), | ||
pycompat.bytestr(changedelete))) | ||||
Matt Mackall
|
r6003 | |||
Siddharth Agarwal
|
r26608 | if tool in internals: | ||
func = internals[tool] | ||||
mergetype = func.mergetype | ||||
onfailure = func.onfailure | ||||
precheck = func.precheck | ||||
else: | ||||
Phil Cohen
|
r35479 | if wctx.isinmemory(): | ||
func = _xmergeimm | ||||
else: | ||||
func = _xmerge | ||||
Siddharth Agarwal
|
r26608 | mergetype = fullmerge | ||
onfailure = _("merging %s failed!\n") | ||||
precheck = None | ||||
Siddharth Agarwal
|
r26512 | |||
Siddharth Agarwal
|
r26608 | toolconf = tool, toolpath, binary, symlink | ||
Matt Mackall
|
r6003 | |||
Siddharth Agarwal
|
r26608 | if mergetype == nomerge: | ||
Simon Farnsworth
|
r29774 | r, deleted = func(repo, mynode, orig, fcd, fco, fca, toolconf, labels) | ||
Siddharth Agarwal
|
r27034 | return True, r, deleted | ||
Siddharth Agarwal
|
r26512 | |||
Siddharth Agarwal
|
r26609 | if premerge: | ||
if orig != fco.path(): | ||||
ui.status(_("merging %s and %s to %s\n") % (orig, fco.path(), fd)) | ||||
else: | ||||
ui.status(_("merging %s\n") % fd) | ||||
Siddharth Agarwal
|
r26528 | |||
Siddharth Agarwal
|
r26608 | ui.debug("my %s other %s ancestor %s\n" % (fcd, fco, fca)) | ||
Siddharth Agarwal
|
r26528 | |||
Siddharth Agarwal
|
r26608 | if precheck and not precheck(repo, mynode, orig, fcd, fco, fca, | ||
toolconf): | ||||
if onfailure: | ||||
Phil Cohen
|
r35283 | if wctx.isinmemory(): | ||
raise error.InMemoryMergeConflictsError('in-memory merge does ' | ||||
'not support merge ' | ||||
'conflicts') | ||||
Siddharth Agarwal
|
r26608 | ui.warn(onfailure % fd) | ||
Siddharth Agarwal
|
r27034 | return True, 1, False | ||
Siddharth Agarwal
|
r26529 | |||
Phil Cohen
|
r34783 | back = _makebackup(repo, ui, wctx, fcd, premerge) | ||
Phil Cohen
|
r34037 | files = (None, None, None, back) | ||
Siddharth Agarwal
|
r26589 | r = 1 | ||
try: | ||||
Jun Wu
|
r33499 | markerstyle = ui.config('ui', 'mergemarkers') | ||
Siddharth Agarwal
|
r26529 | if not labels: | ||
labels = _defaultconflictlabels | ||||
if markerstyle != 'basic': | ||||
labels = _formatlabels(repo, fcd, fco, fca, labels) | ||||
Matt Mackall
|
r6004 | |||
Siddharth Agarwal
|
r26607 | if premerge and mergetype == fullmerge: | ||
Siddharth Agarwal
|
r27041 | r = _premerge(repo, fcd, fco, fca, toolconf, files, labels=labels) | ||
Siddharth Agarwal
|
r26611 | # complete if premerge successful (r is 0) | ||
Siddharth Agarwal
|
r27034 | return not r, r, False | ||
Siddharth Agarwal
|
r26567 | |||
Siddharth Agarwal
|
r27033 | needcheck, r, deleted = func(repo, mynode, orig, fcd, fco, fca, | ||
toolconf, files, labels=labels) | ||||
Siddharth Agarwal
|
r26575 | if needcheck: | ||
Phil Cohen
|
r34036 | r = _check(repo, r, ui, tool, fcd, files) | ||
Durham Goode
|
r21519 | |||
FUJIWARA Katsunori
|
r16125 | if r: | ||
if onfailure: | ||||
Phil Cohen
|
r35283 | if wctx.isinmemory(): | ||
raise error.InMemoryMergeConflictsError('in-memory merge ' | ||||
'does not support ' | ||||
'merge conflicts') | ||||
FUJIWARA Katsunori
|
r16125 | ui.warn(onfailure % fd) | ||
Ryan McElroy
|
r34798 | _onfilemergefailure(ui) | ||
Siddharth Agarwal
|
r26589 | |||
Siddharth Agarwal
|
r27034 | return True, r, deleted | ||
Siddharth Agarwal
|
r26589 | finally: | ||
Siddharth Agarwal
|
r27047 | if not r and back is not None: | ||
Phil Cohen
|
r34783 | back.remove() | ||
Matt Mackall
|
r6004 | |||
Ryan McElroy
|
r34797 | def _haltmerge(): | ||
msg = _('merge halted after failed merge (see hg resolve)') | ||||
raise error.InterventionRequired(msg) | ||||
def _onfilemergefailure(ui): | ||||
action = ui.config('merge', 'on-failure') | ||||
if action == 'prompt': | ||||
msg = _('continue merge operation (yn)?' '$$ &Yes $$ &No') | ||||
if ui.promptchoice(msg, 0) == 1: | ||||
_haltmerge() | ||||
if action == 'halt': | ||||
_haltmerge() | ||||
# default action is 'continue', in which case we neither prompt nor halt | ||||
Phil Cohen
|
r34036 | def _check(repo, r, ui, tool, fcd, files): | ||
Siddharth Agarwal
|
r26575 | fd = fcd.path() | ||
Phil Cohen
|
r34036 | unused, unused, unused, back = files | ||
Siddharth Agarwal
|
r26575 | |||
if not r and (_toolbool(ui, tool, "checkconflicts") or | ||||
'conflicts' in _toollist(ui, tool, "check")): | ||||
if re.search("^(<<<<<<< .*|=======|>>>>>>> .*)$", fcd.data(), | ||||
re.MULTILINE): | ||||
r = 1 | ||||
checked = False | ||||
if 'prompt' in _toollist(ui, tool, "check"): | ||||
checked = True | ||||
if ui.promptchoice(_("was merge of '%s' successful (yn)?" | ||||
"$$ &Yes $$ &No") % fd, 1): | ||||
r = 1 | ||||
if not r and not checked and (_toolbool(ui, tool, "checkchanged") or | ||||
'changed' in | ||||
_toollist(ui, tool, "check")): | ||||
Phil Cohen
|
r34783 | if back is not None and not fcd.cmp(back): | ||
Siddharth Agarwal
|
r26575 | if ui.promptchoice(_(" output file %s appears unchanged\n" | ||
"was merge successful (yn)?" | ||||
"$$ &Yes $$ &No") % fd, 1): | ||||
r = 1 | ||||
Siddharth Agarwal
|
r27047 | if back is not None and _toolbool(ui, tool, "fixeol"): | ||
Phil Cohen
|
r34036 | _matcheol(_workingpath(repo, fcd), back) | ||
Siddharth Agarwal
|
r26575 | |||
return r | ||||
Phil Cohen
|
r34036 | def _workingpath(repo, ctx): | ||
return repo.wjoin(ctx.path()) | ||||
Phil Cohen
|
r34124 | def premerge(repo, wctx, mynode, orig, fcd, fco, fca, labels=None): | ||
return _filemerge(True, repo, wctx, mynode, orig, fcd, fco, fca, | ||||
labels=labels) | ||||
Siddharth Agarwal
|
r26607 | |||
Phil Cohen
|
r34124 | def filemerge(repo, wctx, mynode, orig, fcd, fco, fca, labels=None): | ||
return _filemerge(False, repo, wctx, mynode, orig, fcd, fco, fca, | ||||
labels=labels) | ||||
Siddharth Agarwal
|
r26605 | |||
FUJIWARA Katsunori
|
r33663 | def loadinternalmerge(ui, extname, registrarobj): | ||
"""Load internal merge tool from specified registrarobj | ||||
""" | ||||
for name, func in registrarobj._table.iteritems(): | ||||
fullname = ':' + name | ||||
internals[fullname] = func | ||||
internals['internal:' + name] = func | ||||
internalsdoc[fullname] = func | ||||
# load built-in merge tools explicitly to setup internalsdoc | ||||
loadinternalmerge(None, None, internaltool) | ||||
FUJIWARA Katsunori
|
r16126 | # tell hggettext to extract docstrings from these functions: | ||
i18nfunctions = internals.values() | ||||