rebase.py
2287 lines
| 82.1 KiB
| text/x-python
|
PythonLexer
/ hgext / rebase.py
|
r6906 | # rebase.py - rebasing feature for mercurial | ||
# | ||||
# Copyright 2008 Stefano Tortarolo <stefano.tortarolo at gmail dot com> | ||||
# | ||||
|
r8225 | # This software may be used and distributed according to the terms of the | ||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r6906 | |||
|
r8934 | '''command to move sets of revisions to a different ancestor | ||
|
r6906 | |||
|
r7999 | This extension lets you rebase changesets in an existing Mercurial | ||
repository. | ||||
|
r6906 | |||
For more information: | ||||
|
r26421 | https://mercurial-scm.org/wiki/RebaseExtension | ||
|
r6906 | ''' | ||
|
r29128 | |||
import errno | ||||
import os | ||||
|
r29205 | |||
from mercurial.i18n import _ | ||||
from mercurial.node import ( | ||||
nullrev, | ||||
short, | ||||
|
r46729 | wdirrev, | ||
|
r29205 | ) | ||
|
r43355 | from mercurial.pycompat import open | ||
|
r29128 | from mercurial import ( | ||
bookmarks, | ||||
cmdutil, | ||||
commands, | ||||
copies, | ||||
destutil, | ||||
|
r30490 | dirstateguard, | ||
|
r29128 | error, | ||
extensions, | ||||
|
r48928 | logcmdutil, | ||
|
r30271 | merge as mergemod, | ||
|
r45383 | mergestate as mergestatemod, | ||
|
r30495 | mergeutil, | ||
|
r29128 | obsolete, | ||
r33146 | obsutil, | |||
|
r29128 | patch, | ||
phases, | ||||
|
r35003 | pycompat, | ||
|
r29128 | registrar, | ||
repair, | ||||
revset, | ||||
|
r34007 | revsetlang, | ||
|
r44386 | rewriteutil, | ||
|
r29128 | scmutil, | ||
|
r31023 | smartset, | ||
|
r38535 | state as statemod, | ||
|
r29128 | util, | ||
) | ||||
|
r26669 | # The following constants are used throughout the rebase module. The ordering of | ||
# their values must be maintained. | ||||
# Indicates that a revision needs to be rebased | ||||
|
r23490 | revtodo = -1 | ||
|
r43347 | revtodostr = b'-1' | ||
|
r33842 | |||
# legacy revstates no longer needed in current code | ||||
|
r33844 | # -2: nullmerge, -3: revignored, -4: revprecursor, -5: revpruned | ||
|
r43347 | legacystates = {b'-2', b'-3', b'-4', b'-5'} | ||
|
r10352 | |||
|
r14306 | cmdtable = {} | ||
|
r32337 | command = registrar.command(cmdtable) | ||
|
r47071 | |||
configtable = {} | ||||
configitem = registrar.configitem(configtable) | ||||
configitem( | ||||
b'devel', | ||||
b'rebase.force-in-memory-merge', | ||||
default=False, | ||||
) | ||||
|
r29841 | # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for | ||
|
r25186 | # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should | ||
# be specifying the version(s) of Mercurial they are tested with, or | ||||
# leave the attribute unspecified. | ||||
|
r43347 | testedwith = b'ships-with-hg-core' | ||
|
r14306 | |||
|
r43346 | |||
|
r26671 | def _nothingtorebase(): | ||
return 1 | ||||
|
r43346 | |||
|
r27976 | def _savegraft(ctx, extra): | ||
|
r43347 | s = ctx.extra().get(b'source', None) | ||
|
r27976 | if s is not None: | ||
|
r43347 | extra[b'source'] = s | ||
s = ctx.extra().get(b'intermediate-source', None) | ||||
|
r27976 | if s is not None: | ||
|
r43347 | extra[b'intermediate-source'] = s | ||
|
r27976 | |||
|
r43346 | |||
|
r27976 | def _savebranch(ctx, extra): | ||
|
r43347 | extra[b'branch'] = ctx.branch() | ||
|
r27976 | |||
|
r43346 | |||
|
r29043 | def _destrebase(repo, sourceset, destspace=None): | ||
|
r28189 | """small wrapper around destmerge to pass the right extra args | ||
Please wrap destutil.destmerge instead.""" | ||||
|
r43346 | return destutil.destmerge( | ||
repo, | ||||
|
r43347 | action=b'rebase', | ||
|
r43346 | sourceset=sourceset, | ||
onheadcheck=False, | ||||
destspace=destspace, | ||||
) | ||||
|
r26717 | |||
|
r28394 | revsetpredicate = registrar.revsetpredicate() | ||
|
r27586 | |||
|
r43346 | |||
|
r43347 | @revsetpredicate(b'_destrebase') | ||
|
r26719 | def _revsetdestrebase(repo, subset, x): | ||
|
r26301 | # ``_rebasedefaultdest()`` | ||
# default destination for rebase. | ||||
# # XXX: Currently private because I expect the signature to change. | ||||
# # XXX: - bailing out in case of ambiguity vs returning all data. | ||||
# i18n: "_rebasedefaultdest" is a keyword | ||||
|
r28189 | sourceset = None | ||
if x is not None: | ||||
|
r31023 | sourceset = revset.getset(repo, smartset.fullreposet(repo), x) | ||
return subset & smartset.baseset([_destrebase(repo, sourceset)]) | ||||
|
r26301 | |||
|
r43346 | |||
|
r43347 | @revsetpredicate(b'_destautoorphanrebase') | ||
|
r37805 | def _revsetdestautoorphanrebase(repo, subset, x): | ||
|
r42485 | # ``_destautoorphanrebase()`` | ||
# automatic rebase destination for a single orphan revision. | ||||
|
r37805 | unfi = repo.unfiltered() | ||
|
r43347 | obsoleted = unfi.revs(b'obsolete()') | ||
|
r37805 | |||
src = revset.getset(repo, subset, x).first() | ||||
# Empty src or already obsoleted - Do not return a destination | ||||
if not src or src in obsoleted: | ||||
return smartset.baseset() | ||||
dests = destutil.orphanpossibledestination(repo, src) | ||||
if len(dests) > 1: | ||||
|
r47377 | raise error.StateError( | ||
|
r43347 | _(b"ambiguous automatic rebase: %r could end up on any of %r") | ||
|
r43346 | % (src, dests) | ||
) | ||||
|
r37805 | # We have zero or one destination, so we can just return here. | ||
return smartset.baseset(dests) | ||||
|
r43346 | |||
|
r33840 | def _ctxdesc(ctx): | ||
"""short description for a context""" | ||||
|
r46355 | return cmdutil.format_changeset_summary( | ||
|
r46356 | ctx.repo().ui, ctx, command=b'rebase' | ||
|
r43347 | ) | ||
|
r33840 | |||
|
r43346 | |||
|
r29358 | class rebaseruntime(object): | ||
"""This class is a container for rebase runtime state""" | ||||
|
r43346 | |||
|
r46121 | def __init__(self, repo, ui, inmemory=False, dryrun=False, opts=None): | ||
|
r29399 | if opts is None: | ||
opts = {} | ||||
|
r34096 | # prepared: whether we have rebasestate prepared or not. Currently it | ||
# decides whether "self.repo" is unfiltered or not. | ||||
# The rebasestate has explicit hash to hash instructions not depending | ||||
# on visibility. If rebasestate exists (in-memory or on-disk), use | ||||
# unfiltered repo to avoid visibility issues. | ||||
# Before knowing rebasestate (i.e. when starting a new rebase (not | ||||
# --continue or --abort)), the original repo should be used so | ||||
# visibility-dependent revsets are correct. | ||||
self.prepared = False | ||||
|
r44820 | self.resume = False | ||
|
r34096 | self._repo = repo | ||
|
r29399 | self.ui = ui | ||
self.opts = opts | ||||
|
r29358 | self.originalwd = None | ||
self.external = nullrev | ||||
# Mapping between the old revision id and either what is the new rebased | ||||
# revision or what needs to be done with the old revision. The state | ||||
# dict will be what contains most of the rebase progress state. | ||||
self.state = {} | ||||
self.activebookmark = None | ||||
|
r34006 | self.destmap = {} | ||
|
r29360 | self.skipped = set() | ||
|
r29358 | |||
|
r48253 | self.collapsef = opts.get('collapse', False) | ||
self.collapsemsg = cmdutil.logmessage(ui, pycompat.byteskwargs(opts)) | ||||
self.date = opts.get('date', None) | ||||
|
r29401 | |||
|
r48253 | e = opts.get('extrafn') # internal, used by e.g. hgsubversion | ||
|
r29401 | self.extrafns = [_savegraft] | ||
if e: | ||||
self.extrafns = [e] | ||||
|
r29400 | |||
|
r43347 | self.backupf = ui.configbool(b'rewrite', b'backup-bundle') | ||
|
r48253 | self.keepf = opts.get('keep', False) | ||
self.keepbranchesf = opts.get('keepbranches', False) | ||||
|
r45683 | self.skipemptysuccessorf = rewriteutil.skip_empty_successor( | ||
repo.ui, b'rebase' | ||||
) | ||||
|
r47589 | self.obsolete_with_successor_in_destination = {} | ||
self.obsolete_with_successor_in_rebase_set = set() | ||||
|
r35389 | self.inmemory = inmemory | ||
|
r46121 | self.dryrun = dryrun | ||
|
r43347 | self.stateobj = statemod.cmdstate(repo, b'rebasestate') | ||
|
r29402 | |||
|
r34096 | @property | ||
def repo(self): | ||||
if self.prepared: | ||||
return self._repo.unfiltered() | ||||
else: | ||||
return self._repo | ||||
|
r31224 | def storestatus(self, tr=None): | ||
|
r31223 | """Store the current status to allow recovery""" | ||
|
r31224 | if tr: | ||
|
r43346 | tr.addfilegenerator( | ||
|
r43347 | b'rebasestate', | ||
(b'rebasestate',), | ||||
|
r43346 | self._writestatus, | ||
|
r43347 | location=b'plain', | ||
|
r43346 | ) | ||
|
r31224 | else: | ||
|
r43347 | with self.repo.vfs(b"rebasestate", b"w") as f: | ||
|
r31224 | self._writestatus(f) | ||
def _writestatus(self, f): | ||||
|
r34097 | repo = self.repo | ||
assert repo.filtername is None | ||||
|
r43347 | f.write(repo[self.originalwd].hex() + b'\n') | ||
|
r34006 | # was "dest". we now write dest per src root below. | ||
|
r43347 | f.write(b'\n') | ||
f.write(repo[self.external].hex() + b'\n') | ||||
f.write(b'%d\n' % int(self.collapsef)) | ||||
f.write(b'%d\n' % int(self.keepf)) | ||||
f.write(b'%d\n' % int(self.keepbranchesf)) | ||||
f.write(b'%s\n' % (self.activebookmark or b'')) | ||||
|
r34006 | destmap = self.destmap | ||
|
r43375 | for d, v in pycompat.iteritems(self.state): | ||
|
r31223 | oldrev = repo[d].hex() | ||
if v >= 0: | ||||
newrev = repo[v].hex() | ||||
else: | ||||
|
r43347 | newrev = b"%d" % v | ||
|
r34006 | destnode = repo[destmap[d]].hex() | ||
|
r43347 | f.write(b"%s:%s:%s\n" % (oldrev, newrev, destnode)) | ||
repo.ui.debug(b'rebase status stored\n') | ||||
|
r31223 | |||
|
r29403 | def restorestatus(self): | ||
"""Restore a previously stored status""" | ||||
|
r38537 | if not self.stateobj.exists(): | ||
|
r43347 | cmdutil.wrongtooltocontinue(self.repo, _(b'rebase')) | ||
|
r38537 | |||
|
r38534 | data = self._read() | ||
|
r43347 | self.repo.ui.debug(b'rebase status resumed\n') | ||
|
r38534 | |||
|
r43347 | self.originalwd = data[b'originalwd'] | ||
self.destmap = data[b'destmap'] | ||||
self.state = data[b'state'] | ||||
self.skipped = data[b'skipped'] | ||||
self.collapsef = data[b'collapse'] | ||||
self.keepf = data[b'keep'] | ||||
self.keepbranchesf = data[b'keepbranches'] | ||||
self.external = data[b'external'] | ||||
self.activebookmark = data[b'activebookmark'] | ||||
|
r38534 | |||
def _read(self): | ||||
|
r34096 | self.prepared = True | ||
|
r29403 | repo = self.repo | ||
|
r34097 | assert repo.filtername is None | ||
|
r43346 | data = { | ||
|
r43347 | b'keepbranches': None, | ||
b'collapse': None, | ||||
b'activebookmark': None, | ||||
b'external': nullrev, | ||||
b'keep': None, | ||||
b'originalwd': None, | ||||
|
r43346 | } | ||
|
r34006 | legacydest = None | ||
|
r29403 | state = {} | ||
|
r34006 | destmap = {} | ||
|
r29403 | |||
|
r38537 | if True: | ||
|
r43347 | f = repo.vfs(b"rebasestate") | ||
|
r29403 | for i, l in enumerate(f.read().splitlines()): | ||
if i == 0: | ||||
|
r43347 | data[b'originalwd'] = repo[l].rev() | ||
|
r29403 | elif i == 1: | ||
|
r34006 | # this line should be empty in newer version. but legacy | ||
# clients may still use it | ||||
if l: | ||||
legacydest = repo[l].rev() | ||||
|
r29403 | elif i == 2: | ||
|
r43347 | data[b'external'] = repo[l].rev() | ||
|
r29403 | elif i == 3: | ||
|
r43347 | data[b'collapse'] = bool(int(l)) | ||
|
r29403 | elif i == 4: | ||
|
r43347 | data[b'keep'] = bool(int(l)) | ||
|
r29403 | elif i == 5: | ||
|
r43347 | data[b'keepbranches'] = bool(int(l)) | ||
elif i == 6 and not (len(l) == 81 and b':' in l): | ||||
|
r29403 | # line 6 is a recent addition, so for backwards | ||
# compatibility check that the line doesn't look like the | ||||
# oldrev:newrev lines | ||||
|
r43347 | data[b'activebookmark'] = l | ||
|
r29403 | else: | ||
|
r43347 | args = l.split(b':') | ||
|
r37395 | oldrev = repo[args[0]].rev() | ||
|
r34006 | newrev = args[1] | ||
|
r33842 | if newrev in legacystates: | ||
continue | ||||
|
r34006 | if len(args) > 2: | ||
|
r37394 | destrev = repo[args[2]].rev() | ||
|
r34006 | else: | ||
|
r37394 | destrev = legacydest | ||
|
r37395 | destmap[oldrev] = destrev | ||
|
r37396 | if newrev == revtodostr: | ||
|
r37395 | state[oldrev] = revtodo | ||
|
r29403 | # Legacy compat special case | ||
else: | ||||
|
r37395 | state[oldrev] = repo[newrev].rev() | ||
|
r29403 | |||
|
r43347 | if data[b'keepbranches'] is None: | ||
raise error.Abort(_(b'.hg/rebasestate is incomplete')) | ||||
|
r29403 | |||
|
r43347 | data[b'destmap'] = destmap | ||
data[b'state'] = state | ||||
|
r29403 | skipped = set() | ||
# recompute the set of skipped revs | ||||
|
r43347 | if not data[b'collapse']: | ||
|
r34006 | seen = set(destmap.values()) | ||
|
r29403 | for old, new in sorted(state.items()): | ||
if new != revtodo and new in seen: | ||||
skipped.add(old) | ||||
seen.add(new) | ||||
|
r43347 | data[b'skipped'] = skipped | ||
|
r43346 | repo.ui.debug( | ||
|
r43347 | b'computed skipped revs: %s\n' | ||
% (b' '.join(b'%d' % r for r in sorted(skipped)) or b'') | ||||
|
r43346 | ) | ||
|
r29403 | |||
|
r38534 | return data | ||
|
r29403 | |||
|
r47587 | def _handleskippingobsolete(self): | ||
"""Compute structures necessary for skipping obsolete revisions""" | ||||
|
r47622 | if self.keepf: | ||
return | ||||
|
r43347 | if not self.ui.configbool(b'experimental', b'rebaseskipobsolete'): | ||
|
r29479 | return | ||
|
r47587 | obsoleteset = {r for r in self.state if self.repo[r].obsolete()} | ||
|
r43346 | ( | ||
|
r47589 | self.obsolete_with_successor_in_destination, | ||
self.obsolete_with_successor_in_rebase_set, | ||||
) = _compute_obsolete_sets(self.repo, obsoleteset, self.destmap) | ||||
skippedset = set(self.obsolete_with_successor_in_destination) | ||||
skippedset.update(self.obsolete_with_successor_in_rebase_set) | ||||
|
r33845 | _checkobsrebase(self.repo, self.ui, obsoleteset, skippedset) | ||
r48577 | if obsolete.isenabled(self.repo, obsolete.allowdivergenceopt): | |||
|
r47590 | self.obsolete_with_successor_in_rebase_set = set() | ||
else: | ||||
for rev in self.repo.revs( | ||||
b'descendants(%ld) and not %ld', | ||||
self.obsolete_with_successor_in_rebase_set, | ||||
self.obsolete_with_successor_in_rebase_set, | ||||
): | ||||
self.state.pop(rev, None) | ||||
self.destmap.pop(rev, None) | ||||
|
r29479 | |||
|
r45458 | def _prepareabortorcontinue( | ||
self, isabort, backup=True, suppwarns=False, dryrun=False, confirm=False | ||||
): | ||||
|
r44820 | self.resume = True | ||
|
r29472 | try: | ||
self.restorestatus() | ||||
|
r47589 | # Calculate self.obsolete_* sets | ||
|
r47588 | self._handleskippingobsolete() | ||
|
r31225 | self.collapsemsg = restorecollapsemsg(self.repo, isabort) | ||
|
r29472 | except error.RepoLookupError: | ||
if isabort: | ||||
clearstatus(self.repo) | ||||
clearcollapsemsg(self.repo) | ||||
|
r43346 | self.repo.ui.warn( | ||
_( | ||||
|
r43347 | b'rebase aborted (no revision is removed,' | ||
b' only broken state is cleared)\n' | ||||
|
r43346 | ) | ||
) | ||||
|
r29472 | return 0 | ||
else: | ||||
|
r43347 | msg = _(b'cannot continue inconsistent rebase') | ||
hint = _(b'use "hg rebase --abort" to clear broken state') | ||||
|
r29472 | raise error.Abort(msg, hint=hint) | ||
|
r38852 | |||
|
r29472 | if isabort: | ||
|
r38852 | backup = backup and self.backupf | ||
|
r45458 | return self._abort( | ||
backup=backup, | ||||
suppwarns=suppwarns, | ||||
dryrun=dryrun, | ||||
confirm=confirm, | ||||
) | ||||
|
r29472 | |||
|
r34006 | def _preparenewrebase(self, destmap): | ||
if not destmap: | ||||
|
r29473 | return _nothingtorebase() | ||
|
r34010 | result = buildstate(self.repo, destmap, self.collapsef) | ||
|
r29473 | |||
if not result: | ||||
# Empty state built, nothing to rebase | ||||
|
r43347 | self.ui.status(_(b'nothing to rebase\n')) | ||
|
r29473 | return _nothingtorebase() | ||
|
r34006 | (self.originalwd, self.destmap, self.state) = result | ||
|
r29473 | if self.collapsef: | ||
|
r34006 | dests = set(self.destmap.values()) | ||
if len(dests) != 1: | ||||
|
r47377 | raise error.InputError( | ||
|
r43347 | _(b'--collapse does not work with multiple destinations') | ||
|
r43346 | ) | ||
|
r34006 | destrev = next(iter(dests)) | ||
|
r43346 | destancestors = self.repo.changelog.ancestors( | ||
[destrev], inclusive=True | ||||
) | ||||
|
r33846 | self.external = externalparent(self.repo, self.state, destancestors) | ||
|
r29473 | |||
|
r34006 | for destrev in sorted(set(destmap.values())): | ||
dest = self.repo[destrev] | ||||
if dest.closesbranch() and not self.keepbranchesf: | ||||
|
r43347 | self.ui.status(_(b'reopening closed branch head %s\n') % dest) | ||
|
r29473 | |||
|
r47589 | # Calculate self.obsolete_* sets | ||
|
r47588 | self._handleskippingobsolete() | ||
|
r47592 | if not self.keepf: | ||
|
r47593 | rebaseset = set(destmap.keys()) | ||
rebaseset -= set(self.obsolete_with_successor_in_destination) | ||||
rebaseset -= self.obsolete_with_successor_in_rebase_set | ||||
|
r47784 | # We have our own divergence-checking in the rebase extension | ||
overrides = {} | ||||
if obsolete.isenabled(self.repo, obsolete.createmarkersopt): | ||||
overrides = { | ||||
(b'experimental', b'evolution.allowdivergence'): b'true' | ||||
} | ||||
|
r47592 | try: | ||
|
r47784 | with self.ui.configoverride(overrides): | ||
rewriteutil.precheck(self.repo, rebaseset, action=b'rebase') | ||||
|
r47592 | except error.Abort as e: | ||
if e.hint is None: | ||||
e.hint = _(b'use --keep to keep original changesets') | ||||
raise e | ||||
|
r47591 | self.prepared = True | ||
|
r35334 | def _assignworkingcopy(self): | ||
|
r35291 | if self.inmemory: | ||
from mercurial.context import overlayworkingctx | ||||
|
r43346 | |||
|
r35291 | self.wctx = overlayworkingctx(self.repo) | ||
|
r46082 | self.repo.ui.debug(b"rebasing in memory\n") | ||
|
r35291 | else: | ||
self.wctx = self.repo[None] | ||||
|
r43347 | self.repo.ui.debug(b"rebasing on disk\n") | ||
|
r43346 | self.repo.ui.log( | ||
|
r43347 | b"rebase", | ||
b"using in-memory rebase: %r\n", | ||||
|
r43346 | self.inmemory, | ||
rebase_imm_used=self.inmemory, | ||||
) | ||||
|
r35334 | |||
def _performrebase(self, tr): | ||||
self._assignworkingcopy() | ||||
repo, ui = self.repo, self.ui | ||||
|
r29477 | if self.keepbranchesf: | ||
# insert _savebranch at the start of extrafns so if | ||||
# there's a user-provided extrafn it can clobber branch if | ||||
# desired | ||||
self.extrafns.insert(0, _savebranch) | ||||
if self.collapsef: | ||||
branches = set() | ||||
for rev in self.state: | ||||
branches.add(repo[rev].branch()) | ||||
if len(branches) > 1: | ||||
|
r47377 | raise error.InputError( | ||
|
r43387 | _(b'cannot collapse multiple named branches') | ||
|
r43346 | ) | ||
|
r29477 | |||
|
r33332 | # Keep track of the active bookmarks in order to reset them later | ||
|
r29477 | self.activebookmark = self.activebookmark or repo._activebookmark | ||
if self.activebookmark: | ||||
bookmarks.deactivate(repo) | ||||
|
r31225 | # Store the state before we begin so users can run 'hg rebase --abort' | ||
# if we fail before the transaction closes. | ||||
self.storestatus() | ||||
|
r37049 | if tr: | ||
# When using single transaction, store state when transaction | ||||
# commits. | ||||
self.storestatus(tr) | ||||
|
r31225 | |||
|
r43375 | cands = [k for k, v in pycompat.iteritems(self.state) if v == revtodo] | ||
|
r43346 | p = repo.ui.makeprogress( | ||
|
r43347 | _(b"rebasing"), unit=_(b'changesets'), total=len(cands) | ||
|
r43346 | ) | ||
|
r36951 | def progress(ctx): | ||
|
r43347 | p.increment(item=(b"%d:%s" % (ctx.rev(), ctx))) | ||
|
r43346 | |||
|
r34008 | for subset in sortsource(self.destmap): | ||
|
r43347 | sortedrevs = self.repo.revs(b'sort(%ld, -topo)', subset) | ||
|
r36950 | for rev in sortedrevs: | ||
|
r47590 | self._rebasenode(tr, rev, progress) | ||
|
r38396 | p.complete() | ||
|
r43347 | ui.note(_(b'rebase merging completed\n')) | ||
|
r34008 | |||
|
r46083 | def _concludenode(self, rev, editor, commitmsg=None): | ||
|
r46554 | """Commit the wd changes with parents p1 and p2. | ||
|
r37051 | |||
Reuse commit info from rev but also store useful information in extra. | ||||
|
r46554 | Return node of committed revision.""" | ||
|
r37051 | repo = self.repo | ||
|
r37052 | ctx = repo[rev] | ||
|
r37053 | if commitmsg is None: | ||
commitmsg = ctx.description() | ||||
|
r46377 | |||
# Skip replacement if collapsing, as that degenerates to p1 for all | ||||
# nodes. | ||||
if not self.collapsef: | ||||
cl = repo.changelog | ||||
commitmsg = rewriteutil.update_hash_refs( | ||||
repo, | ||||
commitmsg, | ||||
{ | ||||
cl.node(oldrev): [cl.node(newrev)] | ||||
for oldrev, newrev in self.state.items() | ||||
if newrev != revtodo | ||||
}, | ||||
) | ||||
|
r37059 | date = self.date | ||
if date is None: | ||||
date = ctx.date() | ||||
|
r43347 | extra = {b'rebase_source': ctx.hex()} | ||
|
r37055 | for c in self.extrafns: | ||
c(ctx, extra) | ||||
|
r37057 | destphase = max(ctx.phase(), phases.draft) | ||
|
r45683 | overrides = { | ||
(b'phases', b'new-commit'): destphase, | ||||
(b'ui', b'allowemptycommit'): not self.skipemptysuccessorf, | ||||
} | ||||
|
r43347 | with repo.ui.configoverride(overrides, b'rebase'): | ||
|
r37058 | if self.inmemory: | ||
|
r43346 | newnode = commitmemorynode( | ||
repo, | ||||
|
r37058 | wctx=self.wctx, | ||
extra=extra, | ||||
commitmsg=commitmsg, | ||||
editor=editor, | ||||
|
r37060 | user=ctx.user(), | ||
|
r43346 | date=date, | ||
) | ||||
|
r37058 | else: | ||
|
r43346 | newnode = commitnode( | ||
repo, | ||||
|
r37058 | extra=extra, | ||
commitmsg=commitmsg, | ||||
editor=editor, | ||||
|
r37060 | user=ctx.user(), | ||
|
r43346 | date=date, | ||
) | ||||
|
r37051 | |||
|
r37058 | return newnode | ||
|
r37051 | |||
|
r47590 | def _rebasenode(self, tr, rev, progressfn): | ||
|
r34008 | repo, ui, opts = self.repo, self.ui, self.opts | ||
|
r36949 | ctx = repo[rev] | ||
desc = _ctxdesc(ctx) | ||||
if self.state[rev] == rev: | ||||
|
r43347 | ui.status(_(b'already rebased %s\n') % desc) | ||
|
r47590 | elif rev in self.obsolete_with_successor_in_rebase_set: | ||
|
r43346 | msg = ( | ||
_( | ||||
|
r43347 | b'note: not rebasing %s and its descendants as ' | ||
b'this would cause divergence\n' | ||||
|
r43346 | ) | ||
% desc | ||||
) | ||||
|
r36949 | repo.ui.status(msg) | ||
self.skipped.add(rev) | ||||
|
r47589 | elif rev in self.obsolete_with_successor_in_destination: | ||
succ = self.obsolete_with_successor_in_destination[rev] | ||||
|
r36949 | if succ is None: | ||
|
r43387 | msg = _(b'note: not rebasing %s, it has no successor\n') % desc | ||
|
r36949 | else: | ||
succdesc = _ctxdesc(repo[succ]) | ||||
|
r43346 | msg = _( | ||
|
r43387 | b'note: not rebasing %s, already in destination as %s\n' | ||
|
r43346 | ) % (desc, succdesc) | ||
|
r36949 | repo.ui.status(msg) | ||
# Make clearrebased aware state[rev] is not a true successor | ||||
self.skipped.add(rev) | ||||
# Record rev as moved to its desired destination in self.state. | ||||
# This helps bookmark and working parent movement. | ||||
|
r43346 | dest = max( | ||
adjustdest(repo, rev, self.destmap, self.state, self.skipped) | ||||
) | ||||
|
r36949 | self.state[rev] = dest | ||
elif self.state[rev] == revtodo: | ||||
|
r43347 | ui.status(_(b'rebasing %s\n') % desc) | ||
|
r36949 | progressfn(ctx) | ||
|
r43346 | p1, p2, base = defineparents( | ||
repo, | ||||
rev, | ||||
self.destmap, | ||||
self.state, | ||||
self.skipped, | ||||
|
r47589 | self.obsolete_with_successor_in_destination, | ||
|
r43346 | ) | ||
|
r44820 | if self.resume and self.wctx.p1().rev() == p1: | ||
|
r43347 | repo.ui.debug(b'resuming interrupted rebase\n') | ||
|
r44820 | self.resume = False | ||
|
r36949 | else: | ||
|
r48253 | overrides = {(b'ui', b'forcemerge'): opts.get('tool', b'')} | ||
|
r43347 | with ui.configoverride(overrides, b'rebase'): | ||
|
r46132 | try: | ||
rebasenode( | ||||
repo, | ||||
rev, | ||||
p1, | ||||
p2, | ||||
base, | ||||
self.collapsef, | ||||
wctx=self.wctx, | ||||
) | ||||
except error.InMemoryMergeConflictsError: | ||||
if self.dryrun: | ||||
raise error.ConflictResolutionRequired(b'rebase') | ||||
if self.collapsef: | ||||
# TODO: Make the overlayworkingctx reflected | ||||
# in the working copy here instead of re-raising | ||||
# so the entire rebase operation is retried. | ||||
raise | ||||
ui.status( | ||||
_( | ||||
b"hit merge conflicts; rebasing that " | ||||
b"commit again in the working copy\n" | ||||
) | ||||
) | ||||
|
r46641 | try: | ||
cmdutil.bailifchanged(repo) | ||||
except error.Abort: | ||||
clearstatus(repo) | ||||
clearcollapsemsg(repo) | ||||
raise | ||||
|
r46132 | self.inmemory = False | ||
self._assignworkingcopy() | ||||
|
r46150 | mergemod.update(repo[p1], wc=self.wctx) | ||
|
r46132 | rebasenode( | ||
repo, | ||||
rev, | ||||
p1, | ||||
p2, | ||||
base, | ||||
self.collapsef, | ||||
wctx=self.wctx, | ||||
) | ||||
|
r36949 | if not self.collapsef: | ||
merging = p2 != nullrev | ||||
|
r43347 | editform = cmdutil.mergeeditform(merging, b'rebase') | ||
|
r48253 | editor = cmdutil.getcommiteditor(editform=editform, **opts) | ||
|
r45157 | # We need to set parents again here just in case we're continuing | ||
# a rebase started with an old hg version (before 9c9cfecd4600), | ||||
# because those old versions would have left us with two dirstate | ||||
# parents, and we don't want to create a merge commit here (unless | ||||
# we're rebasing a merge commit). | ||||
self.wctx.setparents(repo[p1].node(), repo[p2].node()) | ||||
|
r46083 | newnode = self._concludenode(rev, editor) | ||
|
r36949 | else: | ||
# Skip commit if we are collapsing | ||||
newnode = None | ||||
# Update the state | ||||
if newnode is not None: | ||||
self.state[rev] = repo[newnode].rev() | ||||
|
r43347 | ui.debug(b'rebased as %s\n' % short(newnode)) | ||
|
r45683 | if repo[newnode].isempty(): | ||
ui.warn( | ||||
_( | ||||
b'note: created empty successor for %s, its ' | ||||
b'destination already has all its changes\n' | ||||
) | ||||
% desc | ||||
) | ||||
|
r36949 | else: | ||
|
r29477 | if not self.collapsef: | ||
|
r43346 | ui.warn( | ||
_( | ||||
|
r43347 | b'note: not rebasing %s, its destination already ' | ||
b'has all its changes\n' | ||||
|
r43346 | ) | ||
% desc | ||||
) | ||||
|
r36949 | self.skipped.add(rev) | ||
self.state[rev] = p1 | ||||
|
r43347 | ui.debug(b'next revision set to %d\n' % p1) | ||
|
r36949 | else: | ||
|
r43346 | ui.status( | ||
|
r43347 | _(b'already rebased %s as %s\n') % (desc, repo[self.state[rev]]) | ||
|
r43346 | ) | ||
|
r37050 | if not tr: | ||
# When not using single transaction, store state after each | ||||
# commit is completely done. On InterventionRequired, we thus | ||||
# won't store the status. Instead, we'll hit the "len(parents) == 2" | ||||
# case and realize that the commit was in progress. | ||||
self.storestatus() | ||||
|
r29477 | |||
|
r38852 | def _finishrebase(self): | ||
|
r29478 | repo, ui, opts = self.repo, self.ui, self.opts | ||
|
r48253 | fm = ui.formatter(b'rebase', pycompat.byteskwargs(opts)) | ||
|
r34884 | fm.startitem() | ||
|
r36789 | if self.collapsef: | ||
|
r43346 | p1, p2, _base = defineparents( | ||
repo, | ||||
min(self.state), | ||||
self.destmap, | ||||
self.state, | ||||
self.skipped, | ||||
|
r47589 | self.obsolete_with_successor_in_destination, | ||
|
r43346 | ) | ||
|
r48253 | editopt = opts.get('edit') | ||
|
r43347 | editform = b'rebase.collapse' | ||
|
r29478 | if self.collapsemsg: | ||
commitmsg = self.collapsemsg | ||||
else: | ||||
|
r43347 | commitmsg = b'Collapsed revision' | ||
|
r33624 | for rebased in sorted(self.state): | ||
|
r33847 | if rebased not in self.skipped: | ||
|
r43347 | commitmsg += b'\n* %s' % repo[rebased].description() | ||
|
r29478 | editopt = True | ||
editor = cmdutil.getcommiteditor(edit=editopt, editform=editform) | ||||
|
r29552 | revtoreuse = max(self.state) | ||
|
r33621 | |||
|
r44822 | self.wctx.setparents(repo[p1].node(), repo[self.external].node()) | ||
|
r43346 | newnode = self._concludenode( | ||
|
r46083 | revtoreuse, editor, commitmsg=commitmsg | ||
|
r43346 | ) | ||
|
r36946 | |||
|
r33864 | if newnode is not None: | ||
|
r29478 | newrev = repo[newnode].rev() | ||
|
r36313 | for oldrev in self.state: | ||
|
r29478 | self.state[oldrev] = newrev | ||
|
r43347 | if b'qtip' in repo.tags(): | ||
|
r48253 | updatemq(repo, self.state, self.skipped, **opts) | ||
|
r29478 | |||
# restore original working directory | ||||
# (we do this before stripping) | ||||
newwd = self.state.get(self.originalwd, self.originalwd) | ||||
|
r33842 | if newwd < 0: | ||
|
r29478 | # original directory is a parent of rebase set root or ignored | ||
newwd = self.originalwd | ||||
|
r36993 | if newwd not in [c.rev() for c in repo[None].parents()]: | ||
|
r43347 | ui.note(_(b"update back to initial working directory parent\n")) | ||
|
r46151 | mergemod.update(repo[newwd]) | ||
|
r29478 | |||
|
r34364 | collapsedas = None | ||
|
r36792 | if self.collapsef and not self.keepf: | ||
collapsedas = newnode | ||||
|
r43346 | clearrebased( | ||
ui, | ||||
repo, | ||||
self.destmap, | ||||
self.state, | ||||
self.skipped, | ||||
collapsedas, | ||||
self.keepf, | ||||
fm=fm, | ||||
backup=self.backupf, | ||||
) | ||||
|
r29478 | |||
clearstatus(repo) | ||||
clearcollapsemsg(repo) | ||||
|
r43347 | ui.note(_(b"rebase completed\n")) | ||
util.unlinkpath(repo.sjoin(b'undo'), ignoremissing=True) | ||||
|
r29478 | if self.skipped: | ||
skippedlen = len(self.skipped) | ||||
|
r43347 | ui.note(_(b"%d revisions have been skipped\n") % skippedlen) | ||
|
r34884 | fm.end() | ||
|
r29478 | |||
|
r43346 | if ( | ||
self.activebookmark | ||||
and self.activebookmark in repo._bookmarks | ||||
|
r43347 | and repo[b'.'].node() == repo._bookmarks[self.activebookmark] | ||
|
r43346 | ): | ||
bookmarks.activate(repo, self.activebookmark) | ||||
|
r29478 | |||
|
r45458 | def _abort(self, backup=True, suppwarns=False, dryrun=False, confirm=False): | ||
|
r40892 | '''Restore the repository to its original state.''' | ||
|
r40891 | |||
|
r40892 | repo = self.repo | ||
|
r40891 | try: | ||
# If the first commits in the rebased set get skipped during the | ||||
# rebase, their values within the state mapping will be the dest | ||||
# rev id. The rebased list must must not contain the dest rev | ||||
# (issue4896) | ||||
|
r43346 | rebased = [ | ||
s | ||||
for r, s in self.state.items() | ||||
if s >= 0 and s != r and s != self.destmap[r] | ||||
] | ||||
|
r40891 | immutable = [d for d in rebased if not repo[d].mutable()] | ||
cleanup = True | ||||
if immutable: | ||||
|
r43346 | repo.ui.warn( | ||
|
r43347 | _(b"warning: can't clean up public changesets %s\n") | ||
% b', '.join(bytes(repo[r]) for r in immutable), | ||||
hint=_(b"see 'hg help phases' for details"), | ||||
|
r43346 | ) | ||
|
r40891 | cleanup = False | ||
descendants = set() | ||||
if rebased: | ||||
descendants = set(repo.changelog.descendants(rebased)) | ||||
if descendants - set(rebased): | ||||
|
r43346 | repo.ui.warn( | ||
_( | ||||
|
r43347 | b"warning: new changesets detected on " | ||
b"destination branch, can't strip\n" | ||||
|
r43346 | ) | ||
) | ||||
|
r40891 | cleanup = False | ||
if cleanup: | ||||
if rebased: | ||||
strippoints = [ | ||||
|
r43347 | c.node() for c in repo.set(b'roots(%ld)', rebased) | ||
|
r43346 | ] | ||
|
r40891 | |||
updateifonnodes = set(rebased) | ||||
|
r40892 | updateifonnodes.update(self.destmap.values()) | ||
|
r45458 | |||
if not dryrun and not confirm: | ||||
updateifonnodes.add(self.originalwd) | ||||
|
r43347 | shouldupdate = repo[b'.'].rev() in updateifonnodes | ||
|
r40891 | |||
# Update away from the rebase if necessary | ||||
|
r44505 | if shouldupdate: | ||
|
r44743 | mergemod.clean_update(repo[self.originalwd]) | ||
|
r40891 | |||
# Strip from the first rebased revision | ||||
if rebased: | ||||
repair.strip(repo.ui, repo, strippoints, backup=backup) | ||||
|
r40892 | if self.activebookmark and self.activebookmark in repo._bookmarks: | ||
bookmarks.activate(repo, self.activebookmark) | ||||
|
r40891 | |||
finally: | ||||
clearstatus(repo) | ||||
clearcollapsemsg(repo) | ||||
if not suppwarns: | ||||
|
r43347 | repo.ui.warn(_(b'rebase aborted\n')) | ||
|
r40891 | return 0 | ||
|
r43346 | |||
@command( | ||||
|
r43347 | b'rebase', | ||
|
r43346 | [ | ||
( | ||||
|
r43347 | b's', | ||
b'source', | ||||
|
r45044 | [], | ||
_(b'rebase the specified changesets and their descendants'), | ||||
|
r43347 | _(b'REV'), | ||
|
r43346 | ), | ||
( | ||||
|
r43347 | b'b', | ||
b'base', | ||||
|
r45045 | [], | ||
|
r43347 | _(b'rebase everything from branching point of specified changeset'), | ||
_(b'REV'), | ||||
|
r43346 | ), | ||
|
r43347 | (b'r', b'rev', [], _(b'rebase these revisions'), _(b'REV')), | ||
|
r43346 | ( | ||
|
r43347 | b'd', | ||
b'dest', | ||||
b'', | ||||
_(b'rebase onto the specified changeset'), | ||||
_(b'REV'), | ||||
|
r43346 | ), | ||
|
r43347 | (b'', b'collapse', False, _(b'collapse the rebased changesets')), | ||
|
r43346 | ( | ||
|
r43347 | b'm', | ||
b'message', | ||||
b'', | ||||
_(b'use text as collapse commit message'), | ||||
_(b'TEXT'), | ||||
), | ||||
(b'e', b'edit', False, _(b'invoke editor on commit messages')), | ||||
( | ||||
b'l', | ||||
b'logfile', | ||||
b'', | ||||
_(b'read collapse commit message from file'), | ||||
_(b'FILE'), | ||||
|
r43346 | ), | ||
|
r43347 | (b'k', b'keep', False, _(b'keep original changesets')), | ||
(b'', b'keepbranches', False, _(b'keep original branch names')), | ||||
(b'D', b'detach', False, _(b'(DEPRECATED)')), | ||||
(b'i', b'interactive', False, _(b'(DEPRECATED)')), | ||||
(b't', b'tool', b'', _(b'specify merge tool')), | ||||
(b'', b'stop', False, _(b'stop interrupted rebase')), | ||||
(b'c', b'continue', False, _(b'continue an interrupted rebase')), | ||||
(b'a', b'abort', False, _(b'abort an interrupted rebase')), | ||||
|
r43346 | ( | ||
|
r43347 | b'', | ||
b'auto-orphans', | ||||
b'', | ||||
|
r43346 | _( | ||
|
r43347 | b'automatically rebase orphan revisions ' | ||
b'in the specified revset (EXPERIMENTAL)' | ||||
|
r43346 | ), | ||
), | ||||
] | ||||
+ cmdutil.dryrunopts | ||||
+ cmdutil.formatteropts | ||||
+ cmdutil.confirmopts, | ||||
|
r45045 | _(b'[[-s REV]... | [-b REV]... | [-r REV]...] [-d REV] [OPTION]...'), | ||
|
r43346 | helpcategory=command.CATEGORY_CHANGE_MANAGEMENT, | ||
) | ||||
|
r6906 | def rebase(ui, repo, **opts): | ||
"""move changeset (and descendants) to a different branch | ||||
|
r7999 | Rebase uses repeated merging to graft changesets from one part of | ||
|
r10646 | history (the source) onto another (the destination). This can be | ||
|
r11188 | useful for linearizing *local* changes relative to a master | ||
|
r10646 | development tree. | ||
|
r27454 | Published commits cannot be rebased (see :hg:`help phases`). | ||
To copy commits, see :hg:`help graft`. | ||||
|
r18516 | |||
|
r28189 | If you don't specify a destination changeset (``-d/--dest``), rebase | ||
will use the same logic as :hg:`merge` to pick a destination. if | ||||
the current branch contains exactly one other head, the other head | ||||
is merged with by default. Otherwise, an explicit revision with | ||||
which to merge with must be provided. (destination changeset is not | ||||
modified by rebasing, but new changesets are added as its | ||||
descendants.) | ||||
|
r10646 | |||
|
r27956 | Here are the ways to select changesets: | ||
|
r27455 | |||
1. Explicitly select them using ``--rev``. | ||||
|
r10646 | |||
|
r27455 | 2. Use ``--source`` to select a root changeset and include all of its | ||
|
r27959 | descendants. | ||
|
r27455 | |||
3. Use ``--base`` to select a changeset; rebase will find ancestors | ||||
|
r27959 | and their descendants which are not also ancestors of the destination. | ||
|
r18518 | |||
|
r41972 | 4. If you do not specify any of ``--rev``, ``--source``, or ``--base``, | ||
|
r27959 | rebase will use ``--base .`` as above. | ||
|
r27932 | |||
|
r35288 | If ``--source`` or ``--rev`` is used, special names ``SRC`` and ``ALLSRC`` | ||
can be used in ``--dest``. Destination would be calculated per source | ||||
revision with ``SRC`` substituted by that single source revision and | ||||
``ALLSRC`` substituted by all source revisions. | ||||
|
r27456 | Rebase will destroy original changesets unless you use ``--keep``. | ||
It will also move your bookmarks (even if you do). | ||||
Some changesets may be dropped if they do not contribute changes | ||||
(e.g. merges from the destination branch). | ||||
|
r10646 | |||
|
r27457 | Unlike ``merge``, rebase will do nothing if you are at the branch tip of | ||
a named branch with two heads. You will need to explicitly specify source | ||||
and/or destination. | ||||
|
r6906 | |||
|
r28001 | If you need to use a tool to automate merge/conflict decisions, you | ||
can specify one with ``--tool``, see :hg:`help merge-tools`. | ||||
|
r28002 | As a caveat: the tool will not be used to mediate when a file was | ||
deleted, there is no hook presently available for this. | ||||
|
r28001 | |||
|
r27458 | If a rebase is interrupted to manually resolve a conflict, it can be | ||
|
r39130 | continued with --continue/-c, aborted with --abort/-a, or stopped with | ||
--stop. | ||||
|
r11205 | |||
|
r22790 | .. container:: verbose | ||
Examples: | ||||
- move "local changes" (current commit back to branching point) | ||||
to the current branch tip after a pull:: | ||||
hg rebase | ||||
- move a single changeset to the stable branch:: | ||||
hg rebase -r 5f493448 -d stable | ||||
- splice a commit and all its descendants onto another part of history:: | ||||
hg rebase --source c0c3 --dest 4cf9 | ||||
- rebase everything on a branch marked by a bookmark onto the | ||||
default branch:: | ||||
hg rebase --base myfeature --dest default | ||||
- collapse a sequence of changes into a single commit:: | ||||
hg rebase --collapse -r 1520:1525 -d . | ||||
- move a named branch while preserving its name:: | ||||
hg rebase -r "branch(featureX)" -d 1.3 --keepbranches | ||||
|
r35288 | - stabilize orphaned changesets so history looks linear:: | ||
hg rebase -r 'orphan()-obsolete()'\ | ||||
-d 'first(max((successors(max(roots(ALLSRC) & ::SRC)^)-obsolete())::) +\ | ||||
max(::((roots(ALLSRC) & ::SRC)^)-obsolete()))' | ||||
|
r31558 | Configuration Options: | ||
You can make rebase require a destination if you set the following config | ||||
|
r32085 | option:: | ||
|
r31558 | |||
[commands] | ||||
|
r32084 | rebase.requiredest = True | ||
|
r31558 | |||
|
r33569 | By default, rebase will close the transaction after each commit. For | ||
performance purposes, you can configure rebase to use a single transaction | ||||
across the entire rebase. WARNING: This setting introduces a significant | ||||
risk of losing the work you've done in a rebase if the rebase aborts | ||||
unexpectedly:: | ||||
[rebase] | ||||
singletransaction = True | ||||
|
r35389 | By default, rebase writes to the working copy, but you can configure it to | ||
|
r42640 | run in-memory for better performance. When the rebase is not moving the | ||
parent(s) of the working copy (AKA the "currently checked out changesets"), | ||||
this may also allow it to run even if the working copy is dirty:: | ||||
|
r35389 | |||
[rebase] | ||||
experimental.inmemory = True | ||||
|
r31558 | Return Values: | ||
|
r19971 | Returns 0 on success, 1 if nothing to rebase or there are | ||
unresolved conflicts. | ||||
|
r22790 | |||
|
r6906 | """ | ||
|
r43347 | inmemory = ui.configbool(b'rebase', b'experimental.inmemory') | ||
|
r48252 | action = cmdutil.check_at_most_one_arg(opts, 'abort', 'stop', 'continue') | ||
|
r44379 | if action: | ||
cmdutil.check_incompatible_arguments( | ||||
|
r48252 | opts, action, ['confirm', 'dry_run'] | ||
|
r44379 | ) | ||
|
r44383 | cmdutil.check_incompatible_arguments( | ||
|
r48252 | opts, action, ['rev', 'source', 'base', 'dest'] | ||
|
r44383 | ) | ||
|
r48252 | cmdutil.check_at_most_one_arg(opts, 'confirm', 'dry_run') | ||
cmdutil.check_at_most_one_arg(opts, 'rev', 'source', 'base') | ||||
|
r38391 | |||
|
r39137 | if action or repo.currenttransaction() is not None: | ||
|
r35321 | # in-memory rebase is not compatible with resuming rebases. | ||
|
r35719 | # (Or if it is run within a transaction, since the restart logic can | ||
# fail the entire transaction.) | ||||
|
r35389 | inmemory = False | ||
|
r35321 | |||
|
r48252 | if opts.get('auto_orphans'): | ||
disallowed_opts = set(opts) - {'auto_orphans'} | ||||
|
r44382 | cmdutil.check_incompatible_arguments( | ||
|
r48252 | opts, 'auto_orphans', disallowed_opts | ||
|
r44382 | ) | ||
|
r48252 | userrevs = list(repo.revs(opts.get('auto_orphans'))) | ||
opts['rev'] = [revsetlang.formatspec(b'%ld and orphan()', userrevs)] | ||||
opts['dest'] = b'_destautoorphanrebase(SRC)' | ||||
|
r37805 | |||
|
r48253 | if opts.get('dry_run') or opts.get('confirm'): | ||
|
r39136 | return _dryrunrebase(ui, repo, action, opts) | ||
|
r48252 | elif action == 'stop': | ||
|
r39128 | rbsrt = rebaseruntime(repo, ui) | ||
with repo.wlock(), repo.lock(): | ||||
|
r39152 | rbsrt.restorestatus() | ||
if rbsrt.collapsef: | ||||
|
r47377 | raise error.StateError(_(b"cannot stop in --collapse session")) | ||
|
r39152 | allowunstable = obsolete.isenabled(repo, obsolete.allowunstableopt) | ||
if not (rbsrt.keepf or allowunstable): | ||||
|
r47377 | raise error.StateError( | ||
|
r43346 | _( | ||
|
r43347 | b"cannot remove original changesets with" | ||
b" unrebased descendants" | ||||
|
r43346 | ), | ||
hint=_( | ||||
|
r43347 | b'either enable obsmarkers to allow unstable ' | ||
b'revisions or use --keep to keep original ' | ||||
b'changesets' | ||||
|
r43346 | ), | ||
) | ||||
|
r44505 | # update to the current working revision | ||
# to clear interrupted merge | ||||
|
r46133 | mergemod.clean_update(repo[rbsrt.originalwd]) | ||
|
r39128 | rbsrt._finishrebase() | ||
return 0 | ||||
|
r38391 | elif inmemory: | ||
|
r35321 | try: | ||
# in-memory merge doesn't support conflicts, so if we hit any, abort | ||||
# and re-run as an on-disk merge. | ||||
|
r43347 | overrides = {(b'rebase', b'singletransaction'): True} | ||
with ui.configoverride(overrides, b'rebase'): | ||||
|
r39136 | return _dorebase(ui, repo, action, opts, inmemory=inmemory) | ||
|
r35321 | except error.InMemoryMergeConflictsError: | ||
|
r47071 | if ui.configbool(b'devel', b'rebase.force-in-memory-merge'): | ||
raise | ||||
|
r43346 | ui.warn( | ||
_( | ||||
|
r43347 | b'hit merge conflicts; re-running rebase without in-memory' | ||
b' merge\n' | ||||
|
r43346 | ) | ||
) | ||||
|
r40835 | clearstatus(repo) | ||
clearcollapsemsg(repo) | ||||
|
r39136 | return _dorebase(ui, repo, action, opts, inmemory=False) | ||
|
r35321 | else: | ||
|
r39136 | return _dorebase(ui, repo, action, opts) | ||
|
r35321 | |||
|
r43346 | |||
|
r39136 | def _dryrunrebase(ui, repo, action, opts): | ||
|
r46121 | rbsrt = rebaseruntime(repo, ui, inmemory=True, dryrun=True, opts=opts) | ||
|
r48253 | confirm = opts.get('confirm') | ||
|
r38689 | if confirm: | ||
|
r43347 | ui.status(_(b'starting in-memory rebase\n')) | ||
|
r38689 | else: | ||
|
r43346 | ui.status( | ||
|
r43387 | _(b'starting dry-run rebase; repository will not be changed\n') | ||
|
r43346 | ) | ||
|
r38515 | with repo.wlock(), repo.lock(): | ||
|
r38689 | needsabort = True | ||
|
r38515 | try: | ||
|
r43347 | overrides = {(b'rebase', b'singletransaction'): True} | ||
with ui.configoverride(overrides, b'rebase'): | ||||
|
r46796 | res = _origrebase( | ||
|
r46554 | ui, | ||
repo, | ||||
action, | ||||
opts, | ||||
rbsrt, | ||||
|
r43346 | ) | ||
|
r46796 | if res == _nothingtorebase(): | ||
needsabort = False | ||||
return res | ||||
|
r46132 | except error.ConflictResolutionRequired: | ||
|
r43347 | ui.status(_(b'hit a merge conflict\n')) | ||
|
r38515 | return 1 | ||
|
r42277 | except error.Abort: | ||
needsabort = False | ||||
raise | ||||
|
r38515 | else: | ||
|
r38689 | if confirm: | ||
|
r43347 | ui.status(_(b'rebase completed successfully\n')) | ||
|
r43387 | if not ui.promptchoice(_(b'apply changes (yn)?$$ &Yes $$ &No')): | ||
|
r38689 | # finish unfinished rebase | ||
|
r38852 | rbsrt._finishrebase() | ||
|
r38689 | else: | ||
|
r43346 | rbsrt._prepareabortorcontinue( | ||
|
r45458 | isabort=True, | ||
backup=False, | ||||
suppwarns=True, | ||||
confirm=confirm, | ||||
|
r43346 | ) | ||
|
r38689 | needsabort = False | ||
else: | ||||
|
r43346 | ui.status( | ||
_( | ||||
|
r43347 | b'dry-run rebase completed successfully; run without' | ||
b' -n/--dry-run to perform this rebase\n' | ||||
|
r43346 | ) | ||
) | ||||
|
r38515 | return 0 | ||
finally: | ||||
|
r38689 | if needsabort: | ||
# no need to store backup in case of dryrun | ||||
|
r43346 | rbsrt._prepareabortorcontinue( | ||
|
r45458 | isabort=True, | ||
backup=False, | ||||
suppwarns=True, | ||||
|
r48253 | dryrun=opts.get('dry_run'), | ||
|
r43346 | ) | ||
|
r38515 | |||
|
r39136 | def _dorebase(ui, repo, action, opts, inmemory=False): | ||
|
r46121 | rbsrt = rebaseruntime(repo, ui, inmemory, opts=opts) | ||
|
r46119 | return _origrebase(ui, repo, action, opts, rbsrt) | ||
|
r38516 | |||
|
r43346 | |||
|
r46121 | def _origrebase(ui, repo, action, opts, rbsrt): | ||
|
r48252 | assert action != 'stop' | ||
|
r32917 | with repo.wlock(), repo.lock(): | ||
|
r48253 | if opts.get('interactive'): | ||
|
r26496 | try: | ||
|
r43347 | if extensions.find(b'histedit'): | ||
enablehistedit = b'' | ||||
|
r26496 | except KeyError: | ||
|
r43347 | enablehistedit = b" --config extensions.histedit=" | ||
help = b"hg%s help -e histedit" % enablehistedit | ||||
|
r43346 | msg = ( | ||
_( | ||||
|
r43347 | b"interactive history editing is supported by the " | ||
b"'histedit' extension (see \"%s\")" | ||||
|
r43346 | ) | ||
% help | ||||
) | ||||
|
r47377 | raise error.InputError(msg) | ||
|
r22382 | |||
|
r29400 | if rbsrt.collapsemsg and not rbsrt.collapsef: | ||
|
r47377 | raise error.InputError( | ||
_(b'message can only be specified with collapse') | ||||
) | ||||
|
r13661 | |||
|
r39136 | if action: | ||
|
r29400 | if rbsrt.collapsef: | ||
|
r47377 | raise error.InputError( | ||
|
r43347 | _(b'cannot use collapse with continue or abort') | ||
|
r43346 | ) | ||
|
r48253 | if action == 'abort' and opts.get('tool', False): | ||
|
r43347 | ui.warn(_(b'tool option will be ignored\n')) | ||
|
r48252 | if action == 'continue': | ||
|
r45383 | ms = mergestatemod.mergestate.read(repo) | ||
|
r30495 | mergeutil.checkunresolved(ms) | ||
|
r6906 | |||
|
r48252 | retcode = rbsrt._prepareabortorcontinue(isabort=(action == 'abort')) | ||
|
r29472 | if retcode is not None: | ||
return retcode | ||||
|
r6906 | else: | ||
|
r44384 | # search default destination in this space | ||
# used in the 'hg pull --rebase' case, see issue 5214. | ||||
|
r48253 | destspace = opts.get('_destspace') | ||
|
r43346 | destmap = _definedestmap( | ||
ui, | ||||
repo, | ||||
|
r46119 | rbsrt.inmemory, | ||
|
r48253 | opts.get('dest', None), | ||
opts.get('source', []), | ||||
opts.get('base', []), | ||||
opts.get('rev', []), | ||||
|
r43346 | destspace=destspace, | ||
) | ||||
|
r34006 | retcode = rbsrt._preparenewrebase(destmap) | ||
|
r29473 | if retcode is not None: | ||
return retcode | ||||
|
r36793 | storecollapsemsg(repo, rbsrt.collapsemsg) | ||
|
r21027 | |||
|
r33569 | tr = None | ||
|
r33621 | |||
|
r43347 | singletr = ui.configbool(b'rebase', b'singletransaction') | ||
|
r33621 | if singletr: | ||
|
r43347 | tr = repo.transaction(b'rebase') | ||
|
r35496 | |||
# If `rebase.singletransaction` is enabled, wrap the entire operation in | ||||
# one transaction here. Otherwise, transactions are obtained when | ||||
# committing each node, which is slower but allows partial success. | ||||
|
r33569 | with util.acceptintervention(tr): | ||
|
r35496 | # Same logic for the dirstate guard, except we don't create one when | ||
# rebasing in-memory (it's not needed). | ||||
|
r36791 | dsguard = None | ||
|
r46119 | if singletr and not rbsrt.inmemory: | ||
|
r43347 | dsguard = dirstateguard.dirstateguard(repo, b'rebase') | ||
|
r33621 | with util.acceptintervention(dsguard): | ||
rbsrt._performrebase(tr) | ||||
|
r46121 | if not rbsrt.dryrun: | ||
|
r38852 | rbsrt._finishrebase() | ||
|
r6906 | |||
|
r43346 | |||
|
r45042 | def _definedestmap(ui, repo, inmemory, destf, srcf, basef, revf, destspace): | ||
|
r34006 | """use revisions argument to define destmap {srcrev: destrev}""" | ||
|
r31431 | if revf is None: | ||
revf = [] | ||||
|
r31395 | |||
|
r29043 | # destspace is here to work around issues with `hg pull --rebase` see | ||
# issue5214 for details | ||||
|
r28136 | |||
|
r43988 | cmdutil.checkunfinished(repo) | ||
|
r37043 | if not inmemory: | ||
|
r35292 | cmdutil.bailifchanged(repo) | ||
|
r28136 | |||
|
r43347 | if ui.configbool(b'commands', b'rebase.requiredest') and not destf: | ||
|
r47377 | raise error.InputError( | ||
|
r43347 | _(b'you must specify a destination'), | ||
hint=_(b'use: hg rebase -d REV'), | ||||
|
r43346 | ) | ||
|
r31731 | |||
|
r34007 | dest = None | ||
|
r28136 | |||
if revf: | ||||
|
r48928 | rebaseset = logcmdutil.revrange(repo, revf) | ||
|
r28136 | if not rebaseset: | ||
|
r43347 | ui.status(_(b'empty "rev" revision set - nothing to rebase\n')) | ||
|
r34006 | return None | ||
|
r28136 | elif srcf: | ||
|
r48928 | src = logcmdutil.revrange(repo, srcf) | ||
|
r28136 | if not src: | ||
|
r43347 | ui.status(_(b'empty "source" revision set - nothing to rebase\n')) | ||
|
r34006 | return None | ||
|
r45044 | # `+ (%ld)` to work around `wdir()::` being empty | ||
rebaseset = repo.revs(b'(%ld):: + (%ld)', src, src) | ||||
|
r28136 | else: | ||
|
r48928 | base = logcmdutil.revrange(repo, basef or [b'.']) | ||
|
r28136 | if not base: | ||
|
r43346 | ui.status( | ||
|
r43347 | _(b'empty "base" revision set - ' b"can't compute rebase set\n") | ||
|
r43346 | ) | ||
|
r34006 | return None | ||
|
r34007 | if destf: | ||
# --base does not support multiple destinations | ||||
|
r48930 | dest = logcmdutil.revsingle(repo, destf) | ||
|
r34007 | else: | ||
|
r29043 | dest = repo[_destrebase(repo, base, destspace=destspace)] | ||
|
r36493 | destf = bytes(dest) | ||
|
r28189 | |||
|
r43346 | roots = [] # selected children of branching points | ||
bpbase = {} # {branchingpoint: [origbase]} | ||||
for b in base: # group bases by branching points | ||||
|
r43347 | bp = repo.revs(b'ancestor(%d, %d)', b, dest.rev()).first() | ||
|
r30580 | bpbase[bp] = bpbase.get(bp, []) + [b] | ||
if None in bpbase: | ||||
# emulate the old behavior, showing "nothing to rebase" (a better | ||||
# behavior may be abort with "cannot find branching point" error) | ||||
bpbase.clear() | ||||
|
r43375 | for bp, bs in pycompat.iteritems(bpbase): # calculate roots | ||
|
r43347 | roots += list(repo.revs(b'children(%d) & ancestors(%ld)', bp, bs)) | ||
|
r30580 | |||
|
r43347 | rebaseset = repo.revs(b'%ld::', roots) | ||
|
r28136 | |||
if not rebaseset: | ||||
# transform to list because smartsets are not comparable to | ||||
# lists. This should be improved to honor laziness of | ||||
# smartset. | ||||
if list(base) == [dest.rev()]: | ||||
if basef: | ||||
|
r43346 | ui.status( | ||
_( | ||||
|
r43347 | b'nothing to rebase - %s is both "base"' | ||
b' and destination\n' | ||||
|
r43346 | ) | ||
% dest | ||||
) | ||||
|
r28136 | else: | ||
|
r43346 | ui.status( | ||
_( | ||||
|
r43347 | b'nothing to rebase - working directory ' | ||
b'parent is also destination\n' | ||||
|
r43346 | ) | ||
) | ||||
|
r43347 | elif not repo.revs(b'%ld - ::%d', base, dest.rev()): | ||
|
r28136 | if basef: | ||
|
r43346 | ui.status( | ||
_( | ||||
|
r43347 | b'nothing to rebase - "base" %s is ' | ||
b'already an ancestor of destination ' | ||||
b'%s\n' | ||||
|
r43346 | ) | ||
|
r43347 | % (b'+'.join(bytes(repo[r]) for r in base), dest) | ||
|
r43346 | ) | ||
|
r28136 | else: | ||
|
r43346 | ui.status( | ||
_( | ||||
|
r43347 | b'nothing to rebase - working ' | ||
b'directory parent is already an ' | ||||
b'ancestor of destination %s\n' | ||||
|
r43346 | ) | ||
% dest | ||||
) | ||||
else: # can it happen? | ||||
ui.status( | ||||
|
r43347 | _(b'nothing to rebase from %s to %s\n') | ||
% (b'+'.join(bytes(repo[r]) for r in base), dest) | ||||
|
r43346 | ) | ||
|
r34006 | return None | ||
|
r36993 | |||
|
r46729 | if wdirrev in rebaseset: | ||
|
r47377 | raise error.InputError(_(b'cannot rebase the working copy')) | ||
|
r43347 | rebasingwcp = repo[b'.'].rev() in rebaseset | ||
|
r43346 | ui.log( | ||
|
r43347 | b"rebase", | ||
b"rebasing working copy parent: %r\n", | ||||
|
r43346 | rebasingwcp, | ||
rebase_rebasing_wcp=rebasingwcp, | ||||
) | ||||
|
r37043 | if inmemory and rebasingwcp: | ||
|
r35333 | # Check these since we did not before. | ||
cmdutil.checkunfinished(repo) | ||||
cmdutil.bailifchanged(repo) | ||||
|
r28189 | |||
if not destf: | ||||
|
r29043 | dest = repo[_destrebase(repo, rebaseset, destspace=destspace)] | ||
|
r36493 | destf = bytes(dest) | ||
|
r28189 | |||
|
r43347 | allsrc = revsetlang.formatspec(b'%ld', rebaseset) | ||
alias = {b'ALLSRC': allsrc} | ||||
|
r34007 | |||
if dest is None: | ||||
try: | ||||
# fast path: try to resolve dest without SRC alias | ||||
dest = scmutil.revsingle(repo, destf, localalias=alias) | ||||
except error.RepoLookupError: | ||||
# multi-dest path: resolve dest for each SRC separately | ||||
destmap = {} | ||||
for r in rebaseset: | ||||
|
r43347 | alias[b'SRC'] = revsetlang.formatspec(b'%d', r) | ||
|
r34007 | # use repo.anyrevs instead of scmutil.revsingle because we | ||
# don't want to abort if destset is empty. | ||||
destset = repo.anyrevs([destf], user=True, localalias=alias) | ||||
size = len(destset) | ||||
if size == 1: | ||||
destmap[r] = destset.first() | ||||
elif size == 0: | ||||
|
r43347 | ui.note(_(b'skipping %s - empty destination\n') % repo[r]) | ||
|
r34007 | else: | ||
|
r47377 | raise error.InputError( | ||
|
r43387 | _(b'rebase destination for %s is not unique') % repo[r] | ||
|
r43346 | ) | ||
|
r34007 | |||
if dest is not None: | ||||
# single-dest case: assign dest to each rev in rebaseset | ||||
destrev = dest.rev() | ||||
|
r43346 | destmap = {r: destrev for r in rebaseset} # {srcrev: destrev} | ||
|
r34007 | |||
if not destmap: | ||||
|
r43347 | ui.status(_(b'nothing to rebase - empty destination\n')) | ||
|
r34007 | return None | ||
|
r34006 | |||
return destmap | ||||
|
r28136 | |||
|
r43346 | |||
|
r32248 | def externalparent(repo, state, destancestors): | ||
|
r19955 | """Return the revision that should be used as the second parent | ||
|
r32248 | when the revisions in state is collapsed on top of destancestors. | ||
|
r19955 | Abort if there is more than one parent. | ||
|
r10351 | """ | ||
|
r19955 | parents = set() | ||
|
r10351 | source = min(state) | ||
for rev in state: | ||||
if rev == source: | ||||
continue | ||||
for p in repo[rev].parents(): | ||||
|
r43346 | if p.rev() not in state and p.rev() not in destancestors: | ||
|
r19955 | parents.add(p.rev()) | ||
if not parents: | ||||
return nullrev | ||||
if len(parents) == 1: | ||||
return parents.pop() | ||||
|
r47377 | raise error.StateError( | ||
|
r43346 | _( | ||
|
r43347 | b'unable to collapse on top of %d, there is more ' | ||
b'than one external parent: %s' | ||||
|
r43346 | ) | ||
|
r43347 | % (max(destancestors), b', '.join(b"%d" % p for p in sorted(parents))) | ||
|
r43346 | ) | ||
|
r10351 | |||
|
r44823 | def commitmemorynode(repo, wctx, editor, extra, user, date, commitmsg): | ||
|
r46554 | """Commit the memory changes with parents p1 and p2. | ||
Return node of committed revision.""" | ||||
|
r37058 | # By convention, ``extra['branch']`` (set by extrafn) clobbers | ||
# ``branch`` (used when passing ``--keepbranches``). | ||||
|
r44502 | branch = None | ||
|
r43347 | if b'branch' in extra: | ||
branch = extra[b'branch'] | ||||
|
r35320 | |||
|
r45799 | # FIXME: We call _compact() because it's required to correctly detect | ||
# changed files. This was added to fix a regression shortly before the 5.5 | ||||
# release. A proper fix will be done in the default branch. | ||||
wctx._compact() | ||||
|
r43346 | memctx = wctx.tomemctx( | ||
commitmsg, | ||||
date=date, | ||||
extra=extra, | ||||
user=user, | ||||
branch=branch, | ||||
editor=editor, | ||||
) | ||||
|
r45650 | if memctx.isempty() and not repo.ui.configbool(b'ui', b'allowemptycommit'): | ||
return None | ||||
|
r37058 | commitres = repo.commitctx(memctx) | ||
|
r43346 | wctx.clean() # Might be reused | ||
|
r37058 | return commitres | ||
|
r35320 | |||
|
r43346 | |||
|
r44823 | def commitnode(repo, editor, extra, user, date, commitmsg): | ||
|
r46554 | """Commit the wd changes with parents p1 and p2. | ||
Return node of committed revision.""" | ||||
|
r33621 | dsguard = util.nullcontextmanager() | ||
|
r43347 | if not repo.ui.configbool(b'rebase', b'singletransaction'): | ||
dsguard = dirstateguard.dirstateguard(repo, b'rebase') | ||||
|
r33621 | with dsguard: | ||
|
r37058 | # Commit might fail if unresolved files exist | ||
|
r43346 | newnode = repo.commit( | ||
text=commitmsg, user=user, date=date, extra=extra, editor=editor | ||||
) | ||||
|
r22038 | |||
r33135 | repo.dirstate.setbranch(repo[newnode].branch()) | |||
return newnode | ||||
|
r6906 | |||
|
r43346 | |||
|
r46120 | def rebasenode(repo, rev, p1, p2, base, collapse, wctx): | ||
|
r44226 | """Rebase a single revision rev on top of p1 using base as merge ancestor""" | ||
|
r6906 | # Merge phase | ||
|
r32248 | # Update to destination and merge it with local | ||
|
r44553 | p1ctx = repo[p1] | ||
|
r35318 | if wctx.isinmemory(): | ||
|
r44553 | wctx.setbase(p1ctx) | ||
|
r6906 | else: | ||
|
r43347 | if repo[b'.'].rev() != p1: | ||
|
r44553 | repo.ui.debug(b" update to %d:%s\n" % (p1, p1ctx)) | ||
|
r44743 | mergemod.clean_update(p1ctx) | ||
|
r35318 | else: | ||
|
r43347 | repo.ui.debug(b" already in destination\n") | ||
|
r35411 | # This is, alas, necessary to invalidate workingctx's manifest cache, | ||
# as well as other data we litter on it in other places. | ||||
wctx = repo[None] | ||||
|
r35318 | repo.dirstate.write(repo.currenttransaction()) | ||
|
r44549 | ctx = repo[rev] | ||
repo.ui.debug(b" merge against %d:%s\n" % (rev, ctx)) | ||||
|
r19969 | if base is not None: | ||
|
r43347 | repo.ui.debug(b" detach base %d:%s\n" % (base, repo[base])) | ||
|
r44554 | |||
# See explanation in merge.graft() | ||||
mergeancestor = repo.changelog.isancestor(p1ctx.node(), ctx.node()) | ||||
|
r46134 | stats = mergemod._update( | ||
|
r43346 | repo, | ||
rev, | ||||
branchmerge=True, | ||||
force=True, | ||||
ancestor=base, | ||||
|
r44554 | mergeancestor=mergeancestor, | ||
|
r49434 | labels=[b'dest', b'source', b'parent of source'], | ||
|
r43346 | wc=wctx, | ||
) | ||||
|
r44821 | wctx.setparents(p1ctx.node(), repo[p2].node()) | ||
|
r22905 | if collapse: | ||
|
r46120 | copies.graftcopies(wctx, ctx, p1ctx) | ||
|
r22905 | else: | ||
# If we're not using --collapse, we need to | ||||
# duplicate copies between the revision we're | ||||
|
r44551 | # rebasing and its first parent. | ||
copies.graftcopies(wctx, ctx, ctx.p1()) | ||||
|
r46122 | |||
if stats.unresolvedcount > 0: | ||||
if wctx.isinmemory(): | ||||
raise error.InMemoryMergeConflictsError() | ||||
else: | ||||
raise error.ConflictResolutionRequired(b'rebase') | ||||
|
r6923 | |||
|
r43346 | |||
|
r34010 | def adjustdest(repo, rev, destmap, state, skipped): | ||
|
r41674 | r"""adjust rebase destination given the current rebase state | ||
|
r33591 | |||
rev is what is being rebased. Return a list of two revs, which are the | ||||
adjusted destinations for rev's p1 and p2, respectively. If a parent is | ||||
nullrev, return dest without adjustment for it. | ||||
|
r34007 | For example, when doing rebasing B+E to F, C to G, rebase will first move B | ||
to B1, and E's destination will be adjusted from F to B1. | ||||
|
r33591 | |||
B1 <- written during rebasing B | ||||
| | ||||
F <- original destination of B, E | ||||
| | ||||
| E <- rev, which is being rebased | ||||
| | | ||||
| D <- prev, one parent of rev being checked | ||||
| | | ||||
| x <- skipped, ex. no successor or successor in (::dest) | ||||
| | | ||||
|
r34007 | | C <- rebased as C', different destination | ||
|
r33591 | | | | ||
|
r34007 | | B <- rebased as B1 C' | ||
|/ | | ||||
A G <- destination of C, different | ||||
|
r33591 | |||
Another example about merge changeset, rebase -r C+G+H -d K, rebase will | ||||
first move C to C1, G to G1, and when it's checking H, the adjusted | ||||
destinations will be [C1, G1]. | ||||
H C1 G1 | ||||
/| | / | ||||
F G |/ | ||||
K | | -> K | ||||
| C D | | ||||
| |/ | | ||||
| B | ... | ||||
|/ |/ | ||||
A A | ||||
|
r34008 | |||
Besides, adjust dest according to existing rebase information. For example, | ||||
B C D B needs to be rebased on top of C, C needs to be rebased on top | ||||
\|/ of D. We will rebase C first. | ||||
A | ||||
C' After rebasing C, when considering B's destination, use C' | ||||
| instead of the original C. | ||||
B D | ||||
\ / | ||||
A | ||||
|
r33591 | """ | ||
|
r34006 | # pick already rebased revs with same dest from state as interesting source | ||
dest = destmap[rev] | ||||
|
r43346 | source = [ | ||
s | ||||
for s, d in state.items() | ||||
if d > 0 and destmap[s] == dest and s not in skipped | ||||
] | ||||
|
r33848 | |||
|
r33591 | result = [] | ||
for prev in repo.changelog.parentrevs(rev): | ||||
adjusted = dest | ||||
if prev != nullrev: | ||||
|
r43347 | candidate = repo.revs(b'max(%ld and (::%d))', source, prev).first() | ||
|
r33591 | if candidate is not None: | ||
adjusted = state[candidate] | ||||
|
r34008 | if adjusted == dest and dest in state: | ||
adjusted = state[dest] | ||||
if adjusted == revtodo: | ||||
# sortsource should produce an order that makes this impossible | ||||
raise error.ProgrammingError( | ||||
|
r43347 | b'rev %d should be rebased already at this time' % dest | ||
|
r43346 | ) | ||
|
r33591 | result.append(adjusted) | ||
return result | ||||
|
r43346 | |||
|
r33845 | def _checkobsrebase(repo, ui, rebaseobsrevs, rebaseobsskipped): | ||
|
r28685 | """ | ||
Abort if rebase will create divergence or rebase is noop because of markers | ||||
`rebaseobsrevs`: set of obsolete revision in source | ||||
`rebaseobsskipped`: set of revisions from source skipped because they have | ||||
|
r36013 | successors in destination or no non-obsolete successor. | ||
|
r28685 | """ | ||
# Obsolete node with successors not in dest leads to divergence | ||||
r48577 | divergenceok = obsolete.isenabled(repo, obsolete.allowdivergenceopt) | |||
|
r28685 | divergencebasecandidates = rebaseobsrevs - rebaseobsskipped | ||
if divergencebasecandidates and not divergenceok: | ||||
|
r43346 | divhashes = (bytes(repo[r]) for r in divergencebasecandidates) | ||
|
r43387 | msg = _(b"this rebase will cause divergences from: %s") | ||
|
r43346 | h = _( | ||
|
r43347 | b"to force the rebase please set " | ||
b"experimental.evolution.allowdivergence=True" | ||||
|
r43346 | ) | ||
|
r47377 | raise error.StateError(msg % (b",".join(divhashes),), hint=h) | ||
|
r28685 | |||
|
r43346 | |||
|
r34097 | def successorrevs(unfi, rev): | ||
|
r33783 | """yield revision numbers for successors of rev""" | ||
|
r34097 | assert unfi.filtername is None | ||
r43966 | get_rev = unfi.changelog.index.get_rev | |||
|
r33783 | for s in obsutil.allsuccessors(unfi.obsstore, [unfi[rev].node()]): | ||
r43966 | r = get_rev(s) | |||
if r is not None: | ||||
yield r | ||||
|
r33783 | |||
|
r43346 | |||
|
r34010 | def defineparents(repo, rev, destmap, state, skipped, obsskipped): | ||
|
r33783 | """Return new parents and optionally a merge base for rev being rebased | ||
The destination specified by "dest" cannot always be used directly because | ||||
previously rebase result could affect destination. For example, | ||||
|
r6906 | |||
|
r33783 | D E rebase -r C+D+E -d B | ||
|/ C will be rebased to C' | ||||
B C D's new destination will be C' instead of B | ||||
|/ E's new destination will be C' instead of B | ||||
A | ||||
|
r6906 | |||
|
r33783 | The new parents of a merge is slightly more complicated. See the comment | ||
block below. | ||||
""" | ||||
|
r34094 | # use unfiltered changelog since successorrevs may return filtered nodes | ||
|
r34097 | assert repo.filtername is None | ||
cl = repo.changelog | ||||
|
r38688 | isancestor = cl.isancestorrev | ||
|
r33783 | |||
|
r34006 | dest = destmap[rev] | ||
|
r43346 | oldps = repo.changelog.parentrevs(rev) # old parents | ||
newps = [nullrev, nullrev] # new parents | ||||
|
r34010 | dests = adjustdest(repo, rev, destmap, state, skipped) | ||
|
r43346 | bases = list(oldps) # merge base candidates, initially just old parents | ||
|
r6906 | |||
|
r33783 | if all(r == nullrev for r in oldps[1:]): | ||
# For non-merge changeset, just move p to adjusted dest as requested. | ||||
newps[0] = dests[0] | ||||
else: | ||||
# For merge changeset, if we move p to dests[i] unconditionally, both | ||||
# parents may change and the end result looks like "the merge loses a | ||||
# parent", which is a surprise. This is a limit because "--dest" only | ||||
# accepts one dest per src. | ||||
# | ||||
# Therefore, only move p with reasonable conditions (in this order): | ||||
# 1. use dest, if dest is a descendent of (p or one of p's successors) | ||||
# 2. use p's rebased result, if p is rebased (state[p] > 0) | ||||
# | ||||
# Comparing with adjustdest, the logic here does some additional work: | ||||
# 1. decide which parents will not be moved towards dest | ||||
# 2. if the above decision is "no", should a parent still be moved | ||||
# because it was rebased? | ||||
# | ||||
# For example: | ||||
# | ||||
# C # "rebase -r C -d D" is an error since none of the parents | ||||
# /| # can be moved. "rebase -r B+C -d D" will move C's parent | ||||
# A B D # B (using rule "2."), since B will be rebased. | ||||
# | ||||
# The loop tries to be not rely on the fact that a Mercurial node has | ||||
# at most 2 parents. | ||||
for i, p in enumerate(oldps): | ||||
|
r43346 | np = p # new parent | ||
|
r33783 | if any(isancestor(x, dests[i]) for x in successorrevs(repo, p)): | ||
np = dests[i] | ||||
elif p in state and state[p] > 0: | ||||
np = state[p] | ||||
|
r23484 | |||
|
r33783 | # If one parent becomes an ancestor of the other, drop the ancestor | ||
for j, x in enumerate(newps[:i]): | ||||
if x == nullrev: | ||||
continue | ||||
|
r43346 | if isancestor(np, x): # CASE-1 | ||
|
r33783 | np = nullrev | ||
|
r43346 | elif isancestor(x, np): # CASE-2 | ||
|
r33783 | newps[j] = np | ||
np = nullrev | ||||
|
r33863 | # New parents forming an ancestor relationship does not | ||
# mean the old parents have a similar relationship. Do not | ||||
# set bases[x] to nullrev. | ||||
|
r33783 | bases[j], bases[i] = bases[i], bases[j] | ||
newps[i] = np | ||||
# "rebasenode" updates to new p1, and the old p1 will be used as merge | ||||
# base. If only p2 changes, merging using unchanged p1 as merge base is | ||||
# suboptimal. Therefore swap parents to make the merge sane. | ||||
if newps[1] != nullrev and oldps[0] == newps[0]: | ||||
assert len(newps) == 2 and len(oldps) == 2 | ||||
newps.reverse() | ||||
bases.reverse() | ||||
# No parent change might be an error because we fail to make rev a | ||||
# descendent of requested dest. This can happen, for example: | ||||
# | ||||
# C # rebase -r C -d D | ||||
# /| # None of A and B will be changed to D and rebase fails. | ||||
# A B D | ||||
if set(newps) == set(oldps) and dest not in newps: | ||||
|
r47377 | raise error.InputError( | ||
|
r43346 | _( | ||
|
r43347 | b'cannot rebase %d:%s without ' | ||
b'moving at least one of its parents' | ||||
|
r43346 | ) | ||
% (rev, repo[rev]) | ||||
) | ||||
|
r33783 | |||
|
r34008 | # Source should not be ancestor of dest. The check here guarantees it's | ||
# impossible. With multi-dest, the initial check does not cover complex | ||||
# cases since we don't have abstractions to dry-run rebase cheaply. | ||||
if any(p != nullrev and isancestor(rev, p) for p in newps): | ||||
|
r47377 | raise error.InputError(_(b'source is ancestor of destination')) | ||
|
r34008 | |||
|
r33783 | # Check if the merge will contain unwanted changes. That may happen if | ||
# there are multiple special (non-changelog ancestor) merge bases, which | ||||
# cannot be handled well by the 3-way merge algorithm. For example: | ||||
# | ||||
# F | ||||
# /| | ||||
# D E # "rebase -r D+E+F -d Z", when rebasing F, if "D" was chosen | ||||
# | | # as merge base, the difference between D and F will include | ||||
# B C # C, so the rebased F will contain C surprisingly. If "E" was | ||||
# |/ # chosen, the rebased F will contain B. | ||||
# A Z | ||||
# | ||||
# But our merge base candidates (D and E in above case) could still be | ||||
# better than the default (ancestor(F, Z) == null). Therefore still | ||||
# pick one (so choose p1 above). | ||||
|
r44815 | if sum(1 for b in set(bases) if b != nullrev and b not in newps) > 1: | ||
|
r43346 | unwanted = [None, None] # unwanted[i]: unwanted revs if choose bases[i] | ||
|
r33863 | for i, base in enumerate(bases): | ||
|
r44815 | if base == nullrev or base in newps: | ||
|
r33863 | continue | ||
# Revisions in the side (not chosen as merge base) branch that | ||||
# might contain "surprising" contents | ||||
|
r44627 | other_bases = set(bases) - {base} | ||
|
r43346 | siderevs = list( | ||
|
r44627 | repo.revs(b'(%ld %% (%d+%d))', other_bases, base, dest) | ||
|
r43346 | ) | ||
|
r23484 | |||
|
r33863 | # If those revisions are covered by rebaseset, the result is good. | ||
# A merge in rebaseset would be considered to cover its ancestors. | ||||
if siderevs: | ||||
|
r43346 | rebaseset = [ | ||
r for r, d in state.items() if d > 0 and r not in obsskipped | ||||
] | ||||
merges = [ | ||||
r for r in rebaseset if cl.parentrevs(r)[1] != nullrev | ||||
] | ||||
unwanted[i] = list( | ||||
repo.revs( | ||||
|
r43347 | b'%ld - (::%ld) - %ld', siderevs, merges, rebaseset | ||
|
r43346 | ) | ||
) | ||||
|
r33783 | |||
|
r44815 | if any(revs is not None for revs in unwanted): | ||
# Choose a merge base that has a minimal number of unwanted revs. | ||||
l, i = min( | ||||
(len(revs), i) | ||||
for i, revs in enumerate(unwanted) | ||||
if revs is not None | ||||
|
r43346 | ) | ||
|
r33863 | |||
|
r44815 | # The merge will include unwanted revisions. Abort now. Revisit this if | ||
# we have a more advanced merge algorithm that handles multiple bases. | ||||
if l > 0: | ||||
unwanteddesc = _(b' or ').join( | ||||
( | ||||
b', '.join(b'%d:%s' % (r, repo[r]) for r in revs) | ||||
for revs in unwanted | ||||
if revs is not None | ||||
) | ||||
) | ||||
|
r47377 | raise error.InputError( | ||
|
r44815 | _(b'rebasing %d:%s will include unwanted changes from %s') | ||
% (rev, repo[rev], unwanteddesc) | ||||
) | ||||
# newps[0] should match merge base if possible. Currently, if newps[i] | ||||
# is nullrev, the only case is newps[i] and newps[j] (j < i), one is | ||||
# the other's ancestor. In that case, it's fine to not swap newps here. | ||||
# (see CASE-1 and CASE-2 above) | ||||
if i != 0: | ||||
if newps[i] != nullrev: | ||||
newps[0], newps[i] = newps[i], newps[0] | ||||
bases[0], bases[i] = bases[i], bases[0] | ||||
|
r44695 | |||
# "rebasenode" updates to new p1, use the corresponding merge base. | ||||
|
r44815 | base = bases[0] | ||
|
r44628 | |||
|
r43347 | repo.ui.debug(b" future parents are %d and %d\n" % tuple(newps)) | ||
|
r33783 | |||
return newps[0], newps[1], base | ||||
|
r6906 | |||
|
r43346 | |||
|
r7955 | def isagitpatch(repo, patchname): | ||
|
r44226 | """Return true if the given patch is in git format""" | ||
|
r7955 | mqpatch = os.path.join(repo.mq.path, patchname) | ||
|
r43347 | for line in patch.linereader(open(mqpatch, b'rb')): | ||
if line.startswith(b'diff --git'): | ||||
|
r7955 | return True | ||
return False | ||||
|
r43346 | |||
|
r6906 | def updatemq(repo, state, skipped, **opts): | ||
|
r44226 | """Update rebased mq patches - finalize and then import them""" | ||
|
r6906 | mqrebase = {} | ||
|
r11537 | mq = repo.mq | ||
|
r14572 | original_series = mq.fullseries[:] | ||
|
r16531 | skippedpatches = set() | ||
|
r14497 | |||
|
r11537 | for p in mq.applied: | ||
rev = repo[p.node].rev() | ||||
if rev in state: | ||||
|
r43346 | repo.ui.debug( | ||
|
r43347 | b'revision %d is an mq patch (%s), finalize it.\n' | ||
|
r43346 | % (rev, p.name) | ||
) | ||||
|
r11537 | mqrebase[rev] = (p.name, isagitpatch(repo, p.name)) | ||
|
r16531 | else: | ||
# Applied but not rebased, not sure this should happen | ||||
skippedpatches.add(p.name) | ||||
|
r6906 | |||
if mqrebase: | ||||
|
r11537 | mq.finish(repo, mqrebase.keys()) | ||
|
r6906 | |||
# We must start import from the newest revision | ||||
|
r8210 | for rev in sorted(mqrebase, reverse=True): | ||
|
r6906 | if rev not in skipped: | ||
|
r11537 | name, isgit = mqrebase[rev] | ||
|
r43346 | repo.ui.note( | ||
|
r43347 | _(b'updating mq patch %s to %d:%s\n') | ||
|
r43346 | % (name, state[rev], repo[state[rev]]) | ||
) | ||||
mq.qimport( | ||||
|
r43347 | repo, | ||
(), | ||||
patchname=name, | ||||
git=isgit, | ||||
rev=[b"%d" % state[rev]], | ||||
|
r43346 | ) | ||
|
r16531 | else: | ||
# Rebased and skipped | ||||
skippedpatches.add(mqrebase[rev][0]) | ||||
|
r14497 | |||
|
r16531 | # Patches were either applied and rebased and imported in | ||
# order, applied and removed or unapplied. Discard the removed | ||||
# ones while preserving the original series order and guards. | ||||
|
r43346 | newseries = [ | ||
s | ||||
for s in original_series | ||||
if mq.guard_re.split(s, 1)[0] not in skippedpatches | ||||
] | ||||
|
r16531 | mq.fullseries[:] = newseries | ||
mq.seriesdirty = True | ||||
|
r14580 | mq.savedirty() | ||
|
r6906 | |||
|
r43346 | |||
|
r28185 | def storecollapsemsg(repo, collapsemsg): | ||
|
r44226 | """Store the collapse message to allow recovery""" | ||
|
r43347 | collapsemsg = collapsemsg or b'' | ||
f = repo.vfs(b"last-message.txt", b"w") | ||||
f.write(b"%s\n" % collapsemsg) | ||||
|
r28185 | f.close() | ||
|
r43346 | |||
|
r28185 | def clearcollapsemsg(repo): | ||
|
r44226 | """Remove collapse message file""" | ||
|
r43347 | repo.vfs.unlinkpath(b"last-message.txt", ignoremissing=True) | ||
|
r28185 | |||
|
r43346 | |||
|
r31225 | def restorecollapsemsg(repo, isabort): | ||
|
r44226 | """Restore previously stored collapse message""" | ||
|
r28185 | try: | ||
|
r43347 | f = repo.vfs(b"last-message.txt") | ||
|
r28185 | collapsemsg = f.readline().strip() | ||
f.close() | ||||
except IOError as err: | ||||
if err.errno != errno.ENOENT: | ||||
raise | ||||
|
r31225 | if isabort: | ||
# Oh well, just abort like normal | ||||
|
r43347 | collapsemsg = b'' | ||
|
r31225 | else: | ||
|
r43347 | raise error.Abort(_(b'missing .hg/last-message.txt for rebase')) | ||
|
r28185 | return collapsemsg | ||
|
r43346 | |||
|
r6906 | def clearstatus(repo): | ||
|
r44226 | """Remove the status files""" | ||
|
r33056 | # Make sure the active transaction won't write the state file | ||
tr = repo.currenttransaction() | ||||
if tr: | ||||
|
r43347 | tr.removefilegenerator(b'rebasestate') | ||
repo.vfs.unlinkpath(b"rebasestate", ignoremissing=True) | ||||
|
r6906 | |||
|
r43346 | |||
|
r34008 | def sortsource(destmap): | ||
"""yield source revisions in an order that we only rebase things once | ||||
If source and destination overlaps, we should filter out revisions | ||||
depending on other revisions which hasn't been rebased yet. | ||||
Yield a sorted list of revisions each time. | ||||
For example, when rebasing A to B, B to C. This function yields [B], then | ||||
[A], indicating B needs to be rebased first. | ||||
Raise if there is a cycle so the rebase is impossible. | ||||
""" | ||||
srcset = set(destmap) | ||||
while srcset: | ||||
srclist = sorted(srcset) | ||||
result = [] | ||||
for r in srclist: | ||||
if destmap[r] not in srcset: | ||||
result.append(r) | ||||
if not result: | ||||
|
r47377 | raise error.InputError(_(b'source and destination form a cycle')) | ||
|
r34008 | srcset -= set(result) | ||
yield result | ||||
|
r43346 | |||
|
r34010 | def buildstate(repo, destmap, collapse): | ||
|
r46554 | """Define which revisions are going to be rebased and where | ||
|
r6906 | |||
|
r15267 | repo: repo | ||
|
r34006 | destmap: {srcrev: destrev} | ||
|
r46554 | """ | ||
|
r34006 | rebaseset = destmap.keys() | ||
|
r43347 | originalwd = repo[b'.'].rev() | ||
|
r6906 | |||
|
r10672 | # This check isn't strictly necessary, since mq detects commits over an | ||
# applied patch. But it prevents messing up the working directory when | ||||
# a partially completed rebase is blocked by mq. | ||||
|
r43347 | if b'qtip' in repo.tags(): | ||
|
r44937 | mqapplied = {repo[s.node].rev() for s in repo.mq.applied} | ||
|
r34006 | if set(destmap.values()) & mqapplied: | ||
|
r47377 | raise error.StateError(_(b'cannot rebase onto an applied mq patch')) | ||
|
r10672 | |||
|
r34008 | # Get "cycle" error early by exhausting the generator. | ||
|
r43346 | sortedsrc = list(sortsource(destmap)) # a list of sorted revs | ||
|
r34008 | if not sortedsrc: | ||
|
r47377 | raise error.InputError(_(b'no matching revisions')) | ||
|
r34008 | |||
# Only check the first batch of revisions to rebase not depending on other | ||||
# rebaseset. This means "source is ancestor of destination" for the second | ||||
# (and following) batches of revisions are not checked here. We rely on | ||||
# "defineparents" to do that check. | ||||
|
r43347 | roots = list(repo.set(b'roots(%ld)', sortedsrc[0])) | ||
|
r15267 | if not roots: | ||
|
r47377 | raise error.InputError(_(b'no matching revisions')) | ||
|
r43346 | |||
|
r36285 | def revof(r): | ||
return r.rev() | ||||
|
r43346 | |||
|
r36285 | roots = sorted(roots, key=revof) | ||
|
r32175 | state = dict.fromkeys(rebaseset, revtodo) | ||
|
r43346 | emptyrebase = len(sortedsrc) == 1 | ||
|
r18424 | for root in roots: | ||
|
r34006 | dest = repo[destmap[root.rev()]] | ||
|
r18424 | commonbase = root.ancestor(dest) | ||
if commonbase == root: | ||||
|
r47377 | raise error.InputError(_(b'source is ancestor of destination')) | ||
|
r18424 | if commonbase == dest: | ||
|
r31380 | wctx = repo[None] | ||
if dest == wctx.p1(): | ||||
# when rebasing to '.', it will use the current wd branch name | ||||
samebranch = root.branch() == wctx.branch() | ||||
else: | ||||
samebranch = root.branch() == dest.branch() | ||||
|
r32900 | if not collapse and samebranch and dest in root.parents(): | ||
|
r32272 | # mark the revision as done by setting its new revision | ||
# equal to its old (current) revisions | ||||
state[root.rev()] = root.rev() | ||||
|
r43347 | repo.ui.debug(b'source is a child of destination\n') | ||
|
r32272 | continue | ||
|
r6906 | |||
|
r32272 | emptyrebase = False | ||
|
r43347 | repo.ui.debug(b'rebase onto %s starting from %s\n' % (dest, root)) | ||
|
r32272 | if emptyrebase: | ||
return None | ||||
for rev in sorted(state): | ||||
parents = [p for p in repo.changelog.parentrevs(rev) if p != nullrev] | ||||
# if all parents of this revision are done, then so is this revision | ||||
if parents and all((state.get(p) == p for p in parents)): | ||||
state[rev] = rev | ||||
|
r34006 | return originalwd, destmap, state | ||
|
r6906 | |||
|
r43346 | |||
def clearrebased( | ||||
ui, | ||||
repo, | ||||
destmap, | ||||
state, | ||||
skipped, | ||||
collapsedas=None, | ||||
keepf=False, | ||||
fm=None, | ||||
backup=True, | ||||
): | ||||
|
r17613 | """dispose of rebased revision at the end of the rebase | ||
If `collapsedas` is not None, the rebase was a collapse whose result if the | ||||
|
r34364 | `collapsedas` node. | ||
If `keepf` is not True, the rebase has --keep set and no nodes should be | ||||
removed (but bookmarks still need to be moved). | ||||
|
r38835 | |||
If `backup` is False, no backup will be stored when stripping rebased | ||||
revisions. | ||||
|
r34364 | """ | ||
|
r33332 | tonode = repo.changelog.node | ||
|
r34364 | replacements = {} | ||
moves = {} | ||||
|
r39951 | stripcleanup = not obsolete.isenabled(repo, obsolete.createmarkersopt) | ||
|
r39955 | |||
collapsednodes = [] | ||||
|
r33333 | for rev, newrev in sorted(state.items()): | ||
if newrev >= 0 and newrev != rev: | ||||
|
r34364 | oldnode = tonode(rev) | ||
newnode = collapsedas or tonode(newrev) | ||||
moves[oldnode] = newnode | ||||
|
r43245 | succs = None | ||
if rev in skipped: | ||||
if stripcleanup or not repo[rev].obsolete(): | ||||
succs = () | ||||
elif collapsedas: | ||||
collapsednodes.append(oldnode) | ||||
else: | ||||
succs = (newnode,) | ||||
if succs is not None: | ||||
replacements[(oldnode,)] = succs | ||||
|
r39955 | if collapsednodes: | ||
replacements[tuple(collapsednodes)] = (collapsedas,) | ||||
|
r34884 | if fm: | ||
|
r35123 | hf = fm.hexfunc | ||
fl = fm.formatlist | ||||
fd = fm.formatdict | ||||
|
r39953 | changes = {} | ||
|
r43375 | for oldns, newn in pycompat.iteritems(replacements): | ||
|
r39954 | for oldn in oldns: | ||
|
r43347 | changes[hf(oldn)] = fl([hf(n) for n in newn], name=b'node') | ||
nodechanges = fd(changes, key=b"oldnode", value=b"newnodes") | ||||
|
r34884 | fm.data(nodechanges=nodechanges) | ||
|
r43245 | if keepf: | ||
replacements = {} | ||||
|
r43347 | scmutil.cleanupnodes(repo, replacements, b'rebase', moves, backup=backup) | ||
|
r17611 | |||
|
r43346 | |||
|
r7216 | def pullrebase(orig, ui, repo, *args, **opts): | ||
|
r44226 | """Call rebase after pull if the latter has been invoked with --rebase""" | ||
|
r43906 | if opts.get('rebase'): | ||
|
r43347 | if ui.configbool(b'commands', b'rebase.requiredest'): | ||
msg = _(b'rebase destination required by configuration') | ||||
hint = _(b'use hg pull followed by hg rebase -d DEST') | ||||
|
r47377 | raise error.InputError(msg, hint=hint) | ||
|
r31733 | |||
|
r32918 | with repo.wlock(), repo.lock(): | ||
|
r43906 | if opts.get('update'): | ||
del opts['update'] | ||||
|
r43346 | ui.debug( | ||
|
r43347 | b'--update and --rebase are not compatible, ignoring ' | ||
b'the update flag\n' | ||||
|
r43346 | ) | ||
|
r6906 | |||
|
r42732 | cmdutil.checkunfinished(repo, skipmerge=True) | ||
|
r43346 | cmdutil.bailifchanged( | ||
repo, | ||||
hint=_( | ||||
|
r43347 | b'cannot pull with rebase: ' | ||
b'please commit or shelve your changes first' | ||||
|
r43346 | ), | ||
) | ||||
|
r30725 | |||
|
r26029 | revsprepull = len(repo) | ||
origpostincoming = commands.postincoming | ||||
|
r43346 | |||
|
r26029 | def _dummy(*args, **kwargs): | ||
pass | ||||
|
r43346 | |||
|
r26029 | commands.postincoming = _dummy | ||
try: | ||||
|
r26960 | ret = orig(ui, repo, *args, **opts) | ||
|
r26029 | finally: | ||
commands.postincoming = origpostincoming | ||||
revspostpull = len(repo) | ||||
if revspostpull > revsprepull: | ||||
# --rev option from pull conflict with rebase own --rev | ||||
# dropping it | ||||
|
r43906 | if 'rev' in opts: | ||
del opts['rev'] | ||||
|
r26029 | # positional argument from pull conflicts with rebase's own | ||
# --source. | ||||
|
r43906 | if 'source' in opts: | ||
del opts['source'] | ||||
|
r29044 | # revsprepull is the len of the repo, not revnum of tip. | ||
destspace = list(repo.changelog.revs(start=revsprepull)) | ||||
|
r43906 | opts['_destspace'] = destspace | ||
|
r28189 | try: | ||
rebase(ui, repo, **opts) | ||||
except error.NoMergeDestAbort: | ||||
# we can maybe update instead | ||||
|
r28118 | rev, _a, _b = destutil.destupdate(repo) | ||
|
r43347 | if rev == repo[b'.'].rev(): | ||
ui.status(_(b'nothing to rebase\n')) | ||||
|
r28189 | else: | ||
|
r43347 | ui.status(_(b'nothing to rebase - updating instead\n')) | ||
|
r28118 | # not passing argument to get the bare update behavior | ||
# with warning and trumpets | ||||
commands.update(ui, repo) | ||||
|
r6906 | else: | ||
|
r43906 | if opts.get('tool'): | ||
|
r47377 | raise error.InputError(_(b'--tool can only be used with --rebase')) | ||
|
r26960 | ret = orig(ui, repo, *args, **opts) | ||
return ret | ||||
|
r6906 | |||
|
r43346 | |||
|
r47589 | def _compute_obsolete_sets(repo, rebaseobsrevs, destmap): | ||
"""Figure out what to do about about obsolete revisions | ||||
|
r35049 | |||
|
r47589 | `obsolete_with_successor_in_destination` is a mapping mapping obsolete => successor for all | ||
|
r35049 | obsolete nodes to be rebased given in `rebaseobsrevs`. | ||
|
r27012 | |||
|
r47589 | `obsolete_with_successor_in_rebase_set` is a set with obsolete revisions, | ||
without a successor in destination, that would cause divergence. | ||||
|
r35049 | """ | ||
|
r47589 | obsolete_with_successor_in_destination = {} | ||
obsolete_with_successor_in_rebase_set = set() | ||||
|
r26349 | |||
|
r26674 | cl = repo.changelog | ||
r43967 | get_rev = cl.index.get_rev | |||
|
r43347 | extinctrevs = set(repo.revs(b'extinct()')) | ||
|
r34005 | for srcrev in rebaseobsrevs: | ||
srcnode = cl.node(srcrev) | ||||
# XXX: more advanced APIs are required to handle split correctly | ||||
|
r36015 | successors = set(obsutil.allsuccessors(repo.obsstore, [srcnode])) | ||
|
r36012 | # obsutil.allsuccessors includes node itself | ||
successors.remove(srcnode) | ||||
r43967 | succrevs = {get_rev(s) for s in successors} | |||
succrevs.discard(None) | ||||
|
r47559 | if not successors or succrevs.issubset(extinctrevs): | ||
# no successor, or all successors are extinct | ||||
|
r47589 | obsolete_with_successor_in_destination[srcrev] = None | ||
|
r34005 | else: | ||
|
r38694 | dstrev = destmap[srcrev] | ||
for succrev in succrevs: | ||||
if cl.isancestorrev(succrev, dstrev): | ||||
|
r47589 | obsolete_with_successor_in_destination[srcrev] = succrev | ||
|
r34005 | break | ||
|
r35049 | else: | ||
# If 'srcrev' has a successor in rebase set but none in | ||||
# destination (which would be catched above), we shall skip it | ||||
# and its descendants to avoid divergence. | ||||
|
r39364 | if srcrev in extinctrevs or any(s in destmap for s in succrevs): | ||
|
r47589 | obsolete_with_successor_in_rebase_set.add(srcrev) | ||
|
r27012 | |||
|
r47589 | return ( | ||
obsolete_with_successor_in_destination, | ||||
obsolete_with_successor_in_rebase_set, | ||||
) | ||||
|
r26349 | |||
|
r43346 | |||
|
r42786 | def abortrebase(ui, repo): | ||
with repo.wlock(), repo.lock(): | ||||
rbsrt = rebaseruntime(repo, ui) | ||||
rbsrt._prepareabortorcontinue(isabort=True) | ||||
|
r43346 | |||
|
r42832 | def continuerebase(ui, repo): | ||
with repo.wlock(), repo.lock(): | ||||
rbsrt = rebaseruntime(repo, ui) | ||||
|
r45383 | ms = mergestatemod.mergestate.read(repo) | ||
|
r42832 | mergeutil.checkunresolved(ms) | ||
retcode = rbsrt._prepareabortorcontinue(isabort=False) | ||||
if retcode is not None: | ||||
return retcode | ||||
rbsrt._performrebase(None) | ||||
rbsrt._finishrebase() | ||||
|
r43346 | |||
|
r19214 | def summaryhook(ui, repo): | ||
|
r43347 | if not repo.vfs.exists(b'rebasestate'): | ||
|
r19214 | return | ||
|
r19849 | try: | ||
|
r29403 | rbsrt = rebaseruntime(repo, ui, {}) | ||
rbsrt.restorestatus() | ||||
state = rbsrt.state | ||||
|
r19849 | except error.RepoLookupError: | ||
# i18n: column positioning for "hg summary" | ||||
|
r43347 | msg = _(b'rebase: (use "hg rebase --abort" to clear broken state)\n') | ||
|
r19849 | ui.write(msg) | ||
return | ||||
|
r43374 | numrebased = len([i for i in pycompat.itervalues(state) if i >= 0]) | ||
|
r19214 | # i18n: column positioning for "hg summary" | ||
|
r43346 | ui.write( | ||
|
r43347 | _(b'rebase: %s, %s (rebase --continue)\n') | ||
|
r43346 | % ( | ||
|
r43347 | ui.label(_(b'%d rebased'), b'rebase.rebased') % numrebased, | ||
ui.label(_(b'%d remaining'), b'rebase.remaining') | ||||
|
r43346 | % (len(state) - numrebased), | ||
) | ||||
) | ||||
|
r19214 | |||
|
r6906 | def uisetup(ui): | ||
|
r43346 | # Replace pull with a decorator to provide --rebase option | ||
|
r43347 | entry = extensions.wrapcommand(commands.table, b'pull', pullrebase) | ||
|
r43346 | entry[1].append( | ||
|
r43347 | (b'', b'rebase', None, _(b"rebase working directory to branch head")) | ||
|
r43346 | ) | ||
|
r43347 | entry[1].append((b't', b'tool', b'', _(b"specify merge tool for rebase"))) | ||
cmdutil.summaryhooks.add(b'rebase', summaryhook) | ||||
|
r43346 | statemod.addunfinished( | ||
|
r43347 | b'rebase', | ||
fname=b'rebasestate', | ||||
|
r43346 | stopflag=True, | ||
continueflag=True, | ||||
abortfunc=abortrebase, | ||||
continuefunc=continuerebase, | ||||
) | ||||