##// END OF EJS Templates
rebase: add missing dashes in help text
Manuel Jacob -
r41972:25fc5b96 stable
parent child Browse files
Show More
@@ -1,1952 +1,1952 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('rewrite', 'backup-bundle')
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 self._abort(backup=backup, suppwarns=suppwarns)
351 351
352 352 def _preparenewrebase(self, destmap):
353 353 if not destmap:
354 354 return _nothingtorebase()
355 355
356 356 rebaseset = destmap.keys()
357 357 allowunstable = obsolete.isenabled(self.repo, obsolete.allowunstableopt)
358 358 if (not (self.keepf or allowunstable)
359 359 and self.repo.revs('first(children(%ld) - %ld)',
360 360 rebaseset, rebaseset)):
361 361 raise error.Abort(
362 362 _("can't remove original changesets with"
363 363 " unrebased descendants"),
364 364 hint=_('use --keep to keep original changesets'))
365 365
366 366 result = buildstate(self.repo, destmap, self.collapsef)
367 367
368 368 if not result:
369 369 # Empty state built, nothing to rebase
370 370 self.ui.status(_('nothing to rebase\n'))
371 371 return _nothingtorebase()
372 372
373 373 for root in self.repo.set('roots(%ld)', rebaseset):
374 374 if not self.keepf and not root.mutable():
375 375 raise error.Abort(_("can't rebase public changeset %s")
376 376 % root,
377 377 hint=_("see 'hg help phases' for details"))
378 378
379 379 (self.originalwd, self.destmap, self.state) = result
380 380 if self.collapsef:
381 381 dests = set(self.destmap.values())
382 382 if len(dests) != 1:
383 383 raise error.Abort(
384 384 _('--collapse does not work with multiple destinations'))
385 385 destrev = next(iter(dests))
386 386 destancestors = self.repo.changelog.ancestors([destrev],
387 387 inclusive=True)
388 388 self.external = externalparent(self.repo, self.state, destancestors)
389 389
390 390 for destrev in sorted(set(destmap.values())):
391 391 dest = self.repo[destrev]
392 392 if dest.closesbranch() and not self.keepbranchesf:
393 393 self.ui.status(_('reopening closed branch head %s\n') % dest)
394 394
395 395 self.prepared = True
396 396
397 397 def _assignworkingcopy(self):
398 398 if self.inmemory:
399 399 from mercurial.context import overlayworkingctx
400 400 self.wctx = overlayworkingctx(self.repo)
401 401 self.repo.ui.debug("rebasing in-memory\n")
402 402 else:
403 403 self.wctx = self.repo[None]
404 404 self.repo.ui.debug("rebasing on disk\n")
405 405 self.repo.ui.log("rebase",
406 406 "using in-memory rebase: %r\n", self.inmemory,
407 407 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 not self.inmemory and 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: not rebasing %s, its destination already '
577 577 'has all its changes\n') % desc)
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 def _abort(self, backup=True, suppwarns=False):
655 655 '''Restore the repository to its original state.'''
656 656
657 657 repo = self.repo
658 658 try:
659 659 # If the first commits in the rebased set get skipped during the
660 660 # rebase, their values within the state mapping will be the dest
661 661 # rev id. The rebased list must must not contain the dest rev
662 662 # (issue4896)
663 663 rebased = [s for r, s in self.state.items()
664 664 if s >= 0 and s != r and s != self.destmap[r]]
665 665 immutable = [d for d in rebased if not repo[d].mutable()]
666 666 cleanup = True
667 667 if immutable:
668 668 repo.ui.warn(_("warning: can't clean up public changesets %s\n")
669 669 % ', '.join(bytes(repo[r]) for r in immutable),
670 670 hint=_("see 'hg help phases' for details"))
671 671 cleanup = False
672 672
673 673 descendants = set()
674 674 if rebased:
675 675 descendants = set(repo.changelog.descendants(rebased))
676 676 if descendants - set(rebased):
677 677 repo.ui.warn(_("warning: new changesets detected on "
678 678 "destination branch, can't strip\n"))
679 679 cleanup = False
680 680
681 681 if cleanup:
682 682 shouldupdate = False
683 683 if rebased:
684 684 strippoints = [
685 685 c.node() for c in repo.set('roots(%ld)', rebased)]
686 686
687 687 updateifonnodes = set(rebased)
688 688 updateifonnodes.update(self.destmap.values())
689 689 updateifonnodes.add(self.originalwd)
690 690 shouldupdate = repo['.'].rev() in updateifonnodes
691 691
692 692 # Update away from the rebase if necessary
693 693 if shouldupdate or needupdate(repo, self.state):
694 694 mergemod.update(repo, self.originalwd, branchmerge=False,
695 695 force=True)
696 696
697 697 # Strip from the first rebased revision
698 698 if rebased:
699 699 repair.strip(repo.ui, repo, strippoints, backup=backup)
700 700
701 701 if self.activebookmark and self.activebookmark in repo._bookmarks:
702 702 bookmarks.activate(repo, self.activebookmark)
703 703
704 704 finally:
705 705 clearstatus(repo)
706 706 clearcollapsemsg(repo)
707 707 if not suppwarns:
708 708 repo.ui.warn(_('rebase aborted\n'))
709 709 return 0
710 710
711 711 @command('rebase',
712 712 [('s', 'source', '',
713 713 _('rebase the specified changeset and descendants'), _('REV')),
714 714 ('b', 'base', '',
715 715 _('rebase everything from branching point of specified changeset'),
716 716 _('REV')),
717 717 ('r', 'rev', [],
718 718 _('rebase these revisions'),
719 719 _('REV')),
720 720 ('d', 'dest', '',
721 721 _('rebase onto the specified changeset'), _('REV')),
722 722 ('', 'collapse', False, _('collapse the rebased changesets')),
723 723 ('m', 'message', '',
724 724 _('use text as collapse commit message'), _('TEXT')),
725 725 ('e', 'edit', False, _('invoke editor on commit messages')),
726 726 ('l', 'logfile', '',
727 727 _('read collapse commit message from file'), _('FILE')),
728 728 ('k', 'keep', False, _('keep original changesets')),
729 729 ('', 'keepbranches', False, _('keep original branch names')),
730 730 ('D', 'detach', False, _('(DEPRECATED)')),
731 731 ('i', 'interactive', False, _('(DEPRECATED)')),
732 732 ('t', 'tool', '', _('specify merge tool')),
733 733 ('', 'stop', False, _('stop interrupted rebase')),
734 734 ('c', 'continue', False, _('continue an interrupted rebase')),
735 735 ('a', 'abort', False, _('abort an interrupted rebase')),
736 736 ('', 'auto-orphans', '', _('automatically rebase orphan revisions '
737 737 'in the specified revset (EXPERIMENTAL)')),
738 738 ] + cmdutil.dryrunopts + cmdutil.formatteropts + cmdutil.confirmopts,
739 739 _('[-s REV | -b REV] [-d REV] [OPTION]'),
740 740 helpcategory=command.CATEGORY_CHANGE_MANAGEMENT)
741 741 def rebase(ui, repo, **opts):
742 742 """move changeset (and descendants) to a different branch
743 743
744 744 Rebase uses repeated merging to graft changesets from one part of
745 745 history (the source) onto another (the destination). This can be
746 746 useful for linearizing *local* changes relative to a master
747 747 development tree.
748 748
749 749 Published commits cannot be rebased (see :hg:`help phases`).
750 750 To copy commits, see :hg:`help graft`.
751 751
752 752 If you don't specify a destination changeset (``-d/--dest``), rebase
753 753 will use the same logic as :hg:`merge` to pick a destination. if
754 754 the current branch contains exactly one other head, the other head
755 755 is merged with by default. Otherwise, an explicit revision with
756 756 which to merge with must be provided. (destination changeset is not
757 757 modified by rebasing, but new changesets are added as its
758 758 descendants.)
759 759
760 760 Here are the ways to select changesets:
761 761
762 762 1. Explicitly select them using ``--rev``.
763 763
764 764 2. Use ``--source`` to select a root changeset and include all of its
765 765 descendants.
766 766
767 767 3. Use ``--base`` to select a changeset; rebase will find ancestors
768 768 and their descendants which are not also ancestors of the destination.
769 769
770 4. If you do not specify any of ``--rev``, ``source``, or ``--base``,
770 4. If you do not specify any of ``--rev``, ``--source``, or ``--base``,
771 771 rebase will use ``--base .`` as above.
772 772
773 773 If ``--source`` or ``--rev`` is used, special names ``SRC`` and ``ALLSRC``
774 774 can be used in ``--dest``. Destination would be calculated per source
775 775 revision with ``SRC`` substituted by that single source revision and
776 776 ``ALLSRC`` substituted by all source revisions.
777 777
778 778 Rebase will destroy original changesets unless you use ``--keep``.
779 779 It will also move your bookmarks (even if you do).
780 780
781 781 Some changesets may be dropped if they do not contribute changes
782 782 (e.g. merges from the destination branch).
783 783
784 784 Unlike ``merge``, rebase will do nothing if you are at the branch tip of
785 785 a named branch with two heads. You will need to explicitly specify source
786 786 and/or destination.
787 787
788 788 If you need to use a tool to automate merge/conflict decisions, you
789 789 can specify one with ``--tool``, see :hg:`help merge-tools`.
790 790 As a caveat: the tool will not be used to mediate when a file was
791 791 deleted, there is no hook presently available for this.
792 792
793 793 If a rebase is interrupted to manually resolve a conflict, it can be
794 794 continued with --continue/-c, aborted with --abort/-a, or stopped with
795 795 --stop.
796 796
797 797 .. container:: verbose
798 798
799 799 Examples:
800 800
801 801 - move "local changes" (current commit back to branching point)
802 802 to the current branch tip after a pull::
803 803
804 804 hg rebase
805 805
806 806 - move a single changeset to the stable branch::
807 807
808 808 hg rebase -r 5f493448 -d stable
809 809
810 810 - splice a commit and all its descendants onto another part of history::
811 811
812 812 hg rebase --source c0c3 --dest 4cf9
813 813
814 814 - rebase everything on a branch marked by a bookmark onto the
815 815 default branch::
816 816
817 817 hg rebase --base myfeature --dest default
818 818
819 819 - collapse a sequence of changes into a single commit::
820 820
821 821 hg rebase --collapse -r 1520:1525 -d .
822 822
823 823 - move a named branch while preserving its name::
824 824
825 825 hg rebase -r "branch(featureX)" -d 1.3 --keepbranches
826 826
827 827 - stabilize orphaned changesets so history looks linear::
828 828
829 829 hg rebase -r 'orphan()-obsolete()'\
830 830 -d 'first(max((successors(max(roots(ALLSRC) & ::SRC)^)-obsolete())::) +\
831 831 max(::((roots(ALLSRC) & ::SRC)^)-obsolete()))'
832 832
833 833 Configuration Options:
834 834
835 835 You can make rebase require a destination if you set the following config
836 836 option::
837 837
838 838 [commands]
839 839 rebase.requiredest = True
840 840
841 841 By default, rebase will close the transaction after each commit. For
842 842 performance purposes, you can configure rebase to use a single transaction
843 843 across the entire rebase. WARNING: This setting introduces a significant
844 844 risk of losing the work you've done in a rebase if the rebase aborts
845 845 unexpectedly::
846 846
847 847 [rebase]
848 848 singletransaction = True
849 849
850 850 By default, rebase writes to the working copy, but you can configure it to
851 851 run in-memory for for better performance, and to allow it to run if the
852 852 working copy is dirty::
853 853
854 854 [rebase]
855 855 experimental.inmemory = True
856 856
857 857 Return Values:
858 858
859 859 Returns 0 on success, 1 if nothing to rebase or there are
860 860 unresolved conflicts.
861 861
862 862 """
863 863 opts = pycompat.byteskwargs(opts)
864 864 inmemory = ui.configbool('rebase', 'experimental.inmemory')
865 865 dryrun = opts.get('dry_run')
866 866 confirm = opts.get('confirm')
867 867 selactions = [k for k in ['abort', 'stop', 'continue'] if opts.get(k)]
868 868 if len(selactions) > 1:
869 869 raise error.Abort(_('cannot use --%s with --%s')
870 870 % tuple(selactions[:2]))
871 871 action = selactions[0] if selactions else None
872 872 if dryrun and action:
873 873 raise error.Abort(_('cannot specify both --dry-run and --%s') % action)
874 874 if confirm and action:
875 875 raise error.Abort(_('cannot specify both --confirm and --%s') % action)
876 876 if dryrun and confirm:
877 877 raise error.Abort(_('cannot specify both --confirm and --dry-run'))
878 878
879 879 if action or repo.currenttransaction() is not None:
880 880 # in-memory rebase is not compatible with resuming rebases.
881 881 # (Or if it is run within a transaction, since the restart logic can
882 882 # fail the entire transaction.)
883 883 inmemory = False
884 884
885 885 if opts.get('auto_orphans'):
886 886 for key in opts:
887 887 if key != 'auto_orphans' and opts.get(key):
888 888 raise error.Abort(_('--auto-orphans is incompatible with %s') %
889 889 ('--' + key))
890 890 userrevs = list(repo.revs(opts.get('auto_orphans')))
891 891 opts['rev'] = [revsetlang.formatspec('%ld and orphan()', userrevs)]
892 892 opts['dest'] = '_destautoorphanrebase(SRC)'
893 893
894 894 if dryrun or confirm:
895 895 return _dryrunrebase(ui, repo, action, opts)
896 896 elif action == 'stop':
897 897 rbsrt = rebaseruntime(repo, ui)
898 898 with repo.wlock(), repo.lock():
899 899 rbsrt.restorestatus()
900 900 if rbsrt.collapsef:
901 901 raise error.Abort(_("cannot stop in --collapse session"))
902 902 allowunstable = obsolete.isenabled(repo, obsolete.allowunstableopt)
903 903 if not (rbsrt.keepf or allowunstable):
904 904 raise error.Abort(_("cannot remove original changesets with"
905 905 " unrebased descendants"),
906 906 hint=_('either enable obsmarkers to allow unstable '
907 907 'revisions or use --keep to keep original '
908 908 'changesets'))
909 909 if needupdate(repo, rbsrt.state):
910 910 # update to the current working revision
911 911 # to clear interrupted merge
912 912 hg.updaterepo(repo, rbsrt.originalwd, overwrite=True)
913 913 rbsrt._finishrebase()
914 914 return 0
915 915 elif inmemory:
916 916 try:
917 917 # in-memory merge doesn't support conflicts, so if we hit any, abort
918 918 # and re-run as an on-disk merge.
919 919 overrides = {('rebase', 'singletransaction'): True}
920 920 with ui.configoverride(overrides, 'rebase'):
921 921 return _dorebase(ui, repo, action, opts, inmemory=inmemory)
922 922 except error.InMemoryMergeConflictsError:
923 923 ui.warn(_('hit merge conflicts; re-running rebase without in-memory'
924 924 ' merge\n'))
925 925 # TODO: Make in-memory merge not use the on-disk merge state, so
926 926 # we don't have to clean it here
927 927 mergemod.mergestate.clean(repo)
928 928 clearstatus(repo)
929 929 clearcollapsemsg(repo)
930 930 return _dorebase(ui, repo, action, opts, inmemory=False)
931 931 else:
932 932 return _dorebase(ui, repo, action, opts)
933 933
934 934 def _dryrunrebase(ui, repo, action, opts):
935 935 rbsrt = rebaseruntime(repo, ui, inmemory=True, opts=opts)
936 936 confirm = opts.get('confirm')
937 937 if confirm:
938 938 ui.status(_('starting in-memory rebase\n'))
939 939 else:
940 940 ui.status(_('starting dry-run rebase; repository will not be '
941 941 'changed\n'))
942 942 with repo.wlock(), repo.lock():
943 943 needsabort = True
944 944 try:
945 945 overrides = {('rebase', 'singletransaction'): True}
946 946 with ui.configoverride(overrides, 'rebase'):
947 947 _origrebase(ui, repo, action, opts, rbsrt, inmemory=True,
948 948 leaveunfinished=True)
949 949 except error.InMemoryMergeConflictsError:
950 950 ui.status(_('hit a merge conflict\n'))
951 951 return 1
952 952 else:
953 953 if confirm:
954 954 ui.status(_('rebase completed successfully\n'))
955 955 if not ui.promptchoice(_(b'apply changes (yn)?'
956 956 b'$$ &Yes $$ &No')):
957 957 # finish unfinished rebase
958 958 rbsrt._finishrebase()
959 959 else:
960 960 rbsrt._prepareabortorcontinue(isabort=True, backup=False,
961 961 suppwarns=True)
962 962 needsabort = False
963 963 else:
964 964 ui.status(_('dry-run rebase completed successfully; run without'
965 965 ' -n/--dry-run to perform this rebase\n'))
966 966 return 0
967 967 finally:
968 968 if needsabort:
969 969 # no need to store backup in case of dryrun
970 970 rbsrt._prepareabortorcontinue(isabort=True, backup=False,
971 971 suppwarns=True)
972 972
973 973 def _dorebase(ui, repo, action, opts, inmemory=False):
974 974 rbsrt = rebaseruntime(repo, ui, inmemory, opts)
975 975 return _origrebase(ui, repo, action, opts, rbsrt, inmemory=inmemory)
976 976
977 977 def _origrebase(ui, repo, action, opts, rbsrt, inmemory=False,
978 978 leaveunfinished=False):
979 979 assert action != 'stop'
980 980 with repo.wlock(), repo.lock():
981 981 # Validate input and define rebasing points
982 982 destf = opts.get('dest', None)
983 983 srcf = opts.get('source', None)
984 984 basef = opts.get('base', None)
985 985 revf = opts.get('rev', [])
986 986 # search default destination in this space
987 987 # used in the 'hg pull --rebase' case, see issue 5214.
988 988 destspace = opts.get('_destspace')
989 989 if opts.get('interactive'):
990 990 try:
991 991 if extensions.find('histedit'):
992 992 enablehistedit = ''
993 993 except KeyError:
994 994 enablehistedit = " --config extensions.histedit="
995 995 help = "hg%s help -e histedit" % enablehistedit
996 996 msg = _("interactive history editing is supported by the "
997 997 "'histedit' extension (see \"%s\")") % help
998 998 raise error.Abort(msg)
999 999
1000 1000 if rbsrt.collapsemsg and not rbsrt.collapsef:
1001 1001 raise error.Abort(
1002 1002 _('message can only be specified with collapse'))
1003 1003
1004 1004 if action:
1005 1005 if rbsrt.collapsef:
1006 1006 raise error.Abort(
1007 1007 _('cannot use collapse with continue or abort'))
1008 1008 if srcf or basef or destf:
1009 1009 raise error.Abort(
1010 1010 _('abort and continue do not allow specifying revisions'))
1011 1011 if action == 'abort' and opts.get('tool', False):
1012 1012 ui.warn(_('tool option will be ignored\n'))
1013 1013 if action == 'continue':
1014 1014 ms = mergemod.mergestate.read(repo)
1015 1015 mergeutil.checkunresolved(ms)
1016 1016
1017 1017 retcode = rbsrt._prepareabortorcontinue(isabort=(action == 'abort'))
1018 1018 if retcode is not None:
1019 1019 return retcode
1020 1020 else:
1021 1021 destmap = _definedestmap(ui, repo, inmemory, destf, srcf, basef,
1022 1022 revf, destspace=destspace)
1023 1023 retcode = rbsrt._preparenewrebase(destmap)
1024 1024 if retcode is not None:
1025 1025 return retcode
1026 1026 storecollapsemsg(repo, rbsrt.collapsemsg)
1027 1027
1028 1028 tr = None
1029 1029
1030 1030 singletr = ui.configbool('rebase', 'singletransaction')
1031 1031 if singletr:
1032 1032 tr = repo.transaction('rebase')
1033 1033
1034 1034 # If `rebase.singletransaction` is enabled, wrap the entire operation in
1035 1035 # one transaction here. Otherwise, transactions are obtained when
1036 1036 # committing each node, which is slower but allows partial success.
1037 1037 with util.acceptintervention(tr):
1038 1038 # Same logic for the dirstate guard, except we don't create one when
1039 1039 # rebasing in-memory (it's not needed).
1040 1040 dsguard = None
1041 1041 if singletr and not inmemory:
1042 1042 dsguard = dirstateguard.dirstateguard(repo, 'rebase')
1043 1043 with util.acceptintervention(dsguard):
1044 1044 rbsrt._performrebase(tr)
1045 1045 if not leaveunfinished:
1046 1046 rbsrt._finishrebase()
1047 1047
1048 1048 def _definedestmap(ui, repo, inmemory, destf=None, srcf=None, basef=None,
1049 1049 revf=None, destspace=None):
1050 1050 """use revisions argument to define destmap {srcrev: destrev}"""
1051 1051 if revf is None:
1052 1052 revf = []
1053 1053
1054 1054 # destspace is here to work around issues with `hg pull --rebase` see
1055 1055 # issue5214 for details
1056 1056 if srcf and basef:
1057 1057 raise error.Abort(_('cannot specify both a source and a base'))
1058 1058 if revf and basef:
1059 1059 raise error.Abort(_('cannot specify both a revision and a base'))
1060 1060 if revf and srcf:
1061 1061 raise error.Abort(_('cannot specify both a revision and a source'))
1062 1062
1063 1063 if not inmemory:
1064 1064 cmdutil.checkunfinished(repo)
1065 1065 cmdutil.bailifchanged(repo)
1066 1066
1067 1067 if ui.configbool('commands', 'rebase.requiredest') and not destf:
1068 1068 raise error.Abort(_('you must specify a destination'),
1069 1069 hint=_('use: hg rebase -d REV'))
1070 1070
1071 1071 dest = None
1072 1072
1073 1073 if revf:
1074 1074 rebaseset = scmutil.revrange(repo, revf)
1075 1075 if not rebaseset:
1076 1076 ui.status(_('empty "rev" revision set - nothing to rebase\n'))
1077 1077 return None
1078 1078 elif srcf:
1079 1079 src = scmutil.revrange(repo, [srcf])
1080 1080 if not src:
1081 1081 ui.status(_('empty "source" revision set - nothing to rebase\n'))
1082 1082 return None
1083 1083 rebaseset = repo.revs('(%ld)::', src)
1084 1084 assert rebaseset
1085 1085 else:
1086 1086 base = scmutil.revrange(repo, [basef or '.'])
1087 1087 if not base:
1088 1088 ui.status(_('empty "base" revision set - '
1089 1089 "can't compute rebase set\n"))
1090 1090 return None
1091 1091 if destf:
1092 1092 # --base does not support multiple destinations
1093 1093 dest = scmutil.revsingle(repo, destf)
1094 1094 else:
1095 1095 dest = repo[_destrebase(repo, base, destspace=destspace)]
1096 1096 destf = bytes(dest)
1097 1097
1098 1098 roots = [] # selected children of branching points
1099 1099 bpbase = {} # {branchingpoint: [origbase]}
1100 1100 for b in base: # group bases by branching points
1101 1101 bp = repo.revs('ancestor(%d, %d)', b, dest.rev()).first()
1102 1102 bpbase[bp] = bpbase.get(bp, []) + [b]
1103 1103 if None in bpbase:
1104 1104 # emulate the old behavior, showing "nothing to rebase" (a better
1105 1105 # behavior may be abort with "cannot find branching point" error)
1106 1106 bpbase.clear()
1107 1107 for bp, bs in bpbase.iteritems(): # calculate roots
1108 1108 roots += list(repo.revs('children(%d) & ancestors(%ld)', bp, bs))
1109 1109
1110 1110 rebaseset = repo.revs('%ld::', roots)
1111 1111
1112 1112 if not rebaseset:
1113 1113 # transform to list because smartsets are not comparable to
1114 1114 # lists. This should be improved to honor laziness of
1115 1115 # smartset.
1116 1116 if list(base) == [dest.rev()]:
1117 1117 if basef:
1118 1118 ui.status(_('nothing to rebase - %s is both "base"'
1119 1119 ' and destination\n') % dest)
1120 1120 else:
1121 1121 ui.status(_('nothing to rebase - working directory '
1122 1122 'parent is also destination\n'))
1123 1123 elif not repo.revs('%ld - ::%d', base, dest.rev()):
1124 1124 if basef:
1125 1125 ui.status(_('nothing to rebase - "base" %s is '
1126 1126 'already an ancestor of destination '
1127 1127 '%s\n') %
1128 1128 ('+'.join(bytes(repo[r]) for r in base),
1129 1129 dest))
1130 1130 else:
1131 1131 ui.status(_('nothing to rebase - working '
1132 1132 'directory parent is already an '
1133 1133 'ancestor of destination %s\n') % dest)
1134 1134 else: # can it happen?
1135 1135 ui.status(_('nothing to rebase from %s to %s\n') %
1136 1136 ('+'.join(bytes(repo[r]) for r in base), dest))
1137 1137 return None
1138 1138
1139 1139 rebasingwcp = repo['.'].rev() in rebaseset
1140 1140 ui.log("rebase", "rebasing working copy parent: %r\n", rebasingwcp,
1141 1141 rebase_rebasing_wcp=rebasingwcp)
1142 1142 if inmemory and rebasingwcp:
1143 1143 # Check these since we did not before.
1144 1144 cmdutil.checkunfinished(repo)
1145 1145 cmdutil.bailifchanged(repo)
1146 1146
1147 1147 if not destf:
1148 1148 dest = repo[_destrebase(repo, rebaseset, destspace=destspace)]
1149 1149 destf = bytes(dest)
1150 1150
1151 1151 allsrc = revsetlang.formatspec('%ld', rebaseset)
1152 1152 alias = {'ALLSRC': allsrc}
1153 1153
1154 1154 if dest is None:
1155 1155 try:
1156 1156 # fast path: try to resolve dest without SRC alias
1157 1157 dest = scmutil.revsingle(repo, destf, localalias=alias)
1158 1158 except error.RepoLookupError:
1159 1159 # multi-dest path: resolve dest for each SRC separately
1160 1160 destmap = {}
1161 1161 for r in rebaseset:
1162 1162 alias['SRC'] = revsetlang.formatspec('%d', r)
1163 1163 # use repo.anyrevs instead of scmutil.revsingle because we
1164 1164 # don't want to abort if destset is empty.
1165 1165 destset = repo.anyrevs([destf], user=True, localalias=alias)
1166 1166 size = len(destset)
1167 1167 if size == 1:
1168 1168 destmap[r] = destset.first()
1169 1169 elif size == 0:
1170 1170 ui.note(_('skipping %s - empty destination\n') % repo[r])
1171 1171 else:
1172 1172 raise error.Abort(_('rebase destination for %s is not '
1173 1173 'unique') % repo[r])
1174 1174
1175 1175 if dest is not None:
1176 1176 # single-dest case: assign dest to each rev in rebaseset
1177 1177 destrev = dest.rev()
1178 1178 destmap = {r: destrev for r in rebaseset} # {srcrev: destrev}
1179 1179
1180 1180 if not destmap:
1181 1181 ui.status(_('nothing to rebase - empty destination\n'))
1182 1182 return None
1183 1183
1184 1184 return destmap
1185 1185
1186 1186 def externalparent(repo, state, destancestors):
1187 1187 """Return the revision that should be used as the second parent
1188 1188 when the revisions in state is collapsed on top of destancestors.
1189 1189 Abort if there is more than one parent.
1190 1190 """
1191 1191 parents = set()
1192 1192 source = min(state)
1193 1193 for rev in state:
1194 1194 if rev == source:
1195 1195 continue
1196 1196 for p in repo[rev].parents():
1197 1197 if (p.rev() not in state
1198 1198 and p.rev() not in destancestors):
1199 1199 parents.add(p.rev())
1200 1200 if not parents:
1201 1201 return nullrev
1202 1202 if len(parents) == 1:
1203 1203 return parents.pop()
1204 1204 raise error.Abort(_('unable to collapse on top of %d, there is more '
1205 1205 'than one external parent: %s') %
1206 1206 (max(destancestors),
1207 1207 ', '.join("%d" % p for p in sorted(parents))))
1208 1208
1209 1209 def commitmemorynode(repo, p1, p2, wctx, editor, extra, user, date, commitmsg):
1210 1210 '''Commit the memory changes with parents p1 and p2.
1211 1211 Return node of committed revision.'''
1212 1212 # Replicates the empty check in ``repo.commit``.
1213 1213 if wctx.isempty() and not repo.ui.configbool('ui', 'allowemptycommit'):
1214 1214 return None
1215 1215
1216 1216 # By convention, ``extra['branch']`` (set by extrafn) clobbers
1217 1217 # ``branch`` (used when passing ``--keepbranches``).
1218 1218 branch = repo[p1].branch()
1219 1219 if 'branch' in extra:
1220 1220 branch = extra['branch']
1221 1221
1222 1222 memctx = wctx.tomemctx(commitmsg, parents=(p1, p2), date=date,
1223 1223 extra=extra, user=user, branch=branch, editor=editor)
1224 1224 commitres = repo.commitctx(memctx)
1225 1225 wctx.clean() # Might be reused
1226 1226 return commitres
1227 1227
1228 1228 def commitnode(repo, p1, p2, editor, extra, user, date, commitmsg):
1229 1229 '''Commit the wd changes with parents p1 and p2.
1230 1230 Return node of committed revision.'''
1231 1231 dsguard = util.nullcontextmanager()
1232 1232 if not repo.ui.configbool('rebase', 'singletransaction'):
1233 1233 dsguard = dirstateguard.dirstateguard(repo, 'rebase')
1234 1234 with dsguard:
1235 1235 repo.setparents(repo[p1].node(), repo[p2].node())
1236 1236
1237 1237 # Commit might fail if unresolved files exist
1238 1238 newnode = repo.commit(text=commitmsg, user=user, date=date,
1239 1239 extra=extra, editor=editor)
1240 1240
1241 1241 repo.dirstate.setbranch(repo[newnode].branch())
1242 1242 return newnode
1243 1243
1244 1244 def rebasenode(repo, rev, p1, base, collapse, dest, wctx):
1245 1245 'Rebase a single revision rev on top of p1 using base as merge ancestor'
1246 1246 # Merge phase
1247 1247 # Update to destination and merge it with local
1248 1248 if wctx.isinmemory():
1249 1249 wctx.setbase(repo[p1])
1250 1250 else:
1251 1251 if repo['.'].rev() != p1:
1252 1252 repo.ui.debug(" update to %d:%s\n" % (p1, repo[p1]))
1253 1253 mergemod.update(repo, p1, branchmerge=False, force=True)
1254 1254 else:
1255 1255 repo.ui.debug(" already in destination\n")
1256 1256 # This is, alas, necessary to invalidate workingctx's manifest cache,
1257 1257 # as well as other data we litter on it in other places.
1258 1258 wctx = repo[None]
1259 1259 repo.dirstate.write(repo.currenttransaction())
1260 1260 repo.ui.debug(" merge against %d:%s\n" % (rev, repo[rev]))
1261 1261 if base is not None:
1262 1262 repo.ui.debug(" detach base %d:%s\n" % (base, repo[base]))
1263 1263 # When collapsing in-place, the parent is the common ancestor, we
1264 1264 # have to allow merging with it.
1265 1265 stats = mergemod.update(repo, rev, branchmerge=True, force=True,
1266 1266 ancestor=base, mergeancestor=collapse,
1267 1267 labels=['dest', 'source'], wc=wctx)
1268 1268 if collapse:
1269 1269 copies.duplicatecopies(repo, wctx, rev, dest)
1270 1270 else:
1271 1271 # If we're not using --collapse, we need to
1272 1272 # duplicate copies between the revision we're
1273 1273 # rebasing and its first parent, but *not*
1274 1274 # duplicate any copies that have already been
1275 1275 # performed in the destination.
1276 1276 p1rev = repo[rev].p1().rev()
1277 1277 copies.duplicatecopies(repo, wctx, rev, p1rev, skiprev=dest)
1278 1278 return stats
1279 1279
1280 1280 def adjustdest(repo, rev, destmap, state, skipped):
1281 1281 """adjust rebase destination given the current rebase state
1282 1282
1283 1283 rev is what is being rebased. Return a list of two revs, which are the
1284 1284 adjusted destinations for rev's p1 and p2, respectively. If a parent is
1285 1285 nullrev, return dest without adjustment for it.
1286 1286
1287 1287 For example, when doing rebasing B+E to F, C to G, rebase will first move B
1288 1288 to B1, and E's destination will be adjusted from F to B1.
1289 1289
1290 1290 B1 <- written during rebasing B
1291 1291 |
1292 1292 F <- original destination of B, E
1293 1293 |
1294 1294 | E <- rev, which is being rebased
1295 1295 | |
1296 1296 | D <- prev, one parent of rev being checked
1297 1297 | |
1298 1298 | x <- skipped, ex. no successor or successor in (::dest)
1299 1299 | |
1300 1300 | C <- rebased as C', different destination
1301 1301 | |
1302 1302 | B <- rebased as B1 C'
1303 1303 |/ |
1304 1304 A G <- destination of C, different
1305 1305
1306 1306 Another example about merge changeset, rebase -r C+G+H -d K, rebase will
1307 1307 first move C to C1, G to G1, and when it's checking H, the adjusted
1308 1308 destinations will be [C1, G1].
1309 1309
1310 1310 H C1 G1
1311 1311 /| | /
1312 1312 F G |/
1313 1313 K | | -> K
1314 1314 | C D |
1315 1315 | |/ |
1316 1316 | B | ...
1317 1317 |/ |/
1318 1318 A A
1319 1319
1320 1320 Besides, adjust dest according to existing rebase information. For example,
1321 1321
1322 1322 B C D B needs to be rebased on top of C, C needs to be rebased on top
1323 1323 \|/ of D. We will rebase C first.
1324 1324 A
1325 1325
1326 1326 C' After rebasing C, when considering B's destination, use C'
1327 1327 | instead of the original C.
1328 1328 B D
1329 1329 \ /
1330 1330 A
1331 1331 """
1332 1332 # pick already rebased revs with same dest from state as interesting source
1333 1333 dest = destmap[rev]
1334 1334 source = [s for s, d in state.items()
1335 1335 if d > 0 and destmap[s] == dest and s not in skipped]
1336 1336
1337 1337 result = []
1338 1338 for prev in repo.changelog.parentrevs(rev):
1339 1339 adjusted = dest
1340 1340 if prev != nullrev:
1341 1341 candidate = repo.revs('max(%ld and (::%d))', source, prev).first()
1342 1342 if candidate is not None:
1343 1343 adjusted = state[candidate]
1344 1344 if adjusted == dest and dest in state:
1345 1345 adjusted = state[dest]
1346 1346 if adjusted == revtodo:
1347 1347 # sortsource should produce an order that makes this impossible
1348 1348 raise error.ProgrammingError(
1349 1349 'rev %d should be rebased already at this time' % dest)
1350 1350 result.append(adjusted)
1351 1351 return result
1352 1352
1353 1353 def _checkobsrebase(repo, ui, rebaseobsrevs, rebaseobsskipped):
1354 1354 """
1355 1355 Abort if rebase will create divergence or rebase is noop because of markers
1356 1356
1357 1357 `rebaseobsrevs`: set of obsolete revision in source
1358 1358 `rebaseobsskipped`: set of revisions from source skipped because they have
1359 1359 successors in destination or no non-obsolete successor.
1360 1360 """
1361 1361 # Obsolete node with successors not in dest leads to divergence
1362 1362 divergenceok = ui.configbool('experimental',
1363 1363 'evolution.allowdivergence')
1364 1364 divergencebasecandidates = rebaseobsrevs - rebaseobsskipped
1365 1365
1366 1366 if divergencebasecandidates and not divergenceok:
1367 1367 divhashes = (bytes(repo[r])
1368 1368 for r in divergencebasecandidates)
1369 1369 msg = _("this rebase will cause "
1370 1370 "divergences from: %s")
1371 1371 h = _("to force the rebase please set "
1372 1372 "experimental.evolution.allowdivergence=True")
1373 1373 raise error.Abort(msg % (",".join(divhashes),), hint=h)
1374 1374
1375 1375 def successorrevs(unfi, rev):
1376 1376 """yield revision numbers for successors of rev"""
1377 1377 assert unfi.filtername is None
1378 1378 nodemap = unfi.changelog.nodemap
1379 1379 for s in obsutil.allsuccessors(unfi.obsstore, [unfi[rev].node()]):
1380 1380 if s in nodemap:
1381 1381 yield nodemap[s]
1382 1382
1383 1383 def defineparents(repo, rev, destmap, state, skipped, obsskipped):
1384 1384 """Return new parents and optionally a merge base for rev being rebased
1385 1385
1386 1386 The destination specified by "dest" cannot always be used directly because
1387 1387 previously rebase result could affect destination. For example,
1388 1388
1389 1389 D E rebase -r C+D+E -d B
1390 1390 |/ C will be rebased to C'
1391 1391 B C D's new destination will be C' instead of B
1392 1392 |/ E's new destination will be C' instead of B
1393 1393 A
1394 1394
1395 1395 The new parents of a merge is slightly more complicated. See the comment
1396 1396 block below.
1397 1397 """
1398 1398 # use unfiltered changelog since successorrevs may return filtered nodes
1399 1399 assert repo.filtername is None
1400 1400 cl = repo.changelog
1401 1401 isancestor = cl.isancestorrev
1402 1402
1403 1403 dest = destmap[rev]
1404 1404 oldps = repo.changelog.parentrevs(rev) # old parents
1405 1405 newps = [nullrev, nullrev] # new parents
1406 1406 dests = adjustdest(repo, rev, destmap, state, skipped)
1407 1407 bases = list(oldps) # merge base candidates, initially just old parents
1408 1408
1409 1409 if all(r == nullrev for r in oldps[1:]):
1410 1410 # For non-merge changeset, just move p to adjusted dest as requested.
1411 1411 newps[0] = dests[0]
1412 1412 else:
1413 1413 # For merge changeset, if we move p to dests[i] unconditionally, both
1414 1414 # parents may change and the end result looks like "the merge loses a
1415 1415 # parent", which is a surprise. This is a limit because "--dest" only
1416 1416 # accepts one dest per src.
1417 1417 #
1418 1418 # Therefore, only move p with reasonable conditions (in this order):
1419 1419 # 1. use dest, if dest is a descendent of (p or one of p's successors)
1420 1420 # 2. use p's rebased result, if p is rebased (state[p] > 0)
1421 1421 #
1422 1422 # Comparing with adjustdest, the logic here does some additional work:
1423 1423 # 1. decide which parents will not be moved towards dest
1424 1424 # 2. if the above decision is "no", should a parent still be moved
1425 1425 # because it was rebased?
1426 1426 #
1427 1427 # For example:
1428 1428 #
1429 1429 # C # "rebase -r C -d D" is an error since none of the parents
1430 1430 # /| # can be moved. "rebase -r B+C -d D" will move C's parent
1431 1431 # A B D # B (using rule "2."), since B will be rebased.
1432 1432 #
1433 1433 # The loop tries to be not rely on the fact that a Mercurial node has
1434 1434 # at most 2 parents.
1435 1435 for i, p in enumerate(oldps):
1436 1436 np = p # new parent
1437 1437 if any(isancestor(x, dests[i]) for x in successorrevs(repo, p)):
1438 1438 np = dests[i]
1439 1439 elif p in state and state[p] > 0:
1440 1440 np = state[p]
1441 1441
1442 1442 # "bases" only record "special" merge bases that cannot be
1443 1443 # calculated from changelog DAG (i.e. isancestor(p, np) is False).
1444 1444 # For example:
1445 1445 #
1446 1446 # B' # rebase -s B -d D, when B was rebased to B'. dest for C
1447 1447 # | C # is B', but merge base for C is B, instead of
1448 1448 # D | # changelog.ancestor(C, B') == A. If changelog DAG and
1449 1449 # | B # "state" edges are merged (so there will be an edge from
1450 1450 # |/ # B to B'), the merge base is still ancestor(C, B') in
1451 1451 # A # the merged graph.
1452 1452 #
1453 1453 # Also see https://bz.mercurial-scm.org/show_bug.cgi?id=1950#c8
1454 1454 # which uses "virtual null merge" to explain this situation.
1455 1455 if isancestor(p, np):
1456 1456 bases[i] = nullrev
1457 1457
1458 1458 # If one parent becomes an ancestor of the other, drop the ancestor
1459 1459 for j, x in enumerate(newps[:i]):
1460 1460 if x == nullrev:
1461 1461 continue
1462 1462 if isancestor(np, x): # CASE-1
1463 1463 np = nullrev
1464 1464 elif isancestor(x, np): # CASE-2
1465 1465 newps[j] = np
1466 1466 np = nullrev
1467 1467 # New parents forming an ancestor relationship does not
1468 1468 # mean the old parents have a similar relationship. Do not
1469 1469 # set bases[x] to nullrev.
1470 1470 bases[j], bases[i] = bases[i], bases[j]
1471 1471
1472 1472 newps[i] = np
1473 1473
1474 1474 # "rebasenode" updates to new p1, and the old p1 will be used as merge
1475 1475 # base. If only p2 changes, merging using unchanged p1 as merge base is
1476 1476 # suboptimal. Therefore swap parents to make the merge sane.
1477 1477 if newps[1] != nullrev and oldps[0] == newps[0]:
1478 1478 assert len(newps) == 2 and len(oldps) == 2
1479 1479 newps.reverse()
1480 1480 bases.reverse()
1481 1481
1482 1482 # No parent change might be an error because we fail to make rev a
1483 1483 # descendent of requested dest. This can happen, for example:
1484 1484 #
1485 1485 # C # rebase -r C -d D
1486 1486 # /| # None of A and B will be changed to D and rebase fails.
1487 1487 # A B D
1488 1488 if set(newps) == set(oldps) and dest not in newps:
1489 1489 raise error.Abort(_('cannot rebase %d:%s without '
1490 1490 'moving at least one of its parents')
1491 1491 % (rev, repo[rev]))
1492 1492
1493 1493 # Source should not be ancestor of dest. The check here guarantees it's
1494 1494 # impossible. With multi-dest, the initial check does not cover complex
1495 1495 # cases since we don't have abstractions to dry-run rebase cheaply.
1496 1496 if any(p != nullrev and isancestor(rev, p) for p in newps):
1497 1497 raise error.Abort(_('source is ancestor of destination'))
1498 1498
1499 1499 # "rebasenode" updates to new p1, use the corresponding merge base.
1500 1500 if bases[0] != nullrev:
1501 1501 base = bases[0]
1502 1502 else:
1503 1503 base = None
1504 1504
1505 1505 # Check if the merge will contain unwanted changes. That may happen if
1506 1506 # there are multiple special (non-changelog ancestor) merge bases, which
1507 1507 # cannot be handled well by the 3-way merge algorithm. For example:
1508 1508 #
1509 1509 # F
1510 1510 # /|
1511 1511 # D E # "rebase -r D+E+F -d Z", when rebasing F, if "D" was chosen
1512 1512 # | | # as merge base, the difference between D and F will include
1513 1513 # B C # C, so the rebased F will contain C surprisingly. If "E" was
1514 1514 # |/ # chosen, the rebased F will contain B.
1515 1515 # A Z
1516 1516 #
1517 1517 # But our merge base candidates (D and E in above case) could still be
1518 1518 # better than the default (ancestor(F, Z) == null). Therefore still
1519 1519 # pick one (so choose p1 above).
1520 1520 if sum(1 for b in bases if b != nullrev) > 1:
1521 1521 unwanted = [None, None] # unwanted[i]: unwanted revs if choose bases[i]
1522 1522 for i, base in enumerate(bases):
1523 1523 if base == nullrev:
1524 1524 continue
1525 1525 # Revisions in the side (not chosen as merge base) branch that
1526 1526 # might contain "surprising" contents
1527 1527 siderevs = list(repo.revs('((%ld-%d) %% (%d+%d))',
1528 1528 bases, base, base, dest))
1529 1529
1530 1530 # If those revisions are covered by rebaseset, the result is good.
1531 1531 # A merge in rebaseset would be considered to cover its ancestors.
1532 1532 if siderevs:
1533 1533 rebaseset = [r for r, d in state.items()
1534 1534 if d > 0 and r not in obsskipped]
1535 1535 merges = [r for r in rebaseset
1536 1536 if cl.parentrevs(r)[1] != nullrev]
1537 1537 unwanted[i] = list(repo.revs('%ld - (::%ld) - %ld',
1538 1538 siderevs, merges, rebaseset))
1539 1539
1540 1540 # Choose a merge base that has a minimal number of unwanted revs.
1541 1541 l, i = min((len(revs), i)
1542 1542 for i, revs in enumerate(unwanted) if revs is not None)
1543 1543 base = bases[i]
1544 1544
1545 1545 # newps[0] should match merge base if possible. Currently, if newps[i]
1546 1546 # is nullrev, the only case is newps[i] and newps[j] (j < i), one is
1547 1547 # the other's ancestor. In that case, it's fine to not swap newps here.
1548 1548 # (see CASE-1 and CASE-2 above)
1549 1549 if i != 0 and newps[i] != nullrev:
1550 1550 newps[0], newps[i] = newps[i], newps[0]
1551 1551
1552 1552 # The merge will include unwanted revisions. Abort now. Revisit this if
1553 1553 # we have a more advanced merge algorithm that handles multiple bases.
1554 1554 if l > 0:
1555 1555 unwanteddesc = _(' or ').join(
1556 1556 (', '.join('%d:%s' % (r, repo[r]) for r in revs)
1557 1557 for revs in unwanted if revs is not None))
1558 1558 raise error.Abort(
1559 1559 _('rebasing %d:%s will include unwanted changes from %s')
1560 1560 % (rev, repo[rev], unwanteddesc))
1561 1561
1562 1562 repo.ui.debug(" future parents are %d and %d\n" % tuple(newps))
1563 1563
1564 1564 return newps[0], newps[1], base
1565 1565
1566 1566 def isagitpatch(repo, patchname):
1567 1567 'Return true if the given patch is in git format'
1568 1568 mqpatch = os.path.join(repo.mq.path, patchname)
1569 1569 for line in patch.linereader(open(mqpatch, 'rb')):
1570 1570 if line.startswith('diff --git'):
1571 1571 return True
1572 1572 return False
1573 1573
1574 1574 def updatemq(repo, state, skipped, **opts):
1575 1575 'Update rebased mq patches - finalize and then import them'
1576 1576 mqrebase = {}
1577 1577 mq = repo.mq
1578 1578 original_series = mq.fullseries[:]
1579 1579 skippedpatches = set()
1580 1580
1581 1581 for p in mq.applied:
1582 1582 rev = repo[p.node].rev()
1583 1583 if rev in state:
1584 1584 repo.ui.debug('revision %d is an mq patch (%s), finalize it.\n' %
1585 1585 (rev, p.name))
1586 1586 mqrebase[rev] = (p.name, isagitpatch(repo, p.name))
1587 1587 else:
1588 1588 # Applied but not rebased, not sure this should happen
1589 1589 skippedpatches.add(p.name)
1590 1590
1591 1591 if mqrebase:
1592 1592 mq.finish(repo, mqrebase.keys())
1593 1593
1594 1594 # We must start import from the newest revision
1595 1595 for rev in sorted(mqrebase, reverse=True):
1596 1596 if rev not in skipped:
1597 1597 name, isgit = mqrebase[rev]
1598 1598 repo.ui.note(_('updating mq patch %s to %d:%s\n') %
1599 1599 (name, state[rev], repo[state[rev]]))
1600 1600 mq.qimport(repo, (), patchname=name, git=isgit,
1601 1601 rev=["%d" % state[rev]])
1602 1602 else:
1603 1603 # Rebased and skipped
1604 1604 skippedpatches.add(mqrebase[rev][0])
1605 1605
1606 1606 # Patches were either applied and rebased and imported in
1607 1607 # order, applied and removed or unapplied. Discard the removed
1608 1608 # ones while preserving the original series order and guards.
1609 1609 newseries = [s for s in original_series
1610 1610 if mq.guard_re.split(s, 1)[0] not in skippedpatches]
1611 1611 mq.fullseries[:] = newseries
1612 1612 mq.seriesdirty = True
1613 1613 mq.savedirty()
1614 1614
1615 1615 def storecollapsemsg(repo, collapsemsg):
1616 1616 'Store the collapse message to allow recovery'
1617 1617 collapsemsg = collapsemsg or ''
1618 1618 f = repo.vfs("last-message.txt", "w")
1619 1619 f.write("%s\n" % collapsemsg)
1620 1620 f.close()
1621 1621
1622 1622 def clearcollapsemsg(repo):
1623 1623 'Remove collapse message file'
1624 1624 repo.vfs.unlinkpath("last-message.txt", ignoremissing=True)
1625 1625
1626 1626 def restorecollapsemsg(repo, isabort):
1627 1627 'Restore previously stored collapse message'
1628 1628 try:
1629 1629 f = repo.vfs("last-message.txt")
1630 1630 collapsemsg = f.readline().strip()
1631 1631 f.close()
1632 1632 except IOError as err:
1633 1633 if err.errno != errno.ENOENT:
1634 1634 raise
1635 1635 if isabort:
1636 1636 # Oh well, just abort like normal
1637 1637 collapsemsg = ''
1638 1638 else:
1639 1639 raise error.Abort(_('missing .hg/last-message.txt for rebase'))
1640 1640 return collapsemsg
1641 1641
1642 1642 def clearstatus(repo):
1643 1643 'Remove the status files'
1644 1644 # Make sure the active transaction won't write the state file
1645 1645 tr = repo.currenttransaction()
1646 1646 if tr:
1647 1647 tr.removefilegenerator('rebasestate')
1648 1648 repo.vfs.unlinkpath("rebasestate", ignoremissing=True)
1649 1649
1650 1650 def needupdate(repo, state):
1651 1651 '''check whether we should `update --clean` away from a merge, or if
1652 1652 somehow the working dir got forcibly updated, e.g. by older hg'''
1653 1653 parents = [p.rev() for p in repo[None].parents()]
1654 1654
1655 1655 # Are we in a merge state at all?
1656 1656 if len(parents) < 2:
1657 1657 return False
1658 1658
1659 1659 # We should be standing on the first as-of-yet unrebased commit.
1660 1660 firstunrebased = min([old for old, new in state.iteritems()
1661 1661 if new == nullrev])
1662 1662 if firstunrebased in parents:
1663 1663 return True
1664 1664
1665 1665 return False
1666 1666
1667 1667 def sortsource(destmap):
1668 1668 """yield source revisions in an order that we only rebase things once
1669 1669
1670 1670 If source and destination overlaps, we should filter out revisions
1671 1671 depending on other revisions which hasn't been rebased yet.
1672 1672
1673 1673 Yield a sorted list of revisions each time.
1674 1674
1675 1675 For example, when rebasing A to B, B to C. This function yields [B], then
1676 1676 [A], indicating B needs to be rebased first.
1677 1677
1678 1678 Raise if there is a cycle so the rebase is impossible.
1679 1679 """
1680 1680 srcset = set(destmap)
1681 1681 while srcset:
1682 1682 srclist = sorted(srcset)
1683 1683 result = []
1684 1684 for r in srclist:
1685 1685 if destmap[r] not in srcset:
1686 1686 result.append(r)
1687 1687 if not result:
1688 1688 raise error.Abort(_('source and destination form a cycle'))
1689 1689 srcset -= set(result)
1690 1690 yield result
1691 1691
1692 1692 def buildstate(repo, destmap, collapse):
1693 1693 '''Define which revisions are going to be rebased and where
1694 1694
1695 1695 repo: repo
1696 1696 destmap: {srcrev: destrev}
1697 1697 '''
1698 1698 rebaseset = destmap.keys()
1699 1699 originalwd = repo['.'].rev()
1700 1700
1701 1701 # This check isn't strictly necessary, since mq detects commits over an
1702 1702 # applied patch. But it prevents messing up the working directory when
1703 1703 # a partially completed rebase is blocked by mq.
1704 1704 if 'qtip' in repo.tags():
1705 1705 mqapplied = set(repo[s.node].rev() for s in repo.mq.applied)
1706 1706 if set(destmap.values()) & mqapplied:
1707 1707 raise error.Abort(_('cannot rebase onto an applied mq patch'))
1708 1708
1709 1709 # Get "cycle" error early by exhausting the generator.
1710 1710 sortedsrc = list(sortsource(destmap)) # a list of sorted revs
1711 1711 if not sortedsrc:
1712 1712 raise error.Abort(_('no matching revisions'))
1713 1713
1714 1714 # Only check the first batch of revisions to rebase not depending on other
1715 1715 # rebaseset. This means "source is ancestor of destination" for the second
1716 1716 # (and following) batches of revisions are not checked here. We rely on
1717 1717 # "defineparents" to do that check.
1718 1718 roots = list(repo.set('roots(%ld)', sortedsrc[0]))
1719 1719 if not roots:
1720 1720 raise error.Abort(_('no matching revisions'))
1721 1721 def revof(r):
1722 1722 return r.rev()
1723 1723 roots = sorted(roots, key=revof)
1724 1724 state = dict.fromkeys(rebaseset, revtodo)
1725 1725 emptyrebase = (len(sortedsrc) == 1)
1726 1726 for root in roots:
1727 1727 dest = repo[destmap[root.rev()]]
1728 1728 commonbase = root.ancestor(dest)
1729 1729 if commonbase == root:
1730 1730 raise error.Abort(_('source is ancestor of destination'))
1731 1731 if commonbase == dest:
1732 1732 wctx = repo[None]
1733 1733 if dest == wctx.p1():
1734 1734 # when rebasing to '.', it will use the current wd branch name
1735 1735 samebranch = root.branch() == wctx.branch()
1736 1736 else:
1737 1737 samebranch = root.branch() == dest.branch()
1738 1738 if not collapse and samebranch and dest in root.parents():
1739 1739 # mark the revision as done by setting its new revision
1740 1740 # equal to its old (current) revisions
1741 1741 state[root.rev()] = root.rev()
1742 1742 repo.ui.debug('source is a child of destination\n')
1743 1743 continue
1744 1744
1745 1745 emptyrebase = False
1746 1746 repo.ui.debug('rebase onto %s starting from %s\n' % (dest, root))
1747 1747 if emptyrebase:
1748 1748 return None
1749 1749 for rev in sorted(state):
1750 1750 parents = [p for p in repo.changelog.parentrevs(rev) if p != nullrev]
1751 1751 # if all parents of this revision are done, then so is this revision
1752 1752 if parents and all((state.get(p) == p for p in parents)):
1753 1753 state[rev] = rev
1754 1754 return originalwd, destmap, state
1755 1755
1756 1756 def clearrebased(ui, repo, destmap, state, skipped, collapsedas=None,
1757 1757 keepf=False, fm=None, backup=True):
1758 1758 """dispose of rebased revision at the end of the rebase
1759 1759
1760 1760 If `collapsedas` is not None, the rebase was a collapse whose result if the
1761 1761 `collapsedas` node.
1762 1762
1763 1763 If `keepf` is not True, the rebase has --keep set and no nodes should be
1764 1764 removed (but bookmarks still need to be moved).
1765 1765
1766 1766 If `backup` is False, no backup will be stored when stripping rebased
1767 1767 revisions.
1768 1768 """
1769 1769 tonode = repo.changelog.node
1770 1770 replacements = {}
1771 1771 moves = {}
1772 1772 stripcleanup = not obsolete.isenabled(repo, obsolete.createmarkersopt)
1773 1773
1774 1774 collapsednodes = []
1775 1775 for rev, newrev in sorted(state.items()):
1776 1776 if newrev >= 0 and newrev != rev:
1777 1777 oldnode = tonode(rev)
1778 1778 newnode = collapsedas or tonode(newrev)
1779 1779 moves[oldnode] = newnode
1780 1780 if not keepf:
1781 1781 succs = None
1782 1782 if rev in skipped:
1783 1783 if stripcleanup or not repo[rev].obsolete():
1784 1784 succs = ()
1785 1785 elif collapsedas:
1786 1786 collapsednodes.append(oldnode)
1787 1787 else:
1788 1788 succs = (newnode,)
1789 1789 if succs is not None:
1790 1790 replacements[(oldnode,)] = succs
1791 1791 if collapsednodes:
1792 1792 replacements[tuple(collapsednodes)] = (collapsedas,)
1793 1793 scmutil.cleanupnodes(repo, replacements, 'rebase', moves, backup=backup)
1794 1794 if fm:
1795 1795 hf = fm.hexfunc
1796 1796 fl = fm.formatlist
1797 1797 fd = fm.formatdict
1798 1798 changes = {}
1799 1799 for oldns, newn in replacements.iteritems():
1800 1800 for oldn in oldns:
1801 1801 changes[hf(oldn)] = fl([hf(n) for n in newn], name='node')
1802 1802 nodechanges = fd(changes, key="oldnode", value="newnodes")
1803 1803 fm.data(nodechanges=nodechanges)
1804 1804
1805 1805 def pullrebase(orig, ui, repo, *args, **opts):
1806 1806 'Call rebase after pull if the latter has been invoked with --rebase'
1807 1807 ret = None
1808 1808 if opts.get(r'rebase'):
1809 1809 if ui.configbool('commands', 'rebase.requiredest'):
1810 1810 msg = _('rebase destination required by configuration')
1811 1811 hint = _('use hg pull followed by hg rebase -d DEST')
1812 1812 raise error.Abort(msg, hint=hint)
1813 1813
1814 1814 with repo.wlock(), repo.lock():
1815 1815 if opts.get(r'update'):
1816 1816 del opts[r'update']
1817 1817 ui.debug('--update and --rebase are not compatible, ignoring '
1818 1818 'the update flag\n')
1819 1819
1820 1820 cmdutil.checkunfinished(repo)
1821 1821 cmdutil.bailifchanged(repo, hint=_('cannot pull with rebase: '
1822 1822 'please commit or shelve your changes first'))
1823 1823
1824 1824 revsprepull = len(repo)
1825 1825 origpostincoming = commands.postincoming
1826 1826 def _dummy(*args, **kwargs):
1827 1827 pass
1828 1828 commands.postincoming = _dummy
1829 1829 try:
1830 1830 ret = orig(ui, repo, *args, **opts)
1831 1831 finally:
1832 1832 commands.postincoming = origpostincoming
1833 1833 revspostpull = len(repo)
1834 1834 if revspostpull > revsprepull:
1835 1835 # --rev option from pull conflict with rebase own --rev
1836 1836 # dropping it
1837 1837 if r'rev' in opts:
1838 1838 del opts[r'rev']
1839 1839 # positional argument from pull conflicts with rebase's own
1840 1840 # --source.
1841 1841 if r'source' in opts:
1842 1842 del opts[r'source']
1843 1843 # revsprepull is the len of the repo, not revnum of tip.
1844 1844 destspace = list(repo.changelog.revs(start=revsprepull))
1845 1845 opts[r'_destspace'] = destspace
1846 1846 try:
1847 1847 rebase(ui, repo, **opts)
1848 1848 except error.NoMergeDestAbort:
1849 1849 # we can maybe update instead
1850 1850 rev, _a, _b = destutil.destupdate(repo)
1851 1851 if rev == repo['.'].rev():
1852 1852 ui.status(_('nothing to rebase\n'))
1853 1853 else:
1854 1854 ui.status(_('nothing to rebase - updating instead\n'))
1855 1855 # not passing argument to get the bare update behavior
1856 1856 # with warning and trumpets
1857 1857 commands.update(ui, repo)
1858 1858 else:
1859 1859 if opts.get(r'tool'):
1860 1860 raise error.Abort(_('--tool can only be used with --rebase'))
1861 1861 ret = orig(ui, repo, *args, **opts)
1862 1862
1863 1863 return ret
1864 1864
1865 1865 def _filterobsoleterevs(repo, revs):
1866 1866 """returns a set of the obsolete revisions in revs"""
1867 1867 return set(r for r in revs if repo[r].obsolete())
1868 1868
1869 1869 def _computeobsoletenotrebased(repo, rebaseobsrevs, destmap):
1870 1870 """Return (obsoletenotrebased, obsoletewithoutsuccessorindestination).
1871 1871
1872 1872 `obsoletenotrebased` is a mapping mapping obsolete => successor for all
1873 1873 obsolete nodes to be rebased given in `rebaseobsrevs`.
1874 1874
1875 1875 `obsoletewithoutsuccessorindestination` is a set with obsolete revisions
1876 1876 without a successor in destination.
1877 1877
1878 1878 `obsoleteextinctsuccessors` is a set of obsolete revisions with only
1879 1879 obsolete successors.
1880 1880 """
1881 1881 obsoletenotrebased = {}
1882 1882 obsoletewithoutsuccessorindestination = set([])
1883 1883 obsoleteextinctsuccessors = set([])
1884 1884
1885 1885 assert repo.filtername is None
1886 1886 cl = repo.changelog
1887 1887 nodemap = cl.nodemap
1888 1888 extinctrevs = set(repo.revs('extinct()'))
1889 1889 for srcrev in rebaseobsrevs:
1890 1890 srcnode = cl.node(srcrev)
1891 1891 # XXX: more advanced APIs are required to handle split correctly
1892 1892 successors = set(obsutil.allsuccessors(repo.obsstore, [srcnode]))
1893 1893 # obsutil.allsuccessors includes node itself
1894 1894 successors.remove(srcnode)
1895 1895 succrevs = {nodemap[s] for s in successors if s in nodemap}
1896 1896 if succrevs.issubset(extinctrevs):
1897 1897 # all successors are extinct
1898 1898 obsoleteextinctsuccessors.add(srcrev)
1899 1899 if not successors:
1900 1900 # no successor
1901 1901 obsoletenotrebased[srcrev] = None
1902 1902 else:
1903 1903 dstrev = destmap[srcrev]
1904 1904 for succrev in succrevs:
1905 1905 if cl.isancestorrev(succrev, dstrev):
1906 1906 obsoletenotrebased[srcrev] = succrev
1907 1907 break
1908 1908 else:
1909 1909 # If 'srcrev' has a successor in rebase set but none in
1910 1910 # destination (which would be catched above), we shall skip it
1911 1911 # and its descendants to avoid divergence.
1912 1912 if srcrev in extinctrevs or any(s in destmap for s in succrevs):
1913 1913 obsoletewithoutsuccessorindestination.add(srcrev)
1914 1914
1915 1915 return (
1916 1916 obsoletenotrebased,
1917 1917 obsoletewithoutsuccessorindestination,
1918 1918 obsoleteextinctsuccessors,
1919 1919 )
1920 1920
1921 1921 def summaryhook(ui, repo):
1922 1922 if not repo.vfs.exists('rebasestate'):
1923 1923 return
1924 1924 try:
1925 1925 rbsrt = rebaseruntime(repo, ui, {})
1926 1926 rbsrt.restorestatus()
1927 1927 state = rbsrt.state
1928 1928 except error.RepoLookupError:
1929 1929 # i18n: column positioning for "hg summary"
1930 1930 msg = _('rebase: (use "hg rebase --abort" to clear broken state)\n')
1931 1931 ui.write(msg)
1932 1932 return
1933 1933 numrebased = len([i for i in state.itervalues() if i >= 0])
1934 1934 # i18n: column positioning for "hg summary"
1935 1935 ui.write(_('rebase: %s, %s (rebase --continue)\n') %
1936 1936 (ui.label(_('%d rebased'), 'rebase.rebased') % numrebased,
1937 1937 ui.label(_('%d remaining'), 'rebase.remaining') %
1938 1938 (len(state) - numrebased)))
1939 1939
1940 1940 def uisetup(ui):
1941 1941 #Replace pull with a decorator to provide --rebase option
1942 1942 entry = extensions.wrapcommand(commands.table, 'pull', pullrebase)
1943 1943 entry[1].append(('', 'rebase', None,
1944 1944 _("rebase working directory to branch head")))
1945 1945 entry[1].append(('t', 'tool', '',
1946 1946 _("specify merge tool for rebase")))
1947 1947 cmdutil.summaryhooks.add('rebase', summaryhook)
1948 1948 cmdutil.unfinishedstates.append(
1949 1949 ['rebasestate', False, False, _('rebase in progress'),
1950 1950 _("use 'hg rebase --continue' or 'hg rebase --abort'")])
1951 1951 cmdutil.afterresolvedstates.append(
1952 1952 ['rebasestate', _('hg rebase --continue')])
General Comments 0
You need to be logged in to leave comments. Login now