##// END OF EJS Templates
merge: split out mostly-non-interactive working dir updates...
Bryan O'Sullivan -
r18630:9b9e2d9e default
parent child Browse files
Show More
@@ -1,667 +1,688 b''
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 node import nullid, nullrev, hex, bin
8 from node import nullid, nullrev, hex, bin
9 from i18n import _
9 from i18n import _
10 import error, util, filemerge, copies, subrepo
10 import error, util, filemerge, copies, subrepo
11 import errno, os, shutil
11 import errno, os, shutil
12
12
13 class mergestate(object):
13 class mergestate(object):
14 '''track 3-way merge state of individual files'''
14 '''track 3-way merge state of individual files'''
15 def __init__(self, repo):
15 def __init__(self, repo):
16 self._repo = repo
16 self._repo = repo
17 self._dirty = False
17 self._dirty = False
18 self._read()
18 self._read()
19 def reset(self, node=None):
19 def reset(self, node=None):
20 self._state = {}
20 self._state = {}
21 if node:
21 if node:
22 self._local = node
22 self._local = node
23 shutil.rmtree(self._repo.join("merge"), True)
23 shutil.rmtree(self._repo.join("merge"), True)
24 self._dirty = False
24 self._dirty = False
25 def _read(self):
25 def _read(self):
26 self._state = {}
26 self._state = {}
27 try:
27 try:
28 f = self._repo.opener("merge/state")
28 f = self._repo.opener("merge/state")
29 for i, l in enumerate(f):
29 for i, l in enumerate(f):
30 if i == 0:
30 if i == 0:
31 self._local = bin(l[:-1])
31 self._local = bin(l[:-1])
32 else:
32 else:
33 bits = l[:-1].split("\0")
33 bits = l[:-1].split("\0")
34 self._state[bits[0]] = bits[1:]
34 self._state[bits[0]] = bits[1:]
35 f.close()
35 f.close()
36 except IOError, err:
36 except IOError, err:
37 if err.errno != errno.ENOENT:
37 if err.errno != errno.ENOENT:
38 raise
38 raise
39 self._dirty = False
39 self._dirty = False
40 def commit(self):
40 def commit(self):
41 if self._dirty:
41 if self._dirty:
42 f = self._repo.opener("merge/state", "w")
42 f = self._repo.opener("merge/state", "w")
43 f.write(hex(self._local) + "\n")
43 f.write(hex(self._local) + "\n")
44 for d, v in self._state.iteritems():
44 for d, v in self._state.iteritems():
45 f.write("\0".join([d] + v) + "\n")
45 f.write("\0".join([d] + v) + "\n")
46 f.close()
46 f.close()
47 self._dirty = False
47 self._dirty = False
48 def add(self, fcl, fco, fca, fd):
48 def add(self, fcl, fco, fca, fd):
49 hash = util.sha1(fcl.path()).hexdigest()
49 hash = util.sha1(fcl.path()).hexdigest()
50 self._repo.opener.write("merge/" + hash, fcl.data())
50 self._repo.opener.write("merge/" + hash, fcl.data())
51 self._state[fd] = ['u', hash, fcl.path(), fca.path(),
51 self._state[fd] = ['u', hash, fcl.path(), fca.path(),
52 hex(fca.filenode()), fco.path(), fcl.flags()]
52 hex(fca.filenode()), fco.path(), fcl.flags()]
53 self._dirty = True
53 self._dirty = True
54 def __contains__(self, dfile):
54 def __contains__(self, dfile):
55 return dfile in self._state
55 return dfile in self._state
56 def __getitem__(self, dfile):
56 def __getitem__(self, dfile):
57 return self._state[dfile][0]
57 return self._state[dfile][0]
58 def __iter__(self):
58 def __iter__(self):
59 l = self._state.keys()
59 l = self._state.keys()
60 l.sort()
60 l.sort()
61 for f in l:
61 for f in l:
62 yield f
62 yield f
63 def mark(self, dfile, state):
63 def mark(self, dfile, state):
64 self._state[dfile][0] = state
64 self._state[dfile][0] = state
65 self._dirty = True
65 self._dirty = True
66 def resolve(self, dfile, wctx, octx):
66 def resolve(self, dfile, wctx, octx):
67 if self[dfile] == 'r':
67 if self[dfile] == 'r':
68 return 0
68 return 0
69 state, hash, lfile, afile, anode, ofile, flags = self._state[dfile]
69 state, hash, lfile, afile, anode, ofile, flags = self._state[dfile]
70 fcd = wctx[dfile]
70 fcd = wctx[dfile]
71 fco = octx[ofile]
71 fco = octx[ofile]
72 fca = self._repo.filectx(afile, fileid=anode)
72 fca = self._repo.filectx(afile, fileid=anode)
73 # "premerge" x flags
73 # "premerge" x flags
74 flo = fco.flags()
74 flo = fco.flags()
75 fla = fca.flags()
75 fla = fca.flags()
76 if 'x' in flags + flo + fla and 'l' not in flags + flo + fla:
76 if 'x' in flags + flo + fla and 'l' not in flags + flo + fla:
77 if fca.node() == nullid:
77 if fca.node() == nullid:
78 self._repo.ui.warn(_('warning: cannot merge flags for %s\n') %
78 self._repo.ui.warn(_('warning: cannot merge flags for %s\n') %
79 afile)
79 afile)
80 elif flags == fla:
80 elif flags == fla:
81 flags = flo
81 flags = flo
82 # restore local
82 # restore local
83 f = self._repo.opener("merge/" + hash)
83 f = self._repo.opener("merge/" + hash)
84 self._repo.wwrite(dfile, f.read(), flags)
84 self._repo.wwrite(dfile, f.read(), flags)
85 f.close()
85 f.close()
86 r = filemerge.filemerge(self._repo, self._local, lfile, fcd, fco, fca)
86 r = filemerge.filemerge(self._repo, self._local, lfile, fcd, fco, fca)
87 if r is None:
87 if r is None:
88 # no real conflict
88 # no real conflict
89 del self._state[dfile]
89 del self._state[dfile]
90 elif not r:
90 elif not r:
91 self.mark(dfile, 'r')
91 self.mark(dfile, 'r')
92 return r
92 return r
93
93
94 def _checkunknownfile(repo, wctx, mctx, f):
94 def _checkunknownfile(repo, wctx, mctx, f):
95 return (not repo.dirstate._ignore(f)
95 return (not repo.dirstate._ignore(f)
96 and os.path.isfile(repo.wjoin(f))
96 and os.path.isfile(repo.wjoin(f))
97 and repo.dirstate.normalize(f) not in repo.dirstate
97 and repo.dirstate.normalize(f) not in repo.dirstate
98 and mctx[f].cmp(wctx[f]))
98 and mctx[f].cmp(wctx[f]))
99
99
100 def _checkunknown(repo, wctx, mctx):
100 def _checkunknown(repo, wctx, mctx):
101 "check for collisions between unknown files and files in mctx"
101 "check for collisions between unknown files and files in mctx"
102
102
103 error = False
103 error = False
104 for f in mctx:
104 for f in mctx:
105 if f not in wctx and _checkunknownfile(repo, wctx, mctx, f):
105 if f not in wctx and _checkunknownfile(repo, wctx, mctx, f):
106 error = True
106 error = True
107 wctx._repo.ui.warn(_("%s: untracked file differs\n") % f)
107 wctx._repo.ui.warn(_("%s: untracked file differs\n") % f)
108 if error:
108 if error:
109 raise util.Abort(_("untracked files in working directory differ "
109 raise util.Abort(_("untracked files in working directory differ "
110 "from files in requested revision"))
110 "from files in requested revision"))
111
111
112 def _remains(f, m, ma, workingctx=False):
112 def _remains(f, m, ma, workingctx=False):
113 """check whether specified file remains after merge.
113 """check whether specified file remains after merge.
114
114
115 It is assumed that specified file is not contained in the manifest
115 It is assumed that specified file is not contained in the manifest
116 of the other context.
116 of the other context.
117 """
117 """
118 if f in ma:
118 if f in ma:
119 n = m[f]
119 n = m[f]
120 if n != ma[f]:
120 if n != ma[f]:
121 return True # because it is changed locally
121 return True # because it is changed locally
122 # even though it doesn't remain, if "remote deleted" is
122 # even though it doesn't remain, if "remote deleted" is
123 # chosen in manifestmerge()
123 # chosen in manifestmerge()
124 elif workingctx and n[20:] == "a":
124 elif workingctx and n[20:] == "a":
125 return True # because it is added locally (linear merge specific)
125 return True # because it is added locally (linear merge specific)
126 else:
126 else:
127 return False # because it is removed remotely
127 return False # because it is removed remotely
128 else:
128 else:
129 return True # because it is added locally
129 return True # because it is added locally
130
130
131 def _checkcollision(mctx, extractxs):
131 def _checkcollision(mctx, extractxs):
132 "check for case folding collisions in the destination context"
132 "check for case folding collisions in the destination context"
133 folded = {}
133 folded = {}
134 for fn in mctx:
134 for fn in mctx:
135 fold = util.normcase(fn)
135 fold = util.normcase(fn)
136 if fold in folded:
136 if fold in folded:
137 raise util.Abort(_("case-folding collision between %s and %s")
137 raise util.Abort(_("case-folding collision between %s and %s")
138 % (fn, folded[fold]))
138 % (fn, folded[fold]))
139 folded[fold] = fn
139 folded[fold] = fn
140
140
141 if extractxs:
141 if extractxs:
142 wctx, actx = extractxs
142 wctx, actx = extractxs
143 # class to delay looking up copy mapping
143 # class to delay looking up copy mapping
144 class pathcopies(object):
144 class pathcopies(object):
145 @util.propertycache
145 @util.propertycache
146 def map(self):
146 def map(self):
147 # {dst@mctx: src@wctx} copy mapping
147 # {dst@mctx: src@wctx} copy mapping
148 return copies.pathcopies(wctx, mctx)
148 return copies.pathcopies(wctx, mctx)
149 pc = pathcopies()
149 pc = pathcopies()
150
150
151 for fn in wctx:
151 for fn in wctx:
152 fold = util.normcase(fn)
152 fold = util.normcase(fn)
153 mfn = folded.get(fold, None)
153 mfn = folded.get(fold, None)
154 if (mfn and mfn != fn and pc.map.get(mfn) != fn and
154 if (mfn and mfn != fn and pc.map.get(mfn) != fn and
155 _remains(fn, wctx.manifest(), actx.manifest(), True) and
155 _remains(fn, wctx.manifest(), actx.manifest(), True) and
156 _remains(mfn, mctx.manifest(), actx.manifest())):
156 _remains(mfn, mctx.manifest(), actx.manifest())):
157 raise util.Abort(_("case-folding collision between %s and %s")
157 raise util.Abort(_("case-folding collision between %s and %s")
158 % (mfn, fn))
158 % (mfn, fn))
159
159
160 def _forgetremoved(wctx, mctx, branchmerge):
160 def _forgetremoved(wctx, mctx, branchmerge):
161 """
161 """
162 Forget removed files
162 Forget removed files
163
163
164 If we're jumping between revisions (as opposed to merging), and if
164 If we're jumping between revisions (as opposed to merging), and if
165 neither the working directory nor the target rev has the file,
165 neither the working directory nor the target rev has the file,
166 then we need to remove it from the dirstate, to prevent the
166 then we need to remove it from the dirstate, to prevent the
167 dirstate from listing the file when it is no longer in the
167 dirstate from listing the file when it is no longer in the
168 manifest.
168 manifest.
169
169
170 If we're merging, and the other revision has removed a file
170 If we're merging, and the other revision has removed a file
171 that is not present in the working directory, we need to mark it
171 that is not present in the working directory, we need to mark it
172 as removed.
172 as removed.
173 """
173 """
174
174
175 actions = []
175 actions = []
176 state = branchmerge and 'r' or 'f'
176 state = branchmerge and 'r' or 'f'
177 for f in wctx.deleted():
177 for f in wctx.deleted():
178 if f not in mctx:
178 if f not in mctx:
179 actions.append((f, state, None, "forget deleted"))
179 actions.append((f, state, None, "forget deleted"))
180
180
181 if not branchmerge:
181 if not branchmerge:
182 for f in wctx.removed():
182 for f in wctx.removed():
183 if f not in mctx:
183 if f not in mctx:
184 actions.append((f, "f", None, "forget removed"))
184 actions.append((f, "f", None, "forget removed"))
185
185
186 return actions
186 return actions
187
187
188 def manifestmerge(repo, wctx, p2, pa, branchmerge, force, partial):
188 def manifestmerge(repo, wctx, p2, pa, branchmerge, force, partial):
189 """
189 """
190 Merge p1 and p2 with ancestor pa and generate merge action list
190 Merge p1 and p2 with ancestor pa and generate merge action list
191
191
192 branchmerge and force are as passed in to update
192 branchmerge and force are as passed in to update
193 partial = function to filter file lists
193 partial = function to filter file lists
194 """
194 """
195
195
196 overwrite = force and not branchmerge
196 overwrite = force and not branchmerge
197 actions, copy, movewithdir = [], {}, {}
197 actions, copy, movewithdir = [], {}, {}
198
198
199 if overwrite:
199 if overwrite:
200 pa = wctx
200 pa = wctx
201 elif pa == p2: # backwards
201 elif pa == p2: # backwards
202 pa = wctx.p1()
202 pa = wctx.p1()
203 elif not branchmerge and not wctx.dirty(missing=True):
203 elif not branchmerge and not wctx.dirty(missing=True):
204 pass
204 pass
205 elif pa and repo.ui.configbool("merge", "followcopies", True):
205 elif pa and repo.ui.configbool("merge", "followcopies", True):
206 ret = copies.mergecopies(repo, wctx, p2, pa)
206 ret = copies.mergecopies(repo, wctx, p2, pa)
207 copy, movewithdir, diverge, renamedelete = ret
207 copy, movewithdir, diverge, renamedelete = ret
208 for of, fl in diverge.iteritems():
208 for of, fl in diverge.iteritems():
209 actions.append((of, "dr", (fl,), "divergent renames"))
209 actions.append((of, "dr", (fl,), "divergent renames"))
210 for of, fl in renamedelete.iteritems():
210 for of, fl in renamedelete.iteritems():
211 actions.append((of, "rd", (fl,), "rename and delete"))
211 actions.append((of, "rd", (fl,), "rename and delete"))
212
212
213 repo.ui.note(_("resolving manifests\n"))
213 repo.ui.note(_("resolving manifests\n"))
214 repo.ui.debug(" branchmerge: %s, force: %s, partial: %s\n"
214 repo.ui.debug(" branchmerge: %s, force: %s, partial: %s\n"
215 % (bool(branchmerge), bool(force), bool(partial)))
215 % (bool(branchmerge), bool(force), bool(partial)))
216 repo.ui.debug(" ancestor: %s, local: %s, remote: %s\n" % (pa, wctx, p2))
216 repo.ui.debug(" ancestor: %s, local: %s, remote: %s\n" % (pa, wctx, p2))
217
217
218 m1, m2, ma = wctx.manifest(), p2.manifest(), pa.manifest()
218 m1, m2, ma = wctx.manifest(), p2.manifest(), pa.manifest()
219 copied = set(copy.values())
219 copied = set(copy.values())
220 copied.update(movewithdir.values())
220 copied.update(movewithdir.values())
221
221
222 if '.hgsubstate' in m1:
222 if '.hgsubstate' in m1:
223 # check whether sub state is modified
223 # check whether sub state is modified
224 for s in sorted(wctx.substate):
224 for s in sorted(wctx.substate):
225 if wctx.sub(s).dirty():
225 if wctx.sub(s).dirty():
226 m1['.hgsubstate'] += "+"
226 m1['.hgsubstate'] += "+"
227 break
227 break
228
228
229 aborts, prompts = [], []
229 aborts, prompts = [], []
230 # Compare manifests
230 # Compare manifests
231 for f, n in m1.iteritems():
231 for f, n in m1.iteritems():
232 if partial and not partial(f):
232 if partial and not partial(f):
233 continue
233 continue
234 if f in m2:
234 if f in m2:
235 n2 = m2[f]
235 n2 = m2[f]
236 fl1, fl2, fla = m1.flags(f), m2.flags(f), ma.flags(f)
236 fl1, fl2, fla = m1.flags(f), m2.flags(f), ma.flags(f)
237 nol = 'l' not in fl1 + fl2 + fla
237 nol = 'l' not in fl1 + fl2 + fla
238 a = ma.get(f, nullid)
238 a = ma.get(f, nullid)
239 if n == n2 and fl1 == fl2:
239 if n == n2 and fl1 == fl2:
240 pass # same - keep local
240 pass # same - keep local
241 elif n2 == a and fl2 == fla:
241 elif n2 == a and fl2 == fla:
242 pass # remote unchanged - keep local
242 pass # remote unchanged - keep local
243 elif n == a and fl1 == fla: # local unchanged - use remote
243 elif n == a and fl1 == fla: # local unchanged - use remote
244 if n == n2: # optimization: keep local content
244 if n == n2: # optimization: keep local content
245 actions.append((f, "e", (fl2,), "update permissions"))
245 actions.append((f, "e", (fl2,), "update permissions"))
246 else:
246 else:
247 actions.append((f, "g", (fl2,), "remote is newer"))
247 actions.append((f, "g", (fl2,), "remote is newer"))
248 elif nol and n2 == a: # remote only changed 'x'
248 elif nol and n2 == a: # remote only changed 'x'
249 actions.append((f, "e", (fl2,), "update permissions"))
249 actions.append((f, "e", (fl2,), "update permissions"))
250 elif nol and n == a: # local only changed 'x'
250 elif nol and n == a: # local only changed 'x'
251 actions.append((f, "g", (fl1,), "remote is newer"))
251 actions.append((f, "g", (fl1,), "remote is newer"))
252 else: # both changed something
252 else: # both changed something
253 actions.append((f, "m", (f, f, False), "versions differ"))
253 actions.append((f, "m", (f, f, False), "versions differ"))
254 elif f in copied: # files we'll deal with on m2 side
254 elif f in copied: # files we'll deal with on m2 side
255 pass
255 pass
256 elif f in movewithdir: # directory rename
256 elif f in movewithdir: # directory rename
257 f2 = movewithdir[f]
257 f2 = movewithdir[f]
258 actions.append((f, "d", (None, f2, m1.flags(f)),
258 actions.append((f, "d", (None, f2, m1.flags(f)),
259 "remote renamed directory to " + f2))
259 "remote renamed directory to " + f2))
260 elif f in copy:
260 elif f in copy:
261 f2 = copy[f]
261 f2 = copy[f]
262 actions.append((f, "m", (f2, f, False),
262 actions.append((f, "m", (f2, f, False),
263 "local copied/moved to " + f2))
263 "local copied/moved to " + f2))
264 elif f in ma: # clean, a different, no remote
264 elif f in ma: # clean, a different, no remote
265 if n != ma[f]:
265 if n != ma[f]:
266 prompts.append((f, "cd")) # prompt changed/deleted
266 prompts.append((f, "cd")) # prompt changed/deleted
267 elif n[20:] == "a": # added, no remote
267 elif n[20:] == "a": # added, no remote
268 actions.append((f, "f", None, "remote deleted"))
268 actions.append((f, "f", None, "remote deleted"))
269 else:
269 else:
270 actions.append((f, "r", None, "other deleted"))
270 actions.append((f, "r", None, "other deleted"))
271
271
272 for f, n in m2.iteritems():
272 for f, n in m2.iteritems():
273 if partial and not partial(f):
273 if partial and not partial(f):
274 continue
274 continue
275 if f in m1 or f in copied: # files already visited
275 if f in m1 or f in copied: # files already visited
276 continue
276 continue
277 if f in movewithdir:
277 if f in movewithdir:
278 f2 = movewithdir[f]
278 f2 = movewithdir[f]
279 actions.append((None, "d", (f, f2, m2.flags(f)),
279 actions.append((None, "d", (f, f2, m2.flags(f)),
280 "local renamed directory to " + f2))
280 "local renamed directory to " + f2))
281 elif f in copy:
281 elif f in copy:
282 f2 = copy[f]
282 f2 = copy[f]
283 if f2 in m2:
283 if f2 in m2:
284 actions.append((f2, "m", (f, f, False),
284 actions.append((f2, "m", (f, f, False),
285 "remote copied to " + f))
285 "remote copied to " + f))
286 else:
286 else:
287 actions.append((f2, "m", (f, f, True),
287 actions.append((f2, "m", (f, f, True),
288 "remote moved to " + f))
288 "remote moved to " + f))
289 elif f not in ma:
289 elif f not in ma:
290 # local unknown, remote created: the logic is described by the
290 # local unknown, remote created: the logic is described by the
291 # following table:
291 # following table:
292 #
292 #
293 # force branchmerge different | action
293 # force branchmerge different | action
294 # n * n | get
294 # n * n | get
295 # n * y | abort
295 # n * y | abort
296 # y n * | get
296 # y n * | get
297 # y y n | get
297 # y y n | get
298 # y y y | merge
298 # y y y | merge
299 #
299 #
300 # Checking whether the files are different is expensive, so we
300 # Checking whether the files are different is expensive, so we
301 # don't do that when we can avoid it.
301 # don't do that when we can avoid it.
302 if force and not branchmerge:
302 if force and not branchmerge:
303 actions.append((f, "g", (m2.flags(f),), "remote created"))
303 actions.append((f, "g", (m2.flags(f),), "remote created"))
304 else:
304 else:
305 different = _checkunknownfile(repo, wctx, p2, f)
305 different = _checkunknownfile(repo, wctx, p2, f)
306 if force and branchmerge and different:
306 if force and branchmerge and different:
307 actions.append((f, "m", (f, f, False),
307 actions.append((f, "m", (f, f, False),
308 "remote differs from untracked local"))
308 "remote differs from untracked local"))
309 elif not force and different:
309 elif not force and different:
310 aborts.append((f, "ud"))
310 aborts.append((f, "ud"))
311 else:
311 else:
312 actions.append((f, "g", (m2.flags(f),), "remote created"))
312 actions.append((f, "g", (m2.flags(f),), "remote created"))
313 elif n != ma[f]:
313 elif n != ma[f]:
314 prompts.append((f, "dc")) # prompt deleted/changed
314 prompts.append((f, "dc")) # prompt deleted/changed
315
315
316 for f, m in sorted(aborts):
316 for f, m in sorted(aborts):
317 if m == "ud":
317 if m == "ud":
318 repo.ui.warn(_("%s: untracked file differs\n") % f)
318 repo.ui.warn(_("%s: untracked file differs\n") % f)
319 else: assert False, m
319 else: assert False, m
320 if aborts:
320 if aborts:
321 raise util.Abort(_("untracked files in working directory differ "
321 raise util.Abort(_("untracked files in working directory differ "
322 "from files in requested revision"))
322 "from files in requested revision"))
323
323
324 for f, m in sorted(prompts):
324 for f, m in sorted(prompts):
325 if m == "cd":
325 if m == "cd":
326 if repo.ui.promptchoice(
326 if repo.ui.promptchoice(
327 _("local changed %s which remote deleted\n"
327 _("local changed %s which remote deleted\n"
328 "use (c)hanged version or (d)elete?") % f,
328 "use (c)hanged version or (d)elete?") % f,
329 (_("&Changed"), _("&Delete")), 0):
329 (_("&Changed"), _("&Delete")), 0):
330 actions.append((f, "r", None, "prompt delete"))
330 actions.append((f, "r", None, "prompt delete"))
331 else:
331 else:
332 actions.append((f, "a", None, "prompt keep"))
332 actions.append((f, "a", None, "prompt keep"))
333 elif m == "dc":
333 elif m == "dc":
334 if repo.ui.promptchoice(
334 if repo.ui.promptchoice(
335 _("remote changed %s which local deleted\n"
335 _("remote changed %s which local deleted\n"
336 "use (c)hanged version or leave (d)eleted?") % f,
336 "use (c)hanged version or leave (d)eleted?") % f,
337 (_("&Changed"), _("&Deleted")), 0) == 0:
337 (_("&Changed"), _("&Deleted")), 0) == 0:
338 actions.append((f, "g", (m2.flags(f),), "prompt recreating"))
338 actions.append((f, "g", (m2.flags(f),), "prompt recreating"))
339 else: assert False, m
339 else: assert False, m
340 return actions
340 return actions
341
341
342 def actionkey(a):
342 def actionkey(a):
343 return a[1] == "r" and -1 or 0, a
343 return a[1] == "r" and -1 or 0, a
344
344
345 def getremove(repo, wctx, mctx, overwrite, args):
346 """apply usually-non-interactive updates to the working directory
347
348 wctx is the working copy context
349 mctx is the context to be merged into the working copy
350
351 yields tuples for progress updates
352 """
353 audit = repo.wopener.audit
354 for i, arg in enumerate(args):
355 f = arg[0]
356 if arg[1] == 'r':
357 repo.ui.note(_("removing %s\n") % f)
358 if f == '.hgsubstate': # subrepo states need updating
359 subrepo.submerge(repo, wctx, mctx, wctx, overwrite)
360 audit(f)
361 try:
362 util.unlinkpath(repo.wjoin(f), ignoremissing=True)
363 except OSError, inst:
364 repo.ui.warn(_("update failed to remove %s: %s!\n") %
365 (f, inst.strerror))
366 else:
367 repo.ui.note(_("getting %s\n") % f)
368 repo.wwrite(f, mctx.filectx(f).data(), arg[2][0])
369 if f == '.hgsubstate': # subrepo states need updating
370 subrepo.submerge(repo, wctx, mctx, wctx, overwrite)
371 yield i, f
372
345 def applyupdates(repo, actions, wctx, mctx, actx, overwrite):
373 def applyupdates(repo, actions, wctx, mctx, actx, overwrite):
346 """apply the merge action list to the working directory
374 """apply the merge action list to the working directory
347
375
348 wctx is the working copy context
376 wctx is the working copy context
349 mctx is the context to be merged into the working copy
377 mctx is the context to be merged into the working copy
350 actx is the context of the common ancestor
378 actx is the context of the common ancestor
351
379
352 Return a tuple of counts (updated, merged, removed, unresolved) that
380 Return a tuple of counts (updated, merged, removed, unresolved) that
353 describes how many files were affected by the update.
381 describes how many files were affected by the update.
354 """
382 """
355
383
356 updated, merged, removed, unresolved = 0, 0, 0, 0
384 updated, merged, removed, unresolved = 0, 0, 0, 0
357 ms = mergestate(repo)
385 ms = mergestate(repo)
358 ms.reset(wctx.p1().node())
386 ms.reset(wctx.p1().node())
359 moves = []
387 moves = []
360 actions.sort(key=actionkey)
388 actions.sort(key=actionkey)
361
389
362 # prescan for merges
390 # prescan for merges
363 for a in actions:
391 for a in actions:
364 f, m, args, msg = a
392 f, m, args, msg = a
365 repo.ui.debug(" %s: %s -> %s\n" % (f, msg, m))
393 repo.ui.debug(" %s: %s -> %s\n" % (f, msg, m))
366 if m == "m": # merge
394 if m == "m": # merge
367 f2, fd, move = args
395 f2, fd, move = args
368 if fd == '.hgsubstate': # merged internally
396 if fd == '.hgsubstate': # merged internally
369 continue
397 continue
370 repo.ui.debug(" preserving %s for resolve of %s\n" % (f, fd))
398 repo.ui.debug(" preserving %s for resolve of %s\n" % (f, fd))
371 fcl = wctx[f]
399 fcl = wctx[f]
372 fco = mctx[f2]
400 fco = mctx[f2]
373 if mctx == actx: # backwards, use working dir parent as ancestor
401 if mctx == actx: # backwards, use working dir parent as ancestor
374 if fcl.parents():
402 if fcl.parents():
375 fca = fcl.p1()
403 fca = fcl.p1()
376 else:
404 else:
377 fca = repo.filectx(f, fileid=nullrev)
405 fca = repo.filectx(f, fileid=nullrev)
378 else:
406 else:
379 fca = fcl.ancestor(fco, actx)
407 fca = fcl.ancestor(fco, actx)
380 if not fca:
408 if not fca:
381 fca = repo.filectx(f, fileid=nullrev)
409 fca = repo.filectx(f, fileid=nullrev)
382 ms.add(fcl, fco, fca, fd)
410 ms.add(fcl, fco, fca, fd)
383 if f != fd and move:
411 if f != fd and move:
384 moves.append(f)
412 moves.append(f)
385
413
386 audit = repo.wopener.audit
414 audit = repo.wopener.audit
387
415
388 # remove renamed files after safely stored
416 # remove renamed files after safely stored
389 for f in moves:
417 for f in moves:
390 if os.path.lexists(repo.wjoin(f)):
418 if os.path.lexists(repo.wjoin(f)):
391 repo.ui.debug("removing %s\n" % f)
419 repo.ui.debug("removing %s\n" % f)
392 audit(f)
420 audit(f)
393 util.unlinkpath(repo.wjoin(f))
421 util.unlinkpath(repo.wjoin(f))
394
422
395 numupdates = len(actions)
423 numupdates = len(actions)
424 workeractions = [a for a in actions if a[1] in 'gr']
425 updated = len([a for a in workeractions if a[1] == 'g'])
426 removed = len([a for a in workeractions if a[1] == 'r'])
427 actions = [a for a in actions if a[1] not in 'gr']
428
429 for i, item in getremove(repo, wctx, mctx, overwrite, workeractions):
430 repo.ui.progress(_('updating'), i + 1, item=item, total=numupdates,
431 unit=_('files'))
432
433 z = len(workeractions)
434
396 for i, a in enumerate(actions):
435 for i, a in enumerate(actions):
397 f, m, args, msg = a
436 f, m, args, msg = a
398 repo.ui.progress(_('updating'), i + 1, item=f, total=numupdates,
437 repo.ui.progress(_('updating'), z + i + 1, item=f, total=numupdates,
399 unit=_('files'))
438 unit=_('files'))
400 if m == "r": # remove
439 if m == "m": # merge
401 repo.ui.note(_("removing %s\n") % f)
402 audit(f)
403 if f == '.hgsubstate': # subrepo states need updating
404 subrepo.submerge(repo, wctx, mctx, wctx, overwrite)
405 try:
406 util.unlinkpath(repo.wjoin(f), ignoremissing=True)
407 except OSError, inst:
408 repo.ui.warn(_("update failed to remove %s: %s!\n") %
409 (f, inst.strerror))
410 removed += 1
411 elif m == "m": # merge
412 if fd == '.hgsubstate': # subrepo states need updating
440 if fd == '.hgsubstate': # subrepo states need updating
413 subrepo.submerge(repo, wctx, mctx, wctx.ancestor(mctx),
441 subrepo.submerge(repo, wctx, mctx, wctx.ancestor(mctx),
414 overwrite)
442 overwrite)
415 continue
443 continue
416 f2, fd, move = args
444 f2, fd, move = args
417 audit(fd)
445 audit(fd)
418 r = ms.resolve(fd, wctx, mctx)
446 r = ms.resolve(fd, wctx, mctx)
419 if r is not None and r > 0:
447 if r is not None and r > 0:
420 unresolved += 1
448 unresolved += 1
421 else:
449 else:
422 if r is None:
450 if r is None:
423 updated += 1
451 updated += 1
424 else:
452 else:
425 merged += 1
453 merged += 1
426 elif m == "g": # get
427 flags, = args
428 repo.ui.note(_("getting %s\n") % f)
429 repo.wwrite(f, mctx.filectx(f).data(), flags)
430 updated += 1
431 if f == '.hgsubstate': # subrepo states need updating
432 subrepo.submerge(repo, wctx, mctx, wctx, overwrite)
433 elif m == "d": # directory rename
454 elif m == "d": # directory rename
434 f2, fd, flags = args
455 f2, fd, flags = args
435 if f:
456 if f:
436 repo.ui.note(_("moving %s to %s\n") % (f, fd))
457 repo.ui.note(_("moving %s to %s\n") % (f, fd))
437 audit(f)
458 audit(f)
438 repo.wwrite(fd, wctx.filectx(f).data(), flags)
459 repo.wwrite(fd, wctx.filectx(f).data(), flags)
439 util.unlinkpath(repo.wjoin(f))
460 util.unlinkpath(repo.wjoin(f))
440 if f2:
461 if f2:
441 repo.ui.note(_("getting %s to %s\n") % (f2, fd))
462 repo.ui.note(_("getting %s to %s\n") % (f2, fd))
442 repo.wwrite(fd, mctx.filectx(f2).data(), flags)
463 repo.wwrite(fd, mctx.filectx(f2).data(), flags)
443 updated += 1
464 updated += 1
444 elif m == "dr": # divergent renames
465 elif m == "dr": # divergent renames
445 fl, = args
466 fl, = args
446 repo.ui.warn(_("note: possible conflict - %s was renamed "
467 repo.ui.warn(_("note: possible conflict - %s was renamed "
447 "multiple times to:\n") % f)
468 "multiple times to:\n") % f)
448 for nf in fl:
469 for nf in fl:
449 repo.ui.warn(" %s\n" % nf)
470 repo.ui.warn(" %s\n" % nf)
450 elif m == "rd": # rename and delete
471 elif m == "rd": # rename and delete
451 fl, = args
472 fl, = args
452 repo.ui.warn(_("note: possible conflict - %s was deleted "
473 repo.ui.warn(_("note: possible conflict - %s was deleted "
453 "and renamed to:\n") % f)
474 "and renamed to:\n") % f)
454 for nf in fl:
475 for nf in fl:
455 repo.ui.warn(" %s\n" % nf)
476 repo.ui.warn(" %s\n" % nf)
456 elif m == "e": # exec
477 elif m == "e": # exec
457 flags, = args
478 flags, = args
458 audit(f)
479 audit(f)
459 util.setflags(repo.wjoin(f), 'l' in flags, 'x' in flags)
480 util.setflags(repo.wjoin(f), 'l' in flags, 'x' in flags)
460 updated += 1
481 updated += 1
461 ms.commit()
482 ms.commit()
462 repo.ui.progress(_('updating'), None, total=numupdates, unit=_('files'))
483 repo.ui.progress(_('updating'), None, total=numupdates, unit=_('files'))
463
484
464 return updated, merged, removed, unresolved
485 return updated, merged, removed, unresolved
465
486
466 def calculateupdates(repo, tctx, mctx, ancestor, branchmerge, force, partial):
487 def calculateupdates(repo, tctx, mctx, ancestor, branchmerge, force, partial):
467 "Calculate the actions needed to merge mctx into tctx"
488 "Calculate the actions needed to merge mctx into tctx"
468 actions = []
489 actions = []
469 folding = not util.checkcase(repo.path)
490 folding = not util.checkcase(repo.path)
470 if folding:
491 if folding:
471 # collision check is not needed for clean update
492 # collision check is not needed for clean update
472 if (not branchmerge and
493 if (not branchmerge and
473 (force or not tctx.dirty(missing=True, branch=False))):
494 (force or not tctx.dirty(missing=True, branch=False))):
474 _checkcollision(mctx, None)
495 _checkcollision(mctx, None)
475 else:
496 else:
476 _checkcollision(mctx, (tctx, ancestor))
497 _checkcollision(mctx, (tctx, ancestor))
477 if tctx.rev() is None:
498 if tctx.rev() is None:
478 actions += _forgetremoved(tctx, mctx, branchmerge)
499 actions += _forgetremoved(tctx, mctx, branchmerge)
479 actions += manifestmerge(repo, tctx, mctx,
500 actions += manifestmerge(repo, tctx, mctx,
480 ancestor,
501 ancestor,
481 branchmerge, force,
502 branchmerge, force,
482 partial)
503 partial)
483 return actions
504 return actions
484
505
485 def recordupdates(repo, actions, branchmerge):
506 def recordupdates(repo, actions, branchmerge):
486 "record merge actions to the dirstate"
507 "record merge actions to the dirstate"
487
508
488 for a in actions:
509 for a in actions:
489 f, m, args, msg = a
510 f, m, args, msg = a
490 if m == "r": # remove
511 if m == "r": # remove
491 if branchmerge:
512 if branchmerge:
492 repo.dirstate.remove(f)
513 repo.dirstate.remove(f)
493 else:
514 else:
494 repo.dirstate.drop(f)
515 repo.dirstate.drop(f)
495 elif m == "a": # re-add
516 elif m == "a": # re-add
496 if not branchmerge:
517 if not branchmerge:
497 repo.dirstate.add(f)
518 repo.dirstate.add(f)
498 elif m == "f": # forget
519 elif m == "f": # forget
499 repo.dirstate.drop(f)
520 repo.dirstate.drop(f)
500 elif m == "e": # exec change
521 elif m == "e": # exec change
501 repo.dirstate.normallookup(f)
522 repo.dirstate.normallookup(f)
502 elif m == "g": # get
523 elif m == "g": # get
503 if branchmerge:
524 if branchmerge:
504 repo.dirstate.otherparent(f)
525 repo.dirstate.otherparent(f)
505 else:
526 else:
506 repo.dirstate.normal(f)
527 repo.dirstate.normal(f)
507 elif m == "m": # merge
528 elif m == "m": # merge
508 f2, fd, move = args
529 f2, fd, move = args
509 if branchmerge:
530 if branchmerge:
510 # We've done a branch merge, mark this file as merged
531 # We've done a branch merge, mark this file as merged
511 # so that we properly record the merger later
532 # so that we properly record the merger later
512 repo.dirstate.merge(fd)
533 repo.dirstate.merge(fd)
513 if f != f2: # copy/rename
534 if f != f2: # copy/rename
514 if move:
535 if move:
515 repo.dirstate.remove(f)
536 repo.dirstate.remove(f)
516 if f != fd:
537 if f != fd:
517 repo.dirstate.copy(f, fd)
538 repo.dirstate.copy(f, fd)
518 else:
539 else:
519 repo.dirstate.copy(f2, fd)
540 repo.dirstate.copy(f2, fd)
520 else:
541 else:
521 # We've update-merged a locally modified file, so
542 # We've update-merged a locally modified file, so
522 # we set the dirstate to emulate a normal checkout
543 # we set the dirstate to emulate a normal checkout
523 # of that file some time in the past. Thus our
544 # of that file some time in the past. Thus our
524 # merge will appear as a normal local file
545 # merge will appear as a normal local file
525 # modification.
546 # modification.
526 if f2 == fd: # file not locally copied/moved
547 if f2 == fd: # file not locally copied/moved
527 repo.dirstate.normallookup(fd)
548 repo.dirstate.normallookup(fd)
528 if move:
549 if move:
529 repo.dirstate.drop(f)
550 repo.dirstate.drop(f)
530 elif m == "d": # directory rename
551 elif m == "d": # directory rename
531 f2, fd, flag = args
552 f2, fd, flag = args
532 if not f2 and f not in repo.dirstate:
553 if not f2 and f not in repo.dirstate:
533 # untracked file moved
554 # untracked file moved
534 continue
555 continue
535 if branchmerge:
556 if branchmerge:
536 repo.dirstate.add(fd)
557 repo.dirstate.add(fd)
537 if f:
558 if f:
538 repo.dirstate.remove(f)
559 repo.dirstate.remove(f)
539 repo.dirstate.copy(f, fd)
560 repo.dirstate.copy(f, fd)
540 if f2:
561 if f2:
541 repo.dirstate.copy(f2, fd)
562 repo.dirstate.copy(f2, fd)
542 else:
563 else:
543 repo.dirstate.normal(fd)
564 repo.dirstate.normal(fd)
544 if f:
565 if f:
545 repo.dirstate.drop(f)
566 repo.dirstate.drop(f)
546
567
547 def update(repo, node, branchmerge, force, partial, ancestor=None,
568 def update(repo, node, branchmerge, force, partial, ancestor=None,
548 mergeancestor=False):
569 mergeancestor=False):
549 """
570 """
550 Perform a merge between the working directory and the given node
571 Perform a merge between the working directory and the given node
551
572
552 node = the node to update to, or None if unspecified
573 node = the node to update to, or None if unspecified
553 branchmerge = whether to merge between branches
574 branchmerge = whether to merge between branches
554 force = whether to force branch merging or file overwriting
575 force = whether to force branch merging or file overwriting
555 partial = a function to filter file lists (dirstate not updated)
576 partial = a function to filter file lists (dirstate not updated)
556 mergeancestor = if false, merging with an ancestor (fast-forward)
577 mergeancestor = if false, merging with an ancestor (fast-forward)
557 is only allowed between different named branches. This flag
578 is only allowed between different named branches. This flag
558 is used by rebase extension as a temporary fix and should be
579 is used by rebase extension as a temporary fix and should be
559 avoided in general.
580 avoided in general.
560
581
561 The table below shows all the behaviors of the update command
582 The table below shows all the behaviors of the update command
562 given the -c and -C or no options, whether the working directory
583 given the -c and -C or no options, whether the working directory
563 is dirty, whether a revision is specified, and the relationship of
584 is dirty, whether a revision is specified, and the relationship of
564 the parent rev to the target rev (linear, on the same named
585 the parent rev to the target rev (linear, on the same named
565 branch, or on another named branch).
586 branch, or on another named branch).
566
587
567 This logic is tested by test-update-branches.t.
588 This logic is tested by test-update-branches.t.
568
589
569 -c -C dirty rev | linear same cross
590 -c -C dirty rev | linear same cross
570 n n n n | ok (1) x
591 n n n n | ok (1) x
571 n n n y | ok ok ok
592 n n n y | ok ok ok
572 n n y * | merge (2) (2)
593 n n y * | merge (2) (2)
573 n y * * | --- discard ---
594 n y * * | --- discard ---
574 y n y * | --- (3) ---
595 y n y * | --- (3) ---
575 y n n * | --- ok ---
596 y n n * | --- ok ---
576 y y * * | --- (4) ---
597 y y * * | --- (4) ---
577
598
578 x = can't happen
599 x = can't happen
579 * = don't-care
600 * = don't-care
580 1 = abort: crosses branches (use 'hg merge' or 'hg update -c')
601 1 = abort: crosses branches (use 'hg merge' or 'hg update -c')
581 2 = abort: crosses branches (use 'hg merge' to merge or
602 2 = abort: crosses branches (use 'hg merge' to merge or
582 use 'hg update -C' to discard changes)
603 use 'hg update -C' to discard changes)
583 3 = abort: uncommitted local changes
604 3 = abort: uncommitted local changes
584 4 = incompatible options (checked in commands.py)
605 4 = incompatible options (checked in commands.py)
585
606
586 Return the same tuple as applyupdates().
607 Return the same tuple as applyupdates().
587 """
608 """
588
609
589 onode = node
610 onode = node
590 wlock = repo.wlock()
611 wlock = repo.wlock()
591 try:
612 try:
592 wc = repo[None]
613 wc = repo[None]
593 if node is None:
614 if node is None:
594 # tip of current branch
615 # tip of current branch
595 try:
616 try:
596 node = repo.branchtip(wc.branch())
617 node = repo.branchtip(wc.branch())
597 except error.RepoLookupError:
618 except error.RepoLookupError:
598 if wc.branch() == "default": # no default branch!
619 if wc.branch() == "default": # no default branch!
599 node = repo.lookup("tip") # update to tip
620 node = repo.lookup("tip") # update to tip
600 else:
621 else:
601 raise util.Abort(_("branch %s not found") % wc.branch())
622 raise util.Abort(_("branch %s not found") % wc.branch())
602 overwrite = force and not branchmerge
623 overwrite = force and not branchmerge
603 pl = wc.parents()
624 pl = wc.parents()
604 p1, p2 = pl[0], repo[node]
625 p1, p2 = pl[0], repo[node]
605 if ancestor:
626 if ancestor:
606 pa = repo[ancestor]
627 pa = repo[ancestor]
607 else:
628 else:
608 pa = p1.ancestor(p2)
629 pa = p1.ancestor(p2)
609
630
610 fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2)
631 fp1, fp2, xp1, xp2 = p1.node(), p2.node(), str(p1), str(p2)
611
632
612 ### check phase
633 ### check phase
613 if not overwrite and len(pl) > 1:
634 if not overwrite and len(pl) > 1:
614 raise util.Abort(_("outstanding uncommitted merges"))
635 raise util.Abort(_("outstanding uncommitted merges"))
615 if branchmerge:
636 if branchmerge:
616 if pa == p2:
637 if pa == p2:
617 raise util.Abort(_("merging with a working directory ancestor"
638 raise util.Abort(_("merging with a working directory ancestor"
618 " has no effect"))
639 " has no effect"))
619 elif pa == p1:
640 elif pa == p1:
620 if not mergeancestor and p1.branch() == p2.branch():
641 if not mergeancestor and p1.branch() == p2.branch():
621 raise util.Abort(_("nothing to merge"),
642 raise util.Abort(_("nothing to merge"),
622 hint=_("use 'hg update' "
643 hint=_("use 'hg update' "
623 "or check 'hg heads'"))
644 "or check 'hg heads'"))
624 if not force and (wc.files() or wc.deleted()):
645 if not force and (wc.files() or wc.deleted()):
625 raise util.Abort(_("outstanding uncommitted changes"),
646 raise util.Abort(_("outstanding uncommitted changes"),
626 hint=_("use 'hg status' to list changes"))
647 hint=_("use 'hg status' to list changes"))
627 for s in sorted(wc.substate):
648 for s in sorted(wc.substate):
628 if wc.sub(s).dirty():
649 if wc.sub(s).dirty():
629 raise util.Abort(_("outstanding uncommitted changes in "
650 raise util.Abort(_("outstanding uncommitted changes in "
630 "subrepository '%s'") % s)
651 "subrepository '%s'") % s)
631
652
632 elif not overwrite:
653 elif not overwrite:
633 if pa == p1 or pa == p2: # linear
654 if pa == p1 or pa == p2: # linear
634 pass # all good
655 pass # all good
635 elif wc.dirty(missing=True):
656 elif wc.dirty(missing=True):
636 raise util.Abort(_("crosses branches (merge branches or use"
657 raise util.Abort(_("crosses branches (merge branches or use"
637 " --clean to discard changes)"))
658 " --clean to discard changes)"))
638 elif onode is None:
659 elif onode is None:
639 raise util.Abort(_("crosses branches (merge branches or update"
660 raise util.Abort(_("crosses branches (merge branches or update"
640 " --check to force update)"))
661 " --check to force update)"))
641 else:
662 else:
642 # Allow jumping branches if clean and specific rev given
663 # Allow jumping branches if clean and specific rev given
643 pa = p1
664 pa = p1
644
665
645 ### calculate phase
666 ### calculate phase
646 actions = calculateupdates(repo, wc, p2, pa,
667 actions = calculateupdates(repo, wc, p2, pa,
647 branchmerge, force, partial)
668 branchmerge, force, partial)
648
669
649 ### apply phase
670 ### apply phase
650 if not branchmerge: # just jump to the new rev
671 if not branchmerge: # just jump to the new rev
651 fp1, fp2, xp1, xp2 = fp2, nullid, xp2, ''
672 fp1, fp2, xp1, xp2 = fp2, nullid, xp2, ''
652 if not partial:
673 if not partial:
653 repo.hook('preupdate', throw=True, parent1=xp1, parent2=xp2)
674 repo.hook('preupdate', throw=True, parent1=xp1, parent2=xp2)
654
675
655 stats = applyupdates(repo, actions, wc, p2, pa, overwrite)
676 stats = applyupdates(repo, actions, wc, p2, pa, overwrite)
656
677
657 if not partial:
678 if not partial:
658 repo.setparents(fp1, fp2)
679 repo.setparents(fp1, fp2)
659 recordupdates(repo, actions, branchmerge)
680 recordupdates(repo, actions, branchmerge)
660 if not branchmerge:
681 if not branchmerge:
661 repo.dirstate.setbranch(p2.branch())
682 repo.dirstate.setbranch(p2.branch())
662 finally:
683 finally:
663 wlock.release()
684 wlock.release()
664
685
665 if not partial:
686 if not partial:
666 repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3])
687 repo.hook('update', parent1=xp1, parent2=xp2, error=stats[3])
667 return stats
688 return stats
General Comments 0
You need to be logged in to leave comments. Login now