##// END OF EJS Templates
rebase: use revlog.findmissingrevs to compute detach set
Siddharth Agarwal -
r17973:fa6be7b8 default
parent child Browse files
Show More
@@ -1,730 +1,733 b''
1 1 # rebase.py - rebasing feature for mercurial
2 2 #
3 3 # Copyright 2008 Stefano Tortarolo <stefano.tortarolo at gmail dot com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 '''command to move sets of revisions to a different ancestor
9 9
10 10 This extension lets you rebase changesets in an existing Mercurial
11 11 repository.
12 12
13 13 For more information:
14 14 http://mercurial.selenic.com/wiki/RebaseExtension
15 15 '''
16 16
17 17 from mercurial import hg, util, repair, merge, cmdutil, commands, bookmarks
18 18 from mercurial import extensions, patch, scmutil, phases, obsolete
19 19 from mercurial.commands import templateopts
20 20 from mercurial.node import nullrev
21 21 from mercurial.lock import release
22 22 from mercurial.i18n import _
23 23 import os, errno
24 24
25 25 nullmerge = -2
26 26
27 27 cmdtable = {}
28 28 command = cmdutil.command(cmdtable)
29 29 testedwith = 'internal'
30 30
31 31 @command('rebase',
32 32 [('s', 'source', '',
33 33 _('rebase from the specified changeset'), _('REV')),
34 34 ('b', 'base', '',
35 35 _('rebase from the base of the specified changeset '
36 36 '(up to greatest common ancestor of base and dest)'),
37 37 _('REV')),
38 38 ('r', 'rev', [],
39 39 _('rebase these revisions'),
40 40 _('REV')),
41 41 ('d', 'dest', '',
42 42 _('rebase onto the specified changeset'), _('REV')),
43 43 ('', 'collapse', False, _('collapse the rebased changesets')),
44 44 ('m', 'message', '',
45 45 _('use text as collapse commit message'), _('TEXT')),
46 46 ('e', 'edit', False, _('invoke editor on commit messages')),
47 47 ('l', 'logfile', '',
48 48 _('read collapse commit message from file'), _('FILE')),
49 49 ('', 'keep', False, _('keep original changesets')),
50 50 ('', 'keepbranches', False, _('keep original branch names')),
51 51 ('D', 'detach', False, _('(DEPRECATED)')),
52 52 ('t', 'tool', '', _('specify merge tool')),
53 53 ('c', 'continue', False, _('continue an interrupted rebase')),
54 54 ('a', 'abort', False, _('abort an interrupted rebase'))] +
55 55 templateopts,
56 56 _('[-s REV | -b REV] [-d REV] [OPTION]'))
57 57 def rebase(ui, repo, **opts):
58 58 """move changeset (and descendants) to a different branch
59 59
60 60 Rebase uses repeated merging to graft changesets from one part of
61 61 history (the source) onto another (the destination). This can be
62 62 useful for linearizing *local* changes relative to a master
63 63 development tree.
64 64
65 65 You should not rebase changesets that have already been shared
66 66 with others. Doing so will force everybody else to perform the
67 67 same rebase or they will end up with duplicated changesets after
68 68 pulling in your rebased changesets.
69 69
70 70 If you don't specify a destination changeset (``-d/--dest``),
71 71 rebase uses the tipmost head of the current named branch as the
72 72 destination. (The destination changeset is not modified by
73 73 rebasing, but new changesets are added as its descendants.)
74 74
75 75 You can specify which changesets to rebase in two ways: as a
76 76 "source" changeset or as a "base" changeset. Both are shorthand
77 77 for a topologically related set of changesets (the "source
78 78 branch"). If you specify source (``-s/--source``), rebase will
79 79 rebase that changeset and all of its descendants onto dest. If you
80 80 specify base (``-b/--base``), rebase will select ancestors of base
81 81 back to but not including the common ancestor with dest. Thus,
82 82 ``-b`` is less precise but more convenient than ``-s``: you can
83 83 specify any changeset in the source branch, and rebase will select
84 84 the whole branch. If you specify neither ``-s`` nor ``-b``, rebase
85 85 uses the parent of the working directory as the base.
86 86
87 87 By default, rebase recreates the changesets in the source branch
88 88 as descendants of dest and then destroys the originals. Use
89 89 ``--keep`` to preserve the original source changesets. Some
90 90 changesets in the source branch (e.g. merges from the destination
91 91 branch) may be dropped if they no longer contribute any change.
92 92
93 93 One result of the rules for selecting the destination changeset
94 94 and source branch is that, unlike ``merge``, rebase will do
95 95 nothing if you are at the latest (tipmost) head of a named branch
96 96 with two heads. You need to explicitly specify source and/or
97 97 destination (or ``update`` to the other head, if it's the head of
98 98 the intended source branch).
99 99
100 100 If a rebase is interrupted to manually resolve a merge, it can be
101 101 continued with --continue/-c or aborted with --abort/-a.
102 102
103 103 Returns 0 on success, 1 if nothing to rebase.
104 104 """
105 105 originalwd = target = None
106 106 external = nullrev
107 107 state = {}
108 108 skipped = set()
109 109 targetancestors = set()
110 110
111 111 editor = None
112 112 if opts.get('edit'):
113 113 editor = cmdutil.commitforceeditor
114 114
115 115 lock = wlock = None
116 116 try:
117 117 wlock = repo.wlock()
118 118 lock = repo.lock()
119 119
120 120 # Validate input and define rebasing points
121 121 destf = opts.get('dest', None)
122 122 srcf = opts.get('source', None)
123 123 basef = opts.get('base', None)
124 124 revf = opts.get('rev', [])
125 125 contf = opts.get('continue')
126 126 abortf = opts.get('abort')
127 127 collapsef = opts.get('collapse', False)
128 128 collapsemsg = cmdutil.logmessage(ui, opts)
129 129 extrafn = opts.get('extrafn') # internal, used by e.g. hgsubversion
130 130 keepf = opts.get('keep', False)
131 131 keepbranchesf = opts.get('keepbranches', False)
132 132 # keepopen is not meant for use on the command line, but by
133 133 # other extensions
134 134 keepopen = opts.get('keepopen', False)
135 135
136 136 if collapsemsg and not collapsef:
137 137 raise util.Abort(
138 138 _('message can only be specified with collapse'))
139 139
140 140 if contf or abortf:
141 141 if contf and abortf:
142 142 raise util.Abort(_('cannot use both abort and continue'))
143 143 if collapsef:
144 144 raise util.Abort(
145 145 _('cannot use collapse with continue or abort'))
146 146 if srcf or basef or destf:
147 147 raise util.Abort(
148 148 _('abort and continue do not allow specifying revisions'))
149 149 if opts.get('tool', False):
150 150 ui.warn(_('tool option will be ignored\n'))
151 151
152 152 (originalwd, target, state, skipped, collapsef, keepf,
153 153 keepbranchesf, external) = restorestatus(repo)
154 154 if abortf:
155 155 return abort(repo, originalwd, target, state)
156 156 else:
157 157 if srcf and basef:
158 158 raise util.Abort(_('cannot specify both a '
159 159 'source and a base'))
160 160 if revf and basef:
161 161 raise util.Abort(_('cannot specify both a '
162 162 'revision and a base'))
163 163 if revf and srcf:
164 164 raise util.Abort(_('cannot specify both a '
165 165 'revision and a source'))
166 166
167 167 cmdutil.bailifchanged(repo)
168 168
169 169 if not destf:
170 170 # Destination defaults to the latest revision in the
171 171 # current branch
172 172 branch = repo[None].branch()
173 173 dest = repo[branch]
174 174 else:
175 175 dest = scmutil.revsingle(repo, destf)
176 176
177 177 if revf:
178 178 rebaseset = repo.revs('%lr', revf)
179 179 elif srcf:
180 180 src = scmutil.revrange(repo, [srcf])
181 181 rebaseset = repo.revs('(%ld)::', src)
182 182 else:
183 183 base = scmutil.revrange(repo, [basef or '.'])
184 184 rebaseset = repo.revs(
185 185 '(children(ancestor(%ld, %d)) and ::(%ld))::',
186 186 base, dest, base)
187 187 # temporary top level filtering of extinct revisions
188 188 rebaseset = repo.revs('%ld - hidden()', rebaseset)
189 189 if rebaseset:
190 190 root = min(rebaseset)
191 191 else:
192 192 root = None
193 193
194 194 if not rebaseset:
195 195 repo.ui.debug('base is ancestor of destination\n')
196 196 result = None
197 197 elif not keepf and repo.revs('first(children(%ld) - %ld)-hidden()',
198 198 rebaseset, rebaseset):
199 199 raise util.Abort(
200 200 _("can't remove original changesets with"
201 201 " unrebased descendants"),
202 202 hint=_('use --keep to keep original changesets'))
203 203 elif not keepf and not repo[root].mutable():
204 204 raise util.Abort(_("can't rebase immutable changeset %s")
205 205 % repo[root],
206 206 hint=_('see hg help phases for details'))
207 207 else:
208 208 result = buildstate(repo, dest, rebaseset, collapsef)
209 209
210 210 if not result:
211 211 # Empty state built, nothing to rebase
212 212 ui.status(_('nothing to rebase\n'))
213 213 return 1
214 214 else:
215 215 originalwd, target, state = result
216 216 if collapsef:
217 217 targetancestors = set(repo.changelog.ancestors([target]))
218 218 targetancestors.add(target)
219 219 external = checkexternal(repo, state, targetancestors)
220 220
221 221 if keepbranchesf:
222 222 assert not extrafn, 'cannot use both keepbranches and extrafn'
223 223 def extrafn(ctx, extra):
224 224 extra['branch'] = ctx.branch()
225 225 if collapsef:
226 226 branches = set()
227 227 for rev in state:
228 228 branches.add(repo[rev].branch())
229 229 if len(branches) > 1:
230 230 raise util.Abort(_('cannot collapse multiple named '
231 231 'branches'))
232 232
233 233
234 234 # Rebase
235 235 if not targetancestors:
236 236 targetancestors = set(repo.changelog.ancestors([target]))
237 237 targetancestors.add(target)
238 238
239 239 # Keep track of the current bookmarks in order to reset them later
240 240 currentbookmarks = repo._bookmarks.copy()
241 241 activebookmark = repo._bookmarkcurrent
242 242 if activebookmark:
243 243 bookmarks.unsetcurrent(repo)
244 244
245 245 sortedstate = sorted(state)
246 246 total = len(sortedstate)
247 247 pos = 0
248 248 for rev in sortedstate:
249 249 pos += 1
250 250 if state[rev] == -1:
251 251 ui.progress(_("rebasing"), pos, ("%d:%s" % (rev, repo[rev])),
252 252 _('changesets'), total)
253 253 storestatus(repo, originalwd, target, state, collapsef, keepf,
254 254 keepbranchesf, external)
255 255 p1, p2 = defineparents(repo, rev, target, state,
256 256 targetancestors)
257 257 if len(repo.parents()) == 2:
258 258 repo.ui.debug('resuming interrupted rebase\n')
259 259 else:
260 260 try:
261 261 ui.setconfig('ui', 'forcemerge', opts.get('tool', ''))
262 262 stats = rebasenode(repo, rev, p1, state, collapsef)
263 263 if stats and stats[3] > 0:
264 264 raise util.Abort(_('unresolved conflicts (see hg '
265 265 'resolve, then hg rebase --continue)'))
266 266 finally:
267 267 ui.setconfig('ui', 'forcemerge', '')
268 268 cmdutil.duplicatecopies(repo, rev, target)
269 269 if not collapsef:
270 270 newrev = concludenode(repo, rev, p1, p2, extrafn=extrafn,
271 271 editor=editor)
272 272 else:
273 273 # Skip commit if we are collapsing
274 274 repo.setparents(repo[p1].node())
275 275 newrev = None
276 276 # Update the state
277 277 if newrev is not None:
278 278 state[rev] = repo[newrev].rev()
279 279 else:
280 280 if not collapsef:
281 281 ui.note(_('no changes, revision %d skipped\n') % rev)
282 282 ui.debug('next revision set to %s\n' % p1)
283 283 skipped.add(rev)
284 284 state[rev] = p1
285 285
286 286 ui.progress(_('rebasing'), None)
287 287 ui.note(_('rebase merging completed\n'))
288 288
289 289 if collapsef and not keepopen:
290 290 p1, p2 = defineparents(repo, min(state), target,
291 291 state, targetancestors)
292 292 if collapsemsg:
293 293 commitmsg = collapsemsg
294 294 else:
295 295 commitmsg = 'Collapsed revision'
296 296 for rebased in state:
297 297 if rebased not in skipped and state[rebased] != nullmerge:
298 298 commitmsg += '\n* %s' % repo[rebased].description()
299 299 commitmsg = ui.edit(commitmsg, repo.ui.username())
300 300 newrev = concludenode(repo, rev, p1, external, commitmsg=commitmsg,
301 301 extrafn=extrafn, editor=editor)
302 302
303 303 if 'qtip' in repo.tags():
304 304 updatemq(repo, state, skipped, **opts)
305 305
306 306 if currentbookmarks:
307 307 # Nodeids are needed to reset bookmarks
308 308 nstate = {}
309 309 for k, v in state.iteritems():
310 310 if v != nullmerge:
311 311 nstate[repo[k].node()] = repo[v].node()
312 312
313 313 if not keepf:
314 314 collapsedas = None
315 315 if collapsef:
316 316 collapsedas = newrev
317 317 clearrebased(ui, repo, state, collapsedas)
318 318
319 319 if currentbookmarks:
320 320 updatebookmarks(repo, nstate, currentbookmarks, **opts)
321 321
322 322 clearstatus(repo)
323 323 ui.note(_("rebase completed\n"))
324 324 if os.path.exists(repo.sjoin('undo')):
325 325 util.unlinkpath(repo.sjoin('undo'))
326 326 if skipped:
327 327 ui.note(_("%d revisions have been skipped\n") % len(skipped))
328 328
329 329 if (activebookmark and
330 330 repo['tip'].node() == repo._bookmarks[activebookmark]):
331 331 bookmarks.setcurrent(repo, activebookmark)
332 332
333 333 finally:
334 334 release(lock, wlock)
335 335
336 336 def checkexternal(repo, state, targetancestors):
337 337 """Check whether one or more external revisions need to be taken in
338 338 consideration. In the latter case, abort.
339 339 """
340 340 external = nullrev
341 341 source = min(state)
342 342 for rev in state:
343 343 if rev == source:
344 344 continue
345 345 # Check externals and fail if there are more than one
346 346 for p in repo[rev].parents():
347 347 if (p.rev() not in state
348 348 and p.rev() not in targetancestors):
349 349 if external != nullrev:
350 350 raise util.Abort(_('unable to collapse, there is more '
351 351 'than one external parent'))
352 352 external = p.rev()
353 353 return external
354 354
355 355 def concludenode(repo, rev, p1, p2, commitmsg=None, editor=None, extrafn=None):
356 356 'Commit the changes and store useful information in extra'
357 357 try:
358 358 repo.setparents(repo[p1].node(), repo[p2].node())
359 359 ctx = repo[rev]
360 360 if commitmsg is None:
361 361 commitmsg = ctx.description()
362 362 extra = {'rebase_source': ctx.hex()}
363 363 if extrafn:
364 364 extrafn(ctx, extra)
365 365 # Commit might fail if unresolved files exist
366 366 newrev = repo.commit(text=commitmsg, user=ctx.user(),
367 367 date=ctx.date(), extra=extra, editor=editor)
368 368 repo.dirstate.setbranch(repo[newrev].branch())
369 369 targetphase = max(ctx.phase(), phases.draft)
370 370 # retractboundary doesn't overwrite upper phase inherited from parent
371 371 newnode = repo[newrev].node()
372 372 if newnode:
373 373 phases.retractboundary(repo, targetphase, [newnode])
374 374 return newrev
375 375 except util.Abort:
376 376 # Invalidate the previous setparents
377 377 repo.dirstate.invalidate()
378 378 raise
379 379
380 380 def rebasenode(repo, rev, p1, state, collapse):
381 381 'Rebase a single revision'
382 382 # Merge phase
383 383 # Update to target and merge it with local
384 384 if repo['.'].rev() != repo[p1].rev():
385 385 repo.ui.debug(" update to %d:%s\n" % (repo[p1].rev(), repo[p1]))
386 386 merge.update(repo, p1, False, True, False)
387 387 else:
388 388 repo.ui.debug(" already in target\n")
389 389 repo.dirstate.write()
390 390 repo.ui.debug(" merge against %d:%s\n" % (repo[rev].rev(), repo[rev]))
391 391 base = None
392 392 if repo[rev].rev() != repo[min(state)].rev():
393 393 base = repo[rev].p1().node()
394 394 # When collapsing in-place, the parent is the common ancestor, we
395 395 # have to allow merging with it.
396 396 return merge.update(repo, rev, True, True, False, base, collapse)
397 397
398 398 def defineparents(repo, rev, target, state, targetancestors):
399 399 'Return the new parent relationship of the revision that will be rebased'
400 400 parents = repo[rev].parents()
401 401 p1 = p2 = nullrev
402 402
403 403 P1n = parents[0].rev()
404 404 if P1n in targetancestors:
405 405 p1 = target
406 406 elif P1n in state:
407 407 if state[P1n] == nullmerge:
408 408 p1 = target
409 409 else:
410 410 p1 = state[P1n]
411 411 else: # P1n external
412 412 p1 = target
413 413 p2 = P1n
414 414
415 415 if len(parents) == 2 and parents[1].rev() not in targetancestors:
416 416 P2n = parents[1].rev()
417 417 # interesting second parent
418 418 if P2n in state:
419 419 if p1 == target: # P1n in targetancestors or external
420 420 p1 = state[P2n]
421 421 else:
422 422 p2 = state[P2n]
423 423 else: # P2n external
424 424 if p2 != nullrev: # P1n external too => rev is a merged revision
425 425 raise util.Abort(_('cannot use revision %d as base, result '
426 426 'would have 3 parents') % rev)
427 427 p2 = P2n
428 428 repo.ui.debug(" future parents are %d and %d\n" %
429 429 (repo[p1].rev(), repo[p2].rev()))
430 430 return p1, p2
431 431
432 432 def isagitpatch(repo, patchname):
433 433 'Return true if the given patch is in git format'
434 434 mqpatch = os.path.join(repo.mq.path, patchname)
435 435 for line in patch.linereader(file(mqpatch, 'rb')):
436 436 if line.startswith('diff --git'):
437 437 return True
438 438 return False
439 439
440 440 def updatemq(repo, state, skipped, **opts):
441 441 'Update rebased mq patches - finalize and then import them'
442 442 mqrebase = {}
443 443 mq = repo.mq
444 444 original_series = mq.fullseries[:]
445 445 skippedpatches = set()
446 446
447 447 for p in mq.applied:
448 448 rev = repo[p.node].rev()
449 449 if rev in state:
450 450 repo.ui.debug('revision %d is an mq patch (%s), finalize it.\n' %
451 451 (rev, p.name))
452 452 mqrebase[rev] = (p.name, isagitpatch(repo, p.name))
453 453 else:
454 454 # Applied but not rebased, not sure this should happen
455 455 skippedpatches.add(p.name)
456 456
457 457 if mqrebase:
458 458 mq.finish(repo, mqrebase.keys())
459 459
460 460 # We must start import from the newest revision
461 461 for rev in sorted(mqrebase, reverse=True):
462 462 if rev not in skipped:
463 463 name, isgit = mqrebase[rev]
464 464 repo.ui.debug('import mq patch %d (%s)\n' % (state[rev], name))
465 465 mq.qimport(repo, (), patchname=name, git=isgit,
466 466 rev=[str(state[rev])])
467 467 else:
468 468 # Rebased and skipped
469 469 skippedpatches.add(mqrebase[rev][0])
470 470
471 471 # Patches were either applied and rebased and imported in
472 472 # order, applied and removed or unapplied. Discard the removed
473 473 # ones while preserving the original series order and guards.
474 474 newseries = [s for s in original_series
475 475 if mq.guard_re.split(s, 1)[0] not in skippedpatches]
476 476 mq.fullseries[:] = newseries
477 477 mq.seriesdirty = True
478 478 mq.savedirty()
479 479
480 480 def updatebookmarks(repo, nstate, originalbookmarks, **opts):
481 481 'Move bookmarks to their correct changesets'
482 482 marks = repo._bookmarks
483 483 for k, v in originalbookmarks.iteritems():
484 484 if v in nstate:
485 485 if nstate[v] != nullmerge:
486 486 # update the bookmarks for revs that have moved
487 487 marks[k] = nstate[v]
488 488
489 489 marks.write()
490 490
491 491 def storestatus(repo, originalwd, target, state, collapse, keep, keepbranches,
492 492 external):
493 493 'Store the current status to allow recovery'
494 494 f = repo.opener("rebasestate", "w")
495 495 f.write(repo[originalwd].hex() + '\n')
496 496 f.write(repo[target].hex() + '\n')
497 497 f.write(repo[external].hex() + '\n')
498 498 f.write('%d\n' % int(collapse))
499 499 f.write('%d\n' % int(keep))
500 500 f.write('%d\n' % int(keepbranches))
501 501 for d, v in state.iteritems():
502 502 oldrev = repo[d].hex()
503 503 if v != nullmerge:
504 504 newrev = repo[v].hex()
505 505 else:
506 506 newrev = v
507 507 f.write("%s:%s\n" % (oldrev, newrev))
508 508 f.close()
509 509 repo.ui.debug('rebase status stored\n')
510 510
511 511 def clearstatus(repo):
512 512 'Remove the status files'
513 513 if os.path.exists(repo.join("rebasestate")):
514 514 util.unlinkpath(repo.join("rebasestate"))
515 515
516 516 def restorestatus(repo):
517 517 'Restore a previously stored status'
518 518 try:
519 519 target = None
520 520 collapse = False
521 521 external = nullrev
522 522 state = {}
523 523 f = repo.opener("rebasestate")
524 524 for i, l in enumerate(f.read().splitlines()):
525 525 if i == 0:
526 526 originalwd = repo[l].rev()
527 527 elif i == 1:
528 528 target = repo[l].rev()
529 529 elif i == 2:
530 530 external = repo[l].rev()
531 531 elif i == 3:
532 532 collapse = bool(int(l))
533 533 elif i == 4:
534 534 keep = bool(int(l))
535 535 elif i == 5:
536 536 keepbranches = bool(int(l))
537 537 else:
538 538 oldrev, newrev = l.split(':')
539 539 if newrev != str(nullmerge):
540 540 state[repo[oldrev].rev()] = repo[newrev].rev()
541 541 else:
542 542 state[repo[oldrev].rev()] = int(newrev)
543 543 skipped = set()
544 544 # recompute the set of skipped revs
545 545 if not collapse:
546 546 seen = set([target])
547 547 for old, new in sorted(state.items()):
548 548 if new != nullrev and new in seen:
549 549 skipped.add(old)
550 550 seen.add(new)
551 551 repo.ui.debug('computed skipped revs: %s\n' % skipped)
552 552 repo.ui.debug('rebase status resumed\n')
553 553 return (originalwd, target, state, skipped,
554 554 collapse, keep, keepbranches, external)
555 555 except IOError, err:
556 556 if err.errno != errno.ENOENT:
557 557 raise
558 558 raise util.Abort(_('no rebase in progress'))
559 559
560 560 def abort(repo, originalwd, target, state):
561 561 'Restore the repository to its original state'
562 562 dstates = [s for s in state.values() if s != nullrev]
563 563 immutable = [d for d in dstates if not repo[d].mutable()]
564 564 if immutable:
565 565 raise util.Abort(_("can't abort rebase due to immutable changesets %s")
566 566 % ', '.join(str(repo[r]) for r in immutable),
567 567 hint=_('see hg help phases for details'))
568 568
569 569 descendants = set()
570 570 if dstates:
571 571 descendants = set(repo.changelog.descendants(dstates))
572 572 if descendants - set(dstates):
573 573 repo.ui.warn(_("warning: new changesets detected on target branch, "
574 574 "can't abort\n"))
575 575 return -1
576 576 else:
577 577 # Strip from the first rebased revision
578 578 merge.update(repo, repo[originalwd].rev(), False, True, False)
579 579 rebased = filter(lambda x: x > -1 and x != target, state.values())
580 580 if rebased:
581 581 strippoint = min(rebased)
582 582 # no backup of rebased cset versions needed
583 583 repair.strip(repo.ui, repo, repo[strippoint].node())
584 584 clearstatus(repo)
585 585 repo.ui.warn(_('rebase aborted\n'))
586 586 return 0
587 587
588 588 def buildstate(repo, dest, rebaseset, collapse):
589 589 '''Define which revisions are going to be rebased and where
590 590
591 591 repo: repo
592 592 dest: context
593 593 rebaseset: set of rev
594 594 '''
595 595
596 596 # This check isn't strictly necessary, since mq detects commits over an
597 597 # applied patch. But it prevents messing up the working directory when
598 598 # a partially completed rebase is blocked by mq.
599 599 if 'qtip' in repo.tags() and (dest.node() in
600 600 [s.node for s in repo.mq.applied]):
601 601 raise util.Abort(_('cannot rebase onto an applied mq patch'))
602 602
603 603 roots = list(repo.set('roots(%ld)', rebaseset))
604 604 if not roots:
605 605 raise util.Abort(_('no matching revisions'))
606 606 if len(roots) > 1:
607 607 raise util.Abort(_("can't rebase multiple roots"))
608 608 root = roots[0]
609 609
610 610 commonbase = root.ancestor(dest)
611 611 if commonbase == root:
612 612 raise util.Abort(_('source is ancestor of destination'))
613 613 if commonbase == dest:
614 614 samebranch = root.branch() == dest.branch()
615 615 if not collapse and samebranch and root in dest.children():
616 616 repo.ui.debug('source is a child of destination\n')
617 617 return None
618 618
619 619 repo.ui.debug('rebase onto %d starting from %d\n' % (dest, root))
620 620 state = dict.fromkeys(rebaseset, nullrev)
621 621 # Rebase tries to turn <dest> into a parent of <root> while
622 622 # preserving the number of parents of rebased changesets:
623 623 #
624 624 # - A changeset with a single parent will always be rebased as a
625 625 # changeset with a single parent.
626 626 #
627 627 # - A merge will be rebased as merge unless its parents are both
628 628 # ancestors of <dest> or are themselves in the rebased set and
629 629 # pruned while rebased.
630 630 #
631 631 # If one parent of <root> is an ancestor of <dest>, the rebased
632 632 # version of this parent will be <dest>. This is always true with
633 633 # --base option.
634 634 #
635 635 # Otherwise, we need to *replace* the original parents with
636 636 # <dest>. This "detaches" the rebased set from its former location
637 637 # and rebases it onto <dest>. Changes introduced by ancestors of
638 638 # <root> not common with <dest> (the detachset, marked as
639 639 # nullmerge) are "removed" from the rebased changesets.
640 640 #
641 641 # - If <root> has a single parent, set it to <dest>.
642 642 #
643 643 # - If <root> is a merge, we cannot decide which parent to
644 644 # replace, the rebase operation is not clearly defined.
645 645 #
646 646 # The table below sums up this behavior:
647 647 #
648 648 # +--------------------+----------------------+-------------------------+
649 649 # | | one parent | merge |
650 650 # +--------------------+----------------------+-------------------------+
651 651 # | parent in ::<dest> | new parent is <dest> | parents in ::<dest> are |
652 652 # | | | remapped to <dest> |
653 653 # +--------------------+----------------------+-------------------------+
654 654 # | unrelated source | new parent is <dest> | ambiguous, abort |
655 655 # +--------------------+----------------------+-------------------------+
656 656 #
657 657 # The actual abort is handled by `defineparents`
658 658 if len(root.parents()) <= 1:
659 # (strict) ancestors of <root> not ancestors of <dest>
660 detachset = repo.revs('::%d - ::%d - %d', root, commonbase, root)
659 # ancestors of <root> not ancestors of <dest>
660 detachset = repo.changelog.findmissingrevs([commonbase.rev()],
661 [root.rev()])
661 662 state.update(dict.fromkeys(detachset, nullmerge))
663 # detachset can have root, and we definitely want to rebase that
664 state[root.rev()] = nullrev
662 665 return repo['.'].rev(), dest.rev(), state
663 666
664 667 def clearrebased(ui, repo, state, collapsedas=None):
665 668 """dispose of rebased revision at the end of the rebase
666 669
667 670 If `collapsedas` is not None, the rebase was a collapse whose result if the
668 671 `collapsedas` node."""
669 672 if obsolete._enabled:
670 673 markers = []
671 674 for rev, newrev in sorted(state.items()):
672 675 if newrev >= 0:
673 676 if collapsedas is not None:
674 677 newrev = collapsedas
675 678 markers.append((repo[rev], (repo[newrev],)))
676 679 if markers:
677 680 obsolete.createmarkers(repo, markers)
678 681 else:
679 682 rebased = [rev for rev in state if state[rev] != nullmerge]
680 683 if rebased:
681 684 if set(repo.changelog.descendants([min(rebased)])) - set(state):
682 685 ui.warn(_("warning: new changesets detected "
683 686 "on source branch, not stripping\n"))
684 687 else:
685 688 # backup the old csets by default
686 689 repair.strip(ui, repo, repo[min(rebased)].node(), "all")
687 690
688 691
689 692 def pullrebase(orig, ui, repo, *args, **opts):
690 693 'Call rebase after pull if the latter has been invoked with --rebase'
691 694 if opts.get('rebase'):
692 695 if opts.get('update'):
693 696 del opts['update']
694 697 ui.debug('--update and --rebase are not compatible, ignoring '
695 698 'the update flag\n')
696 699
697 700 movemarkfrom = repo['.'].node()
698 701 cmdutil.bailifchanged(repo)
699 702 revsprepull = len(repo)
700 703 origpostincoming = commands.postincoming
701 704 def _dummy(*args, **kwargs):
702 705 pass
703 706 commands.postincoming = _dummy
704 707 try:
705 708 orig(ui, repo, *args, **opts)
706 709 finally:
707 710 commands.postincoming = origpostincoming
708 711 revspostpull = len(repo)
709 712 if revspostpull > revsprepull:
710 713 rebase(ui, repo, **opts)
711 714 branch = repo[None].branch()
712 715 dest = repo[branch].rev()
713 716 if dest != repo['.'].rev():
714 717 # there was nothing to rebase we force an update
715 718 hg.update(repo, dest)
716 719 if bookmarks.update(repo, [movemarkfrom], repo['.'].node()):
717 720 ui.status(_("updating bookmark %s\n")
718 721 % repo._bookmarkcurrent)
719 722 else:
720 723 if opts.get('tool'):
721 724 raise util.Abort(_('--tool can only be used with --rebase'))
722 725 orig(ui, repo, *args, **opts)
723 726
724 727 def uisetup(ui):
725 728 'Replace pull with a decorator to provide --rebase option'
726 729 entry = extensions.wrapcommand(commands.table, 'pull', pullrebase)
727 730 entry[1].append(('', 'rebase', None,
728 731 _("rebase working directory to branch head")))
729 732 entry[1].append(('t', 'tool', '',
730 733 _("specify merge tool for rebase")))
General Comments 0
You need to be logged in to leave comments. Login now