##// END OF EJS Templates
rebase: don't try to prune obsolete changeset already in the destination...
Boris Feld -
r39951:7198cdbb default
parent child Browse files
Show More
@@ -1,1934 +1,1938 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 https://mercurial-scm.org/wiki/RebaseExtension
15 15 '''
16 16
17 17 from __future__ import absolute_import
18 18
19 19 import errno
20 20 import os
21 21
22 22 from mercurial.i18n import _
23 23 from mercurial.node import (
24 24 nullrev,
25 25 short,
26 26 )
27 27 from mercurial import (
28 28 bookmarks,
29 29 cmdutil,
30 30 commands,
31 31 copies,
32 32 destutil,
33 33 dirstateguard,
34 34 error,
35 35 extensions,
36 36 hg,
37 37 merge as mergemod,
38 38 mergeutil,
39 39 obsolete,
40 40 obsutil,
41 41 patch,
42 42 phases,
43 43 pycompat,
44 44 registrar,
45 45 repair,
46 46 revset,
47 47 revsetlang,
48 48 scmutil,
49 49 smartset,
50 50 state as statemod,
51 51 util,
52 52 )
53 53
54 54 # The following constants are used throughout the rebase module. The ordering of
55 55 # their values must be maintained.
56 56
57 57 # Indicates that a revision needs to be rebased
58 58 revtodo = -1
59 59 revtodostr = '-1'
60 60
61 61 # legacy revstates no longer needed in current code
62 62 # -2: nullmerge, -3: revignored, -4: revprecursor, -5: revpruned
63 63 legacystates = {'-2', '-3', '-4', '-5'}
64 64
65 65 cmdtable = {}
66 66 command = registrar.command(cmdtable)
67 67 # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for
68 68 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
69 69 # be specifying the version(s) of Mercurial they are tested with, or
70 70 # leave the attribute unspecified.
71 71 testedwith = 'ships-with-hg-core'
72 72
73 73 def _nothingtorebase():
74 74 return 1
75 75
76 76 def _savegraft(ctx, extra):
77 77 s = ctx.extra().get('source', None)
78 78 if s is not None:
79 79 extra['source'] = s
80 80 s = ctx.extra().get('intermediate-source', None)
81 81 if s is not None:
82 82 extra['intermediate-source'] = s
83 83
84 84 def _savebranch(ctx, extra):
85 85 extra['branch'] = ctx.branch()
86 86
87 87 def _destrebase(repo, sourceset, destspace=None):
88 88 """small wrapper around destmerge to pass the right extra args
89 89
90 90 Please wrap destutil.destmerge instead."""
91 91 return destutil.destmerge(repo, action='rebase', sourceset=sourceset,
92 92 onheadcheck=False, destspace=destspace)
93 93
94 94 revsetpredicate = registrar.revsetpredicate()
95 95
96 96 @revsetpredicate('_destrebase')
97 97 def _revsetdestrebase(repo, subset, x):
98 98 # ``_rebasedefaultdest()``
99 99
100 100 # default destination for rebase.
101 101 # # XXX: Currently private because I expect the signature to change.
102 102 # # XXX: - bailing out in case of ambiguity vs returning all data.
103 103 # i18n: "_rebasedefaultdest" is a keyword
104 104 sourceset = None
105 105 if x is not None:
106 106 sourceset = revset.getset(repo, smartset.fullreposet(repo), x)
107 107 return subset & smartset.baseset([_destrebase(repo, sourceset)])
108 108
109 109 @revsetpredicate('_destautoorphanrebase')
110 110 def _revsetdestautoorphanrebase(repo, subset, x):
111 111 """automatic rebase destination for a single orphan revision"""
112 112 unfi = repo.unfiltered()
113 113 obsoleted = unfi.revs('obsolete()')
114 114
115 115 src = revset.getset(repo, subset, x).first()
116 116
117 117 # Empty src or already obsoleted - Do not return a destination
118 118 if not src or src in obsoleted:
119 119 return smartset.baseset()
120 120 dests = destutil.orphanpossibledestination(repo, src)
121 121 if len(dests) > 1:
122 122 raise error.Abort(
123 123 _("ambiguous automatic rebase: %r could end up on any of %r") % (
124 124 src, dests))
125 125 # We have zero or one destination, so we can just return here.
126 126 return smartset.baseset(dests)
127 127
128 128 def _ctxdesc(ctx):
129 129 """short description for a context"""
130 130 desc = '%d:%s "%s"' % (ctx.rev(), ctx,
131 131 ctx.description().split('\n', 1)[0])
132 132 repo = ctx.repo()
133 133 names = []
134 134 for nsname, ns in repo.names.iteritems():
135 135 if nsname == 'branches':
136 136 continue
137 137 names.extend(ns.names(repo, ctx.node()))
138 138 if names:
139 139 desc += ' (%s)' % ' '.join(names)
140 140 return desc
141 141
142 142 class rebaseruntime(object):
143 143 """This class is a container for rebase runtime state"""
144 144 def __init__(self, repo, ui, inmemory=False, opts=None):
145 145 if opts is None:
146 146 opts = {}
147 147
148 148 # prepared: whether we have rebasestate prepared or not. Currently it
149 149 # decides whether "self.repo" is unfiltered or not.
150 150 # The rebasestate has explicit hash to hash instructions not depending
151 151 # on visibility. If rebasestate exists (in-memory or on-disk), use
152 152 # unfiltered repo to avoid visibility issues.
153 153 # Before knowing rebasestate (i.e. when starting a new rebase (not
154 154 # --continue or --abort)), the original repo should be used so
155 155 # visibility-dependent revsets are correct.
156 156 self.prepared = False
157 157 self._repo = repo
158 158
159 159 self.ui = ui
160 160 self.opts = opts
161 161 self.originalwd = None
162 162 self.external = nullrev
163 163 # Mapping between the old revision id and either what is the new rebased
164 164 # revision or what needs to be done with the old revision. The state
165 165 # dict will be what contains most of the rebase progress state.
166 166 self.state = {}
167 167 self.activebookmark = None
168 168 self.destmap = {}
169 169 self.skipped = set()
170 170
171 171 self.collapsef = opts.get('collapse', False)
172 172 self.collapsemsg = cmdutil.logmessage(ui, opts)
173 173 self.date = opts.get('date', None)
174 174
175 175 e = opts.get('extrafn') # internal, used by e.g. hgsubversion
176 176 self.extrafns = [_savegraft]
177 177 if e:
178 178 self.extrafns = [e]
179 179
180 180 self.backupf = ui.configbool('ui', 'history-editing-backup')
181 181 self.keepf = opts.get('keep', False)
182 182 self.keepbranchesf = opts.get('keepbranches', False)
183 183 self.obsoletenotrebased = {}
184 184 self.obsoletewithoutsuccessorindestination = set()
185 185 self.inmemory = inmemory
186 186 self.stateobj = statemod.cmdstate(repo, 'rebasestate')
187 187
188 188 @property
189 189 def repo(self):
190 190 if self.prepared:
191 191 return self._repo.unfiltered()
192 192 else:
193 193 return self._repo
194 194
195 195 def storestatus(self, tr=None):
196 196 """Store the current status to allow recovery"""
197 197 if tr:
198 198 tr.addfilegenerator('rebasestate', ('rebasestate',),
199 199 self._writestatus, location='plain')
200 200 else:
201 201 with self.repo.vfs("rebasestate", "w") as f:
202 202 self._writestatus(f)
203 203
204 204 def _writestatus(self, f):
205 205 repo = self.repo
206 206 assert repo.filtername is None
207 207 f.write(repo[self.originalwd].hex() + '\n')
208 208 # was "dest". we now write dest per src root below.
209 209 f.write('\n')
210 210 f.write(repo[self.external].hex() + '\n')
211 211 f.write('%d\n' % int(self.collapsef))
212 212 f.write('%d\n' % int(self.keepf))
213 213 f.write('%d\n' % int(self.keepbranchesf))
214 214 f.write('%s\n' % (self.activebookmark or ''))
215 215 destmap = self.destmap
216 216 for d, v in self.state.iteritems():
217 217 oldrev = repo[d].hex()
218 218 if v >= 0:
219 219 newrev = repo[v].hex()
220 220 else:
221 221 newrev = "%d" % v
222 222 destnode = repo[destmap[d]].hex()
223 223 f.write("%s:%s:%s\n" % (oldrev, newrev, destnode))
224 224 repo.ui.debug('rebase status stored\n')
225 225
226 226 def restorestatus(self):
227 227 """Restore a previously stored status"""
228 228 if not self.stateobj.exists():
229 229 cmdutil.wrongtooltocontinue(self.repo, _('rebase'))
230 230
231 231 data = self._read()
232 232 self.repo.ui.debug('rebase status resumed\n')
233 233
234 234 self.originalwd = data['originalwd']
235 235 self.destmap = data['destmap']
236 236 self.state = data['state']
237 237 self.skipped = data['skipped']
238 238 self.collapsef = data['collapse']
239 239 self.keepf = data['keep']
240 240 self.keepbranchesf = data['keepbranches']
241 241 self.external = data['external']
242 242 self.activebookmark = data['activebookmark']
243 243
244 244 def _read(self):
245 245 self.prepared = True
246 246 repo = self.repo
247 247 assert repo.filtername is None
248 248 data = {'keepbranches': None, 'collapse': None, 'activebookmark': None,
249 249 'external': nullrev, 'keep': None, 'originalwd': None}
250 250 legacydest = None
251 251 state = {}
252 252 destmap = {}
253 253
254 254 if True:
255 255 f = repo.vfs("rebasestate")
256 256 for i, l in enumerate(f.read().splitlines()):
257 257 if i == 0:
258 258 data['originalwd'] = repo[l].rev()
259 259 elif i == 1:
260 260 # this line should be empty in newer version. but legacy
261 261 # clients may still use it
262 262 if l:
263 263 legacydest = repo[l].rev()
264 264 elif i == 2:
265 265 data['external'] = repo[l].rev()
266 266 elif i == 3:
267 267 data['collapse'] = bool(int(l))
268 268 elif i == 4:
269 269 data['keep'] = bool(int(l))
270 270 elif i == 5:
271 271 data['keepbranches'] = bool(int(l))
272 272 elif i == 6 and not (len(l) == 81 and ':' in l):
273 273 # line 6 is a recent addition, so for backwards
274 274 # compatibility check that the line doesn't look like the
275 275 # oldrev:newrev lines
276 276 data['activebookmark'] = l
277 277 else:
278 278 args = l.split(':')
279 279 oldrev = repo[args[0]].rev()
280 280 newrev = args[1]
281 281 if newrev in legacystates:
282 282 continue
283 283 if len(args) > 2:
284 284 destrev = repo[args[2]].rev()
285 285 else:
286 286 destrev = legacydest
287 287 destmap[oldrev] = destrev
288 288 if newrev == revtodostr:
289 289 state[oldrev] = revtodo
290 290 # Legacy compat special case
291 291 else:
292 292 state[oldrev] = repo[newrev].rev()
293 293
294 294 if data['keepbranches'] is None:
295 295 raise error.Abort(_('.hg/rebasestate is incomplete'))
296 296
297 297 data['destmap'] = destmap
298 298 data['state'] = state
299 299 skipped = set()
300 300 # recompute the set of skipped revs
301 301 if not data['collapse']:
302 302 seen = set(destmap.values())
303 303 for old, new in sorted(state.items()):
304 304 if new != revtodo and new in seen:
305 305 skipped.add(old)
306 306 seen.add(new)
307 307 data['skipped'] = skipped
308 308 repo.ui.debug('computed skipped revs: %s\n' %
309 309 (' '.join('%d' % r for r in sorted(skipped)) or ''))
310 310
311 311 return data
312 312
313 313 def _handleskippingobsolete(self, obsoleterevs, destmap):
314 314 """Compute structures necessary for skipping obsolete revisions
315 315
316 316 obsoleterevs: iterable of all obsolete revisions in rebaseset
317 317 destmap: {srcrev: destrev} destination revisions
318 318 """
319 319 self.obsoletenotrebased = {}
320 320 if not self.ui.configbool('experimental', 'rebaseskipobsolete'):
321 321 return
322 322 obsoleteset = set(obsoleterevs)
323 323 (self.obsoletenotrebased,
324 324 self.obsoletewithoutsuccessorindestination,
325 325 obsoleteextinctsuccessors) = _computeobsoletenotrebased(
326 326 self.repo, obsoleteset, destmap)
327 327 skippedset = set(self.obsoletenotrebased)
328 328 skippedset.update(self.obsoletewithoutsuccessorindestination)
329 329 skippedset.update(obsoleteextinctsuccessors)
330 330 _checkobsrebase(self.repo, self.ui, obsoleteset, skippedset)
331 331
332 332 def _prepareabortorcontinue(self, isabort, backup=True, suppwarns=False):
333 333 try:
334 334 self.restorestatus()
335 335 self.collapsemsg = restorecollapsemsg(self.repo, isabort)
336 336 except error.RepoLookupError:
337 337 if isabort:
338 338 clearstatus(self.repo)
339 339 clearcollapsemsg(self.repo)
340 340 self.repo.ui.warn(_('rebase aborted (no revision is removed,'
341 341 ' only broken state is cleared)\n'))
342 342 return 0
343 343 else:
344 344 msg = _('cannot continue inconsistent rebase')
345 345 hint = _('use "hg rebase --abort" to clear broken state')
346 346 raise error.Abort(msg, hint=hint)
347 347
348 348 if isabort:
349 349 backup = backup and self.backupf
350 350 return abort(self.repo, self.originalwd, self.destmap, self.state,
351 351 activebookmark=self.activebookmark, backup=backup,
352 352 suppwarns=suppwarns)
353 353
354 354 def _preparenewrebase(self, destmap):
355 355 if not destmap:
356 356 return _nothingtorebase()
357 357
358 358 rebaseset = destmap.keys()
359 359 allowunstable = obsolete.isenabled(self.repo, obsolete.allowunstableopt)
360 360 if (not (self.keepf or allowunstable)
361 361 and self.repo.revs('first(children(%ld) - %ld)',
362 362 rebaseset, rebaseset)):
363 363 raise error.Abort(
364 364 _("can't remove original changesets with"
365 365 " unrebased descendants"),
366 366 hint=_('use --keep to keep original changesets'))
367 367
368 368 result = buildstate(self.repo, destmap, self.collapsef)
369 369
370 370 if not result:
371 371 # Empty state built, nothing to rebase
372 372 self.ui.status(_('nothing to rebase\n'))
373 373 return _nothingtorebase()
374 374
375 375 for root in self.repo.set('roots(%ld)', rebaseset):
376 376 if not self.keepf and not root.mutable():
377 377 raise error.Abort(_("can't rebase public changeset %s")
378 378 % root,
379 379 hint=_("see 'hg help phases' for details"))
380 380
381 381 (self.originalwd, self.destmap, self.state) = result
382 382 if self.collapsef:
383 383 dests = set(self.destmap.values())
384 384 if len(dests) != 1:
385 385 raise error.Abort(
386 386 _('--collapse does not work with multiple destinations'))
387 387 destrev = next(iter(dests))
388 388 destancestors = self.repo.changelog.ancestors([destrev],
389 389 inclusive=True)
390 390 self.external = externalparent(self.repo, self.state, destancestors)
391 391
392 392 for destrev in sorted(set(destmap.values())):
393 393 dest = self.repo[destrev]
394 394 if dest.closesbranch() and not self.keepbranchesf:
395 395 self.ui.status(_('reopening closed branch head %s\n') % dest)
396 396
397 397 self.prepared = True
398 398
399 399 def _assignworkingcopy(self):
400 400 if self.inmemory:
401 401 from mercurial.context import overlayworkingctx
402 402 self.wctx = overlayworkingctx(self.repo)
403 403 self.repo.ui.debug("rebasing in-memory\n")
404 404 else:
405 405 self.wctx = self.repo[None]
406 406 self.repo.ui.debug("rebasing on disk\n")
407 407 self.repo.ui.log("rebase", "", rebase_imm_used=self.inmemory)
408 408
409 409 def _performrebase(self, tr):
410 410 self._assignworkingcopy()
411 411 repo, ui = self.repo, self.ui
412 412 if self.keepbranchesf:
413 413 # insert _savebranch at the start of extrafns so if
414 414 # there's a user-provided extrafn it can clobber branch if
415 415 # desired
416 416 self.extrafns.insert(0, _savebranch)
417 417 if self.collapsef:
418 418 branches = set()
419 419 for rev in self.state:
420 420 branches.add(repo[rev].branch())
421 421 if len(branches) > 1:
422 422 raise error.Abort(_('cannot collapse multiple named '
423 423 'branches'))
424 424
425 425 # Calculate self.obsoletenotrebased
426 426 obsrevs = _filterobsoleterevs(self.repo, self.state)
427 427 self._handleskippingobsolete(obsrevs, self.destmap)
428 428
429 429 # Keep track of the active bookmarks in order to reset them later
430 430 self.activebookmark = self.activebookmark or repo._activebookmark
431 431 if self.activebookmark:
432 432 bookmarks.deactivate(repo)
433 433
434 434 # Store the state before we begin so users can run 'hg rebase --abort'
435 435 # if we fail before the transaction closes.
436 436 self.storestatus()
437 437 if tr:
438 438 # When using single transaction, store state when transaction
439 439 # commits.
440 440 self.storestatus(tr)
441 441
442 442 cands = [k for k, v in self.state.iteritems() if v == revtodo]
443 443 p = repo.ui.makeprogress(_("rebasing"), unit=_('changesets'),
444 444 total=len(cands))
445 445 def progress(ctx):
446 446 p.increment(item=("%d:%s" % (ctx.rev(), ctx)))
447 447 allowdivergence = self.ui.configbool(
448 448 'experimental', 'evolution.allowdivergence')
449 449 for subset in sortsource(self.destmap):
450 450 sortedrevs = self.repo.revs('sort(%ld, -topo)', subset)
451 451 if not allowdivergence:
452 452 sortedrevs -= self.repo.revs(
453 453 'descendants(%ld) and not %ld',
454 454 self.obsoletewithoutsuccessorindestination,
455 455 self.obsoletewithoutsuccessorindestination,
456 456 )
457 457 for rev in sortedrevs:
458 458 self._rebasenode(tr, rev, allowdivergence, progress)
459 459 p.complete()
460 460 ui.note(_('rebase merging completed\n'))
461 461
462 462 def _concludenode(self, rev, p1, p2, editor, commitmsg=None):
463 463 '''Commit the wd changes with parents p1 and p2.
464 464
465 465 Reuse commit info from rev but also store useful information in extra.
466 466 Return node of committed revision.'''
467 467 repo = self.repo
468 468 ctx = repo[rev]
469 469 if commitmsg is None:
470 470 commitmsg = ctx.description()
471 471 date = self.date
472 472 if date is None:
473 473 date = ctx.date()
474 474 extra = {'rebase_source': ctx.hex()}
475 475 for c in self.extrafns:
476 476 c(ctx, extra)
477 477 keepbranch = self.keepbranchesf and repo[p1].branch() != ctx.branch()
478 478 destphase = max(ctx.phase(), phases.draft)
479 479 overrides = {('phases', 'new-commit'): destphase}
480 480 if keepbranch:
481 481 overrides[('ui', 'allowemptycommit')] = True
482 482 with repo.ui.configoverride(overrides, 'rebase'):
483 483 if self.inmemory:
484 484 newnode = commitmemorynode(repo, p1, p2,
485 485 wctx=self.wctx,
486 486 extra=extra,
487 487 commitmsg=commitmsg,
488 488 editor=editor,
489 489 user=ctx.user(),
490 490 date=date)
491 491 mergemod.mergestate.clean(repo)
492 492 else:
493 493 newnode = commitnode(repo, p1, p2,
494 494 extra=extra,
495 495 commitmsg=commitmsg,
496 496 editor=editor,
497 497 user=ctx.user(),
498 498 date=date)
499 499
500 500 if newnode is None:
501 501 # If it ended up being a no-op commit, then the normal
502 502 # merge state clean-up path doesn't happen, so do it
503 503 # here. Fix issue5494
504 504 mergemod.mergestate.clean(repo)
505 505 return newnode
506 506
507 507 def _rebasenode(self, tr, rev, allowdivergence, progressfn):
508 508 repo, ui, opts = self.repo, self.ui, self.opts
509 509 dest = self.destmap[rev]
510 510 ctx = repo[rev]
511 511 desc = _ctxdesc(ctx)
512 512 if self.state[rev] == rev:
513 513 ui.status(_('already rebased %s\n') % desc)
514 514 elif (not allowdivergence
515 515 and rev in self.obsoletewithoutsuccessorindestination):
516 516 msg = _('note: not rebasing %s and its descendants as '
517 517 'this would cause divergence\n') % desc
518 518 repo.ui.status(msg)
519 519 self.skipped.add(rev)
520 520 elif rev in self.obsoletenotrebased:
521 521 succ = self.obsoletenotrebased[rev]
522 522 if succ is None:
523 523 msg = _('note: not rebasing %s, it has no '
524 524 'successor\n') % desc
525 525 else:
526 526 succdesc = _ctxdesc(repo[succ])
527 527 msg = (_('note: not rebasing %s, already in '
528 528 'destination as %s\n') % (desc, succdesc))
529 529 repo.ui.status(msg)
530 530 # Make clearrebased aware state[rev] is not a true successor
531 531 self.skipped.add(rev)
532 532 # Record rev as moved to its desired destination in self.state.
533 533 # This helps bookmark and working parent movement.
534 534 dest = max(adjustdest(repo, rev, self.destmap, self.state,
535 535 self.skipped))
536 536 self.state[rev] = dest
537 537 elif self.state[rev] == revtodo:
538 538 ui.status(_('rebasing %s\n') % desc)
539 539 progressfn(ctx)
540 540 p1, p2, base = defineparents(repo, rev, self.destmap,
541 541 self.state, self.skipped,
542 542 self.obsoletenotrebased)
543 543 if len(repo[None].parents()) == 2:
544 544 repo.ui.debug('resuming interrupted rebase\n')
545 545 else:
546 546 overrides = {('ui', 'forcemerge'): opts.get('tool', '')}
547 547 with ui.configoverride(overrides, 'rebase'):
548 548 stats = rebasenode(repo, rev, p1, base, self.collapsef,
549 549 dest, wctx=self.wctx)
550 550 if stats.unresolvedcount > 0:
551 551 if self.inmemory:
552 552 raise error.InMemoryMergeConflictsError()
553 553 else:
554 554 raise error.InterventionRequired(
555 555 _('unresolved conflicts (see hg '
556 556 'resolve, then hg rebase --continue)'))
557 557 if not self.collapsef:
558 558 merging = p2 != nullrev
559 559 editform = cmdutil.mergeeditform(merging, 'rebase')
560 560 editor = cmdutil.getcommiteditor(editform=editform,
561 561 **pycompat.strkwargs(opts))
562 562 newnode = self._concludenode(rev, p1, p2, editor)
563 563 else:
564 564 # Skip commit if we are collapsing
565 565 if self.inmemory:
566 566 self.wctx.setbase(repo[p1])
567 567 else:
568 568 repo.setparents(repo[p1].node())
569 569 newnode = None
570 570 # Update the state
571 571 if newnode is not None:
572 572 self.state[rev] = repo[newnode].rev()
573 573 ui.debug('rebased as %s\n' % short(newnode))
574 574 else:
575 575 if not self.collapsef:
576 576 ui.warn(_('note: rebase of %d:%s created no changes '
577 577 'to commit\n') % (rev, ctx))
578 578 self.skipped.add(rev)
579 579 self.state[rev] = p1
580 580 ui.debug('next revision set to %d\n' % p1)
581 581 else:
582 582 ui.status(_('already rebased %s as %s\n') %
583 583 (desc, repo[self.state[rev]]))
584 584 if not tr:
585 585 # When not using single transaction, store state after each
586 586 # commit is completely done. On InterventionRequired, we thus
587 587 # won't store the status. Instead, we'll hit the "len(parents) == 2"
588 588 # case and realize that the commit was in progress.
589 589 self.storestatus()
590 590
591 591 def _finishrebase(self):
592 592 repo, ui, opts = self.repo, self.ui, self.opts
593 593 fm = ui.formatter('rebase', opts)
594 594 fm.startitem()
595 595 if self.collapsef:
596 596 p1, p2, _base = defineparents(repo, min(self.state), self.destmap,
597 597 self.state, self.skipped,
598 598 self.obsoletenotrebased)
599 599 editopt = opts.get('edit')
600 600 editform = 'rebase.collapse'
601 601 if self.collapsemsg:
602 602 commitmsg = self.collapsemsg
603 603 else:
604 604 commitmsg = 'Collapsed revision'
605 605 for rebased in sorted(self.state):
606 606 if rebased not in self.skipped:
607 607 commitmsg += '\n* %s' % repo[rebased].description()
608 608 editopt = True
609 609 editor = cmdutil.getcommiteditor(edit=editopt, editform=editform)
610 610 revtoreuse = max(self.state)
611 611
612 612 newnode = self._concludenode(revtoreuse, p1, self.external,
613 613 editor, commitmsg=commitmsg)
614 614
615 615 if newnode is not None:
616 616 newrev = repo[newnode].rev()
617 617 for oldrev in self.state:
618 618 self.state[oldrev] = newrev
619 619
620 620 if 'qtip' in repo.tags():
621 621 updatemq(repo, self.state, self.skipped,
622 622 **pycompat.strkwargs(opts))
623 623
624 624 # restore original working directory
625 625 # (we do this before stripping)
626 626 newwd = self.state.get(self.originalwd, self.originalwd)
627 627 if newwd < 0:
628 628 # original directory is a parent of rebase set root or ignored
629 629 newwd = self.originalwd
630 630 if newwd not in [c.rev() for c in repo[None].parents()]:
631 631 ui.note(_("update back to initial working directory parent\n"))
632 632 hg.updaterepo(repo, newwd, overwrite=False)
633 633
634 634 collapsedas = None
635 635 if self.collapsef and not self.keepf:
636 636 collapsedas = newnode
637 637 clearrebased(ui, repo, self.destmap, self.state, self.skipped,
638 638 collapsedas, self.keepf, fm=fm, backup=self.backupf)
639 639
640 640 clearstatus(repo)
641 641 clearcollapsemsg(repo)
642 642
643 643 ui.note(_("rebase completed\n"))
644 644 util.unlinkpath(repo.sjoin('undo'), ignoremissing=True)
645 645 if self.skipped:
646 646 skippedlen = len(self.skipped)
647 647 ui.note(_("%d revisions have been skipped\n") % skippedlen)
648 648 fm.end()
649 649
650 650 if (self.activebookmark and self.activebookmark in repo._bookmarks and
651 651 repo['.'].node() == repo._bookmarks[self.activebookmark]):
652 652 bookmarks.activate(repo, self.activebookmark)
653 653
654 654 @command('rebase',
655 655 [('s', 'source', '',
656 656 _('rebase the specified changeset and descendants'), _('REV')),
657 657 ('b', 'base', '',
658 658 _('rebase everything from branching point of specified changeset'),
659 659 _('REV')),
660 660 ('r', 'rev', [],
661 661 _('rebase these revisions'),
662 662 _('REV')),
663 663 ('d', 'dest', '',
664 664 _('rebase onto the specified changeset'), _('REV')),
665 665 ('', 'collapse', False, _('collapse the rebased changesets')),
666 666 ('m', 'message', '',
667 667 _('use text as collapse commit message'), _('TEXT')),
668 668 ('e', 'edit', False, _('invoke editor on commit messages')),
669 669 ('l', 'logfile', '',
670 670 _('read collapse commit message from file'), _('FILE')),
671 671 ('k', 'keep', False, _('keep original changesets')),
672 672 ('', 'keepbranches', False, _('keep original branch names')),
673 673 ('D', 'detach', False, _('(DEPRECATED)')),
674 674 ('i', 'interactive', False, _('(DEPRECATED)')),
675 675 ('t', 'tool', '', _('specify merge tool')),
676 676 ('', 'stop', False, _('stop interrupted rebase')),
677 677 ('c', 'continue', False, _('continue an interrupted rebase')),
678 678 ('a', 'abort', False, _('abort an interrupted rebase')),
679 679 ('', 'auto-orphans', '', _('automatically rebase orphan revisions '
680 680 'in the specified revset (EXPERIMENTAL)')),
681 681 ] + cmdutil.dryrunopts + cmdutil.formatteropts + cmdutil.confirmopts,
682 682 _('[-s REV | -b REV] [-d REV] [OPTION]'))
683 683 def rebase(ui, repo, **opts):
684 684 """move changeset (and descendants) to a different branch
685 685
686 686 Rebase uses repeated merging to graft changesets from one part of
687 687 history (the source) onto another (the destination). This can be
688 688 useful for linearizing *local* changes relative to a master
689 689 development tree.
690 690
691 691 Published commits cannot be rebased (see :hg:`help phases`).
692 692 To copy commits, see :hg:`help graft`.
693 693
694 694 If you don't specify a destination changeset (``-d/--dest``), rebase
695 695 will use the same logic as :hg:`merge` to pick a destination. if
696 696 the current branch contains exactly one other head, the other head
697 697 is merged with by default. Otherwise, an explicit revision with
698 698 which to merge with must be provided. (destination changeset is not
699 699 modified by rebasing, but new changesets are added as its
700 700 descendants.)
701 701
702 702 Here are the ways to select changesets:
703 703
704 704 1. Explicitly select them using ``--rev``.
705 705
706 706 2. Use ``--source`` to select a root changeset and include all of its
707 707 descendants.
708 708
709 709 3. Use ``--base`` to select a changeset; rebase will find ancestors
710 710 and their descendants which are not also ancestors of the destination.
711 711
712 712 4. If you do not specify any of ``--rev``, ``source``, or ``--base``,
713 713 rebase will use ``--base .`` as above.
714 714
715 715 If ``--source`` or ``--rev`` is used, special names ``SRC`` and ``ALLSRC``
716 716 can be used in ``--dest``. Destination would be calculated per source
717 717 revision with ``SRC`` substituted by that single source revision and
718 718 ``ALLSRC`` substituted by all source revisions.
719 719
720 720 Rebase will destroy original changesets unless you use ``--keep``.
721 721 It will also move your bookmarks (even if you do).
722 722
723 723 Some changesets may be dropped if they do not contribute changes
724 724 (e.g. merges from the destination branch).
725 725
726 726 Unlike ``merge``, rebase will do nothing if you are at the branch tip of
727 727 a named branch with two heads. You will need to explicitly specify source
728 728 and/or destination.
729 729
730 730 If you need to use a tool to automate merge/conflict decisions, you
731 731 can specify one with ``--tool``, see :hg:`help merge-tools`.
732 732 As a caveat: the tool will not be used to mediate when a file was
733 733 deleted, there is no hook presently available for this.
734 734
735 735 If a rebase is interrupted to manually resolve a conflict, it can be
736 736 continued with --continue/-c, aborted with --abort/-a, or stopped with
737 737 --stop.
738 738
739 739 .. container:: verbose
740 740
741 741 Examples:
742 742
743 743 - move "local changes" (current commit back to branching point)
744 744 to the current branch tip after a pull::
745 745
746 746 hg rebase
747 747
748 748 - move a single changeset to the stable branch::
749 749
750 750 hg rebase -r 5f493448 -d stable
751 751
752 752 - splice a commit and all its descendants onto another part of history::
753 753
754 754 hg rebase --source c0c3 --dest 4cf9
755 755
756 756 - rebase everything on a branch marked by a bookmark onto the
757 757 default branch::
758 758
759 759 hg rebase --base myfeature --dest default
760 760
761 761 - collapse a sequence of changes into a single commit::
762 762
763 763 hg rebase --collapse -r 1520:1525 -d .
764 764
765 765 - move a named branch while preserving its name::
766 766
767 767 hg rebase -r "branch(featureX)" -d 1.3 --keepbranches
768 768
769 769 - stabilize orphaned changesets so history looks linear::
770 770
771 771 hg rebase -r 'orphan()-obsolete()'\
772 772 -d 'first(max((successors(max(roots(ALLSRC) & ::SRC)^)-obsolete())::) +\
773 773 max(::((roots(ALLSRC) & ::SRC)^)-obsolete()))'
774 774
775 775 Configuration Options:
776 776
777 777 You can make rebase require a destination if you set the following config
778 778 option::
779 779
780 780 [commands]
781 781 rebase.requiredest = True
782 782
783 783 By default, rebase will close the transaction after each commit. For
784 784 performance purposes, you can configure rebase to use a single transaction
785 785 across the entire rebase. WARNING: This setting introduces a significant
786 786 risk of losing the work you've done in a rebase if the rebase aborts
787 787 unexpectedly::
788 788
789 789 [rebase]
790 790 singletransaction = True
791 791
792 792 By default, rebase writes to the working copy, but you can configure it to
793 793 run in-memory for for better performance, and to allow it to run if the
794 794 working copy is dirty::
795 795
796 796 [rebase]
797 797 experimental.inmemory = True
798 798
799 799 Return Values:
800 800
801 801 Returns 0 on success, 1 if nothing to rebase or there are
802 802 unresolved conflicts.
803 803
804 804 """
805 805 opts = pycompat.byteskwargs(opts)
806 806 inmemory = ui.configbool('rebase', 'experimental.inmemory')
807 807 dryrun = opts.get('dry_run')
808 808 confirm = opts.get('confirm')
809 809 selactions = [k for k in ['abort', 'stop', 'continue'] if opts.get(k)]
810 810 if len(selactions) > 1:
811 811 raise error.Abort(_('cannot use --%s with --%s')
812 812 % tuple(selactions[:2]))
813 813 action = selactions[0] if selactions else None
814 814 if dryrun and action:
815 815 raise error.Abort(_('cannot specify both --dry-run and --%s') % action)
816 816 if confirm and action:
817 817 raise error.Abort(_('cannot specify both --confirm and --%s') % action)
818 818 if dryrun and confirm:
819 819 raise error.Abort(_('cannot specify both --confirm and --dry-run'))
820 820
821 821 if action or repo.currenttransaction() is not None:
822 822 # in-memory rebase is not compatible with resuming rebases.
823 823 # (Or if it is run within a transaction, since the restart logic can
824 824 # fail the entire transaction.)
825 825 inmemory = False
826 826
827 827 if opts.get('auto_orphans'):
828 828 for key in opts:
829 829 if key != 'auto_orphans' and opts.get(key):
830 830 raise error.Abort(_('--auto-orphans is incompatible with %s') %
831 831 ('--' + key))
832 832 userrevs = list(repo.revs(opts.get('auto_orphans')))
833 833 opts['rev'] = [revsetlang.formatspec('%ld and orphan()', userrevs)]
834 834 opts['dest'] = '_destautoorphanrebase(SRC)'
835 835
836 836 if dryrun or confirm:
837 837 return _dryrunrebase(ui, repo, action, opts)
838 838 elif action == 'stop':
839 839 rbsrt = rebaseruntime(repo, ui)
840 840 with repo.wlock(), repo.lock():
841 841 rbsrt.restorestatus()
842 842 if rbsrt.collapsef:
843 843 raise error.Abort(_("cannot stop in --collapse session"))
844 844 allowunstable = obsolete.isenabled(repo, obsolete.allowunstableopt)
845 845 if not (rbsrt.keepf or allowunstable):
846 846 raise error.Abort(_("cannot remove original changesets with"
847 847 " unrebased descendants"),
848 848 hint=_('either enable obsmarkers to allow unstable '
849 849 'revisions or use --keep to keep original '
850 850 'changesets'))
851 851 if needupdate(repo, rbsrt.state):
852 852 # update to the current working revision
853 853 # to clear interrupted merge
854 854 hg.updaterepo(repo, rbsrt.originalwd, overwrite=True)
855 855 rbsrt._finishrebase()
856 856 return 0
857 857 elif inmemory:
858 858 try:
859 859 # in-memory merge doesn't support conflicts, so if we hit any, abort
860 860 # and re-run as an on-disk merge.
861 861 overrides = {('rebase', 'singletransaction'): True}
862 862 with ui.configoverride(overrides, 'rebase'):
863 863 return _dorebase(ui, repo, action, opts, inmemory=inmemory)
864 864 except error.InMemoryMergeConflictsError:
865 865 ui.warn(_('hit merge conflicts; re-running rebase without in-memory'
866 866 ' merge\n'))
867 867 _dorebase(ui, repo, action='abort', opts={})
868 868 return _dorebase(ui, repo, action, opts, inmemory=False)
869 869 else:
870 870 return _dorebase(ui, repo, action, opts)
871 871
872 872 def _dryrunrebase(ui, repo, action, opts):
873 873 rbsrt = rebaseruntime(repo, ui, inmemory=True, opts=opts)
874 874 confirm = opts.get('confirm')
875 875 if confirm:
876 876 ui.status(_('starting in-memory rebase\n'))
877 877 else:
878 878 ui.status(_('starting dry-run rebase; repository will not be '
879 879 'changed\n'))
880 880 with repo.wlock(), repo.lock():
881 881 needsabort = True
882 882 try:
883 883 overrides = {('rebase', 'singletransaction'): True}
884 884 with ui.configoverride(overrides, 'rebase'):
885 885 _origrebase(ui, repo, action, opts, rbsrt, inmemory=True,
886 886 leaveunfinished=True)
887 887 except error.InMemoryMergeConflictsError:
888 888 ui.status(_('hit a merge conflict\n'))
889 889 return 1
890 890 else:
891 891 if confirm:
892 892 ui.status(_('rebase completed successfully\n'))
893 893 if not ui.promptchoice(_(b'apply changes (yn)?'
894 894 b'$$ &Yes $$ &No')):
895 895 # finish unfinished rebase
896 896 rbsrt._finishrebase()
897 897 else:
898 898 rbsrt._prepareabortorcontinue(isabort=True, backup=False,
899 899 suppwarns=True)
900 900 needsabort = False
901 901 else:
902 902 ui.status(_('dry-run rebase completed successfully; run without'
903 903 ' -n/--dry-run to perform this rebase\n'))
904 904 return 0
905 905 finally:
906 906 if needsabort:
907 907 # no need to store backup in case of dryrun
908 908 rbsrt._prepareabortorcontinue(isabort=True, backup=False,
909 909 suppwarns=True)
910 910
911 911 def _dorebase(ui, repo, action, opts, inmemory=False):
912 912 rbsrt = rebaseruntime(repo, ui, inmemory, opts)
913 913 return _origrebase(ui, repo, action, opts, rbsrt, inmemory=inmemory)
914 914
915 915 def _origrebase(ui, repo, action, opts, rbsrt, inmemory=False,
916 916 leaveunfinished=False):
917 917 assert action != 'stop'
918 918 with repo.wlock(), repo.lock():
919 919 # Validate input and define rebasing points
920 920 destf = opts.get('dest', None)
921 921 srcf = opts.get('source', None)
922 922 basef = opts.get('base', None)
923 923 revf = opts.get('rev', [])
924 924 # search default destination in this space
925 925 # used in the 'hg pull --rebase' case, see issue 5214.
926 926 destspace = opts.get('_destspace')
927 927 if opts.get('interactive'):
928 928 try:
929 929 if extensions.find('histedit'):
930 930 enablehistedit = ''
931 931 except KeyError:
932 932 enablehistedit = " --config extensions.histedit="
933 933 help = "hg%s help -e histedit" % enablehistedit
934 934 msg = _("interactive history editing is supported by the "
935 935 "'histedit' extension (see \"%s\")") % help
936 936 raise error.Abort(msg)
937 937
938 938 if rbsrt.collapsemsg and not rbsrt.collapsef:
939 939 raise error.Abort(
940 940 _('message can only be specified with collapse'))
941 941
942 942 if action:
943 943 if rbsrt.collapsef:
944 944 raise error.Abort(
945 945 _('cannot use collapse with continue or abort'))
946 946 if srcf or basef or destf:
947 947 raise error.Abort(
948 948 _('abort and continue do not allow specifying revisions'))
949 949 if action == 'abort' and opts.get('tool', False):
950 950 ui.warn(_('tool option will be ignored\n'))
951 951 if action == 'continue':
952 952 ms = mergemod.mergestate.read(repo)
953 953 mergeutil.checkunresolved(ms)
954 954
955 955 retcode = rbsrt._prepareabortorcontinue(isabort=(action == 'abort'))
956 956 if retcode is not None:
957 957 return retcode
958 958 else:
959 959 destmap = _definedestmap(ui, repo, inmemory, destf, srcf, basef,
960 960 revf, destspace=destspace)
961 961 retcode = rbsrt._preparenewrebase(destmap)
962 962 if retcode is not None:
963 963 return retcode
964 964 storecollapsemsg(repo, rbsrt.collapsemsg)
965 965
966 966 tr = None
967 967
968 968 singletr = ui.configbool('rebase', 'singletransaction')
969 969 if singletr:
970 970 tr = repo.transaction('rebase')
971 971
972 972 # If `rebase.singletransaction` is enabled, wrap the entire operation in
973 973 # one transaction here. Otherwise, transactions are obtained when
974 974 # committing each node, which is slower but allows partial success.
975 975 with util.acceptintervention(tr):
976 976 # Same logic for the dirstate guard, except we don't create one when
977 977 # rebasing in-memory (it's not needed).
978 978 dsguard = None
979 979 if singletr and not inmemory:
980 980 dsguard = dirstateguard.dirstateguard(repo, 'rebase')
981 981 with util.acceptintervention(dsguard):
982 982 rbsrt._performrebase(tr)
983 983 if not leaveunfinished:
984 984 rbsrt._finishrebase()
985 985
986 986 def _definedestmap(ui, repo, inmemory, destf=None, srcf=None, basef=None,
987 987 revf=None, destspace=None):
988 988 """use revisions argument to define destmap {srcrev: destrev}"""
989 989 if revf is None:
990 990 revf = []
991 991
992 992 # destspace is here to work around issues with `hg pull --rebase` see
993 993 # issue5214 for details
994 994 if srcf and basef:
995 995 raise error.Abort(_('cannot specify both a source and a base'))
996 996 if revf and basef:
997 997 raise error.Abort(_('cannot specify both a revision and a base'))
998 998 if revf and srcf:
999 999 raise error.Abort(_('cannot specify both a revision and a source'))
1000 1000
1001 1001 if not inmemory:
1002 1002 cmdutil.checkunfinished(repo)
1003 1003 cmdutil.bailifchanged(repo)
1004 1004
1005 1005 if ui.configbool('commands', 'rebase.requiredest') and not destf:
1006 1006 raise error.Abort(_('you must specify a destination'),
1007 1007 hint=_('use: hg rebase -d REV'))
1008 1008
1009 1009 dest = None
1010 1010
1011 1011 if revf:
1012 1012 rebaseset = scmutil.revrange(repo, revf)
1013 1013 if not rebaseset:
1014 1014 ui.status(_('empty "rev" revision set - nothing to rebase\n'))
1015 1015 return None
1016 1016 elif srcf:
1017 1017 src = scmutil.revrange(repo, [srcf])
1018 1018 if not src:
1019 1019 ui.status(_('empty "source" revision set - nothing to rebase\n'))
1020 1020 return None
1021 1021 rebaseset = repo.revs('(%ld)::', src)
1022 1022 assert rebaseset
1023 1023 else:
1024 1024 base = scmutil.revrange(repo, [basef or '.'])
1025 1025 if not base:
1026 1026 ui.status(_('empty "base" revision set - '
1027 1027 "can't compute rebase set\n"))
1028 1028 return None
1029 1029 if destf:
1030 1030 # --base does not support multiple destinations
1031 1031 dest = scmutil.revsingle(repo, destf)
1032 1032 else:
1033 1033 dest = repo[_destrebase(repo, base, destspace=destspace)]
1034 1034 destf = bytes(dest)
1035 1035
1036 1036 roots = [] # selected children of branching points
1037 1037 bpbase = {} # {branchingpoint: [origbase]}
1038 1038 for b in base: # group bases by branching points
1039 1039 bp = repo.revs('ancestor(%d, %d)', b, dest.rev()).first()
1040 1040 bpbase[bp] = bpbase.get(bp, []) + [b]
1041 1041 if None in bpbase:
1042 1042 # emulate the old behavior, showing "nothing to rebase" (a better
1043 1043 # behavior may be abort with "cannot find branching point" error)
1044 1044 bpbase.clear()
1045 1045 for bp, bs in bpbase.iteritems(): # calculate roots
1046 1046 roots += list(repo.revs('children(%d) & ancestors(%ld)', bp, bs))
1047 1047
1048 1048 rebaseset = repo.revs('%ld::', roots)
1049 1049
1050 1050 if not rebaseset:
1051 1051 # transform to list because smartsets are not comparable to
1052 1052 # lists. This should be improved to honor laziness of
1053 1053 # smartset.
1054 1054 if list(base) == [dest.rev()]:
1055 1055 if basef:
1056 1056 ui.status(_('nothing to rebase - %s is both "base"'
1057 1057 ' and destination\n') % dest)
1058 1058 else:
1059 1059 ui.status(_('nothing to rebase - working directory '
1060 1060 'parent is also destination\n'))
1061 1061 elif not repo.revs('%ld - ::%d', base, dest.rev()):
1062 1062 if basef:
1063 1063 ui.status(_('nothing to rebase - "base" %s is '
1064 1064 'already an ancestor of destination '
1065 1065 '%s\n') %
1066 1066 ('+'.join(bytes(repo[r]) for r in base),
1067 1067 dest))
1068 1068 else:
1069 1069 ui.status(_('nothing to rebase - working '
1070 1070 'directory parent is already an '
1071 1071 'ancestor of destination %s\n') % dest)
1072 1072 else: # can it happen?
1073 1073 ui.status(_('nothing to rebase from %s to %s\n') %
1074 1074 ('+'.join(bytes(repo[r]) for r in base), dest))
1075 1075 return None
1076 1076
1077 1077 rebasingwcp = repo['.'].rev() in rebaseset
1078 1078 ui.log("rebase", "", rebase_rebasing_wcp=rebasingwcp)
1079 1079 if inmemory and rebasingwcp:
1080 1080 # Check these since we did not before.
1081 1081 cmdutil.checkunfinished(repo)
1082 1082 cmdutil.bailifchanged(repo)
1083 1083
1084 1084 if not destf:
1085 1085 dest = repo[_destrebase(repo, rebaseset, destspace=destspace)]
1086 1086 destf = bytes(dest)
1087 1087
1088 1088 allsrc = revsetlang.formatspec('%ld', rebaseset)
1089 1089 alias = {'ALLSRC': allsrc}
1090 1090
1091 1091 if dest is None:
1092 1092 try:
1093 1093 # fast path: try to resolve dest without SRC alias
1094 1094 dest = scmutil.revsingle(repo, destf, localalias=alias)
1095 1095 except error.RepoLookupError:
1096 1096 # multi-dest path: resolve dest for each SRC separately
1097 1097 destmap = {}
1098 1098 for r in rebaseset:
1099 1099 alias['SRC'] = revsetlang.formatspec('%d', r)
1100 1100 # use repo.anyrevs instead of scmutil.revsingle because we
1101 1101 # don't want to abort if destset is empty.
1102 1102 destset = repo.anyrevs([destf], user=True, localalias=alias)
1103 1103 size = len(destset)
1104 1104 if size == 1:
1105 1105 destmap[r] = destset.first()
1106 1106 elif size == 0:
1107 1107 ui.note(_('skipping %s - empty destination\n') % repo[r])
1108 1108 else:
1109 1109 raise error.Abort(_('rebase destination for %s is not '
1110 1110 'unique') % repo[r])
1111 1111
1112 1112 if dest is not None:
1113 1113 # single-dest case: assign dest to each rev in rebaseset
1114 1114 destrev = dest.rev()
1115 1115 destmap = {r: destrev for r in rebaseset} # {srcrev: destrev}
1116 1116
1117 1117 if not destmap:
1118 1118 ui.status(_('nothing to rebase - empty destination\n'))
1119 1119 return None
1120 1120
1121 1121 return destmap
1122 1122
1123 1123 def externalparent(repo, state, destancestors):
1124 1124 """Return the revision that should be used as the second parent
1125 1125 when the revisions in state is collapsed on top of destancestors.
1126 1126 Abort if there is more than one parent.
1127 1127 """
1128 1128 parents = set()
1129 1129 source = min(state)
1130 1130 for rev in state:
1131 1131 if rev == source:
1132 1132 continue
1133 1133 for p in repo[rev].parents():
1134 1134 if (p.rev() not in state
1135 1135 and p.rev() not in destancestors):
1136 1136 parents.add(p.rev())
1137 1137 if not parents:
1138 1138 return nullrev
1139 1139 if len(parents) == 1:
1140 1140 return parents.pop()
1141 1141 raise error.Abort(_('unable to collapse on top of %d, there is more '
1142 1142 'than one external parent: %s') %
1143 1143 (max(destancestors),
1144 1144 ', '.join("%d" % p for p in sorted(parents))))
1145 1145
1146 1146 def commitmemorynode(repo, p1, p2, wctx, editor, extra, user, date, commitmsg):
1147 1147 '''Commit the memory changes with parents p1 and p2.
1148 1148 Return node of committed revision.'''
1149 1149 # Replicates the empty check in ``repo.commit``.
1150 1150 if wctx.isempty() and not repo.ui.configbool('ui', 'allowemptycommit'):
1151 1151 return None
1152 1152
1153 1153 # By convention, ``extra['branch']`` (set by extrafn) clobbers
1154 1154 # ``branch`` (used when passing ``--keepbranches``).
1155 1155 branch = repo[p1].branch()
1156 1156 if 'branch' in extra:
1157 1157 branch = extra['branch']
1158 1158
1159 1159 memctx = wctx.tomemctx(commitmsg, parents=(p1, p2), date=date,
1160 1160 extra=extra, user=user, branch=branch, editor=editor)
1161 1161 commitres = repo.commitctx(memctx)
1162 1162 wctx.clean() # Might be reused
1163 1163 return commitres
1164 1164
1165 1165 def commitnode(repo, p1, p2, editor, extra, user, date, commitmsg):
1166 1166 '''Commit the wd changes with parents p1 and p2.
1167 1167 Return node of committed revision.'''
1168 1168 dsguard = util.nullcontextmanager()
1169 1169 if not repo.ui.configbool('rebase', 'singletransaction'):
1170 1170 dsguard = dirstateguard.dirstateguard(repo, 'rebase')
1171 1171 with dsguard:
1172 1172 repo.setparents(repo[p1].node(), repo[p2].node())
1173 1173
1174 1174 # Commit might fail if unresolved files exist
1175 1175 newnode = repo.commit(text=commitmsg, user=user, date=date,
1176 1176 extra=extra, editor=editor)
1177 1177
1178 1178 repo.dirstate.setbranch(repo[newnode].branch())
1179 1179 return newnode
1180 1180
1181 1181 def rebasenode(repo, rev, p1, base, collapse, dest, wctx):
1182 1182 'Rebase a single revision rev on top of p1 using base as merge ancestor'
1183 1183 # Merge phase
1184 1184 # Update to destination and merge it with local
1185 1185 if wctx.isinmemory():
1186 1186 wctx.setbase(repo[p1])
1187 1187 else:
1188 1188 if repo['.'].rev() != p1:
1189 1189 repo.ui.debug(" update to %d:%s\n" % (p1, repo[p1]))
1190 1190 mergemod.update(repo, p1, False, True)
1191 1191 else:
1192 1192 repo.ui.debug(" already in destination\n")
1193 1193 # This is, alas, necessary to invalidate workingctx's manifest cache,
1194 1194 # as well as other data we litter on it in other places.
1195 1195 wctx = repo[None]
1196 1196 repo.dirstate.write(repo.currenttransaction())
1197 1197 repo.ui.debug(" merge against %d:%s\n" % (rev, repo[rev]))
1198 1198 if base is not None:
1199 1199 repo.ui.debug(" detach base %d:%s\n" % (base, repo[base]))
1200 1200 # When collapsing in-place, the parent is the common ancestor, we
1201 1201 # have to allow merging with it.
1202 1202 stats = mergemod.update(repo, rev, True, True, base, collapse,
1203 1203 labels=['dest', 'source'], wc=wctx)
1204 1204 if collapse:
1205 1205 copies.duplicatecopies(repo, wctx, rev, dest)
1206 1206 else:
1207 1207 # If we're not using --collapse, we need to
1208 1208 # duplicate copies between the revision we're
1209 1209 # rebasing and its first parent, but *not*
1210 1210 # duplicate any copies that have already been
1211 1211 # performed in the destination.
1212 1212 p1rev = repo[rev].p1().rev()
1213 1213 copies.duplicatecopies(repo, wctx, rev, p1rev, skiprev=dest)
1214 1214 return stats
1215 1215
1216 1216 def adjustdest(repo, rev, destmap, state, skipped):
1217 1217 """adjust rebase destination given the current rebase state
1218 1218
1219 1219 rev is what is being rebased. Return a list of two revs, which are the
1220 1220 adjusted destinations for rev's p1 and p2, respectively. If a parent is
1221 1221 nullrev, return dest without adjustment for it.
1222 1222
1223 1223 For example, when doing rebasing B+E to F, C to G, rebase will first move B
1224 1224 to B1, and E's destination will be adjusted from F to B1.
1225 1225
1226 1226 B1 <- written during rebasing B
1227 1227 |
1228 1228 F <- original destination of B, E
1229 1229 |
1230 1230 | E <- rev, which is being rebased
1231 1231 | |
1232 1232 | D <- prev, one parent of rev being checked
1233 1233 | |
1234 1234 | x <- skipped, ex. no successor or successor in (::dest)
1235 1235 | |
1236 1236 | C <- rebased as C', different destination
1237 1237 | |
1238 1238 | B <- rebased as B1 C'
1239 1239 |/ |
1240 1240 A G <- destination of C, different
1241 1241
1242 1242 Another example about merge changeset, rebase -r C+G+H -d K, rebase will
1243 1243 first move C to C1, G to G1, and when it's checking H, the adjusted
1244 1244 destinations will be [C1, G1].
1245 1245
1246 1246 H C1 G1
1247 1247 /| | /
1248 1248 F G |/
1249 1249 K | | -> K
1250 1250 | C D |
1251 1251 | |/ |
1252 1252 | B | ...
1253 1253 |/ |/
1254 1254 A A
1255 1255
1256 1256 Besides, adjust dest according to existing rebase information. For example,
1257 1257
1258 1258 B C D B needs to be rebased on top of C, C needs to be rebased on top
1259 1259 \|/ of D. We will rebase C first.
1260 1260 A
1261 1261
1262 1262 C' After rebasing C, when considering B's destination, use C'
1263 1263 | instead of the original C.
1264 1264 B D
1265 1265 \ /
1266 1266 A
1267 1267 """
1268 1268 # pick already rebased revs with same dest from state as interesting source
1269 1269 dest = destmap[rev]
1270 1270 source = [s for s, d in state.items()
1271 1271 if d > 0 and destmap[s] == dest and s not in skipped]
1272 1272
1273 1273 result = []
1274 1274 for prev in repo.changelog.parentrevs(rev):
1275 1275 adjusted = dest
1276 1276 if prev != nullrev:
1277 1277 candidate = repo.revs('max(%ld and (::%d))', source, prev).first()
1278 1278 if candidate is not None:
1279 1279 adjusted = state[candidate]
1280 1280 if adjusted == dest and dest in state:
1281 1281 adjusted = state[dest]
1282 1282 if adjusted == revtodo:
1283 1283 # sortsource should produce an order that makes this impossible
1284 1284 raise error.ProgrammingError(
1285 1285 'rev %d should be rebased already at this time' % dest)
1286 1286 result.append(adjusted)
1287 1287 return result
1288 1288
1289 1289 def _checkobsrebase(repo, ui, rebaseobsrevs, rebaseobsskipped):
1290 1290 """
1291 1291 Abort if rebase will create divergence or rebase is noop because of markers
1292 1292
1293 1293 `rebaseobsrevs`: set of obsolete revision in source
1294 1294 `rebaseobsskipped`: set of revisions from source skipped because they have
1295 1295 successors in destination or no non-obsolete successor.
1296 1296 """
1297 1297 # Obsolete node with successors not in dest leads to divergence
1298 1298 divergenceok = ui.configbool('experimental',
1299 1299 'evolution.allowdivergence')
1300 1300 divergencebasecandidates = rebaseobsrevs - rebaseobsskipped
1301 1301
1302 1302 if divergencebasecandidates and not divergenceok:
1303 1303 divhashes = (bytes(repo[r])
1304 1304 for r in divergencebasecandidates)
1305 1305 msg = _("this rebase will cause "
1306 1306 "divergences from: %s")
1307 1307 h = _("to force the rebase please set "
1308 1308 "experimental.evolution.allowdivergence=True")
1309 1309 raise error.Abort(msg % (",".join(divhashes),), hint=h)
1310 1310
1311 1311 def successorrevs(unfi, rev):
1312 1312 """yield revision numbers for successors of rev"""
1313 1313 assert unfi.filtername is None
1314 1314 nodemap = unfi.changelog.nodemap
1315 1315 for s in obsutil.allsuccessors(unfi.obsstore, [unfi[rev].node()]):
1316 1316 if s in nodemap:
1317 1317 yield nodemap[s]
1318 1318
1319 1319 def defineparents(repo, rev, destmap, state, skipped, obsskipped):
1320 1320 """Return new parents and optionally a merge base for rev being rebased
1321 1321
1322 1322 The destination specified by "dest" cannot always be used directly because
1323 1323 previously rebase result could affect destination. For example,
1324 1324
1325 1325 D E rebase -r C+D+E -d B
1326 1326 |/ C will be rebased to C'
1327 1327 B C D's new destination will be C' instead of B
1328 1328 |/ E's new destination will be C' instead of B
1329 1329 A
1330 1330
1331 1331 The new parents of a merge is slightly more complicated. See the comment
1332 1332 block below.
1333 1333 """
1334 1334 # use unfiltered changelog since successorrevs may return filtered nodes
1335 1335 assert repo.filtername is None
1336 1336 cl = repo.changelog
1337 1337 isancestor = cl.isancestorrev
1338 1338
1339 1339 dest = destmap[rev]
1340 1340 oldps = repo.changelog.parentrevs(rev) # old parents
1341 1341 newps = [nullrev, nullrev] # new parents
1342 1342 dests = adjustdest(repo, rev, destmap, state, skipped)
1343 1343 bases = list(oldps) # merge base candidates, initially just old parents
1344 1344
1345 1345 if all(r == nullrev for r in oldps[1:]):
1346 1346 # For non-merge changeset, just move p to adjusted dest as requested.
1347 1347 newps[0] = dests[0]
1348 1348 else:
1349 1349 # For merge changeset, if we move p to dests[i] unconditionally, both
1350 1350 # parents may change and the end result looks like "the merge loses a
1351 1351 # parent", which is a surprise. This is a limit because "--dest" only
1352 1352 # accepts one dest per src.
1353 1353 #
1354 1354 # Therefore, only move p with reasonable conditions (in this order):
1355 1355 # 1. use dest, if dest is a descendent of (p or one of p's successors)
1356 1356 # 2. use p's rebased result, if p is rebased (state[p] > 0)
1357 1357 #
1358 1358 # Comparing with adjustdest, the logic here does some additional work:
1359 1359 # 1. decide which parents will not be moved towards dest
1360 1360 # 2. if the above decision is "no", should a parent still be moved
1361 1361 # because it was rebased?
1362 1362 #
1363 1363 # For example:
1364 1364 #
1365 1365 # C # "rebase -r C -d D" is an error since none of the parents
1366 1366 # /| # can be moved. "rebase -r B+C -d D" will move C's parent
1367 1367 # A B D # B (using rule "2."), since B will be rebased.
1368 1368 #
1369 1369 # The loop tries to be not rely on the fact that a Mercurial node has
1370 1370 # at most 2 parents.
1371 1371 for i, p in enumerate(oldps):
1372 1372 np = p # new parent
1373 1373 if any(isancestor(x, dests[i]) for x in successorrevs(repo, p)):
1374 1374 np = dests[i]
1375 1375 elif p in state and state[p] > 0:
1376 1376 np = state[p]
1377 1377
1378 1378 # "bases" only record "special" merge bases that cannot be
1379 1379 # calculated from changelog DAG (i.e. isancestor(p, np) is False).
1380 1380 # For example:
1381 1381 #
1382 1382 # B' # rebase -s B -d D, when B was rebased to B'. dest for C
1383 1383 # | C # is B', but merge base for C is B, instead of
1384 1384 # D | # changelog.ancestor(C, B') == A. If changelog DAG and
1385 1385 # | B # "state" edges are merged (so there will be an edge from
1386 1386 # |/ # B to B'), the merge base is still ancestor(C, B') in
1387 1387 # A # the merged graph.
1388 1388 #
1389 1389 # Also see https://bz.mercurial-scm.org/show_bug.cgi?id=1950#c8
1390 1390 # which uses "virtual null merge" to explain this situation.
1391 1391 if isancestor(p, np):
1392 1392 bases[i] = nullrev
1393 1393
1394 1394 # If one parent becomes an ancestor of the other, drop the ancestor
1395 1395 for j, x in enumerate(newps[:i]):
1396 1396 if x == nullrev:
1397 1397 continue
1398 1398 if isancestor(np, x): # CASE-1
1399 1399 np = nullrev
1400 1400 elif isancestor(x, np): # CASE-2
1401 1401 newps[j] = np
1402 1402 np = nullrev
1403 1403 # New parents forming an ancestor relationship does not
1404 1404 # mean the old parents have a similar relationship. Do not
1405 1405 # set bases[x] to nullrev.
1406 1406 bases[j], bases[i] = bases[i], bases[j]
1407 1407
1408 1408 newps[i] = np
1409 1409
1410 1410 # "rebasenode" updates to new p1, and the old p1 will be used as merge
1411 1411 # base. If only p2 changes, merging using unchanged p1 as merge base is
1412 1412 # suboptimal. Therefore swap parents to make the merge sane.
1413 1413 if newps[1] != nullrev and oldps[0] == newps[0]:
1414 1414 assert len(newps) == 2 and len(oldps) == 2
1415 1415 newps.reverse()
1416 1416 bases.reverse()
1417 1417
1418 1418 # No parent change might be an error because we fail to make rev a
1419 1419 # descendent of requested dest. This can happen, for example:
1420 1420 #
1421 1421 # C # rebase -r C -d D
1422 1422 # /| # None of A and B will be changed to D and rebase fails.
1423 1423 # A B D
1424 1424 if set(newps) == set(oldps) and dest not in newps:
1425 1425 raise error.Abort(_('cannot rebase %d:%s without '
1426 1426 'moving at least one of its parents')
1427 1427 % (rev, repo[rev]))
1428 1428
1429 1429 # Source should not be ancestor of dest. The check here guarantees it's
1430 1430 # impossible. With multi-dest, the initial check does not cover complex
1431 1431 # cases since we don't have abstractions to dry-run rebase cheaply.
1432 1432 if any(p != nullrev and isancestor(rev, p) for p in newps):
1433 1433 raise error.Abort(_('source is ancestor of destination'))
1434 1434
1435 1435 # "rebasenode" updates to new p1, use the corresponding merge base.
1436 1436 if bases[0] != nullrev:
1437 1437 base = bases[0]
1438 1438 else:
1439 1439 base = None
1440 1440
1441 1441 # Check if the merge will contain unwanted changes. That may happen if
1442 1442 # there are multiple special (non-changelog ancestor) merge bases, which
1443 1443 # cannot be handled well by the 3-way merge algorithm. For example:
1444 1444 #
1445 1445 # F
1446 1446 # /|
1447 1447 # D E # "rebase -r D+E+F -d Z", when rebasing F, if "D" was chosen
1448 1448 # | | # as merge base, the difference between D and F will include
1449 1449 # B C # C, so the rebased F will contain C surprisingly. If "E" was
1450 1450 # |/ # chosen, the rebased F will contain B.
1451 1451 # A Z
1452 1452 #
1453 1453 # But our merge base candidates (D and E in above case) could still be
1454 1454 # better than the default (ancestor(F, Z) == null). Therefore still
1455 1455 # pick one (so choose p1 above).
1456 1456 if sum(1 for b in bases if b != nullrev) > 1:
1457 1457 unwanted = [None, None] # unwanted[i]: unwanted revs if choose bases[i]
1458 1458 for i, base in enumerate(bases):
1459 1459 if base == nullrev:
1460 1460 continue
1461 1461 # Revisions in the side (not chosen as merge base) branch that
1462 1462 # might contain "surprising" contents
1463 1463 siderevs = list(repo.revs('((%ld-%d) %% (%d+%d))',
1464 1464 bases, base, base, dest))
1465 1465
1466 1466 # If those revisions are covered by rebaseset, the result is good.
1467 1467 # A merge in rebaseset would be considered to cover its ancestors.
1468 1468 if siderevs:
1469 1469 rebaseset = [r for r, d in state.items()
1470 1470 if d > 0 and r not in obsskipped]
1471 1471 merges = [r for r in rebaseset
1472 1472 if cl.parentrevs(r)[1] != nullrev]
1473 1473 unwanted[i] = list(repo.revs('%ld - (::%ld) - %ld',
1474 1474 siderevs, merges, rebaseset))
1475 1475
1476 1476 # Choose a merge base that has a minimal number of unwanted revs.
1477 1477 l, i = min((len(revs), i)
1478 1478 for i, revs in enumerate(unwanted) if revs is not None)
1479 1479 base = bases[i]
1480 1480
1481 1481 # newps[0] should match merge base if possible. Currently, if newps[i]
1482 1482 # is nullrev, the only case is newps[i] and newps[j] (j < i), one is
1483 1483 # the other's ancestor. In that case, it's fine to not swap newps here.
1484 1484 # (see CASE-1 and CASE-2 above)
1485 1485 if i != 0 and newps[i] != nullrev:
1486 1486 newps[0], newps[i] = newps[i], newps[0]
1487 1487
1488 1488 # The merge will include unwanted revisions. Abort now. Revisit this if
1489 1489 # we have a more advanced merge algorithm that handles multiple bases.
1490 1490 if l > 0:
1491 1491 unwanteddesc = _(' or ').join(
1492 1492 (', '.join('%d:%s' % (r, repo[r]) for r in revs)
1493 1493 for revs in unwanted if revs is not None))
1494 1494 raise error.Abort(
1495 1495 _('rebasing %d:%s will include unwanted changes from %s')
1496 1496 % (rev, repo[rev], unwanteddesc))
1497 1497
1498 1498 repo.ui.debug(" future parents are %d and %d\n" % tuple(newps))
1499 1499
1500 1500 return newps[0], newps[1], base
1501 1501
1502 1502 def isagitpatch(repo, patchname):
1503 1503 'Return true if the given patch is in git format'
1504 1504 mqpatch = os.path.join(repo.mq.path, patchname)
1505 1505 for line in patch.linereader(open(mqpatch, 'rb')):
1506 1506 if line.startswith('diff --git'):
1507 1507 return True
1508 1508 return False
1509 1509
1510 1510 def updatemq(repo, state, skipped, **opts):
1511 1511 'Update rebased mq patches - finalize and then import them'
1512 1512 mqrebase = {}
1513 1513 mq = repo.mq
1514 1514 original_series = mq.fullseries[:]
1515 1515 skippedpatches = set()
1516 1516
1517 1517 for p in mq.applied:
1518 1518 rev = repo[p.node].rev()
1519 1519 if rev in state:
1520 1520 repo.ui.debug('revision %d is an mq patch (%s), finalize it.\n' %
1521 1521 (rev, p.name))
1522 1522 mqrebase[rev] = (p.name, isagitpatch(repo, p.name))
1523 1523 else:
1524 1524 # Applied but not rebased, not sure this should happen
1525 1525 skippedpatches.add(p.name)
1526 1526
1527 1527 if mqrebase:
1528 1528 mq.finish(repo, mqrebase.keys())
1529 1529
1530 1530 # We must start import from the newest revision
1531 1531 for rev in sorted(mqrebase, reverse=True):
1532 1532 if rev not in skipped:
1533 1533 name, isgit = mqrebase[rev]
1534 1534 repo.ui.note(_('updating mq patch %s to %d:%s\n') %
1535 1535 (name, state[rev], repo[state[rev]]))
1536 1536 mq.qimport(repo, (), patchname=name, git=isgit,
1537 1537 rev=["%d" % state[rev]])
1538 1538 else:
1539 1539 # Rebased and skipped
1540 1540 skippedpatches.add(mqrebase[rev][0])
1541 1541
1542 1542 # Patches were either applied and rebased and imported in
1543 1543 # order, applied and removed or unapplied. Discard the removed
1544 1544 # ones while preserving the original series order and guards.
1545 1545 newseries = [s for s in original_series
1546 1546 if mq.guard_re.split(s, 1)[0] not in skippedpatches]
1547 1547 mq.fullseries[:] = newseries
1548 1548 mq.seriesdirty = True
1549 1549 mq.savedirty()
1550 1550
1551 1551 def storecollapsemsg(repo, collapsemsg):
1552 1552 'Store the collapse message to allow recovery'
1553 1553 collapsemsg = collapsemsg or ''
1554 1554 f = repo.vfs("last-message.txt", "w")
1555 1555 f.write("%s\n" % collapsemsg)
1556 1556 f.close()
1557 1557
1558 1558 def clearcollapsemsg(repo):
1559 1559 'Remove collapse message file'
1560 1560 repo.vfs.unlinkpath("last-message.txt", ignoremissing=True)
1561 1561
1562 1562 def restorecollapsemsg(repo, isabort):
1563 1563 'Restore previously stored collapse message'
1564 1564 try:
1565 1565 f = repo.vfs("last-message.txt")
1566 1566 collapsemsg = f.readline().strip()
1567 1567 f.close()
1568 1568 except IOError as err:
1569 1569 if err.errno != errno.ENOENT:
1570 1570 raise
1571 1571 if isabort:
1572 1572 # Oh well, just abort like normal
1573 1573 collapsemsg = ''
1574 1574 else:
1575 1575 raise error.Abort(_('missing .hg/last-message.txt for rebase'))
1576 1576 return collapsemsg
1577 1577
1578 1578 def clearstatus(repo):
1579 1579 'Remove the status files'
1580 1580 # Make sure the active transaction won't write the state file
1581 1581 tr = repo.currenttransaction()
1582 1582 if tr:
1583 1583 tr.removefilegenerator('rebasestate')
1584 1584 repo.vfs.unlinkpath("rebasestate", ignoremissing=True)
1585 1585
1586 1586 def needupdate(repo, state):
1587 1587 '''check whether we should `update --clean` away from a merge, or if
1588 1588 somehow the working dir got forcibly updated, e.g. by older hg'''
1589 1589 parents = [p.rev() for p in repo[None].parents()]
1590 1590
1591 1591 # Are we in a merge state at all?
1592 1592 if len(parents) < 2:
1593 1593 return False
1594 1594
1595 1595 # We should be standing on the first as-of-yet unrebased commit.
1596 1596 firstunrebased = min([old for old, new in state.iteritems()
1597 1597 if new == nullrev])
1598 1598 if firstunrebased in parents:
1599 1599 return True
1600 1600
1601 1601 return False
1602 1602
1603 1603 def abort(repo, originalwd, destmap, state, activebookmark=None, backup=True,
1604 1604 suppwarns=False):
1605 1605 '''Restore the repository to its original state. Additional args:
1606 1606
1607 1607 activebookmark: the name of the bookmark that should be active after the
1608 1608 restore'''
1609 1609
1610 1610 try:
1611 1611 # If the first commits in the rebased set get skipped during the rebase,
1612 1612 # their values within the state mapping will be the dest rev id. The
1613 1613 # rebased list must must not contain the dest rev (issue4896)
1614 1614 rebased = [s for r, s in state.items()
1615 1615 if s >= 0 and s != r and s != destmap[r]]
1616 1616 immutable = [d for d in rebased if not repo[d].mutable()]
1617 1617 cleanup = True
1618 1618 if immutable:
1619 1619 repo.ui.warn(_("warning: can't clean up public changesets %s\n")
1620 1620 % ', '.join(bytes(repo[r]) for r in immutable),
1621 1621 hint=_("see 'hg help phases' for details"))
1622 1622 cleanup = False
1623 1623
1624 1624 descendants = set()
1625 1625 if rebased:
1626 1626 descendants = set(repo.changelog.descendants(rebased))
1627 1627 if descendants - set(rebased):
1628 1628 repo.ui.warn(_("warning: new changesets detected on destination "
1629 1629 "branch, can't strip\n"))
1630 1630 cleanup = False
1631 1631
1632 1632 if cleanup:
1633 1633 shouldupdate = False
1634 1634 if rebased:
1635 1635 strippoints = [
1636 1636 c.node() for c in repo.set('roots(%ld)', rebased)]
1637 1637
1638 1638 updateifonnodes = set(rebased)
1639 1639 updateifonnodes.update(destmap.values())
1640 1640 updateifonnodes.add(originalwd)
1641 1641 shouldupdate = repo['.'].rev() in updateifonnodes
1642 1642
1643 1643 # Update away from the rebase if necessary
1644 1644 if shouldupdate or needupdate(repo, state):
1645 1645 mergemod.update(repo, originalwd, False, True)
1646 1646
1647 1647 # Strip from the first rebased revision
1648 1648 if rebased:
1649 1649 repair.strip(repo.ui, repo, strippoints, backup=backup)
1650 1650
1651 1651 if activebookmark and activebookmark in repo._bookmarks:
1652 1652 bookmarks.activate(repo, activebookmark)
1653 1653
1654 1654 finally:
1655 1655 clearstatus(repo)
1656 1656 clearcollapsemsg(repo)
1657 1657 if not suppwarns:
1658 1658 repo.ui.warn(_('rebase aborted\n'))
1659 1659 return 0
1660 1660
1661 1661 def sortsource(destmap):
1662 1662 """yield source revisions in an order that we only rebase things once
1663 1663
1664 1664 If source and destination overlaps, we should filter out revisions
1665 1665 depending on other revisions which hasn't been rebased yet.
1666 1666
1667 1667 Yield a sorted list of revisions each time.
1668 1668
1669 1669 For example, when rebasing A to B, B to C. This function yields [B], then
1670 1670 [A], indicating B needs to be rebased first.
1671 1671
1672 1672 Raise if there is a cycle so the rebase is impossible.
1673 1673 """
1674 1674 srcset = set(destmap)
1675 1675 while srcset:
1676 1676 srclist = sorted(srcset)
1677 1677 result = []
1678 1678 for r in srclist:
1679 1679 if destmap[r] not in srcset:
1680 1680 result.append(r)
1681 1681 if not result:
1682 1682 raise error.Abort(_('source and destination form a cycle'))
1683 1683 srcset -= set(result)
1684 1684 yield result
1685 1685
1686 1686 def buildstate(repo, destmap, collapse):
1687 1687 '''Define which revisions are going to be rebased and where
1688 1688
1689 1689 repo: repo
1690 1690 destmap: {srcrev: destrev}
1691 1691 '''
1692 1692 rebaseset = destmap.keys()
1693 1693 originalwd = repo['.'].rev()
1694 1694
1695 1695 # This check isn't strictly necessary, since mq detects commits over an
1696 1696 # applied patch. But it prevents messing up the working directory when
1697 1697 # a partially completed rebase is blocked by mq.
1698 1698 if 'qtip' in repo.tags():
1699 1699 mqapplied = set(repo[s.node].rev() for s in repo.mq.applied)
1700 1700 if set(destmap.values()) & mqapplied:
1701 1701 raise error.Abort(_('cannot rebase onto an applied mq patch'))
1702 1702
1703 1703 # Get "cycle" error early by exhausting the generator.
1704 1704 sortedsrc = list(sortsource(destmap)) # a list of sorted revs
1705 1705 if not sortedsrc:
1706 1706 raise error.Abort(_('no matching revisions'))
1707 1707
1708 1708 # Only check the first batch of revisions to rebase not depending on other
1709 1709 # rebaseset. This means "source is ancestor of destination" for the second
1710 1710 # (and following) batches of revisions are not checked here. We rely on
1711 1711 # "defineparents" to do that check.
1712 1712 roots = list(repo.set('roots(%ld)', sortedsrc[0]))
1713 1713 if not roots:
1714 1714 raise error.Abort(_('no matching revisions'))
1715 1715 def revof(r):
1716 1716 return r.rev()
1717 1717 roots = sorted(roots, key=revof)
1718 1718 state = dict.fromkeys(rebaseset, revtodo)
1719 1719 emptyrebase = (len(sortedsrc) == 1)
1720 1720 for root in roots:
1721 1721 dest = repo[destmap[root.rev()]]
1722 1722 commonbase = root.ancestor(dest)
1723 1723 if commonbase == root:
1724 1724 raise error.Abort(_('source is ancestor of destination'))
1725 1725 if commonbase == dest:
1726 1726 wctx = repo[None]
1727 1727 if dest == wctx.p1():
1728 1728 # when rebasing to '.', it will use the current wd branch name
1729 1729 samebranch = root.branch() == wctx.branch()
1730 1730 else:
1731 1731 samebranch = root.branch() == dest.branch()
1732 1732 if not collapse and samebranch and dest in root.parents():
1733 1733 # mark the revision as done by setting its new revision
1734 1734 # equal to its old (current) revisions
1735 1735 state[root.rev()] = root.rev()
1736 1736 repo.ui.debug('source is a child of destination\n')
1737 1737 continue
1738 1738
1739 1739 emptyrebase = False
1740 1740 repo.ui.debug('rebase onto %s starting from %s\n' % (dest, root))
1741 1741 if emptyrebase:
1742 1742 return None
1743 1743 for rev in sorted(state):
1744 1744 parents = [p for p in repo.changelog.parentrevs(rev) if p != nullrev]
1745 1745 # if all parents of this revision are done, then so is this revision
1746 1746 if parents and all((state.get(p) == p for p in parents)):
1747 1747 state[rev] = rev
1748 1748 return originalwd, destmap, state
1749 1749
1750 1750 def clearrebased(ui, repo, destmap, state, skipped, collapsedas=None,
1751 1751 keepf=False, fm=None, backup=True):
1752 1752 """dispose of rebased revision at the end of the rebase
1753 1753
1754 1754 If `collapsedas` is not None, the rebase was a collapse whose result if the
1755 1755 `collapsedas` node.
1756 1756
1757 1757 If `keepf` is not True, the rebase has --keep set and no nodes should be
1758 1758 removed (but bookmarks still need to be moved).
1759 1759
1760 1760 If `backup` is False, no backup will be stored when stripping rebased
1761 1761 revisions.
1762 1762 """
1763 1763 tonode = repo.changelog.node
1764 1764 replacements = {}
1765 1765 moves = {}
1766 stripcleanup = not obsolete.isenabled(repo, obsolete.createmarkersopt)
1766 1767 for rev, newrev in sorted(state.items()):
1767 1768 if newrev >= 0 and newrev != rev:
1768 1769 oldnode = tonode(rev)
1769 1770 newnode = collapsedas or tonode(newrev)
1770 1771 moves[oldnode] = newnode
1771 1772 if not keepf:
1773 succs = None
1772 1774 if rev in skipped:
1773 succs = ()
1775 if stripcleanup or not repo[rev].obsolete():
1776 succs = ()
1774 1777 else:
1775 1778 succs = (newnode,)
1776 replacements[oldnode] = succs
1779 if succs is not None:
1780 replacements[oldnode] = succs
1777 1781 scmutil.cleanupnodes(repo, replacements, 'rebase', moves, backup=backup)
1778 1782 if fm:
1779 1783 hf = fm.hexfunc
1780 1784 fl = fm.formatlist
1781 1785 fd = fm.formatdict
1782 1786 nodechanges = fd({hf(oldn): fl([hf(n) for n in newn], name='node')
1783 1787 for oldn, newn in replacements.iteritems()},
1784 1788 key="oldnode", value="newnodes")
1785 1789 fm.data(nodechanges=nodechanges)
1786 1790
1787 1791 def pullrebase(orig, ui, repo, *args, **opts):
1788 1792 'Call rebase after pull if the latter has been invoked with --rebase'
1789 1793 ret = None
1790 1794 if opts.get(r'rebase'):
1791 1795 if ui.configbool('commands', 'rebase.requiredest'):
1792 1796 msg = _('rebase destination required by configuration')
1793 1797 hint = _('use hg pull followed by hg rebase -d DEST')
1794 1798 raise error.Abort(msg, hint=hint)
1795 1799
1796 1800 with repo.wlock(), repo.lock():
1797 1801 if opts.get(r'update'):
1798 1802 del opts[r'update']
1799 1803 ui.debug('--update and --rebase are not compatible, ignoring '
1800 1804 'the update flag\n')
1801 1805
1802 1806 cmdutil.checkunfinished(repo)
1803 1807 cmdutil.bailifchanged(repo, hint=_('cannot pull with rebase: '
1804 1808 'please commit or shelve your changes first'))
1805 1809
1806 1810 revsprepull = len(repo)
1807 1811 origpostincoming = commands.postincoming
1808 1812 def _dummy(*args, **kwargs):
1809 1813 pass
1810 1814 commands.postincoming = _dummy
1811 1815 try:
1812 1816 ret = orig(ui, repo, *args, **opts)
1813 1817 finally:
1814 1818 commands.postincoming = origpostincoming
1815 1819 revspostpull = len(repo)
1816 1820 if revspostpull > revsprepull:
1817 1821 # --rev option from pull conflict with rebase own --rev
1818 1822 # dropping it
1819 1823 if r'rev' in opts:
1820 1824 del opts[r'rev']
1821 1825 # positional argument from pull conflicts with rebase's own
1822 1826 # --source.
1823 1827 if r'source' in opts:
1824 1828 del opts[r'source']
1825 1829 # revsprepull is the len of the repo, not revnum of tip.
1826 1830 destspace = list(repo.changelog.revs(start=revsprepull))
1827 1831 opts[r'_destspace'] = destspace
1828 1832 try:
1829 1833 rebase(ui, repo, **opts)
1830 1834 except error.NoMergeDestAbort:
1831 1835 # we can maybe update instead
1832 1836 rev, _a, _b = destutil.destupdate(repo)
1833 1837 if rev == repo['.'].rev():
1834 1838 ui.status(_('nothing to rebase\n'))
1835 1839 else:
1836 1840 ui.status(_('nothing to rebase - updating instead\n'))
1837 1841 # not passing argument to get the bare update behavior
1838 1842 # with warning and trumpets
1839 1843 commands.update(ui, repo)
1840 1844 else:
1841 1845 if opts.get(r'tool'):
1842 1846 raise error.Abort(_('--tool can only be used with --rebase'))
1843 1847 ret = orig(ui, repo, *args, **opts)
1844 1848
1845 1849 return ret
1846 1850
1847 1851 def _filterobsoleterevs(repo, revs):
1848 1852 """returns a set of the obsolete revisions in revs"""
1849 1853 return set(r for r in revs if repo[r].obsolete())
1850 1854
1851 1855 def _computeobsoletenotrebased(repo, rebaseobsrevs, destmap):
1852 1856 """Return (obsoletenotrebased, obsoletewithoutsuccessorindestination).
1853 1857
1854 1858 `obsoletenotrebased` is a mapping mapping obsolete => successor for all
1855 1859 obsolete nodes to be rebased given in `rebaseobsrevs`.
1856 1860
1857 1861 `obsoletewithoutsuccessorindestination` is a set with obsolete revisions
1858 1862 without a successor in destination.
1859 1863
1860 1864 `obsoleteextinctsuccessors` is a set of obsolete revisions with only
1861 1865 obsolete successors.
1862 1866 """
1863 1867 obsoletenotrebased = {}
1864 1868 obsoletewithoutsuccessorindestination = set([])
1865 1869 obsoleteextinctsuccessors = set([])
1866 1870
1867 1871 assert repo.filtername is None
1868 1872 cl = repo.changelog
1869 1873 nodemap = cl.nodemap
1870 1874 extinctrevs = set(repo.revs('extinct()'))
1871 1875 for srcrev in rebaseobsrevs:
1872 1876 srcnode = cl.node(srcrev)
1873 1877 # XXX: more advanced APIs are required to handle split correctly
1874 1878 successors = set(obsutil.allsuccessors(repo.obsstore, [srcnode]))
1875 1879 # obsutil.allsuccessors includes node itself
1876 1880 successors.remove(srcnode)
1877 1881 succrevs = {nodemap[s] for s in successors if s in nodemap}
1878 1882 if succrevs.issubset(extinctrevs):
1879 1883 # all successors are extinct
1880 1884 obsoleteextinctsuccessors.add(srcrev)
1881 1885 if not successors:
1882 1886 # no successor
1883 1887 obsoletenotrebased[srcrev] = None
1884 1888 else:
1885 1889 dstrev = destmap[srcrev]
1886 1890 for succrev in succrevs:
1887 1891 if cl.isancestorrev(succrev, dstrev):
1888 1892 obsoletenotrebased[srcrev] = succrev
1889 1893 break
1890 1894 else:
1891 1895 # If 'srcrev' has a successor in rebase set but none in
1892 1896 # destination (which would be catched above), we shall skip it
1893 1897 # and its descendants to avoid divergence.
1894 1898 if srcrev in extinctrevs or any(s in destmap for s in succrevs):
1895 1899 obsoletewithoutsuccessorindestination.add(srcrev)
1896 1900
1897 1901 return (
1898 1902 obsoletenotrebased,
1899 1903 obsoletewithoutsuccessorindestination,
1900 1904 obsoleteextinctsuccessors,
1901 1905 )
1902 1906
1903 1907 def summaryhook(ui, repo):
1904 1908 if not repo.vfs.exists('rebasestate'):
1905 1909 return
1906 1910 try:
1907 1911 rbsrt = rebaseruntime(repo, ui, {})
1908 1912 rbsrt.restorestatus()
1909 1913 state = rbsrt.state
1910 1914 except error.RepoLookupError:
1911 1915 # i18n: column positioning for "hg summary"
1912 1916 msg = _('rebase: (use "hg rebase --abort" to clear broken state)\n')
1913 1917 ui.write(msg)
1914 1918 return
1915 1919 numrebased = len([i for i in state.itervalues() if i >= 0])
1916 1920 # i18n: column positioning for "hg summary"
1917 1921 ui.write(_('rebase: %s, %s (rebase --continue)\n') %
1918 1922 (ui.label(_('%d rebased'), 'rebase.rebased') % numrebased,
1919 1923 ui.label(_('%d remaining'), 'rebase.remaining') %
1920 1924 (len(state) - numrebased)))
1921 1925
1922 1926 def uisetup(ui):
1923 1927 #Replace pull with a decorator to provide --rebase option
1924 1928 entry = extensions.wrapcommand(commands.table, 'pull', pullrebase)
1925 1929 entry[1].append(('', 'rebase', None,
1926 1930 _("rebase working directory to branch head")))
1927 1931 entry[1].append(('t', 'tool', '',
1928 1932 _("specify merge tool for rebase")))
1929 1933 cmdutil.summaryhooks.add('rebase', summaryhook)
1930 1934 cmdutil.unfinishedstates.append(
1931 1935 ['rebasestate', False, False, _('rebase in progress'),
1932 1936 _("use 'hg rebase --continue' or 'hg rebase --abort'")])
1933 1937 cmdutil.afterresolvedstates.append(
1934 1938 ['rebasestate', _('hg rebase --continue')])
General Comments 0
You need to be logged in to leave comments. Login now