merge.py
1653 lines
| 61.6 KiB
| text/x-python
|
PythonLexer
/ mercurial / merge.py
Matt Mackall
|
r2775 | # merge.py - directory-level update/merge handling for Mercurial | ||
# | ||||
Thomas Arendsen Hein
|
r4635 | # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com> | ||
Matt Mackall
|
r2775 | # | ||
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
|
r2775 | |||
Gregory Szorc
|
r25959 | from __future__ import absolute_import | ||
import errno | ||||
import os | ||||
import shutil | ||||
Pierre-Yves David
|
r20590 | import struct | ||
Gregory Szorc
|
r25959 | from .i18n import _ | ||
from .node import ( | ||||
bin, | ||||
hex, | ||||
Siddharth Agarwal
|
r27031 | nullhex, | ||
Gregory Szorc
|
r25959 | nullid, | ||
nullrev, | ||||
) | ||||
from . import ( | ||||
copies, | ||||
Pierre-Yves David
|
r26570 | destutil, | ||
Pierre-Yves David
|
r26587 | error, | ||
Gregory Szorc
|
r25959 | filemerge, | ||
obsolete, | ||||
Siddharth Agarwal
|
r27656 | scmutil, | ||
Gregory Szorc
|
r25959 | subrepo, | ||
util, | ||||
worker, | ||||
) | ||||
Matt Mackall
|
r6512 | |||
Pierre-Yves David
|
r20590 | _pack = struct.pack | ||
_unpack = struct.unpack | ||||
Pierre-Yves David
|
r20593 | def _droponode(data): | ||
# used for compatibility for v1 | ||||
Martin von Zweigbergk
|
r23380 | bits = data.split('\0') | ||
Pierre-Yves David
|
r20593 | bits = bits[:-2] + bits[-1:] | ||
Martin von Zweigbergk
|
r23380 | return '\0'.join(bits) | ||
Pierre-Yves David
|
r20593 | |||
Matt Mackall
|
r6512 | class mergestate(object): | ||
Pierre-Yves David
|
r20590 | '''track 3-way merge state of individual files | ||
Siddharth Agarwal
|
r27022 | The merge state is stored on disk when needed. Two files are used: one with | ||
an old format (version 1), and one with a new format (version 2). Version 2 | ||||
stores a superset of the data in version 1, including new kinds of records | ||||
in the future. For more about the new format, see the documentation for | ||||
`_readrecordsv2`. | ||||
Pierre-Yves David
|
r20590 | |||
Siddharth Agarwal
|
r27022 | Each record can contain arbitrary content, and has an associated type. This | ||
`type` should be a letter. If `type` is uppercase, the record is mandatory: | ||||
versions of Mercurial that don't support it should abort. If `type` is | ||||
lowercase, the record can be safely ignored. | ||||
Pierre-Yves David
|
r20590 | |||
Siddharth Agarwal
|
r27022 | Currently known records: | ||
Pierre-Yves David
|
r20590 | |||
L: the node of the "local" part of the merge (hexified version) | ||||
Pierre-Yves David
|
r20591 | O: the node of the "other" part of the merge (hexified version) | ||
Pierre-Yves David
|
r20590 | F: a file to be merged entry | ||
Siddharth Agarwal
|
r27031 | C: a change/delete or delete/change conflict | ||
Siddharth Agarwal
|
r26650 | D: a file that the external merge driver will merge internally | ||
(experimental) | ||||
Siddharth Agarwal
|
r26649 | m: the external merge driver defined for this merge plus its run state | ||
(experimental) | ||||
Durham Goode
|
r28009 | f: a (filename, dictonary) tuple of optional values for a given file | ||
Siddharth Agarwal
|
r27027 | X: unsupported mandatory record type (used in tests) | ||
x: unsupported advisory record type (used in tests) | ||||
Simon Farnsworth
|
r28634 | l: the labels for the parts of the merge. | ||
Siddharth Agarwal
|
r26649 | |||
Merge driver run states (experimental): | ||||
u: driver-resolved files unmarked -- needs to be run next time we're about | ||||
to resolve or commit | ||||
m: driver-resolved files marked -- only needs to be run before commit | ||||
s: success/skipped -- does not need to be run any more | ||||
Siddharth Agarwal
|
r27022 | |||
Pierre-Yves David
|
r20590 | ''' | ||
Martin von Zweigbergk
|
r23380 | statepathv1 = 'merge/state' | ||
statepathv2 = 'merge/state2' | ||||
Pierre-Yves David
|
r20651 | |||
Siddharth Agarwal
|
r26987 | @staticmethod | ||
Simon Farnsworth
|
r28634 | def clean(repo, node=None, other=None, labels=None): | ||
Siddharth Agarwal
|
r26987 | """Initialize a brand new merge state, removing any existing state on | ||
disk.""" | ||||
ms = mergestate(repo) | ||||
Simon Farnsworth
|
r28634 | ms.reset(node, other, labels) | ||
Siddharth Agarwal
|
r26987 | return ms | ||
Siddharth Agarwal
|
r26991 | @staticmethod | ||
def read(repo): | ||||
"""Initialize the merge state, reading it from disk.""" | ||||
ms = mergestate(repo) | ||||
Siddharth Agarwal
|
r27005 | ms._read() | ||
Siddharth Agarwal
|
r26991 | return ms | ||
Matt Mackall
|
r6512 | def __init__(self, repo): | ||
Siddharth Agarwal
|
r27005 | """Initialize the merge state. | ||
Do not use this directly! Instead call read() or clean().""" | ||||
Matt Mackall
|
r6512 | self._repo = repo | ||
Peter Arrenbrecht
|
r12369 | self._dirty = False | ||
Simon Farnsworth
|
r28634 | self._labels = None | ||
Pierre-Yves David
|
r20651 | |||
Simon Farnsworth
|
r28634 | def reset(self, node=None, other=None, labels=None): | ||
Matt Mackall
|
r6512 | self._state = {} | ||
Durham Goode
|
r28009 | self._stateextras = {} | ||
Gregory Szorc
|
r21261 | self._local = None | ||
self._other = None | ||||
Simon Farnsworth
|
r28634 | self._labels = labels | ||
Siddharth Agarwal
|
r27130 | for var in ('localctx', 'otherctx'): | ||
if var in vars(self): | ||||
delattr(self, var) | ||||
Matt Mackall
|
r7848 | if node: | ||
self._local = node | ||||
Pierre-Yves David
|
r20591 | self._other = other | ||
Siddharth Agarwal
|
r26768 | self._readmergedriver = None | ||
Siddharth Agarwal
|
r26769 | if self.mergedriver: | ||
self._mdstate = 's' | ||||
else: | ||||
self._mdstate = 'u' | ||||
Martin von Zweigbergk
|
r23380 | shutil.rmtree(self._repo.join('merge'), True) | ||
Siddharth Agarwal
|
r27074 | self._results = {} | ||
Peter Arrenbrecht
|
r12369 | self._dirty = False | ||
Pierre-Yves David
|
r20651 | |||
Matt Mackall
|
r6518 | def _read(self): | ||
Pierre-Yves David
|
r20652 | """Analyse each record content to restore a serialized state from disk | ||
This function process "record" entry produced by the de-serialization | ||||
of on disk file. | ||||
""" | ||||
Matt Mackall
|
r6518 | self._state = {} | ||
Durham Goode
|
r28009 | self._stateextras = {} | ||
Gregory Szorc
|
r21261 | self._local = None | ||
self._other = None | ||||
Siddharth Agarwal
|
r27130 | for var in ('localctx', 'otherctx'): | ||
if var in vars(self): | ||||
delattr(self, var) | ||||
Siddharth Agarwal
|
r26768 | self._readmergedriver = None | ||
Siddharth Agarwal
|
r26769 | self._mdstate = 's' | ||
Siddharth Agarwal
|
r26986 | unsupported = set() | ||
Pierre-Yves David
|
r20589 | records = self._readrecords() | ||
for rtype, record in records: | ||||
if rtype == 'L': | ||||
self._local = bin(record) | ||||
Pierre-Yves David
|
r20591 | elif rtype == 'O': | ||
self._other = bin(record) | ||||
Siddharth Agarwal
|
r26649 | elif rtype == 'm': | ||
bits = record.split('\0', 1) | ||||
mdstate = bits[1] | ||||
if len(mdstate) != 1 or mdstate not in 'ums': | ||||
# the merge driver should be idempotent, so just rerun it | ||||
mdstate = 'u' | ||||
Siddharth Agarwal
|
r26768 | self._readmergedriver = bits[0] | ||
Siddharth Agarwal
|
r26649 | self._mdstate = mdstate | ||
Siddharth Agarwal
|
r27031 | elif rtype in 'FDC': | ||
Martin von Zweigbergk
|
r23380 | bits = record.split('\0') | ||
Pierre-Yves David
|
r20589 | self._state[bits[0]] = bits[1:] | ||
Durham Goode
|
r28009 | elif rtype == 'f': | ||
filename, rawextras = record.split('\0', 1) | ||||
extraparts = rawextras.split('\0') | ||||
extras = {} | ||||
i = 0 | ||||
while i < len(extraparts): | ||||
extras[extraparts[i]] = extraparts[i + 1] | ||||
i += 2 | ||||
self._stateextras[filename] = extras | ||||
Simon Farnsworth
|
r28634 | elif rtype == 'l': | ||
labels = record.split('\0', 2) | ||||
self._labels = [l for l in labels if len(l) > 0] | ||||
Pierre-Yves David
|
r20589 | elif not rtype.islower(): | ||
Siddharth Agarwal
|
r26986 | unsupported.add(rtype) | ||
Siddharth Agarwal
|
r27074 | self._results = {} | ||
Pierre-Yves David
|
r20589 | self._dirty = False | ||
Pierre-Yves David
|
r20651 | |||
Siddharth Agarwal
|
r26986 | if unsupported: | ||
raise error.UnsupportedMergeRecords(unsupported) | ||||
Pierre-Yves David
|
r20589 | def _readrecords(self): | ||
Pierre-Yves David
|
r20652 | """Read merge state from disk and return a list of record (TYPE, data) | ||
Mads Kiilerich
|
r21024 | We read data from both v1 and v2 files and decide which one to use. | ||
Pierre-Yves David
|
r20652 | |||
Mads Kiilerich
|
r21024 | V1 has been used by version prior to 2.9.1 and contains less data than | ||
v2. We read both versions and check if no data in v2 contradicts | ||||
Pierre-Yves David
|
r20652 | v1. If there is not contradiction we can safely assume that both v1 | ||
and v2 were written at the same time and use the extract data in v2. If | ||||
there is contradiction we ignore v2 content as we assume an old version | ||||
Mads Kiilerich
|
r21024 | of Mercurial has overwritten the mergestate file and left an old v2 | ||
Pierre-Yves David
|
r20652 | file around. | ||
returns list of record [(TYPE, data), ...]""" | ||||
Pierre-Yves David
|
r20590 | v1records = self._readrecordsv1() | ||
v2records = self._readrecordsv2() | ||||
Siddharth Agarwal
|
r26500 | if self._v1v2match(v1records, v2records): | ||
return v2records | ||||
else: | ||||
# v1 file is newer than v2 file, use it | ||||
# we have to infer the "other" changeset of the merge | ||||
# we cannot do better than that with v1 of the format | ||||
mctx = self._repo[None].parents()[-1] | ||||
v1records.append(('O', mctx.hex())) | ||||
# add place holder "other" file node information | ||||
# nobody is using it yet so we do no need to fetch the data | ||||
# if mctx was wrong `mctx[bits[-2]]` may fails. | ||||
for idx, r in enumerate(v1records): | ||||
if r[0] == 'F': | ||||
bits = r[1].split('\0') | ||||
bits.insert(-2, '') | ||||
v1records[idx] = (r[0], '\0'.join(bits)) | ||||
return v1records | ||||
def _v1v2match(self, v1records, v2records): | ||||
Pierre-Yves David
|
r20593 | oldv2 = set() # old format version of v2 record | ||
for rec in v2records: | ||||
if rec[0] == 'L': | ||||
oldv2.add(rec) | ||||
elif rec[0] == 'F': | ||||
# drop the onode data (not contained in v1) | ||||
oldv2.add(('F', _droponode(rec[1]))) | ||||
for rec in v1records: | ||||
if rec not in oldv2: | ||||
Siddharth Agarwal
|
r26500 | return False | ||
Pierre-Yves David
|
r20590 | else: | ||
Siddharth Agarwal
|
r26500 | return True | ||
Pierre-Yves David
|
r20651 | |||
Pierre-Yves David
|
r20590 | def _readrecordsv1(self): | ||
Pierre-Yves David
|
r20652 | """read on disk merge state for version 1 file | ||
returns list of record [(TYPE, data), ...] | ||||
Note: the "F" data from this file are one entry short | ||||
(no "other file node" entry) | ||||
""" | ||||
Pierre-Yves David
|
r20589 | records = [] | ||
Matt Mackall
|
r6518 | try: | ||
Angel Ezquerra
|
r23877 | f = self._repo.vfs(self.statepathv1) | ||
Patrick Mezard
|
r6530 | for i, l in enumerate(f): | ||
if i == 0: | ||||
Pierre-Yves David
|
r20589 | records.append(('L', l[:-1])) | ||
Patrick Mezard
|
r6530 | else: | ||
Pierre-Yves David
|
r20589 | records.append(('F', l[:-1])) | ||
Dan Villiom Podlaski Christiansen
|
r13400 | f.close() | ||
Gregory Szorc
|
r25660 | except IOError as err: | ||
Matt Mackall
|
r6518 | if err.errno != errno.ENOENT: | ||
raise | ||||
Pierre-Yves David
|
r20589 | return records | ||
Pierre-Yves David
|
r20651 | |||
Pierre-Yves David
|
r20590 | def _readrecordsv2(self): | ||
Pierre-Yves David
|
r20652 | """read on disk merge state for version 2 file | ||
Siddharth Agarwal
|
r27022 | This format is a list of arbitrary records of the form: | ||
[type][length][content] | ||||
`type` is a single character, `length` is a 4 byte integer, and | ||||
`content` is an arbitrary byte sequence of length `length`. | ||||
Siddharth Agarwal
|
r27027 | Mercurial versions prior to 3.7 have a bug where if there are | ||
unsupported mandatory merge records, attempting to clear out the merge | ||||
state with hg update --clean or similar aborts. The 't' record type | ||||
works around that by writing out what those versions treat as an | ||||
advisory record, but later versions interpret as special: the first | ||||
character is the 'real' record type and everything onwards is the data. | ||||
Siddharth Agarwal
|
r27022 | Returns list of records [(TYPE, data), ...].""" | ||
Pierre-Yves David
|
r20590 | records = [] | ||
try: | ||||
Angel Ezquerra
|
r23877 | f = self._repo.vfs(self.statepathv2) | ||
Pierre-Yves David
|
r20590 | data = f.read() | ||
off = 0 | ||||
end = len(data) | ||||
while off < end: | ||||
rtype = data[off] | ||||
off += 1 | ||||
Olle Lundberg
|
r20607 | length = _unpack('>I', data[off:(off + 4)])[0] | ||
Pierre-Yves David
|
r20590 | off += 4 | ||
Olle Lundberg
|
r20607 | record = data[off:(off + length)] | ||
off += length | ||||
Siddharth Agarwal
|
r27027 | if rtype == 't': | ||
rtype, record = record[0], record[1:] | ||||
Pierre-Yves David
|
r20590 | records.append((rtype, record)) | ||
f.close() | ||||
Gregory Szorc
|
r25660 | except IOError as err: | ||
Pierre-Yves David
|
r20590 | if err.errno != errno.ENOENT: | ||
raise | ||||
return records | ||||
Pierre-Yves David
|
r20651 | |||
Siddharth Agarwal
|
r26649 | @util.propertycache | ||
def mergedriver(self): | ||||
Siddharth Agarwal
|
r26768 | # protect against the following: | ||
# - A configures a malicious merge driver in their hgrc, then | ||||
# pauses the merge | ||||
# - A edits their hgrc to remove references to the merge driver | ||||
# - A gives a copy of their entire repo, including .hg, to B | ||||
# - B inspects .hgrc and finds it to be clean | ||||
# - B then continues the merge and the malicious merge driver | ||||
# gets invoked | ||||
configmergedriver = self._repo.ui.config('experimental', 'mergedriver') | ||||
if (self._readmergedriver is not None | ||||
and self._readmergedriver != configmergedriver): | ||||
raise error.ConfigError( | ||||
_("merge driver changed since merge started"), | ||||
hint=_("revert merge driver change or abort merge")) | ||||
return configmergedriver | ||||
Siddharth Agarwal
|
r26765 | @util.propertycache | ||
Siddharth Agarwal
|
r27130 | def localctx(self): | ||
if self._local is None: | ||||
raise RuntimeError("localctx accessed but self._local isn't set") | ||||
return self._repo[self._local] | ||||
@util.propertycache | ||||
Siddharth Agarwal
|
r26765 | def otherctx(self): | ||
Siddharth Agarwal
|
r27129 | if self._other is None: | ||
Simon Farnsworth
|
r28267 | raise RuntimeError("otherctx accessed but self._other isn't set") | ||
Siddharth Agarwal
|
r26765 | return self._repo[self._other] | ||
Siddharth Agarwal
|
r26649 | |||
Gregory Szorc
|
r21264 | def active(self): | ||
"""Whether mergestate is active. | ||||
Returns True if there appears to be mergestate. This is a rough proxy | ||||
for "is a merge in progress." | ||||
""" | ||||
# Check local variables before looking at filesystem for performance | ||||
# reasons. | ||||
return bool(self._local) or bool(self._state) or \ | ||||
Angel Ezquerra
|
r23877 | self._repo.vfs.exists(self.statepathv1) or \ | ||
self._repo.vfs.exists(self.statepathv2) | ||||
Gregory Szorc
|
r21264 | |||
Peter Arrenbrecht
|
r12369 | def commit(self): | ||
Pierre-Yves David
|
r20652 | """Write current state on disk (if necessary)""" | ||
Peter Arrenbrecht
|
r12369 | if self._dirty: | ||
Siddharth Agarwal
|
r27006 | records = self._makerecords() | ||
Pierre-Yves David
|
r20589 | self._writerecords(records) | ||
Peter Arrenbrecht
|
r12369 | self._dirty = False | ||
Pierre-Yves David
|
r20651 | |||
Siddharth Agarwal
|
r27006 | def _makerecords(self): | ||
records = [] | ||||
records.append(('L', hex(self._local))) | ||||
records.append(('O', hex(self._other))) | ||||
if self.mergedriver: | ||||
records.append(('m', '\0'.join([ | ||||
self.mergedriver, self._mdstate]))) | ||||
for d, v in self._state.iteritems(): | ||||
if v[0] == 'd': | ||||
records.append(('D', '\0'.join([d] + v))) | ||||
Siddharth Agarwal
|
r27031 | # v[1] == local ('cd'), v[6] == other ('dc') -- not supported by | ||
# older versions of Mercurial | ||||
elif v[1] == nullhex or v[6] == nullhex: | ||||
records.append(('C', '\0'.join([d] + v))) | ||||
Siddharth Agarwal
|
r27006 | else: | ||
records.append(('F', '\0'.join([d] + v))) | ||||
Durham Goode
|
r28009 | for filename, extras in sorted(self._stateextras.iteritems()): | ||
rawextras = '\0'.join('%s\0%s' % (k, v) for k, v in | ||||
extras.iteritems()) | ||||
records.append(('f', '%s\0%s' % (filename, rawextras))) | ||||
Simon Farnsworth
|
r28634 | if self._labels is not None: | ||
labels = '\0'.join(self._labels) | ||||
records.append(('l', labels)) | ||||
Siddharth Agarwal
|
r27006 | return records | ||
Pierre-Yves David
|
r20589 | def _writerecords(self, records): | ||
Pierre-Yves David
|
r20652 | """Write current state on disk (both v1 and v2)""" | ||
Pierre-Yves David
|
r20590 | self._writerecordsv1(records) | ||
self._writerecordsv2(records) | ||||
Pierre-Yves David
|
r20651 | |||
Pierre-Yves David
|
r20590 | def _writerecordsv1(self, records): | ||
Pierre-Yves David
|
r20652 | """Write current state on disk in a version 1 file""" | ||
Angel Ezquerra
|
r23877 | f = self._repo.vfs(self.statepathv1, 'w') | ||
Pierre-Yves David
|
r20589 | irecords = iter(records) | ||
lrecords = irecords.next() | ||||
assert lrecords[0] == 'L' | ||||
Martin von Zweigbergk
|
r23380 | f.write(hex(self._local) + '\n') | ||
Pierre-Yves David
|
r20589 | for rtype, data in irecords: | ||
Martin von Zweigbergk
|
r23380 | if rtype == 'F': | ||
f.write('%s\n' % _droponode(data)) | ||||
Pierre-Yves David
|
r20589 | f.close() | ||
Pierre-Yves David
|
r20651 | |||
Pierre-Yves David
|
r20590 | def _writerecordsv2(self, records): | ||
Siddharth Agarwal
|
r27027 | """Write current state on disk in a version 2 file | ||
See the docstring for _readrecordsv2 for why we use 't'.""" | ||||
# these are the records that all version 2 clients can read | ||||
whitelist = 'LOF' | ||||
Angel Ezquerra
|
r23877 | f = self._repo.vfs(self.statepathv2, 'w') | ||
Pierre-Yves David
|
r20590 | for key, data in records: | ||
assert len(key) == 1 | ||||
Siddharth Agarwal
|
r27027 | if key not in whitelist: | ||
key, data = 't', '%s%s' % (key, data) | ||||
Martin von Zweigbergk
|
r23380 | format = '>sI%is' % len(data) | ||
Pierre-Yves David
|
r20590 | f.write(_pack(format, key, len(data), data)) | ||
f.close() | ||||
Pierre-Yves David
|
r20651 | |||
Mads Kiilerich
|
r18338 | def add(self, fcl, fco, fca, fd): | ||
Pierre-Yves David
|
r20652 | """add a new (potentially?) conflicting file the merge state | ||
fcl: file context for local, | ||||
fco: file context for remote, | ||||
fca: file context for ancestors, | ||||
fd: file path of the resulting merge. | ||||
note: also write the local version to the `.hg/merge` directory. | ||||
""" | ||||
Siddharth Agarwal
|
r27049 | if fcl.isabsent(): | ||
hash = nullhex | ||||
else: | ||||
hash = util.sha1(fcl.path()).hexdigest() | ||||
self._repo.vfs.write('merge/' + hash, fcl.data()) | ||||
Pierre-Yves David
|
r20593 | self._state[fd] = ['u', hash, fcl.path(), | ||
fca.path(), hex(fca.filenode()), | ||||
fco.path(), hex(fco.filenode()), | ||||
fcl.flags()] | ||||
Durham Goode
|
r28011 | self._stateextras[fd] = { 'ancestorlinknode' : hex(fca.node()) } | ||
Peter Arrenbrecht
|
r12369 | self._dirty = True | ||
Pierre-Yves David
|
r20651 | |||
Matt Mackall
|
r6512 | def __contains__(self, dfile): | ||
return dfile in self._state | ||||
Pierre-Yves David
|
r20651 | |||
Matt Mackall
|
r6512 | def __getitem__(self, dfile): | ||
Matt Mackall
|
r6518 | return self._state[dfile][0] | ||
Pierre-Yves David
|
r20651 | |||
Matt Mackall
|
r6518 | def __iter__(self): | ||
Mads Kiilerich
|
r21268 | return iter(sorted(self._state)) | ||
Pierre-Yves David
|
r20651 | |||
Bryan O'Sullivan
|
r19285 | def files(self): | ||
return self._state.keys() | ||||
Pierre-Yves David
|
r20651 | |||
Matt Mackall
|
r6512 | def mark(self, dfile, state): | ||
Matt Mackall
|
r6518 | self._state[dfile][0] = state | ||
Peter Arrenbrecht
|
r12369 | self._dirty = True | ||
Pierre-Yves David
|
r20651 | |||
Siddharth Agarwal
|
r26766 | def mdstate(self): | ||
return self._mdstate | ||||
Gregory Szorc
|
r21266 | def unresolved(self): | ||
"""Obtain the paths of unresolved files.""" | ||||
for f, entry in self._state.items(): | ||||
if entry[0] == 'u': | ||||
yield f | ||||
Siddharth Agarwal
|
r26740 | def driverresolved(self): | ||
"""Obtain the paths of driver-resolved files.""" | ||||
for f, entry in self._state.items(): | ||||
if entry[0] == 'd': | ||||
yield f | ||||
Durham Goode
|
r28009 | def extras(self, filename): | ||
return self._stateextras.setdefault(filename, {}) | ||||
Simon Farnsworth
|
r28634 | def _resolve(self, preresolve, dfile, wctx): | ||
Pierre-Yves David
|
r20652 | """rerun merge process for file path `dfile`""" | ||
Siddharth Agarwal
|
r26651 | if self[dfile] in 'rd': | ||
Siddharth Agarwal
|
r26616 | return True, 0 | ||
Pierre-Yves David
|
r20593 | stateentry = self._state[dfile] | ||
state, hash, lfile, afile, anode, ofile, onode, flags = stateentry | ||||
Pierre-Yves David
|
r20594 | octx = self._repo[self._other] | ||
Durham Goode
|
r28011 | extras = self.extras(dfile) | ||
anccommitnode = extras.get('ancestorlinknode') | ||||
if anccommitnode: | ||||
actx = self._repo[anccommitnode] | ||||
else: | ||||
actx = None | ||||
Siddharth Agarwal
|
r27048 | fcd = self._filectxorabsent(hash, wctx, dfile) | ||
fco = self._filectxorabsent(onode, octx, ofile) | ||||
# TODO: move this to filectxorabsent | ||||
Durham Goode
|
r28011 | fca = self._repo.filectx(afile, fileid=anode, changeid=actx) | ||
Mads Kiilerich
|
r18338 | # "premerge" x flags | ||
flo = fco.flags() | ||||
fla = fca.flags() | ||||
if 'x' in flags + flo + fla and 'l' not in flags + flo + fla: | ||||
if fca.node() == nullid: | ||||
Siddharth Agarwal
|
r26617 | if preresolve: | ||
self._repo.ui.warn( | ||||
_('warning: cannot merge flags for %s\n') % afile) | ||||
Mads Kiilerich
|
r18338 | elif flags == fla: | ||
flags = flo | ||||
Siddharth Agarwal
|
r26617 | if preresolve: | ||
# restore local | ||||
Siddharth Agarwal
|
r27048 | if hash != nullhex: | ||
f = self._repo.vfs('merge/' + hash) | ||||
self._repo.wwrite(dfile, f.read(), flags) | ||||
f.close() | ||||
else: | ||||
self._repo.wvfs.unlinkpath(dfile, ignoremissing=True) | ||||
Siddharth Agarwal
|
r27034 | complete, r, deleted = filemerge.premerge(self._repo, self._local, | ||
lfile, fcd, fco, fca, | ||||
Simon Farnsworth
|
r28634 | labels=self._labels) | ||
Siddharth Agarwal
|
r26617 | else: | ||
Siddharth Agarwal
|
r27034 | complete, r, deleted = filemerge.filemerge(self._repo, self._local, | ||
lfile, fcd, fco, fca, | ||||
Simon Farnsworth
|
r28634 | labels=self._labels) | ||
Matt Mackall
|
r13536 | if r is None: | ||
# no real conflict | ||||
del self._state[dfile] | ||||
Durham Goode
|
r28009 | self._stateextras.pop(dfile, None) | ||
Mads Kiilerich
|
r20792 | self._dirty = True | ||
Matt Mackall
|
r13536 | elif not r: | ||
Matt Mackall
|
r6512 | self.mark(dfile, 'r') | ||
Siddharth Agarwal
|
r27035 | |||
if complete: | ||||
Siddharth Agarwal
|
r27075 | action = None | ||
Siddharth Agarwal
|
r27035 | if deleted: | ||
Siddharth Agarwal
|
r27122 | if fcd.isabsent(): | ||
# dc: local picked. Need to drop if present, which may | ||||
# happen on re-resolves. | ||||
action = 'f' | ||||
else: | ||||
Siddharth Agarwal
|
r27035 | # cd: remote picked (or otherwise deleted) | ||
action = 'r' | ||||
else: | ||||
if fcd.isabsent(): # dc: remote picked | ||||
action = 'g' | ||||
elif fco.isabsent(): # cd: local picked | ||||
Siddharth Agarwal
|
r27131 | if dfile in self.localctx: | ||
action = 'am' | ||||
else: | ||||
action = 'a' | ||||
Siddharth Agarwal
|
r27035 | # else: regular merges (no action necessary) | ||
Siddharth Agarwal
|
r27074 | self._results[dfile] = r, action | ||
Siddharth Agarwal
|
r27035 | |||
Siddharth Agarwal
|
r26616 | return complete, r | ||
Matt Mackall
|
r2775 | |||
Siddharth Agarwal
|
r27048 | def _filectxorabsent(self, hexnode, ctx, f): | ||
if hexnode == nullhex: | ||||
return filemerge.absentfilectx(ctx, f) | ||||
else: | ||||
return ctx[f] | ||||
Simon Farnsworth
|
r28634 | def preresolve(self, dfile, wctx): | ||
Siddharth Agarwal
|
r26870 | """run premerge process for dfile | ||
Returns whether the merge is complete, and the exit code.""" | ||||
Simon Farnsworth
|
r28634 | return self._resolve(True, dfile, wctx) | ||
Siddharth Agarwal
|
r26617 | |||
Simon Farnsworth
|
r28634 | def resolve(self, dfile, wctx): | ||
Siddharth Agarwal
|
r26870 | """run merge process (assuming premerge was run) for dfile | ||
Returns the exit code of the merge.""" | ||||
Simon Farnsworth
|
r28634 | return self._resolve(False, dfile, wctx)[1] | ||
Siddharth Agarwal
|
r26615 | |||
Siddharth Agarwal
|
r27076 | def counts(self): | ||
"""return counts for updated, merged and removed files in this | ||||
session""" | ||||
updated, merged, removed = 0, 0, 0 | ||||
for r, action in self._results.itervalues(): | ||||
if r is None: | ||||
updated += 1 | ||||
elif r == 0: | ||||
if action == 'r': | ||||
removed += 1 | ||||
else: | ||||
merged += 1 | ||||
return updated, merged, removed | ||||
Siddharth Agarwal
|
r27077 | def unresolvedcount(self): | ||
"""get unresolved count for this merge (persistent)""" | ||||
return len([True for f, entry in self._state.iteritems() | ||||
if entry[0] == 'u']) | ||||
Siddharth Agarwal
|
r27079 | def actions(self): | ||
"""return lists of actions to perform on the dirstate""" | ||||
Siddharth Agarwal
|
r27131 | actions = {'r': [], 'f': [], 'a': [], 'am': [], 'g': []} | ||
Siddharth Agarwal
|
r27079 | for f, (r, action) in self._results.iteritems(): | ||
if action is not None: | ||||
actions[action].append((f, None, "merge result")) | ||||
return actions | ||||
Siddharth Agarwal
|
r27088 | def recordactions(self): | ||
"""record remove/add/get actions in the dirstate""" | ||||
branchmerge = self._repo.dirstate.p2() != nullid | ||||
recordupdates(self._repo, self.actions(), branchmerge) | ||||
Siddharth Agarwal
|
r27090 | def queueremove(self, f): | ||
"""queues a file to be removed from the dirstate | ||||
Meant for use by custom merge drivers.""" | ||||
self._results[f] = 0, 'r' | ||||
def queueadd(self, f): | ||||
"""queues a file to be added to the dirstate | ||||
Meant for use by custom merge drivers.""" | ||||
self._results[f] = 0, 'a' | ||||
def queueget(self, f): | ||||
"""queues a file to be marked modified in the dirstate | ||||
Meant for use by custom merge drivers.""" | ||||
self._results[f] = 0, 'g' | ||||
Siddharth Agarwal
|
r27740 | def _getcheckunknownconfig(repo, section, name): | ||
config = repo.ui.config(section, name, default='abort') | ||||
valid = ['abort', 'ignore', 'warn'] | ||||
if config not in valid: | ||||
validstr = ', '.join(["'" + v + "'" for v in valid]) | ||||
raise error.ConfigError(_("%s.%s not valid " | ||||
"('%s' is none of %s)") | ||||
% (section, name, config, validstr)) | ||||
return config | ||||
Martin von Zweigbergk
|
r23653 | def _checkunknownfile(repo, wctx, mctx, f, f2=None): | ||
if f2 is None: | ||||
f2 = f | ||||
Durham Goode
|
r28088 | return (repo.wvfs.audit.check(f) | ||
and repo.wvfs.isfileorlink(f) | ||||
Matt Mackall
|
r16284 | and repo.dirstate.normalize(f) not in repo.dirstate | ||
Martin von Zweigbergk
|
r23653 | and mctx[f2].cmp(wctx[f])) | ||
Matt Mackall
|
r16093 | |||
Siddharth Agarwal
|
r28020 | def _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce): | ||
Martin von Zweigbergk
|
r23655 | """ | ||
Considers any actions that care about the presence of conflicting unknown | ||||
files. For some actions, the result is to abort; for others, it is to | ||||
choose a different action. | ||||
""" | ||||
Siddharth Agarwal
|
r27654 | conflicts = set() | ||
Siddharth Agarwal
|
r28018 | warnconflicts = set() | ||
abortconflicts = set() | ||||
unknownconfig = _getcheckunknownconfig(repo, 'merge', 'checkunknown') | ||||
ignoredconfig = _getcheckunknownconfig(repo, 'merge', 'checkignored') | ||||
Martin von Zweigbergk
|
r23655 | if not force: | ||
Siddharth Agarwal
|
r27741 | def collectconflicts(conflicts, config): | ||
if config == 'abort': | ||||
abortconflicts.update(conflicts) | ||||
elif config == 'warn': | ||||
warnconflicts.update(conflicts) | ||||
Martin von Zweigbergk
|
r23655 | for f, (m, args, msg) in actions.iteritems(): | ||
if m in ('c', 'dc'): | ||||
if _checkunknownfile(repo, wctx, mctx, f): | ||||
Siddharth Agarwal
|
r27654 | conflicts.add(f) | ||
Martin von Zweigbergk
|
r23655 | elif m == 'dg': | ||
if _checkunknownfile(repo, wctx, mctx, f, args[0]): | ||||
Siddharth Agarwal
|
r27654 | conflicts.add(f) | ||
Martin von Zweigbergk
|
r23655 | |||
Siddharth Agarwal
|
r27742 | ignoredconflicts = set([c for c in conflicts | ||
if repo.dirstate._ignore(c)]) | ||||
unknownconflicts = conflicts - ignoredconflicts | ||||
collectconflicts(ignoredconflicts, ignoredconfig) | ||||
collectconflicts(unknownconflicts, unknownconfig) | ||||
Siddharth Agarwal
|
r28022 | else: | ||
for f, (m, args, msg) in actions.iteritems(): | ||||
if m == 'cm': | ||||
fl2, anc = args | ||||
different = _checkunknownfile(repo, wctx, mctx, f) | ||||
if repo.dirstate._ignore(f): | ||||
config = ignoredconfig | ||||
else: | ||||
config = unknownconfig | ||||
# The behavior when force is True is described by this table: | ||||
# config different mergeforce | action backup | ||||
# * n * | get n | ||||
# * y y | merge - | ||||
# abort y n | merge - (1) | ||||
# warn y n | warn + get y | ||||
# ignore y n | get y | ||||
# | ||||
# (1) this is probably the wrong behavior here -- we should | ||||
# probably abort, but some actions like rebases currently | ||||
# don't like an abort happening in the middle of | ||||
# merge.update. | ||||
if not different: | ||||
actions[f] = ('g', (fl2, False), "remote created") | ||||
elif mergeforce or config == 'abort': | ||||
actions[f] = ('m', (f, f, None, False, anc), | ||||
"remote differs from untracked local") | ||||
elif config == 'abort': | ||||
abortconflicts.add(f) | ||||
else: | ||||
if config == 'warn': | ||||
warnconflicts.add(f) | ||||
actions[f] = ('g', (fl2, True), "remote created") | ||||
Siddharth Agarwal
|
r27741 | |||
Siddharth Agarwal
|
r28018 | for f in sorted(abortconflicts): | ||
repo.ui.warn(_("%s: untracked file differs\n") % f) | ||||
if abortconflicts: | ||||
raise error.Abort(_("untracked files in working directory " | ||||
"differ from files in requested revision")) | ||||
for f in sorted(warnconflicts): | ||||
repo.ui.warn(_("%s: replacing untracked file\n") % f) | ||||
Martin von Zweigbergk
|
r23655 | |||
for f, (m, args, msg) in actions.iteritems(): | ||||
Siddharth Agarwal
|
r27657 | backup = f in conflicts | ||
Martin von Zweigbergk
|
r23655 | if m == 'c': | ||
Siddharth Agarwal
|
r27655 | flags, = args | ||
Siddharth Agarwal
|
r27657 | actions[f] = ('g', (flags, backup), msg) | ||
Martin von Zweigbergk
|
r23655 | |||
Matt Mackall
|
r6269 | def _forgetremoved(wctx, mctx, branchmerge): | ||
Matt Mackall
|
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. | ||||
Alexis S. L. Carvalho
|
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. | ||||
Matt Mackall
|
r3107 | """ | ||
Martin von Zweigbergk
|
r23640 | actions = {} | ||
m = 'f' | ||||
Mads Kiilerich
|
r21545 | if branchmerge: | ||
Martin von Zweigbergk
|
r23640 | m = 'r' | ||
Alexis S. L. Carvalho
|
r6242 | for f in wctx.deleted(): | ||
Matt Mackall
|
r6272 | if f not in mctx: | ||
Martin von Zweigbergk
|
r23640 | actions[f] = m, None, "forget deleted" | ||
Alexis S. L. Carvalho
|
r6242 | |||
if not branchmerge: | ||||
for f in wctx.removed(): | ||||
Matt Mackall
|
r6272 | if f not in mctx: | ||
Martin von Zweigbergk
|
r23640 | actions[f] = 'f', None, "forget removed" | ||
Matt Mackall
|
r3107 | |||
Martin von Zweigbergk
|
r23640 | return actions | ||
Matt Mackall
|
r3107 | |||
Mads Kiilerich
|
r20640 | def _checkcollision(repo, wmf, actions): | ||
FUJIWARA Katsunori
|
r19105 | # build provisional merged manifest up | ||
pmmf = set(wmf) | ||||
Mads Kiilerich
|
r21545 | if actions: | ||
# k, dr, e and rd are no-op | ||||
Siddharth Agarwal
|
r27131 | for m in 'a', 'am', 'f', 'g', 'cd', 'dc': | ||
Mads Kiilerich
|
r21545 | for f, args, msg in actions[m]: | ||
pmmf.add(f) | ||||
for f, args, msg in actions['r']: | ||||
pmmf.discard(f) | ||||
for f, args, msg in actions['dm']: | ||||
f2, flags = args | ||||
pmmf.discard(f2) | ||||
pmmf.add(f) | ||||
for f, args, msg in actions['dg']: | ||||
pmmf.add(f) | ||||
for f, args, msg in actions['m']: | ||||
f1, f2, fa, move, anc = args | ||||
if move: | ||||
pmmf.discard(f1) | ||||
pmmf.add(f) | ||||
FUJIWARA Katsunori
|
r19105 | |||
# check case-folding collision in provisional merged manifest | ||||
foldmap = {} | ||||
for f in sorted(pmmf): | ||||
fold = util.normcase(f) | ||||
if fold in foldmap: | ||||
Pierre-Yves David
|
r26587 | raise error.Abort(_("case-folding collision between %s and %s") | ||
FUJIWARA Katsunori
|
r19105 | % (f, foldmap[fold])) | ||
foldmap[fold] = f | ||||
Mads Kiilerich
|
r26661 | # check case-folding of directories | ||
foldprefix = unfoldprefix = lastfull = '' | ||||
for fold, f in sorted(foldmap.items()): | ||||
if fold.startswith(foldprefix) and not f.startswith(unfoldprefix): | ||||
# the folded prefix matches but actual casing is different | ||||
raise error.Abort(_("case-folding collision between " | ||||
"%s and directory of %s") % (lastfull, f)) | ||||
foldprefix = fold + '/' | ||||
unfoldprefix = f + '/' | ||||
lastfull = f | ||||
Siddharth Agarwal
|
r26785 | def driverpreprocess(repo, ms, wctx, labels=None): | ||
"""run the preprocess step of the merge driver, if any | ||||
This is currently not implemented -- it's an extension point.""" | ||||
return True | ||||
def driverconclude(repo, ms, wctx, labels=None): | ||||
"""run the conclude step of the merge driver, if any | ||||
This is currently not implemented -- it's an extension point.""" | ||||
return True | ||||
Augie Fackler
|
r27346 | def manifestmerge(repo, wctx, p2, pa, branchmerge, force, matcher, | ||
Mads Kiilerich
|
r21080 | acceptremote, followcopies): | ||
Matt Mackall
|
r3105 | """ | ||
Alecs King
|
r11817 | Merge p1 and p2 with ancestor pa and generate merge action list | ||
Matt Mackall
|
r3315 | |||
Siddharth Agarwal
|
r18605 | branchmerge and force are as passed in to update | ||
Augie Fackler
|
r27346 | matcher = matcher to filter file lists | ||
Durham Goode
|
r18778 | acceptremote = accept the incoming changes without prompting | ||
Matt Mackall
|
r3105 | """ | ||
Augie Fackler
|
r27346 | if matcher is not None and matcher.always(): | ||
matcher = None | ||||
Matt Mackall
|
r3105 | |||
Martin von Zweigbergk
|
r23526 | copy, movewithdir, diverge, renamedelete = {}, {}, {}, {} | ||
Matt Mackall
|
r8753 | |||
Siddharth Agarwal
|
r18651 | # manifests fetched in order are going to be faster, so prime the caches | ||
[x.manifest() for x in | ||||
sorted(wctx.parents() + [p2, pa], key=lambda x: x.rev())] | ||||
if followcopies: | ||||
Bryan O'Sullivan
|
r18611 | ret = copies.mergecopies(repo, wctx, p2, pa) | ||
Siddharth Agarwal
|
r18134 | copy, movewithdir, diverge, renamedelete = ret | ||
Matt Mackall
|
r8753 | |||
repo.ui.note(_("resolving manifests\n")) | ||||
Siddharth Agarwal
|
r18605 | repo.ui.debug(" branchmerge: %s, force: %s, partial: %s\n" | ||
Augie Fackler
|
r27346 | % (bool(branchmerge), bool(force), bool(matcher))) | ||
Bryan O'Sullivan
|
r18611 | repo.ui.debug(" ancestor: %s, local: %s, remote: %s\n" % (pa, wctx, p2)) | ||
Matt Mackall
|
r8753 | |||
Bryan O'Sullivan
|
r18611 | m1, m2, ma = wctx.manifest(), p2.manifest(), pa.manifest() | ||
Matt Mackall
|
r8753 | copied = set(copy.values()) | ||
Siddharth Agarwal
|
r18134 | copied.update(movewithdir.values()) | ||
Matt Mackall
|
r3295 | |||
Matt Mackall
|
r11470 | if '.hgsubstate' in m1: | ||
Matt Mackall
|
r9783 | # check whether sub state is modified | ||
Martin von Zweigbergk
|
r28226 | if any(wctx.sub(s).dirty() for s in wctx.substate): | ||
m1['.hgsubstate'] += '+' | ||||
Matt Mackall
|
r9783 | |||
Matt Mackall
|
r3105 | # Compare manifests | ||
Augie Fackler
|
r27429 | if matcher is not None: | ||
m1 = m1.matches(matcher) | ||||
m2 = m2.matches(matcher) | ||||
Martin von Zweigbergk
|
r22964 | diff = m1.diff(m2) | ||
Siddharth Agarwal
|
r18822 | |||
Martin von Zweigbergk
|
r23637 | actions = {} | ||
Martin von Zweigbergk
|
r22966 | for f, ((n1, fl1), (n2, fl2)) in diff.iteritems(): | ||
Martin von Zweigbergk
|
r23473 | if n1 and n2: # file exists on both local and remote side | ||
Martin von Zweigbergk
|
r23396 | if f not in ma: | ||
Martin von Zweigbergk
|
r23397 | fa = copy.get(f, None) | ||
if fa is not None: | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f, f, fa, False, pa.node()), | ||
"both renamed from " + fa) | ||||
Mads Kiilerich
|
r18338 | else: | ||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f, f, None, False, pa.node()), | ||
"both created") | ||||
Matt Mackall
|
r16094 | else: | ||
Martin von Zweigbergk
|
r23396 | a = ma[f] | ||
fla = ma.flags(f) | ||||
Martin von Zweigbergk
|
r23395 | nol = 'l' not in fl1 + fl2 + fla | ||
if n2 == a and fl2 == fla: | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('k' , (), "remote unchanged") | ||
Martin von Zweigbergk
|
r23395 | elif n1 == a and fl1 == fla: # local unchanged - use remote | ||
if n1 == n2: # optimization: keep local content | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('e', (fl2,), "update permissions") | ||
Martin von Zweigbergk
|
r23395 | else: | ||
Siddharth Agarwal
|
r27655 | actions[f] = ('g', (fl2, False), "remote is newer") | ||
Martin von Zweigbergk
|
r23395 | elif nol and n2 == a: # remote only changed 'x' | ||
Martin von Zweigbergk
|
r23637 | actions[f] = ('e', (fl2,), "update permissions") | ||
Martin von Zweigbergk
|
r23395 | elif nol and n1 == a: # local only changed 'x' | ||
Siddharth Agarwal
|
r27655 | actions[f] = ('g', (fl1, False), "remote is newer") | ||
Martin von Zweigbergk
|
r23395 | else: # both changed something | ||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f, f, f, False, pa.node()), | ||
"versions differ") | ||||
Martin von Zweigbergk
|
r23473 | elif n1: # file exists only on local side | ||
Martin von Zweigbergk
|
r23474 | if f in copied: | ||
pass # we'll deal with it on m2 side | ||||
elif f in movewithdir: # directory rename, move local | ||||
Martin von Zweigbergk
|
r23473 | f2 = movewithdir[f] | ||
Martin von Zweigbergk
|
r23475 | if f2 in m2: | ||
Martin von Zweigbergk
|
r23637 | actions[f2] = ('m', (f, f2, None, True, pa.node()), | ||
"remote directory rename, both created") | ||||
Martin von Zweigbergk
|
r23475 | else: | ||
Martin von Zweigbergk
|
r23637 | actions[f2] = ('dm', (f, fl1), | ||
"remote directory rename - move from " + f) | ||||
Martin von Zweigbergk
|
r23473 | elif f in copy: | ||
f2 = copy[f] | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f, f2, f2, False, pa.node()), | ||
"local copied/moved from " + f2) | ||||
Martin von Zweigbergk
|
r23473 | elif f in ma: # clean, a different, no remote | ||
if n1 != ma[f]: | ||||
if acceptremote: | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('r', None, "remote delete") | ||
Martin von Zweigbergk
|
r23473 | else: | ||
Siddharth Agarwal
|
r26962 | actions[f] = ('cd', (f, None, f, False, pa.node()), | ||
"prompt changed/deleted") | ||||
Martin von Zweigbergk
|
r23473 | elif n1[20:] == 'a': | ||
# This extra 'a' is added by working copy manifest to mark | ||||
# the file as locally added. We should forget it instead of | ||||
# deleting it. | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('f', None, "remote deleted") | ||
Mads Kiilerich
|
r20639 | else: | ||
Martin von Zweigbergk
|
r23637 | actions[f] = ('r', None, "other deleted") | ||
Martin von Zweigbergk
|
r23473 | elif n2: # file exists only on remote side | ||
Martin von Zweigbergk
|
r23474 | if f in copied: | ||
pass # we'll deal with it on m1 side | ||||
elif f in movewithdir: | ||||
Martin von Zweigbergk
|
r23473 | f2 = movewithdir[f] | ||
Martin von Zweigbergk
|
r23476 | if f2 in m1: | ||
Martin von Zweigbergk
|
r23637 | actions[f2] = ('m', (f2, f, None, False, pa.node()), | ||
"local directory rename, both created") | ||||
Martin von Zweigbergk
|
r23476 | else: | ||
Martin von Zweigbergk
|
r23637 | actions[f2] = ('dg', (f, fl2), | ||
"local directory rename - get from " + f) | ||||
Martin von Zweigbergk
|
r23473 | elif f in copy: | ||
f2 = copy[f] | ||||
if f2 in m2: | ||||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f2, f, f2, False, pa.node()), | ||
"remote copied from " + f2) | ||||
Martin von Zweigbergk
|
r23473 | else: | ||
Martin von Zweigbergk
|
r23637 | actions[f] = ('m', (f2, f, f2, True, pa.node()), | ||
"remote moved from " + f2) | ||||
Martin von Zweigbergk
|
r23473 | elif f not in ma: | ||
# local unknown, remote created: the logic is described by the | ||||
# following table: | ||||
# | ||||
# force branchmerge different | action | ||||
Martin von Zweigbergk
|
r23651 | # n * * | create | ||
Martin von Zweigbergk
|
r23650 | # y n * | create | ||
# y y n | create | ||||
Martin von Zweigbergk
|
r23473 | # y y y | merge | ||
# | ||||
# Checking whether the files are different is expensive, so we | ||||
# don't do that when we can avoid it. | ||||
Martin von Zweigbergk
|
r23649 | if not force: | ||
Martin von Zweigbergk
|
r23651 | actions[f] = ('c', (fl2,), "remote created") | ||
Martin von Zweigbergk
|
r23649 | elif not branchmerge: | ||
Martin von Zweigbergk
|
r23650 | actions[f] = ('c', (fl2,), "remote created") | ||
Martin von Zweigbergk
|
r23473 | else: | ||
Martin von Zweigbergk
|
r23654 | actions[f] = ('cm', (fl2, pa.node()), | ||
"remote created, get or merge") | ||||
Martin von Zweigbergk
|
r23473 | elif n2 != ma[f]: | ||
Martin von Zweigbergk
|
r23651 | if acceptremote: | ||
actions[f] = ('c', (fl2,), "remote recreating") | ||||
Siddharth Agarwal
|
r18606 | else: | ||
Siddharth Agarwal
|
r26962 | actions[f] = ('dc', (None, f, f, False, pa.node()), | ||
"prompt deleted/changed") | ||||
Martin von Zweigbergk
|
r23651 | |||
Martin von Zweigbergk
|
r23526 | return actions, diverge, renamedelete | ||
Matt Mackall
|
r3105 | |||
Martin von Zweigbergk
|
r23531 | def _resolvetrivial(repo, wctx, mctx, ancestor, actions): | ||
"""Resolves false conflicts where the nodeid changed but the content | ||||
remained the same.""" | ||||
Martin von Zweigbergk
|
r23639 | for f, (m, args, msg) in actions.items(): | ||
if m == 'cd' and f in ancestor and not wctx[f].cmp(ancestor[f]): | ||||
Martin von Zweigbergk
|
r23531 | # local did change but ended up with same content | ||
Martin von Zweigbergk
|
r23639 | actions[f] = 'r', None, "prompt same" | ||
elif m == 'dc' and f in ancestor and not mctx[f].cmp(ancestor[f]): | ||||
Martin von Zweigbergk
|
r23531 | # remote did change but ended up with same content | ||
Martin von Zweigbergk
|
r23639 | del actions[f] # don't get = keep local deleted | ||
Martin von Zweigbergk
|
r23531 | |||
Augie Fackler
|
r27345 | def calculateupdates(repo, wctx, mctx, ancestors, branchmerge, force, | ||
Siddharth Agarwal
|
r28020 | acceptremote, followcopies, matcher=None, | ||
mergeforce=False): | ||||
Martin von Zweigbergk
|
r23385 | "Calculate the actions needed to merge mctx into wctx using ancestors" | ||
if len(ancestors) == 1: # default | ||||
Martin von Zweigbergk
|
r23526 | actions, diverge, renamedelete = manifestmerge( | ||
Augie Fackler
|
r27346 | repo, wctx, mctx, ancestors[0], branchmerge, force, matcher, | ||
Martin von Zweigbergk
|
r23526 | acceptremote, followcopies) | ||
Siddharth Agarwal
|
r28020 | _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce) | ||
Martin von Zweigbergk
|
r23385 | |||
else: # only when merge.preferancestor=* - the default | ||||
repo.ui.note( | ||||
_("note: merging %s and %s using bids from ancestors %s\n") % | ||||
(wctx, mctx, _(' and ').join(str(anc) for anc in ancestors))) | ||||
# Call for bids | ||||
fbids = {} # mapping filename to bids (action method to list af actions) | ||||
Martin von Zweigbergk
|
r23526 | diverge, renamedelete = None, None | ||
Martin von Zweigbergk
|
r23385 | for ancestor in ancestors: | ||
repo.ui.note(_('\ncalculating bids for ancestor %s\n') % ancestor) | ||||
Martin von Zweigbergk
|
r23526 | actions, diverge1, renamedelete1 = manifestmerge( | ||
Augie Fackler
|
r27346 | repo, wctx, mctx, ancestor, branchmerge, force, matcher, | ||
Martin von Zweigbergk
|
r23526 | acceptremote, followcopies) | ||
Siddharth Agarwal
|
r28020 | _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce) | ||
Matt Mackall
|
r26318 | |||
# Track the shortest set of warning on the theory that bid | ||||
# merge will correctly incorporate more information | ||||
if diverge is None or len(diverge1) < len(diverge): | ||||
Martin von Zweigbergk
|
r23526 | diverge = diverge1 | ||
Matt Mackall
|
r26318 | if renamedelete is None or len(renamedelete) < len(renamedelete1): | ||
Martin von Zweigbergk
|
r23526 | renamedelete = renamedelete1 | ||
Matt Mackall
|
r26318 | |||
Martin von Zweigbergk
|
r23638 | for f, a in sorted(actions.iteritems()): | ||
m, args, msg = a | ||||
repo.ui.debug(' %s: %s -> %s\n' % (f, msg, m)) | ||||
if f in fbids: | ||||
d = fbids[f] | ||||
if m in d: | ||||
d[m].append(a) | ||||
Martin von Zweigbergk
|
r23385 | else: | ||
Martin von Zweigbergk
|
r23638 | d[m] = [a] | ||
else: | ||||
fbids[f] = {m: [a]} | ||||
Martin von Zweigbergk
|
r23385 | |||
# Pick the best bid for each file | ||||
repo.ui.note(_('\nauction for merging merge bids\n')) | ||||
Martin von Zweigbergk
|
r23638 | actions = {} | ||
Martin von Zweigbergk
|
r23385 | for f, bids in sorted(fbids.items()): | ||
# bids is a mapping from action method to list af actions | ||||
# Consensus? | ||||
if len(bids) == 1: # all bids are the same kind of method | ||||
m, l = bids.items()[0] | ||||
Augie Fackler
|
r25151 | if all(a == l[0] for a in l[1:]): # len(bids) is > 1 | ||
Martin von Zweigbergk
|
r23385 | repo.ui.note(" %s: consensus for %s\n" % (f, m)) | ||
Martin von Zweigbergk
|
r23638 | actions[f] = l[0] | ||
Martin von Zweigbergk
|
r23385 | continue | ||
# If keep is an option, just do it. | ||||
if 'k' in bids: | ||||
repo.ui.note(" %s: picking 'keep' action\n" % f) | ||||
Martin von Zweigbergk
|
r23638 | actions[f] = bids['k'][0] | ||
Martin von Zweigbergk
|
r23385 | continue | ||
# If there are gets and they all agree [how could they not?], do it. | ||||
if 'g' in bids: | ||||
ga0 = bids['g'][0] | ||||
Augie Fackler
|
r25151 | if all(a == ga0 for a in bids['g'][1:]): | ||
Martin von Zweigbergk
|
r23385 | repo.ui.note(" %s: picking 'get' action\n" % f) | ||
Martin von Zweigbergk
|
r23638 | actions[f] = ga0 | ||
Martin von Zweigbergk
|
r23385 | continue | ||
# TODO: Consider other simple actions such as mode changes | ||||
# Handle inefficient democrazy. | ||||
repo.ui.note(_(' %s: multiple bids for merge action:\n') % f) | ||||
for m, l in sorted(bids.items()): | ||||
for _f, args, msg in l: | ||||
repo.ui.note(' %s -> %s\n' % (msg, m)) | ||||
# Pick random action. TODO: Instead, prompt user when resolving | ||||
m, l = bids.items()[0] | ||||
repo.ui.warn(_(' %s: ambiguous merge - picked %s action\n') % | ||||
(f, m)) | ||||
Martin von Zweigbergk
|
r23638 | actions[f] = l[0] | ||
Martin von Zweigbergk
|
r23385 | continue | ||
repo.ui.note(_('end of auction\n\n')) | ||||
Martin von Zweigbergk
|
r23639 | _resolvetrivial(repo, wctx, mctx, ancestors[0], actions) | ||
Martin von Zweigbergk
|
r23640 | if wctx.rev() is None: | ||
fractions = _forgetremoved(wctx, mctx, branchmerge) | ||||
actions.update(fractions) | ||||
Martin von Zweigbergk
|
r23526 | return actions, diverge, renamedelete | ||
Martin von Zweigbergk
|
r23385 | |||
Mads Kiilerich
|
r21392 | def batchremove(repo, actions): | ||
"""apply removes to the working directory | ||||
Bryan O'Sullivan
|
r18630 | |||
yields tuples for progress updates | ||||
""" | ||||
Bryan O'Sullivan
|
r18640 | verbose = repo.ui.verbose | ||
unlink = util.unlinkpath | ||||
wjoin = repo.wjoin | ||||
Angel Ezquerra
|
r23879 | audit = repo.wvfs.audit | ||
Bryan O'Sullivan
|
r18633 | i = 0 | ||
Mads Kiilerich
|
r21545 | for f, args, msg in actions: | ||
Mads Kiilerich
|
r21392 | repo.ui.debug(" %s: %s -> r\n" % (f, msg)) | ||
Mads Kiilerich
|
r21551 | if verbose: | ||
repo.ui.note(_("removing %s\n") % f) | ||||
audit(f) | ||||
try: | ||||
unlink(wjoin(f), ignoremissing=True) | ||||
Gregory Szorc
|
r25660 | except OSError as inst: | ||
Mads Kiilerich
|
r21551 | repo.ui.warn(_("update failed to remove %s: %s!\n") % | ||
(f, inst.strerror)) | ||||
Mads Kiilerich
|
r21392 | if i == 100: | ||
yield i, f | ||||
i = 0 | ||||
i += 1 | ||||
if i > 0: | ||||
yield i, f | ||||
def batchget(repo, mctx, actions): | ||||
"""apply gets to the working directory | ||||
mctx is the context to get from | ||||
yields tuples for progress updates | ||||
""" | ||||
verbose = repo.ui.verbose | ||||
fctx = mctx.filectx | ||||
wwrite = repo.wwrite | ||||
Siddharth Agarwal
|
r27656 | ui = repo.ui | ||
Mads Kiilerich
|
r21392 | i = 0 | ||
Gregory Szorc
|
r28200 | with repo.wvfs.backgroundclosing(ui, expectedcount=len(actions)): | ||
Gregory Szorc
|
r28199 | for f, (flags, backup), msg in actions: | ||
repo.ui.debug(" %s: %s -> g\n" % (f, msg)) | ||||
if verbose: | ||||
repo.ui.note(_("getting %s\n") % f) | ||||
Siddharth Agarwal
|
r27656 | |||
Gregory Szorc
|
r28199 | if backup: | ||
absf = repo.wjoin(f) | ||||
orig = scmutil.origpath(ui, repo, absf) | ||||
try: | ||||
# TODO Mercurial has always aborted if an untracked | ||||
# directory is replaced by a tracked file, or generally | ||||
# with file/directory merges. This needs to be sorted out. | ||||
if repo.wvfs.isfileorlink(f): | ||||
util.rename(absf, orig) | ||||
except OSError as e: | ||||
if e.errno != errno.ENOENT: | ||||
raise | ||||
Siddharth Agarwal
|
r27656 | |||
Gregory Szorc
|
r28200 | wwrite(f, fctx(f).data(), flags, backgroundclose=True) | ||
Gregory Szorc
|
r28199 | if i == 100: | ||
yield i, f | ||||
i = 0 | ||||
i += 1 | ||||
Bryan O'Sullivan
|
r18633 | if i > 0: | ||
Bryan O'Sullivan
|
r18630 | yield i, f | ||
Durham Goode
|
r21524 | def applyupdates(repo, actions, wctx, mctx, overwrite, labels=None): | ||
Peter Arrenbrecht
|
r11454 | """apply the merge action list to the working directory | ||
wctx is the working copy context | ||||
mctx is the context to be merged into the working copy | ||||
Greg Ward
|
r13162 | |||
Return a tuple of counts (updated, merged, removed, unresolved) that | ||||
describes how many files were affected by the update. | ||||
Peter Arrenbrecht
|
r11454 | """ | ||
Matt Mackall
|
r3315 | |||
Siddharth Agarwal
|
r27078 | updated, merged, removed = 0, 0, 0 | ||
Simon Farnsworth
|
r28634 | ms = mergestate.clean(repo, wctx.p1().node(), mctx.node(), labels) | ||
Matt Mackall
|
r6512 | moves = [] | ||
Mads Kiilerich
|
r21545 | for m, l in actions.items(): | ||
l.sort() | ||||
Matt Mackall
|
r6512 | |||
Siddharth Agarwal
|
r27137 | # 'cd' and 'dc' actions are treated like other merge conflicts | ||
mergeactions = sorted(actions['cd']) | ||||
mergeactions.extend(sorted(actions['dc'])) | ||||
mergeactions.extend(actions['m']) | ||||
for f, args, msg in mergeactions: | ||||
Mads Kiilerich
|
r21551 | f1, f2, fa, move, anc = args | ||
if f == '.hgsubstate': # merged internally | ||||
continue | ||||
Siddharth Agarwal
|
r27091 | if f1 is None: | ||
fcl = filemerge.absentfilectx(wctx, fa) | ||||
else: | ||||
repo.ui.debug(" preserving %s for resolve of %s\n" % (f1, f)) | ||||
fcl = wctx[f1] | ||||
if f2 is None: | ||||
fco = filemerge.absentfilectx(mctx, fa) | ||||
else: | ||||
fco = mctx[f2] | ||||
Mads Kiilerich
|
r21551 | actx = repo[anc] | ||
if fa in actx: | ||||
fca = actx[fa] | ||||
else: | ||||
Siddharth Agarwal
|
r27091 | # TODO: move to absentfilectx | ||
Mads Kiilerich
|
r21551 | fca = repo.filectx(f1, fileid=nullrev) | ||
ms.add(fcl, fco, fca, f) | ||||
if f1 != f and move: | ||||
moves.append(f1) | ||||
Matt Mackall
|
r6512 | |||
Angel Ezquerra
|
r23879 | audit = repo.wvfs.audit | ||
Mads Kiilerich
|
r21390 | _updating = _('updating') | ||
_files = _('files') | ||||
progress = repo.ui.progress | ||||
Adrian Buehlmann
|
r14398 | |||
Matt Mackall
|
r6512 | # remove renamed files after safely stored | ||
for f in moves: | ||||
Martin Geisler
|
r12032 | if os.path.lexists(repo.wjoin(f)): | ||
Martin Geisler
|
r9467 | repo.ui.debug("removing %s\n" % f) | ||
Adrian Buehlmann
|
r14398 | audit(f) | ||
Mads Kiilerich
|
r18333 | util.unlinkpath(repo.wjoin(f)) | ||
Matt Mackall
|
r5042 | |||
Martin von Zweigbergk
|
r23526 | numupdates = sum(len(l) for m, l in actions.items() if m != 'k') | ||
Bryan O'Sullivan
|
r18630 | |||
Mads Kiilerich
|
r21545 | if [a for a in actions['r'] if a[0] == '.hgsubstate']: | ||
Bryan O'Sullivan
|
r18632 | subrepo.submerge(repo, wctx, mctx, wctx, overwrite) | ||
Mads Kiilerich
|
r21390 | # remove in parallel (must come first) | ||
Bryan O'Sullivan
|
r18633 | z = 0 | ||
Mads Kiilerich
|
r21545 | prog = worker.worker(repo.ui, 0.001, batchremove, (repo,), actions['r']) | ||
FUJIWARA Katsunori
|
r19095 | for i, item in prog: | ||
z += i | ||||
Mads Kiilerich
|
r21390 | progress(_updating, z, item=item, total=numupdates, unit=_files) | ||
Mads Kiilerich
|
r21545 | removed = len(actions['r']) | ||
Mads Kiilerich
|
r21390 | |||
# get in parallel | ||||
Mads Kiilerich
|
r21545 | prog = worker.worker(repo.ui, 0.001, batchget, (repo, mctx), actions['g']) | ||
Bryan O'Sullivan
|
r18639 | for i, item in prog: | ||
Bryan O'Sullivan
|
r18633 | z += i | ||
Mads Kiilerich
|
r21390 | progress(_updating, z, item=item, total=numupdates, unit=_files) | ||
Mads Kiilerich
|
r21545 | updated = len(actions['g']) | ||
Bryan O'Sullivan
|
r18630 | |||
Mads Kiilerich
|
r21545 | if [a for a in actions['g'] if a[0] == '.hgsubstate']: | ||
Bryan O'Sullivan
|
r18632 | subrepo.submerge(repo, wctx, mctx, wctx, overwrite) | ||
Mads Kiilerich
|
r21551 | # forget (manifest only, just log it) (must come first) | ||
for f, args, msg in actions['f']: | ||||
repo.ui.debug(" %s: %s -> f\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # re-add (manifest only, just log it) | ||
for f, args, msg in actions['a']: | ||||
repo.ui.debug(" %s: %s -> a\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
Mads Kiilerich
|
r21391 | |||
Siddharth Agarwal
|
r27131 | # re-add/mark as modified (manifest only, just log it) | ||
for f, args, msg in actions['am']: | ||||
repo.ui.debug(" %s: %s -> am\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
Mads Kiilerich
|
r21551 | # keep (noop, just log it) | ||
for f, args, msg in actions['k']: | ||||
repo.ui.debug(" %s: %s -> k\n" % (f, msg)) | ||||
# no progress | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # directory rename, move local | ||
for f, args, msg in actions['dm']: | ||||
repo.ui.debug(" %s: %s -> dm\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
f0, flags = args | ||||
repo.ui.note(_("moving %s to %s\n") % (f0, f)) | ||||
audit(f) | ||||
repo.wwrite(f, wctx.filectx(f0).data(), flags) | ||||
util.unlinkpath(repo.wjoin(f0)) | ||||
updated += 1 | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # local directory rename, get | ||
for f, args, msg in actions['dg']: | ||||
repo.ui.debug(" %s: %s -> dg\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
f0, flags = args | ||||
repo.ui.note(_("getting %s to %s\n") % (f0, f)) | ||||
repo.wwrite(f, mctx.filectx(f0).data(), flags) | ||||
updated += 1 | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # exec | ||
for f, args, msg in actions['e']: | ||||
repo.ui.debug(" %s: %s -> e\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
flags, = args | ||||
audit(f) | ||||
util.setflags(repo.wjoin(f), 'l' in flags, 'x' in flags) | ||||
updated += 1 | ||||
Mads Kiilerich
|
r21391 | |||
Siddharth Agarwal
|
r26786 | # the ordering is important here -- ms.mergedriver will raise if the merge | ||
# driver has changed, and we want to be able to bypass it when overwrite is | ||||
# True | ||||
usemergedriver = not overwrite and mergeactions and ms.mergedriver | ||||
if usemergedriver: | ||||
ms.commit() | ||||
proceed = driverpreprocess(repo, ms, wctx, labels=labels) | ||||
# the driver might leave some files unresolved | ||||
unresolvedf = set(ms.unresolved()) | ||||
if not proceed: | ||||
# XXX setting unresolved to at least 1 is a hack to make sure we | ||||
# error out | ||||
return updated, merged, removed, max(len(unresolvedf), 1) | ||||
newactions = [] | ||||
for f, args, msg in mergeactions: | ||||
if f in unresolvedf: | ||||
newactions.append((f, args, msg)) | ||||
mergeactions = newactions | ||||
Siddharth Agarwal
|
r26618 | # premerge | ||
tocomplete = [] | ||||
Siddharth Agarwal
|
r26949 | for f, args, msg in mergeactions: | ||
Siddharth Agarwal
|
r26618 | repo.ui.debug(" %s: %s -> m (premerge)\n" % (f, msg)) | ||
Siddharth Agarwal
|
r26292 | z += 1 | ||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
if f == '.hgsubstate': # subrepo states need updating | ||||
subrepo.submerge(repo, wctx, mctx, wctx.ancestor(mctx), | ||||
overwrite) | ||||
continue | ||||
audit(f) | ||||
Simon Farnsworth
|
r28634 | complete, r = ms.preresolve(f, wctx) | ||
Siddharth Agarwal
|
r27078 | if not complete: | ||
Siddharth Agarwal
|
r26618 | numupdates += 1 | ||
tocomplete.append((f, args, msg)) | ||||
# merge | ||||
for f, args, msg in tocomplete: | ||||
repo.ui.debug(" %s: %s -> m (merge)\n" % (f, msg)) | ||||
z += 1 | ||||
progress(_updating, z, item=f, total=numupdates, unit=_files) | ||||
Simon Farnsworth
|
r28634 | ms.resolve(f, wctx) | ||
Siddharth Agarwal
|
r26292 | |||
Peter Arrenbrecht
|
r12369 | ms.commit() | ||
Siddharth Agarwal
|
r26787 | |||
Siddharth Agarwal
|
r27078 | unresolved = ms.unresolvedcount() | ||
Siddharth Agarwal
|
r26787 | if usemergedriver and not unresolved and ms.mdstate() != 's': | ||
if not driverconclude(repo, ms, wctx, labels=labels): | ||||
# XXX setting unresolved to at least 1 is a hack to make sure we | ||||
# error out | ||||
Siddharth Agarwal
|
r26975 | unresolved = max(unresolved, 1) | ||
Siddharth Agarwal
|
r26787 | |||
ms.commit() | ||||
Siddharth Agarwal
|
r27078 | msupdated, msmerged, msremoved = ms.counts() | ||
updated += msupdated | ||||
merged += msmerged | ||||
removed += msremoved | ||||
Siddharth Agarwal
|
r27080 | |||
extraactions = ms.actions() | ||||
Siddharth Agarwal
|
r27121 | for k, acts in extraactions.iteritems(): | ||
actions[k].extend(acts) | ||||
Siddharth Agarwal
|
r27080 | |||
Bryan O'Sullivan
|
r18640 | progress(_updating, None, total=numupdates, unit=_files) | ||
Matt Mackall
|
r3111 | |||
return updated, merged, removed, unresolved | ||||
Mads Kiilerich
|
r18330 | def recordupdates(repo, actions, branchmerge): | ||
Matt Mackall
|
r3315 | "record merge actions to the dirstate" | ||
Mads Kiilerich
|
r21551 | # remove (must come first) | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('r', []): | ||
Mads Kiilerich
|
r21551 | if branchmerge: | ||
repo.dirstate.remove(f) | ||||
else: | ||||
Mads Kiilerich
|
r21389 | repo.dirstate.drop(f) | ||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # forget (must come first) | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('f', []): | ||
Mads Kiilerich
|
r21551 | repo.dirstate.drop(f) | ||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # re-add | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('a', []): | ||
Siddharth Agarwal
|
r27132 | repo.dirstate.add(f) | ||
Mads Kiilerich
|
r21391 | |||
Siddharth Agarwal
|
r27131 | # re-add/mark as modified | ||
for f, args, msg in actions.get('am', []): | ||||
if branchmerge: | ||||
repo.dirstate.normallookup(f) | ||||
else: | ||||
Mads Kiilerich
|
r21551 | repo.dirstate.add(f) | ||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # exec change | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('e', []): | ||
Mads Kiilerich
|
r21551 | repo.dirstate.normallookup(f) | ||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # keep | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('k', []): | ||
Mads Kiilerich
|
r21551 | pass | ||
# get | ||||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('g', []): | ||
Mads Kiilerich
|
r21551 | if branchmerge: | ||
repo.dirstate.otherparent(f) | ||||
else: | ||||
repo.dirstate.normal(f) | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # merge | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('m', []): | ||
Mads Kiilerich
|
r21551 | f1, f2, fa, move, anc = args | ||
if branchmerge: | ||||
# We've done a branch merge, mark this file as merged | ||||
# so that we properly record the merger later | ||||
repo.dirstate.merge(f) | ||||
if f1 != f2: # copy/rename | ||||
Matt Mackall
|
r3308 | if move: | ||
Mads Kiilerich
|
r21551 | repo.dirstate.remove(f1) | ||
if f1 != f: | ||||
repo.dirstate.copy(f1, f) | ||||
else: | ||||
repo.dirstate.copy(f2, f) | ||||
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. | ||||
if f2 == f: # file not locally copied/moved | ||||
repo.dirstate.normallookup(f) | ||||
if move: | ||||
repo.dirstate.drop(f1) | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # directory rename, move local | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('dm', []): | ||
Mads Kiilerich
|
r21551 | f0, flag = args | ||
if branchmerge: | ||||
repo.dirstate.add(f) | ||||
repo.dirstate.remove(f0) | ||||
repo.dirstate.copy(f0, f) | ||||
else: | ||||
repo.dirstate.normal(f) | ||||
repo.dirstate.drop(f0) | ||||
Mads Kiilerich
|
r21391 | |||
Mads Kiilerich
|
r21551 | # directory rename, get | ||
Siddharth Agarwal
|
r27087 | for f, args, msg in actions.get('dg', []): | ||
Mads Kiilerich
|
r21551 | f0, flag = args | ||
if branchmerge: | ||||
repo.dirstate.add(f) | ||||
repo.dirstate.copy(f0, f) | ||||
else: | ||||
repo.dirstate.normal(f) | ||||
Matt Mackall
|
r3111 | |||
Augie Fackler
|
r27344 | def update(repo, node, branchmerge, force, ancestor=None, | ||
Siddharth Agarwal
|
r28020 | mergeancestor=False, labels=None, matcher=None, mergeforce=False): | ||
Matt Mackall
|
r3315 | """ | ||
Perform a merge between the working directory and the given node | ||||
Stuart W Marks
|
r9716 | node = the node to update to, or None if unspecified | ||
Matt Mackall
|
r3315 | branchmerge = whether to merge between branches | ||
force = whether to force branch merging or file overwriting | ||||
Augie Fackler
|
r27344 | matcher = a matcher to filter file lists (dirstate not updated) | ||
Durham Goode
|
r18778 | mergeancestor = whether it is merging with an ancestor. If true, | ||
we should accept the incoming changes for any prompts that occur. | ||||
If false, merging with an ancestor (fast-forward) is only allowed | ||||
between different named branches. This flag is used by rebase extension | ||||
as a temporary fix and should be avoided in general. | ||||
Siddharth Agarwal
|
r28019 | labels = labels to use for base, local and other | ||
Siddharth Agarwal
|
r28020 | mergeforce = whether the merge was run with 'merge --force' (deprecated): if | ||
this is True, then 'force' should be True as well. | ||||
Stuart W Marks
|
r9716 | |||
The table below shows all the behaviors of the update command | ||||
given the -c and -C or no options, whether the working directory | ||||
is dirty, whether a revision is specified, and the relationship of | ||||
the parent rev to the target rev (linear, on the same named | ||||
branch, or on another named branch). | ||||
Adrian Buehlmann
|
r12279 | This logic is tested by test-update-branches.t. | ||
Stuart W Marks
|
r9716 | |||
Augie Fackler
|
r26821 | -c -C dirty rev | linear same cross | ||
n n n n | ok (1) x | ||||
n n n y | ok ok ok | ||||
n n y n | merge (2) (2) | ||||
n n y y | merge (3) (3) | ||||
n y * * | discard discard discard | ||||
y n y * | (4) (4) (4) | ||||
y n n * | ok ok ok | ||||
y y * * | (5) (5) (5) | ||||
Stuart W Marks
|
r9716 | |||
x = can't happen | ||||
* = don't-care | ||||
Siddharth Agarwal
|
r19798 | 1 = abort: not a linear update (merge or update --check to force update) | ||
Siddharth Agarwal
|
r19800 | 2 = abort: uncommitted changes (commit and merge, or update --clean to | ||
discard changes) | ||||
Siddharth Agarwal
|
r19799 | 3 = abort: uncommitted changes (commit or update --clean to discard changes) | ||
Siddharth Agarwal
|
r19801 | 4 = abort: uncommitted changes (checked in commands.py) | ||
Siddharth Agarwal
|
r19799 | 5 = incompatible options (checked in commands.py) | ||
Greg Ward
|
r13162 | |||
Return the same tuple as applyupdates(). | ||||
Matt Mackall
|
r3315 | """ | ||
Matt Mackall
|
r2815 | |||
Stuart W Marks
|
r9717 | onode = node | ||
Augie Fackler
|
r27344 | # If we're doing a partial update, we need to skip updating | ||
# the dirstate, so make a note of any partial-ness to the | ||||
# update here. | ||||
if matcher is None or matcher.always(): | ||||
partial = False | ||||
else: | ||||
partial = True | ||||
Bryan O'Sullivan
|
r27852 | with repo.wlock(): | ||
Matt Mackall
|
r6747 | wc = repo[None] | ||
Sean Farley
|
r20279 | pl = wc.parents() | ||
p1 = pl[0] | ||||
Mads Kiilerich
|
r21081 | pas = [None] | ||
Mads Kiilerich
|
r23405 | if ancestor is not None: | ||
Mads Kiilerich
|
r21081 | pas = [repo[ancestor]] | ||
Sean Farley
|
r20279 | |||
Matt Mackall
|
r4915 | if node is None: | ||
Pierre-Yves David
|
r26682 | if (repo.ui.configbool('devel', 'all-warnings') | ||
or repo.ui.configbool('devel', 'oldapi')): | ||||
repo.ui.develwarn('update with no target') | ||||
Pierre-Yves David
|
r26641 | rev, _mark, _act = destutil.destupdate(repo) | ||
node = repo[rev].node() | ||||
Sean Farley
|
r20280 | |||
Matt Mackall
|
r4915 | overwrite = force and not branchmerge | ||
Sean Farley
|
r20279 | |||
p2 = repo[node] | ||||
Mads Kiilerich
|
r21081 | if pas[0] is None: | ||
Matt Mackall
|
r25844 | if repo.ui.configlist('merge', 'preferancestor', ['*']) == ['*']: | ||
Mads Kiilerich
|
r21128 | cahs = repo.changelog.commonancestorsheads(p1.node(), p2.node()) | ||
pas = [repo[anc] for anc in (sorted(cahs) or [nullid])] | ||||
else: | ||||
Mads Kiilerich
|
r22179 | pas = [p1.ancestor(p2, warn=branchmerge)] | ||
Matt Mackall
|
r13874 | |||
Matt Mackall
|
r4915 | fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2) | ||
Matt Mackall
|
r3314 | |||
Matt Mackall
|
r4915 | ### check phase | ||
Martin von Zweigbergk
|
r27316 | if not overwrite: | ||
if len(pl) > 1: | ||||
raise error.Abort(_("outstanding uncommitted merge")) | ||||
ms = mergestate.read(repo) | ||||
if list(ms.unresolved()): | ||||
raise error.Abort(_("outstanding merge conflicts")) | ||||
Matt Mackall
|
r6375 | if branchmerge: | ||
Mads Kiilerich
|
r21081 | if pas == [p2]: | ||
Pierre-Yves David
|
r26587 | raise error.Abort(_("merging with a working directory ancestor" | ||
Matt Mackall
|
r11417 | " has no effect")) | ||
Mads Kiilerich
|
r21081 | elif pas == [p1]: | ||
Patrick Mezard
|
r16696 | if not mergeancestor and p1.branch() == p2.branch(): | ||
Pierre-Yves David
|
r26587 | raise error.Abort(_("nothing to merge"), | ||
Kevin Bullock
|
r15619 | hint=_("use 'hg update' " | ||
"or check 'hg heads'")) | ||||
Matt Mackall
|
r6375 | if not force and (wc.files() or wc.deleted()): | ||
Pierre-Yves David
|
r26587 | raise error.Abort(_("uncommitted changes"), | ||
Kevin Bullock
|
r15619 | hint=_("use 'hg status' to list changes")) | ||
Mads Kiilerich
|
r18364 | for s in sorted(wc.substate): | ||
FUJIWARA Katsunori
|
r24471 | wc.sub(s).bailifchanged() | ||
Oleg Stepanov
|
r13437 | |||
Matt Mackall
|
r6375 | elif not overwrite: | ||
Siddharth Agarwal
|
r19929 | if p1 == p2: # no-op update | ||
# call the hooks and exit early | ||||
repo.hook('preupdate', throw=True, parent1=xp2, parent2='') | ||||
repo.hook('update', parent1=xp2, parent2='', error=0) | ||||
return 0, 0, 0, 0 | ||||
Mads Kiilerich
|
r21081 | if pas not in ([p1], [p2]): # nonlinear | ||
Pierre-Yves David
|
r18985 | dirty = wc.dirty(missing=True) | ||
if dirty or onode is None: | ||||
# Branching is a bit strange to ensure we do the minimal | ||||
# amount of call to obsolete.background. | ||||
foreground = obsolete.foreground(repo, [p1.node()]) | ||||
# note: the <node> variable contains a random identifier | ||||
if repo[node].node() in foreground: | ||||
Mads Kiilerich
|
r21081 | pas = [p1] # allow updating to successors | ||
Siddharth Agarwal
|
r19799 | elif dirty: | ||
msg = _("uncommitted changes") | ||||
Siddharth Agarwal
|
r19800 | if onode is None: | ||
hint = _("commit and merge, or update --clean to" | ||||
" discard changes") | ||||
else: | ||||
hint = _("commit or update --clean to discard" | ||||
" changes") | ||||
Pierre-Yves David
|
r26587 | raise error.Abort(msg, hint=hint) | ||
Pierre-Yves David
|
r18985 | else: # node is none | ||
Siddharth Agarwal
|
r19798 | msg = _("not a linear update") | ||
hint = _("merge or update --check to force update") | ||||
Pierre-Yves David
|
r26587 | raise error.Abort(msg, hint=hint) | ||
Pierre-Yves David
|
r18985 | else: | ||
# Allow jumping branches if clean and specific rev given | ||||
Mads Kiilerich
|
r21081 | pas = [p1] | ||
Matt Mackall
|
r2814 | |||
Matt Mackall
|
r25843 | # deprecated config: merge.followcopies | ||
Mads Kiilerich
|
r21080 | followcopies = False | ||
if overwrite: | ||||
Mads Kiilerich
|
r21081 | pas = [wc] | ||
elif pas == [p2]: # backwards | ||||
pas = [wc.p1()] | ||||
Mads Kiilerich
|
r21080 | elif not branchmerge and not wc.dirty(missing=True): | ||
pass | ||||
Martin von Zweigbergk
|
r23380 | elif pas[0] and repo.ui.configbool('merge', 'followcopies', True): | ||
Mads Kiilerich
|
r21080 | followcopies = True | ||
Matt Mackall
|
r4915 | ### calculate phase | ||
Martin von Zweigbergk
|
r23641 | actionbyfile, diverge, renamedelete = calculateupdates( | ||
Augie Fackler
|
r27345 | repo, wc, p2, pas, branchmerge, force, mergeancestor, | ||
Siddharth Agarwal
|
r28020 | followcopies, matcher=matcher, mergeforce=mergeforce) | ||
Siddharth Agarwal
|
r27951 | |||
# Prompt and create actions. Most of this is in the resolve phase | ||||
# already, but we can't handle .hgsubstate in filemerge or | ||||
# subrepo.submerge yet so we have to keep prompting for it. | ||||
if '.hgsubstate' in actionbyfile: | ||||
f = '.hgsubstate' | ||||
m, args, msg = actionbyfile[f] | ||||
if m == 'cd': | ||||
if repo.ui.promptchoice( | ||||
_("local changed %s which remote deleted\n" | ||||
"use (c)hanged version or (d)elete?" | ||||
"$$ &Changed $$ &Delete") % f, 0): | ||||
actionbyfile[f] = ('r', None, "prompt delete") | ||||
elif f in p1: | ||||
actionbyfile[f] = ('am', None, "prompt keep") | ||||
else: | ||||
actionbyfile[f] = ('a', None, "prompt keep") | ||||
elif m == 'dc': | ||||
f1, f2, fa, move, anc = args | ||||
flags = p2[f2].flags() | ||||
if repo.ui.promptchoice( | ||||
_("remote changed %s which local deleted\n" | ||||
"use (c)hanged version or leave (d)eleted?" | ||||
"$$ &Changed $$ &Deleted") % f, 0) == 0: | ||||
actionbyfile[f] = ('g', (flags, False), "prompt recreating") | ||||
else: | ||||
del actionbyfile[f] | ||||
Martin von Zweigbergk
|
r23641 | # Convert to dictionary-of-lists format | ||
Siddharth Agarwal
|
r27131 | actions = dict((m, []) for m in 'a am f g cd dc r dm dg m e k'.split()) | ||
Martin von Zweigbergk
|
r23641 | for f, (m, args, msg) in actionbyfile.iteritems(): | ||
if m not in actions: | ||||
actions[m] = [] | ||||
actions[m].append((f, args, msg)) | ||||
Matt Mackall
|
r2775 | |||
Martin von Zweigbergk
|
r23544 | if not util.checkcase(repo.path): | ||
# check collision between files only in p2 for clean update | ||||
if (not branchmerge and | ||||
(force or not wc.dirty(missing=True, branch=False))): | ||||
_checkcollision(repo, p2.manifest(), None) | ||||
else: | ||||
_checkcollision(repo, wc.manifest(), actions) | ||||
Martin von Zweigbergk
|
r23525 | # divergent renames | ||
Martin von Zweigbergk
|
r23526 | for f, fl in sorted(diverge.iteritems()): | ||
Martin von Zweigbergk
|
r23525 | repo.ui.warn(_("note: possible conflict - %s was renamed " | ||
"multiple times to:\n") % f) | ||||
for nf in fl: | ||||
repo.ui.warn(" %s\n" % nf) | ||||
# rename and delete | ||||
Martin von Zweigbergk
|
r23526 | for f, fl in sorted(renamedelete.iteritems()): | ||
Martin von Zweigbergk
|
r23525 | repo.ui.warn(_("note: possible conflict - %s was deleted " | ||
"and renamed to:\n") % f) | ||||
for nf in fl: | ||||
repo.ui.warn(" %s\n" % nf) | ||||
Martin von Zweigbergk
|
r26957 | ### 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) | ||||
# note that we're in the middle of an update | ||||
repo.vfs.write('updatestate', p2.hex()) | ||||
stats = applyupdates(repo, actions, wc, p2, overwrite, labels=labels) | ||||
Matt Mackall
|
r4915 | if not partial: | ||
Durham Goode
|
r22405 | repo.dirstate.beginparentchange() | ||
Patrick Mezard
|
r16551 | repo.setparents(fp1, fp2) | ||
Mads Kiilerich
|
r18330 | recordupdates(repo, actions, branchmerge) | ||
Matt Mackall
|
r19482 | # update completed, clear state | ||
util.unlink(repo.join('updatestate')) | ||||
Mads Kiilerich
|
r13561 | if not branchmerge: | ||
Matt Mackall
|
r4915 | repo.dirstate.setbranch(p2.branch()) | ||
Durham Goode
|
r22405 | repo.dirstate.endparentchange() | ||
Sune Foldager
|
r10492 | |||
if not partial: | ||||
FUJIWARA Katsunori
|
r26752 | repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3]) | ||
Sune Foldager
|
r10492 | return stats | ||
Matt Mackall
|
r22902 | |||
Andrew Halberstadt
|
r27267 | def graft(repo, ctx, pctx, labels, keepparent=False): | ||
Matt Mackall
|
r22902 | """Do a graft-like merge. | ||
This is a merge where the merge ancestor is chosen such that one | ||||
or more changesets are grafted onto the current changeset. In | ||||
addition to the merge, this fixes up the dirstate to include only | ||||
Andrew Halberstadt
|
r27267 | a single parent (if keepparent is False) and tries to duplicate any | ||
renames/copies appropriately. | ||||
Matt Mackall
|
r22902 | |||
ctx - changeset to rebase | ||||
pctx - merge base, usually ctx.p1() | ||||
labels - merge labels eg ['local', 'graft'] | ||||
Andrew Halberstadt
|
r27267 | keepparent - keep second parent if any | ||
Matt Mackall
|
r22902 | |||
""" | ||||
Durham Goode
|
r24643 | # If we're grafting a descendant onto an ancestor, be sure to pass | ||
# mergeancestor=True to update. This does two things: 1) allows the merge if | ||||
# the destination is the same as the parent of the ctx (so we can use graft | ||||
# to copy commits), and 2) informs update that the incoming changes are | ||||
# newer than the destination so it doesn't prompt about "remote changed foo | ||||
# which local deleted". | ||||
mergeancestor = repo.changelog.isancestor(repo['.'].node(), ctx.node()) | ||||
Matt Mackall
|
r22902 | |||
Augie Fackler
|
r27344 | stats = update(repo, ctx.node(), True, True, pctx.node(), | ||
Durham Goode
|
r24643 | mergeancestor=mergeancestor, labels=labels) | ||
Andrew Halberstadt
|
r27267 | pother = nullid | ||
parents = ctx.parents() | ||||
if keepparent and len(parents) == 2 and pctx in parents: | ||||
parents.remove(pctx) | ||||
pother = parents[0].node() | ||||
Matt Mackall
|
r22902 | repo.dirstate.beginparentchange() | ||
Andrew Halberstadt
|
r27267 | repo.setparents(repo['.'].node(), pother) | ||
FUJIWARA Katsunori
|
r26748 | repo.dirstate.write(repo.currenttransaction()) | ||
Matt Mackall
|
r22902 | # fix up dirstate for copies and renames | ||
copies.duplicatecopies(repo, ctx.rev(), pctx.rev()) | ||||
repo.dirstate.endparentchange() | ||||
return stats | ||||