##// END OF EJS Templates
merge: don't check for unknown files in IMM...
Phil Cohen -
r35289:5db3c748 default
parent child Browse files
Show More
@@ -1,2054 +1,2066
1 # merge.py - directory-level update/merge handling for Mercurial
1 # merge.py - directory-level update/merge handling for Mercurial
2 #
2 #
3 # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import errno
10 import errno
11 import hashlib
11 import hashlib
12 import shutil
12 import shutil
13 import struct
13 import struct
14
14
15 from .i18n import _
15 from .i18n import _
16 from .node import (
16 from .node import (
17 addednodeid,
17 addednodeid,
18 bin,
18 bin,
19 hex,
19 hex,
20 modifiednodeid,
20 modifiednodeid,
21 nullhex,
21 nullhex,
22 nullid,
22 nullid,
23 nullrev,
23 nullrev,
24 )
24 )
25 from . import (
25 from . import (
26 copies,
26 copies,
27 error,
27 error,
28 extensions,
28 extensions,
29 filemerge,
29 filemerge,
30 match as matchmod,
30 match as matchmod,
31 obsutil,
31 obsutil,
32 pycompat,
32 pycompat,
33 scmutil,
33 scmutil,
34 subrepo,
34 subrepo,
35 util,
35 util,
36 worker,
36 worker,
37 )
37 )
38
38
39 _pack = struct.pack
39 _pack = struct.pack
40 _unpack = struct.unpack
40 _unpack = struct.unpack
41
41
42 def _droponode(data):
42 def _droponode(data):
43 # used for compatibility for v1
43 # used for compatibility for v1
44 bits = data.split('\0')
44 bits = data.split('\0')
45 bits = bits[:-2] + bits[-1:]
45 bits = bits[:-2] + bits[-1:]
46 return '\0'.join(bits)
46 return '\0'.join(bits)
47
47
48 class mergestate(object):
48 class mergestate(object):
49 '''track 3-way merge state of individual files
49 '''track 3-way merge state of individual files
50
50
51 The merge state is stored on disk when needed. Two files are used: one with
51 The merge state is stored on disk when needed. Two files are used: one with
52 an old format (version 1), and one with a new format (version 2). Version 2
52 an old format (version 1), and one with a new format (version 2). Version 2
53 stores a superset of the data in version 1, including new kinds of records
53 stores a superset of the data in version 1, including new kinds of records
54 in the future. For more about the new format, see the documentation for
54 in the future. For more about the new format, see the documentation for
55 `_readrecordsv2`.
55 `_readrecordsv2`.
56
56
57 Each record can contain arbitrary content, and has an associated type. This
57 Each record can contain arbitrary content, and has an associated type. This
58 `type` should be a letter. If `type` is uppercase, the record is mandatory:
58 `type` should be a letter. If `type` is uppercase, the record is mandatory:
59 versions of Mercurial that don't support it should abort. If `type` is
59 versions of Mercurial that don't support it should abort. If `type` is
60 lowercase, the record can be safely ignored.
60 lowercase, the record can be safely ignored.
61
61
62 Currently known records:
62 Currently known records:
63
63
64 L: the node of the "local" part of the merge (hexified version)
64 L: the node of the "local" part of the merge (hexified version)
65 O: the node of the "other" part of the merge (hexified version)
65 O: the node of the "other" part of the merge (hexified version)
66 F: a file to be merged entry
66 F: a file to be merged entry
67 C: a change/delete or delete/change conflict
67 C: a change/delete or delete/change conflict
68 D: a file that the external merge driver will merge internally
68 D: a file that the external merge driver will merge internally
69 (experimental)
69 (experimental)
70 P: a path conflict (file vs directory)
70 P: a path conflict (file vs directory)
71 m: the external merge driver defined for this merge plus its run state
71 m: the external merge driver defined for this merge plus its run state
72 (experimental)
72 (experimental)
73 f: a (filename, dictionary) tuple of optional values for a given file
73 f: a (filename, dictionary) tuple of optional values for a given file
74 X: unsupported mandatory record type (used in tests)
74 X: unsupported mandatory record type (used in tests)
75 x: unsupported advisory record type (used in tests)
75 x: unsupported advisory record type (used in tests)
76 l: the labels for the parts of the merge.
76 l: the labels for the parts of the merge.
77
77
78 Merge driver run states (experimental):
78 Merge driver run states (experimental):
79 u: driver-resolved files unmarked -- needs to be run next time we're about
79 u: driver-resolved files unmarked -- needs to be run next time we're about
80 to resolve or commit
80 to resolve or commit
81 m: driver-resolved files marked -- only needs to be run before commit
81 m: driver-resolved files marked -- only needs to be run before commit
82 s: success/skipped -- does not need to be run any more
82 s: success/skipped -- does not need to be run any more
83
83
84 Merge record states (stored in self._state, indexed by filename):
84 Merge record states (stored in self._state, indexed by filename):
85 u: unresolved conflict
85 u: unresolved conflict
86 r: resolved conflict
86 r: resolved conflict
87 pu: unresolved path conflict (file conflicts with directory)
87 pu: unresolved path conflict (file conflicts with directory)
88 pr: resolved path conflict
88 pr: resolved path conflict
89 d: driver-resolved conflict
89 d: driver-resolved conflict
90
90
91 The resolve command transitions between 'u' and 'r' for conflicts and
91 The resolve command transitions between 'u' and 'r' for conflicts and
92 'pu' and 'pr' for path conflicts.
92 'pu' and 'pr' for path conflicts.
93 '''
93 '''
94 statepathv1 = 'merge/state'
94 statepathv1 = 'merge/state'
95 statepathv2 = 'merge/state2'
95 statepathv2 = 'merge/state2'
96
96
97 @staticmethod
97 @staticmethod
98 def clean(repo, node=None, other=None, labels=None):
98 def clean(repo, node=None, other=None, labels=None):
99 """Initialize a brand new merge state, removing any existing state on
99 """Initialize a brand new merge state, removing any existing state on
100 disk."""
100 disk."""
101 ms = mergestate(repo)
101 ms = mergestate(repo)
102 ms.reset(node, other, labels)
102 ms.reset(node, other, labels)
103 return ms
103 return ms
104
104
105 @staticmethod
105 @staticmethod
106 def read(repo):
106 def read(repo):
107 """Initialize the merge state, reading it from disk."""
107 """Initialize the merge state, reading it from disk."""
108 ms = mergestate(repo)
108 ms = mergestate(repo)
109 ms._read()
109 ms._read()
110 return ms
110 return ms
111
111
112 def __init__(self, repo):
112 def __init__(self, repo):
113 """Initialize the merge state.
113 """Initialize the merge state.
114
114
115 Do not use this directly! Instead call read() or clean()."""
115 Do not use this directly! Instead call read() or clean()."""
116 self._repo = repo
116 self._repo = repo
117 self._dirty = False
117 self._dirty = False
118 self._labels = None
118 self._labels = None
119
119
120 def reset(self, node=None, other=None, labels=None):
120 def reset(self, node=None, other=None, labels=None):
121 self._state = {}
121 self._state = {}
122 self._stateextras = {}
122 self._stateextras = {}
123 self._local = None
123 self._local = None
124 self._other = None
124 self._other = None
125 self._labels = labels
125 self._labels = labels
126 for var in ('localctx', 'otherctx'):
126 for var in ('localctx', 'otherctx'):
127 if var in vars(self):
127 if var in vars(self):
128 delattr(self, var)
128 delattr(self, var)
129 if node:
129 if node:
130 self._local = node
130 self._local = node
131 self._other = other
131 self._other = other
132 self._readmergedriver = None
132 self._readmergedriver = None
133 if self.mergedriver:
133 if self.mergedriver:
134 self._mdstate = 's'
134 self._mdstate = 's'
135 else:
135 else:
136 self._mdstate = 'u'
136 self._mdstate = 'u'
137 shutil.rmtree(self._repo.vfs.join('merge'), True)
137 shutil.rmtree(self._repo.vfs.join('merge'), True)
138 self._results = {}
138 self._results = {}
139 self._dirty = False
139 self._dirty = False
140
140
141 def _read(self):
141 def _read(self):
142 """Analyse each record content to restore a serialized state from disk
142 """Analyse each record content to restore a serialized state from disk
143
143
144 This function process "record" entry produced by the de-serialization
144 This function process "record" entry produced by the de-serialization
145 of on disk file.
145 of on disk file.
146 """
146 """
147 self._state = {}
147 self._state = {}
148 self._stateextras = {}
148 self._stateextras = {}
149 self._local = None
149 self._local = None
150 self._other = None
150 self._other = None
151 for var in ('localctx', 'otherctx'):
151 for var in ('localctx', 'otherctx'):
152 if var in vars(self):
152 if var in vars(self):
153 delattr(self, var)
153 delattr(self, var)
154 self._readmergedriver = None
154 self._readmergedriver = None
155 self._mdstate = 's'
155 self._mdstate = 's'
156 unsupported = set()
156 unsupported = set()
157 records = self._readrecords()
157 records = self._readrecords()
158 for rtype, record in records:
158 for rtype, record in records:
159 if rtype == 'L':
159 if rtype == 'L':
160 self._local = bin(record)
160 self._local = bin(record)
161 elif rtype == 'O':
161 elif rtype == 'O':
162 self._other = bin(record)
162 self._other = bin(record)
163 elif rtype == 'm':
163 elif rtype == 'm':
164 bits = record.split('\0', 1)
164 bits = record.split('\0', 1)
165 mdstate = bits[1]
165 mdstate = bits[1]
166 if len(mdstate) != 1 or mdstate not in 'ums':
166 if len(mdstate) != 1 or mdstate not in 'ums':
167 # the merge driver should be idempotent, so just rerun it
167 # the merge driver should be idempotent, so just rerun it
168 mdstate = 'u'
168 mdstate = 'u'
169
169
170 self._readmergedriver = bits[0]
170 self._readmergedriver = bits[0]
171 self._mdstate = mdstate
171 self._mdstate = mdstate
172 elif rtype in 'FDCP':
172 elif rtype in 'FDCP':
173 bits = record.split('\0')
173 bits = record.split('\0')
174 self._state[bits[0]] = bits[1:]
174 self._state[bits[0]] = bits[1:]
175 elif rtype == 'f':
175 elif rtype == 'f':
176 filename, rawextras = record.split('\0', 1)
176 filename, rawextras = record.split('\0', 1)
177 extraparts = rawextras.split('\0')
177 extraparts = rawextras.split('\0')
178 extras = {}
178 extras = {}
179 i = 0
179 i = 0
180 while i < len(extraparts):
180 while i < len(extraparts):
181 extras[extraparts[i]] = extraparts[i + 1]
181 extras[extraparts[i]] = extraparts[i + 1]
182 i += 2
182 i += 2
183
183
184 self._stateextras[filename] = extras
184 self._stateextras[filename] = extras
185 elif rtype == 'l':
185 elif rtype == 'l':
186 labels = record.split('\0', 2)
186 labels = record.split('\0', 2)
187 self._labels = [l for l in labels if len(l) > 0]
187 self._labels = [l for l in labels if len(l) > 0]
188 elif not rtype.islower():
188 elif not rtype.islower():
189 unsupported.add(rtype)
189 unsupported.add(rtype)
190 self._results = {}
190 self._results = {}
191 self._dirty = False
191 self._dirty = False
192
192
193 if unsupported:
193 if unsupported:
194 raise error.UnsupportedMergeRecords(unsupported)
194 raise error.UnsupportedMergeRecords(unsupported)
195
195
196 def _readrecords(self):
196 def _readrecords(self):
197 """Read merge state from disk and return a list of record (TYPE, data)
197 """Read merge state from disk and return a list of record (TYPE, data)
198
198
199 We read data from both v1 and v2 files and decide which one to use.
199 We read data from both v1 and v2 files and decide which one to use.
200
200
201 V1 has been used by version prior to 2.9.1 and contains less data than
201 V1 has been used by version prior to 2.9.1 and contains less data than
202 v2. We read both versions and check if no data in v2 contradicts
202 v2. We read both versions and check if no data in v2 contradicts
203 v1. If there is not contradiction we can safely assume that both v1
203 v1. If there is not contradiction we can safely assume that both v1
204 and v2 were written at the same time and use the extract data in v2. If
204 and v2 were written at the same time and use the extract data in v2. If
205 there is contradiction we ignore v2 content as we assume an old version
205 there is contradiction we ignore v2 content as we assume an old version
206 of Mercurial has overwritten the mergestate file and left an old v2
206 of Mercurial has overwritten the mergestate file and left an old v2
207 file around.
207 file around.
208
208
209 returns list of record [(TYPE, data), ...]"""
209 returns list of record [(TYPE, data), ...]"""
210 v1records = self._readrecordsv1()
210 v1records = self._readrecordsv1()
211 v2records = self._readrecordsv2()
211 v2records = self._readrecordsv2()
212 if self._v1v2match(v1records, v2records):
212 if self._v1v2match(v1records, v2records):
213 return v2records
213 return v2records
214 else:
214 else:
215 # v1 file is newer than v2 file, use it
215 # v1 file is newer than v2 file, use it
216 # we have to infer the "other" changeset of the merge
216 # we have to infer the "other" changeset of the merge
217 # we cannot do better than that with v1 of the format
217 # we cannot do better than that with v1 of the format
218 mctx = self._repo[None].parents()[-1]
218 mctx = self._repo[None].parents()[-1]
219 v1records.append(('O', mctx.hex()))
219 v1records.append(('O', mctx.hex()))
220 # add place holder "other" file node information
220 # add place holder "other" file node information
221 # nobody is using it yet so we do no need to fetch the data
221 # nobody is using it yet so we do no need to fetch the data
222 # if mctx was wrong `mctx[bits[-2]]` may fails.
222 # if mctx was wrong `mctx[bits[-2]]` may fails.
223 for idx, r in enumerate(v1records):
223 for idx, r in enumerate(v1records):
224 if r[0] == 'F':
224 if r[0] == 'F':
225 bits = r[1].split('\0')
225 bits = r[1].split('\0')
226 bits.insert(-2, '')
226 bits.insert(-2, '')
227 v1records[idx] = (r[0], '\0'.join(bits))
227 v1records[idx] = (r[0], '\0'.join(bits))
228 return v1records
228 return v1records
229
229
230 def _v1v2match(self, v1records, v2records):
230 def _v1v2match(self, v1records, v2records):
231 oldv2 = set() # old format version of v2 record
231 oldv2 = set() # old format version of v2 record
232 for rec in v2records:
232 for rec in v2records:
233 if rec[0] == 'L':
233 if rec[0] == 'L':
234 oldv2.add(rec)
234 oldv2.add(rec)
235 elif rec[0] == 'F':
235 elif rec[0] == 'F':
236 # drop the onode data (not contained in v1)
236 # drop the onode data (not contained in v1)
237 oldv2.add(('F', _droponode(rec[1])))
237 oldv2.add(('F', _droponode(rec[1])))
238 for rec in v1records:
238 for rec in v1records:
239 if rec not in oldv2:
239 if rec not in oldv2:
240 return False
240 return False
241 else:
241 else:
242 return True
242 return True
243
243
244 def _readrecordsv1(self):
244 def _readrecordsv1(self):
245 """read on disk merge state for version 1 file
245 """read on disk merge state for version 1 file
246
246
247 returns list of record [(TYPE, data), ...]
247 returns list of record [(TYPE, data), ...]
248
248
249 Note: the "F" data from this file are one entry short
249 Note: the "F" data from this file are one entry short
250 (no "other file node" entry)
250 (no "other file node" entry)
251 """
251 """
252 records = []
252 records = []
253 try:
253 try:
254 f = self._repo.vfs(self.statepathv1)
254 f = self._repo.vfs(self.statepathv1)
255 for i, l in enumerate(f):
255 for i, l in enumerate(f):
256 if i == 0:
256 if i == 0:
257 records.append(('L', l[:-1]))
257 records.append(('L', l[:-1]))
258 else:
258 else:
259 records.append(('F', l[:-1]))
259 records.append(('F', l[:-1]))
260 f.close()
260 f.close()
261 except IOError as err:
261 except IOError as err:
262 if err.errno != errno.ENOENT:
262 if err.errno != errno.ENOENT:
263 raise
263 raise
264 return records
264 return records
265
265
266 def _readrecordsv2(self):
266 def _readrecordsv2(self):
267 """read on disk merge state for version 2 file
267 """read on disk merge state for version 2 file
268
268
269 This format is a list of arbitrary records of the form:
269 This format is a list of arbitrary records of the form:
270
270
271 [type][length][content]
271 [type][length][content]
272
272
273 `type` is a single character, `length` is a 4 byte integer, and
273 `type` is a single character, `length` is a 4 byte integer, and
274 `content` is an arbitrary byte sequence of length `length`.
274 `content` is an arbitrary byte sequence of length `length`.
275
275
276 Mercurial versions prior to 3.7 have a bug where if there are
276 Mercurial versions prior to 3.7 have a bug where if there are
277 unsupported mandatory merge records, attempting to clear out the merge
277 unsupported mandatory merge records, attempting to clear out the merge
278 state with hg update --clean or similar aborts. The 't' record type
278 state with hg update --clean or similar aborts. The 't' record type
279 works around that by writing out what those versions treat as an
279 works around that by writing out what those versions treat as an
280 advisory record, but later versions interpret as special: the first
280 advisory record, but later versions interpret as special: the first
281 character is the 'real' record type and everything onwards is the data.
281 character is the 'real' record type and everything onwards is the data.
282
282
283 Returns list of records [(TYPE, data), ...]."""
283 Returns list of records [(TYPE, data), ...]."""
284 records = []
284 records = []
285 try:
285 try:
286 f = self._repo.vfs(self.statepathv2)
286 f = self._repo.vfs(self.statepathv2)
287 data = f.read()
287 data = f.read()
288 off = 0
288 off = 0
289 end = len(data)
289 end = len(data)
290 while off < end:
290 while off < end:
291 rtype = data[off]
291 rtype = data[off]
292 off += 1
292 off += 1
293 length = _unpack('>I', data[off:(off + 4)])[0]
293 length = _unpack('>I', data[off:(off + 4)])[0]
294 off += 4
294 off += 4
295 record = data[off:(off + length)]
295 record = data[off:(off + length)]
296 off += length
296 off += length
297 if rtype == 't':
297 if rtype == 't':
298 rtype, record = record[0], record[1:]
298 rtype, record = record[0], record[1:]
299 records.append((rtype, record))
299 records.append((rtype, record))
300 f.close()
300 f.close()
301 except IOError as err:
301 except IOError as err:
302 if err.errno != errno.ENOENT:
302 if err.errno != errno.ENOENT:
303 raise
303 raise
304 return records
304 return records
305
305
306 @util.propertycache
306 @util.propertycache
307 def mergedriver(self):
307 def mergedriver(self):
308 # protect against the following:
308 # protect against the following:
309 # - A configures a malicious merge driver in their hgrc, then
309 # - A configures a malicious merge driver in their hgrc, then
310 # pauses the merge
310 # pauses the merge
311 # - A edits their hgrc to remove references to the merge driver
311 # - A edits their hgrc to remove references to the merge driver
312 # - A gives a copy of their entire repo, including .hg, to B
312 # - A gives a copy of their entire repo, including .hg, to B
313 # - B inspects .hgrc and finds it to be clean
313 # - B inspects .hgrc and finds it to be clean
314 # - B then continues the merge and the malicious merge driver
314 # - B then continues the merge and the malicious merge driver
315 # gets invoked
315 # gets invoked
316 configmergedriver = self._repo.ui.config('experimental', 'mergedriver')
316 configmergedriver = self._repo.ui.config('experimental', 'mergedriver')
317 if (self._readmergedriver is not None
317 if (self._readmergedriver is not None
318 and self._readmergedriver != configmergedriver):
318 and self._readmergedriver != configmergedriver):
319 raise error.ConfigError(
319 raise error.ConfigError(
320 _("merge driver changed since merge started"),
320 _("merge driver changed since merge started"),
321 hint=_("revert merge driver change or abort merge"))
321 hint=_("revert merge driver change or abort merge"))
322
322
323 return configmergedriver
323 return configmergedriver
324
324
325 @util.propertycache
325 @util.propertycache
326 def localctx(self):
326 def localctx(self):
327 if self._local is None:
327 if self._local is None:
328 msg = "localctx accessed but self._local isn't set"
328 msg = "localctx accessed but self._local isn't set"
329 raise error.ProgrammingError(msg)
329 raise error.ProgrammingError(msg)
330 return self._repo[self._local]
330 return self._repo[self._local]
331
331
332 @util.propertycache
332 @util.propertycache
333 def otherctx(self):
333 def otherctx(self):
334 if self._other is None:
334 if self._other is None:
335 msg = "otherctx accessed but self._other isn't set"
335 msg = "otherctx accessed but self._other isn't set"
336 raise error.ProgrammingError(msg)
336 raise error.ProgrammingError(msg)
337 return self._repo[self._other]
337 return self._repo[self._other]
338
338
339 def active(self):
339 def active(self):
340 """Whether mergestate is active.
340 """Whether mergestate is active.
341
341
342 Returns True if there appears to be mergestate. This is a rough proxy
342 Returns True if there appears to be mergestate. This is a rough proxy
343 for "is a merge in progress."
343 for "is a merge in progress."
344 """
344 """
345 # Check local variables before looking at filesystem for performance
345 # Check local variables before looking at filesystem for performance
346 # reasons.
346 # reasons.
347 return bool(self._local) or bool(self._state) or \
347 return bool(self._local) or bool(self._state) or \
348 self._repo.vfs.exists(self.statepathv1) or \
348 self._repo.vfs.exists(self.statepathv1) or \
349 self._repo.vfs.exists(self.statepathv2)
349 self._repo.vfs.exists(self.statepathv2)
350
350
351 def commit(self):
351 def commit(self):
352 """Write current state on disk (if necessary)"""
352 """Write current state on disk (if necessary)"""
353 if self._dirty:
353 if self._dirty:
354 records = self._makerecords()
354 records = self._makerecords()
355 self._writerecords(records)
355 self._writerecords(records)
356 self._dirty = False
356 self._dirty = False
357
357
358 def _makerecords(self):
358 def _makerecords(self):
359 records = []
359 records = []
360 records.append(('L', hex(self._local)))
360 records.append(('L', hex(self._local)))
361 records.append(('O', hex(self._other)))
361 records.append(('O', hex(self._other)))
362 if self.mergedriver:
362 if self.mergedriver:
363 records.append(('m', '\0'.join([
363 records.append(('m', '\0'.join([
364 self.mergedriver, self._mdstate])))
364 self.mergedriver, self._mdstate])))
365 # Write out state items. In all cases, the value of the state map entry
365 # Write out state items. In all cases, the value of the state map entry
366 # is written as the contents of the record. The record type depends on
366 # is written as the contents of the record. The record type depends on
367 # the type of state that is stored, and capital-letter records are used
367 # the type of state that is stored, and capital-letter records are used
368 # to prevent older versions of Mercurial that do not support the feature
368 # to prevent older versions of Mercurial that do not support the feature
369 # from loading them.
369 # from loading them.
370 for filename, v in self._state.iteritems():
370 for filename, v in self._state.iteritems():
371 if v[0] == 'd':
371 if v[0] == 'd':
372 # Driver-resolved merge. These are stored in 'D' records.
372 # Driver-resolved merge. These are stored in 'D' records.
373 records.append(('D', '\0'.join([filename] + v)))
373 records.append(('D', '\0'.join([filename] + v)))
374 elif v[0] in ('pu', 'pr'):
374 elif v[0] in ('pu', 'pr'):
375 # Path conflicts. These are stored in 'P' records. The current
375 # Path conflicts. These are stored in 'P' records. The current
376 # resolution state ('pu' or 'pr') is stored within the record.
376 # resolution state ('pu' or 'pr') is stored within the record.
377 records.append(('P', '\0'.join([filename] + v)))
377 records.append(('P', '\0'.join([filename] + v)))
378 elif v[1] == nullhex or v[6] == nullhex:
378 elif v[1] == nullhex or v[6] == nullhex:
379 # Change/Delete or Delete/Change conflicts. These are stored in
379 # Change/Delete or Delete/Change conflicts. These are stored in
380 # 'C' records. v[1] is the local file, and is nullhex when the
380 # 'C' records. v[1] is the local file, and is nullhex when the
381 # file is deleted locally ('dc'). v[6] is the remote file, and
381 # file is deleted locally ('dc'). v[6] is the remote file, and
382 # is nullhex when the file is deleted remotely ('cd').
382 # is nullhex when the file is deleted remotely ('cd').
383 records.append(('C', '\0'.join([filename] + v)))
383 records.append(('C', '\0'.join([filename] + v)))
384 else:
384 else:
385 # Normal files. These are stored in 'F' records.
385 # Normal files. These are stored in 'F' records.
386 records.append(('F', '\0'.join([filename] + v)))
386 records.append(('F', '\0'.join([filename] + v)))
387 for filename, extras in sorted(self._stateextras.iteritems()):
387 for filename, extras in sorted(self._stateextras.iteritems()):
388 rawextras = '\0'.join('%s\0%s' % (k, v) for k, v in
388 rawextras = '\0'.join('%s\0%s' % (k, v) for k, v in
389 extras.iteritems())
389 extras.iteritems())
390 records.append(('f', '%s\0%s' % (filename, rawextras)))
390 records.append(('f', '%s\0%s' % (filename, rawextras)))
391 if self._labels is not None:
391 if self._labels is not None:
392 labels = '\0'.join(self._labels)
392 labels = '\0'.join(self._labels)
393 records.append(('l', labels))
393 records.append(('l', labels))
394 return records
394 return records
395
395
396 def _writerecords(self, records):
396 def _writerecords(self, records):
397 """Write current state on disk (both v1 and v2)"""
397 """Write current state on disk (both v1 and v2)"""
398 self._writerecordsv1(records)
398 self._writerecordsv1(records)
399 self._writerecordsv2(records)
399 self._writerecordsv2(records)
400
400
401 def _writerecordsv1(self, records):
401 def _writerecordsv1(self, records):
402 """Write current state on disk in a version 1 file"""
402 """Write current state on disk in a version 1 file"""
403 f = self._repo.vfs(self.statepathv1, 'w')
403 f = self._repo.vfs(self.statepathv1, 'w')
404 irecords = iter(records)
404 irecords = iter(records)
405 lrecords = next(irecords)
405 lrecords = next(irecords)
406 assert lrecords[0] == 'L'
406 assert lrecords[0] == 'L'
407 f.write(hex(self._local) + '\n')
407 f.write(hex(self._local) + '\n')
408 for rtype, data in irecords:
408 for rtype, data in irecords:
409 if rtype == 'F':
409 if rtype == 'F':
410 f.write('%s\n' % _droponode(data))
410 f.write('%s\n' % _droponode(data))
411 f.close()
411 f.close()
412
412
413 def _writerecordsv2(self, records):
413 def _writerecordsv2(self, records):
414 """Write current state on disk in a version 2 file
414 """Write current state on disk in a version 2 file
415
415
416 See the docstring for _readrecordsv2 for why we use 't'."""
416 See the docstring for _readrecordsv2 for why we use 't'."""
417 # these are the records that all version 2 clients can read
417 # these are the records that all version 2 clients can read
418 whitelist = 'LOF'
418 whitelist = 'LOF'
419 f = self._repo.vfs(self.statepathv2, 'w')
419 f = self._repo.vfs(self.statepathv2, 'w')
420 for key, data in records:
420 for key, data in records:
421 assert len(key) == 1
421 assert len(key) == 1
422 if key not in whitelist:
422 if key not in whitelist:
423 key, data = 't', '%s%s' % (key, data)
423 key, data = 't', '%s%s' % (key, data)
424 format = '>sI%is' % len(data)
424 format = '>sI%is' % len(data)
425 f.write(_pack(format, key, len(data), data))
425 f.write(_pack(format, key, len(data), data))
426 f.close()
426 f.close()
427
427
428 def add(self, fcl, fco, fca, fd):
428 def add(self, fcl, fco, fca, fd):
429 """add a new (potentially?) conflicting file the merge state
429 """add a new (potentially?) conflicting file the merge state
430 fcl: file context for local,
430 fcl: file context for local,
431 fco: file context for remote,
431 fco: file context for remote,
432 fca: file context for ancestors,
432 fca: file context for ancestors,
433 fd: file path of the resulting merge.
433 fd: file path of the resulting merge.
434
434
435 note: also write the local version to the `.hg/merge` directory.
435 note: also write the local version to the `.hg/merge` directory.
436 """
436 """
437 if fcl.isabsent():
437 if fcl.isabsent():
438 hash = nullhex
438 hash = nullhex
439 else:
439 else:
440 hash = hex(hashlib.sha1(fcl.path()).digest())
440 hash = hex(hashlib.sha1(fcl.path()).digest())
441 self._repo.vfs.write('merge/' + hash, fcl.data())
441 self._repo.vfs.write('merge/' + hash, fcl.data())
442 self._state[fd] = ['u', hash, fcl.path(),
442 self._state[fd] = ['u', hash, fcl.path(),
443 fca.path(), hex(fca.filenode()),
443 fca.path(), hex(fca.filenode()),
444 fco.path(), hex(fco.filenode()),
444 fco.path(), hex(fco.filenode()),
445 fcl.flags()]
445 fcl.flags()]
446 self._stateextras[fd] = {'ancestorlinknode': hex(fca.node())}
446 self._stateextras[fd] = {'ancestorlinknode': hex(fca.node())}
447 self._dirty = True
447 self._dirty = True
448
448
449 def addpath(self, path, frename, forigin):
449 def addpath(self, path, frename, forigin):
450 """add a new conflicting path to the merge state
450 """add a new conflicting path to the merge state
451 path: the path that conflicts
451 path: the path that conflicts
452 frename: the filename the conflicting file was renamed to
452 frename: the filename the conflicting file was renamed to
453 forigin: origin of the file ('l' or 'r' for local/remote)
453 forigin: origin of the file ('l' or 'r' for local/remote)
454 """
454 """
455 self._state[path] = ['pu', frename, forigin]
455 self._state[path] = ['pu', frename, forigin]
456 self._dirty = True
456 self._dirty = True
457
457
458 def __contains__(self, dfile):
458 def __contains__(self, dfile):
459 return dfile in self._state
459 return dfile in self._state
460
460
461 def __getitem__(self, dfile):
461 def __getitem__(self, dfile):
462 return self._state[dfile][0]
462 return self._state[dfile][0]
463
463
464 def __iter__(self):
464 def __iter__(self):
465 return iter(sorted(self._state))
465 return iter(sorted(self._state))
466
466
467 def files(self):
467 def files(self):
468 return self._state.keys()
468 return self._state.keys()
469
469
470 def mark(self, dfile, state):
470 def mark(self, dfile, state):
471 self._state[dfile][0] = state
471 self._state[dfile][0] = state
472 self._dirty = True
472 self._dirty = True
473
473
474 def mdstate(self):
474 def mdstate(self):
475 return self._mdstate
475 return self._mdstate
476
476
477 def unresolved(self):
477 def unresolved(self):
478 """Obtain the paths of unresolved files."""
478 """Obtain the paths of unresolved files."""
479
479
480 for f, entry in self._state.iteritems():
480 for f, entry in self._state.iteritems():
481 if entry[0] in ('u', 'pu'):
481 if entry[0] in ('u', 'pu'):
482 yield f
482 yield f
483
483
484 def driverresolved(self):
484 def driverresolved(self):
485 """Obtain the paths of driver-resolved files."""
485 """Obtain the paths of driver-resolved files."""
486
486
487 for f, entry in self._state.items():
487 for f, entry in self._state.items():
488 if entry[0] == 'd':
488 if entry[0] == 'd':
489 yield f
489 yield f
490
490
491 def extras(self, filename):
491 def extras(self, filename):
492 return self._stateextras.setdefault(filename, {})
492 return self._stateextras.setdefault(filename, {})
493
493
494 def _resolve(self, preresolve, dfile, wctx):
494 def _resolve(self, preresolve, dfile, wctx):
495 """rerun merge process for file path `dfile`"""
495 """rerun merge process for file path `dfile`"""
496 if self[dfile] in 'rd':
496 if self[dfile] in 'rd':
497 return True, 0
497 return True, 0
498 stateentry = self._state[dfile]
498 stateentry = self._state[dfile]
499 state, hash, lfile, afile, anode, ofile, onode, flags = stateentry
499 state, hash, lfile, afile, anode, ofile, onode, flags = stateentry
500 octx = self._repo[self._other]
500 octx = self._repo[self._other]
501 extras = self.extras(dfile)
501 extras = self.extras(dfile)
502 anccommitnode = extras.get('ancestorlinknode')
502 anccommitnode = extras.get('ancestorlinknode')
503 if anccommitnode:
503 if anccommitnode:
504 actx = self._repo[anccommitnode]
504 actx = self._repo[anccommitnode]
505 else:
505 else:
506 actx = None
506 actx = None
507 fcd = self._filectxorabsent(hash, wctx, dfile)
507 fcd = self._filectxorabsent(hash, wctx, dfile)
508 fco = self._filectxorabsent(onode, octx, ofile)
508 fco = self._filectxorabsent(onode, octx, ofile)
509 # TODO: move this to filectxorabsent
509 # TODO: move this to filectxorabsent
510 fca = self._repo.filectx(afile, fileid=anode, changeid=actx)
510 fca = self._repo.filectx(afile, fileid=anode, changeid=actx)
511 # "premerge" x flags
511 # "premerge" x flags
512 flo = fco.flags()
512 flo = fco.flags()
513 fla = fca.flags()
513 fla = fca.flags()
514 if 'x' in flags + flo + fla and 'l' not in flags + flo + fla:
514 if 'x' in flags + flo + fla and 'l' not in flags + flo + fla:
515 if fca.node() == nullid and flags != flo:
515 if fca.node() == nullid and flags != flo:
516 if preresolve:
516 if preresolve:
517 self._repo.ui.warn(
517 self._repo.ui.warn(
518 _('warning: cannot merge flags for %s '
518 _('warning: cannot merge flags for %s '
519 'without common ancestor - keeping local flags\n')
519 'without common ancestor - keeping local flags\n')
520 % afile)
520 % afile)
521 elif flags == fla:
521 elif flags == fla:
522 flags = flo
522 flags = flo
523 if preresolve:
523 if preresolve:
524 # restore local
524 # restore local
525 if hash != nullhex:
525 if hash != nullhex:
526 f = self._repo.vfs('merge/' + hash)
526 f = self._repo.vfs('merge/' + hash)
527 wctx[dfile].write(f.read(), flags)
527 wctx[dfile].write(f.read(), flags)
528 f.close()
528 f.close()
529 else:
529 else:
530 wctx[dfile].remove(ignoremissing=True)
530 wctx[dfile].remove(ignoremissing=True)
531 complete, r, deleted = filemerge.premerge(self._repo, wctx,
531 complete, r, deleted = filemerge.premerge(self._repo, wctx,
532 self._local, lfile, fcd,
532 self._local, lfile, fcd,
533 fco, fca,
533 fco, fca,
534 labels=self._labels)
534 labels=self._labels)
535 else:
535 else:
536 complete, r, deleted = filemerge.filemerge(self._repo, wctx,
536 complete, r, deleted = filemerge.filemerge(self._repo, wctx,
537 self._local, lfile, fcd,
537 self._local, lfile, fcd,
538 fco, fca,
538 fco, fca,
539 labels=self._labels)
539 labels=self._labels)
540 if r is None:
540 if r is None:
541 # no real conflict
541 # no real conflict
542 del self._state[dfile]
542 del self._state[dfile]
543 self._stateextras.pop(dfile, None)
543 self._stateextras.pop(dfile, None)
544 self._dirty = True
544 self._dirty = True
545 elif not r:
545 elif not r:
546 self.mark(dfile, 'r')
546 self.mark(dfile, 'r')
547
547
548 if complete:
548 if complete:
549 action = None
549 action = None
550 if deleted:
550 if deleted:
551 if fcd.isabsent():
551 if fcd.isabsent():
552 # dc: local picked. Need to drop if present, which may
552 # dc: local picked. Need to drop if present, which may
553 # happen on re-resolves.
553 # happen on re-resolves.
554 action = 'f'
554 action = 'f'
555 else:
555 else:
556 # cd: remote picked (or otherwise deleted)
556 # cd: remote picked (or otherwise deleted)
557 action = 'r'
557 action = 'r'
558 else:
558 else:
559 if fcd.isabsent(): # dc: remote picked
559 if fcd.isabsent(): # dc: remote picked
560 action = 'g'
560 action = 'g'
561 elif fco.isabsent(): # cd: local picked
561 elif fco.isabsent(): # cd: local picked
562 if dfile in self.localctx:
562 if dfile in self.localctx:
563 action = 'am'
563 action = 'am'
564 else:
564 else:
565 action = 'a'
565 action = 'a'
566 # else: regular merges (no action necessary)
566 # else: regular merges (no action necessary)
567 self._results[dfile] = r, action
567 self._results[dfile] = r, action
568
568
569 return complete, r
569 return complete, r
570
570
571 def _filectxorabsent(self, hexnode, ctx, f):
571 def _filectxorabsent(self, hexnode, ctx, f):
572 if hexnode == nullhex:
572 if hexnode == nullhex:
573 return filemerge.absentfilectx(ctx, f)
573 return filemerge.absentfilectx(ctx, f)
574 else:
574 else:
575 return ctx[f]
575 return ctx[f]
576
576
577 def preresolve(self, dfile, wctx):
577 def preresolve(self, dfile, wctx):
578 """run premerge process for dfile
578 """run premerge process for dfile
579
579
580 Returns whether the merge is complete, and the exit code."""
580 Returns whether the merge is complete, and the exit code."""
581 return self._resolve(True, dfile, wctx)
581 return self._resolve(True, dfile, wctx)
582
582
583 def resolve(self, dfile, wctx):
583 def resolve(self, dfile, wctx):
584 """run merge process (assuming premerge was run) for dfile
584 """run merge process (assuming premerge was run) for dfile
585
585
586 Returns the exit code of the merge."""
586 Returns the exit code of the merge."""
587 return self._resolve(False, dfile, wctx)[1]
587 return self._resolve(False, dfile, wctx)[1]
588
588
589 def counts(self):
589 def counts(self):
590 """return counts for updated, merged and removed files in this
590 """return counts for updated, merged and removed files in this
591 session"""
591 session"""
592 updated, merged, removed = 0, 0, 0
592 updated, merged, removed = 0, 0, 0
593 for r, action in self._results.itervalues():
593 for r, action in self._results.itervalues():
594 if r is None:
594 if r is None:
595 updated += 1
595 updated += 1
596 elif r == 0:
596 elif r == 0:
597 if action == 'r':
597 if action == 'r':
598 removed += 1
598 removed += 1
599 else:
599 else:
600 merged += 1
600 merged += 1
601 return updated, merged, removed
601 return updated, merged, removed
602
602
603 def unresolvedcount(self):
603 def unresolvedcount(self):
604 """get unresolved count for this merge (persistent)"""
604 """get unresolved count for this merge (persistent)"""
605 return len(list(self.unresolved()))
605 return len(list(self.unresolved()))
606
606
607 def actions(self):
607 def actions(self):
608 """return lists of actions to perform on the dirstate"""
608 """return lists of actions to perform on the dirstate"""
609 actions = {'r': [], 'f': [], 'a': [], 'am': [], 'g': []}
609 actions = {'r': [], 'f': [], 'a': [], 'am': [], 'g': []}
610 for f, (r, action) in self._results.iteritems():
610 for f, (r, action) in self._results.iteritems():
611 if action is not None:
611 if action is not None:
612 actions[action].append((f, None, "merge result"))
612 actions[action].append((f, None, "merge result"))
613 return actions
613 return actions
614
614
615 def recordactions(self):
615 def recordactions(self):
616 """record remove/add/get actions in the dirstate"""
616 """record remove/add/get actions in the dirstate"""
617 branchmerge = self._repo.dirstate.p2() != nullid
617 branchmerge = self._repo.dirstate.p2() != nullid
618 recordupdates(self._repo, self.actions(), branchmerge)
618 recordupdates(self._repo, self.actions(), branchmerge)
619
619
620 def queueremove(self, f):
620 def queueremove(self, f):
621 """queues a file to be removed from the dirstate
621 """queues a file to be removed from the dirstate
622
622
623 Meant for use by custom merge drivers."""
623 Meant for use by custom merge drivers."""
624 self._results[f] = 0, 'r'
624 self._results[f] = 0, 'r'
625
625
626 def queueadd(self, f):
626 def queueadd(self, f):
627 """queues a file to be added to the dirstate
627 """queues a file to be added to the dirstate
628
628
629 Meant for use by custom merge drivers."""
629 Meant for use by custom merge drivers."""
630 self._results[f] = 0, 'a'
630 self._results[f] = 0, 'a'
631
631
632 def queueget(self, f):
632 def queueget(self, f):
633 """queues a file to be marked modified in the dirstate
633 """queues a file to be marked modified in the dirstate
634
634
635 Meant for use by custom merge drivers."""
635 Meant for use by custom merge drivers."""
636 self._results[f] = 0, 'g'
636 self._results[f] = 0, 'g'
637
637
638 def _getcheckunknownconfig(repo, section, name):
638 def _getcheckunknownconfig(repo, section, name):
639 config = repo.ui.config(section, name)
639 config = repo.ui.config(section, name)
640 valid = ['abort', 'ignore', 'warn']
640 valid = ['abort', 'ignore', 'warn']
641 if config not in valid:
641 if config not in valid:
642 validstr = ', '.join(["'" + v + "'" for v in valid])
642 validstr = ', '.join(["'" + v + "'" for v in valid])
643 raise error.ConfigError(_("%s.%s not valid "
643 raise error.ConfigError(_("%s.%s not valid "
644 "('%s' is none of %s)")
644 "('%s' is none of %s)")
645 % (section, name, config, validstr))
645 % (section, name, config, validstr))
646 return config
646 return config
647
647
648 def _checkunknownfile(repo, wctx, mctx, f, f2=None):
648 def _checkunknownfile(repo, wctx, mctx, f, f2=None):
649 if wctx.isinmemory():
650 # Nothing to do in IMM because nothing in the "working copy" can be an
651 # unknown file.
652 #
653 # Note that we should bail out here, not in ``_checkunknownfiles()``,
654 # because that function does other useful work.
655 return False
656
649 if f2 is None:
657 if f2 is None:
650 f2 = f
658 f2 = f
651 return (repo.wvfs.audit.check(f)
659 return (repo.wvfs.audit.check(f)
652 and repo.wvfs.isfileorlink(f)
660 and repo.wvfs.isfileorlink(f)
653 and repo.dirstate.normalize(f) not in repo.dirstate
661 and repo.dirstate.normalize(f) not in repo.dirstate
654 and mctx[f2].cmp(wctx[f]))
662 and mctx[f2].cmp(wctx[f]))
655
663
656 class _unknowndirschecker(object):
664 class _unknowndirschecker(object):
657 """
665 """
658 Look for any unknown files or directories that may have a path conflict
666 Look for any unknown files or directories that may have a path conflict
659 with a file. If any path prefix of the file exists as a file or link,
667 with a file. If any path prefix of the file exists as a file or link,
660 then it conflicts. If the file itself is a directory that contains any
668 then it conflicts. If the file itself is a directory that contains any
661 file that is not tracked, then it conflicts.
669 file that is not tracked, then it conflicts.
662
670
663 Returns the shortest path at which a conflict occurs, or None if there is
671 Returns the shortest path at which a conflict occurs, or None if there is
664 no conflict.
672 no conflict.
665 """
673 """
666 def __init__(self):
674 def __init__(self):
667 # A set of paths known to be good. This prevents repeated checking of
675 # A set of paths known to be good. This prevents repeated checking of
668 # dirs. It will be updated with any new dirs that are checked and found
676 # dirs. It will be updated with any new dirs that are checked and found
669 # to be safe.
677 # to be safe.
670 self._unknowndircache = set()
678 self._unknowndircache = set()
671
679
672 # A set of paths that are known to be absent. This prevents repeated
680 # A set of paths that are known to be absent. This prevents repeated
673 # checking of subdirectories that are known not to exist. It will be
681 # checking of subdirectories that are known not to exist. It will be
674 # updated with any new dirs that are checked and found to be absent.
682 # updated with any new dirs that are checked and found to be absent.
675 self._missingdircache = set()
683 self._missingdircache = set()
676
684
677 def __call__(self, repo, f):
685 def __call__(self, repo, wctx, f):
686 if wctx.isinmemory():
687 # Nothing to do in IMM for the same reason as ``_checkunknownfile``.
688 return False
689
678 # Check for path prefixes that exist as unknown files.
690 # Check for path prefixes that exist as unknown files.
679 for p in reversed(list(util.finddirs(f))):
691 for p in reversed(list(util.finddirs(f))):
680 if p in self._missingdircache:
692 if p in self._missingdircache:
681 return
693 return
682 if p in self._unknowndircache:
694 if p in self._unknowndircache:
683 continue
695 continue
684 if repo.wvfs.audit.check(p):
696 if repo.wvfs.audit.check(p):
685 if (repo.wvfs.isfileorlink(p)
697 if (repo.wvfs.isfileorlink(p)
686 and repo.dirstate.normalize(p) not in repo.dirstate):
698 and repo.dirstate.normalize(p) not in repo.dirstate):
687 return p
699 return p
688 if not repo.wvfs.lexists(p):
700 if not repo.wvfs.lexists(p):
689 self._missingdircache.add(p)
701 self._missingdircache.add(p)
690 return
702 return
691 self._unknowndircache.add(p)
703 self._unknowndircache.add(p)
692
704
693 # Check if the file conflicts with a directory containing unknown files.
705 # Check if the file conflicts with a directory containing unknown files.
694 if repo.wvfs.audit.check(f) and repo.wvfs.isdir(f):
706 if repo.wvfs.audit.check(f) and repo.wvfs.isdir(f):
695 # Does the directory contain any files that are not in the dirstate?
707 # Does the directory contain any files that are not in the dirstate?
696 for p, dirs, files in repo.wvfs.walk(f):
708 for p, dirs, files in repo.wvfs.walk(f):
697 for fn in files:
709 for fn in files:
698 relf = repo.dirstate.normalize(repo.wvfs.reljoin(p, fn))
710 relf = repo.dirstate.normalize(repo.wvfs.reljoin(p, fn))
699 if relf not in repo.dirstate:
711 if relf not in repo.dirstate:
700 return f
712 return f
701 return None
713 return None
702
714
703 def _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce):
715 def _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce):
704 """
716 """
705 Considers any actions that care about the presence of conflicting unknown
717 Considers any actions that care about the presence of conflicting unknown
706 files. For some actions, the result is to abort; for others, it is to
718 files. For some actions, the result is to abort; for others, it is to
707 choose a different action.
719 choose a different action.
708 """
720 """
709 fileconflicts = set()
721 fileconflicts = set()
710 pathconflicts = set()
722 pathconflicts = set()
711 warnconflicts = set()
723 warnconflicts = set()
712 abortconflicts = set()
724 abortconflicts = set()
713 unknownconfig = _getcheckunknownconfig(repo, 'merge', 'checkunknown')
725 unknownconfig = _getcheckunknownconfig(repo, 'merge', 'checkunknown')
714 ignoredconfig = _getcheckunknownconfig(repo, 'merge', 'checkignored')
726 ignoredconfig = _getcheckunknownconfig(repo, 'merge', 'checkignored')
715 pathconfig = repo.ui.configbool('experimental', 'merge.checkpathconflicts')
727 pathconfig = repo.ui.configbool('experimental', 'merge.checkpathconflicts')
716 if not force:
728 if not force:
717 def collectconflicts(conflicts, config):
729 def collectconflicts(conflicts, config):
718 if config == 'abort':
730 if config == 'abort':
719 abortconflicts.update(conflicts)
731 abortconflicts.update(conflicts)
720 elif config == 'warn':
732 elif config == 'warn':
721 warnconflicts.update(conflicts)
733 warnconflicts.update(conflicts)
722
734
723 checkunknowndirs = _unknowndirschecker()
735 checkunknowndirs = _unknowndirschecker()
724 for f, (m, args, msg) in actions.iteritems():
736 for f, (m, args, msg) in actions.iteritems():
725 if m in ('c', 'dc'):
737 if m in ('c', 'dc'):
726 if _checkunknownfile(repo, wctx, mctx, f):
738 if _checkunknownfile(repo, wctx, mctx, f):
727 fileconflicts.add(f)
739 fileconflicts.add(f)
728 elif pathconfig and f not in wctx:
740 elif pathconfig and f not in wctx:
729 path = checkunknowndirs(repo, f)
741 path = checkunknowndirs(repo, wctx, f)
730 if path is not None:
742 if path is not None:
731 pathconflicts.add(path)
743 pathconflicts.add(path)
732 elif m == 'dg':
744 elif m == 'dg':
733 if _checkunknownfile(repo, wctx, mctx, f, args[0]):
745 if _checkunknownfile(repo, wctx, mctx, f, args[0]):
734 fileconflicts.add(f)
746 fileconflicts.add(f)
735
747
736 allconflicts = fileconflicts | pathconflicts
748 allconflicts = fileconflicts | pathconflicts
737 ignoredconflicts = set([c for c in allconflicts
749 ignoredconflicts = set([c for c in allconflicts
738 if repo.dirstate._ignore(c)])
750 if repo.dirstate._ignore(c)])
739 unknownconflicts = allconflicts - ignoredconflicts
751 unknownconflicts = allconflicts - ignoredconflicts
740 collectconflicts(ignoredconflicts, ignoredconfig)
752 collectconflicts(ignoredconflicts, ignoredconfig)
741 collectconflicts(unknownconflicts, unknownconfig)
753 collectconflicts(unknownconflicts, unknownconfig)
742 else:
754 else:
743 for f, (m, args, msg) in actions.iteritems():
755 for f, (m, args, msg) in actions.iteritems():
744 if m == 'cm':
756 if m == 'cm':
745 fl2, anc = args
757 fl2, anc = args
746 different = _checkunknownfile(repo, wctx, mctx, f)
758 different = _checkunknownfile(repo, wctx, mctx, f)
747 if repo.dirstate._ignore(f):
759 if repo.dirstate._ignore(f):
748 config = ignoredconfig
760 config = ignoredconfig
749 else:
761 else:
750 config = unknownconfig
762 config = unknownconfig
751
763
752 # The behavior when force is True is described by this table:
764 # The behavior when force is True is described by this table:
753 # config different mergeforce | action backup
765 # config different mergeforce | action backup
754 # * n * | get n
766 # * n * | get n
755 # * y y | merge -
767 # * y y | merge -
756 # abort y n | merge - (1)
768 # abort y n | merge - (1)
757 # warn y n | warn + get y
769 # warn y n | warn + get y
758 # ignore y n | get y
770 # ignore y n | get y
759 #
771 #
760 # (1) this is probably the wrong behavior here -- we should
772 # (1) this is probably the wrong behavior here -- we should
761 # probably abort, but some actions like rebases currently
773 # probably abort, but some actions like rebases currently
762 # don't like an abort happening in the middle of
774 # don't like an abort happening in the middle of
763 # merge.update.
775 # merge.update.
764 if not different:
776 if not different:
765 actions[f] = ('g', (fl2, False), "remote created")
777 actions[f] = ('g', (fl2, False), "remote created")
766 elif mergeforce or config == 'abort':
778 elif mergeforce or config == 'abort':
767 actions[f] = ('m', (f, f, None, False, anc),
779 actions[f] = ('m', (f, f, None, False, anc),
768 "remote differs from untracked local")
780 "remote differs from untracked local")
769 elif config == 'abort':
781 elif config == 'abort':
770 abortconflicts.add(f)
782 abortconflicts.add(f)
771 else:
783 else:
772 if config == 'warn':
784 if config == 'warn':
773 warnconflicts.add(f)
785 warnconflicts.add(f)
774 actions[f] = ('g', (fl2, True), "remote created")
786 actions[f] = ('g', (fl2, True), "remote created")
775
787
776 for f in sorted(abortconflicts):
788 for f in sorted(abortconflicts):
777 warn = repo.ui.warn
789 warn = repo.ui.warn
778 if f in pathconflicts:
790 if f in pathconflicts:
779 if repo.wvfs.isfileorlink(f):
791 if repo.wvfs.isfileorlink(f):
780 warn(_("%s: untracked file conflicts with directory\n") % f)
792 warn(_("%s: untracked file conflicts with directory\n") % f)
781 else:
793 else:
782 warn(_("%s: untracked directory conflicts with file\n") % f)
794 warn(_("%s: untracked directory conflicts with file\n") % f)
783 else:
795 else:
784 warn(_("%s: untracked file differs\n") % f)
796 warn(_("%s: untracked file differs\n") % f)
785 if abortconflicts:
797 if abortconflicts:
786 raise error.Abort(_("untracked files in working directory "
798 raise error.Abort(_("untracked files in working directory "
787 "differ from files in requested revision"))
799 "differ from files in requested revision"))
788
800
789 for f in sorted(warnconflicts):
801 for f in sorted(warnconflicts):
790 if repo.wvfs.isfileorlink(f):
802 if repo.wvfs.isfileorlink(f):
791 repo.ui.warn(_("%s: replacing untracked file\n") % f)
803 repo.ui.warn(_("%s: replacing untracked file\n") % f)
792 else:
804 else:
793 repo.ui.warn(_("%s: replacing untracked files in directory\n") % f)
805 repo.ui.warn(_("%s: replacing untracked files in directory\n") % f)
794
806
795 for f, (m, args, msg) in actions.iteritems():
807 for f, (m, args, msg) in actions.iteritems():
796 if m == 'c':
808 if m == 'c':
797 backup = (f in fileconflicts or f in pathconflicts or
809 backup = (f in fileconflicts or f in pathconflicts or
798 any(p in pathconflicts for p in util.finddirs(f)))
810 any(p in pathconflicts for p in util.finddirs(f)))
799 flags, = args
811 flags, = args
800 actions[f] = ('g', (flags, backup), msg)
812 actions[f] = ('g', (flags, backup), msg)
801
813
802 def _forgetremoved(wctx, mctx, branchmerge):
814 def _forgetremoved(wctx, mctx, branchmerge):
803 """
815 """
804 Forget removed files
816 Forget removed files
805
817
806 If we're jumping between revisions (as opposed to merging), and if
818 If we're jumping between revisions (as opposed to merging), and if
807 neither the working directory nor the target rev has the file,
819 neither the working directory nor the target rev has the file,
808 then we need to remove it from the dirstate, to prevent the
820 then we need to remove it from the dirstate, to prevent the
809 dirstate from listing the file when it is no longer in the
821 dirstate from listing the file when it is no longer in the
810 manifest.
822 manifest.
811
823
812 If we're merging, and the other revision has removed a file
824 If we're merging, and the other revision has removed a file
813 that is not present in the working directory, we need to mark it
825 that is not present in the working directory, we need to mark it
814 as removed.
826 as removed.
815 """
827 """
816
828
817 actions = {}
829 actions = {}
818 m = 'f'
830 m = 'f'
819 if branchmerge:
831 if branchmerge:
820 m = 'r'
832 m = 'r'
821 for f in wctx.deleted():
833 for f in wctx.deleted():
822 if f not in mctx:
834 if f not in mctx:
823 actions[f] = m, None, "forget deleted"
835 actions[f] = m, None, "forget deleted"
824
836
825 if not branchmerge:
837 if not branchmerge:
826 for f in wctx.removed():
838 for f in wctx.removed():
827 if f not in mctx:
839 if f not in mctx:
828 actions[f] = 'f', None, "forget removed"
840 actions[f] = 'f', None, "forget removed"
829
841
830 return actions
842 return actions
831
843
832 def _checkcollision(repo, wmf, actions):
844 def _checkcollision(repo, wmf, actions):
833 # build provisional merged manifest up
845 # build provisional merged manifest up
834 pmmf = set(wmf)
846 pmmf = set(wmf)
835
847
836 if actions:
848 if actions:
837 # k, dr, e and rd are no-op
849 # k, dr, e and rd are no-op
838 for m in 'a', 'am', 'f', 'g', 'cd', 'dc':
850 for m in 'a', 'am', 'f', 'g', 'cd', 'dc':
839 for f, args, msg in actions[m]:
851 for f, args, msg in actions[m]:
840 pmmf.add(f)
852 pmmf.add(f)
841 for f, args, msg in actions['r']:
853 for f, args, msg in actions['r']:
842 pmmf.discard(f)
854 pmmf.discard(f)
843 for f, args, msg in actions['dm']:
855 for f, args, msg in actions['dm']:
844 f2, flags = args
856 f2, flags = args
845 pmmf.discard(f2)
857 pmmf.discard(f2)
846 pmmf.add(f)
858 pmmf.add(f)
847 for f, args, msg in actions['dg']:
859 for f, args, msg in actions['dg']:
848 pmmf.add(f)
860 pmmf.add(f)
849 for f, args, msg in actions['m']:
861 for f, args, msg in actions['m']:
850 f1, f2, fa, move, anc = args
862 f1, f2, fa, move, anc = args
851 if move:
863 if move:
852 pmmf.discard(f1)
864 pmmf.discard(f1)
853 pmmf.add(f)
865 pmmf.add(f)
854
866
855 # check case-folding collision in provisional merged manifest
867 # check case-folding collision in provisional merged manifest
856 foldmap = {}
868 foldmap = {}
857 for f in pmmf:
869 for f in pmmf:
858 fold = util.normcase(f)
870 fold = util.normcase(f)
859 if fold in foldmap:
871 if fold in foldmap:
860 raise error.Abort(_("case-folding collision between %s and %s")
872 raise error.Abort(_("case-folding collision between %s and %s")
861 % (f, foldmap[fold]))
873 % (f, foldmap[fold]))
862 foldmap[fold] = f
874 foldmap[fold] = f
863
875
864 # check case-folding of directories
876 # check case-folding of directories
865 foldprefix = unfoldprefix = lastfull = ''
877 foldprefix = unfoldprefix = lastfull = ''
866 for fold, f in sorted(foldmap.items()):
878 for fold, f in sorted(foldmap.items()):
867 if fold.startswith(foldprefix) and not f.startswith(unfoldprefix):
879 if fold.startswith(foldprefix) and not f.startswith(unfoldprefix):
868 # the folded prefix matches but actual casing is different
880 # the folded prefix matches but actual casing is different
869 raise error.Abort(_("case-folding collision between "
881 raise error.Abort(_("case-folding collision between "
870 "%s and directory of %s") % (lastfull, f))
882 "%s and directory of %s") % (lastfull, f))
871 foldprefix = fold + '/'
883 foldprefix = fold + '/'
872 unfoldprefix = f + '/'
884 unfoldprefix = f + '/'
873 lastfull = f
885 lastfull = f
874
886
875 def driverpreprocess(repo, ms, wctx, labels=None):
887 def driverpreprocess(repo, ms, wctx, labels=None):
876 """run the preprocess step of the merge driver, if any
888 """run the preprocess step of the merge driver, if any
877
889
878 This is currently not implemented -- it's an extension point."""
890 This is currently not implemented -- it's an extension point."""
879 return True
891 return True
880
892
881 def driverconclude(repo, ms, wctx, labels=None):
893 def driverconclude(repo, ms, wctx, labels=None):
882 """run the conclude step of the merge driver, if any
894 """run the conclude step of the merge driver, if any
883
895
884 This is currently not implemented -- it's an extension point."""
896 This is currently not implemented -- it's an extension point."""
885 return True
897 return True
886
898
887 def _filesindirs(repo, manifest, dirs):
899 def _filesindirs(repo, manifest, dirs):
888 """
900 """
889 Generator that yields pairs of all the files in the manifest that are found
901 Generator that yields pairs of all the files in the manifest that are found
890 inside the directories listed in dirs, and which directory they are found
902 inside the directories listed in dirs, and which directory they are found
891 in.
903 in.
892 """
904 """
893 for f in manifest:
905 for f in manifest:
894 for p in util.finddirs(f):
906 for p in util.finddirs(f):
895 if p in dirs:
907 if p in dirs:
896 yield f, p
908 yield f, p
897 break
909 break
898
910
899 def checkpathconflicts(repo, wctx, mctx, actions):
911 def checkpathconflicts(repo, wctx, mctx, actions):
900 """
912 """
901 Check if any actions introduce path conflicts in the repository, updating
913 Check if any actions introduce path conflicts in the repository, updating
902 actions to record or handle the path conflict accordingly.
914 actions to record or handle the path conflict accordingly.
903 """
915 """
904 mf = wctx.manifest()
916 mf = wctx.manifest()
905
917
906 # The set of local files that conflict with a remote directory.
918 # The set of local files that conflict with a remote directory.
907 localconflicts = set()
919 localconflicts = set()
908
920
909 # The set of directories that conflict with a remote file, and so may cause
921 # The set of directories that conflict with a remote file, and so may cause
910 # conflicts if they still contain any files after the merge.
922 # conflicts if they still contain any files after the merge.
911 remoteconflicts = set()
923 remoteconflicts = set()
912
924
913 # The set of directories that appear as both a file and a directory in the
925 # The set of directories that appear as both a file and a directory in the
914 # remote manifest. These indicate an invalid remote manifest, which
926 # remote manifest. These indicate an invalid remote manifest, which
915 # can't be updated to cleanly.
927 # can't be updated to cleanly.
916 invalidconflicts = set()
928 invalidconflicts = set()
917
929
918 # The set of directories that contain files that are being created.
930 # The set of directories that contain files that are being created.
919 createdfiledirs = set()
931 createdfiledirs = set()
920
932
921 # The set of files deleted by all the actions.
933 # The set of files deleted by all the actions.
922 deletedfiles = set()
934 deletedfiles = set()
923
935
924 for f, (m, args, msg) in actions.items():
936 for f, (m, args, msg) in actions.items():
925 if m in ('c', 'dc', 'm', 'cm'):
937 if m in ('c', 'dc', 'm', 'cm'):
926 # This action may create a new local file.
938 # This action may create a new local file.
927 createdfiledirs.update(util.finddirs(f))
939 createdfiledirs.update(util.finddirs(f))
928 if mf.hasdir(f):
940 if mf.hasdir(f):
929 # The file aliases a local directory. This might be ok if all
941 # The file aliases a local directory. This might be ok if all
930 # the files in the local directory are being deleted. This
942 # the files in the local directory are being deleted. This
931 # will be checked once we know what all the deleted files are.
943 # will be checked once we know what all the deleted files are.
932 remoteconflicts.add(f)
944 remoteconflicts.add(f)
933 # Track the names of all deleted files.
945 # Track the names of all deleted files.
934 if m == 'r':
946 if m == 'r':
935 deletedfiles.add(f)
947 deletedfiles.add(f)
936 if m == 'm':
948 if m == 'm':
937 f1, f2, fa, move, anc = args
949 f1, f2, fa, move, anc = args
938 if move:
950 if move:
939 deletedfiles.add(f1)
951 deletedfiles.add(f1)
940 if m == 'dm':
952 if m == 'dm':
941 f2, flags = args
953 f2, flags = args
942 deletedfiles.add(f2)
954 deletedfiles.add(f2)
943
955
944 # Check all directories that contain created files for path conflicts.
956 # Check all directories that contain created files for path conflicts.
945 for p in createdfiledirs:
957 for p in createdfiledirs:
946 if p in mf:
958 if p in mf:
947 if p in mctx:
959 if p in mctx:
948 # A file is in a directory which aliases both a local
960 # A file is in a directory which aliases both a local
949 # and a remote file. This is an internal inconsistency
961 # and a remote file. This is an internal inconsistency
950 # within the remote manifest.
962 # within the remote manifest.
951 invalidconflicts.add(p)
963 invalidconflicts.add(p)
952 else:
964 else:
953 # A file is in a directory which aliases a local file.
965 # A file is in a directory which aliases a local file.
954 # We will need to rename the local file.
966 # We will need to rename the local file.
955 localconflicts.add(p)
967 localconflicts.add(p)
956 if p in actions and actions[p][0] in ('c', 'dc', 'm', 'cm'):
968 if p in actions and actions[p][0] in ('c', 'dc', 'm', 'cm'):
957 # The file is in a directory which aliases a remote file.
969 # The file is in a directory which aliases a remote file.
958 # This is an internal inconsistency within the remote
970 # This is an internal inconsistency within the remote
959 # manifest.
971 # manifest.
960 invalidconflicts.add(p)
972 invalidconflicts.add(p)
961
973
962 # Rename all local conflicting files that have not been deleted.
974 # Rename all local conflicting files that have not been deleted.
963 for p in localconflicts:
975 for p in localconflicts:
964 if p not in deletedfiles:
976 if p not in deletedfiles:
965 ctxname = str(wctx).rstrip('+')
977 ctxname = str(wctx).rstrip('+')
966 pnew = util.safename(p, ctxname, wctx, set(actions.keys()))
978 pnew = util.safename(p, ctxname, wctx, set(actions.keys()))
967 actions[pnew] = ('pr', (p,), "local path conflict")
979 actions[pnew] = ('pr', (p,), "local path conflict")
968 actions[p] = ('p', (pnew, 'l'), "path conflict")
980 actions[p] = ('p', (pnew, 'l'), "path conflict")
969
981
970 if remoteconflicts:
982 if remoteconflicts:
971 # Check if all files in the conflicting directories have been removed.
983 # Check if all files in the conflicting directories have been removed.
972 ctxname = str(mctx).rstrip('+')
984 ctxname = str(mctx).rstrip('+')
973 for f, p in _filesindirs(repo, mf, remoteconflicts):
985 for f, p in _filesindirs(repo, mf, remoteconflicts):
974 if f not in deletedfiles:
986 if f not in deletedfiles:
975 m, args, msg = actions[p]
987 m, args, msg = actions[p]
976 pnew = util.safename(p, ctxname, wctx, set(actions.keys()))
988 pnew = util.safename(p, ctxname, wctx, set(actions.keys()))
977 if m in ('dc', 'm'):
989 if m in ('dc', 'm'):
978 # Action was merge, just update target.
990 # Action was merge, just update target.
979 actions[pnew] = (m, args, msg)
991 actions[pnew] = (m, args, msg)
980 else:
992 else:
981 # Action was create, change to renamed get action.
993 # Action was create, change to renamed get action.
982 fl = args[0]
994 fl = args[0]
983 actions[pnew] = ('dg', (p, fl), "remote path conflict")
995 actions[pnew] = ('dg', (p, fl), "remote path conflict")
984 actions[p] = ('p', (pnew, 'r'), "path conflict")
996 actions[p] = ('p', (pnew, 'r'), "path conflict")
985 remoteconflicts.remove(p)
997 remoteconflicts.remove(p)
986 break
998 break
987
999
988 if invalidconflicts:
1000 if invalidconflicts:
989 for p in invalidconflicts:
1001 for p in invalidconflicts:
990 repo.ui.warn(_("%s: is both a file and a directory\n") % p)
1002 repo.ui.warn(_("%s: is both a file and a directory\n") % p)
991 raise error.Abort(_("destination manifest contains path conflicts"))
1003 raise error.Abort(_("destination manifest contains path conflicts"))
992
1004
993 def manifestmerge(repo, wctx, p2, pa, branchmerge, force, matcher,
1005 def manifestmerge(repo, wctx, p2, pa, branchmerge, force, matcher,
994 acceptremote, followcopies, forcefulldiff=False):
1006 acceptremote, followcopies, forcefulldiff=False):
995 """
1007 """
996 Merge wctx and p2 with ancestor pa and generate merge action list
1008 Merge wctx and p2 with ancestor pa and generate merge action list
997
1009
998 branchmerge and force are as passed in to update
1010 branchmerge and force are as passed in to update
999 matcher = matcher to filter file lists
1011 matcher = matcher to filter file lists
1000 acceptremote = accept the incoming changes without prompting
1012 acceptremote = accept the incoming changes without prompting
1001 """
1013 """
1002 if matcher is not None and matcher.always():
1014 if matcher is not None and matcher.always():
1003 matcher = None
1015 matcher = None
1004
1016
1005 copy, movewithdir, diverge, renamedelete, dirmove = {}, {}, {}, {}, {}
1017 copy, movewithdir, diverge, renamedelete, dirmove = {}, {}, {}, {}, {}
1006
1018
1007 # manifests fetched in order are going to be faster, so prime the caches
1019 # manifests fetched in order are going to be faster, so prime the caches
1008 [x.manifest() for x in
1020 [x.manifest() for x in
1009 sorted(wctx.parents() + [p2, pa], key=scmutil.intrev)]
1021 sorted(wctx.parents() + [p2, pa], key=scmutil.intrev)]
1010
1022
1011 if followcopies:
1023 if followcopies:
1012 ret = copies.mergecopies(repo, wctx, p2, pa)
1024 ret = copies.mergecopies(repo, wctx, p2, pa)
1013 copy, movewithdir, diverge, renamedelete, dirmove = ret
1025 copy, movewithdir, diverge, renamedelete, dirmove = ret
1014
1026
1015 boolbm = pycompat.bytestr(bool(branchmerge))
1027 boolbm = pycompat.bytestr(bool(branchmerge))
1016 boolf = pycompat.bytestr(bool(force))
1028 boolf = pycompat.bytestr(bool(force))
1017 boolm = pycompat.bytestr(bool(matcher))
1029 boolm = pycompat.bytestr(bool(matcher))
1018 repo.ui.note(_("resolving manifests\n"))
1030 repo.ui.note(_("resolving manifests\n"))
1019 repo.ui.debug(" branchmerge: %s, force: %s, partial: %s\n"
1031 repo.ui.debug(" branchmerge: %s, force: %s, partial: %s\n"
1020 % (boolbm, boolf, boolm))
1032 % (boolbm, boolf, boolm))
1021 repo.ui.debug(" ancestor: %s, local: %s, remote: %s\n" % (pa, wctx, p2))
1033 repo.ui.debug(" ancestor: %s, local: %s, remote: %s\n" % (pa, wctx, p2))
1022
1034
1023 m1, m2, ma = wctx.manifest(), p2.manifest(), pa.manifest()
1035 m1, m2, ma = wctx.manifest(), p2.manifest(), pa.manifest()
1024 copied = set(copy.values())
1036 copied = set(copy.values())
1025 copied.update(movewithdir.values())
1037 copied.update(movewithdir.values())
1026
1038
1027 if '.hgsubstate' in m1:
1039 if '.hgsubstate' in m1:
1028 # check whether sub state is modified
1040 # check whether sub state is modified
1029 if any(wctx.sub(s).dirty() for s in wctx.substate):
1041 if any(wctx.sub(s).dirty() for s in wctx.substate):
1030 m1['.hgsubstate'] = modifiednodeid
1042 m1['.hgsubstate'] = modifiednodeid
1031
1043
1032 # Don't use m2-vs-ma optimization if:
1044 # Don't use m2-vs-ma optimization if:
1033 # - ma is the same as m1 or m2, which we're just going to diff again later
1045 # - ma is the same as m1 or m2, which we're just going to diff again later
1034 # - The caller specifically asks for a full diff, which is useful during bid
1046 # - The caller specifically asks for a full diff, which is useful during bid
1035 # merge.
1047 # merge.
1036 if (pa not in ([wctx, p2] + wctx.parents()) and not forcefulldiff):
1048 if (pa not in ([wctx, p2] + wctx.parents()) and not forcefulldiff):
1037 # Identify which files are relevant to the merge, so we can limit the
1049 # Identify which files are relevant to the merge, so we can limit the
1038 # total m1-vs-m2 diff to just those files. This has significant
1050 # total m1-vs-m2 diff to just those files. This has significant
1039 # performance benefits in large repositories.
1051 # performance benefits in large repositories.
1040 relevantfiles = set(ma.diff(m2).keys())
1052 relevantfiles = set(ma.diff(m2).keys())
1041
1053
1042 # For copied and moved files, we need to add the source file too.
1054 # For copied and moved files, we need to add the source file too.
1043 for copykey, copyvalue in copy.iteritems():
1055 for copykey, copyvalue in copy.iteritems():
1044 if copyvalue in relevantfiles:
1056 if copyvalue in relevantfiles:
1045 relevantfiles.add(copykey)
1057 relevantfiles.add(copykey)
1046 for movedirkey in movewithdir:
1058 for movedirkey in movewithdir:
1047 relevantfiles.add(movedirkey)
1059 relevantfiles.add(movedirkey)
1048 filesmatcher = scmutil.matchfiles(repo, relevantfiles)
1060 filesmatcher = scmutil.matchfiles(repo, relevantfiles)
1049 matcher = matchmod.intersectmatchers(matcher, filesmatcher)
1061 matcher = matchmod.intersectmatchers(matcher, filesmatcher)
1050
1062
1051 diff = m1.diff(m2, match=matcher)
1063 diff = m1.diff(m2, match=matcher)
1052
1064
1053 if matcher is None:
1065 if matcher is None:
1054 matcher = matchmod.always('', '')
1066 matcher = matchmod.always('', '')
1055
1067
1056 actions = {}
1068 actions = {}
1057 for f, ((n1, fl1), (n2, fl2)) in diff.iteritems():
1069 for f, ((n1, fl1), (n2, fl2)) in diff.iteritems():
1058 if n1 and n2: # file exists on both local and remote side
1070 if n1 and n2: # file exists on both local and remote side
1059 if f not in ma:
1071 if f not in ma:
1060 fa = copy.get(f, None)
1072 fa = copy.get(f, None)
1061 if fa is not None:
1073 if fa is not None:
1062 actions[f] = ('m', (f, f, fa, False, pa.node()),
1074 actions[f] = ('m', (f, f, fa, False, pa.node()),
1063 "both renamed from " + fa)
1075 "both renamed from " + fa)
1064 else:
1076 else:
1065 actions[f] = ('m', (f, f, None, False, pa.node()),
1077 actions[f] = ('m', (f, f, None, False, pa.node()),
1066 "both created")
1078 "both created")
1067 else:
1079 else:
1068 a = ma[f]
1080 a = ma[f]
1069 fla = ma.flags(f)
1081 fla = ma.flags(f)
1070 nol = 'l' not in fl1 + fl2 + fla
1082 nol = 'l' not in fl1 + fl2 + fla
1071 if n2 == a and fl2 == fla:
1083 if n2 == a and fl2 == fla:
1072 actions[f] = ('k', (), "remote unchanged")
1084 actions[f] = ('k', (), "remote unchanged")
1073 elif n1 == a and fl1 == fla: # local unchanged - use remote
1085 elif n1 == a and fl1 == fla: # local unchanged - use remote
1074 if n1 == n2: # optimization: keep local content
1086 if n1 == n2: # optimization: keep local content
1075 actions[f] = ('e', (fl2,), "update permissions")
1087 actions[f] = ('e', (fl2,), "update permissions")
1076 else:
1088 else:
1077 actions[f] = ('g', (fl2, False), "remote is newer")
1089 actions[f] = ('g', (fl2, False), "remote is newer")
1078 elif nol and n2 == a: # remote only changed 'x'
1090 elif nol and n2 == a: # remote only changed 'x'
1079 actions[f] = ('e', (fl2,), "update permissions")
1091 actions[f] = ('e', (fl2,), "update permissions")
1080 elif nol and n1 == a: # local only changed 'x'
1092 elif nol and n1 == a: # local only changed 'x'
1081 actions[f] = ('g', (fl1, False), "remote is newer")
1093 actions[f] = ('g', (fl1, False), "remote is newer")
1082 else: # both changed something
1094 else: # both changed something
1083 actions[f] = ('m', (f, f, f, False, pa.node()),
1095 actions[f] = ('m', (f, f, f, False, pa.node()),
1084 "versions differ")
1096 "versions differ")
1085 elif n1: # file exists only on local side
1097 elif n1: # file exists only on local side
1086 if f in copied:
1098 if f in copied:
1087 pass # we'll deal with it on m2 side
1099 pass # we'll deal with it on m2 side
1088 elif f in movewithdir: # directory rename, move local
1100 elif f in movewithdir: # directory rename, move local
1089 f2 = movewithdir[f]
1101 f2 = movewithdir[f]
1090 if f2 in m2:
1102 if f2 in m2:
1091 actions[f2] = ('m', (f, f2, None, True, pa.node()),
1103 actions[f2] = ('m', (f, f2, None, True, pa.node()),
1092 "remote directory rename, both created")
1104 "remote directory rename, both created")
1093 else:
1105 else:
1094 actions[f2] = ('dm', (f, fl1),
1106 actions[f2] = ('dm', (f, fl1),
1095 "remote directory rename - move from " + f)
1107 "remote directory rename - move from " + f)
1096 elif f in copy:
1108 elif f in copy:
1097 f2 = copy[f]
1109 f2 = copy[f]
1098 actions[f] = ('m', (f, f2, f2, False, pa.node()),
1110 actions[f] = ('m', (f, f2, f2, False, pa.node()),
1099 "local copied/moved from " + f2)
1111 "local copied/moved from " + f2)
1100 elif f in ma: # clean, a different, no remote
1112 elif f in ma: # clean, a different, no remote
1101 if n1 != ma[f]:
1113 if n1 != ma[f]:
1102 if acceptremote:
1114 if acceptremote:
1103 actions[f] = ('r', None, "remote delete")
1115 actions[f] = ('r', None, "remote delete")
1104 else:
1116 else:
1105 actions[f] = ('cd', (f, None, f, False, pa.node()),
1117 actions[f] = ('cd', (f, None, f, False, pa.node()),
1106 "prompt changed/deleted")
1118 "prompt changed/deleted")
1107 elif n1 == addednodeid:
1119 elif n1 == addednodeid:
1108 # This extra 'a' is added by working copy manifest to mark
1120 # This extra 'a' is added by working copy manifest to mark
1109 # the file as locally added. We should forget it instead of
1121 # the file as locally added. We should forget it instead of
1110 # deleting it.
1122 # deleting it.
1111 actions[f] = ('f', None, "remote deleted")
1123 actions[f] = ('f', None, "remote deleted")
1112 else:
1124 else:
1113 actions[f] = ('r', None, "other deleted")
1125 actions[f] = ('r', None, "other deleted")
1114 elif n2: # file exists only on remote side
1126 elif n2: # file exists only on remote side
1115 if f in copied:
1127 if f in copied:
1116 pass # we'll deal with it on m1 side
1128 pass # we'll deal with it on m1 side
1117 elif f in movewithdir:
1129 elif f in movewithdir:
1118 f2 = movewithdir[f]
1130 f2 = movewithdir[f]
1119 if f2 in m1:
1131 if f2 in m1:
1120 actions[f2] = ('m', (f2, f, None, False, pa.node()),
1132 actions[f2] = ('m', (f2, f, None, False, pa.node()),
1121 "local directory rename, both created")
1133 "local directory rename, both created")
1122 else:
1134 else:
1123 actions[f2] = ('dg', (f, fl2),
1135 actions[f2] = ('dg', (f, fl2),
1124 "local directory rename - get from " + f)
1136 "local directory rename - get from " + f)
1125 elif f in copy:
1137 elif f in copy:
1126 f2 = copy[f]
1138 f2 = copy[f]
1127 if f2 in m2:
1139 if f2 in m2:
1128 actions[f] = ('m', (f2, f, f2, False, pa.node()),
1140 actions[f] = ('m', (f2, f, f2, False, pa.node()),
1129 "remote copied from " + f2)
1141 "remote copied from " + f2)
1130 else:
1142 else:
1131 actions[f] = ('m', (f2, f, f2, True, pa.node()),
1143 actions[f] = ('m', (f2, f, f2, True, pa.node()),
1132 "remote moved from " + f2)
1144 "remote moved from " + f2)
1133 elif f not in ma:
1145 elif f not in ma:
1134 # local unknown, remote created: the logic is described by the
1146 # local unknown, remote created: the logic is described by the
1135 # following table:
1147 # following table:
1136 #
1148 #
1137 # force branchmerge different | action
1149 # force branchmerge different | action
1138 # n * * | create
1150 # n * * | create
1139 # y n * | create
1151 # y n * | create
1140 # y y n | create
1152 # y y n | create
1141 # y y y | merge
1153 # y y y | merge
1142 #
1154 #
1143 # Checking whether the files are different is expensive, so we
1155 # Checking whether the files are different is expensive, so we
1144 # don't do that when we can avoid it.
1156 # don't do that when we can avoid it.
1145 if not force:
1157 if not force:
1146 actions[f] = ('c', (fl2,), "remote created")
1158 actions[f] = ('c', (fl2,), "remote created")
1147 elif not branchmerge:
1159 elif not branchmerge:
1148 actions[f] = ('c', (fl2,), "remote created")
1160 actions[f] = ('c', (fl2,), "remote created")
1149 else:
1161 else:
1150 actions[f] = ('cm', (fl2, pa.node()),
1162 actions[f] = ('cm', (fl2, pa.node()),
1151 "remote created, get or merge")
1163 "remote created, get or merge")
1152 elif n2 != ma[f]:
1164 elif n2 != ma[f]:
1153 df = None
1165 df = None
1154 for d in dirmove:
1166 for d in dirmove:
1155 if f.startswith(d):
1167 if f.startswith(d):
1156 # new file added in a directory that was moved
1168 # new file added in a directory that was moved
1157 df = dirmove[d] + f[len(d):]
1169 df = dirmove[d] + f[len(d):]
1158 break
1170 break
1159 if df is not None and df in m1:
1171 if df is not None and df in m1:
1160 actions[df] = ('m', (df, f, f, False, pa.node()),
1172 actions[df] = ('m', (df, f, f, False, pa.node()),
1161 "local directory rename - respect move from " + f)
1173 "local directory rename - respect move from " + f)
1162 elif acceptremote:
1174 elif acceptremote:
1163 actions[f] = ('c', (fl2,), "remote recreating")
1175 actions[f] = ('c', (fl2,), "remote recreating")
1164 else:
1176 else:
1165 actions[f] = ('dc', (None, f, f, False, pa.node()),
1177 actions[f] = ('dc', (None, f, f, False, pa.node()),
1166 "prompt deleted/changed")
1178 "prompt deleted/changed")
1167
1179
1168 if repo.ui.configbool('experimental', 'merge.checkpathconflicts'):
1180 if repo.ui.configbool('experimental', 'merge.checkpathconflicts'):
1169 # If we are merging, look for path conflicts.
1181 # If we are merging, look for path conflicts.
1170 checkpathconflicts(repo, wctx, p2, actions)
1182 checkpathconflicts(repo, wctx, p2, actions)
1171
1183
1172 return actions, diverge, renamedelete
1184 return actions, diverge, renamedelete
1173
1185
1174 def _resolvetrivial(repo, wctx, mctx, ancestor, actions):
1186 def _resolvetrivial(repo, wctx, mctx, ancestor, actions):
1175 """Resolves false conflicts where the nodeid changed but the content
1187 """Resolves false conflicts where the nodeid changed but the content
1176 remained the same."""
1188 remained the same."""
1177
1189
1178 for f, (m, args, msg) in actions.items():
1190 for f, (m, args, msg) in actions.items():
1179 if m == 'cd' and f in ancestor and not wctx[f].cmp(ancestor[f]):
1191 if m == 'cd' and f in ancestor and not wctx[f].cmp(ancestor[f]):
1180 # local did change but ended up with same content
1192 # local did change but ended up with same content
1181 actions[f] = 'r', None, "prompt same"
1193 actions[f] = 'r', None, "prompt same"
1182 elif m == 'dc' and f in ancestor and not mctx[f].cmp(ancestor[f]):
1194 elif m == 'dc' and f in ancestor and not mctx[f].cmp(ancestor[f]):
1183 # remote did change but ended up with same content
1195 # remote did change but ended up with same content
1184 del actions[f] # don't get = keep local deleted
1196 del actions[f] # don't get = keep local deleted
1185
1197
1186 def calculateupdates(repo, wctx, mctx, ancestors, branchmerge, force,
1198 def calculateupdates(repo, wctx, mctx, ancestors, branchmerge, force,
1187 acceptremote, followcopies, matcher=None,
1199 acceptremote, followcopies, matcher=None,
1188 mergeforce=False):
1200 mergeforce=False):
1189 """Calculate the actions needed to merge mctx into wctx using ancestors"""
1201 """Calculate the actions needed to merge mctx into wctx using ancestors"""
1190 # Avoid cycle.
1202 # Avoid cycle.
1191 from . import sparse
1203 from . import sparse
1192
1204
1193 if len(ancestors) == 1: # default
1205 if len(ancestors) == 1: # default
1194 actions, diverge, renamedelete = manifestmerge(
1206 actions, diverge, renamedelete = manifestmerge(
1195 repo, wctx, mctx, ancestors[0], branchmerge, force, matcher,
1207 repo, wctx, mctx, ancestors[0], branchmerge, force, matcher,
1196 acceptremote, followcopies)
1208 acceptremote, followcopies)
1197 _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce)
1209 _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce)
1198
1210
1199 else: # only when merge.preferancestor=* - the default
1211 else: # only when merge.preferancestor=* - the default
1200 repo.ui.note(
1212 repo.ui.note(
1201 _("note: merging %s and %s using bids from ancestors %s\n") %
1213 _("note: merging %s and %s using bids from ancestors %s\n") %
1202 (wctx, mctx, _(' and ').join(pycompat.bytestr(anc)
1214 (wctx, mctx, _(' and ').join(pycompat.bytestr(anc)
1203 for anc in ancestors)))
1215 for anc in ancestors)))
1204
1216
1205 # Call for bids
1217 # Call for bids
1206 fbids = {} # mapping filename to bids (action method to list af actions)
1218 fbids = {} # mapping filename to bids (action method to list af actions)
1207 diverge, renamedelete = None, None
1219 diverge, renamedelete = None, None
1208 for ancestor in ancestors:
1220 for ancestor in ancestors:
1209 repo.ui.note(_('\ncalculating bids for ancestor %s\n') % ancestor)
1221 repo.ui.note(_('\ncalculating bids for ancestor %s\n') % ancestor)
1210 actions, diverge1, renamedelete1 = manifestmerge(
1222 actions, diverge1, renamedelete1 = manifestmerge(
1211 repo, wctx, mctx, ancestor, branchmerge, force, matcher,
1223 repo, wctx, mctx, ancestor, branchmerge, force, matcher,
1212 acceptremote, followcopies, forcefulldiff=True)
1224 acceptremote, followcopies, forcefulldiff=True)
1213 _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce)
1225 _checkunknownfiles(repo, wctx, mctx, force, actions, mergeforce)
1214
1226
1215 # Track the shortest set of warning on the theory that bid
1227 # Track the shortest set of warning on the theory that bid
1216 # merge will correctly incorporate more information
1228 # merge will correctly incorporate more information
1217 if diverge is None or len(diverge1) < len(diverge):
1229 if diverge is None or len(diverge1) < len(diverge):
1218 diverge = diverge1
1230 diverge = diverge1
1219 if renamedelete is None or len(renamedelete) < len(renamedelete1):
1231 if renamedelete is None or len(renamedelete) < len(renamedelete1):
1220 renamedelete = renamedelete1
1232 renamedelete = renamedelete1
1221
1233
1222 for f, a in sorted(actions.iteritems()):
1234 for f, a in sorted(actions.iteritems()):
1223 m, args, msg = a
1235 m, args, msg = a
1224 repo.ui.debug(' %s: %s -> %s\n' % (f, msg, m))
1236 repo.ui.debug(' %s: %s -> %s\n' % (f, msg, m))
1225 if f in fbids:
1237 if f in fbids:
1226 d = fbids[f]
1238 d = fbids[f]
1227 if m in d:
1239 if m in d:
1228 d[m].append(a)
1240 d[m].append(a)
1229 else:
1241 else:
1230 d[m] = [a]
1242 d[m] = [a]
1231 else:
1243 else:
1232 fbids[f] = {m: [a]}
1244 fbids[f] = {m: [a]}
1233
1245
1234 # Pick the best bid for each file
1246 # Pick the best bid for each file
1235 repo.ui.note(_('\nauction for merging merge bids\n'))
1247 repo.ui.note(_('\nauction for merging merge bids\n'))
1236 actions = {}
1248 actions = {}
1237 dms = [] # filenames that have dm actions
1249 dms = [] # filenames that have dm actions
1238 for f, bids in sorted(fbids.items()):
1250 for f, bids in sorted(fbids.items()):
1239 # bids is a mapping from action method to list af actions
1251 # bids is a mapping from action method to list af actions
1240 # Consensus?
1252 # Consensus?
1241 if len(bids) == 1: # all bids are the same kind of method
1253 if len(bids) == 1: # all bids are the same kind of method
1242 m, l = list(bids.items())[0]
1254 m, l = list(bids.items())[0]
1243 if all(a == l[0] for a in l[1:]): # len(bids) is > 1
1255 if all(a == l[0] for a in l[1:]): # len(bids) is > 1
1244 repo.ui.note(_(" %s: consensus for %s\n") % (f, m))
1256 repo.ui.note(_(" %s: consensus for %s\n") % (f, m))
1245 actions[f] = l[0]
1257 actions[f] = l[0]
1246 if m == 'dm':
1258 if m == 'dm':
1247 dms.append(f)
1259 dms.append(f)
1248 continue
1260 continue
1249 # If keep is an option, just do it.
1261 # If keep is an option, just do it.
1250 if 'k' in bids:
1262 if 'k' in bids:
1251 repo.ui.note(_(" %s: picking 'keep' action\n") % f)
1263 repo.ui.note(_(" %s: picking 'keep' action\n") % f)
1252 actions[f] = bids['k'][0]
1264 actions[f] = bids['k'][0]
1253 continue
1265 continue
1254 # If there are gets and they all agree [how could they not?], do it.
1266 # If there are gets and they all agree [how could they not?], do it.
1255 if 'g' in bids:
1267 if 'g' in bids:
1256 ga0 = bids['g'][0]
1268 ga0 = bids['g'][0]
1257 if all(a == ga0 for a in bids['g'][1:]):
1269 if all(a == ga0 for a in bids['g'][1:]):
1258 repo.ui.note(_(" %s: picking 'get' action\n") % f)
1270 repo.ui.note(_(" %s: picking 'get' action\n") % f)
1259 actions[f] = ga0
1271 actions[f] = ga0
1260 continue
1272 continue
1261 # TODO: Consider other simple actions such as mode changes
1273 # TODO: Consider other simple actions such as mode changes
1262 # Handle inefficient democrazy.
1274 # Handle inefficient democrazy.
1263 repo.ui.note(_(' %s: multiple bids for merge action:\n') % f)
1275 repo.ui.note(_(' %s: multiple bids for merge action:\n') % f)
1264 for m, l in sorted(bids.items()):
1276 for m, l in sorted(bids.items()):
1265 for _f, args, msg in l:
1277 for _f, args, msg in l:
1266 repo.ui.note(' %s -> %s\n' % (msg, m))
1278 repo.ui.note(' %s -> %s\n' % (msg, m))
1267 # Pick random action. TODO: Instead, prompt user when resolving
1279 # Pick random action. TODO: Instead, prompt user when resolving
1268 m, l = list(bids.items())[0]
1280 m, l = list(bids.items())[0]
1269 repo.ui.warn(_(' %s: ambiguous merge - picked %s action\n') %
1281 repo.ui.warn(_(' %s: ambiguous merge - picked %s action\n') %
1270 (f, m))
1282 (f, m))
1271 actions[f] = l[0]
1283 actions[f] = l[0]
1272 if m == 'dm':
1284 if m == 'dm':
1273 dms.append(f)
1285 dms.append(f)
1274 continue
1286 continue
1275 # Work around 'dm' that can cause multiple actions for the same file
1287 # Work around 'dm' that can cause multiple actions for the same file
1276 for f in dms:
1288 for f in dms:
1277 dm, (f0, flags), msg = actions[f]
1289 dm, (f0, flags), msg = actions[f]
1278 assert dm == 'dm', dm
1290 assert dm == 'dm', dm
1279 if f0 in actions and actions[f0][0] == 'r':
1291 if f0 in actions and actions[f0][0] == 'r':
1280 # We have one bid for removing a file and another for moving it.
1292 # We have one bid for removing a file and another for moving it.
1281 # These two could be merged as first move and then delete ...
1293 # These two could be merged as first move and then delete ...
1282 # but instead drop moving and just delete.
1294 # but instead drop moving and just delete.
1283 del actions[f]
1295 del actions[f]
1284 repo.ui.note(_('end of auction\n\n'))
1296 repo.ui.note(_('end of auction\n\n'))
1285
1297
1286 _resolvetrivial(repo, wctx, mctx, ancestors[0], actions)
1298 _resolvetrivial(repo, wctx, mctx, ancestors[0], actions)
1287
1299
1288 if wctx.rev() is None:
1300 if wctx.rev() is None:
1289 fractions = _forgetremoved(wctx, mctx, branchmerge)
1301 fractions = _forgetremoved(wctx, mctx, branchmerge)
1290 actions.update(fractions)
1302 actions.update(fractions)
1291
1303
1292 prunedactions = sparse.filterupdatesactions(repo, wctx, mctx, branchmerge,
1304 prunedactions = sparse.filterupdatesactions(repo, wctx, mctx, branchmerge,
1293 actions)
1305 actions)
1294
1306
1295 return prunedactions, diverge, renamedelete
1307 return prunedactions, diverge, renamedelete
1296
1308
1297 def _getcwd():
1309 def _getcwd():
1298 try:
1310 try:
1299 return pycompat.getcwd()
1311 return pycompat.getcwd()
1300 except OSError as err:
1312 except OSError as err:
1301 if err.errno == errno.ENOENT:
1313 if err.errno == errno.ENOENT:
1302 return None
1314 return None
1303 raise
1315 raise
1304
1316
1305 def batchremove(repo, wctx, actions):
1317 def batchremove(repo, wctx, actions):
1306 """apply removes to the working directory
1318 """apply removes to the working directory
1307
1319
1308 yields tuples for progress updates
1320 yields tuples for progress updates
1309 """
1321 """
1310 verbose = repo.ui.verbose
1322 verbose = repo.ui.verbose
1311 cwd = _getcwd()
1323 cwd = _getcwd()
1312 i = 0
1324 i = 0
1313 for f, args, msg in actions:
1325 for f, args, msg in actions:
1314 repo.ui.debug(" %s: %s -> r\n" % (f, msg))
1326 repo.ui.debug(" %s: %s -> r\n" % (f, msg))
1315 if verbose:
1327 if verbose:
1316 repo.ui.note(_("removing %s\n") % f)
1328 repo.ui.note(_("removing %s\n") % f)
1317 wctx[f].audit()
1329 wctx[f].audit()
1318 try:
1330 try:
1319 wctx[f].remove(ignoremissing=True)
1331 wctx[f].remove(ignoremissing=True)
1320 except OSError as inst:
1332 except OSError as inst:
1321 repo.ui.warn(_("update failed to remove %s: %s!\n") %
1333 repo.ui.warn(_("update failed to remove %s: %s!\n") %
1322 (f, inst.strerror))
1334 (f, inst.strerror))
1323 if i == 100:
1335 if i == 100:
1324 yield i, f
1336 yield i, f
1325 i = 0
1337 i = 0
1326 i += 1
1338 i += 1
1327 if i > 0:
1339 if i > 0:
1328 yield i, f
1340 yield i, f
1329
1341
1330 if cwd and not _getcwd():
1342 if cwd and not _getcwd():
1331 # cwd was removed in the course of removing files; print a helpful
1343 # cwd was removed in the course of removing files; print a helpful
1332 # warning.
1344 # warning.
1333 repo.ui.warn(_("current directory was removed\n"
1345 repo.ui.warn(_("current directory was removed\n"
1334 "(consider changing to repo root: %s)\n") % repo.root)
1346 "(consider changing to repo root: %s)\n") % repo.root)
1335
1347
1336 def batchget(repo, mctx, wctx, actions):
1348 def batchget(repo, mctx, wctx, actions):
1337 """apply gets to the working directory
1349 """apply gets to the working directory
1338
1350
1339 mctx is the context to get from
1351 mctx is the context to get from
1340
1352
1341 yields tuples for progress updates
1353 yields tuples for progress updates
1342 """
1354 """
1343 verbose = repo.ui.verbose
1355 verbose = repo.ui.verbose
1344 fctx = mctx.filectx
1356 fctx = mctx.filectx
1345 ui = repo.ui
1357 ui = repo.ui
1346 i = 0
1358 i = 0
1347 with repo.wvfs.backgroundclosing(ui, expectedcount=len(actions)):
1359 with repo.wvfs.backgroundclosing(ui, expectedcount=len(actions)):
1348 for f, (flags, backup), msg in actions:
1360 for f, (flags, backup), msg in actions:
1349 repo.ui.debug(" %s: %s -> g\n" % (f, msg))
1361 repo.ui.debug(" %s: %s -> g\n" % (f, msg))
1350 if verbose:
1362 if verbose:
1351 repo.ui.note(_("getting %s\n") % f)
1363 repo.ui.note(_("getting %s\n") % f)
1352
1364
1353 if backup:
1365 if backup:
1354 # If a file or directory exists with the same name, back that
1366 # If a file or directory exists with the same name, back that
1355 # up. Otherwise, look to see if there is a file that conflicts
1367 # up. Otherwise, look to see if there is a file that conflicts
1356 # with a directory this file is in, and if so, back that up.
1368 # with a directory this file is in, and if so, back that up.
1357 absf = repo.wjoin(f)
1369 absf = repo.wjoin(f)
1358 if not repo.wvfs.lexists(f):
1370 if not repo.wvfs.lexists(f):
1359 for p in util.finddirs(f):
1371 for p in util.finddirs(f):
1360 if repo.wvfs.isfileorlink(p):
1372 if repo.wvfs.isfileorlink(p):
1361 absf = repo.wjoin(p)
1373 absf = repo.wjoin(p)
1362 break
1374 break
1363 orig = scmutil.origpath(ui, repo, absf)
1375 orig = scmutil.origpath(ui, repo, absf)
1364 if repo.wvfs.lexists(absf):
1376 if repo.wvfs.lexists(absf):
1365 util.rename(absf, orig)
1377 util.rename(absf, orig)
1366 wctx[f].clearunknown()
1378 wctx[f].clearunknown()
1367 wctx[f].write(fctx(f).data(), flags, backgroundclose=True)
1379 wctx[f].write(fctx(f).data(), flags, backgroundclose=True)
1368 if i == 100:
1380 if i == 100:
1369 yield i, f
1381 yield i, f
1370 i = 0
1382 i = 0
1371 i += 1
1383 i += 1
1372 if i > 0:
1384 if i > 0:
1373 yield i, f
1385 yield i, f
1374
1386
1375
1387
1376 def applyupdates(repo, actions, wctx, mctx, overwrite, labels=None):
1388 def applyupdates(repo, actions, wctx, mctx, overwrite, labels=None):
1377 """apply the merge action list to the working directory
1389 """apply the merge action list to the working directory
1378
1390
1379 wctx is the working copy context
1391 wctx is the working copy context
1380 mctx is the context to be merged into the working copy
1392 mctx is the context to be merged into the working copy
1381
1393
1382 Return a tuple of counts (updated, merged, removed, unresolved) that
1394 Return a tuple of counts (updated, merged, removed, unresolved) that
1383 describes how many files were affected by the update.
1395 describes how many files were affected by the update.
1384 """
1396 """
1385
1397
1386 updated, merged, removed = 0, 0, 0
1398 updated, merged, removed = 0, 0, 0
1387 ms = mergestate.clean(repo, wctx.p1().node(), mctx.node(), labels)
1399 ms = mergestate.clean(repo, wctx.p1().node(), mctx.node(), labels)
1388 moves = []
1400 moves = []
1389 for m, l in actions.items():
1401 for m, l in actions.items():
1390 l.sort()
1402 l.sort()
1391
1403
1392 # 'cd' and 'dc' actions are treated like other merge conflicts
1404 # 'cd' and 'dc' actions are treated like other merge conflicts
1393 mergeactions = sorted(actions['cd'])
1405 mergeactions = sorted(actions['cd'])
1394 mergeactions.extend(sorted(actions['dc']))
1406 mergeactions.extend(sorted(actions['dc']))
1395 mergeactions.extend(actions['m'])
1407 mergeactions.extend(actions['m'])
1396 for f, args, msg in mergeactions:
1408 for f, args, msg in mergeactions:
1397 f1, f2, fa, move, anc = args
1409 f1, f2, fa, move, anc = args
1398 if f == '.hgsubstate': # merged internally
1410 if f == '.hgsubstate': # merged internally
1399 continue
1411 continue
1400 if f1 is None:
1412 if f1 is None:
1401 fcl = filemerge.absentfilectx(wctx, fa)
1413 fcl = filemerge.absentfilectx(wctx, fa)
1402 else:
1414 else:
1403 repo.ui.debug(" preserving %s for resolve of %s\n" % (f1, f))
1415 repo.ui.debug(" preserving %s for resolve of %s\n" % (f1, f))
1404 fcl = wctx[f1]
1416 fcl = wctx[f1]
1405 if f2 is None:
1417 if f2 is None:
1406 fco = filemerge.absentfilectx(mctx, fa)
1418 fco = filemerge.absentfilectx(mctx, fa)
1407 else:
1419 else:
1408 fco = mctx[f2]
1420 fco = mctx[f2]
1409 actx = repo[anc]
1421 actx = repo[anc]
1410 if fa in actx:
1422 if fa in actx:
1411 fca = actx[fa]
1423 fca = actx[fa]
1412 else:
1424 else:
1413 # TODO: move to absentfilectx
1425 # TODO: move to absentfilectx
1414 fca = repo.filectx(f1, fileid=nullrev)
1426 fca = repo.filectx(f1, fileid=nullrev)
1415 ms.add(fcl, fco, fca, f)
1427 ms.add(fcl, fco, fca, f)
1416 if f1 != f and move:
1428 if f1 != f and move:
1417 moves.append(f1)
1429 moves.append(f1)
1418
1430
1419 _updating = _('updating')
1431 _updating = _('updating')
1420 _files = _('files')
1432 _files = _('files')
1421 progress = repo.ui.progress
1433 progress = repo.ui.progress
1422
1434
1423 # remove renamed files after safely stored
1435 # remove renamed files after safely stored
1424 for f in moves:
1436 for f in moves:
1425 if wctx[f].lexists():
1437 if wctx[f].lexists():
1426 repo.ui.debug("removing %s\n" % f)
1438 repo.ui.debug("removing %s\n" % f)
1427 wctx[f].audit()
1439 wctx[f].audit()
1428 wctx[f].remove()
1440 wctx[f].remove()
1429
1441
1430 numupdates = sum(len(l) for m, l in actions.items() if m != 'k')
1442 numupdates = sum(len(l) for m, l in actions.items() if m != 'k')
1431 z = 0
1443 z = 0
1432
1444
1433 if [a for a in actions['r'] if a[0] == '.hgsubstate']:
1445 if [a for a in actions['r'] if a[0] == '.hgsubstate']:
1434 subrepo.submerge(repo, wctx, mctx, wctx, overwrite, labels)
1446 subrepo.submerge(repo, wctx, mctx, wctx, overwrite, labels)
1435
1447
1436 # record path conflicts
1448 # record path conflicts
1437 for f, args, msg in actions['p']:
1449 for f, args, msg in actions['p']:
1438 f1, fo = args
1450 f1, fo = args
1439 s = repo.ui.status
1451 s = repo.ui.status
1440 s(_("%s: path conflict - a file or link has the same name as a "
1452 s(_("%s: path conflict - a file or link has the same name as a "
1441 "directory\n") % f)
1453 "directory\n") % f)
1442 if fo == 'l':
1454 if fo == 'l':
1443 s(_("the local file has been renamed to %s\n") % f1)
1455 s(_("the local file has been renamed to %s\n") % f1)
1444 else:
1456 else:
1445 s(_("the remote file has been renamed to %s\n") % f1)
1457 s(_("the remote file has been renamed to %s\n") % f1)
1446 s(_("resolve manually then use 'hg resolve --mark %s'\n") % f)
1458 s(_("resolve manually then use 'hg resolve --mark %s'\n") % f)
1447 ms.addpath(f, f1, fo)
1459 ms.addpath(f, f1, fo)
1448 z += 1
1460 z += 1
1449 progress(_updating, z, item=f, total=numupdates, unit=_files)
1461 progress(_updating, z, item=f, total=numupdates, unit=_files)
1450
1462
1451 # When merging in-memory, we can't support worker processes, so set the
1463 # When merging in-memory, we can't support worker processes, so set the
1452 # per-item cost at 0 in that case.
1464 # per-item cost at 0 in that case.
1453 cost = 0 if wctx.isinmemory() else 0.001
1465 cost = 0 if wctx.isinmemory() else 0.001
1454
1466
1455 # remove in parallel (must come before resolving path conflicts and getting)
1467 # remove in parallel (must come before resolving path conflicts and getting)
1456 prog = worker.worker(repo.ui, cost, batchremove, (repo, wctx),
1468 prog = worker.worker(repo.ui, cost, batchremove, (repo, wctx),
1457 actions['r'])
1469 actions['r'])
1458 for i, item in prog:
1470 for i, item in prog:
1459 z += i
1471 z += i
1460 progress(_updating, z, item=item, total=numupdates, unit=_files)
1472 progress(_updating, z, item=item, total=numupdates, unit=_files)
1461 removed = len(actions['r'])
1473 removed = len(actions['r'])
1462
1474
1463 # resolve path conflicts (must come before getting)
1475 # resolve path conflicts (must come before getting)
1464 for f, args, msg in actions['pr']:
1476 for f, args, msg in actions['pr']:
1465 repo.ui.debug(" %s: %s -> pr\n" % (f, msg))
1477 repo.ui.debug(" %s: %s -> pr\n" % (f, msg))
1466 f0, = args
1478 f0, = args
1467 if wctx[f0].lexists():
1479 if wctx[f0].lexists():
1468 repo.ui.note(_("moving %s to %s\n") % (f0, f))
1480 repo.ui.note(_("moving %s to %s\n") % (f0, f))
1469 wctx[f].audit()
1481 wctx[f].audit()
1470 wctx[f].write(wctx.filectx(f0).data(), wctx.filectx(f0).flags())
1482 wctx[f].write(wctx.filectx(f0).data(), wctx.filectx(f0).flags())
1471 wctx[f0].remove()
1483 wctx[f0].remove()
1472 z += 1
1484 z += 1
1473 progress(_updating, z, item=f, total=numupdates, unit=_files)
1485 progress(_updating, z, item=f, total=numupdates, unit=_files)
1474
1486
1475 # get in parallel
1487 # get in parallel
1476 prog = worker.worker(repo.ui, cost, batchget, (repo, mctx, wctx),
1488 prog = worker.worker(repo.ui, cost, batchget, (repo, mctx, wctx),
1477 actions['g'])
1489 actions['g'])
1478 for i, item in prog:
1490 for i, item in prog:
1479 z += i
1491 z += i
1480 progress(_updating, z, item=item, total=numupdates, unit=_files)
1492 progress(_updating, z, item=item, total=numupdates, unit=_files)
1481 updated = len(actions['g'])
1493 updated = len(actions['g'])
1482
1494
1483 if [a for a in actions['g'] if a[0] == '.hgsubstate']:
1495 if [a for a in actions['g'] if a[0] == '.hgsubstate']:
1484 subrepo.submerge(repo, wctx, mctx, wctx, overwrite, labels)
1496 subrepo.submerge(repo, wctx, mctx, wctx, overwrite, labels)
1485
1497
1486 # forget (manifest only, just log it) (must come first)
1498 # forget (manifest only, just log it) (must come first)
1487 for f, args, msg in actions['f']:
1499 for f, args, msg in actions['f']:
1488 repo.ui.debug(" %s: %s -> f\n" % (f, msg))
1500 repo.ui.debug(" %s: %s -> f\n" % (f, msg))
1489 z += 1
1501 z += 1
1490 progress(_updating, z, item=f, total=numupdates, unit=_files)
1502 progress(_updating, z, item=f, total=numupdates, unit=_files)
1491
1503
1492 # re-add (manifest only, just log it)
1504 # re-add (manifest only, just log it)
1493 for f, args, msg in actions['a']:
1505 for f, args, msg in actions['a']:
1494 repo.ui.debug(" %s: %s -> a\n" % (f, msg))
1506 repo.ui.debug(" %s: %s -> a\n" % (f, msg))
1495 z += 1
1507 z += 1
1496 progress(_updating, z, item=f, total=numupdates, unit=_files)
1508 progress(_updating, z, item=f, total=numupdates, unit=_files)
1497
1509
1498 # re-add/mark as modified (manifest only, just log it)
1510 # re-add/mark as modified (manifest only, just log it)
1499 for f, args, msg in actions['am']:
1511 for f, args, msg in actions['am']:
1500 repo.ui.debug(" %s: %s -> am\n" % (f, msg))
1512 repo.ui.debug(" %s: %s -> am\n" % (f, msg))
1501 z += 1
1513 z += 1
1502 progress(_updating, z, item=f, total=numupdates, unit=_files)
1514 progress(_updating, z, item=f, total=numupdates, unit=_files)
1503
1515
1504 # keep (noop, just log it)
1516 # keep (noop, just log it)
1505 for f, args, msg in actions['k']:
1517 for f, args, msg in actions['k']:
1506 repo.ui.debug(" %s: %s -> k\n" % (f, msg))
1518 repo.ui.debug(" %s: %s -> k\n" % (f, msg))
1507 # no progress
1519 # no progress
1508
1520
1509 # directory rename, move local
1521 # directory rename, move local
1510 for f, args, msg in actions['dm']:
1522 for f, args, msg in actions['dm']:
1511 repo.ui.debug(" %s: %s -> dm\n" % (f, msg))
1523 repo.ui.debug(" %s: %s -> dm\n" % (f, msg))
1512 z += 1
1524 z += 1
1513 progress(_updating, z, item=f, total=numupdates, unit=_files)
1525 progress(_updating, z, item=f, total=numupdates, unit=_files)
1514 f0, flags = args
1526 f0, flags = args
1515 repo.ui.note(_("moving %s to %s\n") % (f0, f))
1527 repo.ui.note(_("moving %s to %s\n") % (f0, f))
1516 wctx[f].audit()
1528 wctx[f].audit()
1517 wctx[f].write(wctx.filectx(f0).data(), flags)
1529 wctx[f].write(wctx.filectx(f0).data(), flags)
1518 wctx[f0].remove()
1530 wctx[f0].remove()
1519 updated += 1
1531 updated += 1
1520
1532
1521 # local directory rename, get
1533 # local directory rename, get
1522 for f, args, msg in actions['dg']:
1534 for f, args, msg in actions['dg']:
1523 repo.ui.debug(" %s: %s -> dg\n" % (f, msg))
1535 repo.ui.debug(" %s: %s -> dg\n" % (f, msg))
1524 z += 1
1536 z += 1
1525 progress(_updating, z, item=f, total=numupdates, unit=_files)
1537 progress(_updating, z, item=f, total=numupdates, unit=_files)
1526 f0, flags = args
1538 f0, flags = args
1527 repo.ui.note(_("getting %s to %s\n") % (f0, f))
1539 repo.ui.note(_("getting %s to %s\n") % (f0, f))
1528 wctx[f].write(mctx.filectx(f0).data(), flags)
1540 wctx[f].write(mctx.filectx(f0).data(), flags)
1529 updated += 1
1541 updated += 1
1530
1542
1531 # exec
1543 # exec
1532 for f, args, msg in actions['e']:
1544 for f, args, msg in actions['e']:
1533 repo.ui.debug(" %s: %s -> e\n" % (f, msg))
1545 repo.ui.debug(" %s: %s -> e\n" % (f, msg))
1534 z += 1
1546 z += 1
1535 progress(_updating, z, item=f, total=numupdates, unit=_files)
1547 progress(_updating, z, item=f, total=numupdates, unit=_files)
1536 flags, = args
1548 flags, = args
1537 wctx[f].audit()
1549 wctx[f].audit()
1538 wctx[f].setflags('l' in flags, 'x' in flags)
1550 wctx[f].setflags('l' in flags, 'x' in flags)
1539 updated += 1
1551 updated += 1
1540
1552
1541 # the ordering is important here -- ms.mergedriver will raise if the merge
1553 # the ordering is important here -- ms.mergedriver will raise if the merge
1542 # driver has changed, and we want to be able to bypass it when overwrite is
1554 # driver has changed, and we want to be able to bypass it when overwrite is
1543 # True
1555 # True
1544 usemergedriver = not overwrite and mergeactions and ms.mergedriver
1556 usemergedriver = not overwrite and mergeactions and ms.mergedriver
1545
1557
1546 if usemergedriver:
1558 if usemergedriver:
1547 ms.commit()
1559 ms.commit()
1548 proceed = driverpreprocess(repo, ms, wctx, labels=labels)
1560 proceed = driverpreprocess(repo, ms, wctx, labels=labels)
1549 # the driver might leave some files unresolved
1561 # the driver might leave some files unresolved
1550 unresolvedf = set(ms.unresolved())
1562 unresolvedf = set(ms.unresolved())
1551 if not proceed:
1563 if not proceed:
1552 # XXX setting unresolved to at least 1 is a hack to make sure we
1564 # XXX setting unresolved to at least 1 is a hack to make sure we
1553 # error out
1565 # error out
1554 return updated, merged, removed, max(len(unresolvedf), 1)
1566 return updated, merged, removed, max(len(unresolvedf), 1)
1555 newactions = []
1567 newactions = []
1556 for f, args, msg in mergeactions:
1568 for f, args, msg in mergeactions:
1557 if f in unresolvedf:
1569 if f in unresolvedf:
1558 newactions.append((f, args, msg))
1570 newactions.append((f, args, msg))
1559 mergeactions = newactions
1571 mergeactions = newactions
1560
1572
1561 try:
1573 try:
1562 # premerge
1574 # premerge
1563 tocomplete = []
1575 tocomplete = []
1564 for f, args, msg in mergeactions:
1576 for f, args, msg in mergeactions:
1565 repo.ui.debug(" %s: %s -> m (premerge)\n" % (f, msg))
1577 repo.ui.debug(" %s: %s -> m (premerge)\n" % (f, msg))
1566 z += 1
1578 z += 1
1567 progress(_updating, z, item=f, total=numupdates, unit=_files)
1579 progress(_updating, z, item=f, total=numupdates, unit=_files)
1568 if f == '.hgsubstate': # subrepo states need updating
1580 if f == '.hgsubstate': # subrepo states need updating
1569 subrepo.submerge(repo, wctx, mctx, wctx.ancestor(mctx),
1581 subrepo.submerge(repo, wctx, mctx, wctx.ancestor(mctx),
1570 overwrite, labels)
1582 overwrite, labels)
1571 continue
1583 continue
1572 wctx[f].audit()
1584 wctx[f].audit()
1573 complete, r = ms.preresolve(f, wctx)
1585 complete, r = ms.preresolve(f, wctx)
1574 if not complete:
1586 if not complete:
1575 numupdates += 1
1587 numupdates += 1
1576 tocomplete.append((f, args, msg))
1588 tocomplete.append((f, args, msg))
1577
1589
1578 # merge
1590 # merge
1579 for f, args, msg in tocomplete:
1591 for f, args, msg in tocomplete:
1580 repo.ui.debug(" %s: %s -> m (merge)\n" % (f, msg))
1592 repo.ui.debug(" %s: %s -> m (merge)\n" % (f, msg))
1581 z += 1
1593 z += 1
1582 progress(_updating, z, item=f, total=numupdates, unit=_files)
1594 progress(_updating, z, item=f, total=numupdates, unit=_files)
1583 ms.resolve(f, wctx)
1595 ms.resolve(f, wctx)
1584
1596
1585 finally:
1597 finally:
1586 ms.commit()
1598 ms.commit()
1587
1599
1588 unresolved = ms.unresolvedcount()
1600 unresolved = ms.unresolvedcount()
1589
1601
1590 if usemergedriver and not unresolved and ms.mdstate() != 's':
1602 if usemergedriver and not unresolved and ms.mdstate() != 's':
1591 if not driverconclude(repo, ms, wctx, labels=labels):
1603 if not driverconclude(repo, ms, wctx, labels=labels):
1592 # XXX setting unresolved to at least 1 is a hack to make sure we
1604 # XXX setting unresolved to at least 1 is a hack to make sure we
1593 # error out
1605 # error out
1594 unresolved = max(unresolved, 1)
1606 unresolved = max(unresolved, 1)
1595
1607
1596 ms.commit()
1608 ms.commit()
1597
1609
1598 msupdated, msmerged, msremoved = ms.counts()
1610 msupdated, msmerged, msremoved = ms.counts()
1599 updated += msupdated
1611 updated += msupdated
1600 merged += msmerged
1612 merged += msmerged
1601 removed += msremoved
1613 removed += msremoved
1602
1614
1603 extraactions = ms.actions()
1615 extraactions = ms.actions()
1604 if extraactions:
1616 if extraactions:
1605 mfiles = set(a[0] for a in actions['m'])
1617 mfiles = set(a[0] for a in actions['m'])
1606 for k, acts in extraactions.iteritems():
1618 for k, acts in extraactions.iteritems():
1607 actions[k].extend(acts)
1619 actions[k].extend(acts)
1608 # Remove these files from actions['m'] as well. This is important
1620 # Remove these files from actions['m'] as well. This is important
1609 # because in recordupdates, files in actions['m'] are processed
1621 # because in recordupdates, files in actions['m'] are processed
1610 # after files in other actions, and the merge driver might add
1622 # after files in other actions, and the merge driver might add
1611 # files to those actions via extraactions above. This can lead to a
1623 # files to those actions via extraactions above. This can lead to a
1612 # file being recorded twice, with poor results. This is especially
1624 # file being recorded twice, with poor results. This is especially
1613 # problematic for actions['r'] (currently only possible with the
1625 # problematic for actions['r'] (currently only possible with the
1614 # merge driver in the initial merge process; interrupted merges
1626 # merge driver in the initial merge process; interrupted merges
1615 # don't go through this flow).
1627 # don't go through this flow).
1616 #
1628 #
1617 # The real fix here is to have indexes by both file and action so
1629 # The real fix here is to have indexes by both file and action so
1618 # that when the action for a file is changed it is automatically
1630 # that when the action for a file is changed it is automatically
1619 # reflected in the other action lists. But that involves a more
1631 # reflected in the other action lists. But that involves a more
1620 # complex data structure, so this will do for now.
1632 # complex data structure, so this will do for now.
1621 #
1633 #
1622 # We don't need to do the same operation for 'dc' and 'cd' because
1634 # We don't need to do the same operation for 'dc' and 'cd' because
1623 # those lists aren't consulted again.
1635 # those lists aren't consulted again.
1624 mfiles.difference_update(a[0] for a in acts)
1636 mfiles.difference_update(a[0] for a in acts)
1625
1637
1626 actions['m'] = [a for a in actions['m'] if a[0] in mfiles]
1638 actions['m'] = [a for a in actions['m'] if a[0] in mfiles]
1627
1639
1628 progress(_updating, None, total=numupdates, unit=_files)
1640 progress(_updating, None, total=numupdates, unit=_files)
1629
1641
1630 return updated, merged, removed, unresolved
1642 return updated, merged, removed, unresolved
1631
1643
1632 def recordupdates(repo, actions, branchmerge):
1644 def recordupdates(repo, actions, branchmerge):
1633 "record merge actions to the dirstate"
1645 "record merge actions to the dirstate"
1634 # remove (must come first)
1646 # remove (must come first)
1635 for f, args, msg in actions.get('r', []):
1647 for f, args, msg in actions.get('r', []):
1636 if branchmerge:
1648 if branchmerge:
1637 repo.dirstate.remove(f)
1649 repo.dirstate.remove(f)
1638 else:
1650 else:
1639 repo.dirstate.drop(f)
1651 repo.dirstate.drop(f)
1640
1652
1641 # forget (must come first)
1653 # forget (must come first)
1642 for f, args, msg in actions.get('f', []):
1654 for f, args, msg in actions.get('f', []):
1643 repo.dirstate.drop(f)
1655 repo.dirstate.drop(f)
1644
1656
1645 # resolve path conflicts
1657 # resolve path conflicts
1646 for f, args, msg in actions.get('pr', []):
1658 for f, args, msg in actions.get('pr', []):
1647 f0, = args
1659 f0, = args
1648 origf0 = repo.dirstate.copied(f0) or f0
1660 origf0 = repo.dirstate.copied(f0) or f0
1649 repo.dirstate.add(f)
1661 repo.dirstate.add(f)
1650 repo.dirstate.copy(origf0, f)
1662 repo.dirstate.copy(origf0, f)
1651 if f0 == origf0:
1663 if f0 == origf0:
1652 repo.dirstate.remove(f0)
1664 repo.dirstate.remove(f0)
1653 else:
1665 else:
1654 repo.dirstate.drop(f0)
1666 repo.dirstate.drop(f0)
1655
1667
1656 # re-add
1668 # re-add
1657 for f, args, msg in actions.get('a', []):
1669 for f, args, msg in actions.get('a', []):
1658 repo.dirstate.add(f)
1670 repo.dirstate.add(f)
1659
1671
1660 # re-add/mark as modified
1672 # re-add/mark as modified
1661 for f, args, msg in actions.get('am', []):
1673 for f, args, msg in actions.get('am', []):
1662 if branchmerge:
1674 if branchmerge:
1663 repo.dirstate.normallookup(f)
1675 repo.dirstate.normallookup(f)
1664 else:
1676 else:
1665 repo.dirstate.add(f)
1677 repo.dirstate.add(f)
1666
1678
1667 # exec change
1679 # exec change
1668 for f, args, msg in actions.get('e', []):
1680 for f, args, msg in actions.get('e', []):
1669 repo.dirstate.normallookup(f)
1681 repo.dirstate.normallookup(f)
1670
1682
1671 # keep
1683 # keep
1672 for f, args, msg in actions.get('k', []):
1684 for f, args, msg in actions.get('k', []):
1673 pass
1685 pass
1674
1686
1675 # get
1687 # get
1676 for f, args, msg in actions.get('g', []):
1688 for f, args, msg in actions.get('g', []):
1677 if branchmerge:
1689 if branchmerge:
1678 repo.dirstate.otherparent(f)
1690 repo.dirstate.otherparent(f)
1679 else:
1691 else:
1680 repo.dirstate.normal(f)
1692 repo.dirstate.normal(f)
1681
1693
1682 # merge
1694 # merge
1683 for f, args, msg in actions.get('m', []):
1695 for f, args, msg in actions.get('m', []):
1684 f1, f2, fa, move, anc = args
1696 f1, f2, fa, move, anc = args
1685 if branchmerge:
1697 if branchmerge:
1686 # We've done a branch merge, mark this file as merged
1698 # We've done a branch merge, mark this file as merged
1687 # so that we properly record the merger later
1699 # so that we properly record the merger later
1688 repo.dirstate.merge(f)
1700 repo.dirstate.merge(f)
1689 if f1 != f2: # copy/rename
1701 if f1 != f2: # copy/rename
1690 if move:
1702 if move:
1691 repo.dirstate.remove(f1)
1703 repo.dirstate.remove(f1)
1692 if f1 != f:
1704 if f1 != f:
1693 repo.dirstate.copy(f1, f)
1705 repo.dirstate.copy(f1, f)
1694 else:
1706 else:
1695 repo.dirstate.copy(f2, f)
1707 repo.dirstate.copy(f2, f)
1696 else:
1708 else:
1697 # We've update-merged a locally modified file, so
1709 # We've update-merged a locally modified file, so
1698 # we set the dirstate to emulate a normal checkout
1710 # we set the dirstate to emulate a normal checkout
1699 # of that file some time in the past. Thus our
1711 # of that file some time in the past. Thus our
1700 # merge will appear as a normal local file
1712 # merge will appear as a normal local file
1701 # modification.
1713 # modification.
1702 if f2 == f: # file not locally copied/moved
1714 if f2 == f: # file not locally copied/moved
1703 repo.dirstate.normallookup(f)
1715 repo.dirstate.normallookup(f)
1704 if move:
1716 if move:
1705 repo.dirstate.drop(f1)
1717 repo.dirstate.drop(f1)
1706
1718
1707 # directory rename, move local
1719 # directory rename, move local
1708 for f, args, msg in actions.get('dm', []):
1720 for f, args, msg in actions.get('dm', []):
1709 f0, flag = args
1721 f0, flag = args
1710 if branchmerge:
1722 if branchmerge:
1711 repo.dirstate.add(f)
1723 repo.dirstate.add(f)
1712 repo.dirstate.remove(f0)
1724 repo.dirstate.remove(f0)
1713 repo.dirstate.copy(f0, f)
1725 repo.dirstate.copy(f0, f)
1714 else:
1726 else:
1715 repo.dirstate.normal(f)
1727 repo.dirstate.normal(f)
1716 repo.dirstate.drop(f0)
1728 repo.dirstate.drop(f0)
1717
1729
1718 # directory rename, get
1730 # directory rename, get
1719 for f, args, msg in actions.get('dg', []):
1731 for f, args, msg in actions.get('dg', []):
1720 f0, flag = args
1732 f0, flag = args
1721 if branchmerge:
1733 if branchmerge:
1722 repo.dirstate.add(f)
1734 repo.dirstate.add(f)
1723 repo.dirstate.copy(f0, f)
1735 repo.dirstate.copy(f0, f)
1724 else:
1736 else:
1725 repo.dirstate.normal(f)
1737 repo.dirstate.normal(f)
1726
1738
1727 def update(repo, node, branchmerge, force, ancestor=None,
1739 def update(repo, node, branchmerge, force, ancestor=None,
1728 mergeancestor=False, labels=None, matcher=None, mergeforce=False,
1740 mergeancestor=False, labels=None, matcher=None, mergeforce=False,
1729 updatecheck=None, wc=None):
1741 updatecheck=None, wc=None):
1730 """
1742 """
1731 Perform a merge between the working directory and the given node
1743 Perform a merge between the working directory and the given node
1732
1744
1733 node = the node to update to
1745 node = the node to update to
1734 branchmerge = whether to merge between branches
1746 branchmerge = whether to merge between branches
1735 force = whether to force branch merging or file overwriting
1747 force = whether to force branch merging or file overwriting
1736 matcher = a matcher to filter file lists (dirstate not updated)
1748 matcher = a matcher to filter file lists (dirstate not updated)
1737 mergeancestor = whether it is merging with an ancestor. If true,
1749 mergeancestor = whether it is merging with an ancestor. If true,
1738 we should accept the incoming changes for any prompts that occur.
1750 we should accept the incoming changes for any prompts that occur.
1739 If false, merging with an ancestor (fast-forward) is only allowed
1751 If false, merging with an ancestor (fast-forward) is only allowed
1740 between different named branches. This flag is used by rebase extension
1752 between different named branches. This flag is used by rebase extension
1741 as a temporary fix and should be avoided in general.
1753 as a temporary fix and should be avoided in general.
1742 labels = labels to use for base, local and other
1754 labels = labels to use for base, local and other
1743 mergeforce = whether the merge was run with 'merge --force' (deprecated): if
1755 mergeforce = whether the merge was run with 'merge --force' (deprecated): if
1744 this is True, then 'force' should be True as well.
1756 this is True, then 'force' should be True as well.
1745
1757
1746 The table below shows all the behaviors of the update command given the
1758 The table below shows all the behaviors of the update command given the
1747 -c/--check and -C/--clean or no options, whether the working directory is
1759 -c/--check and -C/--clean or no options, whether the working directory is
1748 dirty, whether a revision is specified, and the relationship of the parent
1760 dirty, whether a revision is specified, and the relationship of the parent
1749 rev to the target rev (linear or not). Match from top first. The -n
1761 rev to the target rev (linear or not). Match from top first. The -n
1750 option doesn't exist on the command line, but represents the
1762 option doesn't exist on the command line, but represents the
1751 experimental.updatecheck=noconflict option.
1763 experimental.updatecheck=noconflict option.
1752
1764
1753 This logic is tested by test-update-branches.t.
1765 This logic is tested by test-update-branches.t.
1754
1766
1755 -c -C -n -m dirty rev linear | result
1767 -c -C -n -m dirty rev linear | result
1756 y y * * * * * | (1)
1768 y y * * * * * | (1)
1757 y * y * * * * | (1)
1769 y * y * * * * | (1)
1758 y * * y * * * | (1)
1770 y * * y * * * | (1)
1759 * y y * * * * | (1)
1771 * y y * * * * | (1)
1760 * y * y * * * | (1)
1772 * y * y * * * | (1)
1761 * * y y * * * | (1)
1773 * * y y * * * | (1)
1762 * * * * * n n | x
1774 * * * * * n n | x
1763 * * * * n * * | ok
1775 * * * * n * * | ok
1764 n n n n y * y | merge
1776 n n n n y * y | merge
1765 n n n n y y n | (2)
1777 n n n n y y n | (2)
1766 n n n y y * * | merge
1778 n n n y y * * | merge
1767 n n y n y * * | merge if no conflict
1779 n n y n y * * | merge if no conflict
1768 n y n n y * * | discard
1780 n y n n y * * | discard
1769 y n n n y * * | (3)
1781 y n n n y * * | (3)
1770
1782
1771 x = can't happen
1783 x = can't happen
1772 * = don't-care
1784 * = don't-care
1773 1 = incompatible options (checked in commands.py)
1785 1 = incompatible options (checked in commands.py)
1774 2 = abort: uncommitted changes (commit or update --clean to discard changes)
1786 2 = abort: uncommitted changes (commit or update --clean to discard changes)
1775 3 = abort: uncommitted changes (checked in commands.py)
1787 3 = abort: uncommitted changes (checked in commands.py)
1776
1788
1777 The merge is performed inside ``wc``, a workingctx-like objects. It defaults
1789 The merge is performed inside ``wc``, a workingctx-like objects. It defaults
1778 to repo[None] if None is passed.
1790 to repo[None] if None is passed.
1779
1791
1780 Return the same tuple as applyupdates().
1792 Return the same tuple as applyupdates().
1781 """
1793 """
1782 # Avoid cycle.
1794 # Avoid cycle.
1783 from . import sparse
1795 from . import sparse
1784
1796
1785 # This function used to find the default destination if node was None, but
1797 # This function used to find the default destination if node was None, but
1786 # that's now in destutil.py.
1798 # that's now in destutil.py.
1787 assert node is not None
1799 assert node is not None
1788 if not branchmerge and not force:
1800 if not branchmerge and not force:
1789 # TODO: remove the default once all callers that pass branchmerge=False
1801 # TODO: remove the default once all callers that pass branchmerge=False
1790 # and force=False pass a value for updatecheck. We may want to allow
1802 # and force=False pass a value for updatecheck. We may want to allow
1791 # updatecheck='abort' to better suppport some of these callers.
1803 # updatecheck='abort' to better suppport some of these callers.
1792 if updatecheck is None:
1804 if updatecheck is None:
1793 updatecheck = 'linear'
1805 updatecheck = 'linear'
1794 assert updatecheck in ('none', 'linear', 'noconflict')
1806 assert updatecheck in ('none', 'linear', 'noconflict')
1795 # If we're doing a partial update, we need to skip updating
1807 # If we're doing a partial update, we need to skip updating
1796 # the dirstate, so make a note of any partial-ness to the
1808 # the dirstate, so make a note of any partial-ness to the
1797 # update here.
1809 # update here.
1798 if matcher is None or matcher.always():
1810 if matcher is None or matcher.always():
1799 partial = False
1811 partial = False
1800 else:
1812 else:
1801 partial = True
1813 partial = True
1802 with repo.wlock():
1814 with repo.wlock():
1803 if wc is None:
1815 if wc is None:
1804 wc = repo[None]
1816 wc = repo[None]
1805 pl = wc.parents()
1817 pl = wc.parents()
1806 p1 = pl[0]
1818 p1 = pl[0]
1807 pas = [None]
1819 pas = [None]
1808 if ancestor is not None:
1820 if ancestor is not None:
1809 pas = [repo[ancestor]]
1821 pas = [repo[ancestor]]
1810
1822
1811 overwrite = force and not branchmerge
1823 overwrite = force and not branchmerge
1812
1824
1813 p2 = repo[node]
1825 p2 = repo[node]
1814 if pas[0] is None:
1826 if pas[0] is None:
1815 if repo.ui.configlist('merge', 'preferancestor') == ['*']:
1827 if repo.ui.configlist('merge', 'preferancestor') == ['*']:
1816 cahs = repo.changelog.commonancestorsheads(p1.node(), p2.node())
1828 cahs = repo.changelog.commonancestorsheads(p1.node(), p2.node())
1817 pas = [repo[anc] for anc in (sorted(cahs) or [nullid])]
1829 pas = [repo[anc] for anc in (sorted(cahs) or [nullid])]
1818 else:
1830 else:
1819 pas = [p1.ancestor(p2, warn=branchmerge)]
1831 pas = [p1.ancestor(p2, warn=branchmerge)]
1820
1832
1821 fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2)
1833 fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2)
1822
1834
1823 ### check phase
1835 ### check phase
1824 if not overwrite:
1836 if not overwrite:
1825 if len(pl) > 1:
1837 if len(pl) > 1:
1826 raise error.Abort(_("outstanding uncommitted merge"))
1838 raise error.Abort(_("outstanding uncommitted merge"))
1827 ms = mergestate.read(repo)
1839 ms = mergestate.read(repo)
1828 if list(ms.unresolved()):
1840 if list(ms.unresolved()):
1829 raise error.Abort(_("outstanding merge conflicts"))
1841 raise error.Abort(_("outstanding merge conflicts"))
1830 if branchmerge:
1842 if branchmerge:
1831 if pas == [p2]:
1843 if pas == [p2]:
1832 raise error.Abort(_("merging with a working directory ancestor"
1844 raise error.Abort(_("merging with a working directory ancestor"
1833 " has no effect"))
1845 " has no effect"))
1834 elif pas == [p1]:
1846 elif pas == [p1]:
1835 if not mergeancestor and wc.branch() == p2.branch():
1847 if not mergeancestor and wc.branch() == p2.branch():
1836 raise error.Abort(_("nothing to merge"),
1848 raise error.Abort(_("nothing to merge"),
1837 hint=_("use 'hg update' "
1849 hint=_("use 'hg update' "
1838 "or check 'hg heads'"))
1850 "or check 'hg heads'"))
1839 if not force and (wc.files() or wc.deleted()):
1851 if not force and (wc.files() or wc.deleted()):
1840 raise error.Abort(_("uncommitted changes"),
1852 raise error.Abort(_("uncommitted changes"),
1841 hint=_("use 'hg status' to list changes"))
1853 hint=_("use 'hg status' to list changes"))
1842 if not wc.isinmemory():
1854 if not wc.isinmemory():
1843 for s in sorted(wc.substate):
1855 for s in sorted(wc.substate):
1844 wc.sub(s).bailifchanged()
1856 wc.sub(s).bailifchanged()
1845
1857
1846 elif not overwrite:
1858 elif not overwrite:
1847 if p1 == p2: # no-op update
1859 if p1 == p2: # no-op update
1848 # call the hooks and exit early
1860 # call the hooks and exit early
1849 repo.hook('preupdate', throw=True, parent1=xp2, parent2='')
1861 repo.hook('preupdate', throw=True, parent1=xp2, parent2='')
1850 repo.hook('update', parent1=xp2, parent2='', error=0)
1862 repo.hook('update', parent1=xp2, parent2='', error=0)
1851 return 0, 0, 0, 0
1863 return 0, 0, 0, 0
1852
1864
1853 if (updatecheck == 'linear' and
1865 if (updatecheck == 'linear' and
1854 pas not in ([p1], [p2])): # nonlinear
1866 pas not in ([p1], [p2])): # nonlinear
1855 dirty = wc.dirty(missing=True)
1867 dirty = wc.dirty(missing=True)
1856 if dirty:
1868 if dirty:
1857 # Branching is a bit strange to ensure we do the minimal
1869 # Branching is a bit strange to ensure we do the minimal
1858 # amount of call to obsutil.foreground.
1870 # amount of call to obsutil.foreground.
1859 foreground = obsutil.foreground(repo, [p1.node()])
1871 foreground = obsutil.foreground(repo, [p1.node()])
1860 # note: the <node> variable contains a random identifier
1872 # note: the <node> variable contains a random identifier
1861 if repo[node].node() in foreground:
1873 if repo[node].node() in foreground:
1862 pass # allow updating to successors
1874 pass # allow updating to successors
1863 else:
1875 else:
1864 msg = _("uncommitted changes")
1876 msg = _("uncommitted changes")
1865 hint = _("commit or update --clean to discard changes")
1877 hint = _("commit or update --clean to discard changes")
1866 raise error.UpdateAbort(msg, hint=hint)
1878 raise error.UpdateAbort(msg, hint=hint)
1867 else:
1879 else:
1868 # Allow jumping branches if clean and specific rev given
1880 # Allow jumping branches if clean and specific rev given
1869 pass
1881 pass
1870
1882
1871 if overwrite:
1883 if overwrite:
1872 pas = [wc]
1884 pas = [wc]
1873 elif not branchmerge:
1885 elif not branchmerge:
1874 pas = [p1]
1886 pas = [p1]
1875
1887
1876 # deprecated config: merge.followcopies
1888 # deprecated config: merge.followcopies
1877 followcopies = repo.ui.configbool('merge', 'followcopies')
1889 followcopies = repo.ui.configbool('merge', 'followcopies')
1878 if overwrite:
1890 if overwrite:
1879 followcopies = False
1891 followcopies = False
1880 elif not pas[0]:
1892 elif not pas[0]:
1881 followcopies = False
1893 followcopies = False
1882 if not branchmerge and not wc.dirty(missing=True):
1894 if not branchmerge and not wc.dirty(missing=True):
1883 followcopies = False
1895 followcopies = False
1884
1896
1885 ### calculate phase
1897 ### calculate phase
1886 actionbyfile, diverge, renamedelete = calculateupdates(
1898 actionbyfile, diverge, renamedelete = calculateupdates(
1887 repo, wc, p2, pas, branchmerge, force, mergeancestor,
1899 repo, wc, p2, pas, branchmerge, force, mergeancestor,
1888 followcopies, matcher=matcher, mergeforce=mergeforce)
1900 followcopies, matcher=matcher, mergeforce=mergeforce)
1889
1901
1890 if updatecheck == 'noconflict':
1902 if updatecheck == 'noconflict':
1891 for f, (m, args, msg) in actionbyfile.iteritems():
1903 for f, (m, args, msg) in actionbyfile.iteritems():
1892 if m not in ('g', 'k', 'e', 'r', 'pr'):
1904 if m not in ('g', 'k', 'e', 'r', 'pr'):
1893 msg = _("conflicting changes")
1905 msg = _("conflicting changes")
1894 hint = _("commit or update --clean to discard changes")
1906 hint = _("commit or update --clean to discard changes")
1895 raise error.Abort(msg, hint=hint)
1907 raise error.Abort(msg, hint=hint)
1896
1908
1897 # Prompt and create actions. Most of this is in the resolve phase
1909 # Prompt and create actions. Most of this is in the resolve phase
1898 # already, but we can't handle .hgsubstate in filemerge or
1910 # already, but we can't handle .hgsubstate in filemerge or
1899 # subrepo.submerge yet so we have to keep prompting for it.
1911 # subrepo.submerge yet so we have to keep prompting for it.
1900 if '.hgsubstate' in actionbyfile:
1912 if '.hgsubstate' in actionbyfile:
1901 f = '.hgsubstate'
1913 f = '.hgsubstate'
1902 m, args, msg = actionbyfile[f]
1914 m, args, msg = actionbyfile[f]
1903 prompts = filemerge.partextras(labels)
1915 prompts = filemerge.partextras(labels)
1904 prompts['f'] = f
1916 prompts['f'] = f
1905 if m == 'cd':
1917 if m == 'cd':
1906 if repo.ui.promptchoice(
1918 if repo.ui.promptchoice(
1907 _("local%(l)s changed %(f)s which other%(o)s deleted\n"
1919 _("local%(l)s changed %(f)s which other%(o)s deleted\n"
1908 "use (c)hanged version or (d)elete?"
1920 "use (c)hanged version or (d)elete?"
1909 "$$ &Changed $$ &Delete") % prompts, 0):
1921 "$$ &Changed $$ &Delete") % prompts, 0):
1910 actionbyfile[f] = ('r', None, "prompt delete")
1922 actionbyfile[f] = ('r', None, "prompt delete")
1911 elif f in p1:
1923 elif f in p1:
1912 actionbyfile[f] = ('am', None, "prompt keep")
1924 actionbyfile[f] = ('am', None, "prompt keep")
1913 else:
1925 else:
1914 actionbyfile[f] = ('a', None, "prompt keep")
1926 actionbyfile[f] = ('a', None, "prompt keep")
1915 elif m == 'dc':
1927 elif m == 'dc':
1916 f1, f2, fa, move, anc = args
1928 f1, f2, fa, move, anc = args
1917 flags = p2[f2].flags()
1929 flags = p2[f2].flags()
1918 if repo.ui.promptchoice(
1930 if repo.ui.promptchoice(
1919 _("other%(o)s changed %(f)s which local%(l)s deleted\n"
1931 _("other%(o)s changed %(f)s which local%(l)s deleted\n"
1920 "use (c)hanged version or leave (d)eleted?"
1932 "use (c)hanged version or leave (d)eleted?"
1921 "$$ &Changed $$ &Deleted") % prompts, 0) == 0:
1933 "$$ &Changed $$ &Deleted") % prompts, 0) == 0:
1922 actionbyfile[f] = ('g', (flags, False), "prompt recreating")
1934 actionbyfile[f] = ('g', (flags, False), "prompt recreating")
1923 else:
1935 else:
1924 del actionbyfile[f]
1936 del actionbyfile[f]
1925
1937
1926 # Convert to dictionary-of-lists format
1938 # Convert to dictionary-of-lists format
1927 actions = dict((m, [])
1939 actions = dict((m, [])
1928 for m in 'a am f g cd dc r dm dg m e k p pr'.split())
1940 for m in 'a am f g cd dc r dm dg m e k p pr'.split())
1929 for f, (m, args, msg) in actionbyfile.iteritems():
1941 for f, (m, args, msg) in actionbyfile.iteritems():
1930 if m not in actions:
1942 if m not in actions:
1931 actions[m] = []
1943 actions[m] = []
1932 actions[m].append((f, args, msg))
1944 actions[m].append((f, args, msg))
1933
1945
1934 if not util.fscasesensitive(repo.path):
1946 if not util.fscasesensitive(repo.path):
1935 # check collision between files only in p2 for clean update
1947 # check collision between files only in p2 for clean update
1936 if (not branchmerge and
1948 if (not branchmerge and
1937 (force or not wc.dirty(missing=True, branch=False))):
1949 (force or not wc.dirty(missing=True, branch=False))):
1938 _checkcollision(repo, p2.manifest(), None)
1950 _checkcollision(repo, p2.manifest(), None)
1939 else:
1951 else:
1940 _checkcollision(repo, wc.manifest(), actions)
1952 _checkcollision(repo, wc.manifest(), actions)
1941
1953
1942 # divergent renames
1954 # divergent renames
1943 for f, fl in sorted(diverge.iteritems()):
1955 for f, fl in sorted(diverge.iteritems()):
1944 repo.ui.warn(_("note: possible conflict - %s was renamed "
1956 repo.ui.warn(_("note: possible conflict - %s was renamed "
1945 "multiple times to:\n") % f)
1957 "multiple times to:\n") % f)
1946 for nf in fl:
1958 for nf in fl:
1947 repo.ui.warn(" %s\n" % nf)
1959 repo.ui.warn(" %s\n" % nf)
1948
1960
1949 # rename and delete
1961 # rename and delete
1950 for f, fl in sorted(renamedelete.iteritems()):
1962 for f, fl in sorted(renamedelete.iteritems()):
1951 repo.ui.warn(_("note: possible conflict - %s was deleted "
1963 repo.ui.warn(_("note: possible conflict - %s was deleted "
1952 "and renamed to:\n") % f)
1964 "and renamed to:\n") % f)
1953 for nf in fl:
1965 for nf in fl:
1954 repo.ui.warn(" %s\n" % nf)
1966 repo.ui.warn(" %s\n" % nf)
1955
1967
1956 ### apply phase
1968 ### apply phase
1957 if not branchmerge: # just jump to the new rev
1969 if not branchmerge: # just jump to the new rev
1958 fp1, fp2, xp1, xp2 = fp2, nullid, xp2, ''
1970 fp1, fp2, xp1, xp2 = fp2, nullid, xp2, ''
1959 if not partial and not wc.isinmemory():
1971 if not partial and not wc.isinmemory():
1960 repo.hook('preupdate', throw=True, parent1=xp1, parent2=xp2)
1972 repo.hook('preupdate', throw=True, parent1=xp1, parent2=xp2)
1961 # note that we're in the middle of an update
1973 # note that we're in the middle of an update
1962 repo.vfs.write('updatestate', p2.hex())
1974 repo.vfs.write('updatestate', p2.hex())
1963
1975
1964 # Advertise fsmonitor when its presence could be useful.
1976 # Advertise fsmonitor when its presence could be useful.
1965 #
1977 #
1966 # We only advertise when performing an update from an empty working
1978 # We only advertise when performing an update from an empty working
1967 # directory. This typically only occurs during initial clone.
1979 # directory. This typically only occurs during initial clone.
1968 #
1980 #
1969 # We give users a mechanism to disable the warning in case it is
1981 # We give users a mechanism to disable the warning in case it is
1970 # annoying.
1982 # annoying.
1971 #
1983 #
1972 # We only allow on Linux and MacOS because that's where fsmonitor is
1984 # We only allow on Linux and MacOS because that's where fsmonitor is
1973 # considered stable.
1985 # considered stable.
1974 fsmonitorwarning = repo.ui.configbool('fsmonitor', 'warn_when_unused')
1986 fsmonitorwarning = repo.ui.configbool('fsmonitor', 'warn_when_unused')
1975 fsmonitorthreshold = repo.ui.configint('fsmonitor',
1987 fsmonitorthreshold = repo.ui.configint('fsmonitor',
1976 'warn_update_file_count')
1988 'warn_update_file_count')
1977 try:
1989 try:
1978 extensions.find('fsmonitor')
1990 extensions.find('fsmonitor')
1979 fsmonitorenabled = repo.ui.config('fsmonitor', 'mode') != 'off'
1991 fsmonitorenabled = repo.ui.config('fsmonitor', 'mode') != 'off'
1980 # We intentionally don't look at whether fsmonitor has disabled
1992 # We intentionally don't look at whether fsmonitor has disabled
1981 # itself because a) fsmonitor may have already printed a warning
1993 # itself because a) fsmonitor may have already printed a warning
1982 # b) we only care about the config state here.
1994 # b) we only care about the config state here.
1983 except KeyError:
1995 except KeyError:
1984 fsmonitorenabled = False
1996 fsmonitorenabled = False
1985
1997
1986 if (fsmonitorwarning
1998 if (fsmonitorwarning
1987 and not fsmonitorenabled
1999 and not fsmonitorenabled
1988 and p1.node() == nullid
2000 and p1.node() == nullid
1989 and len(actions['g']) >= fsmonitorthreshold
2001 and len(actions['g']) >= fsmonitorthreshold
1990 and pycompat.sysplatform.startswith(('linux', 'darwin'))):
2002 and pycompat.sysplatform.startswith(('linux', 'darwin'))):
1991 repo.ui.warn(
2003 repo.ui.warn(
1992 _('(warning: large working directory being used without '
2004 _('(warning: large working directory being used without '
1993 'fsmonitor enabled; enable fsmonitor to improve performance; '
2005 'fsmonitor enabled; enable fsmonitor to improve performance; '
1994 'see "hg help -e fsmonitor")\n'))
2006 'see "hg help -e fsmonitor")\n'))
1995
2007
1996 stats = applyupdates(repo, actions, wc, p2, overwrite, labels=labels)
2008 stats = applyupdates(repo, actions, wc, p2, overwrite, labels=labels)
1997
2009
1998 if not partial and not wc.isinmemory():
2010 if not partial and not wc.isinmemory():
1999 with repo.dirstate.parentchange():
2011 with repo.dirstate.parentchange():
2000 repo.setparents(fp1, fp2)
2012 repo.setparents(fp1, fp2)
2001 recordupdates(repo, actions, branchmerge)
2013 recordupdates(repo, actions, branchmerge)
2002 # update completed, clear state
2014 # update completed, clear state
2003 util.unlink(repo.vfs.join('updatestate'))
2015 util.unlink(repo.vfs.join('updatestate'))
2004
2016
2005 if not branchmerge:
2017 if not branchmerge:
2006 repo.dirstate.setbranch(p2.branch())
2018 repo.dirstate.setbranch(p2.branch())
2007
2019
2008 # If we're updating to a location, clean up any stale temporary includes
2020 # If we're updating to a location, clean up any stale temporary includes
2009 # (ex: this happens during hg rebase --abort).
2021 # (ex: this happens during hg rebase --abort).
2010 if not branchmerge:
2022 if not branchmerge:
2011 sparse.prunetemporaryincludes(repo)
2023 sparse.prunetemporaryincludes(repo)
2012
2024
2013 if not partial:
2025 if not partial:
2014 repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3])
2026 repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3])
2015 return stats
2027 return stats
2016
2028
2017 def graft(repo, ctx, pctx, labels, keepparent=False):
2029 def graft(repo, ctx, pctx, labels, keepparent=False):
2018 """Do a graft-like merge.
2030 """Do a graft-like merge.
2019
2031
2020 This is a merge where the merge ancestor is chosen such that one
2032 This is a merge where the merge ancestor is chosen such that one
2021 or more changesets are grafted onto the current changeset. In
2033 or more changesets are grafted onto the current changeset. In
2022 addition to the merge, this fixes up the dirstate to include only
2034 addition to the merge, this fixes up the dirstate to include only
2023 a single parent (if keepparent is False) and tries to duplicate any
2035 a single parent (if keepparent is False) and tries to duplicate any
2024 renames/copies appropriately.
2036 renames/copies appropriately.
2025
2037
2026 ctx - changeset to rebase
2038 ctx - changeset to rebase
2027 pctx - merge base, usually ctx.p1()
2039 pctx - merge base, usually ctx.p1()
2028 labels - merge labels eg ['local', 'graft']
2040 labels - merge labels eg ['local', 'graft']
2029 keepparent - keep second parent if any
2041 keepparent - keep second parent if any
2030
2042
2031 """
2043 """
2032 # If we're grafting a descendant onto an ancestor, be sure to pass
2044 # If we're grafting a descendant onto an ancestor, be sure to pass
2033 # mergeancestor=True to update. This does two things: 1) allows the merge if
2045 # mergeancestor=True to update. This does two things: 1) allows the merge if
2034 # the destination is the same as the parent of the ctx (so we can use graft
2046 # the destination is the same as the parent of the ctx (so we can use graft
2035 # to copy commits), and 2) informs update that the incoming changes are
2047 # to copy commits), and 2) informs update that the incoming changes are
2036 # newer than the destination so it doesn't prompt about "remote changed foo
2048 # newer than the destination so it doesn't prompt about "remote changed foo
2037 # which local deleted".
2049 # which local deleted".
2038 mergeancestor = repo.changelog.isancestor(repo['.'].node(), ctx.node())
2050 mergeancestor = repo.changelog.isancestor(repo['.'].node(), ctx.node())
2039
2051
2040 stats = update(repo, ctx.node(), True, True, pctx.node(),
2052 stats = update(repo, ctx.node(), True, True, pctx.node(),
2041 mergeancestor=mergeancestor, labels=labels)
2053 mergeancestor=mergeancestor, labels=labels)
2042
2054
2043 pother = nullid
2055 pother = nullid
2044 parents = ctx.parents()
2056 parents = ctx.parents()
2045 if keepparent and len(parents) == 2 and pctx in parents:
2057 if keepparent and len(parents) == 2 and pctx in parents:
2046 parents.remove(pctx)
2058 parents.remove(pctx)
2047 pother = parents[0].node()
2059 pother = parents[0].node()
2048
2060
2049 with repo.dirstate.parentchange():
2061 with repo.dirstate.parentchange():
2050 repo.setparents(repo['.'].node(), pother)
2062 repo.setparents(repo['.'].node(), pother)
2051 repo.dirstate.write(repo.currenttransaction())
2063 repo.dirstate.write(repo.currenttransaction())
2052 # fix up dirstate for copies and renames
2064 # fix up dirstate for copies and renames
2053 copies.duplicatecopies(repo, repo[None], ctx.rev(), pctx.rev())
2065 copies.duplicatecopies(repo, repo[None], ctx.rev(), pctx.rev())
2054 return stats
2066 return stats
General Comments 0
You need to be logged in to leave comments. Login now