##// END OF EJS Templates
histedit: minor refactoring of createmarkers check...
Laurent Charignon -
r25808:425839c8 default
parent child Browse files
Show More
@@ -1,1159 +1,1160 b''
1 # histedit.py - interactive history editing for mercurial
1 # histedit.py - interactive history editing for mercurial
2 #
2 #
3 # Copyright 2009 Augie Fackler <raf@durin42.com>
3 # Copyright 2009 Augie Fackler <raf@durin42.com>
4 #
4 #
5 # This software may be used and distributed according to the terms of the
5 # This software may be used and distributed according to the terms of the
6 # GNU General Public License version 2 or any later version.
6 # GNU General Public License version 2 or any later version.
7 """interactive history editing
7 """interactive history editing
8
8
9 With this extension installed, Mercurial gains one new command: histedit. Usage
9 With this extension installed, Mercurial gains one new command: histedit. Usage
10 is as follows, assuming the following history::
10 is as follows, assuming the following history::
11
11
12 @ 3[tip] 7c2fd3b9020c 2009-04-27 18:04 -0500 durin42
12 @ 3[tip] 7c2fd3b9020c 2009-04-27 18:04 -0500 durin42
13 | Add delta
13 | Add delta
14 |
14 |
15 o 2 030b686bedc4 2009-04-27 18:04 -0500 durin42
15 o 2 030b686bedc4 2009-04-27 18:04 -0500 durin42
16 | Add gamma
16 | Add gamma
17 |
17 |
18 o 1 c561b4e977df 2009-04-27 18:04 -0500 durin42
18 o 1 c561b4e977df 2009-04-27 18:04 -0500 durin42
19 | Add beta
19 | Add beta
20 |
20 |
21 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
21 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
22 Add alpha
22 Add alpha
23
23
24 If you were to run ``hg histedit c561b4e977df``, you would see the following
24 If you were to run ``hg histedit c561b4e977df``, you would see the following
25 file open in your editor::
25 file open in your editor::
26
26
27 pick c561b4e977df Add beta
27 pick c561b4e977df Add beta
28 pick 030b686bedc4 Add gamma
28 pick 030b686bedc4 Add gamma
29 pick 7c2fd3b9020c Add delta
29 pick 7c2fd3b9020c Add delta
30
30
31 # Edit history between c561b4e977df and 7c2fd3b9020c
31 # Edit history between c561b4e977df and 7c2fd3b9020c
32 #
32 #
33 # Commits are listed from least to most recent
33 # Commits are listed from least to most recent
34 #
34 #
35 # Commands:
35 # Commands:
36 # p, pick = use commit
36 # p, pick = use commit
37 # e, edit = use commit, but stop for amending
37 # e, edit = use commit, but stop for amending
38 # f, fold = use commit, but combine it with the one above
38 # f, fold = use commit, but combine it with the one above
39 # r, roll = like fold, but discard this commit's description
39 # r, roll = like fold, but discard this commit's description
40 # d, drop = remove commit from history
40 # d, drop = remove commit from history
41 # m, mess = edit message without changing commit content
41 # m, mess = edit message without changing commit content
42 #
42 #
43
43
44 In this file, lines beginning with ``#`` are ignored. You must specify a rule
44 In this file, lines beginning with ``#`` are ignored. You must specify a rule
45 for each revision in your history. For example, if you had meant to add gamma
45 for each revision in your history. For example, if you had meant to add gamma
46 before beta, and then wanted to add delta in the same revision as beta, you
46 before beta, and then wanted to add delta in the same revision as beta, you
47 would reorganize the file to look like this::
47 would reorganize the file to look like this::
48
48
49 pick 030b686bedc4 Add gamma
49 pick 030b686bedc4 Add gamma
50 pick c561b4e977df Add beta
50 pick c561b4e977df Add beta
51 fold 7c2fd3b9020c Add delta
51 fold 7c2fd3b9020c Add delta
52
52
53 # Edit history between c561b4e977df and 7c2fd3b9020c
53 # Edit history between c561b4e977df and 7c2fd3b9020c
54 #
54 #
55 # Commits are listed from least to most recent
55 # Commits are listed from least to most recent
56 #
56 #
57 # Commands:
57 # Commands:
58 # p, pick = use commit
58 # p, pick = use commit
59 # e, edit = use commit, but stop for amending
59 # e, edit = use commit, but stop for amending
60 # f, fold = use commit, but combine it with the one above
60 # f, fold = use commit, but combine it with the one above
61 # r, roll = like fold, but discard this commit's description
61 # r, roll = like fold, but discard this commit's description
62 # d, drop = remove commit from history
62 # d, drop = remove commit from history
63 # m, mess = edit message without changing commit content
63 # m, mess = edit message without changing commit content
64 #
64 #
65
65
66 At which point you close the editor and ``histedit`` starts working. When you
66 At which point you close the editor and ``histedit`` starts working. When you
67 specify a ``fold`` operation, ``histedit`` will open an editor when it folds
67 specify a ``fold`` operation, ``histedit`` will open an editor when it folds
68 those revisions together, offering you a chance to clean up the commit message::
68 those revisions together, offering you a chance to clean up the commit message::
69
69
70 Add beta
70 Add beta
71 ***
71 ***
72 Add delta
72 Add delta
73
73
74 Edit the commit message to your liking, then close the editor. For
74 Edit the commit message to your liking, then close the editor. For
75 this example, let's assume that the commit message was changed to
75 this example, let's assume that the commit message was changed to
76 ``Add beta and delta.`` After histedit has run and had a chance to
76 ``Add beta and delta.`` After histedit has run and had a chance to
77 remove any old or temporary revisions it needed, the history looks
77 remove any old or temporary revisions it needed, the history looks
78 like this::
78 like this::
79
79
80 @ 2[tip] 989b4d060121 2009-04-27 18:04 -0500 durin42
80 @ 2[tip] 989b4d060121 2009-04-27 18:04 -0500 durin42
81 | Add beta and delta.
81 | Add beta and delta.
82 |
82 |
83 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
83 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
84 | Add gamma
84 | Add gamma
85 |
85 |
86 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
86 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
87 Add alpha
87 Add alpha
88
88
89 Note that ``histedit`` does *not* remove any revisions (even its own temporary
89 Note that ``histedit`` does *not* remove any revisions (even its own temporary
90 ones) until after it has completed all the editing operations, so it will
90 ones) until after it has completed all the editing operations, so it will
91 probably perform several strip operations when it's done. For the above example,
91 probably perform several strip operations when it's done. For the above example,
92 it had to run strip twice. Strip can be slow depending on a variety of factors,
92 it had to run strip twice. Strip can be slow depending on a variety of factors,
93 so you might need to be a little patient. You can choose to keep the original
93 so you might need to be a little patient. You can choose to keep the original
94 revisions by passing the ``--keep`` flag.
94 revisions by passing the ``--keep`` flag.
95
95
96 The ``edit`` operation will drop you back to a command prompt,
96 The ``edit`` operation will drop you back to a command prompt,
97 allowing you to edit files freely, or even use ``hg record`` to commit
97 allowing you to edit files freely, or even use ``hg record`` to commit
98 some changes as a separate commit. When you're done, any remaining
98 some changes as a separate commit. When you're done, any remaining
99 uncommitted changes will be committed as well. When done, run ``hg
99 uncommitted changes will be committed as well. When done, run ``hg
100 histedit --continue`` to finish this step. You'll be prompted for a
100 histedit --continue`` to finish this step. You'll be prompted for a
101 new commit message, but the default commit message will be the
101 new commit message, but the default commit message will be the
102 original message for the ``edit`` ed revision.
102 original message for the ``edit`` ed revision.
103
103
104 The ``message`` operation will give you a chance to revise a commit
104 The ``message`` operation will give you a chance to revise a commit
105 message without changing the contents. It's a shortcut for doing
105 message without changing the contents. It's a shortcut for doing
106 ``edit`` immediately followed by `hg histedit --continue``.
106 ``edit`` immediately followed by `hg histedit --continue``.
107
107
108 If ``histedit`` encounters a conflict when moving a revision (while
108 If ``histedit`` encounters a conflict when moving a revision (while
109 handling ``pick`` or ``fold``), it'll stop in a similar manner to
109 handling ``pick`` or ``fold``), it'll stop in a similar manner to
110 ``edit`` with the difference that it won't prompt you for a commit
110 ``edit`` with the difference that it won't prompt you for a commit
111 message when done. If you decide at this point that you don't like how
111 message when done. If you decide at this point that you don't like how
112 much work it will be to rearrange history, or that you made a mistake,
112 much work it will be to rearrange history, or that you made a mistake,
113 you can use ``hg histedit --abort`` to abandon the new changes you
113 you can use ``hg histedit --abort`` to abandon the new changes you
114 have made and return to the state before you attempted to edit your
114 have made and return to the state before you attempted to edit your
115 history.
115 history.
116
116
117 If we clone the histedit-ed example repository above and add four more
117 If we clone the histedit-ed example repository above and add four more
118 changes, such that we have the following history::
118 changes, such that we have the following history::
119
119
120 @ 6[tip] 038383181893 2009-04-27 18:04 -0500 stefan
120 @ 6[tip] 038383181893 2009-04-27 18:04 -0500 stefan
121 | Add theta
121 | Add theta
122 |
122 |
123 o 5 140988835471 2009-04-27 18:04 -0500 stefan
123 o 5 140988835471 2009-04-27 18:04 -0500 stefan
124 | Add eta
124 | Add eta
125 |
125 |
126 o 4 122930637314 2009-04-27 18:04 -0500 stefan
126 o 4 122930637314 2009-04-27 18:04 -0500 stefan
127 | Add zeta
127 | Add zeta
128 |
128 |
129 o 3 836302820282 2009-04-27 18:04 -0500 stefan
129 o 3 836302820282 2009-04-27 18:04 -0500 stefan
130 | Add epsilon
130 | Add epsilon
131 |
131 |
132 o 2 989b4d060121 2009-04-27 18:04 -0500 durin42
132 o 2 989b4d060121 2009-04-27 18:04 -0500 durin42
133 | Add beta and delta.
133 | Add beta and delta.
134 |
134 |
135 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
135 o 1 081603921c3f 2009-04-27 18:04 -0500 durin42
136 | Add gamma
136 | Add gamma
137 |
137 |
138 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
138 o 0 d8d2fcd0e319 2009-04-27 18:04 -0500 durin42
139 Add alpha
139 Add alpha
140
140
141 If you run ``hg histedit --outgoing`` on the clone then it is the same
141 If you run ``hg histedit --outgoing`` on the clone then it is the same
142 as running ``hg histedit 836302820282``. If you need plan to push to a
142 as running ``hg histedit 836302820282``. If you need plan to push to a
143 repository that Mercurial does not detect to be related to the source
143 repository that Mercurial does not detect to be related to the source
144 repo, you can add a ``--force`` option.
144 repo, you can add a ``--force`` option.
145
145
146 Histedit rule lines are truncated to 80 characters by default. You
146 Histedit rule lines are truncated to 80 characters by default. You
147 can customise this behaviour by setting a different length in your
147 can customise this behaviour by setting a different length in your
148 configuration file::
148 configuration file::
149
149
150 [histedit]
150 [histedit]
151 linelen = 120 # truncate rule lines at 120 characters
151 linelen = 120 # truncate rule lines at 120 characters
152 """
152 """
153
153
154 try:
154 try:
155 import cPickle as pickle
155 import cPickle as pickle
156 pickle.dump # import now
156 pickle.dump # import now
157 except ImportError:
157 except ImportError:
158 import pickle
158 import pickle
159 import errno
159 import errno
160 import os
160 import os
161 import sys
161 import sys
162
162
163 from mercurial import cmdutil
163 from mercurial import cmdutil
164 from mercurial import discovery
164 from mercurial import discovery
165 from mercurial import error
165 from mercurial import error
166 from mercurial import changegroup
166 from mercurial import changegroup
167 from mercurial import copies
167 from mercurial import copies
168 from mercurial import context
168 from mercurial import context
169 from mercurial import exchange
169 from mercurial import exchange
170 from mercurial import extensions
170 from mercurial import extensions
171 from mercurial import hg
171 from mercurial import hg
172 from mercurial import node
172 from mercurial import node
173 from mercurial import repair
173 from mercurial import repair
174 from mercurial import scmutil
174 from mercurial import scmutil
175 from mercurial import util
175 from mercurial import util
176 from mercurial import obsolete
176 from mercurial import obsolete
177 from mercurial import merge as mergemod
177 from mercurial import merge as mergemod
178 from mercurial.lock import release
178 from mercurial.lock import release
179 from mercurial.i18n import _
179 from mercurial.i18n import _
180
180
181 cmdtable = {}
181 cmdtable = {}
182 command = cmdutil.command(cmdtable)
182 command = cmdutil.command(cmdtable)
183
183
184 # Note for extension authors: ONLY specify testedwith = 'internal' for
184 # Note for extension authors: ONLY specify testedwith = 'internal' for
185 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
185 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
186 # be specifying the version(s) of Mercurial they are tested with, or
186 # be specifying the version(s) of Mercurial they are tested with, or
187 # leave the attribute unspecified.
187 # leave the attribute unspecified.
188 testedwith = 'internal'
188 testedwith = 'internal'
189
189
190 # i18n: command names and abbreviations must remain untranslated
190 # i18n: command names and abbreviations must remain untranslated
191 editcomment = _("""# Edit history between %s and %s
191 editcomment = _("""# Edit history between %s and %s
192 #
192 #
193 # Commits are listed from least to most recent
193 # Commits are listed from least to most recent
194 #
194 #
195 # Commands:
195 # Commands:
196 # p, pick = use commit
196 # p, pick = use commit
197 # e, edit = use commit, but stop for amending
197 # e, edit = use commit, but stop for amending
198 # f, fold = use commit, but combine it with the one above
198 # f, fold = use commit, but combine it with the one above
199 # r, roll = like fold, but discard this commit's description
199 # r, roll = like fold, but discard this commit's description
200 # d, drop = remove commit from history
200 # d, drop = remove commit from history
201 # m, mess = edit message without changing commit content
201 # m, mess = edit message without changing commit content
202 #
202 #
203 """)
203 """)
204
204
205 class histeditstate(object):
205 class histeditstate(object):
206 def __init__(self, repo, parentctxnode=None, rules=None, keep=None,
206 def __init__(self, repo, parentctxnode=None, rules=None, keep=None,
207 topmost=None, replacements=None, lock=None, wlock=None):
207 topmost=None, replacements=None, lock=None, wlock=None):
208 self.repo = repo
208 self.repo = repo
209 self.rules = rules
209 self.rules = rules
210 self.keep = keep
210 self.keep = keep
211 self.topmost = topmost
211 self.topmost = topmost
212 self.parentctxnode = parentctxnode
212 self.parentctxnode = parentctxnode
213 self.lock = lock
213 self.lock = lock
214 self.wlock = wlock
214 self.wlock = wlock
215 self.backupfile = None
215 self.backupfile = None
216 if replacements is None:
216 if replacements is None:
217 self.replacements = []
217 self.replacements = []
218 else:
218 else:
219 self.replacements = replacements
219 self.replacements = replacements
220
220
221 def read(self):
221 def read(self):
222 """Load histedit state from disk and set fields appropriately."""
222 """Load histedit state from disk and set fields appropriately."""
223 try:
223 try:
224 fp = self.repo.vfs('histedit-state', 'r')
224 fp = self.repo.vfs('histedit-state', 'r')
225 except IOError as err:
225 except IOError as err:
226 if err.errno != errno.ENOENT:
226 if err.errno != errno.ENOENT:
227 raise
227 raise
228 raise util.Abort(_('no histedit in progress'))
228 raise util.Abort(_('no histedit in progress'))
229
229
230 try:
230 try:
231 data = pickle.load(fp)
231 data = pickle.load(fp)
232 parentctxnode, rules, keep, topmost, replacements = data
232 parentctxnode, rules, keep, topmost, replacements = data
233 backupfile = None
233 backupfile = None
234 except pickle.UnpicklingError:
234 except pickle.UnpicklingError:
235 data = self._load()
235 data = self._load()
236 parentctxnode, rules, keep, topmost, replacements, backupfile = data
236 parentctxnode, rules, keep, topmost, replacements, backupfile = data
237
237
238 self.parentctxnode = parentctxnode
238 self.parentctxnode = parentctxnode
239 self.rules = rules
239 self.rules = rules
240 self.keep = keep
240 self.keep = keep
241 self.topmost = topmost
241 self.topmost = topmost
242 self.replacements = replacements
242 self.replacements = replacements
243 self.backupfile = backupfile
243 self.backupfile = backupfile
244
244
245 def write(self):
245 def write(self):
246 fp = self.repo.vfs('histedit-state', 'w')
246 fp = self.repo.vfs('histedit-state', 'w')
247 fp.write('v1\n')
247 fp.write('v1\n')
248 fp.write('%s\n' % node.hex(self.parentctxnode))
248 fp.write('%s\n' % node.hex(self.parentctxnode))
249 fp.write('%s\n' % node.hex(self.topmost))
249 fp.write('%s\n' % node.hex(self.topmost))
250 fp.write('%s\n' % self.keep)
250 fp.write('%s\n' % self.keep)
251 fp.write('%d\n' % len(self.rules))
251 fp.write('%d\n' % len(self.rules))
252 for rule in self.rules:
252 for rule in self.rules:
253 fp.write('%s\n' % rule[0]) # action
253 fp.write('%s\n' % rule[0]) # action
254 fp.write('%s\n' % rule[1]) # remainder
254 fp.write('%s\n' % rule[1]) # remainder
255 fp.write('%d\n' % len(self.replacements))
255 fp.write('%d\n' % len(self.replacements))
256 for replacement in self.replacements:
256 for replacement in self.replacements:
257 fp.write('%s%s\n' % (node.hex(replacement[0]), ''.join(node.hex(r)
257 fp.write('%s%s\n' % (node.hex(replacement[0]), ''.join(node.hex(r)
258 for r in replacement[1])))
258 for r in replacement[1])))
259 backupfile = self.backupfile
259 backupfile = self.backupfile
260 if not backupfile:
260 if not backupfile:
261 backupfile = ''
261 backupfile = ''
262 fp.write('%s\n' % backupfile)
262 fp.write('%s\n' % backupfile)
263 fp.close()
263 fp.close()
264
264
265 def _load(self):
265 def _load(self):
266 fp = self.repo.vfs('histedit-state', 'r')
266 fp = self.repo.vfs('histedit-state', 'r')
267 lines = [l[:-1] for l in fp.readlines()]
267 lines = [l[:-1] for l in fp.readlines()]
268
268
269 index = 0
269 index = 0
270 lines[index] # version number
270 lines[index] # version number
271 index += 1
271 index += 1
272
272
273 parentctxnode = node.bin(lines[index])
273 parentctxnode = node.bin(lines[index])
274 index += 1
274 index += 1
275
275
276 topmost = node.bin(lines[index])
276 topmost = node.bin(lines[index])
277 index += 1
277 index += 1
278
278
279 keep = lines[index] == 'True'
279 keep = lines[index] == 'True'
280 index += 1
280 index += 1
281
281
282 # Rules
282 # Rules
283 rules = []
283 rules = []
284 rulelen = int(lines[index])
284 rulelen = int(lines[index])
285 index += 1
285 index += 1
286 for i in xrange(rulelen):
286 for i in xrange(rulelen):
287 ruleaction = lines[index]
287 ruleaction = lines[index]
288 index += 1
288 index += 1
289 rule = lines[index]
289 rule = lines[index]
290 index += 1
290 index += 1
291 rules.append((ruleaction, rule))
291 rules.append((ruleaction, rule))
292
292
293 # Replacements
293 # Replacements
294 replacements = []
294 replacements = []
295 replacementlen = int(lines[index])
295 replacementlen = int(lines[index])
296 index += 1
296 index += 1
297 for i in xrange(replacementlen):
297 for i in xrange(replacementlen):
298 replacement = lines[index]
298 replacement = lines[index]
299 original = node.bin(replacement[:40])
299 original = node.bin(replacement[:40])
300 succ = [node.bin(replacement[i:i + 40]) for i in
300 succ = [node.bin(replacement[i:i + 40]) for i in
301 range(40, len(replacement), 40)]
301 range(40, len(replacement), 40)]
302 replacements.append((original, succ))
302 replacements.append((original, succ))
303 index += 1
303 index += 1
304
304
305 backupfile = lines[index]
305 backupfile = lines[index]
306 index += 1
306 index += 1
307
307
308 fp.close()
308 fp.close()
309
309
310 return parentctxnode, rules, keep, topmost, replacements, backupfile
310 return parentctxnode, rules, keep, topmost, replacements, backupfile
311
311
312 def clear(self):
312 def clear(self):
313 self.repo.vfs.unlink('histedit-state')
313 self.repo.vfs.unlink('histedit-state')
314
314
315 class histeditaction(object):
315 class histeditaction(object):
316 def __init__(self, state, node):
316 def __init__(self, state, node):
317 self.state = state
317 self.state = state
318 self.repo = state.repo
318 self.repo = state.repo
319 self.node = node
319 self.node = node
320
320
321 @classmethod
321 @classmethod
322 def fromrule(cls, state, rule):
322 def fromrule(cls, state, rule):
323 """Parses the given rule, returning an instance of the histeditaction.
323 """Parses the given rule, returning an instance of the histeditaction.
324 """
324 """
325 repo = state.repo
325 repo = state.repo
326 rulehash = rule.strip().split(' ', 1)[0]
326 rulehash = rule.strip().split(' ', 1)[0]
327 try:
327 try:
328 node = repo[rulehash].node()
328 node = repo[rulehash].node()
329 except error.RepoError:
329 except error.RepoError:
330 raise util.Abort(_('unknown changeset %s listed') % rulehash[:12])
330 raise util.Abort(_('unknown changeset %s listed') % rulehash[:12])
331 return cls(state, node)
331 return cls(state, node)
332
332
333 def run(self):
333 def run(self):
334 """Runs the action. The default behavior is simply apply the action's
334 """Runs the action. The default behavior is simply apply the action's
335 rulectx onto the current parentctx."""
335 rulectx onto the current parentctx."""
336 self.applychange()
336 self.applychange()
337 self.continuedirty()
337 self.continuedirty()
338 return self.continueclean()
338 return self.continueclean()
339
339
340 def applychange(self):
340 def applychange(self):
341 """Applies the changes from this action's rulectx onto the current
341 """Applies the changes from this action's rulectx onto the current
342 parentctx, but does not commit them."""
342 parentctx, but does not commit them."""
343 repo = self.repo
343 repo = self.repo
344 rulectx = repo[self.node]
344 rulectx = repo[self.node]
345 hg.update(repo, self.state.parentctxnode)
345 hg.update(repo, self.state.parentctxnode)
346 stats = applychanges(repo.ui, repo, rulectx, {})
346 stats = applychanges(repo.ui, repo, rulectx, {})
347 if stats and stats[3] > 0:
347 if stats and stats[3] > 0:
348 raise error.InterventionRequired(_('Fix up the change and run '
348 raise error.InterventionRequired(_('Fix up the change and run '
349 'hg histedit --continue'))
349 'hg histedit --continue'))
350
350
351 def continuedirty(self):
351 def continuedirty(self):
352 """Continues the action when changes have been applied to the working
352 """Continues the action when changes have been applied to the working
353 copy. The default behavior is to commit the dirty changes."""
353 copy. The default behavior is to commit the dirty changes."""
354 repo = self.repo
354 repo = self.repo
355 rulectx = repo[self.node]
355 rulectx = repo[self.node]
356
356
357 editor = self.commiteditor()
357 editor = self.commiteditor()
358 commit = commitfuncfor(repo, rulectx)
358 commit = commitfuncfor(repo, rulectx)
359
359
360 commit(text=rulectx.description(), user=rulectx.user(),
360 commit(text=rulectx.description(), user=rulectx.user(),
361 date=rulectx.date(), extra=rulectx.extra(), editor=editor)
361 date=rulectx.date(), extra=rulectx.extra(), editor=editor)
362
362
363 def commiteditor(self):
363 def commiteditor(self):
364 """The editor to be used to edit the commit message."""
364 """The editor to be used to edit the commit message."""
365 return False
365 return False
366
366
367 def continueclean(self):
367 def continueclean(self):
368 """Continues the action when the working copy is clean. The default
368 """Continues the action when the working copy is clean. The default
369 behavior is to accept the current commit as the new version of the
369 behavior is to accept the current commit as the new version of the
370 rulectx."""
370 rulectx."""
371 ctx = self.repo['.']
371 ctx = self.repo['.']
372 if ctx.node() == self.state.parentctxnode:
372 if ctx.node() == self.state.parentctxnode:
373 self.repo.ui.warn(_('%s: empty changeset\n') %
373 self.repo.ui.warn(_('%s: empty changeset\n') %
374 node.short(self.node))
374 node.short(self.node))
375 return ctx, [(self.node, tuple())]
375 return ctx, [(self.node, tuple())]
376 if ctx.node() == self.node:
376 if ctx.node() == self.node:
377 # Nothing changed
377 # Nothing changed
378 return ctx, []
378 return ctx, []
379 return ctx, [(self.node, (ctx.node(),))]
379 return ctx, [(self.node, (ctx.node(),))]
380
380
381 def commitfuncfor(repo, src):
381 def commitfuncfor(repo, src):
382 """Build a commit function for the replacement of <src>
382 """Build a commit function for the replacement of <src>
383
383
384 This function ensure we apply the same treatment to all changesets.
384 This function ensure we apply the same treatment to all changesets.
385
385
386 - Add a 'histedit_source' entry in extra.
386 - Add a 'histedit_source' entry in extra.
387
387
388 Note that fold has its own separated logic because its handling is a bit
388 Note that fold has its own separated logic because its handling is a bit
389 different and not easily factored out of the fold method.
389 different and not easily factored out of the fold method.
390 """
390 """
391 phasemin = src.phase()
391 phasemin = src.phase()
392 def commitfunc(**kwargs):
392 def commitfunc(**kwargs):
393 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
393 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
394 try:
394 try:
395 repo.ui.setconfig('phases', 'new-commit', phasemin,
395 repo.ui.setconfig('phases', 'new-commit', phasemin,
396 'histedit')
396 'histedit')
397 extra = kwargs.get('extra', {}).copy()
397 extra = kwargs.get('extra', {}).copy()
398 extra['histedit_source'] = src.hex()
398 extra['histedit_source'] = src.hex()
399 kwargs['extra'] = extra
399 kwargs['extra'] = extra
400 return repo.commit(**kwargs)
400 return repo.commit(**kwargs)
401 finally:
401 finally:
402 repo.ui.restoreconfig(phasebackup)
402 repo.ui.restoreconfig(phasebackup)
403 return commitfunc
403 return commitfunc
404
404
405 def applychanges(ui, repo, ctx, opts):
405 def applychanges(ui, repo, ctx, opts):
406 """Merge changeset from ctx (only) in the current working directory"""
406 """Merge changeset from ctx (only) in the current working directory"""
407 wcpar = repo.dirstate.parents()[0]
407 wcpar = repo.dirstate.parents()[0]
408 if ctx.p1().node() == wcpar:
408 if ctx.p1().node() == wcpar:
409 # edition ar "in place" we do not need to make any merge,
409 # edition ar "in place" we do not need to make any merge,
410 # just applies changes on parent for edition
410 # just applies changes on parent for edition
411 cmdutil.revert(ui, repo, ctx, (wcpar, node.nullid), all=True)
411 cmdutil.revert(ui, repo, ctx, (wcpar, node.nullid), all=True)
412 stats = None
412 stats = None
413 else:
413 else:
414 try:
414 try:
415 # ui.forcemerge is an internal variable, do not document
415 # ui.forcemerge is an internal variable, do not document
416 repo.ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
416 repo.ui.setconfig('ui', 'forcemerge', opts.get('tool', ''),
417 'histedit')
417 'histedit')
418 stats = mergemod.graft(repo, ctx, ctx.p1(), ['local', 'histedit'])
418 stats = mergemod.graft(repo, ctx, ctx.p1(), ['local', 'histedit'])
419 finally:
419 finally:
420 repo.ui.setconfig('ui', 'forcemerge', '', 'histedit')
420 repo.ui.setconfig('ui', 'forcemerge', '', 'histedit')
421 return stats
421 return stats
422
422
423 def collapse(repo, first, last, commitopts, skipprompt=False):
423 def collapse(repo, first, last, commitopts, skipprompt=False):
424 """collapse the set of revisions from first to last as new one.
424 """collapse the set of revisions from first to last as new one.
425
425
426 Expected commit options are:
426 Expected commit options are:
427 - message
427 - message
428 - date
428 - date
429 - username
429 - username
430 Commit message is edited in all cases.
430 Commit message is edited in all cases.
431
431
432 This function works in memory."""
432 This function works in memory."""
433 ctxs = list(repo.set('%d::%d', first, last))
433 ctxs = list(repo.set('%d::%d', first, last))
434 if not ctxs:
434 if not ctxs:
435 return None
435 return None
436 for c in ctxs:
436 for c in ctxs:
437 if not c.mutable():
437 if not c.mutable():
438 raise util.Abort(
438 raise util.Abort(
439 _("cannot fold into public change %s") % node.short(c.node()))
439 _("cannot fold into public change %s") % node.short(c.node()))
440 base = first.parents()[0]
440 base = first.parents()[0]
441
441
442 # commit a new version of the old changeset, including the update
442 # commit a new version of the old changeset, including the update
443 # collect all files which might be affected
443 # collect all files which might be affected
444 files = set()
444 files = set()
445 for ctx in ctxs:
445 for ctx in ctxs:
446 files.update(ctx.files())
446 files.update(ctx.files())
447
447
448 # Recompute copies (avoid recording a -> b -> a)
448 # Recompute copies (avoid recording a -> b -> a)
449 copied = copies.pathcopies(base, last)
449 copied = copies.pathcopies(base, last)
450
450
451 # prune files which were reverted by the updates
451 # prune files which were reverted by the updates
452 def samefile(f):
452 def samefile(f):
453 if f in last.manifest():
453 if f in last.manifest():
454 a = last.filectx(f)
454 a = last.filectx(f)
455 if f in base.manifest():
455 if f in base.manifest():
456 b = base.filectx(f)
456 b = base.filectx(f)
457 return (a.data() == b.data()
457 return (a.data() == b.data()
458 and a.flags() == b.flags())
458 and a.flags() == b.flags())
459 else:
459 else:
460 return False
460 return False
461 else:
461 else:
462 return f not in base.manifest()
462 return f not in base.manifest()
463 files = [f for f in files if not samefile(f)]
463 files = [f for f in files if not samefile(f)]
464 # commit version of these files as defined by head
464 # commit version of these files as defined by head
465 headmf = last.manifest()
465 headmf = last.manifest()
466 def filectxfn(repo, ctx, path):
466 def filectxfn(repo, ctx, path):
467 if path in headmf:
467 if path in headmf:
468 fctx = last[path]
468 fctx = last[path]
469 flags = fctx.flags()
469 flags = fctx.flags()
470 mctx = context.memfilectx(repo,
470 mctx = context.memfilectx(repo,
471 fctx.path(), fctx.data(),
471 fctx.path(), fctx.data(),
472 islink='l' in flags,
472 islink='l' in flags,
473 isexec='x' in flags,
473 isexec='x' in flags,
474 copied=copied.get(path))
474 copied=copied.get(path))
475 return mctx
475 return mctx
476 return None
476 return None
477
477
478 if commitopts.get('message'):
478 if commitopts.get('message'):
479 message = commitopts['message']
479 message = commitopts['message']
480 else:
480 else:
481 message = first.description()
481 message = first.description()
482 user = commitopts.get('user')
482 user = commitopts.get('user')
483 date = commitopts.get('date')
483 date = commitopts.get('date')
484 extra = commitopts.get('extra')
484 extra = commitopts.get('extra')
485
485
486 parents = (first.p1().node(), first.p2().node())
486 parents = (first.p1().node(), first.p2().node())
487 editor = None
487 editor = None
488 if not skipprompt:
488 if not skipprompt:
489 editor = cmdutil.getcommiteditor(edit=True, editform='histedit.fold')
489 editor = cmdutil.getcommiteditor(edit=True, editform='histedit.fold')
490 new = context.memctx(repo,
490 new = context.memctx(repo,
491 parents=parents,
491 parents=parents,
492 text=message,
492 text=message,
493 files=files,
493 files=files,
494 filectxfn=filectxfn,
494 filectxfn=filectxfn,
495 user=user,
495 user=user,
496 date=date,
496 date=date,
497 extra=extra,
497 extra=extra,
498 editor=editor)
498 editor=editor)
499 return repo.commitctx(new)
499 return repo.commitctx(new)
500
500
501 class pick(histeditaction):
501 class pick(histeditaction):
502 def run(self):
502 def run(self):
503 rulectx = self.repo[self.node]
503 rulectx = self.repo[self.node]
504 if rulectx.parents()[0].node() == self.state.parentctxnode:
504 if rulectx.parents()[0].node() == self.state.parentctxnode:
505 self.repo.ui.debug('node %s unchanged\n' % node.short(self.node))
505 self.repo.ui.debug('node %s unchanged\n' % node.short(self.node))
506 return rulectx, []
506 return rulectx, []
507
507
508 return super(pick, self).run()
508 return super(pick, self).run()
509
509
510 class edit(histeditaction):
510 class edit(histeditaction):
511 def run(self):
511 def run(self):
512 repo = self.repo
512 repo = self.repo
513 rulectx = repo[self.node]
513 rulectx = repo[self.node]
514 hg.update(repo, self.state.parentctxnode)
514 hg.update(repo, self.state.parentctxnode)
515 applychanges(repo.ui, repo, rulectx, {})
515 applychanges(repo.ui, repo, rulectx, {})
516 raise error.InterventionRequired(
516 raise error.InterventionRequired(
517 _('Make changes as needed, you may commit or record as needed '
517 _('Make changes as needed, you may commit or record as needed '
518 'now.\nWhen you are finished, run hg histedit --continue to '
518 'now.\nWhen you are finished, run hg histedit --continue to '
519 'resume.'))
519 'resume.'))
520
520
521 def commiteditor(self):
521 def commiteditor(self):
522 return cmdutil.getcommiteditor(edit=True, editform='histedit.edit')
522 return cmdutil.getcommiteditor(edit=True, editform='histedit.edit')
523
523
524 class fold(histeditaction):
524 class fold(histeditaction):
525 def continuedirty(self):
525 def continuedirty(self):
526 repo = self.repo
526 repo = self.repo
527 rulectx = repo[self.node]
527 rulectx = repo[self.node]
528
528
529 commit = commitfuncfor(repo, rulectx)
529 commit = commitfuncfor(repo, rulectx)
530 commit(text='fold-temp-revision %s' % node.short(self.node),
530 commit(text='fold-temp-revision %s' % node.short(self.node),
531 user=rulectx.user(), date=rulectx.date(),
531 user=rulectx.user(), date=rulectx.date(),
532 extra=rulectx.extra())
532 extra=rulectx.extra())
533
533
534 def continueclean(self):
534 def continueclean(self):
535 repo = self.repo
535 repo = self.repo
536 ctx = repo['.']
536 ctx = repo['.']
537 rulectx = repo[self.node]
537 rulectx = repo[self.node]
538 parentctxnode = self.state.parentctxnode
538 parentctxnode = self.state.parentctxnode
539 if ctx.node() == parentctxnode:
539 if ctx.node() == parentctxnode:
540 repo.ui.warn(_('%s: empty changeset\n') %
540 repo.ui.warn(_('%s: empty changeset\n') %
541 node.short(self.node))
541 node.short(self.node))
542 return ctx, [(self.node, (parentctxnode,))]
542 return ctx, [(self.node, (parentctxnode,))]
543
543
544 parentctx = repo[parentctxnode]
544 parentctx = repo[parentctxnode]
545 newcommits = set(c.node() for c in repo.set('(%d::. - %d)', parentctx,
545 newcommits = set(c.node() for c in repo.set('(%d::. - %d)', parentctx,
546 parentctx))
546 parentctx))
547 if not newcommits:
547 if not newcommits:
548 repo.ui.warn(_('%s: cannot fold - working copy is not a '
548 repo.ui.warn(_('%s: cannot fold - working copy is not a '
549 'descendant of previous commit %s\n') %
549 'descendant of previous commit %s\n') %
550 (node.short(self.node), node.short(parentctxnode)))
550 (node.short(self.node), node.short(parentctxnode)))
551 return ctx, [(self.node, (ctx.node(),))]
551 return ctx, [(self.node, (ctx.node(),))]
552
552
553 middlecommits = newcommits.copy()
553 middlecommits = newcommits.copy()
554 middlecommits.discard(ctx.node())
554 middlecommits.discard(ctx.node())
555
555
556 return self.finishfold(repo.ui, repo, parentctx, rulectx, ctx.node(),
556 return self.finishfold(repo.ui, repo, parentctx, rulectx, ctx.node(),
557 middlecommits)
557 middlecommits)
558
558
559 def skipprompt(self):
559 def skipprompt(self):
560 return False
560 return False
561
561
562 def finishfold(self, ui, repo, ctx, oldctx, newnode, internalchanges):
562 def finishfold(self, ui, repo, ctx, oldctx, newnode, internalchanges):
563 parent = ctx.parents()[0].node()
563 parent = ctx.parents()[0].node()
564 hg.update(repo, parent)
564 hg.update(repo, parent)
565 ### prepare new commit data
565 ### prepare new commit data
566 commitopts = {}
566 commitopts = {}
567 commitopts['user'] = ctx.user()
567 commitopts['user'] = ctx.user()
568 # commit message
568 # commit message
569 if self.skipprompt():
569 if self.skipprompt():
570 newmessage = ctx.description()
570 newmessage = ctx.description()
571 else:
571 else:
572 newmessage = '\n***\n'.join(
572 newmessage = '\n***\n'.join(
573 [ctx.description()] +
573 [ctx.description()] +
574 [repo[r].description() for r in internalchanges] +
574 [repo[r].description() for r in internalchanges] +
575 [oldctx.description()]) + '\n'
575 [oldctx.description()]) + '\n'
576 commitopts['message'] = newmessage
576 commitopts['message'] = newmessage
577 # date
577 # date
578 commitopts['date'] = max(ctx.date(), oldctx.date())
578 commitopts['date'] = max(ctx.date(), oldctx.date())
579 extra = ctx.extra().copy()
579 extra = ctx.extra().copy()
580 # histedit_source
580 # histedit_source
581 # note: ctx is likely a temporary commit but that the best we can do
581 # note: ctx is likely a temporary commit but that the best we can do
582 # here. This is sufficient to solve issue3681 anyway.
582 # here. This is sufficient to solve issue3681 anyway.
583 extra['histedit_source'] = '%s,%s' % (ctx.hex(), oldctx.hex())
583 extra['histedit_source'] = '%s,%s' % (ctx.hex(), oldctx.hex())
584 commitopts['extra'] = extra
584 commitopts['extra'] = extra
585 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
585 phasebackup = repo.ui.backupconfig('phases', 'new-commit')
586 try:
586 try:
587 phasemin = max(ctx.phase(), oldctx.phase())
587 phasemin = max(ctx.phase(), oldctx.phase())
588 repo.ui.setconfig('phases', 'new-commit', phasemin, 'histedit')
588 repo.ui.setconfig('phases', 'new-commit', phasemin, 'histedit')
589 n = collapse(repo, ctx, repo[newnode], commitopts,
589 n = collapse(repo, ctx, repo[newnode], commitopts,
590 skipprompt=self.skipprompt())
590 skipprompt=self.skipprompt())
591 finally:
591 finally:
592 repo.ui.restoreconfig(phasebackup)
592 repo.ui.restoreconfig(phasebackup)
593 if n is None:
593 if n is None:
594 return ctx, []
594 return ctx, []
595 hg.update(repo, n)
595 hg.update(repo, n)
596 replacements = [(oldctx.node(), (newnode,)),
596 replacements = [(oldctx.node(), (newnode,)),
597 (ctx.node(), (n,)),
597 (ctx.node(), (n,)),
598 (newnode, (n,)),
598 (newnode, (n,)),
599 ]
599 ]
600 for ich in internalchanges:
600 for ich in internalchanges:
601 replacements.append((ich, (n,)))
601 replacements.append((ich, (n,)))
602 return repo[n], replacements
602 return repo[n], replacements
603
603
604 class rollup(fold):
604 class rollup(fold):
605 def skipprompt(self):
605 def skipprompt(self):
606 return True
606 return True
607
607
608 class drop(histeditaction):
608 class drop(histeditaction):
609 def run(self):
609 def run(self):
610 parentctx = self.repo[self.state.parentctxnode]
610 parentctx = self.repo[self.state.parentctxnode]
611 return parentctx, [(self.node, tuple())]
611 return parentctx, [(self.node, tuple())]
612
612
613 class message(histeditaction):
613 class message(histeditaction):
614 def commiteditor(self):
614 def commiteditor(self):
615 return cmdutil.getcommiteditor(edit=True, editform='histedit.mess')
615 return cmdutil.getcommiteditor(edit=True, editform='histedit.mess')
616
616
617 def findoutgoing(ui, repo, remote=None, force=False, opts={}):
617 def findoutgoing(ui, repo, remote=None, force=False, opts={}):
618 """utility function to find the first outgoing changeset
618 """utility function to find the first outgoing changeset
619
619
620 Used by initialisation code"""
620 Used by initialisation code"""
621 dest = ui.expandpath(remote or 'default-push', remote or 'default')
621 dest = ui.expandpath(remote or 'default-push', remote or 'default')
622 dest, revs = hg.parseurl(dest, None)[:2]
622 dest, revs = hg.parseurl(dest, None)[:2]
623 ui.status(_('comparing with %s\n') % util.hidepassword(dest))
623 ui.status(_('comparing with %s\n') % util.hidepassword(dest))
624
624
625 revs, checkout = hg.addbranchrevs(repo, repo, revs, None)
625 revs, checkout = hg.addbranchrevs(repo, repo, revs, None)
626 other = hg.peer(repo, opts, dest)
626 other = hg.peer(repo, opts, dest)
627
627
628 if revs:
628 if revs:
629 revs = [repo.lookup(rev) for rev in revs]
629 revs = [repo.lookup(rev) for rev in revs]
630
630
631 outgoing = discovery.findcommonoutgoing(repo, other, revs, force=force)
631 outgoing = discovery.findcommonoutgoing(repo, other, revs, force=force)
632 if not outgoing.missing:
632 if not outgoing.missing:
633 raise util.Abort(_('no outgoing ancestors'))
633 raise util.Abort(_('no outgoing ancestors'))
634 roots = list(repo.revs("roots(%ln)", outgoing.missing))
634 roots = list(repo.revs("roots(%ln)", outgoing.missing))
635 if 1 < len(roots):
635 if 1 < len(roots):
636 msg = _('there are ambiguous outgoing revisions')
636 msg = _('there are ambiguous outgoing revisions')
637 hint = _('see "hg help histedit" for more detail')
637 hint = _('see "hg help histedit" for more detail')
638 raise util.Abort(msg, hint=hint)
638 raise util.Abort(msg, hint=hint)
639 return repo.lookup(roots[0])
639 return repo.lookup(roots[0])
640
640
641 actiontable = {'p': pick,
641 actiontable = {'p': pick,
642 'pick': pick,
642 'pick': pick,
643 'e': edit,
643 'e': edit,
644 'edit': edit,
644 'edit': edit,
645 'f': fold,
645 'f': fold,
646 'fold': fold,
646 'fold': fold,
647 'r': rollup,
647 'r': rollup,
648 'roll': rollup,
648 'roll': rollup,
649 'd': drop,
649 'd': drop,
650 'drop': drop,
650 'drop': drop,
651 'm': message,
651 'm': message,
652 'mess': message,
652 'mess': message,
653 }
653 }
654
654
655 @command('histedit',
655 @command('histedit',
656 [('', 'commands', '',
656 [('', 'commands', '',
657 _('read history edits from the specified file'), _('FILE')),
657 _('read history edits from the specified file'), _('FILE')),
658 ('c', 'continue', False, _('continue an edit already in progress')),
658 ('c', 'continue', False, _('continue an edit already in progress')),
659 ('', 'edit-plan', False, _('edit remaining actions list')),
659 ('', 'edit-plan', False, _('edit remaining actions list')),
660 ('k', 'keep', False,
660 ('k', 'keep', False,
661 _("don't strip old nodes after edit is complete")),
661 _("don't strip old nodes after edit is complete")),
662 ('', 'abort', False, _('abort an edit in progress')),
662 ('', 'abort', False, _('abort an edit in progress')),
663 ('o', 'outgoing', False, _('changesets not found in destination')),
663 ('o', 'outgoing', False, _('changesets not found in destination')),
664 ('f', 'force', False,
664 ('f', 'force', False,
665 _('force outgoing even for unrelated repositories')),
665 _('force outgoing even for unrelated repositories')),
666 ('r', 'rev', [], _('first revision to be edited'), _('REV'))],
666 ('r', 'rev', [], _('first revision to be edited'), _('REV'))],
667 _("ANCESTOR | --outgoing [URL]"))
667 _("ANCESTOR | --outgoing [URL]"))
668 def histedit(ui, repo, *freeargs, **opts):
668 def histedit(ui, repo, *freeargs, **opts):
669 """interactively edit changeset history
669 """interactively edit changeset history
670
670
671 This command edits changesets between ANCESTOR and the parent of
671 This command edits changesets between ANCESTOR and the parent of
672 the working directory.
672 the working directory.
673
673
674 With --outgoing, this edits changesets not found in the
674 With --outgoing, this edits changesets not found in the
675 destination repository. If URL of the destination is omitted, the
675 destination repository. If URL of the destination is omitted, the
676 'default-push' (or 'default') path will be used.
676 'default-push' (or 'default') path will be used.
677
677
678 For safety, this command is aborted, also if there are ambiguous
678 For safety, this command is aborted, also if there are ambiguous
679 outgoing revisions which may confuse users: for example, there are
679 outgoing revisions which may confuse users: for example, there are
680 multiple branches containing outgoing revisions.
680 multiple branches containing outgoing revisions.
681
681
682 Use "min(outgoing() and ::.)" or similar revset specification
682 Use "min(outgoing() and ::.)" or similar revset specification
683 instead of --outgoing to specify edit target revision exactly in
683 instead of --outgoing to specify edit target revision exactly in
684 such ambiguous situation. See :hg:`help revsets` for detail about
684 such ambiguous situation. See :hg:`help revsets` for detail about
685 selecting revisions.
685 selecting revisions.
686
686
687 Returns 0 on success, 1 if user intervention is required (not only
687 Returns 0 on success, 1 if user intervention is required (not only
688 for intentional "edit" command, but also for resolving unexpected
688 for intentional "edit" command, but also for resolving unexpected
689 conflicts).
689 conflicts).
690 """
690 """
691 state = histeditstate(repo)
691 state = histeditstate(repo)
692 try:
692 try:
693 state.wlock = repo.wlock()
693 state.wlock = repo.wlock()
694 state.lock = repo.lock()
694 state.lock = repo.lock()
695 _histedit(ui, repo, state, *freeargs, **opts)
695 _histedit(ui, repo, state, *freeargs, **opts)
696 finally:
696 finally:
697 release(state.lock, state.wlock)
697 release(state.lock, state.wlock)
698
698
699 def _histedit(ui, repo, state, *freeargs, **opts):
699 def _histedit(ui, repo, state, *freeargs, **opts):
700 # TODO only abort if we try and histedit mq patches, not just
700 # TODO only abort if we try and histedit mq patches, not just
701 # blanket if mq patches are applied somewhere
701 # blanket if mq patches are applied somewhere
702 mq = getattr(repo, 'mq', None)
702 mq = getattr(repo, 'mq', None)
703 if mq and mq.applied:
703 if mq and mq.applied:
704 raise util.Abort(_('source has mq patches applied'))
704 raise util.Abort(_('source has mq patches applied'))
705
705
706 # basic argument incompatibility processing
706 # basic argument incompatibility processing
707 outg = opts.get('outgoing')
707 outg = opts.get('outgoing')
708 cont = opts.get('continue')
708 cont = opts.get('continue')
709 editplan = opts.get('edit_plan')
709 editplan = opts.get('edit_plan')
710 abort = opts.get('abort')
710 abort = opts.get('abort')
711 force = opts.get('force')
711 force = opts.get('force')
712 rules = opts.get('commands', '')
712 rules = opts.get('commands', '')
713 revs = opts.get('rev', [])
713 revs = opts.get('rev', [])
714 goal = 'new' # This invocation goal, in new, continue, abort
714 goal = 'new' # This invocation goal, in new, continue, abort
715 if force and not outg:
715 if force and not outg:
716 raise util.Abort(_('--force only allowed with --outgoing'))
716 raise util.Abort(_('--force only allowed with --outgoing'))
717 if cont:
717 if cont:
718 if any((outg, abort, revs, freeargs, rules, editplan)):
718 if any((outg, abort, revs, freeargs, rules, editplan)):
719 raise util.Abort(_('no arguments allowed with --continue'))
719 raise util.Abort(_('no arguments allowed with --continue'))
720 goal = 'continue'
720 goal = 'continue'
721 elif abort:
721 elif abort:
722 if any((outg, revs, freeargs, rules, editplan)):
722 if any((outg, revs, freeargs, rules, editplan)):
723 raise util.Abort(_('no arguments allowed with --abort'))
723 raise util.Abort(_('no arguments allowed with --abort'))
724 goal = 'abort'
724 goal = 'abort'
725 elif editplan:
725 elif editplan:
726 if any((outg, revs, freeargs)):
726 if any((outg, revs, freeargs)):
727 raise util.Abort(_('only --commands argument allowed with '
727 raise util.Abort(_('only --commands argument allowed with '
728 '--edit-plan'))
728 '--edit-plan'))
729 goal = 'edit-plan'
729 goal = 'edit-plan'
730 else:
730 else:
731 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
731 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
732 raise util.Abort(_('history edit already in progress, try '
732 raise util.Abort(_('history edit already in progress, try '
733 '--continue or --abort'))
733 '--continue or --abort'))
734 if outg:
734 if outg:
735 if revs:
735 if revs:
736 raise util.Abort(_('no revisions allowed with --outgoing'))
736 raise util.Abort(_('no revisions allowed with --outgoing'))
737 if len(freeargs) > 1:
737 if len(freeargs) > 1:
738 raise util.Abort(
738 raise util.Abort(
739 _('only one repo argument allowed with --outgoing'))
739 _('only one repo argument allowed with --outgoing'))
740 else:
740 else:
741 revs.extend(freeargs)
741 revs.extend(freeargs)
742 if len(revs) == 0:
742 if len(revs) == 0:
743 histeditdefault = ui.config('histedit', 'defaultrev')
743 histeditdefault = ui.config('histedit', 'defaultrev')
744 if histeditdefault:
744 if histeditdefault:
745 revs.append(histeditdefault)
745 revs.append(histeditdefault)
746 if len(revs) != 1:
746 if len(revs) != 1:
747 raise util.Abort(
747 raise util.Abort(
748 _('histedit requires exactly one ancestor revision'))
748 _('histedit requires exactly one ancestor revision'))
749
749
750
750
751 replacements = []
751 replacements = []
752 state.keep = opts.get('keep', False)
752 state.keep = opts.get('keep', False)
753 supportsmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
753
754
754 # rebuild state
755 # rebuild state
755 if goal == 'continue':
756 if goal == 'continue':
756 state.read()
757 state.read()
757 state = bootstrapcontinue(ui, state, opts)
758 state = bootstrapcontinue(ui, state, opts)
758 elif goal == 'edit-plan':
759 elif goal == 'edit-plan':
759 state.read()
760 state.read()
760 if not rules:
761 if not rules:
761 comment = editcomment % (node.short(state.parentctxnode),
762 comment = editcomment % (node.short(state.parentctxnode),
762 node.short(state.topmost))
763 node.short(state.topmost))
763 rules = ruleeditor(repo, ui, state.rules, comment)
764 rules = ruleeditor(repo, ui, state.rules, comment)
764 else:
765 else:
765 if rules == '-':
766 if rules == '-':
766 f = sys.stdin
767 f = sys.stdin
767 else:
768 else:
768 f = open(rules)
769 f = open(rules)
769 rules = f.read()
770 rules = f.read()
770 f.close()
771 f.close()
771 rules = [l for l in (r.strip() for r in rules.splitlines())
772 rules = [l for l in (r.strip() for r in rules.splitlines())
772 if l and not l.startswith('#')]
773 if l and not l.startswith('#')]
773 rules = verifyrules(rules, repo, [repo[c] for [_a, c] in state.rules])
774 rules = verifyrules(rules, repo, [repo[c] for [_a, c] in state.rules])
774 state.rules = rules
775 state.rules = rules
775 state.write()
776 state.write()
776 return
777 return
777 elif goal == 'abort':
778 elif goal == 'abort':
778 state.read()
779 state.read()
779 mapping, tmpnodes, leafs, _ntm = processreplacement(state)
780 mapping, tmpnodes, leafs, _ntm = processreplacement(state)
780 ui.debug('restore wc to old parent %s\n' % node.short(state.topmost))
781 ui.debug('restore wc to old parent %s\n' % node.short(state.topmost))
781
782
782 # Recover our old commits if necessary
783 # Recover our old commits if necessary
783 if not state.topmost in repo and state.backupfile:
784 if not state.topmost in repo and state.backupfile:
784 backupfile = repo.join(state.backupfile)
785 backupfile = repo.join(state.backupfile)
785 f = hg.openpath(ui, backupfile)
786 f = hg.openpath(ui, backupfile)
786 gen = exchange.readbundle(ui, f, backupfile)
787 gen = exchange.readbundle(ui, f, backupfile)
787 changegroup.addchangegroup(repo, gen, 'histedit',
788 changegroup.addchangegroup(repo, gen, 'histedit',
788 'bundle:' + backupfile)
789 'bundle:' + backupfile)
789 os.remove(backupfile)
790 os.remove(backupfile)
790
791
791 # check whether we should update away
792 # check whether we should update away
792 parentnodes = [c.node() for c in repo[None].parents()]
793 parentnodes = [c.node() for c in repo[None].parents()]
793 for n in leafs | set([state.parentctxnode]):
794 for n in leafs | set([state.parentctxnode]):
794 if n in parentnodes:
795 if n in parentnodes:
795 hg.clean(repo, state.topmost)
796 hg.clean(repo, state.topmost)
796 break
797 break
797 else:
798 else:
798 pass
799 pass
799 cleanupnode(ui, repo, 'created', tmpnodes)
800 cleanupnode(ui, repo, 'created', tmpnodes)
800 cleanupnode(ui, repo, 'temp', leafs)
801 cleanupnode(ui, repo, 'temp', leafs)
801 state.clear()
802 state.clear()
802 return
803 return
803 else:
804 else:
804 cmdutil.checkunfinished(repo)
805 cmdutil.checkunfinished(repo)
805 cmdutil.bailifchanged(repo)
806 cmdutil.bailifchanged(repo)
806
807
807 topmost, empty = repo.dirstate.parents()
808 topmost, empty = repo.dirstate.parents()
808 if outg:
809 if outg:
809 if freeargs:
810 if freeargs:
810 remote = freeargs[0]
811 remote = freeargs[0]
811 else:
812 else:
812 remote = None
813 remote = None
813 root = findoutgoing(ui, repo, remote, force, opts)
814 root = findoutgoing(ui, repo, remote, force, opts)
814 else:
815 else:
815 rr = list(repo.set('roots(%ld)', scmutil.revrange(repo, revs)))
816 rr = list(repo.set('roots(%ld)', scmutil.revrange(repo, revs)))
816 if len(rr) != 1:
817 if len(rr) != 1:
817 raise util.Abort(_('The specified revisions must have '
818 raise util.Abort(_('The specified revisions must have '
818 'exactly one common root'))
819 'exactly one common root'))
819 root = rr[0].node()
820 root = rr[0].node()
820
821
821 revs = between(repo, root, topmost, state.keep)
822 revs = between(repo, root, topmost, state.keep)
822 if not revs:
823 if not revs:
823 raise util.Abort(_('%s is not an ancestor of working directory') %
824 raise util.Abort(_('%s is not an ancestor of working directory') %
824 node.short(root))
825 node.short(root))
825
826
826 ctxs = [repo[r] for r in revs]
827 ctxs = [repo[r] for r in revs]
827 if not rules:
828 if not rules:
828 comment = editcomment % (node.short(root), node.short(topmost))
829 comment = editcomment % (node.short(root), node.short(topmost))
829 rules = ruleeditor(repo, ui, [['pick', c] for c in ctxs], comment)
830 rules = ruleeditor(repo, ui, [['pick', c] for c in ctxs], comment)
830 else:
831 else:
831 if rules == '-':
832 if rules == '-':
832 f = sys.stdin
833 f = sys.stdin
833 else:
834 else:
834 f = open(rules)
835 f = open(rules)
835 rules = f.read()
836 rules = f.read()
836 f.close()
837 f.close()
837 rules = [l for l in (r.strip() for r in rules.splitlines())
838 rules = [l for l in (r.strip() for r in rules.splitlines())
838 if l and not l.startswith('#')]
839 if l and not l.startswith('#')]
839 rules = verifyrules(rules, repo, ctxs)
840 rules = verifyrules(rules, repo, ctxs)
840
841
841 parentctxnode = repo[root].parents()[0].node()
842 parentctxnode = repo[root].parents()[0].node()
842
843
843 state.parentctxnode = parentctxnode
844 state.parentctxnode = parentctxnode
844 state.rules = rules
845 state.rules = rules
845 state.topmost = topmost
846 state.topmost = topmost
846 state.replacements = replacements
847 state.replacements = replacements
847
848
848 # Create a backup so we can always abort completely.
849 # Create a backup so we can always abort completely.
849 backupfile = None
850 backupfile = None
850 if not obsolete.isenabled(repo, obsolete.createmarkersopt):
851 if not obsolete.isenabled(repo, obsolete.createmarkersopt):
851 backupfile = repair._bundle(repo, [parentctxnode], [topmost], root,
852 backupfile = repair._bundle(repo, [parentctxnode], [topmost], root,
852 'histedit')
853 'histedit')
853 state.backupfile = backupfile
854 state.backupfile = backupfile
854
855
855 while state.rules:
856 while state.rules:
856 state.write()
857 state.write()
857 action, ha = state.rules.pop(0)
858 action, ha = state.rules.pop(0)
858 ui.debug('histedit: processing %s %s\n' % (action, ha[:12]))
859 ui.debug('histedit: processing %s %s\n' % (action, ha[:12]))
859 actobj = actiontable[action].fromrule(state, ha)
860 actobj = actiontable[action].fromrule(state, ha)
860 parentctx, replacement_ = actobj.run()
861 parentctx, replacement_ = actobj.run()
861 state.parentctxnode = parentctx.node()
862 state.parentctxnode = parentctx.node()
862 state.replacements.extend(replacement_)
863 state.replacements.extend(replacement_)
863 state.write()
864 state.write()
864
865
865 hg.update(repo, state.parentctxnode)
866 hg.update(repo, state.parentctxnode)
866
867
867 mapping, tmpnodes, created, ntm = processreplacement(state)
868 mapping, tmpnodes, created, ntm = processreplacement(state)
868 if mapping:
869 if mapping:
869 for prec, succs in mapping.iteritems():
870 for prec, succs in mapping.iteritems():
870 if not succs:
871 if not succs:
871 ui.debug('histedit: %s is dropped\n' % node.short(prec))
872 ui.debug('histedit: %s is dropped\n' % node.short(prec))
872 else:
873 else:
873 ui.debug('histedit: %s is replaced by %s\n' % (
874 ui.debug('histedit: %s is replaced by %s\n' % (
874 node.short(prec), node.short(succs[0])))
875 node.short(prec), node.short(succs[0])))
875 if len(succs) > 1:
876 if len(succs) > 1:
876 m = 'histedit: %s'
877 m = 'histedit: %s'
877 for n in succs[1:]:
878 for n in succs[1:]:
878 ui.debug(m % node.short(n))
879 ui.debug(m % node.short(n))
879
880
880 if not state.keep:
881 if not state.keep:
881 if mapping:
882 if mapping:
882 movebookmarks(ui, repo, mapping, state.topmost, ntm)
883 movebookmarks(ui, repo, mapping, state.topmost, ntm)
883 # TODO update mq state
884 # TODO update mq state
884 if obsolete.isenabled(repo, obsolete.createmarkersopt):
885 if supportsmarkers:
885 markers = []
886 markers = []
886 # sort by revision number because it sound "right"
887 # sort by revision number because it sound "right"
887 for prec in sorted(mapping, key=repo.changelog.rev):
888 for prec in sorted(mapping, key=repo.changelog.rev):
888 succs = mapping[prec]
889 succs = mapping[prec]
889 markers.append((repo[prec],
890 markers.append((repo[prec],
890 tuple(repo[s] for s in succs)))
891 tuple(repo[s] for s in succs)))
891 if markers:
892 if markers:
892 obsolete.createmarkers(repo, markers)
893 obsolete.createmarkers(repo, markers)
893 else:
894 else:
894 cleanupnode(ui, repo, 'replaced', mapping)
895 cleanupnode(ui, repo, 'replaced', mapping)
895
896
896 cleanupnode(ui, repo, 'temp', tmpnodes)
897 cleanupnode(ui, repo, 'temp', tmpnodes)
897 state.clear()
898 state.clear()
898 if os.path.exists(repo.sjoin('undo')):
899 if os.path.exists(repo.sjoin('undo')):
899 os.unlink(repo.sjoin('undo'))
900 os.unlink(repo.sjoin('undo'))
900
901
901 def bootstrapcontinue(ui, state, opts):
902 def bootstrapcontinue(ui, state, opts):
902 repo = state.repo
903 repo = state.repo
903 if state.rules:
904 if state.rules:
904 action, currentnode = state.rules.pop(0)
905 action, currentnode = state.rules.pop(0)
905
906
906 actobj = actiontable[action].fromrule(state, currentnode)
907 actobj = actiontable[action].fromrule(state, currentnode)
907
908
908 s = repo.status()
909 s = repo.status()
909 if s.modified or s.added or s.removed or s.deleted:
910 if s.modified or s.added or s.removed or s.deleted:
910 actobj.continuedirty()
911 actobj.continuedirty()
911 s = repo.status()
912 s = repo.status()
912 if s.modified or s.added or s.removed or s.deleted:
913 if s.modified or s.added or s.removed or s.deleted:
913 raise util.Abort(_("working copy still dirty"))
914 raise util.Abort(_("working copy still dirty"))
914
915
915 parentctx, replacements = actobj.continueclean()
916 parentctx, replacements = actobj.continueclean()
916
917
917 state.parentctxnode = parentctx.node()
918 state.parentctxnode = parentctx.node()
918 state.replacements.extend(replacements)
919 state.replacements.extend(replacements)
919
920
920 return state
921 return state
921
922
922 def between(repo, old, new, keep):
923 def between(repo, old, new, keep):
923 """select and validate the set of revision to edit
924 """select and validate the set of revision to edit
924
925
925 When keep is false, the specified set can't have children."""
926 When keep is false, the specified set can't have children."""
926 ctxs = list(repo.set('%n::%n', old, new))
927 ctxs = list(repo.set('%n::%n', old, new))
927 if ctxs and not keep:
928 if ctxs and not keep:
928 if (not obsolete.isenabled(repo, obsolete.allowunstableopt) and
929 if (not obsolete.isenabled(repo, obsolete.allowunstableopt) and
929 repo.revs('(%ld::) - (%ld)', ctxs, ctxs)):
930 repo.revs('(%ld::) - (%ld)', ctxs, ctxs)):
930 raise util.Abort(_('cannot edit history that would orphan nodes'))
931 raise util.Abort(_('cannot edit history that would orphan nodes'))
931 if repo.revs('(%ld) and merge()', ctxs):
932 if repo.revs('(%ld) and merge()', ctxs):
932 raise util.Abort(_('cannot edit history that contains merges'))
933 raise util.Abort(_('cannot edit history that contains merges'))
933 root = ctxs[0] # list is already sorted by repo.set
934 root = ctxs[0] # list is already sorted by repo.set
934 if not root.mutable():
935 if not root.mutable():
935 raise util.Abort(_('cannot edit public changeset: %s') % root,
936 raise util.Abort(_('cannot edit public changeset: %s') % root,
936 hint=_('see "hg help phases" for details'))
937 hint=_('see "hg help phases" for details'))
937 return [c.node() for c in ctxs]
938 return [c.node() for c in ctxs]
938
939
939 def makedesc(repo, action, rev):
940 def makedesc(repo, action, rev):
940 """build a initial action line for a ctx
941 """build a initial action line for a ctx
941
942
942 line are in the form:
943 line are in the form:
943
944
944 <action> <hash> <rev> <summary>
945 <action> <hash> <rev> <summary>
945 """
946 """
946 ctx = repo[rev]
947 ctx = repo[rev]
947 summary = ''
948 summary = ''
948 if ctx.description():
949 if ctx.description():
949 summary = ctx.description().splitlines()[0]
950 summary = ctx.description().splitlines()[0]
950 line = '%s %s %d %s' % (action, ctx, ctx.rev(), summary)
951 line = '%s %s %d %s' % (action, ctx, ctx.rev(), summary)
951 # trim to 80 columns so it's not stupidly wide in my editor
952 # trim to 80 columns so it's not stupidly wide in my editor
952 maxlen = repo.ui.configint('histedit', 'linelen', default=80)
953 maxlen = repo.ui.configint('histedit', 'linelen', default=80)
953 maxlen = max(maxlen, 22) # avoid truncating hash
954 maxlen = max(maxlen, 22) # avoid truncating hash
954 return util.ellipsis(line, maxlen)
955 return util.ellipsis(line, maxlen)
955
956
956 def ruleeditor(repo, ui, rules, editcomment=""):
957 def ruleeditor(repo, ui, rules, editcomment=""):
957 """open an editor to edit rules
958 """open an editor to edit rules
958
959
959 rules are in the format [ [act, ctx], ...] like in state.rules
960 rules are in the format [ [act, ctx], ...] like in state.rules
960 """
961 """
961 rules = '\n'.join([makedesc(repo, act, rev) for [act, rev] in rules])
962 rules = '\n'.join([makedesc(repo, act, rev) for [act, rev] in rules])
962 rules += '\n\n'
963 rules += '\n\n'
963 rules += editcomment
964 rules += editcomment
964 rules = ui.edit(rules, ui.username())
965 rules = ui.edit(rules, ui.username())
965
966
966 # Save edit rules in .hg/histedit-last-edit.txt in case
967 # Save edit rules in .hg/histedit-last-edit.txt in case
967 # the user needs to ask for help after something
968 # the user needs to ask for help after something
968 # surprising happens.
969 # surprising happens.
969 f = open(repo.join('histedit-last-edit.txt'), 'w')
970 f = open(repo.join('histedit-last-edit.txt'), 'w')
970 f.write(rules)
971 f.write(rules)
971 f.close()
972 f.close()
972
973
973 return rules
974 return rules
974
975
975 def verifyrules(rules, repo, ctxs):
976 def verifyrules(rules, repo, ctxs):
976 """Verify that there exists exactly one edit rule per given changeset.
977 """Verify that there exists exactly one edit rule per given changeset.
977
978
978 Will abort if there are to many or too few rules, a malformed rule,
979 Will abort if there are to many or too few rules, a malformed rule,
979 or a rule on a changeset outside of the user-given range.
980 or a rule on a changeset outside of the user-given range.
980 """
981 """
981 parsed = []
982 parsed = []
982 expected = set(c.hex() for c in ctxs)
983 expected = set(c.hex() for c in ctxs)
983 seen = set()
984 seen = set()
984 for r in rules:
985 for r in rules:
985 if ' ' not in r:
986 if ' ' not in r:
986 raise util.Abort(_('malformed line "%s"') % r)
987 raise util.Abort(_('malformed line "%s"') % r)
987 action, rest = r.split(' ', 1)
988 action, rest = r.split(' ', 1)
988 ha = rest.strip().split(' ', 1)[0]
989 ha = rest.strip().split(' ', 1)[0]
989 try:
990 try:
990 ha = repo[ha].hex()
991 ha = repo[ha].hex()
991 except error.RepoError:
992 except error.RepoError:
992 raise util.Abort(_('unknown changeset %s listed') % ha[:12])
993 raise util.Abort(_('unknown changeset %s listed') % ha[:12])
993 if ha not in expected:
994 if ha not in expected:
994 raise util.Abort(
995 raise util.Abort(
995 _('may not use changesets other than the ones listed'))
996 _('may not use changesets other than the ones listed'))
996 if ha in seen:
997 if ha in seen:
997 raise util.Abort(_('duplicated command for changeset %s') %
998 raise util.Abort(_('duplicated command for changeset %s') %
998 ha[:12])
999 ha[:12])
999 seen.add(ha)
1000 seen.add(ha)
1000 if action not in actiontable:
1001 if action not in actiontable:
1001 raise util.Abort(_('unknown action "%s"') % action)
1002 raise util.Abort(_('unknown action "%s"') % action)
1002 parsed.append([action, ha])
1003 parsed.append([action, ha])
1003 missing = sorted(expected - seen) # sort to stabilize output
1004 missing = sorted(expected - seen) # sort to stabilize output
1004 if missing:
1005 if missing:
1005 raise util.Abort(_('missing rules for changeset %s') %
1006 raise util.Abort(_('missing rules for changeset %s') %
1006 missing[0][:12],
1007 missing[0][:12],
1007 hint=_('do you want to use the drop action?'))
1008 hint=_('do you want to use the drop action?'))
1008 return parsed
1009 return parsed
1009
1010
1010 def processreplacement(state):
1011 def processreplacement(state):
1011 """process the list of replacements to return
1012 """process the list of replacements to return
1012
1013
1013 1) the final mapping between original and created nodes
1014 1) the final mapping between original and created nodes
1014 2) the list of temporary node created by histedit
1015 2) the list of temporary node created by histedit
1015 3) the list of new commit created by histedit"""
1016 3) the list of new commit created by histedit"""
1016 replacements = state.replacements
1017 replacements = state.replacements
1017 allsuccs = set()
1018 allsuccs = set()
1018 replaced = set()
1019 replaced = set()
1019 fullmapping = {}
1020 fullmapping = {}
1020 # initialise basic set
1021 # initialise basic set
1021 # fullmapping record all operation recorded in replacement
1022 # fullmapping record all operation recorded in replacement
1022 for rep in replacements:
1023 for rep in replacements:
1023 allsuccs.update(rep[1])
1024 allsuccs.update(rep[1])
1024 replaced.add(rep[0])
1025 replaced.add(rep[0])
1025 fullmapping.setdefault(rep[0], set()).update(rep[1])
1026 fullmapping.setdefault(rep[0], set()).update(rep[1])
1026 new = allsuccs - replaced
1027 new = allsuccs - replaced
1027 tmpnodes = allsuccs & replaced
1028 tmpnodes = allsuccs & replaced
1028 # Reduce content fullmapping into direct relation between original nodes
1029 # Reduce content fullmapping into direct relation between original nodes
1029 # and final node created during history edition
1030 # and final node created during history edition
1030 # Dropped changeset are replaced by an empty list
1031 # Dropped changeset are replaced by an empty list
1031 toproceed = set(fullmapping)
1032 toproceed = set(fullmapping)
1032 final = {}
1033 final = {}
1033 while toproceed:
1034 while toproceed:
1034 for x in list(toproceed):
1035 for x in list(toproceed):
1035 succs = fullmapping[x]
1036 succs = fullmapping[x]
1036 for s in list(succs):
1037 for s in list(succs):
1037 if s in toproceed:
1038 if s in toproceed:
1038 # non final node with unknown closure
1039 # non final node with unknown closure
1039 # We can't process this now
1040 # We can't process this now
1040 break
1041 break
1041 elif s in final:
1042 elif s in final:
1042 # non final node, replace with closure
1043 # non final node, replace with closure
1043 succs.remove(s)
1044 succs.remove(s)
1044 succs.update(final[s])
1045 succs.update(final[s])
1045 else:
1046 else:
1046 final[x] = succs
1047 final[x] = succs
1047 toproceed.remove(x)
1048 toproceed.remove(x)
1048 # remove tmpnodes from final mapping
1049 # remove tmpnodes from final mapping
1049 for n in tmpnodes:
1050 for n in tmpnodes:
1050 del final[n]
1051 del final[n]
1051 # we expect all changes involved in final to exist in the repo
1052 # we expect all changes involved in final to exist in the repo
1052 # turn `final` into list (topologically sorted)
1053 # turn `final` into list (topologically sorted)
1053 nm = state.repo.changelog.nodemap
1054 nm = state.repo.changelog.nodemap
1054 for prec, succs in final.items():
1055 for prec, succs in final.items():
1055 final[prec] = sorted(succs, key=nm.get)
1056 final[prec] = sorted(succs, key=nm.get)
1056
1057
1057 # computed topmost element (necessary for bookmark)
1058 # computed topmost element (necessary for bookmark)
1058 if new:
1059 if new:
1059 newtopmost = sorted(new, key=state.repo.changelog.rev)[-1]
1060 newtopmost = sorted(new, key=state.repo.changelog.rev)[-1]
1060 elif not final:
1061 elif not final:
1061 # Nothing rewritten at all. we won't need `newtopmost`
1062 # Nothing rewritten at all. we won't need `newtopmost`
1062 # It is the same as `oldtopmost` and `processreplacement` know it
1063 # It is the same as `oldtopmost` and `processreplacement` know it
1063 newtopmost = None
1064 newtopmost = None
1064 else:
1065 else:
1065 # every body died. The newtopmost is the parent of the root.
1066 # every body died. The newtopmost is the parent of the root.
1066 r = state.repo.changelog.rev
1067 r = state.repo.changelog.rev
1067 newtopmost = state.repo[sorted(final, key=r)[0]].p1().node()
1068 newtopmost = state.repo[sorted(final, key=r)[0]].p1().node()
1068
1069
1069 return final, tmpnodes, new, newtopmost
1070 return final, tmpnodes, new, newtopmost
1070
1071
1071 def movebookmarks(ui, repo, mapping, oldtopmost, newtopmost):
1072 def movebookmarks(ui, repo, mapping, oldtopmost, newtopmost):
1072 """Move bookmark from old to newly created node"""
1073 """Move bookmark from old to newly created node"""
1073 if not mapping:
1074 if not mapping:
1074 # if nothing got rewritten there is not purpose for this function
1075 # if nothing got rewritten there is not purpose for this function
1075 return
1076 return
1076 moves = []
1077 moves = []
1077 for bk, old in sorted(repo._bookmarks.iteritems()):
1078 for bk, old in sorted(repo._bookmarks.iteritems()):
1078 if old == oldtopmost:
1079 if old == oldtopmost:
1079 # special case ensure bookmark stay on tip.
1080 # special case ensure bookmark stay on tip.
1080 #
1081 #
1081 # This is arguably a feature and we may only want that for the
1082 # This is arguably a feature and we may only want that for the
1082 # active bookmark. But the behavior is kept compatible with the old
1083 # active bookmark. But the behavior is kept compatible with the old
1083 # version for now.
1084 # version for now.
1084 moves.append((bk, newtopmost))
1085 moves.append((bk, newtopmost))
1085 continue
1086 continue
1086 base = old
1087 base = old
1087 new = mapping.get(base, None)
1088 new = mapping.get(base, None)
1088 if new is None:
1089 if new is None:
1089 continue
1090 continue
1090 while not new:
1091 while not new:
1091 # base is killed, trying with parent
1092 # base is killed, trying with parent
1092 base = repo[base].p1().node()
1093 base = repo[base].p1().node()
1093 new = mapping.get(base, (base,))
1094 new = mapping.get(base, (base,))
1094 # nothing to move
1095 # nothing to move
1095 moves.append((bk, new[-1]))
1096 moves.append((bk, new[-1]))
1096 if moves:
1097 if moves:
1097 marks = repo._bookmarks
1098 marks = repo._bookmarks
1098 for mark, new in moves:
1099 for mark, new in moves:
1099 old = marks[mark]
1100 old = marks[mark]
1100 ui.note(_('histedit: moving bookmarks %s from %s to %s\n')
1101 ui.note(_('histedit: moving bookmarks %s from %s to %s\n')
1101 % (mark, node.short(old), node.short(new)))
1102 % (mark, node.short(old), node.short(new)))
1102 marks[mark] = new
1103 marks[mark] = new
1103 marks.write()
1104 marks.write()
1104
1105
1105 def cleanupnode(ui, repo, name, nodes):
1106 def cleanupnode(ui, repo, name, nodes):
1106 """strip a group of nodes from the repository
1107 """strip a group of nodes from the repository
1107
1108
1108 The set of node to strip may contains unknown nodes."""
1109 The set of node to strip may contains unknown nodes."""
1109 ui.debug('should strip %s nodes %s\n' %
1110 ui.debug('should strip %s nodes %s\n' %
1110 (name, ', '.join([node.short(n) for n in nodes])))
1111 (name, ', '.join([node.short(n) for n in nodes])))
1111 lock = None
1112 lock = None
1112 try:
1113 try:
1113 lock = repo.lock()
1114 lock = repo.lock()
1114 # Find all node that need to be stripped
1115 # Find all node that need to be stripped
1115 # (we hg %lr instead of %ln to silently ignore unknown item
1116 # (we hg %lr instead of %ln to silently ignore unknown item
1116 nm = repo.changelog.nodemap
1117 nm = repo.changelog.nodemap
1117 nodes = sorted(n for n in nodes if n in nm)
1118 nodes = sorted(n for n in nodes if n in nm)
1118 roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
1119 roots = [c.node() for c in repo.set("roots(%ln)", nodes)]
1119 for c in roots:
1120 for c in roots:
1120 # We should process node in reverse order to strip tip most first.
1121 # We should process node in reverse order to strip tip most first.
1121 # but this trigger a bug in changegroup hook.
1122 # but this trigger a bug in changegroup hook.
1122 # This would reduce bundle overhead
1123 # This would reduce bundle overhead
1123 repair.strip(ui, repo, c)
1124 repair.strip(ui, repo, c)
1124 finally:
1125 finally:
1125 release(lock)
1126 release(lock)
1126
1127
1127 def stripwrapper(orig, ui, repo, nodelist, *args, **kwargs):
1128 def stripwrapper(orig, ui, repo, nodelist, *args, **kwargs):
1128 if isinstance(nodelist, str):
1129 if isinstance(nodelist, str):
1129 nodelist = [nodelist]
1130 nodelist = [nodelist]
1130 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1131 if os.path.exists(os.path.join(repo.path, 'histedit-state')):
1131 state = histeditstate(repo)
1132 state = histeditstate(repo)
1132 state.read()
1133 state.read()
1133 histedit_nodes = set([repo[rulehash].node() for (action, rulehash)
1134 histedit_nodes = set([repo[rulehash].node() for (action, rulehash)
1134 in state.rules if rulehash in repo])
1135 in state.rules if rulehash in repo])
1135 strip_nodes = set([repo[n].node() for n in nodelist])
1136 strip_nodes = set([repo[n].node() for n in nodelist])
1136 common_nodes = histedit_nodes & strip_nodes
1137 common_nodes = histedit_nodes & strip_nodes
1137 if common_nodes:
1138 if common_nodes:
1138 raise util.Abort(_("histedit in progress, can't strip %s")
1139 raise util.Abort(_("histedit in progress, can't strip %s")
1139 % ', '.join(node.short(x) for x in common_nodes))
1140 % ', '.join(node.short(x) for x in common_nodes))
1140 return orig(ui, repo, nodelist, *args, **kwargs)
1141 return orig(ui, repo, nodelist, *args, **kwargs)
1141
1142
1142 extensions.wrapfunction(repair, 'strip', stripwrapper)
1143 extensions.wrapfunction(repair, 'strip', stripwrapper)
1143
1144
1144 def summaryhook(ui, repo):
1145 def summaryhook(ui, repo):
1145 if not os.path.exists(repo.join('histedit-state')):
1146 if not os.path.exists(repo.join('histedit-state')):
1146 return
1147 return
1147 state = histeditstate(repo)
1148 state = histeditstate(repo)
1148 state.read()
1149 state.read()
1149 if state.rules:
1150 if state.rules:
1150 # i18n: column positioning for "hg summary"
1151 # i18n: column positioning for "hg summary"
1151 ui.write(_('hist: %s (histedit --continue)\n') %
1152 ui.write(_('hist: %s (histedit --continue)\n') %
1152 (ui.label(_('%d remaining'), 'histedit.remaining') %
1153 (ui.label(_('%d remaining'), 'histedit.remaining') %
1153 len(state.rules)))
1154 len(state.rules)))
1154
1155
1155 def extsetup(ui):
1156 def extsetup(ui):
1156 cmdutil.summaryhooks.add('histedit', summaryhook)
1157 cmdutil.summaryhooks.add('histedit', summaryhook)
1157 cmdutil.unfinishedstates.append(
1158 cmdutil.unfinishedstates.append(
1158 ['histedit-state', False, True, _('histedit in progress'),
1159 ['histedit-state', False, True, _('histedit in progress'),
1159 _("use 'hg histedit --continue' or 'hg histedit --abort'")])
1160 _("use 'hg histedit --continue' or 'hg histedit --abort'")])
General Comments 0
You need to be logged in to leave comments. Login now