##// END OF EJS Templates
histedit: break _histedit function into smaller pieces (add _newaction)...
Kostia Balytskyi -
r28132:2d09a400 default
parent child Browse files
Show More
@@ -1,1541 +1,1547 b''
1 1 # histedit.py - interactive history editing for mercurial
2 2 #
3 3 # Copyright 2009 Augie Fackler <raf@durin42.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7 """interactive history editing
8 8
9 9 With this extension installed, Mercurial gains one new command: histedit. Usage
10 10 is as follows, assuming the following history::
11 11
12 12 @ 3[tip] 7c2fd3b9020c 2009-04-27 18:04 -0500 durin42
13 13 | Add delta
14 14 |
15 15 o 2 030b686bedc4 2009-04-27 18:04 -0500 durin42
16 16 | Add gamma
17 17 |
18 18 o 1 c561b4e977df 2009-04-27 18:04 -0500 durin42
19 19 | Add beta
20 20 |
21 21 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
22 22 Add alpha
23 23
24 24 If you were to run ``hg histedit c561b4e977df``, you would see the following
25 25 file open in your editor::
26 26
27 27 pick c561b4e977df Add beta
28 28 pick 030b686bedc4 Add gamma
29 29 pick 7c2fd3b9020c Add delta
30 30
31 31 # Edit history between c561b4e977df and 7c2fd3b9020c
32 32 #
33 33 # Commits are listed from least to most recent
34 34 #
35 35 # Commands:
36 36 # p, pick = use commit
37 37 # e, edit = use commit, but stop for amending
38 38 # f, fold = use commit, but combine it with the one above
39 39 # r, roll = like fold, but discard this commit's description
40 40 # d, drop = remove commit from history
41 41 # m, mess = edit commit message without changing commit content
42 42 #
43 43
44 44 In this file, lines beginning with ``#`` are ignored. You must specify a rule
45 45 for each revision in your history. For example, if you had meant to add gamma
46 46 before beta, and then wanted to add delta in the same revision as beta, you
47 47 would reorganize the file to look like this::
48 48
49 49 pick 030b686bedc4 Add gamma
50 50 pick c561b4e977df Add beta
51 51 fold 7c2fd3b9020c Add delta
52 52
53 53 # Edit history between c561b4e977df and 7c2fd3b9020c
54 54 #
55 55 # Commits are listed from least to most recent
56 56 #
57 57 # Commands:
58 58 # p, pick = use commit
59 59 # e, edit = use commit, but stop for amending
60 60 # f, fold = use commit, but combine it with the one above
61 61 # r, roll = like fold, but discard this commit's description
62 62 # d, drop = remove commit from history
63 63 # m, mess = edit commit message without changing commit content
64 64 #
65 65
66 66 At which point you close the editor and ``histedit`` starts working. When you
67 67 specify a ``fold`` operation, ``histedit`` will open an editor when it folds
68 68 those revisions together, offering you a chance to clean up the commit message::
69 69
70 70 Add beta
71 71 ***
72 72 Add delta
73 73
74 74 Edit the commit message to your liking, then close the editor. For
75 75 this example, let's assume that the commit message was changed to
76 76 ``Add beta and delta.`` After histedit has run and had a chance to
77 77 remove any old or temporary revisions it needed, the history looks
78 78 like this::
79 79
80 80 @ 2[tip] 989b4d060121 2009-04-27 18:04 -0500 durin42
81 81 | Add beta and delta.
82 82 |
83 83 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
84 84 | Add gamma
85 85 |
86 86 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
87 87 Add alpha
88 88
89 89 Note that ``histedit`` does *not* remove any revisions (even its own temporary
90 90 ones) until after it has completed all the editing operations, so it will
91 91 probably perform several strip operations when it's done. For the above example,
92 92 it had to run strip twice. Strip can be slow depending on a variety of factors,
93 93 so you might need to be a little patient. You can choose to keep the original
94 94 revisions by passing the ``--keep`` flag.
95 95
96 96 The ``edit`` operation will drop you back to a command prompt,
97 97 allowing you to edit files freely, or even use ``hg record`` to commit
98 98 some changes as a separate commit. When you're done, any remaining
99 99 uncommitted changes will be committed as well. When done, run ``hg
100 100 histedit --continue`` to finish this step. You'll be prompted for a
101 101 new commit message, but the default commit message will be the
102 102 original message for the ``edit`` ed revision.
103 103
104 104 The ``message`` operation will give you a chance to revise a commit
105 105 message without changing the contents. It's a shortcut for doing
106 106 ``edit`` immediately followed by `hg histedit --continue``.
107 107
108 108 If ``histedit`` encounters a conflict when moving a revision (while
109 109 handling ``pick`` or ``fold``), it'll stop in a similar manner to
110 110 ``edit`` with the difference that it won't prompt you for a commit
111 111 message when done. If you decide at this point that you don't like how
112 112 much work it will be to rearrange history, or that you made a mistake,
113 113 you can use ``hg histedit --abort`` to abandon the new changes you
114 114 have made and return to the state before you attempted to edit your
115 115 history.
116 116
117 117 If we clone the histedit-ed example repository above and add four more
118 118 changes, such that we have the following history::
119 119
120 120 @ 6[tip] 038383181893 2009-04-27 18:04 -0500 stefan
121 121 | Add theta
122 122 |
123 123 o 5 140988835471 2009-04-27 18:04 -0500 stefan
124 124 | Add eta
125 125 |
126 126 o 4 122930637314 2009-04-27 18:04 -0500 stefan
127 127 | Add zeta
128 128 |
129 129 o 3 836302820282 2009-04-27 18:04 -0500 stefan
130 130 | Add epsilon
131 131 |
132 132 o 2 989b4d060121 2009-04-27 18:04 -0500 durin42
133 133 | Add beta and delta.
134 134 |
135 135 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
136 136 | Add gamma
137 137 |
138 138 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
139 139 Add alpha
140 140
141 141 If you run ``hg histedit --outgoing`` on the clone then it is the same
142 142 as running ``hg histedit 836302820282``. If you need plan to push to a
143 143 repository that Mercurial does not detect to be related to the source
144 144 repo, you can add a ``--force`` option.
145 145
146 146 Config
147 147 ------
148 148
149 149 Histedit rule lines are truncated to 80 characters by default. You
150 150 can customize this behavior by setting a different length in your
151 151 configuration file::
152 152
153 153 [histedit]
154 154 linelen = 120 # truncate rule lines at 120 characters
155 155
156 156 ``hg histedit`` attempts to automatically choose an appropriate base
157 157 revision to use. To change which base revision is used, define a
158 158 revset in your configuration file::
159 159
160 160 [histedit]
161 161 defaultrev = only(.) & draft()
162 162
163 163 By default each edited revision needs to be present in histedit commands.
164 164 To remove revision you need to use ``drop`` operation. You can configure
165 165 the drop to be implicit for missing commits by adding::
166 166
167 167 [histedit]
168 168 dropmissing = True
169 169
170 170 """
171 171
172 172 import pickle
173 173 import errno
174 174 import os
175 175 import sys
176 176
177 177 from mercurial import bundle2
178 178 from mercurial import cmdutil
179 179 from mercurial import discovery
180 180 from mercurial import error
181 181 from mercurial import copies
182 182 from mercurial import context
183 183 from mercurial import destutil
184 184 from mercurial import exchange
185 185 from mercurial import extensions
186 186 from mercurial import hg
187 187 from mercurial import node
188 188 from mercurial import repair
189 189 from mercurial import scmutil
190 190 from mercurial import util
191 191 from mercurial import obsolete
192 192 from mercurial import merge as mergemod
193 193 from mercurial.lock import release
194 194 from mercurial.i18n import _
195 195
196 196 cmdtable = {}
197 197 command = cmdutil.command(cmdtable)
198 198
199 199 class _constraints(object):
200 200 # aborts if there are multiple rules for one node
201 201 noduplicates = 'noduplicates'
202 202 # abort if the node does belong to edited stack
203 203 forceother = 'forceother'
204 204 # abort if the node doesn't belong to edited stack
205 205 noother = 'noother'
206 206
207 207 @classmethod
208 208 def known(cls):
209 209 return set([v for k, v in cls.__dict__.items() if k[0] != '_'])
210 210
211 211 # Note for extension authors: ONLY specify testedwith = 'internal' for
212 212 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
213 213 # be specifying the version(s) of Mercurial they are tested with, or
214 214 # leave the attribute unspecified.
215 215 testedwith = 'internal'
216 216
217 217 actiontable = {}
218 218 primaryactions = set()
219 219 secondaryactions = set()
220 220 tertiaryactions = set()
221 221 internalactions = set()
222 222
223 223 def geteditcomment(first, last):
224 224 """ construct the editor comment
225 225 The comment includes::
226 226 - an intro
227 227 - sorted primary commands
228 228 - sorted short commands
229 229 - sorted long commands
230 230
231 231 Commands are only included once.
232 232 """
233 233 intro = _("""Edit history between %s and %s
234 234
235 235 Commits are listed from least to most recent
236 236
237 237 Commands:
238 238 """)
239 239 actions = []
240 240 def addverb(v):
241 241 a = actiontable[v]
242 242 lines = a.message.split("\n")
243 243 if len(a.verbs):
244 244 v = ', '.join(sorted(a.verbs, key=lambda v: len(v)))
245 245 actions.append(" %s = %s" % (v, lines[0]))
246 246 actions.extend([' %s' for l in lines[1:]])
247 247
248 248 for v in (
249 249 sorted(primaryactions) +
250 250 sorted(secondaryactions) +
251 251 sorted(tertiaryactions)
252 252 ):
253 253 addverb(v)
254 254 actions.append('')
255 255
256 256 return ''.join(['# %s\n' % l if l else '#\n'
257 257 for l in ((intro % (first, last)).split('\n')) + actions])
258 258
259 259 class histeditstate(object):
260 260 def __init__(self, repo, parentctxnode=None, actions=None, keep=None,
261 261 topmost=None, replacements=None, lock=None, wlock=None):
262 262 self.repo = repo
263 263 self.actions = actions
264 264 self.keep = keep
265 265 self.topmost = topmost
266 266 self.parentctxnode = parentctxnode
267 267 self.lock = lock
268 268 self.wlock = wlock
269 269 self.backupfile = None
270 270 if replacements is None:
271 271 self.replacements = []
272 272 else:
273 273 self.replacements = replacements
274 274
275 275 def read(self):
276 276 """Load histedit state from disk and set fields appropriately."""
277 277 try:
278 278 state = self.repo.vfs.read('histedit-state')
279 279 except IOError as err:
280 280 if err.errno != errno.ENOENT:
281 281 raise
282 282 cmdutil.wrongtooltocontinue(self.repo, _('histedit'))
283 283
284 284 if state.startswith('v1\n'):
285 285 data = self._load()
286 286 parentctxnode, rules, keep, topmost, replacements, backupfile = data
287 287 else:
288 288 data = pickle.loads(state)
289 289 parentctxnode, rules, keep, topmost, replacements = data
290 290 backupfile = None
291 291
292 292 self.parentctxnode = parentctxnode
293 293 rules = "\n".join(["%s %s" % (verb, rest) for [verb, rest] in rules])
294 294 actions = parserules(rules, self)
295 295 self.actions = actions
296 296 self.keep = keep
297 297 self.topmost = topmost
298 298 self.replacements = replacements
299 299 self.backupfile = backupfile
300 300
301 301 def write(self):
302 302 fp = self.repo.vfs('histedit-state', 'w')
303 303 fp.write('v1\n')
304 304 fp.write('%s\n' % node.hex(self.parentctxnode))
305 305 fp.write('%s\n' % node.hex(self.topmost))
306 306 fp.write('%s\n' % self.keep)
307 307 fp.write('%d\n' % len(self.actions))
308 308 for action in self.actions:
309 309 fp.write('%s\n' % action.tostate())
310 310 fp.write('%d\n' % len(self.replacements))
311 311 for replacement in self.replacements:
312 312 fp.write('%s%s\n' % (node.hex(replacement[0]), ''.join(node.hex(r)
313 313 for r in replacement[1])))
314 314 backupfile = self.backupfile
315 315 if not backupfile:
316 316 backupfile = ''
317 317 fp.write('%s\n' % backupfile)
318 318 fp.close()
319 319
320 320 def _load(self):
321 321 fp = self.repo.vfs('histedit-state', 'r')
322 322 lines = [l[:-1] for l in fp.readlines()]
323 323
324 324 index = 0
325 325 lines[index] # version number
326 326 index += 1
327 327
328 328 parentctxnode = node.bin(lines[index])
329 329 index += 1
330 330
331 331 topmost = node.bin(lines[index])
332 332 index += 1
333 333
334 334 keep = lines[index] == 'True'
335 335 index += 1
336 336
337 337 # Rules
338 338 rules = []
339 339 rulelen = int(lines[index])
340 340 index += 1
341 341 for i in xrange(rulelen):
342 342 ruleaction = lines[index]
343 343 index += 1
344 344 rule = lines[index]
345 345 index += 1
346 346 rules.append((ruleaction, rule))
347 347
348 348 # Replacements
349 349 replacements = []
350 350 replacementlen = int(lines[index])
351 351 index += 1
352 352 for i in xrange(replacementlen):
353 353 replacement = lines[index]
354 354 original = node.bin(replacement[:40])
355 355 succ = [node.bin(replacement[i:i + 40]) for i in
356 356 range(40, len(replacement), 40)]
357 357 replacements.append((original, succ))
358 358 index += 1
359 359
360 360 backupfile = lines[index]
361 361 index += 1
362 362
363 363 fp.close()
364 364
365 365 return parentctxnode, rules, keep, topmost, replacements, backupfile
366 366
367 367 def clear(self):
368 368 if self.inprogress():
369 369 self.repo.vfs.unlink('histedit-state')
370 370
371 371 def inprogress(self):
372 372 return self.repo.vfs.exists('histedit-state')
373 373
374 374
375 375 class histeditaction(object):
376 376 def __init__(self, state, node):
377 377 self.state = state
378 378 self.repo = state.repo
379 379 self.node = node
380 380
381 381 @classmethod
382 382 def fromrule(cls, state, rule):
383 383 """Parses the given rule, returning an instance of the histeditaction.
384 384 """
385 385 rulehash = rule.strip().split(' ', 1)[0]
386 386 try:
387 387 rev = node.bin(rulehash)
388 388 except TypeError:
389 389 raise error.ParseError("invalid changeset %s" % rulehash)
390 390 return cls(state, rev)
391 391
392 392 def verify(self, prev):
393 393 """ Verifies semantic correctness of the rule"""
394 394 repo = self.repo
395 395 ha = node.hex(self.node)
396 396 try:
397 397 self.node = repo[ha].node()
398 398 except error.RepoError:
399 399 raise error.ParseError(_('unknown changeset %s listed')
400 400 % ha[:12])
401 401
402 402 def torule(self):
403 403 """build a histedit rule line for an action
404 404
405 405 by default lines are in the form:
406 406 <hash> <rev> <summary>
407 407 """
408 408 ctx = self.repo[self.node]
409 409 summary = ''
410 410 if ctx.description():
411 411 summary = ctx.description().splitlines()[0]
412 412 line = '%s %s %d %s' % (self.verb, ctx, ctx.rev(), summary)
413 413 # trim to 75 columns by default so it's not stupidly wide in my editor
414 414 # (the 5 more are left for verb)
415 415 maxlen = self.repo.ui.configint('histedit', 'linelen', default=80)
416 416 maxlen = max(maxlen, 22) # avoid truncating hash
417 417 return util.ellipsis(line, maxlen)
418 418
419 419 def tostate(self):
420 420 """Print an action in format used by histedit state files
421 421 (the first line is a verb, the remainder is the second)
422 422 """
423 423 return "%s\n%s" % (self.verb, node.hex(self.node))
424 424
425 425 def constraints(self):
426 426 """Return a set of constrains that this action should be verified for
427 427 """
428 428 return set([_constraints.noduplicates, _constraints.noother])
429 429
430 430 def nodetoverify(self):
431 431 """Returns a node associated with the action that will be used for
432 432 verification purposes.
433 433
434 434 If the action doesn't correspond to node it should return None
435 435 """
436 436 return self.node
437 437
438 438 def run(self):
439 439 """Runs the action. The default behavior is simply apply the action's
440 440 rulectx onto the current parentctx."""
441 441 self.applychange()
442 442 self.continuedirty()
443 443 return self.continueclean()
444 444
445 445 def applychange(self):
446 446 """Applies the changes from this action's rulectx onto the current
447 447 parentctx, but does not commit them."""
448 448 repo = self.repo
449 449 rulectx = repo[self.node]
450 450 repo.ui.pushbuffer(error=True, labeled=True)
451 451 hg.update(repo, self.state.parentctxnode, quietempty=True)
452 452 stats = applychanges(repo.ui, repo, rulectx, {})
453 453 if stats and stats[3] > 0:
454 454 buf = repo.ui.popbuffer()
455 455 repo.ui.write(*buf)
456 456 raise error.InterventionRequired(
457 457 _('Fix up the change (%s %s)') %
458 458 (self.verb, node.short(self.node)),
459 459 hint=_('hg histedit --continue to resume'))
460 460 else:
461 461 repo.ui.popbuffer()
462 462
463 463 def continuedirty(self):
464 464 """Continues the action when changes have been applied to the working
465 465 copy. The default behavior is to commit the dirty changes."""
466 466 repo = self.repo
467 467 rulectx = repo[self.node]
468 468
469 469 editor = self.commiteditor()
470 470 commit = commitfuncfor(repo, rulectx)
471 471
472 472 commit(text=rulectx.description(), user=rulectx.user(),
473 473 date=rulectx.date(), extra=rulectx.extra(), editor=editor)
474 474
475 475 def commiteditor(self):
476 476 """The editor to be used to edit the commit message."""
477 477 return False
478 478
479 479 def continueclean(self):
480 480 """Continues the action when the working copy is clean. The default
481 481 behavior is to accept the current commit as the new version of the
482 482 rulectx."""
483 483 ctx = self.repo['.']
484 484 if ctx.node() == self.state.parentctxnode:
485 485 self.repo.ui.warn(_('%s: empty changeset\n') %
486 486 node.short(self.node))
487 487 return ctx, [(self.node, tuple())]
488 488 if ctx.node() == self.node:
489 489 # Nothing changed
490 490 return ctx, []
491 491 return ctx, [(self.node, (ctx.node(),))]
492 492
493 493 def commitfuncfor(repo, src):
494 494 """Build a commit function for the replacement of <src>
495 495
496 496 This function ensure we apply the same treatment to all changesets.
497 497
498 498 - Add a 'histedit_source' entry in extra.
499 499
500 500 Note that fold has its own separated logic because its handling is a bit
501 501 different and not easily factored out of the fold method.
502 502 """
503 503 phasemin = src.phase()
504 504 def commitfunc(**kwargs):
505 505 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
506 506 try:
507 507 repo.ui.setconfig('phases', 'new-commit', phasemin,
508 508 'histedit')
509 509 extra = kwargs.get('extra', {}).copy()
510 510 extra['histedit_source'] = src.hex()
511 511 kwargs['extra'] = extra
512 512 return repo.commit(**kwargs)
513 513 finally:
514 514 repo.ui.restoreconfig(phasebackup)
515 515 return commitfunc
516 516
517 517 def applychanges(ui, repo, ctx, opts):
518 518 """Merge changeset from ctx (only) in the current working directory"""
519 519 wcpar = repo.dirstate.parents()[0]
520 520 if ctx.p1().node() == wcpar:
521 521 # edits are "in place" we do not need to make any merge,
522 522 # just applies changes on parent for editing
523 523 cmdutil.revert(ui, repo, ctx, (wcpar, node.nullid), all=True)
524 524 stats = None
525 525 else:
526 526 try:
527 527 # ui.forcemerge is an internal variable, do not document
528 528 repo.ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
529 529 'histedit')
530 530 stats = mergemod.graft(repo, ctx, ctx.p1(), ['local', 'histedit'])
531 531 finally:
532 532 repo.ui.setconfig('ui', 'forcemerge', '', 'histedit')
533 533 return stats
534 534
535 535 def collapse(repo, first, last, commitopts, skipprompt=False):
536 536 """collapse the set of revisions from first to last as new one.
537 537
538 538 Expected commit options are:
539 539 - message
540 540 - date
541 541 - username
542 542 Commit message is edited in all cases.
543 543
544 544 This function works in memory."""
545 545 ctxs = list(repo.set('%d::%d', first, last))
546 546 if not ctxs:
547 547 return None
548 548 for c in ctxs:
549 549 if not c.mutable():
550 550 raise error.ParseError(
551 551 _("cannot fold into public change %s") % node.short(c.node()))
552 552 base = first.parents()[0]
553 553
554 554 # commit a new version of the old changeset, including the update
555 555 # collect all files which might be affected
556 556 files = set()
557 557 for ctx in ctxs:
558 558 files.update(ctx.files())
559 559
560 560 # Recompute copies (avoid recording a -> b -> a)
561 561 copied = copies.pathcopies(base, last)
562 562
563 563 # prune files which were reverted by the updates
564 564 def samefile(f):
565 565 if f in last.manifest():
566 566 a = last.filectx(f)
567 567 if f in base.manifest():
568 568 b = base.filectx(f)
569 569 return (a.data() == b.data()
570 570 and a.flags() == b.flags())
571 571 else:
572 572 return False
573 573 else:
574 574 return f not in base.manifest()
575 575 files = [f for f in files if not samefile(f)]
576 576 # commit version of these files as defined by head
577 577 headmf = last.manifest()
578 578 def filectxfn(repo, ctx, path):
579 579 if path in headmf:
580 580 fctx = last[path]
581 581 flags = fctx.flags()
582 582 mctx = context.memfilectx(repo,
583 583 fctx.path(), fctx.data(),
584 584 islink='l' in flags,
585 585 isexec='x' in flags,
586 586 copied=copied.get(path))
587 587 return mctx
588 588 return None
589 589
590 590 if commitopts.get('message'):
591 591 message = commitopts['message']
592 592 else:
593 593 message = first.description()
594 594 user = commitopts.get('user')
595 595 date = commitopts.get('date')
596 596 extra = commitopts.get('extra')
597 597
598 598 parents = (first.p1().node(), first.p2().node())
599 599 editor = None
600 600 if not skipprompt:
601 601 editor = cmdutil.getcommiteditor(edit=True, editform='histedit.fold')
602 602 new = context.memctx(repo,
603 603 parents=parents,
604 604 text=message,
605 605 files=files,
606 606 filectxfn=filectxfn,
607 607 user=user,
608 608 date=date,
609 609 extra=extra,
610 610 editor=editor)
611 611 return repo.commitctx(new)
612 612
613 613 def _isdirtywc(repo):
614 614 return repo[None].dirty(missing=True)
615 615
616 616 def abortdirty():
617 617 raise error.Abort(_('working copy has pending changes'),
618 618 hint=_('amend, commit, or revert them and run histedit '
619 619 '--continue, or abort with histedit --abort'))
620 620
621 621 def action(verbs, message, priority=False, internal=False):
622 622 def wrap(cls):
623 623 assert not priority or not internal
624 624 verb = verbs[0]
625 625 if priority:
626 626 primaryactions.add(verb)
627 627 elif internal:
628 628 internalactions.add(verb)
629 629 elif len(verbs) > 1:
630 630 secondaryactions.add(verb)
631 631 else:
632 632 tertiaryactions.add(verb)
633 633
634 634 cls.verb = verb
635 635 cls.verbs = verbs
636 636 cls.message = message
637 637 for verb in verbs:
638 638 actiontable[verb] = cls
639 639 return cls
640 640 return wrap
641 641
642 642 @action(['pick', 'p'],
643 643 _('use commit'),
644 644 priority=True)
645 645 class pick(histeditaction):
646 646 def run(self):
647 647 rulectx = self.repo[self.node]
648 648 if rulectx.parents()[0].node() == self.state.parentctxnode:
649 649 self.repo.ui.debug('node %s unchanged\n' % node.short(self.node))
650 650 return rulectx, []
651 651
652 652 return super(pick, self).run()
653 653
654 654 @action(['edit', 'e'],
655 655 _('use commit, but stop for amending'),
656 656 priority=True)
657 657 class edit(histeditaction):
658 658 def run(self):
659 659 repo = self.repo
660 660 rulectx = repo[self.node]
661 661 hg.update(repo, self.state.parentctxnode, quietempty=True)
662 662 applychanges(repo.ui, repo, rulectx, {})
663 663 raise error.InterventionRequired(
664 664 _('Editing (%s), you may commit or record as needed now.')
665 665 % node.short(self.node),
666 666 hint=_('hg histedit --continue to resume'))
667 667
668 668 def commiteditor(self):
669 669 return cmdutil.getcommiteditor(edit=True, editform='histedit.edit')
670 670
671 671 @action(['fold', 'f'],
672 672 _('use commit, but combine it with the one above'))
673 673 class fold(histeditaction):
674 674 def verify(self, prev):
675 675 """ Verifies semantic correctness of the fold rule"""
676 676 super(fold, self).verify(prev)
677 677 repo = self.repo
678 678 if not prev:
679 679 c = repo[self.node].parents()[0]
680 680 elif not prev.verb in ('pick', 'base'):
681 681 return
682 682 else:
683 683 c = repo[prev.node]
684 684 if not c.mutable():
685 685 raise error.ParseError(
686 686 _("cannot fold into public change %s") % node.short(c.node()))
687 687
688 688
689 689 def continuedirty(self):
690 690 repo = self.repo
691 691 rulectx = repo[self.node]
692 692
693 693 commit = commitfuncfor(repo, rulectx)
694 694 commit(text='fold-temp-revision %s' % node.short(self.node),
695 695 user=rulectx.user(), date=rulectx.date(),
696 696 extra=rulectx.extra())
697 697
698 698 def continueclean(self):
699 699 repo = self.repo
700 700 ctx = repo['.']
701 701 rulectx = repo[self.node]
702 702 parentctxnode = self.state.parentctxnode
703 703 if ctx.node() == parentctxnode:
704 704 repo.ui.warn(_('%s: empty changeset\n') %
705 705 node.short(self.node))
706 706 return ctx, [(self.node, (parentctxnode,))]
707 707
708 708 parentctx = repo[parentctxnode]
709 709 newcommits = set(c.node() for c in repo.set('(%d::. - %d)', parentctx,
710 710 parentctx))
711 711 if not newcommits:
712 712 repo.ui.warn(_('%s: cannot fold - working copy is not a '
713 713 'descendant of previous commit %s\n') %
714 714 (node.short(self.node), node.short(parentctxnode)))
715 715 return ctx, [(self.node, (ctx.node(),))]
716 716
717 717 middlecommits = newcommits.copy()
718 718 middlecommits.discard(ctx.node())
719 719
720 720 return self.finishfold(repo.ui, repo, parentctx, rulectx, ctx.node(),
721 721 middlecommits)
722 722
723 723 def skipprompt(self):
724 724 """Returns true if the rule should skip the message editor.
725 725
726 726 For example, 'fold' wants to show an editor, but 'rollup'
727 727 doesn't want to.
728 728 """
729 729 return False
730 730
731 731 def mergedescs(self):
732 732 """Returns true if the rule should merge messages of multiple changes.
733 733
734 734 This exists mainly so that 'rollup' rules can be a subclass of
735 735 'fold'.
736 736 """
737 737 return True
738 738
739 739 def finishfold(self, ui, repo, ctx, oldctx, newnode, internalchanges):
740 740 parent = ctx.parents()[0].node()
741 741 repo.ui.pushbuffer()
742 742 hg.update(repo, parent)
743 743 repo.ui.popbuffer()
744 744 ### prepare new commit data
745 745 commitopts = {}
746 746 commitopts['user'] = ctx.user()
747 747 # commit message
748 748 if not self.mergedescs():
749 749 newmessage = ctx.description()
750 750 else:
751 751 newmessage = '\n***\n'.join(
752 752 [ctx.description()] +
753 753 [repo[r].description() for r in internalchanges] +
754 754 [oldctx.description()]) + '\n'
755 755 commitopts['message'] = newmessage
756 756 # date
757 757 commitopts['date'] = max(ctx.date(), oldctx.date())
758 758 extra = ctx.extra().copy()
759 759 # histedit_source
760 760 # note: ctx is likely a temporary commit but that the best we can do
761 761 # here. This is sufficient to solve issue3681 anyway.
762 762 extra['histedit_source'] = '%s,%s' % (ctx.hex(), oldctx.hex())
763 763 commitopts['extra'] = extra
764 764 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
765 765 try:
766 766 phasemin = max(ctx.phase(), oldctx.phase())
767 767 repo.ui.setconfig('phases', 'new-commit', phasemin, 'histedit')
768 768 n = collapse(repo, ctx, repo[newnode], commitopts,
769 769 skipprompt=self.skipprompt())
770 770 finally:
771 771 repo.ui.restoreconfig(phasebackup)
772 772 if n is None:
773 773 return ctx, []
774 774 repo.ui.pushbuffer()
775 775 hg.update(repo, n)
776 776 repo.ui.popbuffer()
777 777 replacements = [(oldctx.node(), (newnode,)),
778 778 (ctx.node(), (n,)),
779 779 (newnode, (n,)),
780 780 ]
781 781 for ich in internalchanges:
782 782 replacements.append((ich, (n,)))
783 783 return repo[n], replacements
784 784
785 785 class base(histeditaction):
786 786 def constraints(self):
787 787 return set([_constraints.forceother])
788 788
789 789 def run(self):
790 790 if self.repo['.'].node() != self.node:
791 791 mergemod.update(self.repo, self.node, False, True)
792 792 # branchmerge, force)
793 793 return self.continueclean()
794 794
795 795 def continuedirty(self):
796 796 abortdirty()
797 797
798 798 def continueclean(self):
799 799 basectx = self.repo['.']
800 800 return basectx, []
801 801
802 802 @action(['_multifold'],
803 803 _(
804 804 """fold subclass used for when multiple folds happen in a row
805 805
806 806 We only want to fire the editor for the folded message once when
807 807 (say) four changes are folded down into a single change. This is
808 808 similar to rollup, but we should preserve both messages so that
809 809 when the last fold operation runs we can show the user all the
810 810 commit messages in their editor.
811 811 """),
812 812 internal=True)
813 813 class _multifold(fold):
814 814 def skipprompt(self):
815 815 return True
816 816
817 817 @action(["roll", "r"],
818 818 _("like fold, but discard this commit's description"))
819 819 class rollup(fold):
820 820 def mergedescs(self):
821 821 return False
822 822
823 823 def skipprompt(self):
824 824 return True
825 825
826 826 @action(["drop", "d"],
827 827 _('remove commit from history'))
828 828 class drop(histeditaction):
829 829 def run(self):
830 830 parentctx = self.repo[self.state.parentctxnode]
831 831 return parentctx, [(self.node, tuple())]
832 832
833 833 @action(["mess", "m"],
834 834 _('edit commit message without changing commit content'),
835 835 priority=True)
836 836 class message(histeditaction):
837 837 def commiteditor(self):
838 838 return cmdutil.getcommiteditor(edit=True, editform='histedit.mess')
839 839
840 840 def findoutgoing(ui, repo, remote=None, force=False, opts=None):
841 841 """utility function to find the first outgoing changeset
842 842
843 843 Used by initialization code"""
844 844 if opts is None:
845 845 opts = {}
846 846 dest = ui.expandpath(remote or 'default-push', remote or 'default')
847 847 dest, revs = hg.parseurl(dest, None)[:2]
848 848 ui.status(_('comparing with %s\n') % util.hidepassword(dest))
849 849
850 850 revs, checkout = hg.addbranchrevs(repo, repo, revs, None)
851 851 other = hg.peer(repo, opts, dest)
852 852
853 853 if revs:
854 854 revs = [repo.lookup(rev) for rev in revs]
855 855
856 856 outgoing = discovery.findcommonoutgoing(repo, other, revs, force=force)
857 857 if not outgoing.missing:
858 858 raise error.Abort(_('no outgoing ancestors'))
859 859 roots = list(repo.revs("roots(%ln)", outgoing.missing))
860 860 if 1 < len(roots):
861 861 msg = _('there are ambiguous outgoing revisions')
862 862 hint = _('see "hg help histedit" for more detail')
863 863 raise error.Abort(msg, hint=hint)
864 864 return repo.lookup(roots[0])
865 865
866 866
867 867 @command('histedit',
868 868 [('', 'commands', '',
869 869 _('read history edits from the specified file'), _('FILE')),
870 870 ('c', 'continue', False, _('continue an edit already in progress')),
871 871 ('', 'edit-plan', False, _('edit remaining actions list')),
872 872 ('k', 'keep', False,
873 873 _("don't strip old nodes after edit is complete")),
874 874 ('', 'abort', False, _('abort an edit in progress')),
875 875 ('o', 'outgoing', False, _('changesets not found in destination')),
876 876 ('f', 'force', False,
877 877 _('force outgoing even for unrelated repositories')),
878 878 ('r', 'rev', [], _('first revision to be edited'), _('REV'))],
879 879 _("[OPTIONS] ([ANCESTOR] | --outgoing [URL])"))
880 880 def histedit(ui, repo, *freeargs, **opts):
881 881 """interactively edit changeset history
882 882
883 883 This command lets you edit a linear series of changesets (up to
884 884 and including the working directory, which should be clean).
885 885 You can:
886 886
887 887 - `pick` to [re]order a changeset
888 888
889 889 - `drop` to omit changeset
890 890
891 891 - `mess` to reword the changeset commit message
892 892
893 893 - `fold` to combine it with the preceding changeset
894 894
895 895 - `roll` like fold, but discarding this commit's description
896 896
897 897 - `edit` to edit this changeset
898 898
899 899 There are a number of ways to select the root changeset:
900 900
901 901 - Specify ANCESTOR directly
902 902
903 903 - Use --outgoing -- it will be the first linear changeset not
904 904 included in destination. (See :hg:`help config.paths.default-push`)
905 905
906 906 - Otherwise, the value from the "histedit.defaultrev" config option
907 907 is used as a revset to select the base revision when ANCESTOR is not
908 908 specified. The first revision returned by the revset is used. By
909 909 default, this selects the editable history that is unique to the
910 910 ancestry of the working directory.
911 911
912 912 .. container:: verbose
913 913
914 914 If you use --outgoing, this command will abort if there are ambiguous
915 915 outgoing revisions. For example, if there are multiple branches
916 916 containing outgoing revisions.
917 917
918 918 Use "min(outgoing() and ::.)" or similar revset specification
919 919 instead of --outgoing to specify edit target revision exactly in
920 920 such ambiguous situation. See :hg:`help revsets` for detail about
921 921 selecting revisions.
922 922
923 923 .. container:: verbose
924 924
925 925 Examples:
926 926
927 927 - A number of changes have been made.
928 928 Revision 3 is no longer needed.
929 929
930 930 Start history editing from revision 3::
931 931
932 932 hg histedit -r 3
933 933
934 934 An editor opens, containing the list of revisions,
935 935 with specific actions specified::
936 936
937 937 pick 5339bf82f0ca 3 Zworgle the foobar
938 938 pick 8ef592ce7cc4 4 Bedazzle the zerlog
939 939 pick 0a9639fcda9d 5 Morgify the cromulancy
940 940
941 941 Additional information about the possible actions
942 942 to take appears below the list of revisions.
943 943
944 944 To remove revision 3 from the history,
945 945 its action (at the beginning of the relevant line)
946 946 is changed to 'drop'::
947 947
948 948 drop 5339bf82f0ca 3 Zworgle the foobar
949 949 pick 8ef592ce7cc4 4 Bedazzle the zerlog
950 950 pick 0a9639fcda9d 5 Morgify the cromulancy
951 951
952 952 - A number of changes have been made.
953 953 Revision 2 and 4 need to be swapped.
954 954
955 955 Start history editing from revision 2::
956 956
957 957 hg histedit -r 2
958 958
959 959 An editor opens, containing the list of revisions,
960 960 with specific actions specified::
961 961
962 962 pick 252a1af424ad 2 Blorb a morgwazzle
963 963 pick 5339bf82f0ca 3 Zworgle the foobar
964 964 pick 8ef592ce7cc4 4 Bedazzle the zerlog
965 965
966 966 To swap revision 2 and 4, its lines are swapped
967 967 in the editor::
968 968
969 969 pick 8ef592ce7cc4 4 Bedazzle the zerlog
970 970 pick 5339bf82f0ca 3 Zworgle the foobar
971 971 pick 252a1af424ad 2 Blorb a morgwazzle
972 972
973 973 Returns 0 on success, 1 if user intervention is required (not only
974 974 for intentional "edit" command, but also for resolving unexpected
975 975 conflicts).
976 976 """
977 977 state = histeditstate(repo)
978 978 try:
979 979 state.wlock = repo.wlock()
980 980 state.lock = repo.lock()
981 981 _histedit(ui, repo, state, *freeargs, **opts)
982 982 finally:
983 983 release(state.lock, state.wlock)
984 984
985 985 def _histedit(ui, repo, state, *freeargs, **opts):
986 986 # TODO only abort if we try to histedit mq patches, not just
987 987 # blanket if mq patches are applied somewhere
988 988 mq = getattr(repo, 'mq', None)
989 989 if mq and mq.applied:
990 990 raise error.Abort(_('source has mq patches applied'))
991 991
992 992 # basic argument incompatibility processing
993 993 outg = opts.get('outgoing')
994 994 cont = opts.get('continue')
995 995 editplan = opts.get('edit_plan')
996 996 abort = opts.get('abort')
997 997 force = opts.get('force')
998 998 rules = opts.get('commands', '')
999 999 revs = opts.get('rev', [])
1000 1000 goal = 'new' # This invocation goal, in new, continue, abort
1001 1001 if force and not outg:
1002 1002 raise error.Abort(_('--force only allowed with --outgoing'))
1003 1003 if cont:
1004 1004 if any((outg, abort, revs, freeargs, rules, editplan)):
1005 1005 raise error.Abort(_('no arguments allowed with --continue'))
1006 1006 goal = 'continue'
1007 1007 elif abort:
1008 1008 if any((outg, revs, freeargs, rules, editplan)):
1009 1009 raise error.Abort(_('no arguments allowed with --abort'))
1010 1010 goal = 'abort'
1011 1011 elif editplan:
1012 1012 if any((outg, revs, freeargs)):
1013 1013 raise error.Abort(_('only --commands argument allowed with '
1014 1014 '--edit-plan'))
1015 1015 goal = 'edit-plan'
1016 1016 else:
1017 1017 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1018 1018 raise error.Abort(_('history edit already in progress, try '
1019 1019 '--continue or --abort'))
1020 1020 if outg:
1021 1021 if revs:
1022 1022 raise error.Abort(_('no revisions allowed with --outgoing'))
1023 1023 if len(freeargs) > 1:
1024 1024 raise error.Abort(
1025 1025 _('only one repo argument allowed with --outgoing'))
1026 1026 else:
1027 1027 revs.extend(freeargs)
1028 1028 if len(revs) == 0:
1029 1029 defaultrev = destutil.desthistedit(ui, repo)
1030 1030 if defaultrev is not None:
1031 1031 revs.append(defaultrev)
1032 1032
1033 1033 if len(revs) != 1:
1034 1034 raise error.Abort(
1035 1035 _('histedit requires exactly one ancestor revision'))
1036 1036
1037 1037
1038 replacements = []
1039 1038 state.keep = opts.get('keep', False)
1040 1039 supportsmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
1041 1040
1042 1041 # rebuild state
1043 1042 if goal == 'continue':
1044 1043 state.read()
1045 1044 state = bootstrapcontinue(ui, state, opts)
1046 1045 elif goal == 'edit-plan':
1047 1046 _editplanaction(ui, repo, state, rules)
1048 1047 return
1049 1048 elif goal == 'abort':
1050 1049 _abortaction(ui, repo, state)
1051 1050 return
1052 1051 else:
1053 cmdutil.checkunfinished(repo)
1054 cmdutil.bailifchanged(repo)
1055
1056 topmost, empty = repo.dirstate.parents()
1057 if outg:
1058 if freeargs:
1059 remote = freeargs[0]
1060 else:
1061 remote = None
1062 root = findoutgoing(ui, repo, remote, force, opts)
1063 else:
1064 rr = list(repo.set('roots(%ld)', scmutil.revrange(repo, revs)))
1065 if len(rr) != 1:
1066 raise error.Abort(_('The specified revisions must have '
1067 'exactly one common root'))
1068 root = rr[0].node()
1069
1070 revs = between(repo, root, topmost, state.keep)
1071 if not revs:
1072 raise error.Abort(_('%s is not an ancestor of working directory') %
1073 node.short(root))
1074
1075 ctxs = [repo[r] for r in revs]
1076 if not rules:
1077 comment = geteditcomment(node.short(root), node.short(topmost))
1078 actions = [pick(state, r) for r in revs]
1079 rules = ruleeditor(repo, ui, actions, comment)
1080 else:
1081 if rules == '-':
1082 f = sys.stdin
1083 else:
1084 f = open(rules)
1085 rules = f.read()
1086 f.close()
1087 actions = parserules(rules, state)
1088 warnverifyactions(ui, repo, actions, state, ctxs)
1089
1090 parentctxnode = repo[root].parents()[0].node()
1091
1092 state.parentctxnode = parentctxnode
1093 state.actions = actions
1094 state.topmost = topmost
1095 state.replacements = replacements
1096
1097 # Create a backup so we can always abort completely.
1098 backupfile = None
1099 if not obsolete.isenabled(repo, obsolete.createmarkersopt):
1100 backupfile = repair._bundle(repo, [parentctxnode], [topmost], root,
1101 'histedit')
1102 state.backupfile = backupfile
1052 _newaction(ui, repo, state, revs, freeargs, opts)
1103 1053
1104 1054 # preprocess rules so that we can hide inner folds from the user
1105 1055 # and only show one editor
1106 1056 actions = state.actions[:]
1107 1057 for idx, (action, nextact) in enumerate(
1108 1058 zip(actions, actions[1:] + [None])):
1109 1059 if action.verb == 'fold' and nextact and nextact.verb == 'fold':
1110 1060 state.actions[idx].__class__ = _multifold
1111 1061
1112 1062 total = len(state.actions)
1113 1063 pos = 0
1114 1064 while state.actions:
1115 1065 state.write()
1116 1066 actobj = state.actions.pop(0)
1117 1067 pos += 1
1118 1068 ui.progress(_("editing"), pos, actobj.torule(),
1119 1069 _('changes'), total)
1120 1070 ui.debug('histedit: processing %s %s\n' % (actobj.verb,\
1121 1071 actobj.torule()))
1122 1072 parentctx, replacement_ = actobj.run()
1123 1073 state.parentctxnode = parentctx.node()
1124 1074 state.replacements.extend(replacement_)
1125 1075 state.write()
1126 1076 ui.progress(_("editing"), None)
1127 1077
1128 1078 repo.ui.pushbuffer()
1129 1079 hg.update(repo, state.parentctxnode, quietempty=True)
1130 1080 repo.ui.popbuffer()
1131 1081
1132 1082 mapping, tmpnodes, created, ntm = processreplacement(state)
1133 1083 if mapping:
1134 1084 for prec, succs in mapping.iteritems():
1135 1085 if not succs:
1136 1086 ui.debug('histedit: %s is dropped\n' % node.short(prec))
1137 1087 else:
1138 1088 ui.debug('histedit: %s is replaced by %s\n' % (
1139 1089 node.short(prec), node.short(succs[0])))
1140 1090 if len(succs) > 1:
1141 1091 m = 'histedit: %s'
1142 1092 for n in succs[1:]:
1143 1093 ui.debug(m % node.short(n))
1144 1094
1145 1095 if supportsmarkers:
1146 1096 # Only create markers if the temp nodes weren't already removed.
1147 1097 obsolete.createmarkers(repo, ((repo[t],()) for t in sorted(tmpnodes)
1148 1098 if t in repo))
1149 1099 else:
1150 1100 cleanupnode(ui, repo, 'temp', tmpnodes)
1151 1101
1152 1102 if not state.keep:
1153 1103 if mapping:
1154 1104 movebookmarks(ui, repo, mapping, state.topmost, ntm)
1155 1105 # TODO update mq state
1156 1106 if supportsmarkers:
1157 1107 markers = []
1158 1108 # sort by revision number because it sound "right"
1159 1109 for prec in sorted(mapping, key=repo.changelog.rev):
1160 1110 succs = mapping[prec]
1161 1111 markers.append((repo[prec],
1162 1112 tuple(repo[s] for s in succs)))
1163 1113 if markers:
1164 1114 obsolete.createmarkers(repo, markers)
1165 1115 else:
1166 1116 cleanupnode(ui, repo, 'replaced', mapping)
1167 1117
1168 1118 state.clear()
1169 1119 if os.path.exists(repo.sjoin('undo')):
1170 1120 os.unlink(repo.sjoin('undo'))
1171 1121 if repo.vfs.exists('histedit-last-edit.txt'):
1172 1122 repo.vfs.unlink('histedit-last-edit.txt')
1173 1123
1174 1124 def _abortaction(ui, repo, state):
1175 1125 try:
1176 1126 state.read()
1177 1127 tmpnodes, leafs = newnodestoabort(state)
1178 1128 ui.debug('restore wc to old parent %s\n'
1179 1129 % node.short(state.topmost))
1180 1130
1181 1131 # Recover our old commits if necessary
1182 1132 if not state.topmost in repo and state.backupfile:
1183 1133 backupfile = repo.join(state.backupfile)
1184 1134 f = hg.openpath(ui, backupfile)
1185 1135 gen = exchange.readbundle(ui, f, backupfile)
1186 1136 with repo.transaction('histedit.abort') as tr:
1187 1137 if not isinstance(gen, bundle2.unbundle20):
1188 1138 gen.apply(repo, 'histedit', 'bundle:' + backupfile)
1189 1139 if isinstance(gen, bundle2.unbundle20):
1190 1140 bundle2.applybundle(repo, gen, tr,
1191 1141 source='histedit',
1192 1142 url='bundle:' + backupfile)
1193 1143
1194 1144 os.remove(backupfile)
1195 1145
1196 1146 # check whether we should update away
1197 1147 if repo.unfiltered().revs('parents() and (%n or %ln::)',
1198 1148 state.parentctxnode, leafs | tmpnodes):
1199 1149 hg.clean(repo, state.topmost, show_stats=True, quietempty=True)
1200 1150 cleanupnode(ui, repo, 'created', tmpnodes)
1201 1151 cleanupnode(ui, repo, 'temp', leafs)
1202 1152 except Exception:
1203 1153 if state.inprogress():
1204 1154 ui.warn(_('warning: encountered an exception during histedit '
1205 1155 '--abort; the repository may not have been completely '
1206 1156 'cleaned up\n'))
1207 1157 raise
1208 1158 finally:
1209 1159 state.clear()
1210 1160
1211 1161 def _editplanaction(ui, repo, state, rules):
1212 1162 state.read()
1213 1163 if not rules:
1214 1164 comment = geteditcomment(node.short(state.parentctxnode),
1215 1165 node.short(state.topmost))
1216 1166 rules = ruleeditor(repo, ui, state.actions, comment)
1217 1167 else:
1218 1168 if rules == '-':
1219 1169 f = sys.stdin
1220 1170 else:
1221 1171 f = open(rules)
1222 1172 rules = f.read()
1223 1173 f.close()
1224 1174 actions = parserules(rules, state)
1225 1175 ctxs = [repo[act.nodetoverify()] \
1226 1176 for act in state.actions if act.nodetoverify()]
1227 1177 warnverifyactions(ui, repo, actions, state, ctxs)
1228 1178 state.actions = actions
1229 1179 state.write()
1230 1180
1181 def _newaction(ui, repo, state, revs, freeargs, opts):
1182 outg = opts.get('outgoing')
1183 rules = opts.get('commands', '')
1184 force = opts.get('force')
1185
1186 cmdutil.checkunfinished(repo)
1187 cmdutil.bailifchanged(repo)
1188
1189 topmost, empty = repo.dirstate.parents()
1190 if outg:
1191 if freeargs:
1192 remote = freeargs[0]
1193 else:
1194 remote = None
1195 root = findoutgoing(ui, repo, remote, force, opts)
1196 else:
1197 rr = list(repo.set('roots(%ld)', scmutil.revrange(repo, revs)))
1198 if len(rr) != 1:
1199 raise error.Abort(_('The specified revisions must have '
1200 'exactly one common root'))
1201 root = rr[0].node()
1202
1203 revs = between(repo, root, topmost, state.keep)
1204 if not revs:
1205 raise error.Abort(_('%s is not an ancestor of working directory') %
1206 node.short(root))
1207
1208 ctxs = [repo[r] for r in revs]
1209 if not rules:
1210 comment = geteditcomment(node.short(root), node.short(topmost))
1211 actions = [pick(state, r) for r in revs]
1212 rules = ruleeditor(repo, ui, actions, comment)
1213 else:
1214 if rules == '-':
1215 f = sys.stdin
1216 else:
1217 f = open(rules)
1218 rules = f.read()
1219 f.close()
1220 actions = parserules(rules, state)
1221 warnverifyactions(ui, repo, actions, state, ctxs)
1222
1223 parentctxnode = repo[root].parents()[0].node()
1224
1225 state.parentctxnode = parentctxnode
1226 state.actions = actions
1227 state.topmost = topmost
1228 state.replacements = []
1229
1230 # Create a backup so we can always abort completely.
1231 backupfile = None
1232 if not obsolete.isenabled(repo, obsolete.createmarkersopt):
1233 backupfile = repair._bundle(repo, [parentctxnode], [topmost], root,
1234 'histedit')
1235 state.backupfile = backupfile
1236
1231 1237 def bootstrapcontinue(ui, state, opts):
1232 1238 repo = state.repo
1233 1239 if state.actions:
1234 1240 actobj = state.actions.pop(0)
1235 1241
1236 1242 if _isdirtywc(repo):
1237 1243 actobj.continuedirty()
1238 1244 if _isdirtywc(repo):
1239 1245 abortdirty()
1240 1246
1241 1247 parentctx, replacements = actobj.continueclean()
1242 1248
1243 1249 state.parentctxnode = parentctx.node()
1244 1250 state.replacements.extend(replacements)
1245 1251
1246 1252 return state
1247 1253
1248 1254 def between(repo, old, new, keep):
1249 1255 """select and validate the set of revision to edit
1250 1256
1251 1257 When keep is false, the specified set can't have children."""
1252 1258 ctxs = list(repo.set('%n::%n', old, new))
1253 1259 if ctxs and not keep:
1254 1260 if (not obsolete.isenabled(repo, obsolete.allowunstableopt) and
1255 1261 repo.revs('(%ld::) - (%ld)', ctxs, ctxs)):
1256 1262 raise error.Abort(_('cannot edit history that would orphan nodes'))
1257 1263 if repo.revs('(%ld) and merge()', ctxs):
1258 1264 raise error.Abort(_('cannot edit history that contains merges'))
1259 1265 root = ctxs[0] # list is already sorted by repo.set
1260 1266 if not root.mutable():
1261 1267 raise error.Abort(_('cannot edit public changeset: %s') % root,
1262 1268 hint=_('see "hg help phases" for details'))
1263 1269 return [c.node() for c in ctxs]
1264 1270
1265 1271 def ruleeditor(repo, ui, actions, editcomment=""):
1266 1272 """open an editor to edit rules
1267 1273
1268 1274 rules are in the format [ [act, ctx], ...] like in state.rules
1269 1275 """
1270 1276 rules = '\n'.join([act.torule() for act in actions])
1271 1277 rules += '\n\n'
1272 1278 rules += editcomment
1273 1279 rules = ui.edit(rules, ui.username(), {'prefix': 'histedit'})
1274 1280
1275 1281 # Save edit rules in .hg/histedit-last-edit.txt in case
1276 1282 # the user needs to ask for help after something
1277 1283 # surprising happens.
1278 1284 f = open(repo.join('histedit-last-edit.txt'), 'w')
1279 1285 f.write(rules)
1280 1286 f.close()
1281 1287
1282 1288 return rules
1283 1289
1284 1290 def parserules(rules, state):
1285 1291 """Read the histedit rules string and return list of action objects """
1286 1292 rules = [l for l in (r.strip() for r in rules.splitlines())
1287 1293 if l and not l.startswith('#')]
1288 1294 actions = []
1289 1295 for r in rules:
1290 1296 if ' ' not in r:
1291 1297 raise error.ParseError(_('malformed line "%s"') % r)
1292 1298 verb, rest = r.split(' ', 1)
1293 1299
1294 1300 if verb not in actiontable:
1295 1301 raise error.ParseError(_('unknown action "%s"') % verb)
1296 1302
1297 1303 action = actiontable[verb].fromrule(state, rest)
1298 1304 actions.append(action)
1299 1305 return actions
1300 1306
1301 1307 def warnverifyactions(ui, repo, actions, state, ctxs):
1302 1308 try:
1303 1309 verifyactions(actions, state, ctxs)
1304 1310 except error.ParseError:
1305 1311 if repo.vfs.exists('histedit-last-edit.txt'):
1306 1312 ui.warn(_('warning: histedit rules saved '
1307 1313 'to: .hg/histedit-last-edit.txt\n'))
1308 1314 raise
1309 1315
1310 1316 def verifyactions(actions, state, ctxs):
1311 1317 """Verify that there exists exactly one action per given changeset and
1312 1318 other constraints.
1313 1319
1314 1320 Will abort if there are to many or too few rules, a malformed rule,
1315 1321 or a rule on a changeset outside of the user-given range.
1316 1322 """
1317 1323 expected = set(c.hex() for c in ctxs)
1318 1324 seen = set()
1319 1325 prev = None
1320 1326 for action in actions:
1321 1327 action.verify(prev)
1322 1328 prev = action
1323 1329 constraints = action.constraints()
1324 1330 for constraint in constraints:
1325 1331 if constraint not in _constraints.known():
1326 1332 raise error.ParseError(_('unknown constraint "%s"') %
1327 1333 constraint)
1328 1334
1329 1335 nodetoverify = action.nodetoverify()
1330 1336 if nodetoverify is not None:
1331 1337 ha = node.hex(nodetoverify)
1332 1338 if _constraints.noother in constraints and ha not in expected:
1333 1339 raise error.ParseError(
1334 1340 _('%s "%s" changeset was not a candidate')
1335 1341 % (action.verb, ha[:12]),
1336 1342 hint=_('only use listed changesets'))
1337 1343 if _constraints.forceother in constraints and ha in expected:
1338 1344 raise error.ParseError(
1339 1345 _('%s "%s" changeset was not an edited list candidate')
1340 1346 % (action.verb, ha[:12]),
1341 1347 hint=_('only use listed changesets'))
1342 1348 if _constraints.noduplicates in constraints and ha in seen:
1343 1349 raise error.ParseError(_(
1344 1350 'duplicated command for changeset %s') %
1345 1351 ha[:12])
1346 1352 seen.add(ha)
1347 1353 missing = sorted(expected - seen) # sort to stabilize output
1348 1354
1349 1355 if state.repo.ui.configbool('histedit', 'dropmissing'):
1350 1356 drops = [drop(state, node.bin(n)) for n in missing]
1351 1357 # put the in the beginning so they execute immediately and
1352 1358 # don't show in the edit-plan in the future
1353 1359 actions[:0] = drops
1354 1360 elif missing:
1355 1361 raise error.ParseError(_('missing rules for changeset %s') %
1356 1362 missing[0][:12],
1357 1363 hint=_('use "drop %s" to discard, see also: '
1358 1364 '"hg help -e histedit.config"') % missing[0][:12])
1359 1365
1360 1366 def newnodestoabort(state):
1361 1367 """process the list of replacements to return
1362 1368
1363 1369 1) the list of final node
1364 1370 2) the list of temporary node
1365 1371
1366 1372 This is meant to be used on abort as less data are required in this case.
1367 1373 """
1368 1374 replacements = state.replacements
1369 1375 allsuccs = set()
1370 1376 replaced = set()
1371 1377 for rep in replacements:
1372 1378 allsuccs.update(rep[1])
1373 1379 replaced.add(rep[0])
1374 1380 newnodes = allsuccs - replaced
1375 1381 tmpnodes = allsuccs & replaced
1376 1382 return newnodes, tmpnodes
1377 1383
1378 1384
1379 1385 def processreplacement(state):
1380 1386 """process the list of replacements to return
1381 1387
1382 1388 1) the final mapping between original and created nodes
1383 1389 2) the list of temporary node created by histedit
1384 1390 3) the list of new commit created by histedit"""
1385 1391 replacements = state.replacements
1386 1392 allsuccs = set()
1387 1393 replaced = set()
1388 1394 fullmapping = {}
1389 1395 # initialize basic set
1390 1396 # fullmapping records all operations recorded in replacement
1391 1397 for rep in replacements:
1392 1398 allsuccs.update(rep[1])
1393 1399 replaced.add(rep[0])
1394 1400 fullmapping.setdefault(rep[0], set()).update(rep[1])
1395 1401 new = allsuccs - replaced
1396 1402 tmpnodes = allsuccs & replaced
1397 1403 # Reduce content fullmapping into direct relation between original nodes
1398 1404 # and final node created during history edition
1399 1405 # Dropped changeset are replaced by an empty list
1400 1406 toproceed = set(fullmapping)
1401 1407 final = {}
1402 1408 while toproceed:
1403 1409 for x in list(toproceed):
1404 1410 succs = fullmapping[x]
1405 1411 for s in list(succs):
1406 1412 if s in toproceed:
1407 1413 # non final node with unknown closure
1408 1414 # We can't process this now
1409 1415 break
1410 1416 elif s in final:
1411 1417 # non final node, replace with closure
1412 1418 succs.remove(s)
1413 1419 succs.update(final[s])
1414 1420 else:
1415 1421 final[x] = succs
1416 1422 toproceed.remove(x)
1417 1423 # remove tmpnodes from final mapping
1418 1424 for n in tmpnodes:
1419 1425 del final[n]
1420 1426 # we expect all changes involved in final to exist in the repo
1421 1427 # turn `final` into list (topologically sorted)
1422 1428 nm = state.repo.changelog.nodemap
1423 1429 for prec, succs in final.items():
1424 1430 final[prec] = sorted(succs, key=nm.get)
1425 1431
1426 1432 # computed topmost element (necessary for bookmark)
1427 1433 if new:
1428 1434 newtopmost = sorted(new, key=state.repo.changelog.rev)[-1]
1429 1435 elif not final:
1430 1436 # Nothing rewritten at all. we won't need `newtopmost`
1431 1437 # It is the same as `oldtopmost` and `processreplacement` know it
1432 1438 newtopmost = None
1433 1439 else:
1434 1440 # every body died. The newtopmost is the parent of the root.
1435 1441 r = state.repo.changelog.rev
1436 1442 newtopmost = state.repo[sorted(final, key=r)[0]].p1().node()
1437 1443
1438 1444 return final, tmpnodes, new, newtopmost
1439 1445
1440 1446 def movebookmarks(ui, repo, mapping, oldtopmost, newtopmost):
1441 1447 """Move bookmark from old to newly created node"""
1442 1448 if not mapping:
1443 1449 # if nothing got rewritten there is not purpose for this function
1444 1450 return
1445 1451 moves = []
1446 1452 for bk, old in sorted(repo._bookmarks.iteritems()):
1447 1453 if old == oldtopmost:
1448 1454 # special case ensure bookmark stay on tip.
1449 1455 #
1450 1456 # This is arguably a feature and we may only want that for the
1451 1457 # active bookmark. But the behavior is kept compatible with the old
1452 1458 # version for now.
1453 1459 moves.append((bk, newtopmost))
1454 1460 continue
1455 1461 base = old
1456 1462 new = mapping.get(base, None)
1457 1463 if new is None:
1458 1464 continue
1459 1465 while not new:
1460 1466 # base is killed, trying with parent
1461 1467 base = repo[base].p1().node()
1462 1468 new = mapping.get(base, (base,))
1463 1469 # nothing to move
1464 1470 moves.append((bk, new[-1]))
1465 1471 if moves:
1466 1472 lock = tr = None
1467 1473 try:
1468 1474 lock = repo.lock()
1469 1475 tr = repo.transaction('histedit')
1470 1476 marks = repo._bookmarks
1471 1477 for mark, new in moves:
1472 1478 old = marks[mark]
1473 1479 ui.note(_('histedit: moving bookmarks %s from %s to %s\n')
1474 1480 % (mark, node.short(old), node.short(new)))
1475 1481 marks[mark] = new
1476 1482 marks.recordchange(tr)
1477 1483 tr.close()
1478 1484 finally:
1479 1485 release(tr, lock)
1480 1486
1481 1487 def cleanupnode(ui, repo, name, nodes):
1482 1488 """strip a group of nodes from the repository
1483 1489
1484 1490 The set of node to strip may contains unknown nodes."""
1485 1491 ui.debug('should strip %s nodes %s\n' %
1486 1492 (name, ', '.join([node.short(n) for n in nodes])))
1487 1493 with repo.lock():
1488 1494 # do not let filtering get in the way of the cleanse
1489 1495 # we should probably get rid of obsolescence marker created during the
1490 1496 # histedit, but we currently do not have such information.
1491 1497 repo = repo.unfiltered()
1492 1498 # Find all nodes that need to be stripped
1493 1499 # (we use %lr instead of %ln to silently ignore unknown items)
1494 1500 nm = repo.changelog.nodemap
1495 1501 nodes = sorted(n for n in nodes if n in nm)
1496 1502 roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
1497 1503 for c in roots:
1498 1504 # We should process node in reverse order to strip tip most first.
1499 1505 # but this trigger a bug in changegroup hook.
1500 1506 # This would reduce bundle overhead
1501 1507 repair.strip(ui, repo, c)
1502 1508
1503 1509 def stripwrapper(orig, ui, repo, nodelist, *args, **kwargs):
1504 1510 if isinstance(nodelist, str):
1505 1511 nodelist = [nodelist]
1506 1512 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1507 1513 state = histeditstate(repo)
1508 1514 state.read()
1509 1515 histedit_nodes = set([action.nodetoverify() for action
1510 1516 in state.actions if action.nodetoverify()])
1511 1517 strip_nodes = set([repo[n].node() for n in nodelist])
1512 1518 common_nodes = histedit_nodes & strip_nodes
1513 1519 if common_nodes:
1514 1520 raise error.Abort(_("histedit in progress, can't strip %s")
1515 1521 % ', '.join(node.short(x) for x in common_nodes))
1516 1522 return orig(ui, repo, nodelist, *args, **kwargs)
1517 1523
1518 1524 extensions.wrapfunction(repair, 'strip', stripwrapper)
1519 1525
1520 1526 def summaryhook(ui, repo):
1521 1527 if not os.path.exists(repo.join('histedit-state')):
1522 1528 return
1523 1529 state = histeditstate(repo)
1524 1530 state.read()
1525 1531 if state.actions:
1526 1532 # i18n: column positioning for "hg summary"
1527 1533 ui.write(_('hist: %s (histedit --continue)\n') %
1528 1534 (ui.label(_('%d remaining'), 'histedit.remaining') %
1529 1535 len(state.actions)))
1530 1536
1531 1537 def extsetup(ui):
1532 1538 cmdutil.summaryhooks.add('histedit', summaryhook)
1533 1539 cmdutil.unfinishedstates.append(
1534 1540 ['histedit-state', False, True, _('histedit in progress'),
1535 1541 _("use 'hg histedit --continue' or 'hg histedit --abort'")])
1536 1542 cmdutil.afterresolvedstates.append(
1537 1543 ['histedit-state', _('hg histedit --continue')])
1538 1544 if ui.configbool("experimental", "histeditng"):
1539 1545 globals()['base'] = action(['base', 'b'],
1540 1546 _('checkout changeset and apply further changesets from there')
1541 1547 )(base)
General Comments 0
You need to be logged in to leave comments. Login now