##// END OF EJS Templates
with: use context manager for lock in histedit cleanupnode
Bryan O'Sullivan -
r27833:321759df default
parent child Browse files
Show More
@@ -1,1532 +1,1528 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 raise error.Abort(_('no histedit in progress'))
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 hg.update(repo, self.state.parentctxnode, quietempty=True)
451 451 stats = applychanges(repo.ui, repo, rulectx, {})
452 452 if stats and stats[3] > 0:
453 453 raise error.InterventionRequired(
454 454 _('Fix up the change (%s %s)') %
455 455 (self.verb, node.short(self.node)),
456 456 hint=_('hg histedit --continue to resume'))
457 457
458 458 def continuedirty(self):
459 459 """Continues the action when changes have been applied to the working
460 460 copy. The default behavior is to commit the dirty changes."""
461 461 repo = self.repo
462 462 rulectx = repo[self.node]
463 463
464 464 editor = self.commiteditor()
465 465 commit = commitfuncfor(repo, rulectx)
466 466
467 467 commit(text=rulectx.description(), user=rulectx.user(),
468 468 date=rulectx.date(), extra=rulectx.extra(), editor=editor)
469 469
470 470 def commiteditor(self):
471 471 """The editor to be used to edit the commit message."""
472 472 return False
473 473
474 474 def continueclean(self):
475 475 """Continues the action when the working copy is clean. The default
476 476 behavior is to accept the current commit as the new version of the
477 477 rulectx."""
478 478 ctx = self.repo['.']
479 479 if ctx.node() == self.state.parentctxnode:
480 480 self.repo.ui.warn(_('%s: empty changeset\n') %
481 481 node.short(self.node))
482 482 return ctx, [(self.node, tuple())]
483 483 if ctx.node() == self.node:
484 484 # Nothing changed
485 485 return ctx, []
486 486 return ctx, [(self.node, (ctx.node(),))]
487 487
488 488 def commitfuncfor(repo, src):
489 489 """Build a commit function for the replacement of <src>
490 490
491 491 This function ensure we apply the same treatment to all changesets.
492 492
493 493 - Add a 'histedit_source' entry in extra.
494 494
495 495 Note that fold has its own separated logic because its handling is a bit
496 496 different and not easily factored out of the fold method.
497 497 """
498 498 phasemin = src.phase()
499 499 def commitfunc(**kwargs):
500 500 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
501 501 try:
502 502 repo.ui.setconfig('phases', 'new-commit', phasemin,
503 503 'histedit')
504 504 extra = kwargs.get('extra', {}).copy()
505 505 extra['histedit_source'] = src.hex()
506 506 kwargs['extra'] = extra
507 507 return repo.commit(**kwargs)
508 508 finally:
509 509 repo.ui.restoreconfig(phasebackup)
510 510 return commitfunc
511 511
512 512 def applychanges(ui, repo, ctx, opts):
513 513 """Merge changeset from ctx (only) in the current working directory"""
514 514 wcpar = repo.dirstate.parents()[0]
515 515 if ctx.p1().node() == wcpar:
516 516 # edits are "in place" we do not need to make any merge,
517 517 # just applies changes on parent for editing
518 518 cmdutil.revert(ui, repo, ctx, (wcpar, node.nullid), all=True)
519 519 stats = None
520 520 else:
521 521 try:
522 522 # ui.forcemerge is an internal variable, do not document
523 523 repo.ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
524 524 'histedit')
525 525 stats = mergemod.graft(repo, ctx, ctx.p1(), ['local', 'histedit'])
526 526 finally:
527 527 repo.ui.setconfig('ui', 'forcemerge', '', 'histedit')
528 528 return stats
529 529
530 530 def collapse(repo, first, last, commitopts, skipprompt=False):
531 531 """collapse the set of revisions from first to last as new one.
532 532
533 533 Expected commit options are:
534 534 - message
535 535 - date
536 536 - username
537 537 Commit message is edited in all cases.
538 538
539 539 This function works in memory."""
540 540 ctxs = list(repo.set('%d::%d', first, last))
541 541 if not ctxs:
542 542 return None
543 543 for c in ctxs:
544 544 if not c.mutable():
545 545 raise error.ParseError(
546 546 _("cannot fold into public change %s") % node.short(c.node()))
547 547 base = first.parents()[0]
548 548
549 549 # commit a new version of the old changeset, including the update
550 550 # collect all files which might be affected
551 551 files = set()
552 552 for ctx in ctxs:
553 553 files.update(ctx.files())
554 554
555 555 # Recompute copies (avoid recording a -> b -> a)
556 556 copied = copies.pathcopies(base, last)
557 557
558 558 # prune files which were reverted by the updates
559 559 def samefile(f):
560 560 if f in last.manifest():
561 561 a = last.filectx(f)
562 562 if f in base.manifest():
563 563 b = base.filectx(f)
564 564 return (a.data() == b.data()
565 565 and a.flags() == b.flags())
566 566 else:
567 567 return False
568 568 else:
569 569 return f not in base.manifest()
570 570 files = [f for f in files if not samefile(f)]
571 571 # commit version of these files as defined by head
572 572 headmf = last.manifest()
573 573 def filectxfn(repo, ctx, path):
574 574 if path in headmf:
575 575 fctx = last[path]
576 576 flags = fctx.flags()
577 577 mctx = context.memfilectx(repo,
578 578 fctx.path(), fctx.data(),
579 579 islink='l' in flags,
580 580 isexec='x' in flags,
581 581 copied=copied.get(path))
582 582 return mctx
583 583 return None
584 584
585 585 if commitopts.get('message'):
586 586 message = commitopts['message']
587 587 else:
588 588 message = first.description()
589 589 user = commitopts.get('user')
590 590 date = commitopts.get('date')
591 591 extra = commitopts.get('extra')
592 592
593 593 parents = (first.p1().node(), first.p2().node())
594 594 editor = None
595 595 if not skipprompt:
596 596 editor = cmdutil.getcommiteditor(edit=True, editform='histedit.fold')
597 597 new = context.memctx(repo,
598 598 parents=parents,
599 599 text=message,
600 600 files=files,
601 601 filectxfn=filectxfn,
602 602 user=user,
603 603 date=date,
604 604 extra=extra,
605 605 editor=editor)
606 606 return repo.commitctx(new)
607 607
608 608 def _isdirtywc(repo):
609 609 return repo[None].dirty(missing=True)
610 610
611 611 def abortdirty():
612 612 raise error.Abort(_('working copy has pending changes'),
613 613 hint=_('amend, commit, or revert them and run histedit '
614 614 '--continue, or abort with histedit --abort'))
615 615
616 616 def action(verbs, message, priority=False, internal=False):
617 617 def wrap(cls):
618 618 assert not priority or not internal
619 619 verb = verbs[0]
620 620 if priority:
621 621 primaryactions.add(verb)
622 622 elif internal:
623 623 internalactions.add(verb)
624 624 elif len(verbs) > 1:
625 625 secondaryactions.add(verb)
626 626 else:
627 627 tertiaryactions.add(verb)
628 628
629 629 cls.verb = verb
630 630 cls.verbs = verbs
631 631 cls.message = message
632 632 for verb in verbs:
633 633 actiontable[verb] = cls
634 634 return cls
635 635 return wrap
636 636
637 637 @action(['pick', 'p'],
638 638 _('use commit'),
639 639 priority=True)
640 640 class pick(histeditaction):
641 641 def run(self):
642 642 rulectx = self.repo[self.node]
643 643 if rulectx.parents()[0].node() == self.state.parentctxnode:
644 644 self.repo.ui.debug('node %s unchanged\n' % node.short(self.node))
645 645 return rulectx, []
646 646
647 647 return super(pick, self).run()
648 648
649 649 @action(['edit', 'e'],
650 650 _('use commit, but stop for amending'),
651 651 priority=True)
652 652 class edit(histeditaction):
653 653 def run(self):
654 654 repo = self.repo
655 655 rulectx = repo[self.node]
656 656 hg.update(repo, self.state.parentctxnode, quietempty=True)
657 657 applychanges(repo.ui, repo, rulectx, {})
658 658 raise error.InterventionRequired(
659 659 _('Editing (%s), you may commit or record as needed now.')
660 660 % node.short(self.node),
661 661 hint=_('hg histedit --continue to resume'))
662 662
663 663 def commiteditor(self):
664 664 return cmdutil.getcommiteditor(edit=True, editform='histedit.edit')
665 665
666 666 @action(['fold', 'f'],
667 667 _('use commit, but combine it with the one above'))
668 668 class fold(histeditaction):
669 669 def verify(self, prev):
670 670 """ Verifies semantic correctness of the fold rule"""
671 671 super(fold, self).verify(prev)
672 672 repo = self.repo
673 673 if not prev:
674 674 c = repo[self.node].parents()[0]
675 675 elif not prev.verb in ('pick', 'base'):
676 676 return
677 677 else:
678 678 c = repo[prev.node]
679 679 if not c.mutable():
680 680 raise error.ParseError(
681 681 _("cannot fold into public change %s") % node.short(c.node()))
682 682
683 683
684 684 def continuedirty(self):
685 685 repo = self.repo
686 686 rulectx = repo[self.node]
687 687
688 688 commit = commitfuncfor(repo, rulectx)
689 689 commit(text='fold-temp-revision %s' % node.short(self.node),
690 690 user=rulectx.user(), date=rulectx.date(),
691 691 extra=rulectx.extra())
692 692
693 693 def continueclean(self):
694 694 repo = self.repo
695 695 ctx = repo['.']
696 696 rulectx = repo[self.node]
697 697 parentctxnode = self.state.parentctxnode
698 698 if ctx.node() == parentctxnode:
699 699 repo.ui.warn(_('%s: empty changeset\n') %
700 700 node.short(self.node))
701 701 return ctx, [(self.node, (parentctxnode,))]
702 702
703 703 parentctx = repo[parentctxnode]
704 704 newcommits = set(c.node() for c in repo.set('(%d::. - %d)', parentctx,
705 705 parentctx))
706 706 if not newcommits:
707 707 repo.ui.warn(_('%s: cannot fold - working copy is not a '
708 708 'descendant of previous commit %s\n') %
709 709 (node.short(self.node), node.short(parentctxnode)))
710 710 return ctx, [(self.node, (ctx.node(),))]
711 711
712 712 middlecommits = newcommits.copy()
713 713 middlecommits.discard(ctx.node())
714 714
715 715 return self.finishfold(repo.ui, repo, parentctx, rulectx, ctx.node(),
716 716 middlecommits)
717 717
718 718 def skipprompt(self):
719 719 """Returns true if the rule should skip the message editor.
720 720
721 721 For example, 'fold' wants to show an editor, but 'rollup'
722 722 doesn't want to.
723 723 """
724 724 return False
725 725
726 726 def mergedescs(self):
727 727 """Returns true if the rule should merge messages of multiple changes.
728 728
729 729 This exists mainly so that 'rollup' rules can be a subclass of
730 730 'fold'.
731 731 """
732 732 return True
733 733
734 734 def finishfold(self, ui, repo, ctx, oldctx, newnode, internalchanges):
735 735 parent = ctx.parents()[0].node()
736 736 hg.update(repo, parent)
737 737 ### prepare new commit data
738 738 commitopts = {}
739 739 commitopts['user'] = ctx.user()
740 740 # commit message
741 741 if not self.mergedescs():
742 742 newmessage = ctx.description()
743 743 else:
744 744 newmessage = '\n***\n'.join(
745 745 [ctx.description()] +
746 746 [repo[r].description() for r in internalchanges] +
747 747 [oldctx.description()]) + '\n'
748 748 commitopts['message'] = newmessage
749 749 # date
750 750 commitopts['date'] = max(ctx.date(), oldctx.date())
751 751 extra = ctx.extra().copy()
752 752 # histedit_source
753 753 # note: ctx is likely a temporary commit but that the best we can do
754 754 # here. This is sufficient to solve issue3681 anyway.
755 755 extra['histedit_source'] = '%s,%s' % (ctx.hex(), oldctx.hex())
756 756 commitopts['extra'] = extra
757 757 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
758 758 try:
759 759 phasemin = max(ctx.phase(), oldctx.phase())
760 760 repo.ui.setconfig('phases', 'new-commit', phasemin, 'histedit')
761 761 n = collapse(repo, ctx, repo[newnode], commitopts,
762 762 skipprompt=self.skipprompt())
763 763 finally:
764 764 repo.ui.restoreconfig(phasebackup)
765 765 if n is None:
766 766 return ctx, []
767 767 hg.update(repo, n)
768 768 replacements = [(oldctx.node(), (newnode,)),
769 769 (ctx.node(), (n,)),
770 770 (newnode, (n,)),
771 771 ]
772 772 for ich in internalchanges:
773 773 replacements.append((ich, (n,)))
774 774 return repo[n], replacements
775 775
776 776 class base(histeditaction):
777 777 def constraints(self):
778 778 return set([_constraints.forceother])
779 779
780 780 def run(self):
781 781 if self.repo['.'].node() != self.node:
782 782 mergemod.update(self.repo, self.node, False, True)
783 783 # branchmerge, force)
784 784 return self.continueclean()
785 785
786 786 def continuedirty(self):
787 787 abortdirty()
788 788
789 789 def continueclean(self):
790 790 basectx = self.repo['.']
791 791 return basectx, []
792 792
793 793 @action(['_multifold'],
794 794 _(
795 795 """fold subclass used for when multiple folds happen in a row
796 796
797 797 We only want to fire the editor for the folded message once when
798 798 (say) four changes are folded down into a single change. This is
799 799 similar to rollup, but we should preserve both messages so that
800 800 when the last fold operation runs we can show the user all the
801 801 commit messages in their editor.
802 802 """),
803 803 internal=True)
804 804 class _multifold(fold):
805 805 def skipprompt(self):
806 806 return True
807 807
808 808 @action(["roll", "r"],
809 809 _("like fold, but discard this commit's description"))
810 810 class rollup(fold):
811 811 def mergedescs(self):
812 812 return False
813 813
814 814 def skipprompt(self):
815 815 return True
816 816
817 817 @action(["drop", "d"],
818 818 _('remove commit from history'))
819 819 class drop(histeditaction):
820 820 def run(self):
821 821 parentctx = self.repo[self.state.parentctxnode]
822 822 return parentctx, [(self.node, tuple())]
823 823
824 824 @action(["mess", "m"],
825 825 _('edit commit message without changing commit content'),
826 826 priority=True)
827 827 class message(histeditaction):
828 828 def commiteditor(self):
829 829 return cmdutil.getcommiteditor(edit=True, editform='histedit.mess')
830 830
831 831 def findoutgoing(ui, repo, remote=None, force=False, opts=None):
832 832 """utility function to find the first outgoing changeset
833 833
834 834 Used by initialization code"""
835 835 if opts is None:
836 836 opts = {}
837 837 dest = ui.expandpath(remote or 'default-push', remote or 'default')
838 838 dest, revs = hg.parseurl(dest, None)[:2]
839 839 ui.status(_('comparing with %s\n') % util.hidepassword(dest))
840 840
841 841 revs, checkout = hg.addbranchrevs(repo, repo, revs, None)
842 842 other = hg.peer(repo, opts, dest)
843 843
844 844 if revs:
845 845 revs = [repo.lookup(rev) for rev in revs]
846 846
847 847 outgoing = discovery.findcommonoutgoing(repo, other, revs, force=force)
848 848 if not outgoing.missing:
849 849 raise error.Abort(_('no outgoing ancestors'))
850 850 roots = list(repo.revs("roots(%ln)", outgoing.missing))
851 851 if 1 < len(roots):
852 852 msg = _('there are ambiguous outgoing revisions')
853 853 hint = _('see "hg help histedit" for more detail')
854 854 raise error.Abort(msg, hint=hint)
855 855 return repo.lookup(roots[0])
856 856
857 857
858 858 @command('histedit',
859 859 [('', 'commands', '',
860 860 _('read history edits from the specified file'), _('FILE')),
861 861 ('c', 'continue', False, _('continue an edit already in progress')),
862 862 ('', 'edit-plan', False, _('edit remaining actions list')),
863 863 ('k', 'keep', False,
864 864 _("don't strip old nodes after edit is complete")),
865 865 ('', 'abort', False, _('abort an edit in progress')),
866 866 ('o', 'outgoing', False, _('changesets not found in destination')),
867 867 ('f', 'force', False,
868 868 _('force outgoing even for unrelated repositories')),
869 869 ('r', 'rev', [], _('first revision to be edited'), _('REV'))],
870 870 _("[OPTIONS] ([ANCESTOR] | --outgoing [URL])"))
871 871 def histedit(ui, repo, *freeargs, **opts):
872 872 """interactively edit changeset history
873 873
874 874 This command lets you edit a linear series of changesets (up to
875 875 and including the working directory, which should be clean).
876 876 You can::
877 877
878 878 - `pick` to [re]order a changeset
879 879
880 880 - `drop` to omit changeset
881 881
882 882 - `mess` to reword the changeset commit message
883 883
884 884 - `fold` to combine it with the preceding changeset
885 885
886 886 - `roll` like fold, but discarding this commit's description
887 887
888 888 - `edit` to edit this changeset
889 889
890 890 There are a number of ways to select the root changset::
891 891
892 892 - Specify ANCESTOR directly
893 893
894 894 - Use --outgoing -- it will be the first linear changeset not
895 895 included in destination. (See :hg:"help default-push")
896 896
897 897 - Otherwise, the value from the "histedit.defaultrev" config option
898 898 is used as a revset to select the base revision when ANCESTOR is not
899 899 specified. The first revision returned by the revset is used. By
900 900 default, this selects the editable history that is unique to the
901 901 ancestry of the working directory.
902 902
903 903 .. container:: verbose
904 904
905 905 If you use --outgoing, this command will abort if there are ambiguous
906 906 outgoing revisions. For example, if there are multiple branches
907 907 containing outgoing revisions.
908 908
909 909 Use "min(outgoing() and ::.)" or similar revset specification
910 910 instead of --outgoing to specify edit target revision exactly in
911 911 such ambiguous situation. See :hg:`help revsets` for detail about
912 912 selecting revisions.
913 913
914 914 .. container:: verbose
915 915
916 916 Examples:
917 917
918 918 - A number of changes have been made.
919 919 Revision 3 is no longer needed.
920 920
921 921 Start history editing from revision 3::
922 922
923 923 hg histedit -r 3
924 924
925 925 An editor opens, containing the list of revisions,
926 926 with specific actions specified::
927 927
928 928 pick 5339bf82f0ca 3 Zworgle the foobar
929 929 pick 8ef592ce7cc4 4 Bedazzle the zerlog
930 930 pick 0a9639fcda9d 5 Morgify the cromulancy
931 931
932 932 Additional information about the possible actions
933 933 to take appears below the list of revisions.
934 934
935 935 To remove revision 3 from the history,
936 936 its action (at the beginning of the relevant line)
937 937 is changed to 'drop'::
938 938
939 939 drop 5339bf82f0ca 3 Zworgle the foobar
940 940 pick 8ef592ce7cc4 4 Bedazzle the zerlog
941 941 pick 0a9639fcda9d 5 Morgify the cromulancy
942 942
943 943 - A number of changes have been made.
944 944 Revision 2 and 4 need to be swapped.
945 945
946 946 Start history editing from revision 2::
947 947
948 948 hg histedit -r 2
949 949
950 950 An editor opens, containing the list of revisions,
951 951 with specific actions specified::
952 952
953 953 pick 252a1af424ad 2 Blorb a morgwazzle
954 954 pick 5339bf82f0ca 3 Zworgle the foobar
955 955 pick 8ef592ce7cc4 4 Bedazzle the zerlog
956 956
957 957 To swap revision 2 and 4, its lines are swapped
958 958 in the editor::
959 959
960 960 pick 8ef592ce7cc4 4 Bedazzle the zerlog
961 961 pick 5339bf82f0ca 3 Zworgle the foobar
962 962 pick 252a1af424ad 2 Blorb a morgwazzle
963 963
964 964 Returns 0 on success, 1 if user intervention is required (not only
965 965 for intentional "edit" command, but also for resolving unexpected
966 966 conflicts).
967 967 """
968 968 state = histeditstate(repo)
969 969 try:
970 970 state.wlock = repo.wlock()
971 971 state.lock = repo.lock()
972 972 _histedit(ui, repo, state, *freeargs, **opts)
973 973 finally:
974 974 release(state.lock, state.wlock)
975 975
976 976 def _histedit(ui, repo, state, *freeargs, **opts):
977 977 # TODO only abort if we try to histedit mq patches, not just
978 978 # blanket if mq patches are applied somewhere
979 979 mq = getattr(repo, 'mq', None)
980 980 if mq and mq.applied:
981 981 raise error.Abort(_('source has mq patches applied'))
982 982
983 983 # basic argument incompatibility processing
984 984 outg = opts.get('outgoing')
985 985 cont = opts.get('continue')
986 986 editplan = opts.get('edit_plan')
987 987 abort = opts.get('abort')
988 988 force = opts.get('force')
989 989 rules = opts.get('commands', '')
990 990 revs = opts.get('rev', [])
991 991 goal = 'new' # This invocation goal, in new, continue, abort
992 992 if force and not outg:
993 993 raise error.Abort(_('--force only allowed with --outgoing'))
994 994 if cont:
995 995 if any((outg, abort, revs, freeargs, rules, editplan)):
996 996 raise error.Abort(_('no arguments allowed with --continue'))
997 997 goal = 'continue'
998 998 elif abort:
999 999 if any((outg, revs, freeargs, rules, editplan)):
1000 1000 raise error.Abort(_('no arguments allowed with --abort'))
1001 1001 goal = 'abort'
1002 1002 elif editplan:
1003 1003 if any((outg, revs, freeargs)):
1004 1004 raise error.Abort(_('only --commands argument allowed with '
1005 1005 '--edit-plan'))
1006 1006 goal = 'edit-plan'
1007 1007 else:
1008 1008 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1009 1009 raise error.Abort(_('history edit already in progress, try '
1010 1010 '--continue or --abort'))
1011 1011 if outg:
1012 1012 if revs:
1013 1013 raise error.Abort(_('no revisions allowed with --outgoing'))
1014 1014 if len(freeargs) > 1:
1015 1015 raise error.Abort(
1016 1016 _('only one repo argument allowed with --outgoing'))
1017 1017 else:
1018 1018 revs.extend(freeargs)
1019 1019 if len(revs) == 0:
1020 1020 defaultrev = destutil.desthistedit(ui, repo)
1021 1021 if defaultrev is not None:
1022 1022 revs.append(defaultrev)
1023 1023
1024 1024 if len(revs) != 1:
1025 1025 raise error.Abort(
1026 1026 _('histedit requires exactly one ancestor revision'))
1027 1027
1028 1028
1029 1029 replacements = []
1030 1030 state.keep = opts.get('keep', False)
1031 1031 supportsmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
1032 1032
1033 1033 # rebuild state
1034 1034 if goal == 'continue':
1035 1035 state.read()
1036 1036 state = bootstrapcontinue(ui, state, opts)
1037 1037 elif goal == 'edit-plan':
1038 1038 state.read()
1039 1039 if not rules:
1040 1040 comment = geteditcomment(node.short(state.parentctxnode),
1041 1041 node.short(state.topmost))
1042 1042 rules = ruleeditor(repo, ui, state.actions, comment)
1043 1043 else:
1044 1044 if rules == '-':
1045 1045 f = sys.stdin
1046 1046 else:
1047 1047 f = open(rules)
1048 1048 rules = f.read()
1049 1049 f.close()
1050 1050 actions = parserules(rules, state)
1051 1051 ctxs = [repo[act.nodetoverify()] \
1052 1052 for act in state.actions if act.nodetoverify()]
1053 1053 warnverifyactions(ui, repo, actions, state, ctxs)
1054 1054 state.actions = actions
1055 1055 state.write()
1056 1056 return
1057 1057 elif goal == 'abort':
1058 1058 try:
1059 1059 state.read()
1060 1060 tmpnodes, leafs = newnodestoabort(state)
1061 1061 ui.debug('restore wc to old parent %s\n'
1062 1062 % node.short(state.topmost))
1063 1063
1064 1064 # Recover our old commits if necessary
1065 1065 if not state.topmost in repo and state.backupfile:
1066 1066 backupfile = repo.join(state.backupfile)
1067 1067 f = hg.openpath(ui, backupfile)
1068 1068 gen = exchange.readbundle(ui, f, backupfile)
1069 1069 tr = repo.transaction('histedit.abort')
1070 1070 try:
1071 1071 if not isinstance(gen, bundle2.unbundle20):
1072 1072 gen.apply(repo, 'histedit', 'bundle:' + backupfile)
1073 1073 if isinstance(gen, bundle2.unbundle20):
1074 1074 bundle2.applybundle(repo, gen, tr,
1075 1075 source='histedit',
1076 1076 url='bundle:' + backupfile)
1077 1077 tr.close()
1078 1078 finally:
1079 1079 tr.release()
1080 1080
1081 1081 os.remove(backupfile)
1082 1082
1083 1083 # check whether we should update away
1084 1084 if repo.unfiltered().revs('parents() and (%n or %ln::)',
1085 1085 state.parentctxnode, leafs | tmpnodes):
1086 1086 hg.clean(repo, state.topmost, show_stats=True, quietempty=True)
1087 1087 cleanupnode(ui, repo, 'created', tmpnodes)
1088 1088 cleanupnode(ui, repo, 'temp', leafs)
1089 1089 except Exception:
1090 1090 if state.inprogress():
1091 1091 ui.warn(_('warning: encountered an exception during histedit '
1092 1092 '--abort; the repository may not have been completely '
1093 1093 'cleaned up\n'))
1094 1094 raise
1095 1095 finally:
1096 1096 state.clear()
1097 1097 return
1098 1098 else:
1099 1099 cmdutil.checkunfinished(repo)
1100 1100 cmdutil.bailifchanged(repo)
1101 1101
1102 1102 topmost, empty = repo.dirstate.parents()
1103 1103 if outg:
1104 1104 if freeargs:
1105 1105 remote = freeargs[0]
1106 1106 else:
1107 1107 remote = None
1108 1108 root = findoutgoing(ui, repo, remote, force, opts)
1109 1109 else:
1110 1110 rr = list(repo.set('roots(%ld)', scmutil.revrange(repo, revs)))
1111 1111 if len(rr) != 1:
1112 1112 raise error.Abort(_('The specified revisions must have '
1113 1113 'exactly one common root'))
1114 1114 root = rr[0].node()
1115 1115
1116 1116 revs = between(repo, root, topmost, state.keep)
1117 1117 if not revs:
1118 1118 raise error.Abort(_('%s is not an ancestor of working directory') %
1119 1119 node.short(root))
1120 1120
1121 1121 ctxs = [repo[r] for r in revs]
1122 1122 if not rules:
1123 1123 comment = geteditcomment(node.short(root), node.short(topmost))
1124 1124 actions = [pick(state, r) for r in revs]
1125 1125 rules = ruleeditor(repo, ui, actions, comment)
1126 1126 else:
1127 1127 if rules == '-':
1128 1128 f = sys.stdin
1129 1129 else:
1130 1130 f = open(rules)
1131 1131 rules = f.read()
1132 1132 f.close()
1133 1133 actions = parserules(rules, state)
1134 1134 warnverifyactions(ui, repo, actions, state, ctxs)
1135 1135
1136 1136 parentctxnode = repo[root].parents()[0].node()
1137 1137
1138 1138 state.parentctxnode = parentctxnode
1139 1139 state.actions = actions
1140 1140 state.topmost = topmost
1141 1141 state.replacements = replacements
1142 1142
1143 1143 # Create a backup so we can always abort completely.
1144 1144 backupfile = None
1145 1145 if not obsolete.isenabled(repo, obsolete.createmarkersopt):
1146 1146 backupfile = repair._bundle(repo, [parentctxnode], [topmost], root,
1147 1147 'histedit')
1148 1148 state.backupfile = backupfile
1149 1149
1150 1150 # preprocess rules so that we can hide inner folds from the user
1151 1151 # and only show one editor
1152 1152 actions = state.actions[:]
1153 1153 for idx, (action, nextact) in enumerate(
1154 1154 zip(actions, actions[1:] + [None])):
1155 1155 if action.verb == 'fold' and nextact and nextact.verb == 'fold':
1156 1156 state.actions[idx].__class__ = _multifold
1157 1157
1158 1158 total = len(state.actions)
1159 1159 pos = 0
1160 1160 while state.actions:
1161 1161 state.write()
1162 1162 actobj = state.actions.pop(0)
1163 1163 pos += 1
1164 1164 ui.progress(_("editing"), pos, actobj.torule(),
1165 1165 _('changes'), total)
1166 1166 ui.debug('histedit: processing %s %s\n' % (actobj.verb,\
1167 1167 actobj.torule()))
1168 1168 parentctx, replacement_ = actobj.run()
1169 1169 state.parentctxnode = parentctx.node()
1170 1170 state.replacements.extend(replacement_)
1171 1171 state.write()
1172 1172 ui.progress(_("editing"), None)
1173 1173
1174 1174 hg.update(repo, state.parentctxnode, quietempty=True)
1175 1175
1176 1176 mapping, tmpnodes, created, ntm = processreplacement(state)
1177 1177 if mapping:
1178 1178 for prec, succs in mapping.iteritems():
1179 1179 if not succs:
1180 1180 ui.debug('histedit: %s is dropped\n' % node.short(prec))
1181 1181 else:
1182 1182 ui.debug('histedit: %s is replaced by %s\n' % (
1183 1183 node.short(prec), node.short(succs[0])))
1184 1184 if len(succs) > 1:
1185 1185 m = 'histedit: %s'
1186 1186 for n in succs[1:]:
1187 1187 ui.debug(m % node.short(n))
1188 1188
1189 1189 if supportsmarkers:
1190 1190 # Only create markers if the temp nodes weren't already removed.
1191 1191 obsolete.createmarkers(repo, ((repo[t],()) for t in sorted(tmpnodes)
1192 1192 if t in repo))
1193 1193 else:
1194 1194 cleanupnode(ui, repo, 'temp', tmpnodes)
1195 1195
1196 1196 if not state.keep:
1197 1197 if mapping:
1198 1198 movebookmarks(ui, repo, mapping, state.topmost, ntm)
1199 1199 # TODO update mq state
1200 1200 if supportsmarkers:
1201 1201 markers = []
1202 1202 # sort by revision number because it sound "right"
1203 1203 for prec in sorted(mapping, key=repo.changelog.rev):
1204 1204 succs = mapping[prec]
1205 1205 markers.append((repo[prec],
1206 1206 tuple(repo[s] for s in succs)))
1207 1207 if markers:
1208 1208 obsolete.createmarkers(repo, markers)
1209 1209 else:
1210 1210 cleanupnode(ui, repo, 'replaced', mapping)
1211 1211
1212 1212 state.clear()
1213 1213 if os.path.exists(repo.sjoin('undo')):
1214 1214 os.unlink(repo.sjoin('undo'))
1215 1215 if repo.vfs.exists('histedit-last-edit.txt'):
1216 1216 repo.vfs.unlink('histedit-last-edit.txt')
1217 1217
1218 1218 def bootstrapcontinue(ui, state, opts):
1219 1219 repo = state.repo
1220 1220 if state.actions:
1221 1221 actobj = state.actions.pop(0)
1222 1222
1223 1223 if _isdirtywc(repo):
1224 1224 actobj.continuedirty()
1225 1225 if _isdirtywc(repo):
1226 1226 abortdirty()
1227 1227
1228 1228 parentctx, replacements = actobj.continueclean()
1229 1229
1230 1230 state.parentctxnode = parentctx.node()
1231 1231 state.replacements.extend(replacements)
1232 1232
1233 1233 return state
1234 1234
1235 1235 def between(repo, old, new, keep):
1236 1236 """select and validate the set of revision to edit
1237 1237
1238 1238 When keep is false, the specified set can't have children."""
1239 1239 ctxs = list(repo.set('%n::%n', old, new))
1240 1240 if ctxs and not keep:
1241 1241 if (not obsolete.isenabled(repo, obsolete.allowunstableopt) and
1242 1242 repo.revs('(%ld::) - (%ld)', ctxs, ctxs)):
1243 1243 raise error.Abort(_('cannot edit history that would orphan nodes'))
1244 1244 if repo.revs('(%ld) and merge()', ctxs):
1245 1245 raise error.Abort(_('cannot edit history that contains merges'))
1246 1246 root = ctxs[0] # list is already sorted by repo.set
1247 1247 if not root.mutable():
1248 1248 raise error.Abort(_('cannot edit public changeset: %s') % root,
1249 1249 hint=_('see "hg help phases" for details'))
1250 1250 return [c.node() for c in ctxs]
1251 1251
1252 1252 def ruleeditor(repo, ui, actions, editcomment=""):
1253 1253 """open an editor to edit rules
1254 1254
1255 1255 rules are in the format [ [act, ctx], ...] like in state.rules
1256 1256 """
1257 1257 rules = '\n'.join([act.torule() for act in actions])
1258 1258 rules += '\n\n'
1259 1259 rules += editcomment
1260 1260 rules = ui.edit(rules, ui.username(), {'prefix': 'histedit'})
1261 1261
1262 1262 # Save edit rules in .hg/histedit-last-edit.txt in case
1263 1263 # the user needs to ask for help after something
1264 1264 # surprising happens.
1265 1265 f = open(repo.join('histedit-last-edit.txt'), 'w')
1266 1266 f.write(rules)
1267 1267 f.close()
1268 1268
1269 1269 return rules
1270 1270
1271 1271 def parserules(rules, state):
1272 1272 """Read the histedit rules string and return list of action objects """
1273 1273 rules = [l for l in (r.strip() for r in rules.splitlines())
1274 1274 if l and not l.startswith('#')]
1275 1275 actions = []
1276 1276 for r in rules:
1277 1277 if ' ' not in r:
1278 1278 raise error.ParseError(_('malformed line "%s"') % r)
1279 1279 verb, rest = r.split(' ', 1)
1280 1280
1281 1281 if verb not in actiontable:
1282 1282 raise error.ParseError(_('unknown action "%s"') % verb)
1283 1283
1284 1284 action = actiontable[verb].fromrule(state, rest)
1285 1285 actions.append(action)
1286 1286 return actions
1287 1287
1288 1288 def warnverifyactions(ui, repo, actions, state, ctxs):
1289 1289 try:
1290 1290 verifyactions(actions, state, ctxs)
1291 1291 except error.ParseError:
1292 1292 if repo.vfs.exists('histedit-last-edit.txt'):
1293 1293 ui.warn(_('warning: histedit rules saved '
1294 1294 'to: .hg/histedit-last-edit.txt\n'))
1295 1295 raise
1296 1296
1297 1297 def verifyactions(actions, state, ctxs):
1298 1298 """Verify that there exists exactly one action per given changeset and
1299 1299 other constraints.
1300 1300
1301 1301 Will abort if there are to many or too few rules, a malformed rule,
1302 1302 or a rule on a changeset outside of the user-given range.
1303 1303 """
1304 1304 expected = set(c.hex() for c in ctxs)
1305 1305 seen = set()
1306 1306 prev = None
1307 1307 for action in actions:
1308 1308 action.verify(prev)
1309 1309 prev = action
1310 1310 constraints = action.constraints()
1311 1311 for constraint in constraints:
1312 1312 if constraint not in _constraints.known():
1313 1313 raise error.ParseError(_('unknown constraint "%s"') %
1314 1314 constraint)
1315 1315
1316 1316 nodetoverify = action.nodetoverify()
1317 1317 if nodetoverify is not None:
1318 1318 ha = node.hex(nodetoverify)
1319 1319 if _constraints.noother in constraints and ha not in expected:
1320 1320 raise error.ParseError(
1321 1321 _('%s "%s" changeset was not a candidate')
1322 1322 % (action.verb, node.short(ha)),
1323 1323 hint=_('only use listed changesets'))
1324 1324 if _constraints.forceother in constraints and ha in expected:
1325 1325 raise error.ParseError(
1326 1326 _('%s "%s" changeset was not an edited list candidate')
1327 1327 % (action.verb, node.short(ha)),
1328 1328 hint=_('only use listed changesets'))
1329 1329 if _constraints.noduplicates in constraints and ha in seen:
1330 1330 raise error.ParseError(_(
1331 1331 'duplicated command for changeset %s') %
1332 1332 ha[:12])
1333 1333 seen.add(ha)
1334 1334 missing = sorted(expected - seen) # sort to stabilize output
1335 1335
1336 1336 if state.repo.ui.configbool('histedit', 'dropmissing'):
1337 1337 drops = [drop(state, node.bin(n)) for n in missing]
1338 1338 # put the in the beginning so they execute immediately and
1339 1339 # don't show in the edit-plan in the future
1340 1340 actions[:0] = drops
1341 1341 elif missing:
1342 1342 raise error.ParseError(_('missing rules for changeset %s') %
1343 1343 missing[0][:12],
1344 1344 hint=_('use "drop %s" to discard, see also: '
1345 1345 '"hg help -e histedit.config"') % missing[0][:12])
1346 1346
1347 1347 def newnodestoabort(state):
1348 1348 """process the list of replacements to return
1349 1349
1350 1350 1) the list of final node
1351 1351 2) the list of temporary node
1352 1352
1353 1353 This is meant to be used on abort as less data are required in this case.
1354 1354 """
1355 1355 replacements = state.replacements
1356 1356 allsuccs = set()
1357 1357 replaced = set()
1358 1358 for rep in replacements:
1359 1359 allsuccs.update(rep[1])
1360 1360 replaced.add(rep[0])
1361 1361 newnodes = allsuccs - replaced
1362 1362 tmpnodes = allsuccs & replaced
1363 1363 return newnodes, tmpnodes
1364 1364
1365 1365
1366 1366 def processreplacement(state):
1367 1367 """process the list of replacements to return
1368 1368
1369 1369 1) the final mapping between original and created nodes
1370 1370 2) the list of temporary node created by histedit
1371 1371 3) the list of new commit created by histedit"""
1372 1372 replacements = state.replacements
1373 1373 allsuccs = set()
1374 1374 replaced = set()
1375 1375 fullmapping = {}
1376 1376 # initialize basic set
1377 1377 # fullmapping records all operations recorded in replacement
1378 1378 for rep in replacements:
1379 1379 allsuccs.update(rep[1])
1380 1380 replaced.add(rep[0])
1381 1381 fullmapping.setdefault(rep[0], set()).update(rep[1])
1382 1382 new = allsuccs - replaced
1383 1383 tmpnodes = allsuccs & replaced
1384 1384 # Reduce content fullmapping into direct relation between original nodes
1385 1385 # and final node created during history edition
1386 1386 # Dropped changeset are replaced by an empty list
1387 1387 toproceed = set(fullmapping)
1388 1388 final = {}
1389 1389 while toproceed:
1390 1390 for x in list(toproceed):
1391 1391 succs = fullmapping[x]
1392 1392 for s in list(succs):
1393 1393 if s in toproceed:
1394 1394 # non final node with unknown closure
1395 1395 # We can't process this now
1396 1396 break
1397 1397 elif s in final:
1398 1398 # non final node, replace with closure
1399 1399 succs.remove(s)
1400 1400 succs.update(final[s])
1401 1401 else:
1402 1402 final[x] = succs
1403 1403 toproceed.remove(x)
1404 1404 # remove tmpnodes from final mapping
1405 1405 for n in tmpnodes:
1406 1406 del final[n]
1407 1407 # we expect all changes involved in final to exist in the repo
1408 1408 # turn `final` into list (topologically sorted)
1409 1409 nm = state.repo.changelog.nodemap
1410 1410 for prec, succs in final.items():
1411 1411 final[prec] = sorted(succs, key=nm.get)
1412 1412
1413 1413 # computed topmost element (necessary for bookmark)
1414 1414 if new:
1415 1415 newtopmost = sorted(new, key=state.repo.changelog.rev)[-1]
1416 1416 elif not final:
1417 1417 # Nothing rewritten at all. we won't need `newtopmost`
1418 1418 # It is the same as `oldtopmost` and `processreplacement` know it
1419 1419 newtopmost = None
1420 1420 else:
1421 1421 # every body died. The newtopmost is the parent of the root.
1422 1422 r = state.repo.changelog.rev
1423 1423 newtopmost = state.repo[sorted(final, key=r)[0]].p1().node()
1424 1424
1425 1425 return final, tmpnodes, new, newtopmost
1426 1426
1427 1427 def movebookmarks(ui, repo, mapping, oldtopmost, newtopmost):
1428 1428 """Move bookmark from old to newly created node"""
1429 1429 if not mapping:
1430 1430 # if nothing got rewritten there is not purpose for this function
1431 1431 return
1432 1432 moves = []
1433 1433 for bk, old in sorted(repo._bookmarks.iteritems()):
1434 1434 if old == oldtopmost:
1435 1435 # special case ensure bookmark stay on tip.
1436 1436 #
1437 1437 # This is arguably a feature and we may only want that for the
1438 1438 # active bookmark. But the behavior is kept compatible with the old
1439 1439 # version for now.
1440 1440 moves.append((bk, newtopmost))
1441 1441 continue
1442 1442 base = old
1443 1443 new = mapping.get(base, None)
1444 1444 if new is None:
1445 1445 continue
1446 1446 while not new:
1447 1447 # base is killed, trying with parent
1448 1448 base = repo[base].p1().node()
1449 1449 new = mapping.get(base, (base,))
1450 1450 # nothing to move
1451 1451 moves.append((bk, new[-1]))
1452 1452 if moves:
1453 1453 lock = tr = None
1454 1454 try:
1455 1455 lock = repo.lock()
1456 1456 tr = repo.transaction('histedit')
1457 1457 marks = repo._bookmarks
1458 1458 for mark, new in moves:
1459 1459 old = marks[mark]
1460 1460 ui.note(_('histedit: moving bookmarks %s from %s to %s\n')
1461 1461 % (mark, node.short(old), node.short(new)))
1462 1462 marks[mark] = new
1463 1463 marks.recordchange(tr)
1464 1464 tr.close()
1465 1465 finally:
1466 1466 release(tr, lock)
1467 1467
1468 1468 def cleanupnode(ui, repo, name, nodes):
1469 1469 """strip a group of nodes from the repository
1470 1470
1471 1471 The set of node to strip may contains unknown nodes."""
1472 1472 ui.debug('should strip %s nodes %s\n' %
1473 1473 (name, ', '.join([node.short(n) for n in nodes])))
1474 lock = None
1475 try:
1476 lock = repo.lock()
1474 with repo.lock():
1477 1475 # do not let filtering get in the way of the cleanse
1478 1476 # we should probably get rid of obsolescence marker created during the
1479 1477 # histedit, but we currently do not have such information.
1480 1478 repo = repo.unfiltered()
1481 1479 # Find all nodes that need to be stripped
1482 1480 # (we use %lr instead of %ln to silently ignore unknown items)
1483 1481 nm = repo.changelog.nodemap
1484 1482 nodes = sorted(n for n in nodes if n in nm)
1485 1483 roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
1486 1484 for c in roots:
1487 1485 # We should process node in reverse order to strip tip most first.
1488 1486 # but this trigger a bug in changegroup hook.
1489 1487 # This would reduce bundle overhead
1490 1488 repair.strip(ui, repo, c)
1491 finally:
1492 release(lock)
1493 1489
1494 1490 def stripwrapper(orig, ui, repo, nodelist, *args, **kwargs):
1495 1491 if isinstance(nodelist, str):
1496 1492 nodelist = [nodelist]
1497 1493 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1498 1494 state = histeditstate(repo)
1499 1495 state.read()
1500 1496 histedit_nodes = set([action.nodetoverify() for action
1501 1497 in state.actions if action.nodetoverify()])
1502 1498 strip_nodes = set([repo[n].node() for n in nodelist])
1503 1499 common_nodes = histedit_nodes & strip_nodes
1504 1500 if common_nodes:
1505 1501 raise error.Abort(_("histedit in progress, can't strip %s")
1506 1502 % ', '.join(node.short(x) for x in common_nodes))
1507 1503 return orig(ui, repo, nodelist, *args, **kwargs)
1508 1504
1509 1505 extensions.wrapfunction(repair, 'strip', stripwrapper)
1510 1506
1511 1507 def summaryhook(ui, repo):
1512 1508 if not os.path.exists(repo.join('histedit-state')):
1513 1509 return
1514 1510 state = histeditstate(repo)
1515 1511 state.read()
1516 1512 if state.actions:
1517 1513 # i18n: column positioning for "hg summary"
1518 1514 ui.write(_('hist: %s (histedit --continue)\n') %
1519 1515 (ui.label(_('%d remaining'), 'histedit.remaining') %
1520 1516 len(state.actions)))
1521 1517
1522 1518 def extsetup(ui):
1523 1519 cmdutil.summaryhooks.add('histedit', summaryhook)
1524 1520 cmdutil.unfinishedstates.append(
1525 1521 ['histedit-state', False, True, _('histedit in progress'),
1526 1522 _("use 'hg histedit --continue' or 'hg histedit --abort'")])
1527 1523 cmdutil.afterresolvedstates.append(
1528 1524 ['histedit-state', _('hg histedit --continue')])
1529 1525 if ui.configbool("experimental", "histeditng"):
1530 1526 globals()['base'] = action(['base', 'b'],
1531 1527 _('checkout changeset and apply further changesets from there')
1532 1528 )(base)
General Comments 0
You need to be logged in to leave comments. Login now