histedit.py
770 lines
| 27.8 KiB
| text/x-python
|
PythonLexer
/ hgext / histedit.py
Augie Fackler
|
r17064 | # histedit.py - interactive history editing for mercurial | ||
# | ||||
# Copyright 2009 Augie Fackler <raf@durin42.com> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
Augie Fackler
|
r17131 | """interactive history editing | ||
With this extension installed, Mercurial gains one new command: histedit. Usage | ||||
is as follows, assuming the following history:: | ||||
@ 3[tip] 7c2fd3b9020c 2009-04-27 18:04 -0500 durin42 | ||||
| Add delta | ||||
| | ||||
o 2 030b686bedc4 2009-04-27 18:04 -0500 durin42 | ||||
| Add gamma | ||||
| | ||||
o 1 c561b4e977df 2009-04-27 18:04 -0500 durin42 | ||||
| Add beta | ||||
| | ||||
o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42 | ||||
Add alpha | ||||
If you were to run ``hg histedit c561b4e977df``, you would see the following | ||||
file open in your editor:: | ||||
pick c561b4e977df Add beta | ||||
pick 030b686bedc4 Add gamma | ||||
pick 7c2fd3b9020c Add delta | ||||
# Edit history between 633536316234 and 7c2fd3b9020c | ||||
# | ||||
# Commands: | ||||
# p, pick = use commit | ||||
# e, edit = use commit, but stop for amending | ||||
Wagner Bruna
|
r17335 | # f, fold = use commit, but fold into previous commit (combines N and N-1) | ||
Augie Fackler
|
r17131 | # d, drop = remove commit from history | ||
# m, mess = edit message without changing commit content | ||||
# | ||||
In this file, lines beginning with ``#`` are ignored. You must specify a rule | ||||
for each revision in your history. For example, if you had meant to add gamma | ||||
before beta, and then wanted to add delta in the same revision as beta, you | ||||
would reorganize the file to look like this:: | ||||
pick 030b686bedc4 Add gamma | ||||
pick c561b4e977df Add beta | ||||
fold 7c2fd3b9020c Add delta | ||||
# Edit history between 633536316234 and 7c2fd3b9020c | ||||
# | ||||
# Commands: | ||||
# p, pick = use commit | ||||
# e, edit = use commit, but stop for amending | ||||
Wagner Bruna
|
r17335 | # f, fold = use commit, but fold into previous commit (combines N and N-1) | ||
Augie Fackler
|
r17131 | # d, drop = remove commit from history | ||
# m, mess = edit message without changing commit content | ||||
# | ||||
At which point you close the editor and ``histedit`` starts working. When you | ||||
specify a ``fold`` operation, ``histedit`` will open an editor when it folds | ||||
those revisions together, offering you a chance to clean up the commit message:: | ||||
Add beta | ||||
*** | ||||
Add delta | ||||
Augie Fackler
|
r17064 | |||
Augie Fackler
|
r17131 | Edit the commit message to your liking, then close the editor. For | ||
this example, let's assume that the commit message was changed to | ||||
``Add beta and delta.`` After histedit has run and had a chance to | ||||
remove any old or temporary revisions it needed, the history looks | ||||
like this:: | ||||
@ 2[tip] 989b4d060121 2009-04-27 18:04 -0500 durin42 | ||||
| Add beta and delta. | ||||
| | ||||
o 1 081603921c3f 2009-04-27 18:04 -0500 durin42 | ||||
| Add gamma | ||||
| | ||||
o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42 | ||||
Add alpha | ||||
Note that ``histedit`` does *not* remove any revisions (even its own temporary | ||||
ones) until after it has completed all the editing operations, so it will | ||||
probably perform several strip operations when it's done. For the above example, | ||||
it had to run strip twice. Strip can be slow depending on a variety of factors, | ||||
so you might need to be a little patient. You can choose to keep the original | ||||
revisions by passing the ``--keep`` flag. | ||||
The ``edit`` operation will drop you back to a command prompt, | ||||
allowing you to edit files freely, or even use ``hg record`` to commit | ||||
some changes as a separate commit. When you're done, any remaining | ||||
uncommitted changes will be committed as well. When done, run ``hg | ||||
histedit --continue`` to finish this step. You'll be prompted for a | ||||
new commit message, but the default commit message will be the | ||||
original message for the ``edit`` ed revision. | ||||
The ``message`` operation will give you a chance to revise a commit | ||||
message without changing the contents. It's a shortcut for doing | ||||
``edit`` immediately followed by `hg histedit --continue``. | ||||
If ``histedit`` encounters a conflict when moving a revision (while | ||||
handling ``pick`` or ``fold``), it'll stop in a similar manner to | ||||
``edit`` with the difference that it won't prompt you for a commit | ||||
message when done. If you decide at this point that you don't like how | ||||
much work it will be to rearrange history, or that you made a mistake, | ||||
you can use ``hg histedit --abort`` to abandon the new changes you | ||||
have made and return to the state before you attempted to edit your | ||||
history. | ||||
If we clone the example repository above and add three more changes, such that | ||||
we have the following history:: | ||||
@ 6[tip] 038383181893 2009-04-27 18:04 -0500 stefan | ||||
| Add theta | ||||
| | ||||
o 5 140988835471 2009-04-27 18:04 -0500 stefan | ||||
| Add eta | ||||
| | ||||
o 4 122930637314 2009-04-27 18:04 -0500 stefan | ||||
| Add zeta | ||||
| | ||||
o 3 836302820282 2009-04-27 18:04 -0500 stefan | ||||
| Add epsilon | ||||
| | ||||
o 2 989b4d060121 2009-04-27 18:04 -0500 durin42 | ||||
| Add beta and delta. | ||||
| | ||||
o 1 081603921c3f 2009-04-27 18:04 -0500 durin42 | ||||
| Add gamma | ||||
| | ||||
o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42 | ||||
Add alpha | ||||
If you run ``hg histedit --outgoing`` on the clone then it is the same | ||||
as running ``hg histedit 836302820282``. If you need plan to push to a | ||||
repository that Mercurial does not detect to be related to the source | ||||
repo, you can add a ``--force`` option. | ||||
Augie Fackler
|
r17064 | """ | ||
Augie Fackler
|
r17131 | |||
Augie Fackler
|
r17064 | try: | ||
import cPickle as pickle | ||||
except ImportError: | ||||
import pickle | ||||
import os | ||||
from mercurial import bookmarks | ||||
from mercurial import cmdutil | ||||
from mercurial import discovery | ||||
from mercurial import error | ||||
Pierre-Yves David
|
r17644 | from mercurial import copies | ||
from mercurial import context | ||||
Augie Fackler
|
r17064 | from mercurial import hg | ||
Augie Fackler
|
r17326 | from mercurial import lock as lockmod | ||
Augie Fackler
|
r17064 | from mercurial import node | ||
from mercurial import repair | ||||
from mercurial import scmutil | ||||
from mercurial import util | ||||
Pierre-Yves David
|
r17759 | from mercurial import obsolete | ||
Pierre-Yves David
|
r17647 | from mercurial import merge as mergemod | ||
Augie Fackler
|
r17064 | from mercurial.i18n import _ | ||
Adrian Buehlmann
|
r17147 | cmdtable = {} | ||
command = cmdutil.command(cmdtable) | ||||
Augie Fackler
|
r17069 | testedwith = 'internal' | ||
Augie Fackler
|
r17064 | |||
Wagner Bruna
|
r17337 | # i18n: command names and abbreviations must remain untranslated | ||
FUJIWARA Katsunori
|
r17315 | editcomment = _("""# Edit history between %s and %s | ||
Augie Fackler
|
r17064 | # | ||
# Commands: | ||||
# p, pick = use commit | ||||
# e, edit = use commit, but stop for amending | ||||
# f, fold = use commit, but fold into previous commit (combines N and N-1) | ||||
# d, drop = remove commit from history | ||||
# m, mess = edit message without changing commit content | ||||
# | ||||
FUJIWARA Katsunori
|
r17315 | """) | ||
Augie Fackler
|
r17064 | |||
Pierre-Yves David
|
r17647 | def applychanges(ui, repo, ctx, opts): | ||
"""Merge changeset from ctx (only) in the current working directory""" | ||||
wcpar = repo.dirstate.parents()[0] | ||||
if ctx.p1().node() == wcpar: | ||||
# edition ar "in place" we do not need to make any merge, | ||||
# just applies changes on parent for edition | ||||
cmdutil.revert(ui, repo, ctx, (wcpar, node.nullid), all=True) | ||||
stats = None | ||||
else: | ||||
try: | ||||
# ui.forcemerge is an internal variable, do not document | ||||
repo.ui.setconfig('ui', 'forcemerge', opts.get('tool', '')) | ||||
stats = mergemod.update(repo, ctx.node(), True, True, False, | ||||
ctx.p1().node()) | ||||
finally: | ||||
repo.ui.setconfig('ui', 'forcemerge', '') | ||||
repo.setparents(wcpar, node.nullid) | ||||
repo.dirstate.write() | ||||
# fix up dirstate for copies and renames | ||||
cmdutil.duplicatecopies(repo, ctx.rev(), ctx.p1().rev()) | ||||
return stats | ||||
Leah Xue
|
r17407 | |||
Pierre-Yves David
|
r17644 | def collapse(repo, first, last, commitopts): | ||
"""collapse the set of revisions from first to last as new one. | ||||
Expected commit options are: | ||||
- message | ||||
- date | ||||
- username | ||||
Mads Kiilerich
|
r17738 | Commit message is edited in all cases. | ||
Pierre-Yves David
|
r17644 | |||
This function works in memory.""" | ||||
ctxs = list(repo.set('%d::%d', first, last)) | ||||
if not ctxs: | ||||
return None | ||||
base = first.parents()[0] | ||||
# commit a new version of the old changeset, including the update | ||||
# collect all files which might be affected | ||||
files = set() | ||||
for ctx in ctxs: | ||||
files.update(ctx.files()) | ||||
# Recompute copies (avoid recording a -> b -> a) | ||||
copied = copies.pathcopies(first, last) | ||||
# prune files which were reverted by the updates | ||||
def samefile(f): | ||||
if f in last.manifest(): | ||||
a = last.filectx(f) | ||||
if f in base.manifest(): | ||||
b = base.filectx(f) | ||||
return (a.data() == b.data() | ||||
and a.flags() == b.flags()) | ||||
else: | ||||
return False | ||||
else: | ||||
return f not in base.manifest() | ||||
files = [f for f in files if not samefile(f)] | ||||
# commit version of these files as defined by head | ||||
headmf = last.manifest() | ||||
def filectxfn(repo, ctx, path): | ||||
if path in headmf: | ||||
fctx = last[path] | ||||
flags = fctx.flags() | ||||
mctx = context.memfilectx(fctx.path(), fctx.data(), | ||||
islink='l' in flags, | ||||
isexec='x' in flags, | ||||
copied=copied.get(path)) | ||||
return mctx | ||||
raise IOError() | ||||
if commitopts.get('message'): | ||||
message = commitopts['message'] | ||||
else: | ||||
message = first.description() | ||||
user = commitopts.get('user') | ||||
date = commitopts.get('date') | ||||
extra = first.extra() | ||||
parents = (first.p1().node(), first.p2().node()) | ||||
new = context.memctx(repo, | ||||
parents=parents, | ||||
text=message, | ||||
files=files, | ||||
filectxfn=filectxfn, | ||||
user=user, | ||||
date=date, | ||||
extra=extra) | ||||
new._text = cmdutil.commitforceeditor(repo, new, []) | ||||
return repo.commitctx(new) | ||||
Augie Fackler
|
r17064 | def pick(ui, repo, ctx, ha, opts): | ||
oldctx = repo[ha] | ||||
if oldctx.parents()[0] == ctx: | ||||
ui.debug('node %s unchanged\n' % ha) | ||||
Pierre-Yves David
|
r17758 | return oldctx, [] | ||
Augie Fackler
|
r17064 | hg.update(repo, ctx.node()) | ||
Pierre-Yves David
|
r17647 | stats = applychanges(ui, repo, oldctx, opts) | ||
if stats and stats[3] > 0: | ||||
Augie Fackler
|
r17064 | raise util.Abort(_('Fix up the change and run ' | ||
'hg histedit --continue')) | ||||
Pierre-Yves David
|
r17647 | # drop the second merge parent | ||
Augie Fackler
|
r17066 | n = repo.commit(text=oldctx.description(), user=oldctx.user(), | ||
date=oldctx.date(), extra=oldctx.extra()) | ||||
Pierre-Yves David
|
r17647 | if n is None: | ||
ui.warn(_('%s: empty changeset\n') | ||||
% node.hex(ha)) | ||||
Pierre-Yves David
|
r17758 | return ctx, [] | ||
new = repo[n] | ||||
return new, [(oldctx.node(), (n,))] | ||||
Augie Fackler
|
r17064 | |||
def edit(ui, repo, ctx, ha, opts): | ||||
oldctx = repo[ha] | ||||
hg.update(repo, ctx.node()) | ||||
Pierre-Yves David
|
r17647 | applychanges(ui, repo, oldctx, opts) | ||
Augie Fackler
|
r17064 | raise util.Abort(_('Make changes as needed, you may commit or record as ' | ||
'needed now.\nWhen you are finished, run hg' | ||||
' histedit --continue to resume.')) | ||||
def fold(ui, repo, ctx, ha, opts): | ||||
oldctx = repo[ha] | ||||
hg.update(repo, ctx.node()) | ||||
Pierre-Yves David
|
r17647 | stats = applychanges(ui, repo, oldctx, opts) | ||
if stats and stats[3] > 0: | ||||
Augie Fackler
|
r17064 | raise util.Abort(_('Fix up the change and run ' | ||
'hg histedit --continue')) | ||||
Augie Fackler
|
r17066 | n = repo.commit(text='fold-temp-revision %s' % ha, user=oldctx.user(), | ||
date=oldctx.date(), extra=oldctx.extra()) | ||||
Pierre-Yves David
|
r17647 | if n is None: | ||
ui.warn(_('%s: empty changeset') | ||||
% node.hex(ha)) | ||||
Pierre-Yves David
|
r17758 | return ctx, [] | ||
Augie Fackler
|
r17064 | return finishfold(ui, repo, ctx, oldctx, n, opts, []) | ||
def finishfold(ui, repo, ctx, oldctx, newnode, opts, internalchanges): | ||||
parent = ctx.parents()[0].node() | ||||
hg.update(repo, parent) | ||||
Pierre-Yves David
|
r17644 | ### prepare new commit data | ||
commitopts = opts.copy() | ||||
# username | ||||
if ctx.user() == oldctx.user(): | ||||
username = ctx.user() | ||||
else: | ||||
username = ui.username() | ||||
commitopts['user'] = username | ||||
# commit message | ||||
Augie Fackler
|
r17064 | newmessage = '\n***\n'.join( | ||
Augie Fackler
|
r17066 | [ctx.description()] + | ||
Augie Fackler
|
r17064 | [repo[r].description() for r in internalchanges] + | ||
Patrick Mezard
|
r17241 | [oldctx.description()]) + '\n' | ||
Pierre-Yves David
|
r17644 | commitopts['message'] = newmessage | ||
# date | ||||
commitopts['date'] = max(ctx.date(), oldctx.date()) | ||||
n = collapse(repo, ctx, repo[newnode], commitopts) | ||||
if n is None: | ||||
Pierre-Yves David
|
r17758 | return ctx, [] | ||
Pierre-Yves David
|
r17644 | hg.update(repo, n) | ||
Pierre-Yves David
|
r17758 | replacements = [(oldctx.node(), (newnode,)), | ||
(ctx.node(), (n,)), | ||||
(newnode, (n,)), | ||||
] | ||||
for ich in internalchanges: | ||||
replacements.append((ich, (n,))) | ||||
return repo[n], replacements | ||||
Augie Fackler
|
r17064 | |||
def drop(ui, repo, ctx, ha, opts): | ||||
Pierre-Yves David
|
r17758 | return ctx, [(repo[ha].node(), ())] | ||
Augie Fackler
|
r17064 | |||
def message(ui, repo, ctx, ha, opts): | ||||
oldctx = repo[ha] | ||||
hg.update(repo, ctx.node()) | ||||
Pierre-Yves David
|
r17647 | stats = applychanges(ui, repo, oldctx, opts) | ||
if stats and stats[3] > 0: | ||||
Augie Fackler
|
r17064 | raise util.Abort(_('Fix up the change and run ' | ||
'hg histedit --continue')) | ||||
Mads Kiilerich
|
r17285 | message = oldctx.description() + '\n' | ||
Augie Fackler
|
r17064 | message = ui.edit(message, ui.username()) | ||
new = repo.commit(text=message, user=oldctx.user(), date=oldctx.date(), | ||||
extra=oldctx.extra()) | ||||
newctx = repo[new] | ||||
if oldctx.node() != newctx.node(): | ||||
Pierre-Yves David
|
r17758 | return newctx, [(oldctx.node(), (new,))] | ||
Augie Fackler
|
r17064 | # We didn't make an edit, so just indicate no replaced nodes | ||
Pierre-Yves David
|
r17758 | return newctx, [] | ||
Augie Fackler
|
r17064 | |||
actiontable = {'p': pick, | ||||
'pick': pick, | ||||
'e': edit, | ||||
'edit': edit, | ||||
'f': fold, | ||||
'fold': fold, | ||||
'd': drop, | ||||
'drop': drop, | ||||
'm': message, | ||||
'mess': message, | ||||
} | ||||
Adrian Buehlmann
|
r17147 | |||
@command('histedit', | ||||
[('', 'commands', '', | ||||
_('Read history edits from the specified file.')), | ||||
('c', 'continue', False, _('continue an edit already in progress')), | ||||
('k', 'keep', False, | ||||
_("don't strip old nodes after edit is complete")), | ||||
('', 'abort', False, _('abort an edit in progress')), | ||||
('o', 'outgoing', False, _('changesets not found in destination')), | ||||
('f', 'force', False, | ||||
_('force outgoing even for unrelated repositories')), | ||||
('r', 'rev', [], _('first revision to be edited'))], | ||||
_("[PARENT]")) | ||||
Augie Fackler
|
r17064 | def histedit(ui, repo, *parent, **opts): | ||
Augie Fackler
|
r17131 | """interactively edit changeset history | ||
Augie Fackler
|
r17064 | """ | ||
# TODO only abort if we try and histedit mq patches, not just | ||||
# blanket if mq patches are applied somewhere | ||||
mq = getattr(repo, 'mq', None) | ||||
if mq and mq.applied: | ||||
raise util.Abort(_('source has mq patches applied')) | ||||
parent = list(parent) + opts.get('rev', []) | ||||
if opts.get('outgoing'): | ||||
if len(parent) > 1: | ||||
Augie Fackler
|
r17066 | raise util.Abort( | ||
_('only one repo argument allowed with --outgoing')) | ||||
Augie Fackler
|
r17064 | elif parent: | ||
parent = parent[0] | ||||
dest = ui.expandpath(parent or 'default-push', parent or 'default') | ||||
dest, revs = hg.parseurl(dest, None)[:2] | ||||
ui.status(_('comparing with %s\n') % util.hidepassword(dest)) | ||||
revs, checkout = hg.addbranchrevs(repo, repo, revs, None) | ||||
Sune Foldager
|
r17191 | other = hg.peer(repo, opts, dest) | ||
Augie Fackler
|
r17064 | |||
if revs: | ||||
revs = [repo.lookup(rev) for rev in revs] | ||||
parent = discovery.findcommonoutgoing( | ||||
repo, other, [], force=opts.get('force')).missing[0:1] | ||||
else: | ||||
if opts.get('force'): | ||||
raise util.Abort(_('--force only allowed with --outgoing')) | ||||
if opts.get('continue', False): | ||||
if len(parent) != 0: | ||||
raise util.Abort(_('no arguments allowed with --continue')) | ||||
Pierre-Yves David
|
r17758 | (parentctxnode, rules, keep, topmost, replacements) = readstate(repo) | ||
currentparent, wantnull = repo.dirstate.parents() | ||||
Augie Fackler
|
r17064 | parentctx = repo[parentctxnode] | ||
Pierre-Yves David
|
r17758 | parentctx, repl = bootstrapcontinue(ui, repo, parentctx, rules, opts) | ||
replacements.extend(repl) | ||||
Augie Fackler
|
r17064 | elif opts.get('abort', False): | ||
if len(parent) != 0: | ||||
raise util.Abort(_('no arguments allowed with --abort')) | ||||
Pierre-Yves David
|
r17758 | (parentctxnode, rules, keep, topmost, replacements) = readstate(repo) | ||
mapping, tmpnodes, leafs, _ntm = processreplacement(repo, replacements) | ||||
Pierre-Yves David
|
r17665 | ui.debug('restore wc to old parent %s\n' % node.short(topmost)) | ||
hg.clean(repo, topmost) | ||||
Pierre-Yves David
|
r17758 | cleanupnode(ui, repo, 'created', tmpnodes) | ||
cleanupnode(ui, repo, 'temp', leafs) | ||||
Augie Fackler
|
r17064 | os.unlink(os.path.join(repo.path, 'histedit-state')) | ||
return | ||||
else: | ||||
cmdutil.bailifchanged(repo) | ||||
if os.path.exists(os.path.join(repo.path, 'histedit-state')): | ||||
raise util.Abort(_('history edit already in progress, try ' | ||||
'--continue or --abort')) | ||||
Pierre-Yves David
|
r17665 | topmost, empty = repo.dirstate.parents() | ||
Augie Fackler
|
r17064 | |||
if len(parent) != 1: | ||||
raise util.Abort(_('histedit requires exactly one parent revision')) | ||||
parent = scmutil.revsingle(repo, parent[0]).node() | ||||
keep = opts.get('keep', False) | ||||
Pierre-Yves David
|
r17665 | revs = between(repo, parent, topmost, keep) | ||
Pierre-Yves David
|
r17766 | if not revs: | ||
ui.warn(_('nothing to edit\n')) | ||||
return 1 | ||||
Augie Fackler
|
r17064 | |||
ctxs = [repo[r] for r in revs] | ||||
rules = opts.get('commands', '') | ||||
if not rules: | ||||
rules = '\n'.join([makedesc(c) for c in ctxs]) | ||||
FUJIWARA Katsunori
|
r17315 | rules += '\n\n' | ||
Pierre-Yves David
|
r17665 | rules += editcomment % (node.short(parent), node.short(topmost)) | ||
Augie Fackler
|
r17064 | rules = ui.edit(rules, ui.username()) | ||
# Save edit rules in .hg/histedit-last-edit.txt in case | ||||
# the user needs to ask for help after something | ||||
# surprising happens. | ||||
f = open(repo.join('histedit-last-edit.txt'), 'w') | ||||
f.write(rules) | ||||
f.close() | ||||
else: | ||||
f = open(rules) | ||||
rules = f.read() | ||||
f.close() | ||||
rules = [l for l in (r.strip() for r in rules.splitlines()) | ||||
if l and not l[0] == '#'] | ||||
rules = verifyrules(rules, repo, ctxs) | ||||
parentctx = repo[parent].parents()[0] | ||||
keep = opts.get('keep', False) | ||||
Pierre-Yves David
|
r17758 | replacements = [] | ||
Augie Fackler
|
r17064 | |||
while rules: | ||||
Pierre-Yves David
|
r17758 | writestate(repo, parentctx.node(), rules, keep, topmost, replacements) | ||
Augie Fackler
|
r17064 | action, ha = rules.pop(0) | ||
Pierre-Yves David
|
r17645 | ui.debug('histedit: processing %s %s\n' % (action, ha)) | ||
Pierre-Yves David
|
r17758 | actfunc = actiontable[action] | ||
parentctx, replacement_ = actfunc(ui, repo, parentctx, ha, opts) | ||||
replacements.extend(replacement_) | ||||
Augie Fackler
|
r17064 | |||
hg.update(repo, parentctx.node()) | ||||
Pierre-Yves David
|
r17758 | mapping, tmpnodes, created, ntm = processreplacement(repo, replacements) | ||
if mapping: | ||||
for prec, succs in mapping.iteritems(): | ||||
if not succs: | ||||
ui.debug('histedit: %s is dropped\n' % node.short(prec)) | ||||
else: | ||||
ui.debug('histedit: %s is replaced by %s\n' % ( | ||||
node.short(prec), node.short(succs[0]))) | ||||
if len(succs) > 1: | ||||
m = 'histedit: %s' | ||||
for n in succs[1:]: | ||||
ui.debug(m % node.short(n)) | ||||
Augie Fackler
|
r17064 | if not keep: | ||
Pierre-Yves David
|
r17758 | if mapping: | ||
movebookmarks(ui, repo, mapping, topmost, ntm) | ||||
Pierre-Yves David
|
r17663 | # TODO update mq state | ||
Pierre-Yves David
|
r17759 | if obsolete._enabled: | ||
markers = [] | ||||
Pierre-Yves David
|
r17771 | # sort by revision number because it sound "right" | ||
for prec in sorted(mapping, key=repo.changelog.rev): | ||||
succs = mapping[prec] | ||||
Pierre-Yves David
|
r17759 | markers.append((repo[prec], | ||
tuple(repo[s] for s in succs))) | ||||
if markers: | ||||
obsolete.createmarkers(repo, markers) | ||||
else: | ||||
cleanupnode(ui, repo, 'replaced', mapping) | ||||
Augie Fackler
|
r17064 | |||
Pierre-Yves David
|
r17664 | cleanupnode(ui, repo, 'temp', tmpnodes) | ||
Augie Fackler
|
r17064 | os.unlink(os.path.join(repo.path, 'histedit-state')) | ||
if os.path.exists(repo.sjoin('undo')): | ||||
os.unlink(repo.sjoin('undo')) | ||||
Pierre-Yves David
|
r17758 | def bootstrapcontinue(ui, repo, parentctx, rules, opts): | ||
Pierre-Yves David
|
r17666 | action, currentnode = rules.pop(0) | ||
Pierre-Yves David
|
r17758 | ctx = repo[currentnode] | ||
Pierre-Yves David
|
r17749 | # is there any new commit between the expected parent and "." | ||
# | ||||
# note: does not take non linear new change in account (but previous | ||||
# implementation didn't used them anyway (issue3655) | ||||
newchildren = [c.node() for c in repo.set('(%d::.)', parentctx)] | ||||
if not newchildren: | ||||
# `parentctxnode` should match but no result. This means that | ||||
# currentnode is not a descendant from parentctxnode. | ||||
msg = _('working directory parent is not a descendant of %s') | ||||
hint = _('update to %s or descendant and run "hg histedit ' | ||||
'--continue" again') % parentctx | ||||
raise util.Abort(msg % parentctx, hint=hint) | ||||
newchildren.pop(0) # remove parentctxnode | ||||
Pierre-Yves David
|
r17758 | # Commit dirty working directory if necessary | ||
new = None | ||||
Pierre-Yves David
|
r17666 | m, a, r, d = repo.status()[:4] | ||
if m or a or r or d: | ||||
Pierre-Yves David
|
r17758 | # prepare the message for the commit to comes | ||
if action in ('f', 'fold'): | ||||
message = 'fold-temp-revision %s' % currentnode | ||||
else: | ||||
message = ctx.description() + '\n' | ||||
if action in ('e', 'edit', 'm', 'mess'): | ||||
editor = cmdutil.commitforceeditor | ||||
else: | ||||
editor = False | ||||
new = repo.commit(text=message, user=ctx.user(), | ||||
date=ctx.date(), extra=ctx.extra(), | ||||
editor=editor) | ||||
if new is not None: | ||||
newchildren.append(new) | ||||
Pierre-Yves David
|
r17666 | |||
Pierre-Yves David
|
r17758 | replacements = [] | ||
# track replacements | ||||
if ctx.node() not in newchildren: | ||||
# note: new children may be empty when the changeset is dropped. | ||||
# this happen e.g during conflicting pick where we revert content | ||||
# to parent. | ||||
replacements.append((ctx.node(), tuple(newchildren))) | ||||
if action in ('f', 'fold'): | ||||
# finalize fold operation if applicable | ||||
if new is None: | ||||
Pierre-Yves David
|
r17666 | new = newchildren[-1] | ||
Pierre-Yves David
|
r17758 | else: | ||
newchildren.pop() # remove new from internal changes | ||||
parentctx, repl = finishfold(ui, repo, parentctx, ctx, new, opts, | ||||
newchildren) | ||||
replacements.extend(repl) | ||||
elif newchildren: | ||||
# otherwize update "parentctx" before proceding to further operation | ||||
parentctx = repo[newchildren[-1]] | ||||
return parentctx, replacements | ||||
Pierre-Yves David
|
r17666 | |||
Pierre-Yves David
|
r17642 | def between(repo, old, new, keep): | ||
"""select and validate the set of revision to edit | ||||
When keep is false, the specified set can't have children.""" | ||||
Pierre-Yves David
|
r17765 | ctxs = list(repo.set('%n::%n', old, new)) | ||
Pierre-Yves David
|
r17766 | if ctxs and not keep: | ||
Pierre-Yves David
|
r17765 | if repo.revs('(%ld::) - (%ld + hidden())', ctxs, ctxs): | ||
Pierre-Yves David
|
r17762 | raise util.Abort(_('cannot edit history that would orphan nodes')) | ||
Pierre-Yves David
|
r17767 | root = ctxs[0] # list is already sorted by repo.set | ||
Pierre-Yves David
|
r17762 | if not root.phase(): | ||
raise util.Abort(_('cannot edit immutable changeset: %s') % root) | ||||
Pierre-Yves David
|
r17765 | return [c.node() for c in ctxs] | ||
Pierre-Yves David
|
r17642 | |||
Pierre-Yves David
|
r17758 | def writestate(repo, parentnode, rules, keep, topmost, replacements): | ||
Augie Fackler
|
r17064 | fp = open(os.path.join(repo.path, 'histedit-state'), 'w') | ||
Pierre-Yves David
|
r17758 | pickle.dump((parentnode, rules, keep, topmost, replacements), fp) | ||
Augie Fackler
|
r17064 | fp.close() | ||
def readstate(repo): | ||||
Pierre-Yves David
|
r17758 | """Returns a tuple of (parentnode, rules, keep, topmost, replacements). | ||
Augie Fackler
|
r17064 | """ | ||
fp = open(os.path.join(repo.path, 'histedit-state')) | ||||
return pickle.load(fp) | ||||
Pierre-Yves David
|
r17643 | def makedesc(c): | ||
"""build a initial action line for a ctx `c` | ||||
line are in the form: | ||||
pick <hash> <rev> <summary> | ||||
""" | ||||
summary = '' | ||||
if c.description(): | ||||
summary = c.description().splitlines()[0] | ||||
Pierre-Yves David
|
r17662 | line = 'pick %s %d %s' % (c, c.rev(), summary) | ||
Pierre-Yves David
|
r17643 | return line[:80] # trim to 80 chars so it's not stupidly wide in my editor | ||
Augie Fackler
|
r17064 | def verifyrules(rules, repo, ctxs): | ||
"""Verify that there exists exactly one edit rule per given changeset. | ||||
Will abort if there are to many or too few rules, a malformed rule, | ||||
or a rule on a changeset outside of the user-given range. | ||||
""" | ||||
parsed = [] | ||||
if len(rules) != len(ctxs): | ||||
raise util.Abort(_('must specify a rule for each changeset once')) | ||||
for r in rules: | ||||
if ' ' not in r: | ||||
raise util.Abort(_('malformed line "%s"') % r) | ||||
action, rest = r.split(' ', 1) | ||||
if ' ' in rest.strip(): | ||||
ha, rest = rest.split(' ', 1) | ||||
else: | ||||
ha = r.strip() | ||||
try: | ||||
if repo[ha] not in ctxs: | ||||
Augie Fackler
|
r17066 | raise util.Abort( | ||
_('may not use changesets other than the ones listed')) | ||||
Augie Fackler
|
r17064 | except error.RepoError: | ||
raise util.Abort(_('unknown changeset %s listed') % ha) | ||||
if action not in actiontable: | ||||
raise util.Abort(_('unknown action "%s"') % action) | ||||
parsed.append([action, ha]) | ||||
return parsed | ||||
Pierre-Yves David
|
r17663 | |||
Pierre-Yves David
|
r17758 | def processreplacement(repo, replacements): | ||
"""process the list of replacements to return | ||||
1) the final mapping between original and created nodes | ||||
2) the list of temporary node created by histedit | ||||
3) the list of new commit created by histedit""" | ||||
allsuccs = set() | ||||
replaced = set() | ||||
fullmapping = {} | ||||
# initialise basic set | ||||
# fullmapping record all operation recorded in replacement | ||||
for rep in replacements: | ||||
allsuccs.update(rep[1]) | ||||
replaced.add(rep[0]) | ||||
fullmapping.setdefault(rep[0], set()).update(rep[1]) | ||||
new = allsuccs - replaced | ||||
tmpnodes = allsuccs & replaced | ||||
# Reduce content fullmapping into direct relation between original nodes | ||||
# and final node created during history edition | ||||
# Dropped changeset are replaced by an empty list | ||||
toproceed = set(fullmapping) | ||||
final = {} | ||||
while toproceed: | ||||
for x in list(toproceed): | ||||
succs = fullmapping[x] | ||||
for s in list(succs): | ||||
if s in toproceed: | ||||
# non final node with unknown closure | ||||
# We can't process this now | ||||
break | ||||
elif s in final: | ||||
# non final node, replace with closure | ||||
succs.remove(s) | ||||
succs.update(final[s]) | ||||
else: | ||||
final[x] = succs | ||||
toproceed.remove(x) | ||||
# remove tmpnodes from final mapping | ||||
for n in tmpnodes: | ||||
del final[n] | ||||
# we expect all changes involved in final to exist in the repo | ||||
# turn `final` into list (topologically sorted) | ||||
nm = repo.changelog.nodemap | ||||
for prec, succs in final.items(): | ||||
final[prec] = sorted(succs, key=nm.get) | ||||
Pierre-Yves David
|
r17663 | |||
Pierre-Yves David
|
r17758 | # computed topmost element (necessary for bookmark) | ||
if new: | ||||
Thomas Arendsen Hein
|
r17769 | newtopmost = sorted(new, key=repo.changelog.rev)[-1] | ||
Pierre-Yves David
|
r17758 | elif not final: | ||
# Nothing rewritten at all. we won't need `newtopmost` | ||||
# It is the same as `oldtopmost` and `processreplacement` know it | ||||
newtopmost = None | ||||
else: | ||||
# every body died. The newtopmost is the parent of the root. | ||||
Thomas Arendsen Hein
|
r17769 | newtopmost = repo[sorted(final, key=repo.changelog.rev)[0]].p1().node() | ||
Pierre-Yves David
|
r17758 | |||
return final, tmpnodes, new, newtopmost | ||||
Pierre-Yves David
|
r17663 | |||
Pierre-Yves David
|
r17758 | def movebookmarks(ui, repo, mapping, oldtopmost, newtopmost): | ||
"""Move bookmark from old to newly created node""" | ||||
if not mapping: | ||||
# if nothing got rewritten there is not purpose for this function | ||||
return | ||||
moves = [] | ||||
for bk, old in repo._bookmarks.iteritems(): | ||||
if old == oldtopmost: | ||||
# special case ensure bookmark stay on tip. | ||||
# | ||||
# This is arguably a feature and we may only want that for the | ||||
# active bookmark. But the behavior is kept compatible with the old | ||||
# version for now. | ||||
moves.append((bk, newtopmost)) | ||||
continue | ||||
base = old | ||||
new = mapping.get(base, None) | ||||
if new is None: | ||||
continue | ||||
while not new: | ||||
# base is killed, trying with parent | ||||
base = repo[base].p1().node() | ||||
new = mapping.get(base, (base,)) | ||||
# nothing to move | ||||
moves.append((bk, new[-1])) | ||||
if moves: | ||||
for mark, new in moves: | ||||
old = repo._bookmarks[mark] | ||||
ui.note(_('histedit: moving bookmarks %s from %s to %s\n') | ||||
% (mark, node.short(old), node.short(new))) | ||||
repo._bookmarks[mark] = new | ||||
bookmarks.write(repo) | ||||
Pierre-Yves David
|
r17664 | |||
def cleanupnode(ui, repo, name, nodes): | ||||
"""strip a group of nodes from the repository | ||||
The set of node to strip may contains unknown nodes.""" | ||||
ui.debug('should strip %s nodes %s\n' % | ||||
(name, ', '.join([node.short(n) for n in nodes]))) | ||||
lock = None | ||||
try: | ||||
lock = repo.lock() | ||||
# Find all node that need to be stripped | ||||
# (we hg %lr instead of %ln to silently ignore unknown item | ||||
nm = repo.changelog.nodemap | ||||
nodes = [n for n in nodes if n in nm] | ||||
roots = [c.node() for c in repo.set("roots(%ln)", nodes)] | ||||
for c in roots: | ||||
# We should process node in reverse order to strip tip most first. | ||||
# but this trigger a bug in changegroup hook. | ||||
# This would reduce bundle overhead | ||||
repair.strip(ui, repo, c) | ||||
finally: | ||||
lockmod.release(lock) | ||||