rebase.py
441 lines
| 16.4 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> | ||||
# | ||||
# This software may be used and distributed according to the terms | ||||
# of the GNU General Public License, incorporated herein by reference. | ||||
|
r7127 | '''move sets of revisions to a different ancestor | ||
|
r6906 | |||
This extension lets you rebase changesets in an existing Mercurial repository. | ||||
For more information: | ||||
http://www.selenic.com/mercurial/wiki/index.cgi/RebaseProject | ||||
''' | ||||
|
r7636 | from mercurial import util, repair, merge, cmdutil, commands, error | ||
|
r7278 | from mercurial import extensions, ancestor | ||
|
r6906 | from mercurial.commands import templateopts | ||
from mercurial.node import nullrev | ||||
from mercurial.i18n import _ | ||||
import os, errno | ||||
|
r7278 | def rebasemerge(repo, rev, first=False): | ||
'return the correct ancestor' | ||||
oldancestor = ancestor.ancestor | ||||
|
r7298 | |||
|
r7278 | def newancestor(a, b, pfunc): | ||
ancestor.ancestor = oldancestor | ||||
if b == rev: | ||||
return repo[rev].parents()[0].rev() | ||||
return ancestor.ancestor(a, b, pfunc) | ||||
if not first: | ||||
ancestor.ancestor = newancestor | ||||
else: | ||||
|
r7599 | repo.ui.debug(_("first revision, do not change ancestor\n")) | ||
|
r7278 | stats = merge.update(repo, rev, True, True, False) | ||
return stats | ||||
|
r6906 | def rebase(ui, repo, **opts): | ||
"""move changeset (and descendants) to a different branch | ||||
Rebase uses repeated merging to graft changesets from one part of history | ||||
onto another. This can be useful for linearizing local changes relative to | ||||
a master development tree. | ||||
If a rebase is interrupted to manually resolve a merge, it can be continued | ||||
with --continue or aborted with --abort. | ||||
""" | ||||
|
r7280 | originalwd = target = None | ||
|
r6906 | external = nullrev | ||
state = skipped = {} | ||||
lock = wlock = None | ||||
try: | ||||
lock = repo.lock() | ||||
wlock = repo.wlock() | ||||
# Validate input and define rebasing points | ||||
destf = opts.get('dest', None) | ||||
srcf = opts.get('source', None) | ||||
basef = opts.get('base', None) | ||||
contf = opts.get('continue') | ||||
abortf = opts.get('abort') | ||||
collapsef = opts.get('collapse', False) | ||||
|
r7468 | extrafn = opts.get('extrafn') | ||
if opts.get('keepbranches', None): | ||||
if extrafn: | ||||
|
r7636 | raise error.ParseError( | ||
'rebase', _('cannot use both keepbranches and extrafn')) | ||||
|
r7468 | def extrafn(ctx, extra): | ||
extra['branch'] = ctx.branch() | ||||
|
r6906 | if contf or abortf: | ||
if contf and abortf: | ||||
|
r7636 | raise error.ParseError('rebase', | ||
_('cannot use both abort and continue')) | ||||
|
r6906 | if collapsef: | ||
|
r7636 | raise error.ParseError( | ||
'rebase', _('cannot use collapse with continue or abort')) | ||||
|
r6906 | |||
if (srcf or basef or destf): | ||||
|
r7636 | raise error.ParseError('rebase', | ||
|
r6906 | _('abort and continue do not allow specifying revisions')) | ||
originalwd, target, state, collapsef, external = restorestatus(repo) | ||||
if abortf: | ||||
abort(repo, originalwd, target, state) | ||||
return | ||||
else: | ||||
if srcf and basef: | ||||
|
r7636 | raise error.ParseError('rebase', _('cannot specify both a ' | ||
'revision and a base')) | ||||
|
r6906 | cmdutil.bail_if_changed(repo) | ||
result = buildstate(repo, destf, srcf, basef, collapsef) | ||||
if result: | ||||
originalwd, target, state, external = result | ||||
else: # Empty state built, nothing to rebase | ||||
repo.ui.status(_('nothing to rebase\n')) | ||||
return | ||||
# Rebase | ||||
targetancestors = list(repo.changelog.ancestors(target)) | ||||
targetancestors.append(target) | ||||
for rev in util.sort(state): | ||||
if state[rev] == -1: | ||||
storestatus(repo, originalwd, target, state, collapsef, | ||||
external) | ||||
rebasenode(repo, rev, target, state, skipped, targetancestors, | ||||
|
r7468 | collapsef, extrafn) | ||
|
r6906 | ui.note(_('rebase merging completed\n')) | ||
if collapsef: | ||||
|
r6923 | p1, p2 = defineparents(repo, min(state), target, | ||
|
r6906 | state, targetancestors) | ||
concludenode(repo, rev, p1, external, state, collapsef, | ||||
|
r7468 | last=True, skipped=skipped, extrafn=extrafn) | ||
|
r6906 | |||
if 'qtip' in repo.tags(): | ||||
updatemq(repo, state, skipped, **opts) | ||||
if not opts.get('keep'): | ||||
# Remove no more useful revisions | ||||
|
r6923 | if (util.set(repo.changelog.descendants(min(state))) | ||
|
r6906 | - util.set(state.keys())): | ||
ui.warn(_("warning: new changesets detected on source branch, " | ||||
"not stripping\n")) | ||||
else: | ||||
repair.strip(repo.ui, repo, repo[min(state)].node(), "strip") | ||||
clearstatus(repo) | ||||
ui.status(_("rebase completed\n")) | ||||
|
r7130 | if os.path.exists(repo.sjoin('undo')): | ||
util.unlink(repo.sjoin('undo')) | ||||
|
r6906 | if skipped: | ||
ui.note(_("%d revisions have been skipped\n") % len(skipped)) | ||||
finally: | ||||
del lock, wlock | ||||
|
r7468 | def concludenode(repo, rev, p1, p2, state, collapse, last=False, skipped={}, | ||
extrafn=None): | ||||
|
r6906 | """Skip commit if collapsing has been required and rev is not the last | ||
revision, commit otherwise | ||||
""" | ||||
|
r7278 | repo.ui.debug(_(" set parents\n")) | ||
if collapse and not last: | ||||
repo.dirstate.setparents(repo[p1].node()) | ||||
return None | ||||
|
r6906 | |||
|
r7278 | repo.dirstate.setparents(repo[p1].node(), repo[p2].node()) | ||
|
r6906 | |||
# Commit, record the old nodeid | ||||
m, a, r = repo.status()[:3] | ||||
newrev = nullrev | ||||
try: | ||||
if last: | ||||
commitmsg = 'Collapsed revision' | ||||
for rebased in state: | ||||
if rebased not in skipped: | ||||
commitmsg += '\n* %s' % repo[rebased].description() | ||||
commitmsg = repo.ui.edit(commitmsg, repo.ui.username()) | ||||
else: | ||||
commitmsg = repo[rev].description() | ||||
# Commit might fail if unresolved files exist | ||||
|
r7468 | extra = {'rebase_source': repo[rev].hex()} | ||
if extrafn: | ||||
extrafn(repo[rev], extra) | ||||
|
r6906 | newrev = repo.commit(m+a+r, | ||
text=commitmsg, | ||||
user=repo[rev].user(), | ||||
date=repo[rev].date(), | ||||
|
r7468 | extra=extra) | ||
|
r6906 | return newrev | ||
except util.Abort: | ||||
# Invalidate the previous setparents | ||||
repo.dirstate.invalidate() | ||||
raise | ||||
|
r7468 | def rebasenode(repo, rev, target, state, skipped, targetancestors, collapse, | ||
extrafn): | ||||
|
r6906 | 'Rebase a single revision' | ||
|
r7278 | repo.ui.debug(_("rebasing %d:%s\n") % (rev, repo[rev])) | ||
|
r6906 | |||
p1, p2 = defineparents(repo, rev, target, state, targetancestors) | ||||
|
r7298 | repo.ui.debug(_(" future parents are %d and %d\n") % (repo[p1].rev(), | ||
|
r7278 | repo[p2].rev())) | ||
|
r7298 | |||
|
r6906 | # Merge phase | ||
if len(repo.parents()) != 2: | ||||
# Update to target and merge it with local | ||||
|
r7278 | if repo['.'].rev() != repo[p1].rev(): | ||
repo.ui.debug(_(" update to %d:%s\n") % (repo[p1].rev(), repo[p1])) | ||||
merge.update(repo, p1, False, True, False) | ||||
else: | ||||
repo.ui.debug(_(" already in target\n")) | ||||
|
r6906 | repo.dirstate.write() | ||
|
r7278 | repo.ui.debug(_(" merge against %d:%s\n") % (repo[rev].rev(), repo[rev])) | ||
first = repo[rev].rev() == repo[min(state)].rev() | ||||
stats = rebasemerge(repo, rev, first) | ||||
|
r6906 | |||
if stats[3] > 0: | ||||
raise util.Abort(_('fix unresolved conflicts with hg resolve then ' | ||||
'run hg rebase --continue')) | ||||
else: # we have an interrupted rebase | ||||
repo.ui.debug(_('resuming interrupted rebase\n')) | ||||
|
r7468 | newrev = concludenode(repo, rev, p1, p2, state, collapse, | ||
extrafn=extrafn) | ||||
|
r6906 | |||
# Update the state | ||||
if newrev is not None: | ||||
state[rev] = repo[newrev].rev() | ||||
else: | ||||
if not collapse: | ||||
|
r6964 | repo.ui.note(_('no changes, revision %d skipped\n') % rev) | ||
repo.ui.debug(_('next revision set to %s\n') % p1) | ||||
|
r6906 | skipped[rev] = True | ||
state[rev] = p1 | ||||
|
r6923 | |||
|
r6906 | def defineparents(repo, rev, target, state, targetancestors): | ||
'Return the new parent relationship of the revision that will be rebased' | ||||
parents = repo[rev].parents() | ||||
p1 = p2 = nullrev | ||||
P1n = parents[0].rev() | ||||
if P1n in targetancestors: | ||||
p1 = target | ||||
elif P1n in state: | ||||
p1 = state[P1n] | ||||
else: # P1n external | ||||
p1 = target | ||||
p2 = P1n | ||||
if len(parents) == 2 and parents[1].rev() not in targetancestors: | ||||
P2n = parents[1].rev() | ||||
# interesting second parent | ||||
if P2n in state: | ||||
if p1 == target: # P1n in targetancestors or external | ||||
p1 = state[P2n] | ||||
else: | ||||
p2 = state[P2n] | ||||
else: # P2n external | ||||
if p2 != nullrev: # P1n external too => rev is a merged revision | ||||
raise util.Abort(_('cannot use revision %d as base, result ' | ||||
'would have 3 parents') % rev) | ||||
p2 = P2n | ||||
return p1, p2 | ||||
def updatemq(repo, state, skipped, **opts): | ||||
'Update rebased mq patches - finalize and then import them' | ||||
mqrebase = {} | ||||
for p in repo.mq.applied: | ||||
if repo[p.rev].rev() in state: | ||||
|
r6964 | repo.ui.debug(_('revision %d is an mq patch (%s), finalize it.\n') % | ||
|
r6906 | (repo[p.rev].rev(), p.name)) | ||
mqrebase[repo[p.rev].rev()] = p.name | ||||
if mqrebase: | ||||
repo.mq.finish(repo, mqrebase.keys()) | ||||
# We must start import from the newest revision | ||||
mq = mqrebase.keys() | ||||
mq.sort() | ||||
mq.reverse() | ||||
for rev in mq: | ||||
if rev not in skipped: | ||||
|
r6964 | repo.ui.debug(_('import mq patch %d (%s)\n') | ||
% (state[rev], mqrebase[rev])) | ||||
|
r6923 | repo.mq.qimport(repo, (), patchname=mqrebase[rev], | ||
|
r6906 | git=opts.get('git', False),rev=[str(state[rev])]) | ||
repo.mq.save_dirty() | ||||
def storestatus(repo, originalwd, target, state, collapse, external): | ||||
'Store the current status to allow recovery' | ||||
f = repo.opener("rebasestate", "w") | ||||
f.write(repo[originalwd].hex() + '\n') | ||||
f.write(repo[target].hex() + '\n') | ||||
f.write(repo[external].hex() + '\n') | ||||
f.write('%d\n' % int(collapse)) | ||||
|
r7622 | for d, v in state.iteritems(): | ||
|
r6906 | oldrev = repo[d].hex() | ||
newrev = repo[v].hex() | ||||
f.write("%s:%s\n" % (oldrev, newrev)) | ||||
f.close() | ||||
repo.ui.debug(_('rebase status stored\n')) | ||||
def clearstatus(repo): | ||||
'Remove the status files' | ||||
if os.path.exists(repo.join("rebasestate")): | ||||
util.unlink(repo.join("rebasestate")) | ||||
def restorestatus(repo): | ||||
'Restore a previously stored status' | ||||
try: | ||||
target = None | ||||
collapse = False | ||||
external = nullrev | ||||
state = {} | ||||
f = repo.opener("rebasestate") | ||||
for i, l in enumerate(f.read().splitlines()): | ||||
if i == 0: | ||||
originalwd = repo[l].rev() | ||||
elif i == 1: | ||||
target = repo[l].rev() | ||||
elif i == 2: | ||||
external = repo[l].rev() | ||||
elif i == 3: | ||||
collapse = bool(int(l)) | ||||
else: | ||||
oldrev, newrev = l.split(':') | ||||
state[repo[oldrev].rev()] = repo[newrev].rev() | ||||
repo.ui.debug(_('rebase status resumed\n')) | ||||
return originalwd, target, state, collapse, external | ||||
except IOError, err: | ||||
if err.errno != errno.ENOENT: | ||||
raise | ||||
raise util.Abort(_('no rebase in progress')) | ||||
def abort(repo, originalwd, target, state): | ||||
'Restore the repository to its original state' | ||||
if util.set(repo.changelog.descendants(target)) - util.set(state.values()): | ||||
repo.ui.warn(_("warning: new changesets detected on target branch, " | ||||
"not stripping\n")) | ||||
else: | ||||
# Strip from the first rebased revision | ||||
merge.update(repo, repo[originalwd].rev(), False, True, False) | ||||
rebased = filter(lambda x: x > -1, state.values()) | ||||
if rebased: | ||||
strippoint = min(rebased) | ||||
repair.strip(repo.ui, repo, repo[strippoint].node(), "strip") | ||||
clearstatus(repo) | ||||
repo.ui.status(_('rebase aborted\n')) | ||||
def buildstate(repo, dest, src, base, collapse): | ||||
'Define which revisions are going to be rebased and where' | ||||
targetancestors = util.set() | ||||
if not dest: | ||||
|
r7877 | # Destination defaults to the latest revision in the current branch | ||
|
r6906 | branch = repo[None].branch() | ||
dest = repo[branch].rev() | ||||
else: | ||||
if 'qtip' in repo.tags() and (repo[dest].hex() in | ||||
[s.rev for s in repo.mq.applied]): | ||||
raise util.Abort(_('cannot rebase onto an applied mq patch')) | ||||
dest = repo[dest].rev() | ||||
if src: | ||||
commonbase = repo[src].ancestor(repo[dest]) | ||||
if commonbase == repo[src]: | ||||
raise util.Abort(_('cannot rebase an ancestor')) | ||||
if commonbase == repo[dest]: | ||||
raise util.Abort(_('cannot rebase a descendant')) | ||||
source = repo[src].rev() | ||||
else: | ||||
if base: | ||||
cwd = repo[base].rev() | ||||
else: | ||||
cwd = repo['.'].rev() | ||||
if cwd == dest: | ||||
repo.ui.debug(_('already working on current\n')) | ||||
return None | ||||
targetancestors = util.set(repo.changelog.ancestors(dest)) | ||||
if cwd in targetancestors: | ||||
repo.ui.debug(_('already working on the current branch\n')) | ||||
return None | ||||
cwdancestors = util.set(repo.changelog.ancestors(cwd)) | ||||
cwdancestors.add(cwd) | ||||
rebasingbranch = cwdancestors - targetancestors | ||||
source = min(rebasingbranch) | ||||
|
r6923 | |||
|
r6906 | repo.ui.debug(_('rebase onto %d starting from %d\n') % (dest, source)) | ||
state = dict.fromkeys(repo.changelog.descendants(source), nullrev) | ||||
external = nullrev | ||||
if collapse: | ||||
if not targetancestors: | ||||
targetancestors = util.set(repo.changelog.ancestors(dest)) | ||||
for rev in state: | ||||
# Check externals and fail if there are more than one | ||||
for p in repo[rev].parents(): | ||||
if (p.rev() not in state and p.rev() != source | ||||
and p.rev() not in targetancestors): | ||||
if external != nullrev: | ||||
raise util.Abort(_('unable to collapse, there is more ' | ||||
'than one external parent')) | ||||
external = p.rev() | ||||
state[source] = nullrev | ||||
return repo['.'].rev(), repo[dest].rev(), state, external | ||||
|
r7216 | def pullrebase(orig, ui, repo, *args, **opts): | ||
|
r6906 | 'Call rebase after pull if the latter has been invoked with --rebase' | ||
if opts.get('rebase'): | ||||
if opts.get('update'): | ||||
|
r7786 | del opts.get['update'] | ||
ui.debug(_('--update and --rebase are not compatible, ignoring ' | ||||
'the update flag\n')) | ||||
|
r6906 | |||
cmdutil.bail_if_changed(repo) | ||||
revsprepull = len(repo) | ||||
|
r7216 | orig(ui, repo, *args, **opts) | ||
|
r6906 | revspostpull = len(repo) | ||
if revspostpull > revsprepull: | ||||
|
r7216 | rebase(ui, repo, **opts) | ||
|
r7786 | branch = repo[None].branch() | ||
dest = repo[branch].rev() | ||||
if dest != repo['.'].rev(): | ||||
# there was nothing to rebase we force an update | ||||
merge.update(repo, dest, False, False, False) | ||||
|
r6906 | else: | ||
|
r7216 | orig(ui, repo, *args, **opts) | ||
|
r6906 | |||
def uisetup(ui): | ||||
'Replace pull with a decorator to provide --rebase option' | ||||
|
r7216 | entry = extensions.wrapcommand(commands.table, 'pull', pullrebase) | ||
entry[1].append(('', 'rebase', None, | ||||
_("rebase working directory to branch head")) | ||||
) | ||||
|
r6906 | |||
cmdtable = { | ||||
"rebase": | ||||
(rebase, | ||||
[ | ||||
('', 'keep', False, _('keep original revisions')), | ||||
|
r7468 | ('', 'keepbranches', False, _('keep original branches')), | ||
|
r6906 | ('s', 'source', '', _('rebase from a given revision')), | ||
('b', 'base', '', _('rebase from the base of a given revision')), | ||||
('d', 'dest', '', _('rebase onto a given revision')), | ||||
('', 'collapse', False, _('collapse the rebased revisions')), | ||||
('c', 'continue', False, _('continue an interrupted rebase')), | ||||
('a', 'abort', False, _('abort an interrupted rebase')),] + | ||||
templateopts, | ||||
_('hg rebase [-s rev | -b rev] [-d rev] [--collapse] | [-c] | [-a] | ' | ||||
'[--keep]')), | ||||
} | ||||