##// END OF EJS Templates
crecord: drop unused "operation" parameter from filterpatch function
Denis Laxalde -
r29327:baab9ea4 default
parent child Browse files
Show More
@@ -1,3558 +1,3558 b''
1 # cmdutil.py - help for command processing in mercurial
1 # cmdutil.py - help for command processing in mercurial
2 #
2 #
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
3 # Copyright 2005-2007 Matt Mackall <mpm@selenic.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
7
8 from __future__ import absolute_import
8 from __future__ import absolute_import
9
9
10 import errno
10 import errno
11 import os
11 import os
12 import re
12 import re
13 import sys
13 import sys
14 import tempfile
14 import tempfile
15
15
16 from .i18n import _
16 from .i18n import _
17 from .node import (
17 from .node import (
18 bin,
18 bin,
19 hex,
19 hex,
20 nullid,
20 nullid,
21 nullrev,
21 nullrev,
22 short,
22 short,
23 )
23 )
24
24
25 from . import (
25 from . import (
26 bookmarks,
26 bookmarks,
27 changelog,
27 changelog,
28 copies,
28 copies,
29 crecord as crecordmod,
29 crecord as crecordmod,
30 encoding,
30 encoding,
31 error,
31 error,
32 formatter,
32 formatter,
33 graphmod,
33 graphmod,
34 lock as lockmod,
34 lock as lockmod,
35 match as matchmod,
35 match as matchmod,
36 obsolete,
36 obsolete,
37 patch,
37 patch,
38 pathutil,
38 pathutil,
39 phases,
39 phases,
40 repair,
40 repair,
41 revlog,
41 revlog,
42 revset,
42 revset,
43 scmutil,
43 scmutil,
44 templatekw,
44 templatekw,
45 templater,
45 templater,
46 util,
46 util,
47 )
47 )
48 stringio = util.stringio
48 stringio = util.stringio
49
49
50 def ishunk(x):
50 def ishunk(x):
51 hunkclasses = (crecordmod.uihunk, patch.recordhunk)
51 hunkclasses = (crecordmod.uihunk, patch.recordhunk)
52 return isinstance(x, hunkclasses)
52 return isinstance(x, hunkclasses)
53
53
54 def newandmodified(chunks, originalchunks):
54 def newandmodified(chunks, originalchunks):
55 newlyaddedandmodifiedfiles = set()
55 newlyaddedandmodifiedfiles = set()
56 for chunk in chunks:
56 for chunk in chunks:
57 if ishunk(chunk) and chunk.header.isnewfile() and chunk not in \
57 if ishunk(chunk) and chunk.header.isnewfile() and chunk not in \
58 originalchunks:
58 originalchunks:
59 newlyaddedandmodifiedfiles.add(chunk.header.filename())
59 newlyaddedandmodifiedfiles.add(chunk.header.filename())
60 return newlyaddedandmodifiedfiles
60 return newlyaddedandmodifiedfiles
61
61
62 def parsealiases(cmd):
62 def parsealiases(cmd):
63 return cmd.lstrip("^").split("|")
63 return cmd.lstrip("^").split("|")
64
64
65 def setupwrapcolorwrite(ui):
65 def setupwrapcolorwrite(ui):
66 # wrap ui.write so diff output can be labeled/colorized
66 # wrap ui.write so diff output can be labeled/colorized
67 def wrapwrite(orig, *args, **kw):
67 def wrapwrite(orig, *args, **kw):
68 label = kw.pop('label', '')
68 label = kw.pop('label', '')
69 for chunk, l in patch.difflabel(lambda: args):
69 for chunk, l in patch.difflabel(lambda: args):
70 orig(chunk, label=label + l)
70 orig(chunk, label=label + l)
71
71
72 oldwrite = ui.write
72 oldwrite = ui.write
73 def wrap(*args, **kwargs):
73 def wrap(*args, **kwargs):
74 return wrapwrite(oldwrite, *args, **kwargs)
74 return wrapwrite(oldwrite, *args, **kwargs)
75 setattr(ui, 'write', wrap)
75 setattr(ui, 'write', wrap)
76 return oldwrite
76 return oldwrite
77
77
78 def filterchunks(ui, originalhunks, usecurses, testfile, operation=None):
78 def filterchunks(ui, originalhunks, usecurses, testfile, operation=None):
79 if usecurses:
79 if usecurses:
80 if testfile:
80 if testfile:
81 recordfn = crecordmod.testdecorator(testfile,
81 recordfn = crecordmod.testdecorator(testfile,
82 crecordmod.testchunkselector)
82 crecordmod.testchunkselector)
83 else:
83 else:
84 recordfn = crecordmod.chunkselector
84 recordfn = crecordmod.chunkselector
85
85
86 return crecordmod.filterpatch(ui, originalhunks, recordfn, operation)
86 return crecordmod.filterpatch(ui, originalhunks, recordfn)
87
87
88 else:
88 else:
89 return patch.filterpatch(ui, originalhunks, operation)
89 return patch.filterpatch(ui, originalhunks, operation)
90
90
91 def recordfilter(ui, originalhunks, operation=None):
91 def recordfilter(ui, originalhunks, operation=None):
92 """ Prompts the user to filter the originalhunks and return a list of
92 """ Prompts the user to filter the originalhunks and return a list of
93 selected hunks.
93 selected hunks.
94 *operation* is used for to build ui messages to indicate the user what
94 *operation* is used for to build ui messages to indicate the user what
95 kind of filtering they are doing: reverting, committing, shelving, etc.
95 kind of filtering they are doing: reverting, committing, shelving, etc.
96 (see patch.filterpatch).
96 (see patch.filterpatch).
97 """
97 """
98 usecurses = crecordmod.checkcurses(ui)
98 usecurses = crecordmod.checkcurses(ui)
99 testfile = ui.config('experimental', 'crecordtest', None)
99 testfile = ui.config('experimental', 'crecordtest', None)
100 oldwrite = setupwrapcolorwrite(ui)
100 oldwrite = setupwrapcolorwrite(ui)
101 try:
101 try:
102 newchunks, newopts = filterchunks(ui, originalhunks, usecurses,
102 newchunks, newopts = filterchunks(ui, originalhunks, usecurses,
103 testfile, operation)
103 testfile, operation)
104 finally:
104 finally:
105 ui.write = oldwrite
105 ui.write = oldwrite
106 return newchunks, newopts
106 return newchunks, newopts
107
107
108 def dorecord(ui, repo, commitfunc, cmdsuggest, backupall,
108 def dorecord(ui, repo, commitfunc, cmdsuggest, backupall,
109 filterfn, *pats, **opts):
109 filterfn, *pats, **opts):
110 from . import merge as mergemod
110 from . import merge as mergemod
111 if not ui.interactive():
111 if not ui.interactive():
112 if cmdsuggest:
112 if cmdsuggest:
113 msg = _('running non-interactively, use %s instead') % cmdsuggest
113 msg = _('running non-interactively, use %s instead') % cmdsuggest
114 else:
114 else:
115 msg = _('running non-interactively')
115 msg = _('running non-interactively')
116 raise error.Abort(msg)
116 raise error.Abort(msg)
117
117
118 # make sure username is set before going interactive
118 # make sure username is set before going interactive
119 if not opts.get('user'):
119 if not opts.get('user'):
120 ui.username() # raise exception, username not provided
120 ui.username() # raise exception, username not provided
121
121
122 def recordfunc(ui, repo, message, match, opts):
122 def recordfunc(ui, repo, message, match, opts):
123 """This is generic record driver.
123 """This is generic record driver.
124
124
125 Its job is to interactively filter local changes, and
125 Its job is to interactively filter local changes, and
126 accordingly prepare working directory into a state in which the
126 accordingly prepare working directory into a state in which the
127 job can be delegated to a non-interactive commit command such as
127 job can be delegated to a non-interactive commit command such as
128 'commit' or 'qrefresh'.
128 'commit' or 'qrefresh'.
129
129
130 After the actual job is done by non-interactive command, the
130 After the actual job is done by non-interactive command, the
131 working directory is restored to its original state.
131 working directory is restored to its original state.
132
132
133 In the end we'll record interesting changes, and everything else
133 In the end we'll record interesting changes, and everything else
134 will be left in place, so the user can continue working.
134 will be left in place, so the user can continue working.
135 """
135 """
136
136
137 checkunfinished(repo, commit=True)
137 checkunfinished(repo, commit=True)
138 wctx = repo[None]
138 wctx = repo[None]
139 merge = len(wctx.parents()) > 1
139 merge = len(wctx.parents()) > 1
140 if merge:
140 if merge:
141 raise error.Abort(_('cannot partially commit a merge '
141 raise error.Abort(_('cannot partially commit a merge '
142 '(use "hg commit" instead)'))
142 '(use "hg commit" instead)'))
143
143
144 def fail(f, msg):
144 def fail(f, msg):
145 raise error.Abort('%s: %s' % (f, msg))
145 raise error.Abort('%s: %s' % (f, msg))
146
146
147 force = opts.get('force')
147 force = opts.get('force')
148 if not force:
148 if not force:
149 vdirs = []
149 vdirs = []
150 match.explicitdir = vdirs.append
150 match.explicitdir = vdirs.append
151 match.bad = fail
151 match.bad = fail
152
152
153 status = repo.status(match=match)
153 status = repo.status(match=match)
154 if not force:
154 if not force:
155 repo.checkcommitpatterns(wctx, vdirs, match, status, fail)
155 repo.checkcommitpatterns(wctx, vdirs, match, status, fail)
156 diffopts = patch.difffeatureopts(ui, opts=opts, whitespace=True)
156 diffopts = patch.difffeatureopts(ui, opts=opts, whitespace=True)
157 diffopts.nodates = True
157 diffopts.nodates = True
158 diffopts.git = True
158 diffopts.git = True
159 diffopts.showfunc = True
159 diffopts.showfunc = True
160 originaldiff = patch.diff(repo, changes=status, opts=diffopts)
160 originaldiff = patch.diff(repo, changes=status, opts=diffopts)
161 originalchunks = patch.parsepatch(originaldiff)
161 originalchunks = patch.parsepatch(originaldiff)
162
162
163 # 1. filter patch, since we are intending to apply subset of it
163 # 1. filter patch, since we are intending to apply subset of it
164 try:
164 try:
165 chunks, newopts = filterfn(ui, originalchunks)
165 chunks, newopts = filterfn(ui, originalchunks)
166 except patch.PatchError as err:
166 except patch.PatchError as err:
167 raise error.Abort(_('error parsing patch: %s') % err)
167 raise error.Abort(_('error parsing patch: %s') % err)
168 opts.update(newopts)
168 opts.update(newopts)
169
169
170 # We need to keep a backup of files that have been newly added and
170 # We need to keep a backup of files that have been newly added and
171 # modified during the recording process because there is a previous
171 # modified during the recording process because there is a previous
172 # version without the edit in the workdir
172 # version without the edit in the workdir
173 newlyaddedandmodifiedfiles = newandmodified(chunks, originalchunks)
173 newlyaddedandmodifiedfiles = newandmodified(chunks, originalchunks)
174 contenders = set()
174 contenders = set()
175 for h in chunks:
175 for h in chunks:
176 try:
176 try:
177 contenders.update(set(h.files()))
177 contenders.update(set(h.files()))
178 except AttributeError:
178 except AttributeError:
179 pass
179 pass
180
180
181 changed = status.modified + status.added + status.removed
181 changed = status.modified + status.added + status.removed
182 newfiles = [f for f in changed if f in contenders]
182 newfiles = [f for f in changed if f in contenders]
183 if not newfiles:
183 if not newfiles:
184 ui.status(_('no changes to record\n'))
184 ui.status(_('no changes to record\n'))
185 return 0
185 return 0
186
186
187 modified = set(status.modified)
187 modified = set(status.modified)
188
188
189 # 2. backup changed files, so we can restore them in the end
189 # 2. backup changed files, so we can restore them in the end
190
190
191 if backupall:
191 if backupall:
192 tobackup = changed
192 tobackup = changed
193 else:
193 else:
194 tobackup = [f for f in newfiles if f in modified or f in \
194 tobackup = [f for f in newfiles if f in modified or f in \
195 newlyaddedandmodifiedfiles]
195 newlyaddedandmodifiedfiles]
196 backups = {}
196 backups = {}
197 if tobackup:
197 if tobackup:
198 backupdir = repo.join('record-backups')
198 backupdir = repo.join('record-backups')
199 try:
199 try:
200 os.mkdir(backupdir)
200 os.mkdir(backupdir)
201 except OSError as err:
201 except OSError as err:
202 if err.errno != errno.EEXIST:
202 if err.errno != errno.EEXIST:
203 raise
203 raise
204 try:
204 try:
205 # backup continues
205 # backup continues
206 for f in tobackup:
206 for f in tobackup:
207 fd, tmpname = tempfile.mkstemp(prefix=f.replace('/', '_')+'.',
207 fd, tmpname = tempfile.mkstemp(prefix=f.replace('/', '_')+'.',
208 dir=backupdir)
208 dir=backupdir)
209 os.close(fd)
209 os.close(fd)
210 ui.debug('backup %r as %r\n' % (f, tmpname))
210 ui.debug('backup %r as %r\n' % (f, tmpname))
211 util.copyfile(repo.wjoin(f), tmpname, copystat=True)
211 util.copyfile(repo.wjoin(f), tmpname, copystat=True)
212 backups[f] = tmpname
212 backups[f] = tmpname
213
213
214 fp = stringio()
214 fp = stringio()
215 for c in chunks:
215 for c in chunks:
216 fname = c.filename()
216 fname = c.filename()
217 if fname in backups:
217 if fname in backups:
218 c.write(fp)
218 c.write(fp)
219 dopatch = fp.tell()
219 dopatch = fp.tell()
220 fp.seek(0)
220 fp.seek(0)
221
221
222 # 2.5 optionally review / modify patch in text editor
222 # 2.5 optionally review / modify patch in text editor
223 if opts.get('review', False):
223 if opts.get('review', False):
224 patchtext = (crecordmod.diffhelptext
224 patchtext = (crecordmod.diffhelptext
225 + crecordmod.patchhelptext
225 + crecordmod.patchhelptext
226 + fp.read())
226 + fp.read())
227 reviewedpatch = ui.edit(patchtext, "",
227 reviewedpatch = ui.edit(patchtext, "",
228 extra={"suffix": ".diff"})
228 extra={"suffix": ".diff"})
229 fp.truncate(0)
229 fp.truncate(0)
230 fp.write(reviewedpatch)
230 fp.write(reviewedpatch)
231 fp.seek(0)
231 fp.seek(0)
232
232
233 [os.unlink(repo.wjoin(c)) for c in newlyaddedandmodifiedfiles]
233 [os.unlink(repo.wjoin(c)) for c in newlyaddedandmodifiedfiles]
234 # 3a. apply filtered patch to clean repo (clean)
234 # 3a. apply filtered patch to clean repo (clean)
235 if backups:
235 if backups:
236 # Equivalent to hg.revert
236 # Equivalent to hg.revert
237 m = scmutil.matchfiles(repo, backups.keys())
237 m = scmutil.matchfiles(repo, backups.keys())
238 mergemod.update(repo, repo.dirstate.p1(),
238 mergemod.update(repo, repo.dirstate.p1(),
239 False, True, matcher=m)
239 False, True, matcher=m)
240
240
241 # 3b. (apply)
241 # 3b. (apply)
242 if dopatch:
242 if dopatch:
243 try:
243 try:
244 ui.debug('applying patch\n')
244 ui.debug('applying patch\n')
245 ui.debug(fp.getvalue())
245 ui.debug(fp.getvalue())
246 patch.internalpatch(ui, repo, fp, 1, eolmode=None)
246 patch.internalpatch(ui, repo, fp, 1, eolmode=None)
247 except patch.PatchError as err:
247 except patch.PatchError as err:
248 raise error.Abort(str(err))
248 raise error.Abort(str(err))
249 del fp
249 del fp
250
250
251 # 4. We prepared working directory according to filtered
251 # 4. We prepared working directory according to filtered
252 # patch. Now is the time to delegate the job to
252 # patch. Now is the time to delegate the job to
253 # commit/qrefresh or the like!
253 # commit/qrefresh or the like!
254
254
255 # Make all of the pathnames absolute.
255 # Make all of the pathnames absolute.
256 newfiles = [repo.wjoin(nf) for nf in newfiles]
256 newfiles = [repo.wjoin(nf) for nf in newfiles]
257 return commitfunc(ui, repo, *newfiles, **opts)
257 return commitfunc(ui, repo, *newfiles, **opts)
258 finally:
258 finally:
259 # 5. finally restore backed-up files
259 # 5. finally restore backed-up files
260 try:
260 try:
261 dirstate = repo.dirstate
261 dirstate = repo.dirstate
262 for realname, tmpname in backups.iteritems():
262 for realname, tmpname in backups.iteritems():
263 ui.debug('restoring %r to %r\n' % (tmpname, realname))
263 ui.debug('restoring %r to %r\n' % (tmpname, realname))
264
264
265 if dirstate[realname] == 'n':
265 if dirstate[realname] == 'n':
266 # without normallookup, restoring timestamp
266 # without normallookup, restoring timestamp
267 # may cause partially committed files
267 # may cause partially committed files
268 # to be treated as unmodified
268 # to be treated as unmodified
269 dirstate.normallookup(realname)
269 dirstate.normallookup(realname)
270
270
271 # copystat=True here and above are a hack to trick any
271 # copystat=True here and above are a hack to trick any
272 # editors that have f open that we haven't modified them.
272 # editors that have f open that we haven't modified them.
273 #
273 #
274 # Also note that this racy as an editor could notice the
274 # Also note that this racy as an editor could notice the
275 # file's mtime before we've finished writing it.
275 # file's mtime before we've finished writing it.
276 util.copyfile(tmpname, repo.wjoin(realname), copystat=True)
276 util.copyfile(tmpname, repo.wjoin(realname), copystat=True)
277 os.unlink(tmpname)
277 os.unlink(tmpname)
278 if tobackup:
278 if tobackup:
279 os.rmdir(backupdir)
279 os.rmdir(backupdir)
280 except OSError:
280 except OSError:
281 pass
281 pass
282
282
283 def recordinwlock(ui, repo, message, match, opts):
283 def recordinwlock(ui, repo, message, match, opts):
284 with repo.wlock():
284 with repo.wlock():
285 return recordfunc(ui, repo, message, match, opts)
285 return recordfunc(ui, repo, message, match, opts)
286
286
287 return commit(ui, repo, recordinwlock, pats, opts)
287 return commit(ui, repo, recordinwlock, pats, opts)
288
288
289 def findpossible(cmd, table, strict=False):
289 def findpossible(cmd, table, strict=False):
290 """
290 """
291 Return cmd -> (aliases, command table entry)
291 Return cmd -> (aliases, command table entry)
292 for each matching command.
292 for each matching command.
293 Return debug commands (or their aliases) only if no normal command matches.
293 Return debug commands (or their aliases) only if no normal command matches.
294 """
294 """
295 choice = {}
295 choice = {}
296 debugchoice = {}
296 debugchoice = {}
297
297
298 if cmd in table:
298 if cmd in table:
299 # short-circuit exact matches, "log" alias beats "^log|history"
299 # short-circuit exact matches, "log" alias beats "^log|history"
300 keys = [cmd]
300 keys = [cmd]
301 else:
301 else:
302 keys = table.keys()
302 keys = table.keys()
303
303
304 allcmds = []
304 allcmds = []
305 for e in keys:
305 for e in keys:
306 aliases = parsealiases(e)
306 aliases = parsealiases(e)
307 allcmds.extend(aliases)
307 allcmds.extend(aliases)
308 found = None
308 found = None
309 if cmd in aliases:
309 if cmd in aliases:
310 found = cmd
310 found = cmd
311 elif not strict:
311 elif not strict:
312 for a in aliases:
312 for a in aliases:
313 if a.startswith(cmd):
313 if a.startswith(cmd):
314 found = a
314 found = a
315 break
315 break
316 if found is not None:
316 if found is not None:
317 if aliases[0].startswith("debug") or found.startswith("debug"):
317 if aliases[0].startswith("debug") or found.startswith("debug"):
318 debugchoice[found] = (aliases, table[e])
318 debugchoice[found] = (aliases, table[e])
319 else:
319 else:
320 choice[found] = (aliases, table[e])
320 choice[found] = (aliases, table[e])
321
321
322 if not choice and debugchoice:
322 if not choice and debugchoice:
323 choice = debugchoice
323 choice = debugchoice
324
324
325 return choice, allcmds
325 return choice, allcmds
326
326
327 def findcmd(cmd, table, strict=True):
327 def findcmd(cmd, table, strict=True):
328 """Return (aliases, command table entry) for command string."""
328 """Return (aliases, command table entry) for command string."""
329 choice, allcmds = findpossible(cmd, table, strict)
329 choice, allcmds = findpossible(cmd, table, strict)
330
330
331 if cmd in choice:
331 if cmd in choice:
332 return choice[cmd]
332 return choice[cmd]
333
333
334 if len(choice) > 1:
334 if len(choice) > 1:
335 clist = choice.keys()
335 clist = choice.keys()
336 clist.sort()
336 clist.sort()
337 raise error.AmbiguousCommand(cmd, clist)
337 raise error.AmbiguousCommand(cmd, clist)
338
338
339 if choice:
339 if choice:
340 return choice.values()[0]
340 return choice.values()[0]
341
341
342 raise error.UnknownCommand(cmd, allcmds)
342 raise error.UnknownCommand(cmd, allcmds)
343
343
344 def findrepo(p):
344 def findrepo(p):
345 while not os.path.isdir(os.path.join(p, ".hg")):
345 while not os.path.isdir(os.path.join(p, ".hg")):
346 oldp, p = p, os.path.dirname(p)
346 oldp, p = p, os.path.dirname(p)
347 if p == oldp:
347 if p == oldp:
348 return None
348 return None
349
349
350 return p
350 return p
351
351
352 def bailifchanged(repo, merge=True):
352 def bailifchanged(repo, merge=True):
353 if merge and repo.dirstate.p2() != nullid:
353 if merge and repo.dirstate.p2() != nullid:
354 raise error.Abort(_('outstanding uncommitted merge'))
354 raise error.Abort(_('outstanding uncommitted merge'))
355 modified, added, removed, deleted = repo.status()[:4]
355 modified, added, removed, deleted = repo.status()[:4]
356 if modified or added or removed or deleted:
356 if modified or added or removed or deleted:
357 raise error.Abort(_('uncommitted changes'))
357 raise error.Abort(_('uncommitted changes'))
358 ctx = repo[None]
358 ctx = repo[None]
359 for s in sorted(ctx.substate):
359 for s in sorted(ctx.substate):
360 ctx.sub(s).bailifchanged()
360 ctx.sub(s).bailifchanged()
361
361
362 def logmessage(ui, opts):
362 def logmessage(ui, opts):
363 """ get the log message according to -m and -l option """
363 """ get the log message according to -m and -l option """
364 message = opts.get('message')
364 message = opts.get('message')
365 logfile = opts.get('logfile')
365 logfile = opts.get('logfile')
366
366
367 if message and logfile:
367 if message and logfile:
368 raise error.Abort(_('options --message and --logfile are mutually '
368 raise error.Abort(_('options --message and --logfile are mutually '
369 'exclusive'))
369 'exclusive'))
370 if not message and logfile:
370 if not message and logfile:
371 try:
371 try:
372 if logfile == '-':
372 if logfile == '-':
373 message = ui.fin.read()
373 message = ui.fin.read()
374 else:
374 else:
375 message = '\n'.join(util.readfile(logfile).splitlines())
375 message = '\n'.join(util.readfile(logfile).splitlines())
376 except IOError as inst:
376 except IOError as inst:
377 raise error.Abort(_("can't read commit message '%s': %s") %
377 raise error.Abort(_("can't read commit message '%s': %s") %
378 (logfile, inst.strerror))
378 (logfile, inst.strerror))
379 return message
379 return message
380
380
381 def mergeeditform(ctxorbool, baseformname):
381 def mergeeditform(ctxorbool, baseformname):
382 """return appropriate editform name (referencing a committemplate)
382 """return appropriate editform name (referencing a committemplate)
383
383
384 'ctxorbool' is either a ctx to be committed, or a bool indicating whether
384 'ctxorbool' is either a ctx to be committed, or a bool indicating whether
385 merging is committed.
385 merging is committed.
386
386
387 This returns baseformname with '.merge' appended if it is a merge,
387 This returns baseformname with '.merge' appended if it is a merge,
388 otherwise '.normal' is appended.
388 otherwise '.normal' is appended.
389 """
389 """
390 if isinstance(ctxorbool, bool):
390 if isinstance(ctxorbool, bool):
391 if ctxorbool:
391 if ctxorbool:
392 return baseformname + ".merge"
392 return baseformname + ".merge"
393 elif 1 < len(ctxorbool.parents()):
393 elif 1 < len(ctxorbool.parents()):
394 return baseformname + ".merge"
394 return baseformname + ".merge"
395
395
396 return baseformname + ".normal"
396 return baseformname + ".normal"
397
397
398 def getcommiteditor(edit=False, finishdesc=None, extramsg=None,
398 def getcommiteditor(edit=False, finishdesc=None, extramsg=None,
399 editform='', **opts):
399 editform='', **opts):
400 """get appropriate commit message editor according to '--edit' option
400 """get appropriate commit message editor according to '--edit' option
401
401
402 'finishdesc' is a function to be called with edited commit message
402 'finishdesc' is a function to be called with edited commit message
403 (= 'description' of the new changeset) just after editing, but
403 (= 'description' of the new changeset) just after editing, but
404 before checking empty-ness. It should return actual text to be
404 before checking empty-ness. It should return actual text to be
405 stored into history. This allows to change description before
405 stored into history. This allows to change description before
406 storing.
406 storing.
407
407
408 'extramsg' is a extra message to be shown in the editor instead of
408 'extramsg' is a extra message to be shown in the editor instead of
409 'Leave message empty to abort commit' line. 'HG: ' prefix and EOL
409 'Leave message empty to abort commit' line. 'HG: ' prefix and EOL
410 is automatically added.
410 is automatically added.
411
411
412 'editform' is a dot-separated list of names, to distinguish
412 'editform' is a dot-separated list of names, to distinguish
413 the purpose of commit text editing.
413 the purpose of commit text editing.
414
414
415 'getcommiteditor' returns 'commitforceeditor' regardless of
415 'getcommiteditor' returns 'commitforceeditor' regardless of
416 'edit', if one of 'finishdesc' or 'extramsg' is specified, because
416 'edit', if one of 'finishdesc' or 'extramsg' is specified, because
417 they are specific for usage in MQ.
417 they are specific for usage in MQ.
418 """
418 """
419 if edit or finishdesc or extramsg:
419 if edit or finishdesc or extramsg:
420 return lambda r, c, s: commitforceeditor(r, c, s,
420 return lambda r, c, s: commitforceeditor(r, c, s,
421 finishdesc=finishdesc,
421 finishdesc=finishdesc,
422 extramsg=extramsg,
422 extramsg=extramsg,
423 editform=editform)
423 editform=editform)
424 elif editform:
424 elif editform:
425 return lambda r, c, s: commiteditor(r, c, s, editform=editform)
425 return lambda r, c, s: commiteditor(r, c, s, editform=editform)
426 else:
426 else:
427 return commiteditor
427 return commiteditor
428
428
429 def loglimit(opts):
429 def loglimit(opts):
430 """get the log limit according to option -l/--limit"""
430 """get the log limit according to option -l/--limit"""
431 limit = opts.get('limit')
431 limit = opts.get('limit')
432 if limit:
432 if limit:
433 try:
433 try:
434 limit = int(limit)
434 limit = int(limit)
435 except ValueError:
435 except ValueError:
436 raise error.Abort(_('limit must be a positive integer'))
436 raise error.Abort(_('limit must be a positive integer'))
437 if limit <= 0:
437 if limit <= 0:
438 raise error.Abort(_('limit must be positive'))
438 raise error.Abort(_('limit must be positive'))
439 else:
439 else:
440 limit = None
440 limit = None
441 return limit
441 return limit
442
442
443 def makefilename(repo, pat, node, desc=None,
443 def makefilename(repo, pat, node, desc=None,
444 total=None, seqno=None, revwidth=None, pathname=None):
444 total=None, seqno=None, revwidth=None, pathname=None):
445 node_expander = {
445 node_expander = {
446 'H': lambda: hex(node),
446 'H': lambda: hex(node),
447 'R': lambda: str(repo.changelog.rev(node)),
447 'R': lambda: str(repo.changelog.rev(node)),
448 'h': lambda: short(node),
448 'h': lambda: short(node),
449 'm': lambda: re.sub('[^\w]', '_', str(desc))
449 'm': lambda: re.sub('[^\w]', '_', str(desc))
450 }
450 }
451 expander = {
451 expander = {
452 '%': lambda: '%',
452 '%': lambda: '%',
453 'b': lambda: os.path.basename(repo.root),
453 'b': lambda: os.path.basename(repo.root),
454 }
454 }
455
455
456 try:
456 try:
457 if node:
457 if node:
458 expander.update(node_expander)
458 expander.update(node_expander)
459 if node:
459 if node:
460 expander['r'] = (lambda:
460 expander['r'] = (lambda:
461 str(repo.changelog.rev(node)).zfill(revwidth or 0))
461 str(repo.changelog.rev(node)).zfill(revwidth or 0))
462 if total is not None:
462 if total is not None:
463 expander['N'] = lambda: str(total)
463 expander['N'] = lambda: str(total)
464 if seqno is not None:
464 if seqno is not None:
465 expander['n'] = lambda: str(seqno)
465 expander['n'] = lambda: str(seqno)
466 if total is not None and seqno is not None:
466 if total is not None and seqno is not None:
467 expander['n'] = lambda: str(seqno).zfill(len(str(total)))
467 expander['n'] = lambda: str(seqno).zfill(len(str(total)))
468 if pathname is not None:
468 if pathname is not None:
469 expander['s'] = lambda: os.path.basename(pathname)
469 expander['s'] = lambda: os.path.basename(pathname)
470 expander['d'] = lambda: os.path.dirname(pathname) or '.'
470 expander['d'] = lambda: os.path.dirname(pathname) or '.'
471 expander['p'] = lambda: pathname
471 expander['p'] = lambda: pathname
472
472
473 newname = []
473 newname = []
474 patlen = len(pat)
474 patlen = len(pat)
475 i = 0
475 i = 0
476 while i < patlen:
476 while i < patlen:
477 c = pat[i]
477 c = pat[i]
478 if c == '%':
478 if c == '%':
479 i += 1
479 i += 1
480 c = pat[i]
480 c = pat[i]
481 c = expander[c]()
481 c = expander[c]()
482 newname.append(c)
482 newname.append(c)
483 i += 1
483 i += 1
484 return ''.join(newname)
484 return ''.join(newname)
485 except KeyError as inst:
485 except KeyError as inst:
486 raise error.Abort(_("invalid format spec '%%%s' in output filename") %
486 raise error.Abort(_("invalid format spec '%%%s' in output filename") %
487 inst.args[0])
487 inst.args[0])
488
488
489 class _unclosablefile(object):
489 class _unclosablefile(object):
490 def __init__(self, fp):
490 def __init__(self, fp):
491 self._fp = fp
491 self._fp = fp
492
492
493 def close(self):
493 def close(self):
494 pass
494 pass
495
495
496 def __iter__(self):
496 def __iter__(self):
497 return iter(self._fp)
497 return iter(self._fp)
498
498
499 def __getattr__(self, attr):
499 def __getattr__(self, attr):
500 return getattr(self._fp, attr)
500 return getattr(self._fp, attr)
501
501
502 def makefileobj(repo, pat, node=None, desc=None, total=None,
502 def makefileobj(repo, pat, node=None, desc=None, total=None,
503 seqno=None, revwidth=None, mode='wb', modemap=None,
503 seqno=None, revwidth=None, mode='wb', modemap=None,
504 pathname=None):
504 pathname=None):
505
505
506 writable = mode not in ('r', 'rb')
506 writable = mode not in ('r', 'rb')
507
507
508 if not pat or pat == '-':
508 if not pat or pat == '-':
509 if writable:
509 if writable:
510 fp = repo.ui.fout
510 fp = repo.ui.fout
511 else:
511 else:
512 fp = repo.ui.fin
512 fp = repo.ui.fin
513 return _unclosablefile(fp)
513 return _unclosablefile(fp)
514 if util.safehasattr(pat, 'write') and writable:
514 if util.safehasattr(pat, 'write') and writable:
515 return pat
515 return pat
516 if util.safehasattr(pat, 'read') and 'r' in mode:
516 if util.safehasattr(pat, 'read') and 'r' in mode:
517 return pat
517 return pat
518 fn = makefilename(repo, pat, node, desc, total, seqno, revwidth, pathname)
518 fn = makefilename(repo, pat, node, desc, total, seqno, revwidth, pathname)
519 if modemap is not None:
519 if modemap is not None:
520 mode = modemap.get(fn, mode)
520 mode = modemap.get(fn, mode)
521 if mode == 'wb':
521 if mode == 'wb':
522 modemap[fn] = 'ab'
522 modemap[fn] = 'ab'
523 return open(fn, mode)
523 return open(fn, mode)
524
524
525 def openrevlog(repo, cmd, file_, opts):
525 def openrevlog(repo, cmd, file_, opts):
526 """opens the changelog, manifest, a filelog or a given revlog"""
526 """opens the changelog, manifest, a filelog or a given revlog"""
527 cl = opts['changelog']
527 cl = opts['changelog']
528 mf = opts['manifest']
528 mf = opts['manifest']
529 dir = opts['dir']
529 dir = opts['dir']
530 msg = None
530 msg = None
531 if cl and mf:
531 if cl and mf:
532 msg = _('cannot specify --changelog and --manifest at the same time')
532 msg = _('cannot specify --changelog and --manifest at the same time')
533 elif cl and dir:
533 elif cl and dir:
534 msg = _('cannot specify --changelog and --dir at the same time')
534 msg = _('cannot specify --changelog and --dir at the same time')
535 elif cl or mf:
535 elif cl or mf:
536 if file_:
536 if file_:
537 msg = _('cannot specify filename with --changelog or --manifest')
537 msg = _('cannot specify filename with --changelog or --manifest')
538 elif not repo:
538 elif not repo:
539 msg = _('cannot specify --changelog or --manifest or --dir '
539 msg = _('cannot specify --changelog or --manifest or --dir '
540 'without a repository')
540 'without a repository')
541 if msg:
541 if msg:
542 raise error.Abort(msg)
542 raise error.Abort(msg)
543
543
544 r = None
544 r = None
545 if repo:
545 if repo:
546 if cl:
546 if cl:
547 r = repo.unfiltered().changelog
547 r = repo.unfiltered().changelog
548 elif dir:
548 elif dir:
549 if 'treemanifest' not in repo.requirements:
549 if 'treemanifest' not in repo.requirements:
550 raise error.Abort(_("--dir can only be used on repos with "
550 raise error.Abort(_("--dir can only be used on repos with "
551 "treemanifest enabled"))
551 "treemanifest enabled"))
552 dirlog = repo.dirlog(file_)
552 dirlog = repo.dirlog(file_)
553 if len(dirlog):
553 if len(dirlog):
554 r = dirlog
554 r = dirlog
555 elif mf:
555 elif mf:
556 r = repo.manifest
556 r = repo.manifest
557 elif file_:
557 elif file_:
558 filelog = repo.file(file_)
558 filelog = repo.file(file_)
559 if len(filelog):
559 if len(filelog):
560 r = filelog
560 r = filelog
561 if not r:
561 if not r:
562 if not file_:
562 if not file_:
563 raise error.CommandError(cmd, _('invalid arguments'))
563 raise error.CommandError(cmd, _('invalid arguments'))
564 if not os.path.isfile(file_):
564 if not os.path.isfile(file_):
565 raise error.Abort(_("revlog '%s' not found") % file_)
565 raise error.Abort(_("revlog '%s' not found") % file_)
566 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False),
566 r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False),
567 file_[:-2] + ".i")
567 file_[:-2] + ".i")
568 return r
568 return r
569
569
570 def copy(ui, repo, pats, opts, rename=False):
570 def copy(ui, repo, pats, opts, rename=False):
571 # called with the repo lock held
571 # called with the repo lock held
572 #
572 #
573 # hgsep => pathname that uses "/" to separate directories
573 # hgsep => pathname that uses "/" to separate directories
574 # ossep => pathname that uses os.sep to separate directories
574 # ossep => pathname that uses os.sep to separate directories
575 cwd = repo.getcwd()
575 cwd = repo.getcwd()
576 targets = {}
576 targets = {}
577 after = opts.get("after")
577 after = opts.get("after")
578 dryrun = opts.get("dry_run")
578 dryrun = opts.get("dry_run")
579 wctx = repo[None]
579 wctx = repo[None]
580
580
581 def walkpat(pat):
581 def walkpat(pat):
582 srcs = []
582 srcs = []
583 if after:
583 if after:
584 badstates = '?'
584 badstates = '?'
585 else:
585 else:
586 badstates = '?r'
586 badstates = '?r'
587 m = scmutil.match(repo[None], [pat], opts, globbed=True)
587 m = scmutil.match(repo[None], [pat], opts, globbed=True)
588 for abs in repo.walk(m):
588 for abs in repo.walk(m):
589 state = repo.dirstate[abs]
589 state = repo.dirstate[abs]
590 rel = m.rel(abs)
590 rel = m.rel(abs)
591 exact = m.exact(abs)
591 exact = m.exact(abs)
592 if state in badstates:
592 if state in badstates:
593 if exact and state == '?':
593 if exact and state == '?':
594 ui.warn(_('%s: not copying - file is not managed\n') % rel)
594 ui.warn(_('%s: not copying - file is not managed\n') % rel)
595 if exact and state == 'r':
595 if exact and state == 'r':
596 ui.warn(_('%s: not copying - file has been marked for'
596 ui.warn(_('%s: not copying - file has been marked for'
597 ' remove\n') % rel)
597 ' remove\n') % rel)
598 continue
598 continue
599 # abs: hgsep
599 # abs: hgsep
600 # rel: ossep
600 # rel: ossep
601 srcs.append((abs, rel, exact))
601 srcs.append((abs, rel, exact))
602 return srcs
602 return srcs
603
603
604 # abssrc: hgsep
604 # abssrc: hgsep
605 # relsrc: ossep
605 # relsrc: ossep
606 # otarget: ossep
606 # otarget: ossep
607 def copyfile(abssrc, relsrc, otarget, exact):
607 def copyfile(abssrc, relsrc, otarget, exact):
608 abstarget = pathutil.canonpath(repo.root, cwd, otarget)
608 abstarget = pathutil.canonpath(repo.root, cwd, otarget)
609 if '/' in abstarget:
609 if '/' in abstarget:
610 # We cannot normalize abstarget itself, this would prevent
610 # We cannot normalize abstarget itself, this would prevent
611 # case only renames, like a => A.
611 # case only renames, like a => A.
612 abspath, absname = abstarget.rsplit('/', 1)
612 abspath, absname = abstarget.rsplit('/', 1)
613 abstarget = repo.dirstate.normalize(abspath) + '/' + absname
613 abstarget = repo.dirstate.normalize(abspath) + '/' + absname
614 reltarget = repo.pathto(abstarget, cwd)
614 reltarget = repo.pathto(abstarget, cwd)
615 target = repo.wjoin(abstarget)
615 target = repo.wjoin(abstarget)
616 src = repo.wjoin(abssrc)
616 src = repo.wjoin(abssrc)
617 state = repo.dirstate[abstarget]
617 state = repo.dirstate[abstarget]
618
618
619 scmutil.checkportable(ui, abstarget)
619 scmutil.checkportable(ui, abstarget)
620
620
621 # check for collisions
621 # check for collisions
622 prevsrc = targets.get(abstarget)
622 prevsrc = targets.get(abstarget)
623 if prevsrc is not None:
623 if prevsrc is not None:
624 ui.warn(_('%s: not overwriting - %s collides with %s\n') %
624 ui.warn(_('%s: not overwriting - %s collides with %s\n') %
625 (reltarget, repo.pathto(abssrc, cwd),
625 (reltarget, repo.pathto(abssrc, cwd),
626 repo.pathto(prevsrc, cwd)))
626 repo.pathto(prevsrc, cwd)))
627 return
627 return
628
628
629 # check for overwrites
629 # check for overwrites
630 exists = os.path.lexists(target)
630 exists = os.path.lexists(target)
631 samefile = False
631 samefile = False
632 if exists and abssrc != abstarget:
632 if exists and abssrc != abstarget:
633 if (repo.dirstate.normalize(abssrc) ==
633 if (repo.dirstate.normalize(abssrc) ==
634 repo.dirstate.normalize(abstarget)):
634 repo.dirstate.normalize(abstarget)):
635 if not rename:
635 if not rename:
636 ui.warn(_("%s: can't copy - same file\n") % reltarget)
636 ui.warn(_("%s: can't copy - same file\n") % reltarget)
637 return
637 return
638 exists = False
638 exists = False
639 samefile = True
639 samefile = True
640
640
641 if not after and exists or after and state in 'mn':
641 if not after and exists or after and state in 'mn':
642 if not opts['force']:
642 if not opts['force']:
643 ui.warn(_('%s: not overwriting - file exists\n') %
643 ui.warn(_('%s: not overwriting - file exists\n') %
644 reltarget)
644 reltarget)
645 return
645 return
646
646
647 if after:
647 if after:
648 if not exists:
648 if not exists:
649 if rename:
649 if rename:
650 ui.warn(_('%s: not recording move - %s does not exist\n') %
650 ui.warn(_('%s: not recording move - %s does not exist\n') %
651 (relsrc, reltarget))
651 (relsrc, reltarget))
652 else:
652 else:
653 ui.warn(_('%s: not recording copy - %s does not exist\n') %
653 ui.warn(_('%s: not recording copy - %s does not exist\n') %
654 (relsrc, reltarget))
654 (relsrc, reltarget))
655 return
655 return
656 elif not dryrun:
656 elif not dryrun:
657 try:
657 try:
658 if exists:
658 if exists:
659 os.unlink(target)
659 os.unlink(target)
660 targetdir = os.path.dirname(target) or '.'
660 targetdir = os.path.dirname(target) or '.'
661 if not os.path.isdir(targetdir):
661 if not os.path.isdir(targetdir):
662 os.makedirs(targetdir)
662 os.makedirs(targetdir)
663 if samefile:
663 if samefile:
664 tmp = target + "~hgrename"
664 tmp = target + "~hgrename"
665 os.rename(src, tmp)
665 os.rename(src, tmp)
666 os.rename(tmp, target)
666 os.rename(tmp, target)
667 else:
667 else:
668 util.copyfile(src, target)
668 util.copyfile(src, target)
669 srcexists = True
669 srcexists = True
670 except IOError as inst:
670 except IOError as inst:
671 if inst.errno == errno.ENOENT:
671 if inst.errno == errno.ENOENT:
672 ui.warn(_('%s: deleted in working directory\n') % relsrc)
672 ui.warn(_('%s: deleted in working directory\n') % relsrc)
673 srcexists = False
673 srcexists = False
674 else:
674 else:
675 ui.warn(_('%s: cannot copy - %s\n') %
675 ui.warn(_('%s: cannot copy - %s\n') %
676 (relsrc, inst.strerror))
676 (relsrc, inst.strerror))
677 return True # report a failure
677 return True # report a failure
678
678
679 if ui.verbose or not exact:
679 if ui.verbose or not exact:
680 if rename:
680 if rename:
681 ui.status(_('moving %s to %s\n') % (relsrc, reltarget))
681 ui.status(_('moving %s to %s\n') % (relsrc, reltarget))
682 else:
682 else:
683 ui.status(_('copying %s to %s\n') % (relsrc, reltarget))
683 ui.status(_('copying %s to %s\n') % (relsrc, reltarget))
684
684
685 targets[abstarget] = abssrc
685 targets[abstarget] = abssrc
686
686
687 # fix up dirstate
687 # fix up dirstate
688 scmutil.dirstatecopy(ui, repo, wctx, abssrc, abstarget,
688 scmutil.dirstatecopy(ui, repo, wctx, abssrc, abstarget,
689 dryrun=dryrun, cwd=cwd)
689 dryrun=dryrun, cwd=cwd)
690 if rename and not dryrun:
690 if rename and not dryrun:
691 if not after and srcexists and not samefile:
691 if not after and srcexists and not samefile:
692 util.unlinkpath(repo.wjoin(abssrc))
692 util.unlinkpath(repo.wjoin(abssrc))
693 wctx.forget([abssrc])
693 wctx.forget([abssrc])
694
694
695 # pat: ossep
695 # pat: ossep
696 # dest ossep
696 # dest ossep
697 # srcs: list of (hgsep, hgsep, ossep, bool)
697 # srcs: list of (hgsep, hgsep, ossep, bool)
698 # return: function that takes hgsep and returns ossep
698 # return: function that takes hgsep and returns ossep
699 def targetpathfn(pat, dest, srcs):
699 def targetpathfn(pat, dest, srcs):
700 if os.path.isdir(pat):
700 if os.path.isdir(pat):
701 abspfx = pathutil.canonpath(repo.root, cwd, pat)
701 abspfx = pathutil.canonpath(repo.root, cwd, pat)
702 abspfx = util.localpath(abspfx)
702 abspfx = util.localpath(abspfx)
703 if destdirexists:
703 if destdirexists:
704 striplen = len(os.path.split(abspfx)[0])
704 striplen = len(os.path.split(abspfx)[0])
705 else:
705 else:
706 striplen = len(abspfx)
706 striplen = len(abspfx)
707 if striplen:
707 if striplen:
708 striplen += len(os.sep)
708 striplen += len(os.sep)
709 res = lambda p: os.path.join(dest, util.localpath(p)[striplen:])
709 res = lambda p: os.path.join(dest, util.localpath(p)[striplen:])
710 elif destdirexists:
710 elif destdirexists:
711 res = lambda p: os.path.join(dest,
711 res = lambda p: os.path.join(dest,
712 os.path.basename(util.localpath(p)))
712 os.path.basename(util.localpath(p)))
713 else:
713 else:
714 res = lambda p: dest
714 res = lambda p: dest
715 return res
715 return res
716
716
717 # pat: ossep
717 # pat: ossep
718 # dest ossep
718 # dest ossep
719 # srcs: list of (hgsep, hgsep, ossep, bool)
719 # srcs: list of (hgsep, hgsep, ossep, bool)
720 # return: function that takes hgsep and returns ossep
720 # return: function that takes hgsep and returns ossep
721 def targetpathafterfn(pat, dest, srcs):
721 def targetpathafterfn(pat, dest, srcs):
722 if matchmod.patkind(pat):
722 if matchmod.patkind(pat):
723 # a mercurial pattern
723 # a mercurial pattern
724 res = lambda p: os.path.join(dest,
724 res = lambda p: os.path.join(dest,
725 os.path.basename(util.localpath(p)))
725 os.path.basename(util.localpath(p)))
726 else:
726 else:
727 abspfx = pathutil.canonpath(repo.root, cwd, pat)
727 abspfx = pathutil.canonpath(repo.root, cwd, pat)
728 if len(abspfx) < len(srcs[0][0]):
728 if len(abspfx) < len(srcs[0][0]):
729 # A directory. Either the target path contains the last
729 # A directory. Either the target path contains the last
730 # component of the source path or it does not.
730 # component of the source path or it does not.
731 def evalpath(striplen):
731 def evalpath(striplen):
732 score = 0
732 score = 0
733 for s in srcs:
733 for s in srcs:
734 t = os.path.join(dest, util.localpath(s[0])[striplen:])
734 t = os.path.join(dest, util.localpath(s[0])[striplen:])
735 if os.path.lexists(t):
735 if os.path.lexists(t):
736 score += 1
736 score += 1
737 return score
737 return score
738
738
739 abspfx = util.localpath(abspfx)
739 abspfx = util.localpath(abspfx)
740 striplen = len(abspfx)
740 striplen = len(abspfx)
741 if striplen:
741 if striplen:
742 striplen += len(os.sep)
742 striplen += len(os.sep)
743 if os.path.isdir(os.path.join(dest, os.path.split(abspfx)[1])):
743 if os.path.isdir(os.path.join(dest, os.path.split(abspfx)[1])):
744 score = evalpath(striplen)
744 score = evalpath(striplen)
745 striplen1 = len(os.path.split(abspfx)[0])
745 striplen1 = len(os.path.split(abspfx)[0])
746 if striplen1:
746 if striplen1:
747 striplen1 += len(os.sep)
747 striplen1 += len(os.sep)
748 if evalpath(striplen1) > score:
748 if evalpath(striplen1) > score:
749 striplen = striplen1
749 striplen = striplen1
750 res = lambda p: os.path.join(dest,
750 res = lambda p: os.path.join(dest,
751 util.localpath(p)[striplen:])
751 util.localpath(p)[striplen:])
752 else:
752 else:
753 # a file
753 # a file
754 if destdirexists:
754 if destdirexists:
755 res = lambda p: os.path.join(dest,
755 res = lambda p: os.path.join(dest,
756 os.path.basename(util.localpath(p)))
756 os.path.basename(util.localpath(p)))
757 else:
757 else:
758 res = lambda p: dest
758 res = lambda p: dest
759 return res
759 return res
760
760
761 pats = scmutil.expandpats(pats)
761 pats = scmutil.expandpats(pats)
762 if not pats:
762 if not pats:
763 raise error.Abort(_('no source or destination specified'))
763 raise error.Abort(_('no source or destination specified'))
764 if len(pats) == 1:
764 if len(pats) == 1:
765 raise error.Abort(_('no destination specified'))
765 raise error.Abort(_('no destination specified'))
766 dest = pats.pop()
766 dest = pats.pop()
767 destdirexists = os.path.isdir(dest) and not os.path.islink(dest)
767 destdirexists = os.path.isdir(dest) and not os.path.islink(dest)
768 if not destdirexists:
768 if not destdirexists:
769 if len(pats) > 1 or matchmod.patkind(pats[0]):
769 if len(pats) > 1 or matchmod.patkind(pats[0]):
770 raise error.Abort(_('with multiple sources, destination must be an '
770 raise error.Abort(_('with multiple sources, destination must be an '
771 'existing directory'))
771 'existing directory'))
772 if util.endswithsep(dest):
772 if util.endswithsep(dest):
773 raise error.Abort(_('destination %s is not a directory') % dest)
773 raise error.Abort(_('destination %s is not a directory') % dest)
774
774
775 tfn = targetpathfn
775 tfn = targetpathfn
776 if after:
776 if after:
777 tfn = targetpathafterfn
777 tfn = targetpathafterfn
778 copylist = []
778 copylist = []
779 for pat in pats:
779 for pat in pats:
780 srcs = walkpat(pat)
780 srcs = walkpat(pat)
781 if not srcs:
781 if not srcs:
782 continue
782 continue
783 copylist.append((tfn(pat, dest, srcs), srcs))
783 copylist.append((tfn(pat, dest, srcs), srcs))
784 if not copylist:
784 if not copylist:
785 raise error.Abort(_('no files to copy'))
785 raise error.Abort(_('no files to copy'))
786
786
787 errors = 0
787 errors = 0
788 for targetpath, srcs in copylist:
788 for targetpath, srcs in copylist:
789 for abssrc, relsrc, exact in srcs:
789 for abssrc, relsrc, exact in srcs:
790 if copyfile(abssrc, relsrc, targetpath(abssrc), exact):
790 if copyfile(abssrc, relsrc, targetpath(abssrc), exact):
791 errors += 1
791 errors += 1
792
792
793 if errors:
793 if errors:
794 ui.warn(_('(consider using --after)\n'))
794 ui.warn(_('(consider using --after)\n'))
795
795
796 return errors != 0
796 return errors != 0
797
797
798 def service(opts, parentfn=None, initfn=None, runfn=None, logfile=None,
798 def service(opts, parentfn=None, initfn=None, runfn=None, logfile=None,
799 runargs=None, appendpid=False):
799 runargs=None, appendpid=False):
800 '''Run a command as a service.'''
800 '''Run a command as a service.'''
801
801
802 def writepid(pid):
802 def writepid(pid):
803 if opts['pid_file']:
803 if opts['pid_file']:
804 if appendpid:
804 if appendpid:
805 mode = 'a'
805 mode = 'a'
806 else:
806 else:
807 mode = 'w'
807 mode = 'w'
808 fp = open(opts['pid_file'], mode)
808 fp = open(opts['pid_file'], mode)
809 fp.write(str(pid) + '\n')
809 fp.write(str(pid) + '\n')
810 fp.close()
810 fp.close()
811
811
812 if opts['daemon'] and not opts['daemon_postexec']:
812 if opts['daemon'] and not opts['daemon_postexec']:
813 # Signal child process startup with file removal
813 # Signal child process startup with file removal
814 lockfd, lockpath = tempfile.mkstemp(prefix='hg-service-')
814 lockfd, lockpath = tempfile.mkstemp(prefix='hg-service-')
815 os.close(lockfd)
815 os.close(lockfd)
816 try:
816 try:
817 if not runargs:
817 if not runargs:
818 runargs = util.hgcmd() + sys.argv[1:]
818 runargs = util.hgcmd() + sys.argv[1:]
819 runargs.append('--daemon-postexec=unlink:%s' % lockpath)
819 runargs.append('--daemon-postexec=unlink:%s' % lockpath)
820 # Don't pass --cwd to the child process, because we've already
820 # Don't pass --cwd to the child process, because we've already
821 # changed directory.
821 # changed directory.
822 for i in xrange(1, len(runargs)):
822 for i in xrange(1, len(runargs)):
823 if runargs[i].startswith('--cwd='):
823 if runargs[i].startswith('--cwd='):
824 del runargs[i]
824 del runargs[i]
825 break
825 break
826 elif runargs[i].startswith('--cwd'):
826 elif runargs[i].startswith('--cwd'):
827 del runargs[i:i + 2]
827 del runargs[i:i + 2]
828 break
828 break
829 def condfn():
829 def condfn():
830 return not os.path.exists(lockpath)
830 return not os.path.exists(lockpath)
831 pid = util.rundetached(runargs, condfn)
831 pid = util.rundetached(runargs, condfn)
832 if pid < 0:
832 if pid < 0:
833 raise error.Abort(_('child process failed to start'))
833 raise error.Abort(_('child process failed to start'))
834 writepid(pid)
834 writepid(pid)
835 finally:
835 finally:
836 try:
836 try:
837 os.unlink(lockpath)
837 os.unlink(lockpath)
838 except OSError as e:
838 except OSError as e:
839 if e.errno != errno.ENOENT:
839 if e.errno != errno.ENOENT:
840 raise
840 raise
841 if parentfn:
841 if parentfn:
842 return parentfn(pid)
842 return parentfn(pid)
843 else:
843 else:
844 return
844 return
845
845
846 if initfn:
846 if initfn:
847 initfn()
847 initfn()
848
848
849 if not opts['daemon']:
849 if not opts['daemon']:
850 writepid(util.getpid())
850 writepid(util.getpid())
851
851
852 if opts['daemon_postexec']:
852 if opts['daemon_postexec']:
853 try:
853 try:
854 os.setsid()
854 os.setsid()
855 except AttributeError:
855 except AttributeError:
856 pass
856 pass
857 for inst in opts['daemon_postexec']:
857 for inst in opts['daemon_postexec']:
858 if inst.startswith('unlink:'):
858 if inst.startswith('unlink:'):
859 lockpath = inst[7:]
859 lockpath = inst[7:]
860 os.unlink(lockpath)
860 os.unlink(lockpath)
861 elif inst.startswith('chdir:'):
861 elif inst.startswith('chdir:'):
862 os.chdir(inst[6:])
862 os.chdir(inst[6:])
863 elif inst != 'none':
863 elif inst != 'none':
864 raise error.Abort(_('invalid value for --daemon-postexec: %s')
864 raise error.Abort(_('invalid value for --daemon-postexec: %s')
865 % inst)
865 % inst)
866 util.hidewindow()
866 util.hidewindow()
867 sys.stdout.flush()
867 sys.stdout.flush()
868 sys.stderr.flush()
868 sys.stderr.flush()
869
869
870 nullfd = os.open(os.devnull, os.O_RDWR)
870 nullfd = os.open(os.devnull, os.O_RDWR)
871 logfilefd = nullfd
871 logfilefd = nullfd
872 if logfile:
872 if logfile:
873 logfilefd = os.open(logfile, os.O_RDWR | os.O_CREAT | os.O_APPEND)
873 logfilefd = os.open(logfile, os.O_RDWR | os.O_CREAT | os.O_APPEND)
874 os.dup2(nullfd, 0)
874 os.dup2(nullfd, 0)
875 os.dup2(logfilefd, 1)
875 os.dup2(logfilefd, 1)
876 os.dup2(logfilefd, 2)
876 os.dup2(logfilefd, 2)
877 if nullfd not in (0, 1, 2):
877 if nullfd not in (0, 1, 2):
878 os.close(nullfd)
878 os.close(nullfd)
879 if logfile and logfilefd not in (0, 1, 2):
879 if logfile and logfilefd not in (0, 1, 2):
880 os.close(logfilefd)
880 os.close(logfilefd)
881
881
882 if runfn:
882 if runfn:
883 return runfn()
883 return runfn()
884
884
885 ## facility to let extension process additional data into an import patch
885 ## facility to let extension process additional data into an import patch
886 # list of identifier to be executed in order
886 # list of identifier to be executed in order
887 extrapreimport = [] # run before commit
887 extrapreimport = [] # run before commit
888 extrapostimport = [] # run after commit
888 extrapostimport = [] # run after commit
889 # mapping from identifier to actual import function
889 # mapping from identifier to actual import function
890 #
890 #
891 # 'preimport' are run before the commit is made and are provided the following
891 # 'preimport' are run before the commit is made and are provided the following
892 # arguments:
892 # arguments:
893 # - repo: the localrepository instance,
893 # - repo: the localrepository instance,
894 # - patchdata: data extracted from patch header (cf m.patch.patchheadermap),
894 # - patchdata: data extracted from patch header (cf m.patch.patchheadermap),
895 # - extra: the future extra dictionary of the changeset, please mutate it,
895 # - extra: the future extra dictionary of the changeset, please mutate it,
896 # - opts: the import options.
896 # - opts: the import options.
897 # XXX ideally, we would just pass an ctx ready to be computed, that would allow
897 # XXX ideally, we would just pass an ctx ready to be computed, that would allow
898 # mutation of in memory commit and more. Feel free to rework the code to get
898 # mutation of in memory commit and more. Feel free to rework the code to get
899 # there.
899 # there.
900 extrapreimportmap = {}
900 extrapreimportmap = {}
901 # 'postimport' are run after the commit is made and are provided the following
901 # 'postimport' are run after the commit is made and are provided the following
902 # argument:
902 # argument:
903 # - ctx: the changectx created by import.
903 # - ctx: the changectx created by import.
904 extrapostimportmap = {}
904 extrapostimportmap = {}
905
905
906 def tryimportone(ui, repo, hunk, parents, opts, msgs, updatefunc):
906 def tryimportone(ui, repo, hunk, parents, opts, msgs, updatefunc):
907 """Utility function used by commands.import to import a single patch
907 """Utility function used by commands.import to import a single patch
908
908
909 This function is explicitly defined here to help the evolve extension to
909 This function is explicitly defined here to help the evolve extension to
910 wrap this part of the import logic.
910 wrap this part of the import logic.
911
911
912 The API is currently a bit ugly because it a simple code translation from
912 The API is currently a bit ugly because it a simple code translation from
913 the import command. Feel free to make it better.
913 the import command. Feel free to make it better.
914
914
915 :hunk: a patch (as a binary string)
915 :hunk: a patch (as a binary string)
916 :parents: nodes that will be parent of the created commit
916 :parents: nodes that will be parent of the created commit
917 :opts: the full dict of option passed to the import command
917 :opts: the full dict of option passed to the import command
918 :msgs: list to save commit message to.
918 :msgs: list to save commit message to.
919 (used in case we need to save it when failing)
919 (used in case we need to save it when failing)
920 :updatefunc: a function that update a repo to a given node
920 :updatefunc: a function that update a repo to a given node
921 updatefunc(<repo>, <node>)
921 updatefunc(<repo>, <node>)
922 """
922 """
923 # avoid cycle context -> subrepo -> cmdutil
923 # avoid cycle context -> subrepo -> cmdutil
924 from . import context
924 from . import context
925 extractdata = patch.extract(ui, hunk)
925 extractdata = patch.extract(ui, hunk)
926 tmpname = extractdata.get('filename')
926 tmpname = extractdata.get('filename')
927 message = extractdata.get('message')
927 message = extractdata.get('message')
928 user = opts.get('user') or extractdata.get('user')
928 user = opts.get('user') or extractdata.get('user')
929 date = opts.get('date') or extractdata.get('date')
929 date = opts.get('date') or extractdata.get('date')
930 branch = extractdata.get('branch')
930 branch = extractdata.get('branch')
931 nodeid = extractdata.get('nodeid')
931 nodeid = extractdata.get('nodeid')
932 p1 = extractdata.get('p1')
932 p1 = extractdata.get('p1')
933 p2 = extractdata.get('p2')
933 p2 = extractdata.get('p2')
934
934
935 nocommit = opts.get('no_commit')
935 nocommit = opts.get('no_commit')
936 importbranch = opts.get('import_branch')
936 importbranch = opts.get('import_branch')
937 update = not opts.get('bypass')
937 update = not opts.get('bypass')
938 strip = opts["strip"]
938 strip = opts["strip"]
939 prefix = opts["prefix"]
939 prefix = opts["prefix"]
940 sim = float(opts.get('similarity') or 0)
940 sim = float(opts.get('similarity') or 0)
941 if not tmpname:
941 if not tmpname:
942 return (None, None, False)
942 return (None, None, False)
943
943
944 rejects = False
944 rejects = False
945
945
946 try:
946 try:
947 cmdline_message = logmessage(ui, opts)
947 cmdline_message = logmessage(ui, opts)
948 if cmdline_message:
948 if cmdline_message:
949 # pickup the cmdline msg
949 # pickup the cmdline msg
950 message = cmdline_message
950 message = cmdline_message
951 elif message:
951 elif message:
952 # pickup the patch msg
952 # pickup the patch msg
953 message = message.strip()
953 message = message.strip()
954 else:
954 else:
955 # launch the editor
955 # launch the editor
956 message = None
956 message = None
957 ui.debug('message:\n%s\n' % message)
957 ui.debug('message:\n%s\n' % message)
958
958
959 if len(parents) == 1:
959 if len(parents) == 1:
960 parents.append(repo[nullid])
960 parents.append(repo[nullid])
961 if opts.get('exact'):
961 if opts.get('exact'):
962 if not nodeid or not p1:
962 if not nodeid or not p1:
963 raise error.Abort(_('not a Mercurial patch'))
963 raise error.Abort(_('not a Mercurial patch'))
964 p1 = repo[p1]
964 p1 = repo[p1]
965 p2 = repo[p2 or nullid]
965 p2 = repo[p2 or nullid]
966 elif p2:
966 elif p2:
967 try:
967 try:
968 p1 = repo[p1]
968 p1 = repo[p1]
969 p2 = repo[p2]
969 p2 = repo[p2]
970 # Without any options, consider p2 only if the
970 # Without any options, consider p2 only if the
971 # patch is being applied on top of the recorded
971 # patch is being applied on top of the recorded
972 # first parent.
972 # first parent.
973 if p1 != parents[0]:
973 if p1 != parents[0]:
974 p1 = parents[0]
974 p1 = parents[0]
975 p2 = repo[nullid]
975 p2 = repo[nullid]
976 except error.RepoError:
976 except error.RepoError:
977 p1, p2 = parents
977 p1, p2 = parents
978 if p2.node() == nullid:
978 if p2.node() == nullid:
979 ui.warn(_("warning: import the patch as a normal revision\n"
979 ui.warn(_("warning: import the patch as a normal revision\n"
980 "(use --exact to import the patch as a merge)\n"))
980 "(use --exact to import the patch as a merge)\n"))
981 else:
981 else:
982 p1, p2 = parents
982 p1, p2 = parents
983
983
984 n = None
984 n = None
985 if update:
985 if update:
986 if p1 != parents[0]:
986 if p1 != parents[0]:
987 updatefunc(repo, p1.node())
987 updatefunc(repo, p1.node())
988 if p2 != parents[1]:
988 if p2 != parents[1]:
989 repo.setparents(p1.node(), p2.node())
989 repo.setparents(p1.node(), p2.node())
990
990
991 if opts.get('exact') or importbranch:
991 if opts.get('exact') or importbranch:
992 repo.dirstate.setbranch(branch or 'default')
992 repo.dirstate.setbranch(branch or 'default')
993
993
994 partial = opts.get('partial', False)
994 partial = opts.get('partial', False)
995 files = set()
995 files = set()
996 try:
996 try:
997 patch.patch(ui, repo, tmpname, strip=strip, prefix=prefix,
997 patch.patch(ui, repo, tmpname, strip=strip, prefix=prefix,
998 files=files, eolmode=None, similarity=sim / 100.0)
998 files=files, eolmode=None, similarity=sim / 100.0)
999 except patch.PatchError as e:
999 except patch.PatchError as e:
1000 if not partial:
1000 if not partial:
1001 raise error.Abort(str(e))
1001 raise error.Abort(str(e))
1002 if partial:
1002 if partial:
1003 rejects = True
1003 rejects = True
1004
1004
1005 files = list(files)
1005 files = list(files)
1006 if nocommit:
1006 if nocommit:
1007 if message:
1007 if message:
1008 msgs.append(message)
1008 msgs.append(message)
1009 else:
1009 else:
1010 if opts.get('exact') or p2:
1010 if opts.get('exact') or p2:
1011 # If you got here, you either use --force and know what
1011 # If you got here, you either use --force and know what
1012 # you are doing or used --exact or a merge patch while
1012 # you are doing or used --exact or a merge patch while
1013 # being updated to its first parent.
1013 # being updated to its first parent.
1014 m = None
1014 m = None
1015 else:
1015 else:
1016 m = scmutil.matchfiles(repo, files or [])
1016 m = scmutil.matchfiles(repo, files or [])
1017 editform = mergeeditform(repo[None], 'import.normal')
1017 editform = mergeeditform(repo[None], 'import.normal')
1018 if opts.get('exact'):
1018 if opts.get('exact'):
1019 editor = None
1019 editor = None
1020 else:
1020 else:
1021 editor = getcommiteditor(editform=editform, **opts)
1021 editor = getcommiteditor(editform=editform, **opts)
1022 allowemptyback = repo.ui.backupconfig('ui', 'allowemptycommit')
1022 allowemptyback = repo.ui.backupconfig('ui', 'allowemptycommit')
1023 extra = {}
1023 extra = {}
1024 for idfunc in extrapreimport:
1024 for idfunc in extrapreimport:
1025 extrapreimportmap[idfunc](repo, extractdata, extra, opts)
1025 extrapreimportmap[idfunc](repo, extractdata, extra, opts)
1026 try:
1026 try:
1027 if partial:
1027 if partial:
1028 repo.ui.setconfig('ui', 'allowemptycommit', True)
1028 repo.ui.setconfig('ui', 'allowemptycommit', True)
1029 n = repo.commit(message, user,
1029 n = repo.commit(message, user,
1030 date, match=m,
1030 date, match=m,
1031 editor=editor, extra=extra)
1031 editor=editor, extra=extra)
1032 for idfunc in extrapostimport:
1032 for idfunc in extrapostimport:
1033 extrapostimportmap[idfunc](repo[n])
1033 extrapostimportmap[idfunc](repo[n])
1034 finally:
1034 finally:
1035 repo.ui.restoreconfig(allowemptyback)
1035 repo.ui.restoreconfig(allowemptyback)
1036 else:
1036 else:
1037 if opts.get('exact') or importbranch:
1037 if opts.get('exact') or importbranch:
1038 branch = branch or 'default'
1038 branch = branch or 'default'
1039 else:
1039 else:
1040 branch = p1.branch()
1040 branch = p1.branch()
1041 store = patch.filestore()
1041 store = patch.filestore()
1042 try:
1042 try:
1043 files = set()
1043 files = set()
1044 try:
1044 try:
1045 patch.patchrepo(ui, repo, p1, store, tmpname, strip, prefix,
1045 patch.patchrepo(ui, repo, p1, store, tmpname, strip, prefix,
1046 files, eolmode=None)
1046 files, eolmode=None)
1047 except patch.PatchError as e:
1047 except patch.PatchError as e:
1048 raise error.Abort(str(e))
1048 raise error.Abort(str(e))
1049 if opts.get('exact'):
1049 if opts.get('exact'):
1050 editor = None
1050 editor = None
1051 else:
1051 else:
1052 editor = getcommiteditor(editform='import.bypass')
1052 editor = getcommiteditor(editform='import.bypass')
1053 memctx = context.makememctx(repo, (p1.node(), p2.node()),
1053 memctx = context.makememctx(repo, (p1.node(), p2.node()),
1054 message,
1054 message,
1055 user,
1055 user,
1056 date,
1056 date,
1057 branch, files, store,
1057 branch, files, store,
1058 editor=editor)
1058 editor=editor)
1059 n = memctx.commit()
1059 n = memctx.commit()
1060 finally:
1060 finally:
1061 store.close()
1061 store.close()
1062 if opts.get('exact') and nocommit:
1062 if opts.get('exact') and nocommit:
1063 # --exact with --no-commit is still useful in that it does merge
1063 # --exact with --no-commit is still useful in that it does merge
1064 # and branch bits
1064 # and branch bits
1065 ui.warn(_("warning: can't check exact import with --no-commit\n"))
1065 ui.warn(_("warning: can't check exact import with --no-commit\n"))
1066 elif opts.get('exact') and hex(n) != nodeid:
1066 elif opts.get('exact') and hex(n) != nodeid:
1067 raise error.Abort(_('patch is damaged or loses information'))
1067 raise error.Abort(_('patch is damaged or loses information'))
1068 msg = _('applied to working directory')
1068 msg = _('applied to working directory')
1069 if n:
1069 if n:
1070 # i18n: refers to a short changeset id
1070 # i18n: refers to a short changeset id
1071 msg = _('created %s') % short(n)
1071 msg = _('created %s') % short(n)
1072 return (msg, n, rejects)
1072 return (msg, n, rejects)
1073 finally:
1073 finally:
1074 os.unlink(tmpname)
1074 os.unlink(tmpname)
1075
1075
1076 # facility to let extensions include additional data in an exported patch
1076 # facility to let extensions include additional data in an exported patch
1077 # list of identifiers to be executed in order
1077 # list of identifiers to be executed in order
1078 extraexport = []
1078 extraexport = []
1079 # mapping from identifier to actual export function
1079 # mapping from identifier to actual export function
1080 # function as to return a string to be added to the header or None
1080 # function as to return a string to be added to the header or None
1081 # it is given two arguments (sequencenumber, changectx)
1081 # it is given two arguments (sequencenumber, changectx)
1082 extraexportmap = {}
1082 extraexportmap = {}
1083
1083
1084 def export(repo, revs, template='hg-%h.patch', fp=None, switch_parent=False,
1084 def export(repo, revs, template='hg-%h.patch', fp=None, switch_parent=False,
1085 opts=None, match=None):
1085 opts=None, match=None):
1086 '''export changesets as hg patches.'''
1086 '''export changesets as hg patches.'''
1087
1087
1088 total = len(revs)
1088 total = len(revs)
1089 revwidth = max([len(str(rev)) for rev in revs])
1089 revwidth = max([len(str(rev)) for rev in revs])
1090 filemode = {}
1090 filemode = {}
1091
1091
1092 def single(rev, seqno, fp):
1092 def single(rev, seqno, fp):
1093 ctx = repo[rev]
1093 ctx = repo[rev]
1094 node = ctx.node()
1094 node = ctx.node()
1095 parents = [p.node() for p in ctx.parents() if p]
1095 parents = [p.node() for p in ctx.parents() if p]
1096 branch = ctx.branch()
1096 branch = ctx.branch()
1097 if switch_parent:
1097 if switch_parent:
1098 parents.reverse()
1098 parents.reverse()
1099
1099
1100 if parents:
1100 if parents:
1101 prev = parents[0]
1101 prev = parents[0]
1102 else:
1102 else:
1103 prev = nullid
1103 prev = nullid
1104
1104
1105 shouldclose = False
1105 shouldclose = False
1106 if not fp and len(template) > 0:
1106 if not fp and len(template) > 0:
1107 desc_lines = ctx.description().rstrip().split('\n')
1107 desc_lines = ctx.description().rstrip().split('\n')
1108 desc = desc_lines[0] #Commit always has a first line.
1108 desc = desc_lines[0] #Commit always has a first line.
1109 fp = makefileobj(repo, template, node, desc=desc, total=total,
1109 fp = makefileobj(repo, template, node, desc=desc, total=total,
1110 seqno=seqno, revwidth=revwidth, mode='wb',
1110 seqno=seqno, revwidth=revwidth, mode='wb',
1111 modemap=filemode)
1111 modemap=filemode)
1112 shouldclose = True
1112 shouldclose = True
1113 if fp and not getattr(fp, 'name', '<unnamed>').startswith('<'):
1113 if fp and not getattr(fp, 'name', '<unnamed>').startswith('<'):
1114 repo.ui.note("%s\n" % fp.name)
1114 repo.ui.note("%s\n" % fp.name)
1115
1115
1116 if not fp:
1116 if not fp:
1117 write = repo.ui.write
1117 write = repo.ui.write
1118 else:
1118 else:
1119 def write(s, **kw):
1119 def write(s, **kw):
1120 fp.write(s)
1120 fp.write(s)
1121
1121
1122 write("# HG changeset patch\n")
1122 write("# HG changeset patch\n")
1123 write("# User %s\n" % ctx.user())
1123 write("# User %s\n" % ctx.user())
1124 write("# Date %d %d\n" % ctx.date())
1124 write("# Date %d %d\n" % ctx.date())
1125 write("# %s\n" % util.datestr(ctx.date()))
1125 write("# %s\n" % util.datestr(ctx.date()))
1126 if branch and branch != 'default':
1126 if branch and branch != 'default':
1127 write("# Branch %s\n" % branch)
1127 write("# Branch %s\n" % branch)
1128 write("# Node ID %s\n" % hex(node))
1128 write("# Node ID %s\n" % hex(node))
1129 write("# Parent %s\n" % hex(prev))
1129 write("# Parent %s\n" % hex(prev))
1130 if len(parents) > 1:
1130 if len(parents) > 1:
1131 write("# Parent %s\n" % hex(parents[1]))
1131 write("# Parent %s\n" % hex(parents[1]))
1132
1132
1133 for headerid in extraexport:
1133 for headerid in extraexport:
1134 header = extraexportmap[headerid](seqno, ctx)
1134 header = extraexportmap[headerid](seqno, ctx)
1135 if header is not None:
1135 if header is not None:
1136 write('# %s\n' % header)
1136 write('# %s\n' % header)
1137 write(ctx.description().rstrip())
1137 write(ctx.description().rstrip())
1138 write("\n\n")
1138 write("\n\n")
1139
1139
1140 for chunk, label in patch.diffui(repo, prev, node, match, opts=opts):
1140 for chunk, label in patch.diffui(repo, prev, node, match, opts=opts):
1141 write(chunk, label=label)
1141 write(chunk, label=label)
1142
1142
1143 if shouldclose:
1143 if shouldclose:
1144 fp.close()
1144 fp.close()
1145
1145
1146 for seqno, rev in enumerate(revs):
1146 for seqno, rev in enumerate(revs):
1147 single(rev, seqno + 1, fp)
1147 single(rev, seqno + 1, fp)
1148
1148
1149 def diffordiffstat(ui, repo, diffopts, node1, node2, match,
1149 def diffordiffstat(ui, repo, diffopts, node1, node2, match,
1150 changes=None, stat=False, fp=None, prefix='',
1150 changes=None, stat=False, fp=None, prefix='',
1151 root='', listsubrepos=False):
1151 root='', listsubrepos=False):
1152 '''show diff or diffstat.'''
1152 '''show diff or diffstat.'''
1153 if fp is None:
1153 if fp is None:
1154 write = ui.write
1154 write = ui.write
1155 else:
1155 else:
1156 def write(s, **kw):
1156 def write(s, **kw):
1157 fp.write(s)
1157 fp.write(s)
1158
1158
1159 if root:
1159 if root:
1160 relroot = pathutil.canonpath(repo.root, repo.getcwd(), root)
1160 relroot = pathutil.canonpath(repo.root, repo.getcwd(), root)
1161 else:
1161 else:
1162 relroot = ''
1162 relroot = ''
1163 if relroot != '':
1163 if relroot != '':
1164 # XXX relative roots currently don't work if the root is within a
1164 # XXX relative roots currently don't work if the root is within a
1165 # subrepo
1165 # subrepo
1166 uirelroot = match.uipath(relroot)
1166 uirelroot = match.uipath(relroot)
1167 relroot += '/'
1167 relroot += '/'
1168 for matchroot in match.files():
1168 for matchroot in match.files():
1169 if not matchroot.startswith(relroot):
1169 if not matchroot.startswith(relroot):
1170 ui.warn(_('warning: %s not inside relative root %s\n') % (
1170 ui.warn(_('warning: %s not inside relative root %s\n') % (
1171 match.uipath(matchroot), uirelroot))
1171 match.uipath(matchroot), uirelroot))
1172
1172
1173 if stat:
1173 if stat:
1174 diffopts = diffopts.copy(context=0)
1174 diffopts = diffopts.copy(context=0)
1175 width = 80
1175 width = 80
1176 if not ui.plain():
1176 if not ui.plain():
1177 width = ui.termwidth()
1177 width = ui.termwidth()
1178 chunks = patch.diff(repo, node1, node2, match, changes, diffopts,
1178 chunks = patch.diff(repo, node1, node2, match, changes, diffopts,
1179 prefix=prefix, relroot=relroot)
1179 prefix=prefix, relroot=relroot)
1180 for chunk, label in patch.diffstatui(util.iterlines(chunks),
1180 for chunk, label in patch.diffstatui(util.iterlines(chunks),
1181 width=width,
1181 width=width,
1182 git=diffopts.git):
1182 git=diffopts.git):
1183 write(chunk, label=label)
1183 write(chunk, label=label)
1184 else:
1184 else:
1185 for chunk, label in patch.diffui(repo, node1, node2, match,
1185 for chunk, label in patch.diffui(repo, node1, node2, match,
1186 changes, diffopts, prefix=prefix,
1186 changes, diffopts, prefix=prefix,
1187 relroot=relroot):
1187 relroot=relroot):
1188 write(chunk, label=label)
1188 write(chunk, label=label)
1189
1189
1190 if listsubrepos:
1190 if listsubrepos:
1191 ctx1 = repo[node1]
1191 ctx1 = repo[node1]
1192 ctx2 = repo[node2]
1192 ctx2 = repo[node2]
1193 for subpath, sub in scmutil.itersubrepos(ctx1, ctx2):
1193 for subpath, sub in scmutil.itersubrepos(ctx1, ctx2):
1194 tempnode2 = node2
1194 tempnode2 = node2
1195 try:
1195 try:
1196 if node2 is not None:
1196 if node2 is not None:
1197 tempnode2 = ctx2.substate[subpath][1]
1197 tempnode2 = ctx2.substate[subpath][1]
1198 except KeyError:
1198 except KeyError:
1199 # A subrepo that existed in node1 was deleted between node1 and
1199 # A subrepo that existed in node1 was deleted between node1 and
1200 # node2 (inclusive). Thus, ctx2's substate won't contain that
1200 # node2 (inclusive). Thus, ctx2's substate won't contain that
1201 # subpath. The best we can do is to ignore it.
1201 # subpath. The best we can do is to ignore it.
1202 tempnode2 = None
1202 tempnode2 = None
1203 submatch = matchmod.subdirmatcher(subpath, match)
1203 submatch = matchmod.subdirmatcher(subpath, match)
1204 sub.diff(ui, diffopts, tempnode2, submatch, changes=changes,
1204 sub.diff(ui, diffopts, tempnode2, submatch, changes=changes,
1205 stat=stat, fp=fp, prefix=prefix)
1205 stat=stat, fp=fp, prefix=prefix)
1206
1206
1207 class changeset_printer(object):
1207 class changeset_printer(object):
1208 '''show changeset information when templating not requested.'''
1208 '''show changeset information when templating not requested.'''
1209
1209
1210 def __init__(self, ui, repo, matchfn, diffopts, buffered):
1210 def __init__(self, ui, repo, matchfn, diffopts, buffered):
1211 self.ui = ui
1211 self.ui = ui
1212 self.repo = repo
1212 self.repo = repo
1213 self.buffered = buffered
1213 self.buffered = buffered
1214 self.matchfn = matchfn
1214 self.matchfn = matchfn
1215 self.diffopts = diffopts
1215 self.diffopts = diffopts
1216 self.header = {}
1216 self.header = {}
1217 self.hunk = {}
1217 self.hunk = {}
1218 self.lastheader = None
1218 self.lastheader = None
1219 self.footer = None
1219 self.footer = None
1220
1220
1221 def flush(self, ctx):
1221 def flush(self, ctx):
1222 rev = ctx.rev()
1222 rev = ctx.rev()
1223 if rev in self.header:
1223 if rev in self.header:
1224 h = self.header[rev]
1224 h = self.header[rev]
1225 if h != self.lastheader:
1225 if h != self.lastheader:
1226 self.lastheader = h
1226 self.lastheader = h
1227 self.ui.write(h)
1227 self.ui.write(h)
1228 del self.header[rev]
1228 del self.header[rev]
1229 if rev in self.hunk:
1229 if rev in self.hunk:
1230 self.ui.write(self.hunk[rev])
1230 self.ui.write(self.hunk[rev])
1231 del self.hunk[rev]
1231 del self.hunk[rev]
1232 return 1
1232 return 1
1233 return 0
1233 return 0
1234
1234
1235 def close(self):
1235 def close(self):
1236 if self.footer:
1236 if self.footer:
1237 self.ui.write(self.footer)
1237 self.ui.write(self.footer)
1238
1238
1239 def show(self, ctx, copies=None, matchfn=None, **props):
1239 def show(self, ctx, copies=None, matchfn=None, **props):
1240 if self.buffered:
1240 if self.buffered:
1241 self.ui.pushbuffer(labeled=True)
1241 self.ui.pushbuffer(labeled=True)
1242 self._show(ctx, copies, matchfn, props)
1242 self._show(ctx, copies, matchfn, props)
1243 self.hunk[ctx.rev()] = self.ui.popbuffer()
1243 self.hunk[ctx.rev()] = self.ui.popbuffer()
1244 else:
1244 else:
1245 self._show(ctx, copies, matchfn, props)
1245 self._show(ctx, copies, matchfn, props)
1246
1246
1247 def _show(self, ctx, copies, matchfn, props):
1247 def _show(self, ctx, copies, matchfn, props):
1248 '''show a single changeset or file revision'''
1248 '''show a single changeset or file revision'''
1249 changenode = ctx.node()
1249 changenode = ctx.node()
1250 rev = ctx.rev()
1250 rev = ctx.rev()
1251 if self.ui.debugflag:
1251 if self.ui.debugflag:
1252 hexfunc = hex
1252 hexfunc = hex
1253 else:
1253 else:
1254 hexfunc = short
1254 hexfunc = short
1255 # as of now, wctx.node() and wctx.rev() return None, but we want to
1255 # as of now, wctx.node() and wctx.rev() return None, but we want to
1256 # show the same values as {node} and {rev} templatekw
1256 # show the same values as {node} and {rev} templatekw
1257 revnode = (scmutil.intrev(rev), hexfunc(bin(ctx.hex())))
1257 revnode = (scmutil.intrev(rev), hexfunc(bin(ctx.hex())))
1258
1258
1259 if self.ui.quiet:
1259 if self.ui.quiet:
1260 self.ui.write("%d:%s\n" % revnode, label='log.node')
1260 self.ui.write("%d:%s\n" % revnode, label='log.node')
1261 return
1261 return
1262
1262
1263 date = util.datestr(ctx.date())
1263 date = util.datestr(ctx.date())
1264
1264
1265 # i18n: column positioning for "hg log"
1265 # i18n: column positioning for "hg log"
1266 self.ui.write(_("changeset: %d:%s\n") % revnode,
1266 self.ui.write(_("changeset: %d:%s\n") % revnode,
1267 label='log.changeset changeset.%s' % ctx.phasestr())
1267 label='log.changeset changeset.%s' % ctx.phasestr())
1268
1268
1269 # branches are shown first before any other names due to backwards
1269 # branches are shown first before any other names due to backwards
1270 # compatibility
1270 # compatibility
1271 branch = ctx.branch()
1271 branch = ctx.branch()
1272 # don't show the default branch name
1272 # don't show the default branch name
1273 if branch != 'default':
1273 if branch != 'default':
1274 # i18n: column positioning for "hg log"
1274 # i18n: column positioning for "hg log"
1275 self.ui.write(_("branch: %s\n") % branch,
1275 self.ui.write(_("branch: %s\n") % branch,
1276 label='log.branch')
1276 label='log.branch')
1277
1277
1278 for nsname, ns in self.repo.names.iteritems():
1278 for nsname, ns in self.repo.names.iteritems():
1279 # branches has special logic already handled above, so here we just
1279 # branches has special logic already handled above, so here we just
1280 # skip it
1280 # skip it
1281 if nsname == 'branches':
1281 if nsname == 'branches':
1282 continue
1282 continue
1283 # we will use the templatename as the color name since those two
1283 # we will use the templatename as the color name since those two
1284 # should be the same
1284 # should be the same
1285 for name in ns.names(self.repo, changenode):
1285 for name in ns.names(self.repo, changenode):
1286 self.ui.write(ns.logfmt % name,
1286 self.ui.write(ns.logfmt % name,
1287 label='log.%s' % ns.colorname)
1287 label='log.%s' % ns.colorname)
1288 if self.ui.debugflag:
1288 if self.ui.debugflag:
1289 # i18n: column positioning for "hg log"
1289 # i18n: column positioning for "hg log"
1290 self.ui.write(_("phase: %s\n") % ctx.phasestr(),
1290 self.ui.write(_("phase: %s\n") % ctx.phasestr(),
1291 label='log.phase')
1291 label='log.phase')
1292 for pctx in scmutil.meaningfulparents(self.repo, ctx):
1292 for pctx in scmutil.meaningfulparents(self.repo, ctx):
1293 label = 'log.parent changeset.%s' % pctx.phasestr()
1293 label = 'log.parent changeset.%s' % pctx.phasestr()
1294 # i18n: column positioning for "hg log"
1294 # i18n: column positioning for "hg log"
1295 self.ui.write(_("parent: %d:%s\n")
1295 self.ui.write(_("parent: %d:%s\n")
1296 % (pctx.rev(), hexfunc(pctx.node())),
1296 % (pctx.rev(), hexfunc(pctx.node())),
1297 label=label)
1297 label=label)
1298
1298
1299 if self.ui.debugflag and rev is not None:
1299 if self.ui.debugflag and rev is not None:
1300 mnode = ctx.manifestnode()
1300 mnode = ctx.manifestnode()
1301 # i18n: column positioning for "hg log"
1301 # i18n: column positioning for "hg log"
1302 self.ui.write(_("manifest: %d:%s\n") %
1302 self.ui.write(_("manifest: %d:%s\n") %
1303 (self.repo.manifest.rev(mnode), hex(mnode)),
1303 (self.repo.manifest.rev(mnode), hex(mnode)),
1304 label='ui.debug log.manifest')
1304 label='ui.debug log.manifest')
1305 # i18n: column positioning for "hg log"
1305 # i18n: column positioning for "hg log"
1306 self.ui.write(_("user: %s\n") % ctx.user(),
1306 self.ui.write(_("user: %s\n") % ctx.user(),
1307 label='log.user')
1307 label='log.user')
1308 # i18n: column positioning for "hg log"
1308 # i18n: column positioning for "hg log"
1309 self.ui.write(_("date: %s\n") % date,
1309 self.ui.write(_("date: %s\n") % date,
1310 label='log.date')
1310 label='log.date')
1311
1311
1312 if self.ui.debugflag:
1312 if self.ui.debugflag:
1313 files = ctx.p1().status(ctx)[:3]
1313 files = ctx.p1().status(ctx)[:3]
1314 for key, value in zip([# i18n: column positioning for "hg log"
1314 for key, value in zip([# i18n: column positioning for "hg log"
1315 _("files:"),
1315 _("files:"),
1316 # i18n: column positioning for "hg log"
1316 # i18n: column positioning for "hg log"
1317 _("files+:"),
1317 _("files+:"),
1318 # i18n: column positioning for "hg log"
1318 # i18n: column positioning for "hg log"
1319 _("files-:")], files):
1319 _("files-:")], files):
1320 if value:
1320 if value:
1321 self.ui.write("%-12s %s\n" % (key, " ".join(value)),
1321 self.ui.write("%-12s %s\n" % (key, " ".join(value)),
1322 label='ui.debug log.files')
1322 label='ui.debug log.files')
1323 elif ctx.files() and self.ui.verbose:
1323 elif ctx.files() and self.ui.verbose:
1324 # i18n: column positioning for "hg log"
1324 # i18n: column positioning for "hg log"
1325 self.ui.write(_("files: %s\n") % " ".join(ctx.files()),
1325 self.ui.write(_("files: %s\n") % " ".join(ctx.files()),
1326 label='ui.note log.files')
1326 label='ui.note log.files')
1327 if copies and self.ui.verbose:
1327 if copies and self.ui.verbose:
1328 copies = ['%s (%s)' % c for c in copies]
1328 copies = ['%s (%s)' % c for c in copies]
1329 # i18n: column positioning for "hg log"
1329 # i18n: column positioning for "hg log"
1330 self.ui.write(_("copies: %s\n") % ' '.join(copies),
1330 self.ui.write(_("copies: %s\n") % ' '.join(copies),
1331 label='ui.note log.copies')
1331 label='ui.note log.copies')
1332
1332
1333 extra = ctx.extra()
1333 extra = ctx.extra()
1334 if extra and self.ui.debugflag:
1334 if extra and self.ui.debugflag:
1335 for key, value in sorted(extra.items()):
1335 for key, value in sorted(extra.items()):
1336 # i18n: column positioning for "hg log"
1336 # i18n: column positioning for "hg log"
1337 self.ui.write(_("extra: %s=%s\n")
1337 self.ui.write(_("extra: %s=%s\n")
1338 % (key, value.encode('string_escape')),
1338 % (key, value.encode('string_escape')),
1339 label='ui.debug log.extra')
1339 label='ui.debug log.extra')
1340
1340
1341 description = ctx.description().strip()
1341 description = ctx.description().strip()
1342 if description:
1342 if description:
1343 if self.ui.verbose:
1343 if self.ui.verbose:
1344 self.ui.write(_("description:\n"),
1344 self.ui.write(_("description:\n"),
1345 label='ui.note log.description')
1345 label='ui.note log.description')
1346 self.ui.write(description,
1346 self.ui.write(description,
1347 label='ui.note log.description')
1347 label='ui.note log.description')
1348 self.ui.write("\n\n")
1348 self.ui.write("\n\n")
1349 else:
1349 else:
1350 # i18n: column positioning for "hg log"
1350 # i18n: column positioning for "hg log"
1351 self.ui.write(_("summary: %s\n") %
1351 self.ui.write(_("summary: %s\n") %
1352 description.splitlines()[0],
1352 description.splitlines()[0],
1353 label='log.summary')
1353 label='log.summary')
1354 self.ui.write("\n")
1354 self.ui.write("\n")
1355
1355
1356 self.showpatch(ctx, matchfn)
1356 self.showpatch(ctx, matchfn)
1357
1357
1358 def showpatch(self, ctx, matchfn):
1358 def showpatch(self, ctx, matchfn):
1359 if not matchfn:
1359 if not matchfn:
1360 matchfn = self.matchfn
1360 matchfn = self.matchfn
1361 if matchfn:
1361 if matchfn:
1362 stat = self.diffopts.get('stat')
1362 stat = self.diffopts.get('stat')
1363 diff = self.diffopts.get('patch')
1363 diff = self.diffopts.get('patch')
1364 diffopts = patch.diffallopts(self.ui, self.diffopts)
1364 diffopts = patch.diffallopts(self.ui, self.diffopts)
1365 node = ctx.node()
1365 node = ctx.node()
1366 prev = ctx.p1().node()
1366 prev = ctx.p1().node()
1367 if stat:
1367 if stat:
1368 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1368 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1369 match=matchfn, stat=True)
1369 match=matchfn, stat=True)
1370 if diff:
1370 if diff:
1371 if stat:
1371 if stat:
1372 self.ui.write("\n")
1372 self.ui.write("\n")
1373 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1373 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1374 match=matchfn, stat=False)
1374 match=matchfn, stat=False)
1375 self.ui.write("\n")
1375 self.ui.write("\n")
1376
1376
1377 class jsonchangeset(changeset_printer):
1377 class jsonchangeset(changeset_printer):
1378 '''format changeset information.'''
1378 '''format changeset information.'''
1379
1379
1380 def __init__(self, ui, repo, matchfn, diffopts, buffered):
1380 def __init__(self, ui, repo, matchfn, diffopts, buffered):
1381 changeset_printer.__init__(self, ui, repo, matchfn, diffopts, buffered)
1381 changeset_printer.__init__(self, ui, repo, matchfn, diffopts, buffered)
1382 self.cache = {}
1382 self.cache = {}
1383 self._first = True
1383 self._first = True
1384
1384
1385 def close(self):
1385 def close(self):
1386 if not self._first:
1386 if not self._first:
1387 self.ui.write("\n]\n")
1387 self.ui.write("\n]\n")
1388 else:
1388 else:
1389 self.ui.write("[]\n")
1389 self.ui.write("[]\n")
1390
1390
1391 def _show(self, ctx, copies, matchfn, props):
1391 def _show(self, ctx, copies, matchfn, props):
1392 '''show a single changeset or file revision'''
1392 '''show a single changeset or file revision'''
1393 rev = ctx.rev()
1393 rev = ctx.rev()
1394 if rev is None:
1394 if rev is None:
1395 jrev = jnode = 'null'
1395 jrev = jnode = 'null'
1396 else:
1396 else:
1397 jrev = str(rev)
1397 jrev = str(rev)
1398 jnode = '"%s"' % hex(ctx.node())
1398 jnode = '"%s"' % hex(ctx.node())
1399 j = encoding.jsonescape
1399 j = encoding.jsonescape
1400
1400
1401 if self._first:
1401 if self._first:
1402 self.ui.write("[\n {")
1402 self.ui.write("[\n {")
1403 self._first = False
1403 self._first = False
1404 else:
1404 else:
1405 self.ui.write(",\n {")
1405 self.ui.write(",\n {")
1406
1406
1407 if self.ui.quiet:
1407 if self.ui.quiet:
1408 self.ui.write('\n "rev": %s' % jrev)
1408 self.ui.write('\n "rev": %s' % jrev)
1409 self.ui.write(',\n "node": %s' % jnode)
1409 self.ui.write(',\n "node": %s' % jnode)
1410 self.ui.write('\n }')
1410 self.ui.write('\n }')
1411 return
1411 return
1412
1412
1413 self.ui.write('\n "rev": %s' % jrev)
1413 self.ui.write('\n "rev": %s' % jrev)
1414 self.ui.write(',\n "node": %s' % jnode)
1414 self.ui.write(',\n "node": %s' % jnode)
1415 self.ui.write(',\n "branch": "%s"' % j(ctx.branch()))
1415 self.ui.write(',\n "branch": "%s"' % j(ctx.branch()))
1416 self.ui.write(',\n "phase": "%s"' % ctx.phasestr())
1416 self.ui.write(',\n "phase": "%s"' % ctx.phasestr())
1417 self.ui.write(',\n "user": "%s"' % j(ctx.user()))
1417 self.ui.write(',\n "user": "%s"' % j(ctx.user()))
1418 self.ui.write(',\n "date": [%d, %d]' % ctx.date())
1418 self.ui.write(',\n "date": [%d, %d]' % ctx.date())
1419 self.ui.write(',\n "desc": "%s"' % j(ctx.description()))
1419 self.ui.write(',\n "desc": "%s"' % j(ctx.description()))
1420
1420
1421 self.ui.write(',\n "bookmarks": [%s]' %
1421 self.ui.write(',\n "bookmarks": [%s]' %
1422 ", ".join('"%s"' % j(b) for b in ctx.bookmarks()))
1422 ", ".join('"%s"' % j(b) for b in ctx.bookmarks()))
1423 self.ui.write(',\n "tags": [%s]' %
1423 self.ui.write(',\n "tags": [%s]' %
1424 ", ".join('"%s"' % j(t) for t in ctx.tags()))
1424 ", ".join('"%s"' % j(t) for t in ctx.tags()))
1425 self.ui.write(',\n "parents": [%s]' %
1425 self.ui.write(',\n "parents": [%s]' %
1426 ", ".join('"%s"' % c.hex() for c in ctx.parents()))
1426 ", ".join('"%s"' % c.hex() for c in ctx.parents()))
1427
1427
1428 if self.ui.debugflag:
1428 if self.ui.debugflag:
1429 if rev is None:
1429 if rev is None:
1430 jmanifestnode = 'null'
1430 jmanifestnode = 'null'
1431 else:
1431 else:
1432 jmanifestnode = '"%s"' % hex(ctx.manifestnode())
1432 jmanifestnode = '"%s"' % hex(ctx.manifestnode())
1433 self.ui.write(',\n "manifest": %s' % jmanifestnode)
1433 self.ui.write(',\n "manifest": %s' % jmanifestnode)
1434
1434
1435 self.ui.write(',\n "extra": {%s}' %
1435 self.ui.write(',\n "extra": {%s}' %
1436 ", ".join('"%s": "%s"' % (j(k), j(v))
1436 ", ".join('"%s": "%s"' % (j(k), j(v))
1437 for k, v in ctx.extra().items()))
1437 for k, v in ctx.extra().items()))
1438
1438
1439 files = ctx.p1().status(ctx)
1439 files = ctx.p1().status(ctx)
1440 self.ui.write(',\n "modified": [%s]' %
1440 self.ui.write(',\n "modified": [%s]' %
1441 ", ".join('"%s"' % j(f) for f in files[0]))
1441 ", ".join('"%s"' % j(f) for f in files[0]))
1442 self.ui.write(',\n "added": [%s]' %
1442 self.ui.write(',\n "added": [%s]' %
1443 ", ".join('"%s"' % j(f) for f in files[1]))
1443 ", ".join('"%s"' % j(f) for f in files[1]))
1444 self.ui.write(',\n "removed": [%s]' %
1444 self.ui.write(',\n "removed": [%s]' %
1445 ", ".join('"%s"' % j(f) for f in files[2]))
1445 ", ".join('"%s"' % j(f) for f in files[2]))
1446
1446
1447 elif self.ui.verbose:
1447 elif self.ui.verbose:
1448 self.ui.write(',\n "files": [%s]' %
1448 self.ui.write(',\n "files": [%s]' %
1449 ", ".join('"%s"' % j(f) for f in ctx.files()))
1449 ", ".join('"%s"' % j(f) for f in ctx.files()))
1450
1450
1451 if copies:
1451 if copies:
1452 self.ui.write(',\n "copies": {%s}' %
1452 self.ui.write(',\n "copies": {%s}' %
1453 ", ".join('"%s": "%s"' % (j(k), j(v))
1453 ", ".join('"%s": "%s"' % (j(k), j(v))
1454 for k, v in copies))
1454 for k, v in copies))
1455
1455
1456 matchfn = self.matchfn
1456 matchfn = self.matchfn
1457 if matchfn:
1457 if matchfn:
1458 stat = self.diffopts.get('stat')
1458 stat = self.diffopts.get('stat')
1459 diff = self.diffopts.get('patch')
1459 diff = self.diffopts.get('patch')
1460 diffopts = patch.difffeatureopts(self.ui, self.diffopts, git=True)
1460 diffopts = patch.difffeatureopts(self.ui, self.diffopts, git=True)
1461 node, prev = ctx.node(), ctx.p1().node()
1461 node, prev = ctx.node(), ctx.p1().node()
1462 if stat:
1462 if stat:
1463 self.ui.pushbuffer()
1463 self.ui.pushbuffer()
1464 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1464 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1465 match=matchfn, stat=True)
1465 match=matchfn, stat=True)
1466 self.ui.write(',\n "diffstat": "%s"' % j(self.ui.popbuffer()))
1466 self.ui.write(',\n "diffstat": "%s"' % j(self.ui.popbuffer()))
1467 if diff:
1467 if diff:
1468 self.ui.pushbuffer()
1468 self.ui.pushbuffer()
1469 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1469 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1470 match=matchfn, stat=False)
1470 match=matchfn, stat=False)
1471 self.ui.write(',\n "diff": "%s"' % j(self.ui.popbuffer()))
1471 self.ui.write(',\n "diff": "%s"' % j(self.ui.popbuffer()))
1472
1472
1473 self.ui.write("\n }")
1473 self.ui.write("\n }")
1474
1474
1475 class changeset_templater(changeset_printer):
1475 class changeset_templater(changeset_printer):
1476 '''format changeset information.'''
1476 '''format changeset information.'''
1477
1477
1478 def __init__(self, ui, repo, matchfn, diffopts, tmpl, mapfile, buffered):
1478 def __init__(self, ui, repo, matchfn, diffopts, tmpl, mapfile, buffered):
1479 changeset_printer.__init__(self, ui, repo, matchfn, diffopts, buffered)
1479 changeset_printer.__init__(self, ui, repo, matchfn, diffopts, buffered)
1480 formatnode = ui.debugflag and (lambda x: x) or (lambda x: x[:12])
1480 formatnode = ui.debugflag and (lambda x: x) or (lambda x: x[:12])
1481 filters = {'formatnode': formatnode}
1481 filters = {'formatnode': formatnode}
1482 defaulttempl = {
1482 defaulttempl = {
1483 'parent': '{rev}:{node|formatnode} ',
1483 'parent': '{rev}:{node|formatnode} ',
1484 'manifest': '{rev}:{node|formatnode}',
1484 'manifest': '{rev}:{node|formatnode}',
1485 'file_copy': '{name} ({source})',
1485 'file_copy': '{name} ({source})',
1486 'extra': '{key}={value|stringescape}'
1486 'extra': '{key}={value|stringescape}'
1487 }
1487 }
1488 # filecopy is preserved for compatibility reasons
1488 # filecopy is preserved for compatibility reasons
1489 defaulttempl['filecopy'] = defaulttempl['file_copy']
1489 defaulttempl['filecopy'] = defaulttempl['file_copy']
1490 assert not (tmpl and mapfile)
1490 assert not (tmpl and mapfile)
1491 if mapfile:
1491 if mapfile:
1492 self.t = templater.templater.frommapfile(mapfile, filters=filters,
1492 self.t = templater.templater.frommapfile(mapfile, filters=filters,
1493 cache=defaulttempl)
1493 cache=defaulttempl)
1494 else:
1494 else:
1495 self.t = formatter.maketemplater(ui, 'changeset', tmpl,
1495 self.t = formatter.maketemplater(ui, 'changeset', tmpl,
1496 filters=filters,
1496 filters=filters,
1497 cache=defaulttempl)
1497 cache=defaulttempl)
1498
1498
1499 self.cache = {}
1499 self.cache = {}
1500
1500
1501 # find correct templates for current mode
1501 # find correct templates for current mode
1502 tmplmodes = [
1502 tmplmodes = [
1503 (True, None),
1503 (True, None),
1504 (self.ui.verbose, 'verbose'),
1504 (self.ui.verbose, 'verbose'),
1505 (self.ui.quiet, 'quiet'),
1505 (self.ui.quiet, 'quiet'),
1506 (self.ui.debugflag, 'debug'),
1506 (self.ui.debugflag, 'debug'),
1507 ]
1507 ]
1508
1508
1509 self._parts = {'header': '', 'footer': '', 'changeset': 'changeset',
1509 self._parts = {'header': '', 'footer': '', 'changeset': 'changeset',
1510 'docheader': '', 'docfooter': ''}
1510 'docheader': '', 'docfooter': ''}
1511 for mode, postfix in tmplmodes:
1511 for mode, postfix in tmplmodes:
1512 for t in self._parts:
1512 for t in self._parts:
1513 cur = t
1513 cur = t
1514 if postfix:
1514 if postfix:
1515 cur += "_" + postfix
1515 cur += "_" + postfix
1516 if mode and cur in self.t:
1516 if mode and cur in self.t:
1517 self._parts[t] = cur
1517 self._parts[t] = cur
1518
1518
1519 if self._parts['docheader']:
1519 if self._parts['docheader']:
1520 self.ui.write(templater.stringify(self.t(self._parts['docheader'])))
1520 self.ui.write(templater.stringify(self.t(self._parts['docheader'])))
1521
1521
1522 def close(self):
1522 def close(self):
1523 if self._parts['docfooter']:
1523 if self._parts['docfooter']:
1524 if not self.footer:
1524 if not self.footer:
1525 self.footer = ""
1525 self.footer = ""
1526 self.footer += templater.stringify(self.t(self._parts['docfooter']))
1526 self.footer += templater.stringify(self.t(self._parts['docfooter']))
1527 return super(changeset_templater, self).close()
1527 return super(changeset_templater, self).close()
1528
1528
1529 def _show(self, ctx, copies, matchfn, props):
1529 def _show(self, ctx, copies, matchfn, props):
1530 '''show a single changeset or file revision'''
1530 '''show a single changeset or file revision'''
1531 props = props.copy()
1531 props = props.copy()
1532 props.update(templatekw.keywords)
1532 props.update(templatekw.keywords)
1533 props['templ'] = self.t
1533 props['templ'] = self.t
1534 props['ctx'] = ctx
1534 props['ctx'] = ctx
1535 props['repo'] = self.repo
1535 props['repo'] = self.repo
1536 props['ui'] = self.repo.ui
1536 props['ui'] = self.repo.ui
1537 props['revcache'] = {'copies': copies}
1537 props['revcache'] = {'copies': copies}
1538 props['cache'] = self.cache
1538 props['cache'] = self.cache
1539
1539
1540 # write header
1540 # write header
1541 if self._parts['header']:
1541 if self._parts['header']:
1542 h = templater.stringify(self.t(self._parts['header'], **props))
1542 h = templater.stringify(self.t(self._parts['header'], **props))
1543 if self.buffered:
1543 if self.buffered:
1544 self.header[ctx.rev()] = h
1544 self.header[ctx.rev()] = h
1545 else:
1545 else:
1546 if self.lastheader != h:
1546 if self.lastheader != h:
1547 self.lastheader = h
1547 self.lastheader = h
1548 self.ui.write(h)
1548 self.ui.write(h)
1549
1549
1550 # write changeset metadata, then patch if requested
1550 # write changeset metadata, then patch if requested
1551 key = self._parts['changeset']
1551 key = self._parts['changeset']
1552 self.ui.write(templater.stringify(self.t(key, **props)))
1552 self.ui.write(templater.stringify(self.t(key, **props)))
1553 self.showpatch(ctx, matchfn)
1553 self.showpatch(ctx, matchfn)
1554
1554
1555 if self._parts['footer']:
1555 if self._parts['footer']:
1556 if not self.footer:
1556 if not self.footer:
1557 self.footer = templater.stringify(
1557 self.footer = templater.stringify(
1558 self.t(self._parts['footer'], **props))
1558 self.t(self._parts['footer'], **props))
1559
1559
1560 def gettemplate(ui, tmpl, style):
1560 def gettemplate(ui, tmpl, style):
1561 """
1561 """
1562 Find the template matching the given template spec or style.
1562 Find the template matching the given template spec or style.
1563 """
1563 """
1564
1564
1565 # ui settings
1565 # ui settings
1566 if not tmpl and not style: # template are stronger than style
1566 if not tmpl and not style: # template are stronger than style
1567 tmpl = ui.config('ui', 'logtemplate')
1567 tmpl = ui.config('ui', 'logtemplate')
1568 if tmpl:
1568 if tmpl:
1569 return templater.unquotestring(tmpl), None
1569 return templater.unquotestring(tmpl), None
1570 else:
1570 else:
1571 style = util.expandpath(ui.config('ui', 'style', ''))
1571 style = util.expandpath(ui.config('ui', 'style', ''))
1572
1572
1573 if not tmpl and style:
1573 if not tmpl and style:
1574 mapfile = style
1574 mapfile = style
1575 if not os.path.split(mapfile)[0]:
1575 if not os.path.split(mapfile)[0]:
1576 mapname = (templater.templatepath('map-cmdline.' + mapfile)
1576 mapname = (templater.templatepath('map-cmdline.' + mapfile)
1577 or templater.templatepath(mapfile))
1577 or templater.templatepath(mapfile))
1578 if mapname:
1578 if mapname:
1579 mapfile = mapname
1579 mapfile = mapname
1580 return None, mapfile
1580 return None, mapfile
1581
1581
1582 if not tmpl:
1582 if not tmpl:
1583 return None, None
1583 return None, None
1584
1584
1585 return formatter.lookuptemplate(ui, 'changeset', tmpl)
1585 return formatter.lookuptemplate(ui, 'changeset', tmpl)
1586
1586
1587 def show_changeset(ui, repo, opts, buffered=False):
1587 def show_changeset(ui, repo, opts, buffered=False):
1588 """show one changeset using template or regular display.
1588 """show one changeset using template or regular display.
1589
1589
1590 Display format will be the first non-empty hit of:
1590 Display format will be the first non-empty hit of:
1591 1. option 'template'
1591 1. option 'template'
1592 2. option 'style'
1592 2. option 'style'
1593 3. [ui] setting 'logtemplate'
1593 3. [ui] setting 'logtemplate'
1594 4. [ui] setting 'style'
1594 4. [ui] setting 'style'
1595 If all of these values are either the unset or the empty string,
1595 If all of these values are either the unset or the empty string,
1596 regular display via changeset_printer() is done.
1596 regular display via changeset_printer() is done.
1597 """
1597 """
1598 # options
1598 # options
1599 matchfn = None
1599 matchfn = None
1600 if opts.get('patch') or opts.get('stat'):
1600 if opts.get('patch') or opts.get('stat'):
1601 matchfn = scmutil.matchall(repo)
1601 matchfn = scmutil.matchall(repo)
1602
1602
1603 if opts.get('template') == 'json':
1603 if opts.get('template') == 'json':
1604 return jsonchangeset(ui, repo, matchfn, opts, buffered)
1604 return jsonchangeset(ui, repo, matchfn, opts, buffered)
1605
1605
1606 tmpl, mapfile = gettemplate(ui, opts.get('template'), opts.get('style'))
1606 tmpl, mapfile = gettemplate(ui, opts.get('template'), opts.get('style'))
1607
1607
1608 if not tmpl and not mapfile:
1608 if not tmpl and not mapfile:
1609 return changeset_printer(ui, repo, matchfn, opts, buffered)
1609 return changeset_printer(ui, repo, matchfn, opts, buffered)
1610
1610
1611 return changeset_templater(ui, repo, matchfn, opts, tmpl, mapfile, buffered)
1611 return changeset_templater(ui, repo, matchfn, opts, tmpl, mapfile, buffered)
1612
1612
1613 def showmarker(ui, marker, index=None):
1613 def showmarker(ui, marker, index=None):
1614 """utility function to display obsolescence marker in a readable way
1614 """utility function to display obsolescence marker in a readable way
1615
1615
1616 To be used by debug function."""
1616 To be used by debug function."""
1617 if index is not None:
1617 if index is not None:
1618 ui.write("%i " % index)
1618 ui.write("%i " % index)
1619 ui.write(hex(marker.precnode()))
1619 ui.write(hex(marker.precnode()))
1620 for repl in marker.succnodes():
1620 for repl in marker.succnodes():
1621 ui.write(' ')
1621 ui.write(' ')
1622 ui.write(hex(repl))
1622 ui.write(hex(repl))
1623 ui.write(' %X ' % marker.flags())
1623 ui.write(' %X ' % marker.flags())
1624 parents = marker.parentnodes()
1624 parents = marker.parentnodes()
1625 if parents is not None:
1625 if parents is not None:
1626 ui.write('{%s} ' % ', '.join(hex(p) for p in parents))
1626 ui.write('{%s} ' % ', '.join(hex(p) for p in parents))
1627 ui.write('(%s) ' % util.datestr(marker.date()))
1627 ui.write('(%s) ' % util.datestr(marker.date()))
1628 ui.write('{%s}' % (', '.join('%r: %r' % t for t in
1628 ui.write('{%s}' % (', '.join('%r: %r' % t for t in
1629 sorted(marker.metadata().items())
1629 sorted(marker.metadata().items())
1630 if t[0] != 'date')))
1630 if t[0] != 'date')))
1631 ui.write('\n')
1631 ui.write('\n')
1632
1632
1633 def finddate(ui, repo, date):
1633 def finddate(ui, repo, date):
1634 """Find the tipmost changeset that matches the given date spec"""
1634 """Find the tipmost changeset that matches the given date spec"""
1635
1635
1636 df = util.matchdate(date)
1636 df = util.matchdate(date)
1637 m = scmutil.matchall(repo)
1637 m = scmutil.matchall(repo)
1638 results = {}
1638 results = {}
1639
1639
1640 def prep(ctx, fns):
1640 def prep(ctx, fns):
1641 d = ctx.date()
1641 d = ctx.date()
1642 if df(d[0]):
1642 if df(d[0]):
1643 results[ctx.rev()] = d
1643 results[ctx.rev()] = d
1644
1644
1645 for ctx in walkchangerevs(repo, m, {'rev': None}, prep):
1645 for ctx in walkchangerevs(repo, m, {'rev': None}, prep):
1646 rev = ctx.rev()
1646 rev = ctx.rev()
1647 if rev in results:
1647 if rev in results:
1648 ui.status(_("found revision %s from %s\n") %
1648 ui.status(_("found revision %s from %s\n") %
1649 (rev, util.datestr(results[rev])))
1649 (rev, util.datestr(results[rev])))
1650 return str(rev)
1650 return str(rev)
1651
1651
1652 raise error.Abort(_("revision matching date not found"))
1652 raise error.Abort(_("revision matching date not found"))
1653
1653
1654 def increasingwindows(windowsize=8, sizelimit=512):
1654 def increasingwindows(windowsize=8, sizelimit=512):
1655 while True:
1655 while True:
1656 yield windowsize
1656 yield windowsize
1657 if windowsize < sizelimit:
1657 if windowsize < sizelimit:
1658 windowsize *= 2
1658 windowsize *= 2
1659
1659
1660 class FileWalkError(Exception):
1660 class FileWalkError(Exception):
1661 pass
1661 pass
1662
1662
1663 def walkfilerevs(repo, match, follow, revs, fncache):
1663 def walkfilerevs(repo, match, follow, revs, fncache):
1664 '''Walks the file history for the matched files.
1664 '''Walks the file history for the matched files.
1665
1665
1666 Returns the changeset revs that are involved in the file history.
1666 Returns the changeset revs that are involved in the file history.
1667
1667
1668 Throws FileWalkError if the file history can't be walked using
1668 Throws FileWalkError if the file history can't be walked using
1669 filelogs alone.
1669 filelogs alone.
1670 '''
1670 '''
1671 wanted = set()
1671 wanted = set()
1672 copies = []
1672 copies = []
1673 minrev, maxrev = min(revs), max(revs)
1673 minrev, maxrev = min(revs), max(revs)
1674 def filerevgen(filelog, last):
1674 def filerevgen(filelog, last):
1675 """
1675 """
1676 Only files, no patterns. Check the history of each file.
1676 Only files, no patterns. Check the history of each file.
1677
1677
1678 Examines filelog entries within minrev, maxrev linkrev range
1678 Examines filelog entries within minrev, maxrev linkrev range
1679 Returns an iterator yielding (linkrev, parentlinkrevs, copied)
1679 Returns an iterator yielding (linkrev, parentlinkrevs, copied)
1680 tuples in backwards order
1680 tuples in backwards order
1681 """
1681 """
1682 cl_count = len(repo)
1682 cl_count = len(repo)
1683 revs = []
1683 revs = []
1684 for j in xrange(0, last + 1):
1684 for j in xrange(0, last + 1):
1685 linkrev = filelog.linkrev(j)
1685 linkrev = filelog.linkrev(j)
1686 if linkrev < minrev:
1686 if linkrev < minrev:
1687 continue
1687 continue
1688 # only yield rev for which we have the changelog, it can
1688 # only yield rev for which we have the changelog, it can
1689 # happen while doing "hg log" during a pull or commit
1689 # happen while doing "hg log" during a pull or commit
1690 if linkrev >= cl_count:
1690 if linkrev >= cl_count:
1691 break
1691 break
1692
1692
1693 parentlinkrevs = []
1693 parentlinkrevs = []
1694 for p in filelog.parentrevs(j):
1694 for p in filelog.parentrevs(j):
1695 if p != nullrev:
1695 if p != nullrev:
1696 parentlinkrevs.append(filelog.linkrev(p))
1696 parentlinkrevs.append(filelog.linkrev(p))
1697 n = filelog.node(j)
1697 n = filelog.node(j)
1698 revs.append((linkrev, parentlinkrevs,
1698 revs.append((linkrev, parentlinkrevs,
1699 follow and filelog.renamed(n)))
1699 follow and filelog.renamed(n)))
1700
1700
1701 return reversed(revs)
1701 return reversed(revs)
1702 def iterfiles():
1702 def iterfiles():
1703 pctx = repo['.']
1703 pctx = repo['.']
1704 for filename in match.files():
1704 for filename in match.files():
1705 if follow:
1705 if follow:
1706 if filename not in pctx:
1706 if filename not in pctx:
1707 raise error.Abort(_('cannot follow file not in parent '
1707 raise error.Abort(_('cannot follow file not in parent '
1708 'revision: "%s"') % filename)
1708 'revision: "%s"') % filename)
1709 yield filename, pctx[filename].filenode()
1709 yield filename, pctx[filename].filenode()
1710 else:
1710 else:
1711 yield filename, None
1711 yield filename, None
1712 for filename_node in copies:
1712 for filename_node in copies:
1713 yield filename_node
1713 yield filename_node
1714
1714
1715 for file_, node in iterfiles():
1715 for file_, node in iterfiles():
1716 filelog = repo.file(file_)
1716 filelog = repo.file(file_)
1717 if not len(filelog):
1717 if not len(filelog):
1718 if node is None:
1718 if node is None:
1719 # A zero count may be a directory or deleted file, so
1719 # A zero count may be a directory or deleted file, so
1720 # try to find matching entries on the slow path.
1720 # try to find matching entries on the slow path.
1721 if follow:
1721 if follow:
1722 raise error.Abort(
1722 raise error.Abort(
1723 _('cannot follow nonexistent file: "%s"') % file_)
1723 _('cannot follow nonexistent file: "%s"') % file_)
1724 raise FileWalkError("Cannot walk via filelog")
1724 raise FileWalkError("Cannot walk via filelog")
1725 else:
1725 else:
1726 continue
1726 continue
1727
1727
1728 if node is None:
1728 if node is None:
1729 last = len(filelog) - 1
1729 last = len(filelog) - 1
1730 else:
1730 else:
1731 last = filelog.rev(node)
1731 last = filelog.rev(node)
1732
1732
1733 # keep track of all ancestors of the file
1733 # keep track of all ancestors of the file
1734 ancestors = set([filelog.linkrev(last)])
1734 ancestors = set([filelog.linkrev(last)])
1735
1735
1736 # iterate from latest to oldest revision
1736 # iterate from latest to oldest revision
1737 for rev, flparentlinkrevs, copied in filerevgen(filelog, last):
1737 for rev, flparentlinkrevs, copied in filerevgen(filelog, last):
1738 if not follow:
1738 if not follow:
1739 if rev > maxrev:
1739 if rev > maxrev:
1740 continue
1740 continue
1741 else:
1741 else:
1742 # Note that last might not be the first interesting
1742 # Note that last might not be the first interesting
1743 # rev to us:
1743 # rev to us:
1744 # if the file has been changed after maxrev, we'll
1744 # if the file has been changed after maxrev, we'll
1745 # have linkrev(last) > maxrev, and we still need
1745 # have linkrev(last) > maxrev, and we still need
1746 # to explore the file graph
1746 # to explore the file graph
1747 if rev not in ancestors:
1747 if rev not in ancestors:
1748 continue
1748 continue
1749 # XXX insert 1327 fix here
1749 # XXX insert 1327 fix here
1750 if flparentlinkrevs:
1750 if flparentlinkrevs:
1751 ancestors.update(flparentlinkrevs)
1751 ancestors.update(flparentlinkrevs)
1752
1752
1753 fncache.setdefault(rev, []).append(file_)
1753 fncache.setdefault(rev, []).append(file_)
1754 wanted.add(rev)
1754 wanted.add(rev)
1755 if copied:
1755 if copied:
1756 copies.append(copied)
1756 copies.append(copied)
1757
1757
1758 return wanted
1758 return wanted
1759
1759
1760 class _followfilter(object):
1760 class _followfilter(object):
1761 def __init__(self, repo, onlyfirst=False):
1761 def __init__(self, repo, onlyfirst=False):
1762 self.repo = repo
1762 self.repo = repo
1763 self.startrev = nullrev
1763 self.startrev = nullrev
1764 self.roots = set()
1764 self.roots = set()
1765 self.onlyfirst = onlyfirst
1765 self.onlyfirst = onlyfirst
1766
1766
1767 def match(self, rev):
1767 def match(self, rev):
1768 def realparents(rev):
1768 def realparents(rev):
1769 if self.onlyfirst:
1769 if self.onlyfirst:
1770 return self.repo.changelog.parentrevs(rev)[0:1]
1770 return self.repo.changelog.parentrevs(rev)[0:1]
1771 else:
1771 else:
1772 return filter(lambda x: x != nullrev,
1772 return filter(lambda x: x != nullrev,
1773 self.repo.changelog.parentrevs(rev))
1773 self.repo.changelog.parentrevs(rev))
1774
1774
1775 if self.startrev == nullrev:
1775 if self.startrev == nullrev:
1776 self.startrev = rev
1776 self.startrev = rev
1777 return True
1777 return True
1778
1778
1779 if rev > self.startrev:
1779 if rev > self.startrev:
1780 # forward: all descendants
1780 # forward: all descendants
1781 if not self.roots:
1781 if not self.roots:
1782 self.roots.add(self.startrev)
1782 self.roots.add(self.startrev)
1783 for parent in realparents(rev):
1783 for parent in realparents(rev):
1784 if parent in self.roots:
1784 if parent in self.roots:
1785 self.roots.add(rev)
1785 self.roots.add(rev)
1786 return True
1786 return True
1787 else:
1787 else:
1788 # backwards: all parents
1788 # backwards: all parents
1789 if not self.roots:
1789 if not self.roots:
1790 self.roots.update(realparents(self.startrev))
1790 self.roots.update(realparents(self.startrev))
1791 if rev in self.roots:
1791 if rev in self.roots:
1792 self.roots.remove(rev)
1792 self.roots.remove(rev)
1793 self.roots.update(realparents(rev))
1793 self.roots.update(realparents(rev))
1794 return True
1794 return True
1795
1795
1796 return False
1796 return False
1797
1797
1798 def walkchangerevs(repo, match, opts, prepare):
1798 def walkchangerevs(repo, match, opts, prepare):
1799 '''Iterate over files and the revs in which they changed.
1799 '''Iterate over files and the revs in which they changed.
1800
1800
1801 Callers most commonly need to iterate backwards over the history
1801 Callers most commonly need to iterate backwards over the history
1802 in which they are interested. Doing so has awful (quadratic-looking)
1802 in which they are interested. Doing so has awful (quadratic-looking)
1803 performance, so we use iterators in a "windowed" way.
1803 performance, so we use iterators in a "windowed" way.
1804
1804
1805 We walk a window of revisions in the desired order. Within the
1805 We walk a window of revisions in the desired order. Within the
1806 window, we first walk forwards to gather data, then in the desired
1806 window, we first walk forwards to gather data, then in the desired
1807 order (usually backwards) to display it.
1807 order (usually backwards) to display it.
1808
1808
1809 This function returns an iterator yielding contexts. Before
1809 This function returns an iterator yielding contexts. Before
1810 yielding each context, the iterator will first call the prepare
1810 yielding each context, the iterator will first call the prepare
1811 function on each context in the window in forward order.'''
1811 function on each context in the window in forward order.'''
1812
1812
1813 follow = opts.get('follow') or opts.get('follow_first')
1813 follow = opts.get('follow') or opts.get('follow_first')
1814 revs = _logrevs(repo, opts)
1814 revs = _logrevs(repo, opts)
1815 if not revs:
1815 if not revs:
1816 return []
1816 return []
1817 wanted = set()
1817 wanted = set()
1818 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
1818 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
1819 opts.get('removed'))
1819 opts.get('removed'))
1820 fncache = {}
1820 fncache = {}
1821 change = repo.changectx
1821 change = repo.changectx
1822
1822
1823 # First step is to fill wanted, the set of revisions that we want to yield.
1823 # First step is to fill wanted, the set of revisions that we want to yield.
1824 # When it does not induce extra cost, we also fill fncache for revisions in
1824 # When it does not induce extra cost, we also fill fncache for revisions in
1825 # wanted: a cache of filenames that were changed (ctx.files()) and that
1825 # wanted: a cache of filenames that were changed (ctx.files()) and that
1826 # match the file filtering conditions.
1826 # match the file filtering conditions.
1827
1827
1828 if match.always():
1828 if match.always():
1829 # No files, no patterns. Display all revs.
1829 # No files, no patterns. Display all revs.
1830 wanted = revs
1830 wanted = revs
1831 elif not slowpath:
1831 elif not slowpath:
1832 # We only have to read through the filelog to find wanted revisions
1832 # We only have to read through the filelog to find wanted revisions
1833
1833
1834 try:
1834 try:
1835 wanted = walkfilerevs(repo, match, follow, revs, fncache)
1835 wanted = walkfilerevs(repo, match, follow, revs, fncache)
1836 except FileWalkError:
1836 except FileWalkError:
1837 slowpath = True
1837 slowpath = True
1838
1838
1839 # We decided to fall back to the slowpath because at least one
1839 # We decided to fall back to the slowpath because at least one
1840 # of the paths was not a file. Check to see if at least one of them
1840 # of the paths was not a file. Check to see if at least one of them
1841 # existed in history, otherwise simply return
1841 # existed in history, otherwise simply return
1842 for path in match.files():
1842 for path in match.files():
1843 if path == '.' or path in repo.store:
1843 if path == '.' or path in repo.store:
1844 break
1844 break
1845 else:
1845 else:
1846 return []
1846 return []
1847
1847
1848 if slowpath:
1848 if slowpath:
1849 # We have to read the changelog to match filenames against
1849 # We have to read the changelog to match filenames against
1850 # changed files
1850 # changed files
1851
1851
1852 if follow:
1852 if follow:
1853 raise error.Abort(_('can only follow copies/renames for explicit '
1853 raise error.Abort(_('can only follow copies/renames for explicit '
1854 'filenames'))
1854 'filenames'))
1855
1855
1856 # The slow path checks files modified in every changeset.
1856 # The slow path checks files modified in every changeset.
1857 # This is really slow on large repos, so compute the set lazily.
1857 # This is really slow on large repos, so compute the set lazily.
1858 class lazywantedset(object):
1858 class lazywantedset(object):
1859 def __init__(self):
1859 def __init__(self):
1860 self.set = set()
1860 self.set = set()
1861 self.revs = set(revs)
1861 self.revs = set(revs)
1862
1862
1863 # No need to worry about locality here because it will be accessed
1863 # No need to worry about locality here because it will be accessed
1864 # in the same order as the increasing window below.
1864 # in the same order as the increasing window below.
1865 def __contains__(self, value):
1865 def __contains__(self, value):
1866 if value in self.set:
1866 if value in self.set:
1867 return True
1867 return True
1868 elif not value in self.revs:
1868 elif not value in self.revs:
1869 return False
1869 return False
1870 else:
1870 else:
1871 self.revs.discard(value)
1871 self.revs.discard(value)
1872 ctx = change(value)
1872 ctx = change(value)
1873 matches = filter(match, ctx.files())
1873 matches = filter(match, ctx.files())
1874 if matches:
1874 if matches:
1875 fncache[value] = matches
1875 fncache[value] = matches
1876 self.set.add(value)
1876 self.set.add(value)
1877 return True
1877 return True
1878 return False
1878 return False
1879
1879
1880 def discard(self, value):
1880 def discard(self, value):
1881 self.revs.discard(value)
1881 self.revs.discard(value)
1882 self.set.discard(value)
1882 self.set.discard(value)
1883
1883
1884 wanted = lazywantedset()
1884 wanted = lazywantedset()
1885
1885
1886 # it might be worthwhile to do this in the iterator if the rev range
1886 # it might be worthwhile to do this in the iterator if the rev range
1887 # is descending and the prune args are all within that range
1887 # is descending and the prune args are all within that range
1888 for rev in opts.get('prune', ()):
1888 for rev in opts.get('prune', ()):
1889 rev = repo[rev].rev()
1889 rev = repo[rev].rev()
1890 ff = _followfilter(repo)
1890 ff = _followfilter(repo)
1891 stop = min(revs[0], revs[-1])
1891 stop = min(revs[0], revs[-1])
1892 for x in xrange(rev, stop - 1, -1):
1892 for x in xrange(rev, stop - 1, -1):
1893 if ff.match(x):
1893 if ff.match(x):
1894 wanted = wanted - [x]
1894 wanted = wanted - [x]
1895
1895
1896 # Now that wanted is correctly initialized, we can iterate over the
1896 # Now that wanted is correctly initialized, we can iterate over the
1897 # revision range, yielding only revisions in wanted.
1897 # revision range, yielding only revisions in wanted.
1898 def iterate():
1898 def iterate():
1899 if follow and match.always():
1899 if follow and match.always():
1900 ff = _followfilter(repo, onlyfirst=opts.get('follow_first'))
1900 ff = _followfilter(repo, onlyfirst=opts.get('follow_first'))
1901 def want(rev):
1901 def want(rev):
1902 return ff.match(rev) and rev in wanted
1902 return ff.match(rev) and rev in wanted
1903 else:
1903 else:
1904 def want(rev):
1904 def want(rev):
1905 return rev in wanted
1905 return rev in wanted
1906
1906
1907 it = iter(revs)
1907 it = iter(revs)
1908 stopiteration = False
1908 stopiteration = False
1909 for windowsize in increasingwindows():
1909 for windowsize in increasingwindows():
1910 nrevs = []
1910 nrevs = []
1911 for i in xrange(windowsize):
1911 for i in xrange(windowsize):
1912 rev = next(it, None)
1912 rev = next(it, None)
1913 if rev is None:
1913 if rev is None:
1914 stopiteration = True
1914 stopiteration = True
1915 break
1915 break
1916 elif want(rev):
1916 elif want(rev):
1917 nrevs.append(rev)
1917 nrevs.append(rev)
1918 for rev in sorted(nrevs):
1918 for rev in sorted(nrevs):
1919 fns = fncache.get(rev)
1919 fns = fncache.get(rev)
1920 ctx = change(rev)
1920 ctx = change(rev)
1921 if not fns:
1921 if not fns:
1922 def fns_generator():
1922 def fns_generator():
1923 for f in ctx.files():
1923 for f in ctx.files():
1924 if match(f):
1924 if match(f):
1925 yield f
1925 yield f
1926 fns = fns_generator()
1926 fns = fns_generator()
1927 prepare(ctx, fns)
1927 prepare(ctx, fns)
1928 for rev in nrevs:
1928 for rev in nrevs:
1929 yield change(rev)
1929 yield change(rev)
1930
1930
1931 if stopiteration:
1931 if stopiteration:
1932 break
1932 break
1933
1933
1934 return iterate()
1934 return iterate()
1935
1935
1936 def _makefollowlogfilematcher(repo, files, followfirst):
1936 def _makefollowlogfilematcher(repo, files, followfirst):
1937 # When displaying a revision with --patch --follow FILE, we have
1937 # When displaying a revision with --patch --follow FILE, we have
1938 # to know which file of the revision must be diffed. With
1938 # to know which file of the revision must be diffed. With
1939 # --follow, we want the names of the ancestors of FILE in the
1939 # --follow, we want the names of the ancestors of FILE in the
1940 # revision, stored in "fcache". "fcache" is populated by
1940 # revision, stored in "fcache". "fcache" is populated by
1941 # reproducing the graph traversal already done by --follow revset
1941 # reproducing the graph traversal already done by --follow revset
1942 # and relating linkrevs to file names (which is not "correct" but
1942 # and relating linkrevs to file names (which is not "correct" but
1943 # good enough).
1943 # good enough).
1944 fcache = {}
1944 fcache = {}
1945 fcacheready = [False]
1945 fcacheready = [False]
1946 pctx = repo['.']
1946 pctx = repo['.']
1947
1947
1948 def populate():
1948 def populate():
1949 for fn in files:
1949 for fn in files:
1950 for i in ((pctx[fn],), pctx[fn].ancestors(followfirst=followfirst)):
1950 for i in ((pctx[fn],), pctx[fn].ancestors(followfirst=followfirst)):
1951 for c in i:
1951 for c in i:
1952 fcache.setdefault(c.linkrev(), set()).add(c.path())
1952 fcache.setdefault(c.linkrev(), set()).add(c.path())
1953
1953
1954 def filematcher(rev):
1954 def filematcher(rev):
1955 if not fcacheready[0]:
1955 if not fcacheready[0]:
1956 # Lazy initialization
1956 # Lazy initialization
1957 fcacheready[0] = True
1957 fcacheready[0] = True
1958 populate()
1958 populate()
1959 return scmutil.matchfiles(repo, fcache.get(rev, []))
1959 return scmutil.matchfiles(repo, fcache.get(rev, []))
1960
1960
1961 return filematcher
1961 return filematcher
1962
1962
1963 def _makenofollowlogfilematcher(repo, pats, opts):
1963 def _makenofollowlogfilematcher(repo, pats, opts):
1964 '''hook for extensions to override the filematcher for non-follow cases'''
1964 '''hook for extensions to override the filematcher for non-follow cases'''
1965 return None
1965 return None
1966
1966
1967 def _makelogrevset(repo, pats, opts, revs):
1967 def _makelogrevset(repo, pats, opts, revs):
1968 """Return (expr, filematcher) where expr is a revset string built
1968 """Return (expr, filematcher) where expr is a revset string built
1969 from log options and file patterns or None. If --stat or --patch
1969 from log options and file patterns or None. If --stat or --patch
1970 are not passed filematcher is None. Otherwise it is a callable
1970 are not passed filematcher is None. Otherwise it is a callable
1971 taking a revision number and returning a match objects filtering
1971 taking a revision number and returning a match objects filtering
1972 the files to be detailed when displaying the revision.
1972 the files to be detailed when displaying the revision.
1973 """
1973 """
1974 opt2revset = {
1974 opt2revset = {
1975 'no_merges': ('not merge()', None),
1975 'no_merges': ('not merge()', None),
1976 'only_merges': ('merge()', None),
1976 'only_merges': ('merge()', None),
1977 '_ancestors': ('ancestors(%(val)s)', None),
1977 '_ancestors': ('ancestors(%(val)s)', None),
1978 '_fancestors': ('_firstancestors(%(val)s)', None),
1978 '_fancestors': ('_firstancestors(%(val)s)', None),
1979 '_descendants': ('descendants(%(val)s)', None),
1979 '_descendants': ('descendants(%(val)s)', None),
1980 '_fdescendants': ('_firstdescendants(%(val)s)', None),
1980 '_fdescendants': ('_firstdescendants(%(val)s)', None),
1981 '_matchfiles': ('_matchfiles(%(val)s)', None),
1981 '_matchfiles': ('_matchfiles(%(val)s)', None),
1982 'date': ('date(%(val)r)', None),
1982 'date': ('date(%(val)r)', None),
1983 'branch': ('branch(%(val)r)', ' or '),
1983 'branch': ('branch(%(val)r)', ' or '),
1984 '_patslog': ('filelog(%(val)r)', ' or '),
1984 '_patslog': ('filelog(%(val)r)', ' or '),
1985 '_patsfollow': ('follow(%(val)r)', ' or '),
1985 '_patsfollow': ('follow(%(val)r)', ' or '),
1986 '_patsfollowfirst': ('_followfirst(%(val)r)', ' or '),
1986 '_patsfollowfirst': ('_followfirst(%(val)r)', ' or '),
1987 'keyword': ('keyword(%(val)r)', ' or '),
1987 'keyword': ('keyword(%(val)r)', ' or '),
1988 'prune': ('not (%(val)r or ancestors(%(val)r))', ' and '),
1988 'prune': ('not (%(val)r or ancestors(%(val)r))', ' and '),
1989 'user': ('user(%(val)r)', ' or '),
1989 'user': ('user(%(val)r)', ' or '),
1990 }
1990 }
1991
1991
1992 opts = dict(opts)
1992 opts = dict(opts)
1993 # follow or not follow?
1993 # follow or not follow?
1994 follow = opts.get('follow') or opts.get('follow_first')
1994 follow = opts.get('follow') or opts.get('follow_first')
1995 if opts.get('follow_first'):
1995 if opts.get('follow_first'):
1996 followfirst = 1
1996 followfirst = 1
1997 else:
1997 else:
1998 followfirst = 0
1998 followfirst = 0
1999 # --follow with FILE behavior depends on revs...
1999 # --follow with FILE behavior depends on revs...
2000 it = iter(revs)
2000 it = iter(revs)
2001 startrev = next(it)
2001 startrev = next(it)
2002 followdescendants = startrev < next(it, startrev)
2002 followdescendants = startrev < next(it, startrev)
2003
2003
2004 # branch and only_branch are really aliases and must be handled at
2004 # branch and only_branch are really aliases and must be handled at
2005 # the same time
2005 # the same time
2006 opts['branch'] = opts.get('branch', []) + opts.get('only_branch', [])
2006 opts['branch'] = opts.get('branch', []) + opts.get('only_branch', [])
2007 opts['branch'] = [repo.lookupbranch(b) for b in opts['branch']]
2007 opts['branch'] = [repo.lookupbranch(b) for b in opts['branch']]
2008 # pats/include/exclude are passed to match.match() directly in
2008 # pats/include/exclude are passed to match.match() directly in
2009 # _matchfiles() revset but walkchangerevs() builds its matcher with
2009 # _matchfiles() revset but walkchangerevs() builds its matcher with
2010 # scmutil.match(). The difference is input pats are globbed on
2010 # scmutil.match(). The difference is input pats are globbed on
2011 # platforms without shell expansion (windows).
2011 # platforms without shell expansion (windows).
2012 wctx = repo[None]
2012 wctx = repo[None]
2013 match, pats = scmutil.matchandpats(wctx, pats, opts)
2013 match, pats = scmutil.matchandpats(wctx, pats, opts)
2014 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
2014 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
2015 opts.get('removed'))
2015 opts.get('removed'))
2016 if not slowpath:
2016 if not slowpath:
2017 for f in match.files():
2017 for f in match.files():
2018 if follow and f not in wctx:
2018 if follow and f not in wctx:
2019 # If the file exists, it may be a directory, so let it
2019 # If the file exists, it may be a directory, so let it
2020 # take the slow path.
2020 # take the slow path.
2021 if os.path.exists(repo.wjoin(f)):
2021 if os.path.exists(repo.wjoin(f)):
2022 slowpath = True
2022 slowpath = True
2023 continue
2023 continue
2024 else:
2024 else:
2025 raise error.Abort(_('cannot follow file not in parent '
2025 raise error.Abort(_('cannot follow file not in parent '
2026 'revision: "%s"') % f)
2026 'revision: "%s"') % f)
2027 filelog = repo.file(f)
2027 filelog = repo.file(f)
2028 if not filelog:
2028 if not filelog:
2029 # A zero count may be a directory or deleted file, so
2029 # A zero count may be a directory or deleted file, so
2030 # try to find matching entries on the slow path.
2030 # try to find matching entries on the slow path.
2031 if follow:
2031 if follow:
2032 raise error.Abort(
2032 raise error.Abort(
2033 _('cannot follow nonexistent file: "%s"') % f)
2033 _('cannot follow nonexistent file: "%s"') % f)
2034 slowpath = True
2034 slowpath = True
2035
2035
2036 # We decided to fall back to the slowpath because at least one
2036 # We decided to fall back to the slowpath because at least one
2037 # of the paths was not a file. Check to see if at least one of them
2037 # of the paths was not a file. Check to see if at least one of them
2038 # existed in history - in that case, we'll continue down the
2038 # existed in history - in that case, we'll continue down the
2039 # slowpath; otherwise, we can turn off the slowpath
2039 # slowpath; otherwise, we can turn off the slowpath
2040 if slowpath:
2040 if slowpath:
2041 for path in match.files():
2041 for path in match.files():
2042 if path == '.' or path in repo.store:
2042 if path == '.' or path in repo.store:
2043 break
2043 break
2044 else:
2044 else:
2045 slowpath = False
2045 slowpath = False
2046
2046
2047 fpats = ('_patsfollow', '_patsfollowfirst')
2047 fpats = ('_patsfollow', '_patsfollowfirst')
2048 fnopats = (('_ancestors', '_fancestors'),
2048 fnopats = (('_ancestors', '_fancestors'),
2049 ('_descendants', '_fdescendants'))
2049 ('_descendants', '_fdescendants'))
2050 if slowpath:
2050 if slowpath:
2051 # See walkchangerevs() slow path.
2051 # See walkchangerevs() slow path.
2052 #
2052 #
2053 # pats/include/exclude cannot be represented as separate
2053 # pats/include/exclude cannot be represented as separate
2054 # revset expressions as their filtering logic applies at file
2054 # revset expressions as their filtering logic applies at file
2055 # level. For instance "-I a -X a" matches a revision touching
2055 # level. For instance "-I a -X a" matches a revision touching
2056 # "a" and "b" while "file(a) and not file(b)" does
2056 # "a" and "b" while "file(a) and not file(b)" does
2057 # not. Besides, filesets are evaluated against the working
2057 # not. Besides, filesets are evaluated against the working
2058 # directory.
2058 # directory.
2059 matchargs = ['r:', 'd:relpath']
2059 matchargs = ['r:', 'd:relpath']
2060 for p in pats:
2060 for p in pats:
2061 matchargs.append('p:' + p)
2061 matchargs.append('p:' + p)
2062 for p in opts.get('include', []):
2062 for p in opts.get('include', []):
2063 matchargs.append('i:' + p)
2063 matchargs.append('i:' + p)
2064 for p in opts.get('exclude', []):
2064 for p in opts.get('exclude', []):
2065 matchargs.append('x:' + p)
2065 matchargs.append('x:' + p)
2066 matchargs = ','.join(('%r' % p) for p in matchargs)
2066 matchargs = ','.join(('%r' % p) for p in matchargs)
2067 opts['_matchfiles'] = matchargs
2067 opts['_matchfiles'] = matchargs
2068 if follow:
2068 if follow:
2069 opts[fnopats[0][followfirst]] = '.'
2069 opts[fnopats[0][followfirst]] = '.'
2070 else:
2070 else:
2071 if follow:
2071 if follow:
2072 if pats:
2072 if pats:
2073 # follow() revset interprets its file argument as a
2073 # follow() revset interprets its file argument as a
2074 # manifest entry, so use match.files(), not pats.
2074 # manifest entry, so use match.files(), not pats.
2075 opts[fpats[followfirst]] = list(match.files())
2075 opts[fpats[followfirst]] = list(match.files())
2076 else:
2076 else:
2077 op = fnopats[followdescendants][followfirst]
2077 op = fnopats[followdescendants][followfirst]
2078 opts[op] = 'rev(%d)' % startrev
2078 opts[op] = 'rev(%d)' % startrev
2079 else:
2079 else:
2080 opts['_patslog'] = list(pats)
2080 opts['_patslog'] = list(pats)
2081
2081
2082 filematcher = None
2082 filematcher = None
2083 if opts.get('patch') or opts.get('stat'):
2083 if opts.get('patch') or opts.get('stat'):
2084 # When following files, track renames via a special matcher.
2084 # When following files, track renames via a special matcher.
2085 # If we're forced to take the slowpath it means we're following
2085 # If we're forced to take the slowpath it means we're following
2086 # at least one pattern/directory, so don't bother with rename tracking.
2086 # at least one pattern/directory, so don't bother with rename tracking.
2087 if follow and not match.always() and not slowpath:
2087 if follow and not match.always() and not slowpath:
2088 # _makefollowlogfilematcher expects its files argument to be
2088 # _makefollowlogfilematcher expects its files argument to be
2089 # relative to the repo root, so use match.files(), not pats.
2089 # relative to the repo root, so use match.files(), not pats.
2090 filematcher = _makefollowlogfilematcher(repo, match.files(),
2090 filematcher = _makefollowlogfilematcher(repo, match.files(),
2091 followfirst)
2091 followfirst)
2092 else:
2092 else:
2093 filematcher = _makenofollowlogfilematcher(repo, pats, opts)
2093 filematcher = _makenofollowlogfilematcher(repo, pats, opts)
2094 if filematcher is None:
2094 if filematcher is None:
2095 filematcher = lambda rev: match
2095 filematcher = lambda rev: match
2096
2096
2097 expr = []
2097 expr = []
2098 for op, val in sorted(opts.iteritems()):
2098 for op, val in sorted(opts.iteritems()):
2099 if not val:
2099 if not val:
2100 continue
2100 continue
2101 if op not in opt2revset:
2101 if op not in opt2revset:
2102 continue
2102 continue
2103 revop, andor = opt2revset[op]
2103 revop, andor = opt2revset[op]
2104 if '%(val)' not in revop:
2104 if '%(val)' not in revop:
2105 expr.append(revop)
2105 expr.append(revop)
2106 else:
2106 else:
2107 if not isinstance(val, list):
2107 if not isinstance(val, list):
2108 e = revop % {'val': val}
2108 e = revop % {'val': val}
2109 else:
2109 else:
2110 e = '(' + andor.join((revop % {'val': v}) for v in val) + ')'
2110 e = '(' + andor.join((revop % {'val': v}) for v in val) + ')'
2111 expr.append(e)
2111 expr.append(e)
2112
2112
2113 if expr:
2113 if expr:
2114 expr = '(' + ' and '.join(expr) + ')'
2114 expr = '(' + ' and '.join(expr) + ')'
2115 else:
2115 else:
2116 expr = None
2116 expr = None
2117 return expr, filematcher
2117 return expr, filematcher
2118
2118
2119 def _logrevs(repo, opts):
2119 def _logrevs(repo, opts):
2120 # Default --rev value depends on --follow but --follow behavior
2120 # Default --rev value depends on --follow but --follow behavior
2121 # depends on revisions resolved from --rev...
2121 # depends on revisions resolved from --rev...
2122 follow = opts.get('follow') or opts.get('follow_first')
2122 follow = opts.get('follow') or opts.get('follow_first')
2123 if opts.get('rev'):
2123 if opts.get('rev'):
2124 revs = scmutil.revrange(repo, opts['rev'])
2124 revs = scmutil.revrange(repo, opts['rev'])
2125 elif follow and repo.dirstate.p1() == nullid:
2125 elif follow and repo.dirstate.p1() == nullid:
2126 revs = revset.baseset()
2126 revs = revset.baseset()
2127 elif follow:
2127 elif follow:
2128 revs = repo.revs('reverse(:.)')
2128 revs = repo.revs('reverse(:.)')
2129 else:
2129 else:
2130 revs = revset.spanset(repo)
2130 revs = revset.spanset(repo)
2131 revs.reverse()
2131 revs.reverse()
2132 return revs
2132 return revs
2133
2133
2134 def getgraphlogrevs(repo, pats, opts):
2134 def getgraphlogrevs(repo, pats, opts):
2135 """Return (revs, expr, filematcher) where revs is an iterable of
2135 """Return (revs, expr, filematcher) where revs is an iterable of
2136 revision numbers, expr is a revset string built from log options
2136 revision numbers, expr is a revset string built from log options
2137 and file patterns or None, and used to filter 'revs'. If --stat or
2137 and file patterns or None, and used to filter 'revs'. If --stat or
2138 --patch are not passed filematcher is None. Otherwise it is a
2138 --patch are not passed filematcher is None. Otherwise it is a
2139 callable taking a revision number and returning a match objects
2139 callable taking a revision number and returning a match objects
2140 filtering the files to be detailed when displaying the revision.
2140 filtering the files to be detailed when displaying the revision.
2141 """
2141 """
2142 limit = loglimit(opts)
2142 limit = loglimit(opts)
2143 revs = _logrevs(repo, opts)
2143 revs = _logrevs(repo, opts)
2144 if not revs:
2144 if not revs:
2145 return revset.baseset(), None, None
2145 return revset.baseset(), None, None
2146 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2146 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2147 if opts.get('rev'):
2147 if opts.get('rev'):
2148 # User-specified revs might be unsorted, but don't sort before
2148 # User-specified revs might be unsorted, but don't sort before
2149 # _makelogrevset because it might depend on the order of revs
2149 # _makelogrevset because it might depend on the order of revs
2150 revs.sort(reverse=True)
2150 revs.sort(reverse=True)
2151 if expr:
2151 if expr:
2152 # Revset matchers often operate faster on revisions in changelog
2152 # Revset matchers often operate faster on revisions in changelog
2153 # order, because most filters deal with the changelog.
2153 # order, because most filters deal with the changelog.
2154 revs.reverse()
2154 revs.reverse()
2155 matcher = revset.match(repo.ui, expr)
2155 matcher = revset.match(repo.ui, expr)
2156 # Revset matches can reorder revisions. "A or B" typically returns
2156 # Revset matches can reorder revisions. "A or B" typically returns
2157 # returns the revision matching A then the revision matching B. Sort
2157 # returns the revision matching A then the revision matching B. Sort
2158 # again to fix that.
2158 # again to fix that.
2159 revs = matcher(repo, revs)
2159 revs = matcher(repo, revs)
2160 revs.sort(reverse=True)
2160 revs.sort(reverse=True)
2161 if limit is not None:
2161 if limit is not None:
2162 limitedrevs = []
2162 limitedrevs = []
2163 for idx, rev in enumerate(revs):
2163 for idx, rev in enumerate(revs):
2164 if idx >= limit:
2164 if idx >= limit:
2165 break
2165 break
2166 limitedrevs.append(rev)
2166 limitedrevs.append(rev)
2167 revs = revset.baseset(limitedrevs)
2167 revs = revset.baseset(limitedrevs)
2168
2168
2169 return revs, expr, filematcher
2169 return revs, expr, filematcher
2170
2170
2171 def getlogrevs(repo, pats, opts):
2171 def getlogrevs(repo, pats, opts):
2172 """Return (revs, expr, filematcher) where revs is an iterable of
2172 """Return (revs, expr, filematcher) where revs is an iterable of
2173 revision numbers, expr is a revset string built from log options
2173 revision numbers, expr is a revset string built from log options
2174 and file patterns or None, and used to filter 'revs'. If --stat or
2174 and file patterns or None, and used to filter 'revs'. If --stat or
2175 --patch are not passed filematcher is None. Otherwise it is a
2175 --patch are not passed filematcher is None. Otherwise it is a
2176 callable taking a revision number and returning a match objects
2176 callable taking a revision number and returning a match objects
2177 filtering the files to be detailed when displaying the revision.
2177 filtering the files to be detailed when displaying the revision.
2178 """
2178 """
2179 limit = loglimit(opts)
2179 limit = loglimit(opts)
2180 revs = _logrevs(repo, opts)
2180 revs = _logrevs(repo, opts)
2181 if not revs:
2181 if not revs:
2182 return revset.baseset([]), None, None
2182 return revset.baseset([]), None, None
2183 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2183 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2184 if expr:
2184 if expr:
2185 # Revset matchers often operate faster on revisions in changelog
2185 # Revset matchers often operate faster on revisions in changelog
2186 # order, because most filters deal with the changelog.
2186 # order, because most filters deal with the changelog.
2187 if not opts.get('rev'):
2187 if not opts.get('rev'):
2188 revs.reverse()
2188 revs.reverse()
2189 matcher = revset.match(repo.ui, expr)
2189 matcher = revset.match(repo.ui, expr)
2190 # Revset matches can reorder revisions. "A or B" typically returns
2190 # Revset matches can reorder revisions. "A or B" typically returns
2191 # returns the revision matching A then the revision matching B. Sort
2191 # returns the revision matching A then the revision matching B. Sort
2192 # again to fix that.
2192 # again to fix that.
2193 fixopts = ['branch', 'only_branch', 'keyword', 'user']
2193 fixopts = ['branch', 'only_branch', 'keyword', 'user']
2194 oldrevs = revs
2194 oldrevs = revs
2195 revs = matcher(repo, revs)
2195 revs = matcher(repo, revs)
2196 if not opts.get('rev'):
2196 if not opts.get('rev'):
2197 revs.sort(reverse=True)
2197 revs.sort(reverse=True)
2198 elif len(pats) > 1 or any(len(opts.get(op, [])) > 1 for op in fixopts):
2198 elif len(pats) > 1 or any(len(opts.get(op, [])) > 1 for op in fixopts):
2199 # XXX "A or B" is known to change the order; fix it by filtering
2199 # XXX "A or B" is known to change the order; fix it by filtering
2200 # matched set again (issue5100)
2200 # matched set again (issue5100)
2201 revs = oldrevs & revs
2201 revs = oldrevs & revs
2202 if limit is not None:
2202 if limit is not None:
2203 limitedrevs = []
2203 limitedrevs = []
2204 for idx, r in enumerate(revs):
2204 for idx, r in enumerate(revs):
2205 if limit <= idx:
2205 if limit <= idx:
2206 break
2206 break
2207 limitedrevs.append(r)
2207 limitedrevs.append(r)
2208 revs = revset.baseset(limitedrevs)
2208 revs = revset.baseset(limitedrevs)
2209
2209
2210 return revs, expr, filematcher
2210 return revs, expr, filematcher
2211
2211
2212 def _graphnodeformatter(ui, displayer):
2212 def _graphnodeformatter(ui, displayer):
2213 spec = ui.config('ui', 'graphnodetemplate')
2213 spec = ui.config('ui', 'graphnodetemplate')
2214 if not spec:
2214 if not spec:
2215 return templatekw.showgraphnode # fast path for "{graphnode}"
2215 return templatekw.showgraphnode # fast path for "{graphnode}"
2216
2216
2217 templ = formatter.gettemplater(ui, 'graphnode', spec)
2217 templ = formatter.gettemplater(ui, 'graphnode', spec)
2218 cache = {}
2218 cache = {}
2219 if isinstance(displayer, changeset_templater):
2219 if isinstance(displayer, changeset_templater):
2220 cache = displayer.cache # reuse cache of slow templates
2220 cache = displayer.cache # reuse cache of slow templates
2221 props = templatekw.keywords.copy()
2221 props = templatekw.keywords.copy()
2222 props['templ'] = templ
2222 props['templ'] = templ
2223 props['cache'] = cache
2223 props['cache'] = cache
2224 def formatnode(repo, ctx):
2224 def formatnode(repo, ctx):
2225 props['ctx'] = ctx
2225 props['ctx'] = ctx
2226 props['repo'] = repo
2226 props['repo'] = repo
2227 props['ui'] = repo.ui
2227 props['ui'] = repo.ui
2228 props['revcache'] = {}
2228 props['revcache'] = {}
2229 return templater.stringify(templ('graphnode', **props))
2229 return templater.stringify(templ('graphnode', **props))
2230 return formatnode
2230 return formatnode
2231
2231
2232 def displaygraph(ui, repo, dag, displayer, edgefn, getrenamed=None,
2232 def displaygraph(ui, repo, dag, displayer, edgefn, getrenamed=None,
2233 filematcher=None):
2233 filematcher=None):
2234 formatnode = _graphnodeformatter(ui, displayer)
2234 formatnode = _graphnodeformatter(ui, displayer)
2235 state = graphmod.asciistate()
2235 state = graphmod.asciistate()
2236 styles = state['styles']
2236 styles = state['styles']
2237
2237
2238 # only set graph styling if HGPLAIN is not set.
2238 # only set graph styling if HGPLAIN is not set.
2239 if ui.plain('graph'):
2239 if ui.plain('graph'):
2240 # set all edge styles to |, the default pre-3.8 behaviour
2240 # set all edge styles to |, the default pre-3.8 behaviour
2241 styles.update(dict.fromkeys(styles, '|'))
2241 styles.update(dict.fromkeys(styles, '|'))
2242 else:
2242 else:
2243 edgetypes = {
2243 edgetypes = {
2244 'parent': graphmod.PARENT,
2244 'parent': graphmod.PARENT,
2245 'grandparent': graphmod.GRANDPARENT,
2245 'grandparent': graphmod.GRANDPARENT,
2246 'missing': graphmod.MISSINGPARENT
2246 'missing': graphmod.MISSINGPARENT
2247 }
2247 }
2248 for name, key in edgetypes.items():
2248 for name, key in edgetypes.items():
2249 # experimental config: experimental.graphstyle.*
2249 # experimental config: experimental.graphstyle.*
2250 styles[key] = ui.config('experimental', 'graphstyle.%s' % name,
2250 styles[key] = ui.config('experimental', 'graphstyle.%s' % name,
2251 styles[key])
2251 styles[key])
2252 if not styles[key]:
2252 if not styles[key]:
2253 styles[key] = None
2253 styles[key] = None
2254
2254
2255 # experimental config: experimental.graphshorten
2255 # experimental config: experimental.graphshorten
2256 state['graphshorten'] = ui.configbool('experimental', 'graphshorten')
2256 state['graphshorten'] = ui.configbool('experimental', 'graphshorten')
2257
2257
2258 for rev, type, ctx, parents in dag:
2258 for rev, type, ctx, parents in dag:
2259 char = formatnode(repo, ctx)
2259 char = formatnode(repo, ctx)
2260 copies = None
2260 copies = None
2261 if getrenamed and ctx.rev():
2261 if getrenamed and ctx.rev():
2262 copies = []
2262 copies = []
2263 for fn in ctx.files():
2263 for fn in ctx.files():
2264 rename = getrenamed(fn, ctx.rev())
2264 rename = getrenamed(fn, ctx.rev())
2265 if rename:
2265 if rename:
2266 copies.append((fn, rename[0]))
2266 copies.append((fn, rename[0]))
2267 revmatchfn = None
2267 revmatchfn = None
2268 if filematcher is not None:
2268 if filematcher is not None:
2269 revmatchfn = filematcher(ctx.rev())
2269 revmatchfn = filematcher(ctx.rev())
2270 displayer.show(ctx, copies=copies, matchfn=revmatchfn)
2270 displayer.show(ctx, copies=copies, matchfn=revmatchfn)
2271 lines = displayer.hunk.pop(rev).split('\n')
2271 lines = displayer.hunk.pop(rev).split('\n')
2272 if not lines[-1]:
2272 if not lines[-1]:
2273 del lines[-1]
2273 del lines[-1]
2274 displayer.flush(ctx)
2274 displayer.flush(ctx)
2275 edges = edgefn(type, char, lines, state, rev, parents)
2275 edges = edgefn(type, char, lines, state, rev, parents)
2276 for type, char, lines, coldata in edges:
2276 for type, char, lines, coldata in edges:
2277 graphmod.ascii(ui, state, type, char, lines, coldata)
2277 graphmod.ascii(ui, state, type, char, lines, coldata)
2278 displayer.close()
2278 displayer.close()
2279
2279
2280 def graphlog(ui, repo, *pats, **opts):
2280 def graphlog(ui, repo, *pats, **opts):
2281 # Parameters are identical to log command ones
2281 # Parameters are identical to log command ones
2282 revs, expr, filematcher = getgraphlogrevs(repo, pats, opts)
2282 revs, expr, filematcher = getgraphlogrevs(repo, pats, opts)
2283 revdag = graphmod.dagwalker(repo, revs)
2283 revdag = graphmod.dagwalker(repo, revs)
2284
2284
2285 getrenamed = None
2285 getrenamed = None
2286 if opts.get('copies'):
2286 if opts.get('copies'):
2287 endrev = None
2287 endrev = None
2288 if opts.get('rev'):
2288 if opts.get('rev'):
2289 endrev = scmutil.revrange(repo, opts.get('rev')).max() + 1
2289 endrev = scmutil.revrange(repo, opts.get('rev')).max() + 1
2290 getrenamed = templatekw.getrenamedfn(repo, endrev=endrev)
2290 getrenamed = templatekw.getrenamedfn(repo, endrev=endrev)
2291 displayer = show_changeset(ui, repo, opts, buffered=True)
2291 displayer = show_changeset(ui, repo, opts, buffered=True)
2292 displaygraph(ui, repo, revdag, displayer, graphmod.asciiedges, getrenamed,
2292 displaygraph(ui, repo, revdag, displayer, graphmod.asciiedges, getrenamed,
2293 filematcher)
2293 filematcher)
2294
2294
2295 def checkunsupportedgraphflags(pats, opts):
2295 def checkunsupportedgraphflags(pats, opts):
2296 for op in ["newest_first"]:
2296 for op in ["newest_first"]:
2297 if op in opts and opts[op]:
2297 if op in opts and opts[op]:
2298 raise error.Abort(_("-G/--graph option is incompatible with --%s")
2298 raise error.Abort(_("-G/--graph option is incompatible with --%s")
2299 % op.replace("_", "-"))
2299 % op.replace("_", "-"))
2300
2300
2301 def graphrevs(repo, nodes, opts):
2301 def graphrevs(repo, nodes, opts):
2302 limit = loglimit(opts)
2302 limit = loglimit(opts)
2303 nodes.reverse()
2303 nodes.reverse()
2304 if limit is not None:
2304 if limit is not None:
2305 nodes = nodes[:limit]
2305 nodes = nodes[:limit]
2306 return graphmod.nodes(repo, nodes)
2306 return graphmod.nodes(repo, nodes)
2307
2307
2308 def add(ui, repo, match, prefix, explicitonly, **opts):
2308 def add(ui, repo, match, prefix, explicitonly, **opts):
2309 join = lambda f: os.path.join(prefix, f)
2309 join = lambda f: os.path.join(prefix, f)
2310 bad = []
2310 bad = []
2311
2311
2312 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2312 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2313 names = []
2313 names = []
2314 wctx = repo[None]
2314 wctx = repo[None]
2315 cca = None
2315 cca = None
2316 abort, warn = scmutil.checkportabilityalert(ui)
2316 abort, warn = scmutil.checkportabilityalert(ui)
2317 if abort or warn:
2317 if abort or warn:
2318 cca = scmutil.casecollisionauditor(ui, abort, repo.dirstate)
2318 cca = scmutil.casecollisionauditor(ui, abort, repo.dirstate)
2319
2319
2320 badmatch = matchmod.badmatch(match, badfn)
2320 badmatch = matchmod.badmatch(match, badfn)
2321 dirstate = repo.dirstate
2321 dirstate = repo.dirstate
2322 # We don't want to just call wctx.walk here, since it would return a lot of
2322 # We don't want to just call wctx.walk here, since it would return a lot of
2323 # clean files, which we aren't interested in and takes time.
2323 # clean files, which we aren't interested in and takes time.
2324 for f in sorted(dirstate.walk(badmatch, sorted(wctx.substate),
2324 for f in sorted(dirstate.walk(badmatch, sorted(wctx.substate),
2325 True, False, full=False)):
2325 True, False, full=False)):
2326 exact = match.exact(f)
2326 exact = match.exact(f)
2327 if exact or not explicitonly and f not in wctx and repo.wvfs.lexists(f):
2327 if exact or not explicitonly and f not in wctx and repo.wvfs.lexists(f):
2328 if cca:
2328 if cca:
2329 cca(f)
2329 cca(f)
2330 names.append(f)
2330 names.append(f)
2331 if ui.verbose or not exact:
2331 if ui.verbose or not exact:
2332 ui.status(_('adding %s\n') % match.rel(f))
2332 ui.status(_('adding %s\n') % match.rel(f))
2333
2333
2334 for subpath in sorted(wctx.substate):
2334 for subpath in sorted(wctx.substate):
2335 sub = wctx.sub(subpath)
2335 sub = wctx.sub(subpath)
2336 try:
2336 try:
2337 submatch = matchmod.subdirmatcher(subpath, match)
2337 submatch = matchmod.subdirmatcher(subpath, match)
2338 if opts.get('subrepos'):
2338 if opts.get('subrepos'):
2339 bad.extend(sub.add(ui, submatch, prefix, False, **opts))
2339 bad.extend(sub.add(ui, submatch, prefix, False, **opts))
2340 else:
2340 else:
2341 bad.extend(sub.add(ui, submatch, prefix, True, **opts))
2341 bad.extend(sub.add(ui, submatch, prefix, True, **opts))
2342 except error.LookupError:
2342 except error.LookupError:
2343 ui.status(_("skipping missing subrepository: %s\n")
2343 ui.status(_("skipping missing subrepository: %s\n")
2344 % join(subpath))
2344 % join(subpath))
2345
2345
2346 if not opts.get('dry_run'):
2346 if not opts.get('dry_run'):
2347 rejected = wctx.add(names, prefix)
2347 rejected = wctx.add(names, prefix)
2348 bad.extend(f for f in rejected if f in match.files())
2348 bad.extend(f for f in rejected if f in match.files())
2349 return bad
2349 return bad
2350
2350
2351 def forget(ui, repo, match, prefix, explicitonly):
2351 def forget(ui, repo, match, prefix, explicitonly):
2352 join = lambda f: os.path.join(prefix, f)
2352 join = lambda f: os.path.join(prefix, f)
2353 bad = []
2353 bad = []
2354 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2354 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2355 wctx = repo[None]
2355 wctx = repo[None]
2356 forgot = []
2356 forgot = []
2357
2357
2358 s = repo.status(match=matchmod.badmatch(match, badfn), clean=True)
2358 s = repo.status(match=matchmod.badmatch(match, badfn), clean=True)
2359 forget = sorted(s[0] + s[1] + s[3] + s[6])
2359 forget = sorted(s[0] + s[1] + s[3] + s[6])
2360 if explicitonly:
2360 if explicitonly:
2361 forget = [f for f in forget if match.exact(f)]
2361 forget = [f for f in forget if match.exact(f)]
2362
2362
2363 for subpath in sorted(wctx.substate):
2363 for subpath in sorted(wctx.substate):
2364 sub = wctx.sub(subpath)
2364 sub = wctx.sub(subpath)
2365 try:
2365 try:
2366 submatch = matchmod.subdirmatcher(subpath, match)
2366 submatch = matchmod.subdirmatcher(subpath, match)
2367 subbad, subforgot = sub.forget(submatch, prefix)
2367 subbad, subforgot = sub.forget(submatch, prefix)
2368 bad.extend([subpath + '/' + f for f in subbad])
2368 bad.extend([subpath + '/' + f for f in subbad])
2369 forgot.extend([subpath + '/' + f for f in subforgot])
2369 forgot.extend([subpath + '/' + f for f in subforgot])
2370 except error.LookupError:
2370 except error.LookupError:
2371 ui.status(_("skipping missing subrepository: %s\n")
2371 ui.status(_("skipping missing subrepository: %s\n")
2372 % join(subpath))
2372 % join(subpath))
2373
2373
2374 if not explicitonly:
2374 if not explicitonly:
2375 for f in match.files():
2375 for f in match.files():
2376 if f not in repo.dirstate and not repo.wvfs.isdir(f):
2376 if f not in repo.dirstate and not repo.wvfs.isdir(f):
2377 if f not in forgot:
2377 if f not in forgot:
2378 if repo.wvfs.exists(f):
2378 if repo.wvfs.exists(f):
2379 # Don't complain if the exact case match wasn't given.
2379 # Don't complain if the exact case match wasn't given.
2380 # But don't do this until after checking 'forgot', so
2380 # But don't do this until after checking 'forgot', so
2381 # that subrepo files aren't normalized, and this op is
2381 # that subrepo files aren't normalized, and this op is
2382 # purely from data cached by the status walk above.
2382 # purely from data cached by the status walk above.
2383 if repo.dirstate.normalize(f) in repo.dirstate:
2383 if repo.dirstate.normalize(f) in repo.dirstate:
2384 continue
2384 continue
2385 ui.warn(_('not removing %s: '
2385 ui.warn(_('not removing %s: '
2386 'file is already untracked\n')
2386 'file is already untracked\n')
2387 % match.rel(f))
2387 % match.rel(f))
2388 bad.append(f)
2388 bad.append(f)
2389
2389
2390 for f in forget:
2390 for f in forget:
2391 if ui.verbose or not match.exact(f):
2391 if ui.verbose or not match.exact(f):
2392 ui.status(_('removing %s\n') % match.rel(f))
2392 ui.status(_('removing %s\n') % match.rel(f))
2393
2393
2394 rejected = wctx.forget(forget, prefix)
2394 rejected = wctx.forget(forget, prefix)
2395 bad.extend(f for f in rejected if f in match.files())
2395 bad.extend(f for f in rejected if f in match.files())
2396 forgot.extend(f for f in forget if f not in rejected)
2396 forgot.extend(f for f in forget if f not in rejected)
2397 return bad, forgot
2397 return bad, forgot
2398
2398
2399 def files(ui, ctx, m, fm, fmt, subrepos):
2399 def files(ui, ctx, m, fm, fmt, subrepos):
2400 rev = ctx.rev()
2400 rev = ctx.rev()
2401 ret = 1
2401 ret = 1
2402 ds = ctx.repo().dirstate
2402 ds = ctx.repo().dirstate
2403
2403
2404 for f in ctx.matches(m):
2404 for f in ctx.matches(m):
2405 if rev is None and ds[f] == 'r':
2405 if rev is None and ds[f] == 'r':
2406 continue
2406 continue
2407 fm.startitem()
2407 fm.startitem()
2408 if ui.verbose:
2408 if ui.verbose:
2409 fc = ctx[f]
2409 fc = ctx[f]
2410 fm.write('size flags', '% 10d % 1s ', fc.size(), fc.flags())
2410 fm.write('size flags', '% 10d % 1s ', fc.size(), fc.flags())
2411 fm.data(abspath=f)
2411 fm.data(abspath=f)
2412 fm.write('path', fmt, m.rel(f))
2412 fm.write('path', fmt, m.rel(f))
2413 ret = 0
2413 ret = 0
2414
2414
2415 for subpath in sorted(ctx.substate):
2415 for subpath in sorted(ctx.substate):
2416 def matchessubrepo(subpath):
2416 def matchessubrepo(subpath):
2417 return (m.exact(subpath)
2417 return (m.exact(subpath)
2418 or any(f.startswith(subpath + '/') for f in m.files()))
2418 or any(f.startswith(subpath + '/') for f in m.files()))
2419
2419
2420 if subrepos or matchessubrepo(subpath):
2420 if subrepos or matchessubrepo(subpath):
2421 sub = ctx.sub(subpath)
2421 sub = ctx.sub(subpath)
2422 try:
2422 try:
2423 submatch = matchmod.subdirmatcher(subpath, m)
2423 submatch = matchmod.subdirmatcher(subpath, m)
2424 recurse = m.exact(subpath) or subrepos
2424 recurse = m.exact(subpath) or subrepos
2425 if sub.printfiles(ui, submatch, fm, fmt, recurse) == 0:
2425 if sub.printfiles(ui, submatch, fm, fmt, recurse) == 0:
2426 ret = 0
2426 ret = 0
2427 except error.LookupError:
2427 except error.LookupError:
2428 ui.status(_("skipping missing subrepository: %s\n")
2428 ui.status(_("skipping missing subrepository: %s\n")
2429 % m.abs(subpath))
2429 % m.abs(subpath))
2430
2430
2431 return ret
2431 return ret
2432
2432
2433 def remove(ui, repo, m, prefix, after, force, subrepos, warnings=None):
2433 def remove(ui, repo, m, prefix, after, force, subrepos, warnings=None):
2434 join = lambda f: os.path.join(prefix, f)
2434 join = lambda f: os.path.join(prefix, f)
2435 ret = 0
2435 ret = 0
2436 s = repo.status(match=m, clean=True)
2436 s = repo.status(match=m, clean=True)
2437 modified, added, deleted, clean = s[0], s[1], s[3], s[6]
2437 modified, added, deleted, clean = s[0], s[1], s[3], s[6]
2438
2438
2439 wctx = repo[None]
2439 wctx = repo[None]
2440
2440
2441 if warnings is None:
2441 if warnings is None:
2442 warnings = []
2442 warnings = []
2443 warn = True
2443 warn = True
2444 else:
2444 else:
2445 warn = False
2445 warn = False
2446
2446
2447 subs = sorted(wctx.substate)
2447 subs = sorted(wctx.substate)
2448 total = len(subs)
2448 total = len(subs)
2449 count = 0
2449 count = 0
2450 for subpath in subs:
2450 for subpath in subs:
2451 def matchessubrepo(matcher, subpath):
2451 def matchessubrepo(matcher, subpath):
2452 if matcher.exact(subpath):
2452 if matcher.exact(subpath):
2453 return True
2453 return True
2454 for f in matcher.files():
2454 for f in matcher.files():
2455 if f.startswith(subpath):
2455 if f.startswith(subpath):
2456 return True
2456 return True
2457 return False
2457 return False
2458
2458
2459 count += 1
2459 count += 1
2460 if subrepos or matchessubrepo(m, subpath):
2460 if subrepos or matchessubrepo(m, subpath):
2461 ui.progress(_('searching'), count, total=total, unit=_('subrepos'))
2461 ui.progress(_('searching'), count, total=total, unit=_('subrepos'))
2462
2462
2463 sub = wctx.sub(subpath)
2463 sub = wctx.sub(subpath)
2464 try:
2464 try:
2465 submatch = matchmod.subdirmatcher(subpath, m)
2465 submatch = matchmod.subdirmatcher(subpath, m)
2466 if sub.removefiles(submatch, prefix, after, force, subrepos,
2466 if sub.removefiles(submatch, prefix, after, force, subrepos,
2467 warnings):
2467 warnings):
2468 ret = 1
2468 ret = 1
2469 except error.LookupError:
2469 except error.LookupError:
2470 warnings.append(_("skipping missing subrepository: %s\n")
2470 warnings.append(_("skipping missing subrepository: %s\n")
2471 % join(subpath))
2471 % join(subpath))
2472 ui.progress(_('searching'), None)
2472 ui.progress(_('searching'), None)
2473
2473
2474 # warn about failure to delete explicit files/dirs
2474 # warn about failure to delete explicit files/dirs
2475 deleteddirs = util.dirs(deleted)
2475 deleteddirs = util.dirs(deleted)
2476 files = m.files()
2476 files = m.files()
2477 total = len(files)
2477 total = len(files)
2478 count = 0
2478 count = 0
2479 for f in files:
2479 for f in files:
2480 def insubrepo():
2480 def insubrepo():
2481 for subpath in wctx.substate:
2481 for subpath in wctx.substate:
2482 if f.startswith(subpath):
2482 if f.startswith(subpath):
2483 return True
2483 return True
2484 return False
2484 return False
2485
2485
2486 count += 1
2486 count += 1
2487 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2487 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2488 isdir = f in deleteddirs or wctx.hasdir(f)
2488 isdir = f in deleteddirs or wctx.hasdir(f)
2489 if f in repo.dirstate or isdir or f == '.' or insubrepo():
2489 if f in repo.dirstate or isdir or f == '.' or insubrepo():
2490 continue
2490 continue
2491
2491
2492 if repo.wvfs.exists(f):
2492 if repo.wvfs.exists(f):
2493 if repo.wvfs.isdir(f):
2493 if repo.wvfs.isdir(f):
2494 warnings.append(_('not removing %s: no tracked files\n')
2494 warnings.append(_('not removing %s: no tracked files\n')
2495 % m.rel(f))
2495 % m.rel(f))
2496 else:
2496 else:
2497 warnings.append(_('not removing %s: file is untracked\n')
2497 warnings.append(_('not removing %s: file is untracked\n')
2498 % m.rel(f))
2498 % m.rel(f))
2499 # missing files will generate a warning elsewhere
2499 # missing files will generate a warning elsewhere
2500 ret = 1
2500 ret = 1
2501 ui.progress(_('deleting'), None)
2501 ui.progress(_('deleting'), None)
2502
2502
2503 if force:
2503 if force:
2504 list = modified + deleted + clean + added
2504 list = modified + deleted + clean + added
2505 elif after:
2505 elif after:
2506 list = deleted
2506 list = deleted
2507 remaining = modified + added + clean
2507 remaining = modified + added + clean
2508 total = len(remaining)
2508 total = len(remaining)
2509 count = 0
2509 count = 0
2510 for f in remaining:
2510 for f in remaining:
2511 count += 1
2511 count += 1
2512 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2512 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2513 warnings.append(_('not removing %s: file still exists\n')
2513 warnings.append(_('not removing %s: file still exists\n')
2514 % m.rel(f))
2514 % m.rel(f))
2515 ret = 1
2515 ret = 1
2516 ui.progress(_('skipping'), None)
2516 ui.progress(_('skipping'), None)
2517 else:
2517 else:
2518 list = deleted + clean
2518 list = deleted + clean
2519 total = len(modified) + len(added)
2519 total = len(modified) + len(added)
2520 count = 0
2520 count = 0
2521 for f in modified:
2521 for f in modified:
2522 count += 1
2522 count += 1
2523 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2523 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2524 warnings.append(_('not removing %s: file is modified (use -f'
2524 warnings.append(_('not removing %s: file is modified (use -f'
2525 ' to force removal)\n') % m.rel(f))
2525 ' to force removal)\n') % m.rel(f))
2526 ret = 1
2526 ret = 1
2527 for f in added:
2527 for f in added:
2528 count += 1
2528 count += 1
2529 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2529 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2530 warnings.append(_('not removing %s: file has been marked for add'
2530 warnings.append(_('not removing %s: file has been marked for add'
2531 ' (use forget to undo)\n') % m.rel(f))
2531 ' (use forget to undo)\n') % m.rel(f))
2532 ret = 1
2532 ret = 1
2533 ui.progress(_('skipping'), None)
2533 ui.progress(_('skipping'), None)
2534
2534
2535 list = sorted(list)
2535 list = sorted(list)
2536 total = len(list)
2536 total = len(list)
2537 count = 0
2537 count = 0
2538 for f in list:
2538 for f in list:
2539 count += 1
2539 count += 1
2540 if ui.verbose or not m.exact(f):
2540 if ui.verbose or not m.exact(f):
2541 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2541 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2542 ui.status(_('removing %s\n') % m.rel(f))
2542 ui.status(_('removing %s\n') % m.rel(f))
2543 ui.progress(_('deleting'), None)
2543 ui.progress(_('deleting'), None)
2544
2544
2545 with repo.wlock():
2545 with repo.wlock():
2546 if not after:
2546 if not after:
2547 for f in list:
2547 for f in list:
2548 if f in added:
2548 if f in added:
2549 continue # we never unlink added files on remove
2549 continue # we never unlink added files on remove
2550 util.unlinkpath(repo.wjoin(f), ignoremissing=True)
2550 util.unlinkpath(repo.wjoin(f), ignoremissing=True)
2551 repo[None].forget(list)
2551 repo[None].forget(list)
2552
2552
2553 if warn:
2553 if warn:
2554 for warning in warnings:
2554 for warning in warnings:
2555 ui.warn(warning)
2555 ui.warn(warning)
2556
2556
2557 return ret
2557 return ret
2558
2558
2559 def cat(ui, repo, ctx, matcher, prefix, **opts):
2559 def cat(ui, repo, ctx, matcher, prefix, **opts):
2560 err = 1
2560 err = 1
2561
2561
2562 def write(path):
2562 def write(path):
2563 fp = makefileobj(repo, opts.get('output'), ctx.node(),
2563 fp = makefileobj(repo, opts.get('output'), ctx.node(),
2564 pathname=os.path.join(prefix, path))
2564 pathname=os.path.join(prefix, path))
2565 data = ctx[path].data()
2565 data = ctx[path].data()
2566 if opts.get('decode'):
2566 if opts.get('decode'):
2567 data = repo.wwritedata(path, data)
2567 data = repo.wwritedata(path, data)
2568 fp.write(data)
2568 fp.write(data)
2569 fp.close()
2569 fp.close()
2570
2570
2571 # Automation often uses hg cat on single files, so special case it
2571 # Automation often uses hg cat on single files, so special case it
2572 # for performance to avoid the cost of parsing the manifest.
2572 # for performance to avoid the cost of parsing the manifest.
2573 if len(matcher.files()) == 1 and not matcher.anypats():
2573 if len(matcher.files()) == 1 and not matcher.anypats():
2574 file = matcher.files()[0]
2574 file = matcher.files()[0]
2575 mf = repo.manifest
2575 mf = repo.manifest
2576 mfnode = ctx.manifestnode()
2576 mfnode = ctx.manifestnode()
2577 if mfnode and mf.find(mfnode, file)[0]:
2577 if mfnode and mf.find(mfnode, file)[0]:
2578 write(file)
2578 write(file)
2579 return 0
2579 return 0
2580
2580
2581 # Don't warn about "missing" files that are really in subrepos
2581 # Don't warn about "missing" files that are really in subrepos
2582 def badfn(path, msg):
2582 def badfn(path, msg):
2583 for subpath in ctx.substate:
2583 for subpath in ctx.substate:
2584 if path.startswith(subpath):
2584 if path.startswith(subpath):
2585 return
2585 return
2586 matcher.bad(path, msg)
2586 matcher.bad(path, msg)
2587
2587
2588 for abs in ctx.walk(matchmod.badmatch(matcher, badfn)):
2588 for abs in ctx.walk(matchmod.badmatch(matcher, badfn)):
2589 write(abs)
2589 write(abs)
2590 err = 0
2590 err = 0
2591
2591
2592 for subpath in sorted(ctx.substate):
2592 for subpath in sorted(ctx.substate):
2593 sub = ctx.sub(subpath)
2593 sub = ctx.sub(subpath)
2594 try:
2594 try:
2595 submatch = matchmod.subdirmatcher(subpath, matcher)
2595 submatch = matchmod.subdirmatcher(subpath, matcher)
2596
2596
2597 if not sub.cat(submatch, os.path.join(prefix, sub._path),
2597 if not sub.cat(submatch, os.path.join(prefix, sub._path),
2598 **opts):
2598 **opts):
2599 err = 0
2599 err = 0
2600 except error.RepoLookupError:
2600 except error.RepoLookupError:
2601 ui.status(_("skipping missing subrepository: %s\n")
2601 ui.status(_("skipping missing subrepository: %s\n")
2602 % os.path.join(prefix, subpath))
2602 % os.path.join(prefix, subpath))
2603
2603
2604 return err
2604 return err
2605
2605
2606 def commit(ui, repo, commitfunc, pats, opts):
2606 def commit(ui, repo, commitfunc, pats, opts):
2607 '''commit the specified files or all outstanding changes'''
2607 '''commit the specified files or all outstanding changes'''
2608 date = opts.get('date')
2608 date = opts.get('date')
2609 if date:
2609 if date:
2610 opts['date'] = util.parsedate(date)
2610 opts['date'] = util.parsedate(date)
2611 message = logmessage(ui, opts)
2611 message = logmessage(ui, opts)
2612 matcher = scmutil.match(repo[None], pats, opts)
2612 matcher = scmutil.match(repo[None], pats, opts)
2613
2613
2614 # extract addremove carefully -- this function can be called from a command
2614 # extract addremove carefully -- this function can be called from a command
2615 # that doesn't support addremove
2615 # that doesn't support addremove
2616 if opts.get('addremove'):
2616 if opts.get('addremove'):
2617 if scmutil.addremove(repo, matcher, "", opts) != 0:
2617 if scmutil.addremove(repo, matcher, "", opts) != 0:
2618 raise error.Abort(
2618 raise error.Abort(
2619 _("failed to mark all new/missing files as added/removed"))
2619 _("failed to mark all new/missing files as added/removed"))
2620
2620
2621 return commitfunc(ui, repo, message, matcher, opts)
2621 return commitfunc(ui, repo, message, matcher, opts)
2622
2622
2623 def amend(ui, repo, commitfunc, old, extra, pats, opts):
2623 def amend(ui, repo, commitfunc, old, extra, pats, opts):
2624 # avoid cycle context -> subrepo -> cmdutil
2624 # avoid cycle context -> subrepo -> cmdutil
2625 from . import context
2625 from . import context
2626
2626
2627 # amend will reuse the existing user if not specified, but the obsolete
2627 # amend will reuse the existing user if not specified, but the obsolete
2628 # marker creation requires that the current user's name is specified.
2628 # marker creation requires that the current user's name is specified.
2629 if obsolete.isenabled(repo, obsolete.createmarkersopt):
2629 if obsolete.isenabled(repo, obsolete.createmarkersopt):
2630 ui.username() # raise exception if username not set
2630 ui.username() # raise exception if username not set
2631
2631
2632 ui.note(_('amending changeset %s\n') % old)
2632 ui.note(_('amending changeset %s\n') % old)
2633 base = old.p1()
2633 base = old.p1()
2634 createmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
2634 createmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
2635
2635
2636 wlock = lock = newid = None
2636 wlock = lock = newid = None
2637 try:
2637 try:
2638 wlock = repo.wlock()
2638 wlock = repo.wlock()
2639 lock = repo.lock()
2639 lock = repo.lock()
2640 with repo.transaction('amend') as tr:
2640 with repo.transaction('amend') as tr:
2641 # See if we got a message from -m or -l, if not, open the editor
2641 # See if we got a message from -m or -l, if not, open the editor
2642 # with the message of the changeset to amend
2642 # with the message of the changeset to amend
2643 message = logmessage(ui, opts)
2643 message = logmessage(ui, opts)
2644 # ensure logfile does not conflict with later enforcement of the
2644 # ensure logfile does not conflict with later enforcement of the
2645 # message. potential logfile content has been processed by
2645 # message. potential logfile content has been processed by
2646 # `logmessage` anyway.
2646 # `logmessage` anyway.
2647 opts.pop('logfile')
2647 opts.pop('logfile')
2648 # First, do a regular commit to record all changes in the working
2648 # First, do a regular commit to record all changes in the working
2649 # directory (if there are any)
2649 # directory (if there are any)
2650 ui.callhooks = False
2650 ui.callhooks = False
2651 activebookmark = repo._bookmarks.active
2651 activebookmark = repo._bookmarks.active
2652 try:
2652 try:
2653 repo._bookmarks.active = None
2653 repo._bookmarks.active = None
2654 opts['message'] = 'temporary amend commit for %s' % old
2654 opts['message'] = 'temporary amend commit for %s' % old
2655 node = commit(ui, repo, commitfunc, pats, opts)
2655 node = commit(ui, repo, commitfunc, pats, opts)
2656 finally:
2656 finally:
2657 repo._bookmarks.active = activebookmark
2657 repo._bookmarks.active = activebookmark
2658 repo._bookmarks.recordchange(tr)
2658 repo._bookmarks.recordchange(tr)
2659 ui.callhooks = True
2659 ui.callhooks = True
2660 ctx = repo[node]
2660 ctx = repo[node]
2661
2661
2662 # Participating changesets:
2662 # Participating changesets:
2663 #
2663 #
2664 # node/ctx o - new (intermediate) commit that contains changes
2664 # node/ctx o - new (intermediate) commit that contains changes
2665 # | from working dir to go into amending commit
2665 # | from working dir to go into amending commit
2666 # | (or a workingctx if there were no changes)
2666 # | (or a workingctx if there were no changes)
2667 # |
2667 # |
2668 # old o - changeset to amend
2668 # old o - changeset to amend
2669 # |
2669 # |
2670 # base o - parent of amending changeset
2670 # base o - parent of amending changeset
2671
2671
2672 # Update extra dict from amended commit (e.g. to preserve graft
2672 # Update extra dict from amended commit (e.g. to preserve graft
2673 # source)
2673 # source)
2674 extra.update(old.extra())
2674 extra.update(old.extra())
2675
2675
2676 # Also update it from the intermediate commit or from the wctx
2676 # Also update it from the intermediate commit or from the wctx
2677 extra.update(ctx.extra())
2677 extra.update(ctx.extra())
2678
2678
2679 if len(old.parents()) > 1:
2679 if len(old.parents()) > 1:
2680 # ctx.files() isn't reliable for merges, so fall back to the
2680 # ctx.files() isn't reliable for merges, so fall back to the
2681 # slower repo.status() method
2681 # slower repo.status() method
2682 files = set([fn for st in repo.status(base, old)[:3]
2682 files = set([fn for st in repo.status(base, old)[:3]
2683 for fn in st])
2683 for fn in st])
2684 else:
2684 else:
2685 files = set(old.files())
2685 files = set(old.files())
2686
2686
2687 # Second, we use either the commit we just did, or if there were no
2687 # Second, we use either the commit we just did, or if there were no
2688 # changes the parent of the working directory as the version of the
2688 # changes the parent of the working directory as the version of the
2689 # files in the final amend commit
2689 # files in the final amend commit
2690 if node:
2690 if node:
2691 ui.note(_('copying changeset %s to %s\n') % (ctx, base))
2691 ui.note(_('copying changeset %s to %s\n') % (ctx, base))
2692
2692
2693 user = ctx.user()
2693 user = ctx.user()
2694 date = ctx.date()
2694 date = ctx.date()
2695 # Recompute copies (avoid recording a -> b -> a)
2695 # Recompute copies (avoid recording a -> b -> a)
2696 copied = copies.pathcopies(base, ctx)
2696 copied = copies.pathcopies(base, ctx)
2697 if old.p2:
2697 if old.p2:
2698 copied.update(copies.pathcopies(old.p2(), ctx))
2698 copied.update(copies.pathcopies(old.p2(), ctx))
2699
2699
2700 # Prune files which were reverted by the updates: if old
2700 # Prune files which were reverted by the updates: if old
2701 # introduced file X and our intermediate commit, node,
2701 # introduced file X and our intermediate commit, node,
2702 # renamed that file, then those two files are the same and
2702 # renamed that file, then those two files are the same and
2703 # we can discard X from our list of files. Likewise if X
2703 # we can discard X from our list of files. Likewise if X
2704 # was deleted, it's no longer relevant
2704 # was deleted, it's no longer relevant
2705 files.update(ctx.files())
2705 files.update(ctx.files())
2706
2706
2707 def samefile(f):
2707 def samefile(f):
2708 if f in ctx.manifest():
2708 if f in ctx.manifest():
2709 a = ctx.filectx(f)
2709 a = ctx.filectx(f)
2710 if f in base.manifest():
2710 if f in base.manifest():
2711 b = base.filectx(f)
2711 b = base.filectx(f)
2712 return (not a.cmp(b)
2712 return (not a.cmp(b)
2713 and a.flags() == b.flags())
2713 and a.flags() == b.flags())
2714 else:
2714 else:
2715 return False
2715 return False
2716 else:
2716 else:
2717 return f not in base.manifest()
2717 return f not in base.manifest()
2718 files = [f for f in files if not samefile(f)]
2718 files = [f for f in files if not samefile(f)]
2719
2719
2720 def filectxfn(repo, ctx_, path):
2720 def filectxfn(repo, ctx_, path):
2721 try:
2721 try:
2722 fctx = ctx[path]
2722 fctx = ctx[path]
2723 flags = fctx.flags()
2723 flags = fctx.flags()
2724 mctx = context.memfilectx(repo,
2724 mctx = context.memfilectx(repo,
2725 fctx.path(), fctx.data(),
2725 fctx.path(), fctx.data(),
2726 islink='l' in flags,
2726 islink='l' in flags,
2727 isexec='x' in flags,
2727 isexec='x' in flags,
2728 copied=copied.get(path))
2728 copied=copied.get(path))
2729 return mctx
2729 return mctx
2730 except KeyError:
2730 except KeyError:
2731 return None
2731 return None
2732 else:
2732 else:
2733 ui.note(_('copying changeset %s to %s\n') % (old, base))
2733 ui.note(_('copying changeset %s to %s\n') % (old, base))
2734
2734
2735 # Use version of files as in the old cset
2735 # Use version of files as in the old cset
2736 def filectxfn(repo, ctx_, path):
2736 def filectxfn(repo, ctx_, path):
2737 try:
2737 try:
2738 return old.filectx(path)
2738 return old.filectx(path)
2739 except KeyError:
2739 except KeyError:
2740 return None
2740 return None
2741
2741
2742 user = opts.get('user') or old.user()
2742 user = opts.get('user') or old.user()
2743 date = opts.get('date') or old.date()
2743 date = opts.get('date') or old.date()
2744 editform = mergeeditform(old, 'commit.amend')
2744 editform = mergeeditform(old, 'commit.amend')
2745 editor = getcommiteditor(editform=editform, **opts)
2745 editor = getcommiteditor(editform=editform, **opts)
2746 if not message:
2746 if not message:
2747 editor = getcommiteditor(edit=True, editform=editform)
2747 editor = getcommiteditor(edit=True, editform=editform)
2748 message = old.description()
2748 message = old.description()
2749
2749
2750 pureextra = extra.copy()
2750 pureextra = extra.copy()
2751 extra['amend_source'] = old.hex()
2751 extra['amend_source'] = old.hex()
2752
2752
2753 new = context.memctx(repo,
2753 new = context.memctx(repo,
2754 parents=[base.node(), old.p2().node()],
2754 parents=[base.node(), old.p2().node()],
2755 text=message,
2755 text=message,
2756 files=files,
2756 files=files,
2757 filectxfn=filectxfn,
2757 filectxfn=filectxfn,
2758 user=user,
2758 user=user,
2759 date=date,
2759 date=date,
2760 extra=extra,
2760 extra=extra,
2761 editor=editor)
2761 editor=editor)
2762
2762
2763 newdesc = changelog.stripdesc(new.description())
2763 newdesc = changelog.stripdesc(new.description())
2764 if ((not node)
2764 if ((not node)
2765 and newdesc == old.description()
2765 and newdesc == old.description()
2766 and user == old.user()
2766 and user == old.user()
2767 and date == old.date()
2767 and date == old.date()
2768 and pureextra == old.extra()):
2768 and pureextra == old.extra()):
2769 # nothing changed. continuing here would create a new node
2769 # nothing changed. continuing here would create a new node
2770 # anyway because of the amend_source noise.
2770 # anyway because of the amend_source noise.
2771 #
2771 #
2772 # This not what we expect from amend.
2772 # This not what we expect from amend.
2773 return old.node()
2773 return old.node()
2774
2774
2775 ph = repo.ui.config('phases', 'new-commit', phases.draft)
2775 ph = repo.ui.config('phases', 'new-commit', phases.draft)
2776 try:
2776 try:
2777 if opts.get('secret'):
2777 if opts.get('secret'):
2778 commitphase = 'secret'
2778 commitphase = 'secret'
2779 else:
2779 else:
2780 commitphase = old.phase()
2780 commitphase = old.phase()
2781 repo.ui.setconfig('phases', 'new-commit', commitphase, 'amend')
2781 repo.ui.setconfig('phases', 'new-commit', commitphase, 'amend')
2782 newid = repo.commitctx(new)
2782 newid = repo.commitctx(new)
2783 finally:
2783 finally:
2784 repo.ui.setconfig('phases', 'new-commit', ph, 'amend')
2784 repo.ui.setconfig('phases', 'new-commit', ph, 'amend')
2785 if newid != old.node():
2785 if newid != old.node():
2786 # Reroute the working copy parent to the new changeset
2786 # Reroute the working copy parent to the new changeset
2787 repo.setparents(newid, nullid)
2787 repo.setparents(newid, nullid)
2788
2788
2789 # Move bookmarks from old parent to amend commit
2789 # Move bookmarks from old parent to amend commit
2790 bms = repo.nodebookmarks(old.node())
2790 bms = repo.nodebookmarks(old.node())
2791 if bms:
2791 if bms:
2792 marks = repo._bookmarks
2792 marks = repo._bookmarks
2793 for bm in bms:
2793 for bm in bms:
2794 ui.debug('moving bookmarks %r from %s to %s\n' %
2794 ui.debug('moving bookmarks %r from %s to %s\n' %
2795 (marks, old.hex(), hex(newid)))
2795 (marks, old.hex(), hex(newid)))
2796 marks[bm] = newid
2796 marks[bm] = newid
2797 marks.recordchange(tr)
2797 marks.recordchange(tr)
2798 #commit the whole amend process
2798 #commit the whole amend process
2799 if createmarkers:
2799 if createmarkers:
2800 # mark the new changeset as successor of the rewritten one
2800 # mark the new changeset as successor of the rewritten one
2801 new = repo[newid]
2801 new = repo[newid]
2802 obs = [(old, (new,))]
2802 obs = [(old, (new,))]
2803 if node:
2803 if node:
2804 obs.append((ctx, ()))
2804 obs.append((ctx, ()))
2805
2805
2806 obsolete.createmarkers(repo, obs)
2806 obsolete.createmarkers(repo, obs)
2807 if not createmarkers and newid != old.node():
2807 if not createmarkers and newid != old.node():
2808 # Strip the intermediate commit (if there was one) and the amended
2808 # Strip the intermediate commit (if there was one) and the amended
2809 # commit
2809 # commit
2810 if node:
2810 if node:
2811 ui.note(_('stripping intermediate changeset %s\n') % ctx)
2811 ui.note(_('stripping intermediate changeset %s\n') % ctx)
2812 ui.note(_('stripping amended changeset %s\n') % old)
2812 ui.note(_('stripping amended changeset %s\n') % old)
2813 repair.strip(ui, repo, old.node(), topic='amend-backup')
2813 repair.strip(ui, repo, old.node(), topic='amend-backup')
2814 finally:
2814 finally:
2815 lockmod.release(lock, wlock)
2815 lockmod.release(lock, wlock)
2816 return newid
2816 return newid
2817
2817
2818 def commiteditor(repo, ctx, subs, editform=''):
2818 def commiteditor(repo, ctx, subs, editform=''):
2819 if ctx.description():
2819 if ctx.description():
2820 return ctx.description()
2820 return ctx.description()
2821 return commitforceeditor(repo, ctx, subs, editform=editform,
2821 return commitforceeditor(repo, ctx, subs, editform=editform,
2822 unchangedmessagedetection=True)
2822 unchangedmessagedetection=True)
2823
2823
2824 def commitforceeditor(repo, ctx, subs, finishdesc=None, extramsg=None,
2824 def commitforceeditor(repo, ctx, subs, finishdesc=None, extramsg=None,
2825 editform='', unchangedmessagedetection=False):
2825 editform='', unchangedmessagedetection=False):
2826 if not extramsg:
2826 if not extramsg:
2827 extramsg = _("Leave message empty to abort commit.")
2827 extramsg = _("Leave message empty to abort commit.")
2828
2828
2829 forms = [e for e in editform.split('.') if e]
2829 forms = [e for e in editform.split('.') if e]
2830 forms.insert(0, 'changeset')
2830 forms.insert(0, 'changeset')
2831 templatetext = None
2831 templatetext = None
2832 while forms:
2832 while forms:
2833 tmpl = repo.ui.config('committemplate', '.'.join(forms))
2833 tmpl = repo.ui.config('committemplate', '.'.join(forms))
2834 if tmpl:
2834 if tmpl:
2835 templatetext = committext = buildcommittemplate(
2835 templatetext = committext = buildcommittemplate(
2836 repo, ctx, subs, extramsg, tmpl)
2836 repo, ctx, subs, extramsg, tmpl)
2837 break
2837 break
2838 forms.pop()
2838 forms.pop()
2839 else:
2839 else:
2840 committext = buildcommittext(repo, ctx, subs, extramsg)
2840 committext = buildcommittext(repo, ctx, subs, extramsg)
2841
2841
2842 # run editor in the repository root
2842 # run editor in the repository root
2843 olddir = os.getcwd()
2843 olddir = os.getcwd()
2844 os.chdir(repo.root)
2844 os.chdir(repo.root)
2845
2845
2846 # make in-memory changes visible to external process
2846 # make in-memory changes visible to external process
2847 tr = repo.currenttransaction()
2847 tr = repo.currenttransaction()
2848 repo.dirstate.write(tr)
2848 repo.dirstate.write(tr)
2849 pending = tr and tr.writepending() and repo.root
2849 pending = tr and tr.writepending() and repo.root
2850
2850
2851 editortext = repo.ui.edit(committext, ctx.user(), ctx.extra(),
2851 editortext = repo.ui.edit(committext, ctx.user(), ctx.extra(),
2852 editform=editform, pending=pending)
2852 editform=editform, pending=pending)
2853 text = re.sub("(?m)^HG:.*(\n|$)", "", editortext)
2853 text = re.sub("(?m)^HG:.*(\n|$)", "", editortext)
2854 os.chdir(olddir)
2854 os.chdir(olddir)
2855
2855
2856 if finishdesc:
2856 if finishdesc:
2857 text = finishdesc(text)
2857 text = finishdesc(text)
2858 if not text.strip():
2858 if not text.strip():
2859 raise error.Abort(_("empty commit message"))
2859 raise error.Abort(_("empty commit message"))
2860 if unchangedmessagedetection and editortext == templatetext:
2860 if unchangedmessagedetection and editortext == templatetext:
2861 raise error.Abort(_("commit message unchanged"))
2861 raise error.Abort(_("commit message unchanged"))
2862
2862
2863 return text
2863 return text
2864
2864
2865 def buildcommittemplate(repo, ctx, subs, extramsg, tmpl):
2865 def buildcommittemplate(repo, ctx, subs, extramsg, tmpl):
2866 ui = repo.ui
2866 ui = repo.ui
2867 tmpl, mapfile = gettemplate(ui, tmpl, None)
2867 tmpl, mapfile = gettemplate(ui, tmpl, None)
2868
2868
2869 t = changeset_templater(ui, repo, None, {}, tmpl, mapfile, False)
2869 t = changeset_templater(ui, repo, None, {}, tmpl, mapfile, False)
2870
2870
2871 for k, v in repo.ui.configitems('committemplate'):
2871 for k, v in repo.ui.configitems('committemplate'):
2872 if k != 'changeset':
2872 if k != 'changeset':
2873 t.t.cache[k] = v
2873 t.t.cache[k] = v
2874
2874
2875 if not extramsg:
2875 if not extramsg:
2876 extramsg = '' # ensure that extramsg is string
2876 extramsg = '' # ensure that extramsg is string
2877
2877
2878 ui.pushbuffer()
2878 ui.pushbuffer()
2879 t.show(ctx, extramsg=extramsg)
2879 t.show(ctx, extramsg=extramsg)
2880 return ui.popbuffer()
2880 return ui.popbuffer()
2881
2881
2882 def hgprefix(msg):
2882 def hgprefix(msg):
2883 return "\n".join(["HG: %s" % a for a in msg.split("\n") if a])
2883 return "\n".join(["HG: %s" % a for a in msg.split("\n") if a])
2884
2884
2885 def buildcommittext(repo, ctx, subs, extramsg):
2885 def buildcommittext(repo, ctx, subs, extramsg):
2886 edittext = []
2886 edittext = []
2887 modified, added, removed = ctx.modified(), ctx.added(), ctx.removed()
2887 modified, added, removed = ctx.modified(), ctx.added(), ctx.removed()
2888 if ctx.description():
2888 if ctx.description():
2889 edittext.append(ctx.description())
2889 edittext.append(ctx.description())
2890 edittext.append("")
2890 edittext.append("")
2891 edittext.append("") # Empty line between message and comments.
2891 edittext.append("") # Empty line between message and comments.
2892 edittext.append(hgprefix(_("Enter commit message."
2892 edittext.append(hgprefix(_("Enter commit message."
2893 " Lines beginning with 'HG:' are removed.")))
2893 " Lines beginning with 'HG:' are removed.")))
2894 edittext.append(hgprefix(extramsg))
2894 edittext.append(hgprefix(extramsg))
2895 edittext.append("HG: --")
2895 edittext.append("HG: --")
2896 edittext.append(hgprefix(_("user: %s") % ctx.user()))
2896 edittext.append(hgprefix(_("user: %s") % ctx.user()))
2897 if ctx.p2():
2897 if ctx.p2():
2898 edittext.append(hgprefix(_("branch merge")))
2898 edittext.append(hgprefix(_("branch merge")))
2899 if ctx.branch():
2899 if ctx.branch():
2900 edittext.append(hgprefix(_("branch '%s'") % ctx.branch()))
2900 edittext.append(hgprefix(_("branch '%s'") % ctx.branch()))
2901 if bookmarks.isactivewdirparent(repo):
2901 if bookmarks.isactivewdirparent(repo):
2902 edittext.append(hgprefix(_("bookmark '%s'") % repo._activebookmark))
2902 edittext.append(hgprefix(_("bookmark '%s'") % repo._activebookmark))
2903 edittext.extend([hgprefix(_("subrepo %s") % s) for s in subs])
2903 edittext.extend([hgprefix(_("subrepo %s") % s) for s in subs])
2904 edittext.extend([hgprefix(_("added %s") % f) for f in added])
2904 edittext.extend([hgprefix(_("added %s") % f) for f in added])
2905 edittext.extend([hgprefix(_("changed %s") % f) for f in modified])
2905 edittext.extend([hgprefix(_("changed %s") % f) for f in modified])
2906 edittext.extend([hgprefix(_("removed %s") % f) for f in removed])
2906 edittext.extend([hgprefix(_("removed %s") % f) for f in removed])
2907 if not added and not modified and not removed:
2907 if not added and not modified and not removed:
2908 edittext.append(hgprefix(_("no files changed")))
2908 edittext.append(hgprefix(_("no files changed")))
2909 edittext.append("")
2909 edittext.append("")
2910
2910
2911 return "\n".join(edittext)
2911 return "\n".join(edittext)
2912
2912
2913 def commitstatus(repo, node, branch, bheads=None, opts=None):
2913 def commitstatus(repo, node, branch, bheads=None, opts=None):
2914 if opts is None:
2914 if opts is None:
2915 opts = {}
2915 opts = {}
2916 ctx = repo[node]
2916 ctx = repo[node]
2917 parents = ctx.parents()
2917 parents = ctx.parents()
2918
2918
2919 if (not opts.get('amend') and bheads and node not in bheads and not
2919 if (not opts.get('amend') and bheads and node not in bheads and not
2920 [x for x in parents if x.node() in bheads and x.branch() == branch]):
2920 [x for x in parents if x.node() in bheads and x.branch() == branch]):
2921 repo.ui.status(_('created new head\n'))
2921 repo.ui.status(_('created new head\n'))
2922 # The message is not printed for initial roots. For the other
2922 # The message is not printed for initial roots. For the other
2923 # changesets, it is printed in the following situations:
2923 # changesets, it is printed in the following situations:
2924 #
2924 #
2925 # Par column: for the 2 parents with ...
2925 # Par column: for the 2 parents with ...
2926 # N: null or no parent
2926 # N: null or no parent
2927 # B: parent is on another named branch
2927 # B: parent is on another named branch
2928 # C: parent is a regular non head changeset
2928 # C: parent is a regular non head changeset
2929 # H: parent was a branch head of the current branch
2929 # H: parent was a branch head of the current branch
2930 # Msg column: whether we print "created new head" message
2930 # Msg column: whether we print "created new head" message
2931 # In the following, it is assumed that there already exists some
2931 # In the following, it is assumed that there already exists some
2932 # initial branch heads of the current branch, otherwise nothing is
2932 # initial branch heads of the current branch, otherwise nothing is
2933 # printed anyway.
2933 # printed anyway.
2934 #
2934 #
2935 # Par Msg Comment
2935 # Par Msg Comment
2936 # N N y additional topo root
2936 # N N y additional topo root
2937 #
2937 #
2938 # B N y additional branch root
2938 # B N y additional branch root
2939 # C N y additional topo head
2939 # C N y additional topo head
2940 # H N n usual case
2940 # H N n usual case
2941 #
2941 #
2942 # B B y weird additional branch root
2942 # B B y weird additional branch root
2943 # C B y branch merge
2943 # C B y branch merge
2944 # H B n merge with named branch
2944 # H B n merge with named branch
2945 #
2945 #
2946 # C C y additional head from merge
2946 # C C y additional head from merge
2947 # C H n merge with a head
2947 # C H n merge with a head
2948 #
2948 #
2949 # H H n head merge: head count decreases
2949 # H H n head merge: head count decreases
2950
2950
2951 if not opts.get('close_branch'):
2951 if not opts.get('close_branch'):
2952 for r in parents:
2952 for r in parents:
2953 if r.closesbranch() and r.branch() == branch:
2953 if r.closesbranch() and r.branch() == branch:
2954 repo.ui.status(_('reopening closed branch head %d\n') % r)
2954 repo.ui.status(_('reopening closed branch head %d\n') % r)
2955
2955
2956 if repo.ui.debugflag:
2956 if repo.ui.debugflag:
2957 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx.hex()))
2957 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx.hex()))
2958 elif repo.ui.verbose:
2958 elif repo.ui.verbose:
2959 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx))
2959 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx))
2960
2960
2961 def postcommitstatus(repo, pats, opts):
2961 def postcommitstatus(repo, pats, opts):
2962 return repo.status(match=scmutil.match(repo[None], pats, opts))
2962 return repo.status(match=scmutil.match(repo[None], pats, opts))
2963
2963
2964 def revert(ui, repo, ctx, parents, *pats, **opts):
2964 def revert(ui, repo, ctx, parents, *pats, **opts):
2965 parent, p2 = parents
2965 parent, p2 = parents
2966 node = ctx.node()
2966 node = ctx.node()
2967
2967
2968 mf = ctx.manifest()
2968 mf = ctx.manifest()
2969 if node == p2:
2969 if node == p2:
2970 parent = p2
2970 parent = p2
2971
2971
2972 # need all matching names in dirstate and manifest of target rev,
2972 # need all matching names in dirstate and manifest of target rev,
2973 # so have to walk both. do not print errors if files exist in one
2973 # so have to walk both. do not print errors if files exist in one
2974 # but not other. in both cases, filesets should be evaluated against
2974 # but not other. in both cases, filesets should be evaluated against
2975 # workingctx to get consistent result (issue4497). this means 'set:**'
2975 # workingctx to get consistent result (issue4497). this means 'set:**'
2976 # cannot be used to select missing files from target rev.
2976 # cannot be used to select missing files from target rev.
2977
2977
2978 # `names` is a mapping for all elements in working copy and target revision
2978 # `names` is a mapping for all elements in working copy and target revision
2979 # The mapping is in the form:
2979 # The mapping is in the form:
2980 # <asb path in repo> -> (<path from CWD>, <exactly specified by matcher?>)
2980 # <asb path in repo> -> (<path from CWD>, <exactly specified by matcher?>)
2981 names = {}
2981 names = {}
2982
2982
2983 with repo.wlock():
2983 with repo.wlock():
2984 ## filling of the `names` mapping
2984 ## filling of the `names` mapping
2985 # walk dirstate to fill `names`
2985 # walk dirstate to fill `names`
2986
2986
2987 interactive = opts.get('interactive', False)
2987 interactive = opts.get('interactive', False)
2988 wctx = repo[None]
2988 wctx = repo[None]
2989 m = scmutil.match(wctx, pats, opts)
2989 m = scmutil.match(wctx, pats, opts)
2990
2990
2991 # we'll need this later
2991 # we'll need this later
2992 targetsubs = sorted(s for s in wctx.substate if m(s))
2992 targetsubs = sorted(s for s in wctx.substate if m(s))
2993
2993
2994 if not m.always():
2994 if not m.always():
2995 for abs in repo.walk(matchmod.badmatch(m, lambda x, y: False)):
2995 for abs in repo.walk(matchmod.badmatch(m, lambda x, y: False)):
2996 names[abs] = m.rel(abs), m.exact(abs)
2996 names[abs] = m.rel(abs), m.exact(abs)
2997
2997
2998 # walk target manifest to fill `names`
2998 # walk target manifest to fill `names`
2999
2999
3000 def badfn(path, msg):
3000 def badfn(path, msg):
3001 if path in names:
3001 if path in names:
3002 return
3002 return
3003 if path in ctx.substate:
3003 if path in ctx.substate:
3004 return
3004 return
3005 path_ = path + '/'
3005 path_ = path + '/'
3006 for f in names:
3006 for f in names:
3007 if f.startswith(path_):
3007 if f.startswith(path_):
3008 return
3008 return
3009 ui.warn("%s: %s\n" % (m.rel(path), msg))
3009 ui.warn("%s: %s\n" % (m.rel(path), msg))
3010
3010
3011 for abs in ctx.walk(matchmod.badmatch(m, badfn)):
3011 for abs in ctx.walk(matchmod.badmatch(m, badfn)):
3012 if abs not in names:
3012 if abs not in names:
3013 names[abs] = m.rel(abs), m.exact(abs)
3013 names[abs] = m.rel(abs), m.exact(abs)
3014
3014
3015 # Find status of all file in `names`.
3015 # Find status of all file in `names`.
3016 m = scmutil.matchfiles(repo, names)
3016 m = scmutil.matchfiles(repo, names)
3017
3017
3018 changes = repo.status(node1=node, match=m,
3018 changes = repo.status(node1=node, match=m,
3019 unknown=True, ignored=True, clean=True)
3019 unknown=True, ignored=True, clean=True)
3020 else:
3020 else:
3021 changes = repo.status(node1=node, match=m)
3021 changes = repo.status(node1=node, match=m)
3022 for kind in changes:
3022 for kind in changes:
3023 for abs in kind:
3023 for abs in kind:
3024 names[abs] = m.rel(abs), m.exact(abs)
3024 names[abs] = m.rel(abs), m.exact(abs)
3025
3025
3026 m = scmutil.matchfiles(repo, names)
3026 m = scmutil.matchfiles(repo, names)
3027
3027
3028 modified = set(changes.modified)
3028 modified = set(changes.modified)
3029 added = set(changes.added)
3029 added = set(changes.added)
3030 removed = set(changes.removed)
3030 removed = set(changes.removed)
3031 _deleted = set(changes.deleted)
3031 _deleted = set(changes.deleted)
3032 unknown = set(changes.unknown)
3032 unknown = set(changes.unknown)
3033 unknown.update(changes.ignored)
3033 unknown.update(changes.ignored)
3034 clean = set(changes.clean)
3034 clean = set(changes.clean)
3035 modadded = set()
3035 modadded = set()
3036
3036
3037 # split between files known in target manifest and the others
3037 # split between files known in target manifest and the others
3038 smf = set(mf)
3038 smf = set(mf)
3039
3039
3040 # determine the exact nature of the deleted changesets
3040 # determine the exact nature of the deleted changesets
3041 deladded = _deleted - smf
3041 deladded = _deleted - smf
3042 deleted = _deleted - deladded
3042 deleted = _deleted - deladded
3043
3043
3044 # We need to account for the state of the file in the dirstate,
3044 # We need to account for the state of the file in the dirstate,
3045 # even when we revert against something else than parent. This will
3045 # even when we revert against something else than parent. This will
3046 # slightly alter the behavior of revert (doing back up or not, delete
3046 # slightly alter the behavior of revert (doing back up or not, delete
3047 # or just forget etc).
3047 # or just forget etc).
3048 if parent == node:
3048 if parent == node:
3049 dsmodified = modified
3049 dsmodified = modified
3050 dsadded = added
3050 dsadded = added
3051 dsremoved = removed
3051 dsremoved = removed
3052 # store all local modifications, useful later for rename detection
3052 # store all local modifications, useful later for rename detection
3053 localchanges = dsmodified | dsadded
3053 localchanges = dsmodified | dsadded
3054 modified, added, removed = set(), set(), set()
3054 modified, added, removed = set(), set(), set()
3055 else:
3055 else:
3056 changes = repo.status(node1=parent, match=m)
3056 changes = repo.status(node1=parent, match=m)
3057 dsmodified = set(changes.modified)
3057 dsmodified = set(changes.modified)
3058 dsadded = set(changes.added)
3058 dsadded = set(changes.added)
3059 dsremoved = set(changes.removed)
3059 dsremoved = set(changes.removed)
3060 # store all local modifications, useful later for rename detection
3060 # store all local modifications, useful later for rename detection
3061 localchanges = dsmodified | dsadded
3061 localchanges = dsmodified | dsadded
3062
3062
3063 # only take into account for removes between wc and target
3063 # only take into account for removes between wc and target
3064 clean |= dsremoved - removed
3064 clean |= dsremoved - removed
3065 dsremoved &= removed
3065 dsremoved &= removed
3066 # distinct between dirstate remove and other
3066 # distinct between dirstate remove and other
3067 removed -= dsremoved
3067 removed -= dsremoved
3068
3068
3069 modadded = added & dsmodified
3069 modadded = added & dsmodified
3070 added -= modadded
3070 added -= modadded
3071
3071
3072 # tell newly modified apart.
3072 # tell newly modified apart.
3073 dsmodified &= modified
3073 dsmodified &= modified
3074 dsmodified |= modified & dsadded # dirstate added may need backup
3074 dsmodified |= modified & dsadded # dirstate added may need backup
3075 modified -= dsmodified
3075 modified -= dsmodified
3076
3076
3077 # We need to wait for some post-processing to update this set
3077 # We need to wait for some post-processing to update this set
3078 # before making the distinction. The dirstate will be used for
3078 # before making the distinction. The dirstate will be used for
3079 # that purpose.
3079 # that purpose.
3080 dsadded = added
3080 dsadded = added
3081
3081
3082 # in case of merge, files that are actually added can be reported as
3082 # in case of merge, files that are actually added can be reported as
3083 # modified, we need to post process the result
3083 # modified, we need to post process the result
3084 if p2 != nullid:
3084 if p2 != nullid:
3085 mergeadd = dsmodified - smf
3085 mergeadd = dsmodified - smf
3086 dsadded |= mergeadd
3086 dsadded |= mergeadd
3087 dsmodified -= mergeadd
3087 dsmodified -= mergeadd
3088
3088
3089 # if f is a rename, update `names` to also revert the source
3089 # if f is a rename, update `names` to also revert the source
3090 cwd = repo.getcwd()
3090 cwd = repo.getcwd()
3091 for f in localchanges:
3091 for f in localchanges:
3092 src = repo.dirstate.copied(f)
3092 src = repo.dirstate.copied(f)
3093 # XXX should we check for rename down to target node?
3093 # XXX should we check for rename down to target node?
3094 if src and src not in names and repo.dirstate[src] == 'r':
3094 if src and src not in names and repo.dirstate[src] == 'r':
3095 dsremoved.add(src)
3095 dsremoved.add(src)
3096 names[src] = (repo.pathto(src, cwd), True)
3096 names[src] = (repo.pathto(src, cwd), True)
3097
3097
3098 # distinguish between file to forget and the other
3098 # distinguish between file to forget and the other
3099 added = set()
3099 added = set()
3100 for abs in dsadded:
3100 for abs in dsadded:
3101 if repo.dirstate[abs] != 'a':
3101 if repo.dirstate[abs] != 'a':
3102 added.add(abs)
3102 added.add(abs)
3103 dsadded -= added
3103 dsadded -= added
3104
3104
3105 for abs in deladded:
3105 for abs in deladded:
3106 if repo.dirstate[abs] == 'a':
3106 if repo.dirstate[abs] == 'a':
3107 dsadded.add(abs)
3107 dsadded.add(abs)
3108 deladded -= dsadded
3108 deladded -= dsadded
3109
3109
3110 # For files marked as removed, we check if an unknown file is present at
3110 # For files marked as removed, we check if an unknown file is present at
3111 # the same path. If a such file exists it may need to be backed up.
3111 # the same path. If a such file exists it may need to be backed up.
3112 # Making the distinction at this stage helps have simpler backup
3112 # Making the distinction at this stage helps have simpler backup
3113 # logic.
3113 # logic.
3114 removunk = set()
3114 removunk = set()
3115 for abs in removed:
3115 for abs in removed:
3116 target = repo.wjoin(abs)
3116 target = repo.wjoin(abs)
3117 if os.path.lexists(target):
3117 if os.path.lexists(target):
3118 removunk.add(abs)
3118 removunk.add(abs)
3119 removed -= removunk
3119 removed -= removunk
3120
3120
3121 dsremovunk = set()
3121 dsremovunk = set()
3122 for abs in dsremoved:
3122 for abs in dsremoved:
3123 target = repo.wjoin(abs)
3123 target = repo.wjoin(abs)
3124 if os.path.lexists(target):
3124 if os.path.lexists(target):
3125 dsremovunk.add(abs)
3125 dsremovunk.add(abs)
3126 dsremoved -= dsremovunk
3126 dsremoved -= dsremovunk
3127
3127
3128 # action to be actually performed by revert
3128 # action to be actually performed by revert
3129 # (<list of file>, message>) tuple
3129 # (<list of file>, message>) tuple
3130 actions = {'revert': ([], _('reverting %s\n')),
3130 actions = {'revert': ([], _('reverting %s\n')),
3131 'add': ([], _('adding %s\n')),
3131 'add': ([], _('adding %s\n')),
3132 'remove': ([], _('removing %s\n')),
3132 'remove': ([], _('removing %s\n')),
3133 'drop': ([], _('removing %s\n')),
3133 'drop': ([], _('removing %s\n')),
3134 'forget': ([], _('forgetting %s\n')),
3134 'forget': ([], _('forgetting %s\n')),
3135 'undelete': ([], _('undeleting %s\n')),
3135 'undelete': ([], _('undeleting %s\n')),
3136 'noop': (None, _('no changes needed to %s\n')),
3136 'noop': (None, _('no changes needed to %s\n')),
3137 'unknown': (None, _('file not managed: %s\n')),
3137 'unknown': (None, _('file not managed: %s\n')),
3138 }
3138 }
3139
3139
3140 # "constant" that convey the backup strategy.
3140 # "constant" that convey the backup strategy.
3141 # All set to `discard` if `no-backup` is set do avoid checking
3141 # All set to `discard` if `no-backup` is set do avoid checking
3142 # no_backup lower in the code.
3142 # no_backup lower in the code.
3143 # These values are ordered for comparison purposes
3143 # These values are ordered for comparison purposes
3144 backup = 2 # unconditionally do backup
3144 backup = 2 # unconditionally do backup
3145 check = 1 # check if the existing file differs from target
3145 check = 1 # check if the existing file differs from target
3146 discard = 0 # never do backup
3146 discard = 0 # never do backup
3147 if opts.get('no_backup'):
3147 if opts.get('no_backup'):
3148 backup = check = discard
3148 backup = check = discard
3149
3149
3150 backupanddel = actions['remove']
3150 backupanddel = actions['remove']
3151 if not opts.get('no_backup'):
3151 if not opts.get('no_backup'):
3152 backupanddel = actions['drop']
3152 backupanddel = actions['drop']
3153
3153
3154 disptable = (
3154 disptable = (
3155 # dispatch table:
3155 # dispatch table:
3156 # file state
3156 # file state
3157 # action
3157 # action
3158 # make backup
3158 # make backup
3159
3159
3160 ## Sets that results that will change file on disk
3160 ## Sets that results that will change file on disk
3161 # Modified compared to target, no local change
3161 # Modified compared to target, no local change
3162 (modified, actions['revert'], discard),
3162 (modified, actions['revert'], discard),
3163 # Modified compared to target, but local file is deleted
3163 # Modified compared to target, but local file is deleted
3164 (deleted, actions['revert'], discard),
3164 (deleted, actions['revert'], discard),
3165 # Modified compared to target, local change
3165 # Modified compared to target, local change
3166 (dsmodified, actions['revert'], backup),
3166 (dsmodified, actions['revert'], backup),
3167 # Added since target
3167 # Added since target
3168 (added, actions['remove'], discard),
3168 (added, actions['remove'], discard),
3169 # Added in working directory
3169 # Added in working directory
3170 (dsadded, actions['forget'], discard),
3170 (dsadded, actions['forget'], discard),
3171 # Added since target, have local modification
3171 # Added since target, have local modification
3172 (modadded, backupanddel, backup),
3172 (modadded, backupanddel, backup),
3173 # Added since target but file is missing in working directory
3173 # Added since target but file is missing in working directory
3174 (deladded, actions['drop'], discard),
3174 (deladded, actions['drop'], discard),
3175 # Removed since target, before working copy parent
3175 # Removed since target, before working copy parent
3176 (removed, actions['add'], discard),
3176 (removed, actions['add'], discard),
3177 # Same as `removed` but an unknown file exists at the same path
3177 # Same as `removed` but an unknown file exists at the same path
3178 (removunk, actions['add'], check),
3178 (removunk, actions['add'], check),
3179 # Removed since targe, marked as such in working copy parent
3179 # Removed since targe, marked as such in working copy parent
3180 (dsremoved, actions['undelete'], discard),
3180 (dsremoved, actions['undelete'], discard),
3181 # Same as `dsremoved` but an unknown file exists at the same path
3181 # Same as `dsremoved` but an unknown file exists at the same path
3182 (dsremovunk, actions['undelete'], check),
3182 (dsremovunk, actions['undelete'], check),
3183 ## the following sets does not result in any file changes
3183 ## the following sets does not result in any file changes
3184 # File with no modification
3184 # File with no modification
3185 (clean, actions['noop'], discard),
3185 (clean, actions['noop'], discard),
3186 # Existing file, not tracked anywhere
3186 # Existing file, not tracked anywhere
3187 (unknown, actions['unknown'], discard),
3187 (unknown, actions['unknown'], discard),
3188 )
3188 )
3189
3189
3190 for abs, (rel, exact) in sorted(names.items()):
3190 for abs, (rel, exact) in sorted(names.items()):
3191 # target file to be touch on disk (relative to cwd)
3191 # target file to be touch on disk (relative to cwd)
3192 target = repo.wjoin(abs)
3192 target = repo.wjoin(abs)
3193 # search the entry in the dispatch table.
3193 # search the entry in the dispatch table.
3194 # if the file is in any of these sets, it was touched in the working
3194 # if the file is in any of these sets, it was touched in the working
3195 # directory parent and we are sure it needs to be reverted.
3195 # directory parent and we are sure it needs to be reverted.
3196 for table, (xlist, msg), dobackup in disptable:
3196 for table, (xlist, msg), dobackup in disptable:
3197 if abs not in table:
3197 if abs not in table:
3198 continue
3198 continue
3199 if xlist is not None:
3199 if xlist is not None:
3200 xlist.append(abs)
3200 xlist.append(abs)
3201 if dobackup and (backup <= dobackup
3201 if dobackup and (backup <= dobackup
3202 or wctx[abs].cmp(ctx[abs])):
3202 or wctx[abs].cmp(ctx[abs])):
3203 bakname = scmutil.origpath(ui, repo, rel)
3203 bakname = scmutil.origpath(ui, repo, rel)
3204 ui.note(_('saving current version of %s as %s\n') %
3204 ui.note(_('saving current version of %s as %s\n') %
3205 (rel, bakname))
3205 (rel, bakname))
3206 if not opts.get('dry_run'):
3206 if not opts.get('dry_run'):
3207 if interactive:
3207 if interactive:
3208 util.copyfile(target, bakname)
3208 util.copyfile(target, bakname)
3209 else:
3209 else:
3210 util.rename(target, bakname)
3210 util.rename(target, bakname)
3211 if ui.verbose or not exact:
3211 if ui.verbose or not exact:
3212 if not isinstance(msg, basestring):
3212 if not isinstance(msg, basestring):
3213 msg = msg(abs)
3213 msg = msg(abs)
3214 ui.status(msg % rel)
3214 ui.status(msg % rel)
3215 elif exact:
3215 elif exact:
3216 ui.warn(msg % rel)
3216 ui.warn(msg % rel)
3217 break
3217 break
3218
3218
3219 if not opts.get('dry_run'):
3219 if not opts.get('dry_run'):
3220 needdata = ('revert', 'add', 'undelete')
3220 needdata = ('revert', 'add', 'undelete')
3221 _revertprefetch(repo, ctx, *[actions[name][0] for name in needdata])
3221 _revertprefetch(repo, ctx, *[actions[name][0] for name in needdata])
3222 _performrevert(repo, parents, ctx, actions, interactive)
3222 _performrevert(repo, parents, ctx, actions, interactive)
3223
3223
3224 if targetsubs:
3224 if targetsubs:
3225 # Revert the subrepos on the revert list
3225 # Revert the subrepos on the revert list
3226 for sub in targetsubs:
3226 for sub in targetsubs:
3227 try:
3227 try:
3228 wctx.sub(sub).revert(ctx.substate[sub], *pats, **opts)
3228 wctx.sub(sub).revert(ctx.substate[sub], *pats, **opts)
3229 except KeyError:
3229 except KeyError:
3230 raise error.Abort("subrepository '%s' does not exist in %s!"
3230 raise error.Abort("subrepository '%s' does not exist in %s!"
3231 % (sub, short(ctx.node())))
3231 % (sub, short(ctx.node())))
3232
3232
3233 def _revertprefetch(repo, ctx, *files):
3233 def _revertprefetch(repo, ctx, *files):
3234 """Let extension changing the storage layer prefetch content"""
3234 """Let extension changing the storage layer prefetch content"""
3235 pass
3235 pass
3236
3236
3237 def _performrevert(repo, parents, ctx, actions, interactive=False):
3237 def _performrevert(repo, parents, ctx, actions, interactive=False):
3238 """function that actually perform all the actions computed for revert
3238 """function that actually perform all the actions computed for revert
3239
3239
3240 This is an independent function to let extension to plug in and react to
3240 This is an independent function to let extension to plug in and react to
3241 the imminent revert.
3241 the imminent revert.
3242
3242
3243 Make sure you have the working directory locked when calling this function.
3243 Make sure you have the working directory locked when calling this function.
3244 """
3244 """
3245 parent, p2 = parents
3245 parent, p2 = parents
3246 node = ctx.node()
3246 node = ctx.node()
3247 excluded_files = []
3247 excluded_files = []
3248 matcher_opts = {"exclude": excluded_files}
3248 matcher_opts = {"exclude": excluded_files}
3249
3249
3250 def checkout(f):
3250 def checkout(f):
3251 fc = ctx[f]
3251 fc = ctx[f]
3252 repo.wwrite(f, fc.data(), fc.flags())
3252 repo.wwrite(f, fc.data(), fc.flags())
3253
3253
3254 audit_path = pathutil.pathauditor(repo.root)
3254 audit_path = pathutil.pathauditor(repo.root)
3255 for f in actions['forget'][0]:
3255 for f in actions['forget'][0]:
3256 if interactive:
3256 if interactive:
3257 choice = \
3257 choice = \
3258 repo.ui.promptchoice(
3258 repo.ui.promptchoice(
3259 _("forget added file %s (yn)?$$ &Yes $$ &No")
3259 _("forget added file %s (yn)?$$ &Yes $$ &No")
3260 % f)
3260 % f)
3261 if choice == 0:
3261 if choice == 0:
3262 repo.dirstate.drop(f)
3262 repo.dirstate.drop(f)
3263 else:
3263 else:
3264 excluded_files.append(repo.wjoin(f))
3264 excluded_files.append(repo.wjoin(f))
3265 else:
3265 else:
3266 repo.dirstate.drop(f)
3266 repo.dirstate.drop(f)
3267 for f in actions['remove'][0]:
3267 for f in actions['remove'][0]:
3268 audit_path(f)
3268 audit_path(f)
3269 try:
3269 try:
3270 util.unlinkpath(repo.wjoin(f))
3270 util.unlinkpath(repo.wjoin(f))
3271 except OSError:
3271 except OSError:
3272 pass
3272 pass
3273 repo.dirstate.remove(f)
3273 repo.dirstate.remove(f)
3274 for f in actions['drop'][0]:
3274 for f in actions['drop'][0]:
3275 audit_path(f)
3275 audit_path(f)
3276 repo.dirstate.remove(f)
3276 repo.dirstate.remove(f)
3277
3277
3278 normal = None
3278 normal = None
3279 if node == parent:
3279 if node == parent:
3280 # We're reverting to our parent. If possible, we'd like status
3280 # We're reverting to our parent. If possible, we'd like status
3281 # to report the file as clean. We have to use normallookup for
3281 # to report the file as clean. We have to use normallookup for
3282 # merges to avoid losing information about merged/dirty files.
3282 # merges to avoid losing information about merged/dirty files.
3283 if p2 != nullid:
3283 if p2 != nullid:
3284 normal = repo.dirstate.normallookup
3284 normal = repo.dirstate.normallookup
3285 else:
3285 else:
3286 normal = repo.dirstate.normal
3286 normal = repo.dirstate.normal
3287
3287
3288 newlyaddedandmodifiedfiles = set()
3288 newlyaddedandmodifiedfiles = set()
3289 if interactive:
3289 if interactive:
3290 # Prompt the user for changes to revert
3290 # Prompt the user for changes to revert
3291 torevert = [repo.wjoin(f) for f in actions['revert'][0]]
3291 torevert = [repo.wjoin(f) for f in actions['revert'][0]]
3292 m = scmutil.match(ctx, torevert, matcher_opts)
3292 m = scmutil.match(ctx, torevert, matcher_opts)
3293 diffopts = patch.difffeatureopts(repo.ui, whitespace=True)
3293 diffopts = patch.difffeatureopts(repo.ui, whitespace=True)
3294 diffopts.nodates = True
3294 diffopts.nodates = True
3295 diffopts.git = True
3295 diffopts.git = True
3296 reversehunks = repo.ui.configbool('experimental',
3296 reversehunks = repo.ui.configbool('experimental',
3297 'revertalternateinteractivemode',
3297 'revertalternateinteractivemode',
3298 True)
3298 True)
3299 if reversehunks:
3299 if reversehunks:
3300 diff = patch.diff(repo, ctx.node(), None, m, opts=diffopts)
3300 diff = patch.diff(repo, ctx.node(), None, m, opts=diffopts)
3301 else:
3301 else:
3302 diff = patch.diff(repo, None, ctx.node(), m, opts=diffopts)
3302 diff = patch.diff(repo, None, ctx.node(), m, opts=diffopts)
3303 originalchunks = patch.parsepatch(diff)
3303 originalchunks = patch.parsepatch(diff)
3304 operation = 'discard' if node == parent else 'revert'
3304 operation = 'discard' if node == parent else 'revert'
3305
3305
3306 try:
3306 try:
3307
3307
3308 chunks, opts = recordfilter(repo.ui, originalchunks,
3308 chunks, opts = recordfilter(repo.ui, originalchunks,
3309 operation=operation)
3309 operation=operation)
3310 if reversehunks:
3310 if reversehunks:
3311 chunks = patch.reversehunks(chunks)
3311 chunks = patch.reversehunks(chunks)
3312
3312
3313 except patch.PatchError as err:
3313 except patch.PatchError as err:
3314 raise error.Abort(_('error parsing patch: %s') % err)
3314 raise error.Abort(_('error parsing patch: %s') % err)
3315
3315
3316 newlyaddedandmodifiedfiles = newandmodified(chunks, originalchunks)
3316 newlyaddedandmodifiedfiles = newandmodified(chunks, originalchunks)
3317 # Apply changes
3317 # Apply changes
3318 fp = stringio()
3318 fp = stringio()
3319 for c in chunks:
3319 for c in chunks:
3320 c.write(fp)
3320 c.write(fp)
3321 dopatch = fp.tell()
3321 dopatch = fp.tell()
3322 fp.seek(0)
3322 fp.seek(0)
3323 if dopatch:
3323 if dopatch:
3324 try:
3324 try:
3325 patch.internalpatch(repo.ui, repo, fp, 1, eolmode=None)
3325 patch.internalpatch(repo.ui, repo, fp, 1, eolmode=None)
3326 except patch.PatchError as err:
3326 except patch.PatchError as err:
3327 raise error.Abort(str(err))
3327 raise error.Abort(str(err))
3328 del fp
3328 del fp
3329 else:
3329 else:
3330 for f in actions['revert'][0]:
3330 for f in actions['revert'][0]:
3331 checkout(f)
3331 checkout(f)
3332 if normal:
3332 if normal:
3333 normal(f)
3333 normal(f)
3334
3334
3335 for f in actions['add'][0]:
3335 for f in actions['add'][0]:
3336 # Don't checkout modified files, they are already created by the diff
3336 # Don't checkout modified files, they are already created by the diff
3337 if f not in newlyaddedandmodifiedfiles:
3337 if f not in newlyaddedandmodifiedfiles:
3338 checkout(f)
3338 checkout(f)
3339 repo.dirstate.add(f)
3339 repo.dirstate.add(f)
3340
3340
3341 normal = repo.dirstate.normallookup
3341 normal = repo.dirstate.normallookup
3342 if node == parent and p2 == nullid:
3342 if node == parent and p2 == nullid:
3343 normal = repo.dirstate.normal
3343 normal = repo.dirstate.normal
3344 for f in actions['undelete'][0]:
3344 for f in actions['undelete'][0]:
3345 checkout(f)
3345 checkout(f)
3346 normal(f)
3346 normal(f)
3347
3347
3348 copied = copies.pathcopies(repo[parent], ctx)
3348 copied = copies.pathcopies(repo[parent], ctx)
3349
3349
3350 for f in actions['add'][0] + actions['undelete'][0] + actions['revert'][0]:
3350 for f in actions['add'][0] + actions['undelete'][0] + actions['revert'][0]:
3351 if f in copied:
3351 if f in copied:
3352 repo.dirstate.copy(copied[f], f)
3352 repo.dirstate.copy(copied[f], f)
3353
3353
3354 def command(table):
3354 def command(table):
3355 """Returns a function object to be used as a decorator for making commands.
3355 """Returns a function object to be used as a decorator for making commands.
3356
3356
3357 This function receives a command table as its argument. The table should
3357 This function receives a command table as its argument. The table should
3358 be a dict.
3358 be a dict.
3359
3359
3360 The returned function can be used as a decorator for adding commands
3360 The returned function can be used as a decorator for adding commands
3361 to that command table. This function accepts multiple arguments to define
3361 to that command table. This function accepts multiple arguments to define
3362 a command.
3362 a command.
3363
3363
3364 The first argument is the command name.
3364 The first argument is the command name.
3365
3365
3366 The options argument is an iterable of tuples defining command arguments.
3366 The options argument is an iterable of tuples defining command arguments.
3367 See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.
3367 See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.
3368
3368
3369 The synopsis argument defines a short, one line summary of how to use the
3369 The synopsis argument defines a short, one line summary of how to use the
3370 command. This shows up in the help output.
3370 command. This shows up in the help output.
3371
3371
3372 The norepo argument defines whether the command does not require a
3372 The norepo argument defines whether the command does not require a
3373 local repository. Most commands operate against a repository, thus the
3373 local repository. Most commands operate against a repository, thus the
3374 default is False.
3374 default is False.
3375
3375
3376 The optionalrepo argument defines whether the command optionally requires
3376 The optionalrepo argument defines whether the command optionally requires
3377 a local repository.
3377 a local repository.
3378
3378
3379 The inferrepo argument defines whether to try to find a repository from the
3379 The inferrepo argument defines whether to try to find a repository from the
3380 command line arguments. If True, arguments will be examined for potential
3380 command line arguments. If True, arguments will be examined for potential
3381 repository locations. See ``findrepo()``. If a repository is found, it
3381 repository locations. See ``findrepo()``. If a repository is found, it
3382 will be used.
3382 will be used.
3383 """
3383 """
3384 def cmd(name, options=(), synopsis=None, norepo=False, optionalrepo=False,
3384 def cmd(name, options=(), synopsis=None, norepo=False, optionalrepo=False,
3385 inferrepo=False):
3385 inferrepo=False):
3386 def decorator(func):
3386 def decorator(func):
3387 func.norepo = norepo
3387 func.norepo = norepo
3388 func.optionalrepo = optionalrepo
3388 func.optionalrepo = optionalrepo
3389 func.inferrepo = inferrepo
3389 func.inferrepo = inferrepo
3390 if synopsis:
3390 if synopsis:
3391 table[name] = func, list(options), synopsis
3391 table[name] = func, list(options), synopsis
3392 else:
3392 else:
3393 table[name] = func, list(options)
3393 table[name] = func, list(options)
3394 return func
3394 return func
3395 return decorator
3395 return decorator
3396
3396
3397 return cmd
3397 return cmd
3398
3398
3399 # a list of (ui, repo, otherpeer, opts, missing) functions called by
3399 # a list of (ui, repo, otherpeer, opts, missing) functions called by
3400 # commands.outgoing. "missing" is "missing" of the result of
3400 # commands.outgoing. "missing" is "missing" of the result of
3401 # "findcommonoutgoing()"
3401 # "findcommonoutgoing()"
3402 outgoinghooks = util.hooks()
3402 outgoinghooks = util.hooks()
3403
3403
3404 # a list of (ui, repo) functions called by commands.summary
3404 # a list of (ui, repo) functions called by commands.summary
3405 summaryhooks = util.hooks()
3405 summaryhooks = util.hooks()
3406
3406
3407 # a list of (ui, repo, opts, changes) functions called by commands.summary.
3407 # a list of (ui, repo, opts, changes) functions called by commands.summary.
3408 #
3408 #
3409 # functions should return tuple of booleans below, if 'changes' is None:
3409 # functions should return tuple of booleans below, if 'changes' is None:
3410 # (whether-incomings-are-needed, whether-outgoings-are-needed)
3410 # (whether-incomings-are-needed, whether-outgoings-are-needed)
3411 #
3411 #
3412 # otherwise, 'changes' is a tuple of tuples below:
3412 # otherwise, 'changes' is a tuple of tuples below:
3413 # - (sourceurl, sourcebranch, sourcepeer, incoming)
3413 # - (sourceurl, sourcebranch, sourcepeer, incoming)
3414 # - (desturl, destbranch, destpeer, outgoing)
3414 # - (desturl, destbranch, destpeer, outgoing)
3415 summaryremotehooks = util.hooks()
3415 summaryremotehooks = util.hooks()
3416
3416
3417 # A list of state files kept by multistep operations like graft.
3417 # A list of state files kept by multistep operations like graft.
3418 # Since graft cannot be aborted, it is considered 'clearable' by update.
3418 # Since graft cannot be aborted, it is considered 'clearable' by update.
3419 # note: bisect is intentionally excluded
3419 # note: bisect is intentionally excluded
3420 # (state file, clearable, allowcommit, error, hint)
3420 # (state file, clearable, allowcommit, error, hint)
3421 unfinishedstates = [
3421 unfinishedstates = [
3422 ('graftstate', True, False, _('graft in progress'),
3422 ('graftstate', True, False, _('graft in progress'),
3423 _("use 'hg graft --continue' or 'hg update' to abort")),
3423 _("use 'hg graft --continue' or 'hg update' to abort")),
3424 ('updatestate', True, False, _('last update was interrupted'),
3424 ('updatestate', True, False, _('last update was interrupted'),
3425 _("use 'hg update' to get a consistent checkout"))
3425 _("use 'hg update' to get a consistent checkout"))
3426 ]
3426 ]
3427
3427
3428 def checkunfinished(repo, commit=False):
3428 def checkunfinished(repo, commit=False):
3429 '''Look for an unfinished multistep operation, like graft, and abort
3429 '''Look for an unfinished multistep operation, like graft, and abort
3430 if found. It's probably good to check this right before
3430 if found. It's probably good to check this right before
3431 bailifchanged().
3431 bailifchanged().
3432 '''
3432 '''
3433 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3433 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3434 if commit and allowcommit:
3434 if commit and allowcommit:
3435 continue
3435 continue
3436 if repo.vfs.exists(f):
3436 if repo.vfs.exists(f):
3437 raise error.Abort(msg, hint=hint)
3437 raise error.Abort(msg, hint=hint)
3438
3438
3439 def clearunfinished(repo):
3439 def clearunfinished(repo):
3440 '''Check for unfinished operations (as above), and clear the ones
3440 '''Check for unfinished operations (as above), and clear the ones
3441 that are clearable.
3441 that are clearable.
3442 '''
3442 '''
3443 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3443 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3444 if not clearable and repo.vfs.exists(f):
3444 if not clearable and repo.vfs.exists(f):
3445 raise error.Abort(msg, hint=hint)
3445 raise error.Abort(msg, hint=hint)
3446 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3446 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3447 if clearable and repo.vfs.exists(f):
3447 if clearable and repo.vfs.exists(f):
3448 util.unlink(repo.join(f))
3448 util.unlink(repo.join(f))
3449
3449
3450 afterresolvedstates = [
3450 afterresolvedstates = [
3451 ('graftstate',
3451 ('graftstate',
3452 _('hg graft --continue')),
3452 _('hg graft --continue')),
3453 ]
3453 ]
3454
3454
3455 def howtocontinue(repo):
3455 def howtocontinue(repo):
3456 '''Check for an unfinished operation and return the command to finish
3456 '''Check for an unfinished operation and return the command to finish
3457 it.
3457 it.
3458
3458
3459 afterresolvedstates tupples define a .hg/{file} and the corresponding
3459 afterresolvedstates tupples define a .hg/{file} and the corresponding
3460 command needed to finish it.
3460 command needed to finish it.
3461
3461
3462 Returns a (msg, warning) tuple. 'msg' is a string and 'warning' is
3462 Returns a (msg, warning) tuple. 'msg' is a string and 'warning' is
3463 a boolean.
3463 a boolean.
3464 '''
3464 '''
3465 contmsg = _("continue: %s")
3465 contmsg = _("continue: %s")
3466 for f, msg in afterresolvedstates:
3466 for f, msg in afterresolvedstates:
3467 if repo.vfs.exists(f):
3467 if repo.vfs.exists(f):
3468 return contmsg % msg, True
3468 return contmsg % msg, True
3469 workingctx = repo[None]
3469 workingctx = repo[None]
3470 dirty = any(repo.status()) or any(workingctx.sub(s).dirty()
3470 dirty = any(repo.status()) or any(workingctx.sub(s).dirty()
3471 for s in workingctx.substate)
3471 for s in workingctx.substate)
3472 if dirty:
3472 if dirty:
3473 return contmsg % _("hg commit"), False
3473 return contmsg % _("hg commit"), False
3474 return None, None
3474 return None, None
3475
3475
3476 def checkafterresolved(repo):
3476 def checkafterresolved(repo):
3477 '''Inform the user about the next action after completing hg resolve
3477 '''Inform the user about the next action after completing hg resolve
3478
3478
3479 If there's a matching afterresolvedstates, howtocontinue will yield
3479 If there's a matching afterresolvedstates, howtocontinue will yield
3480 repo.ui.warn as the reporter.
3480 repo.ui.warn as the reporter.
3481
3481
3482 Otherwise, it will yield repo.ui.note.
3482 Otherwise, it will yield repo.ui.note.
3483 '''
3483 '''
3484 msg, warning = howtocontinue(repo)
3484 msg, warning = howtocontinue(repo)
3485 if msg is not None:
3485 if msg is not None:
3486 if warning:
3486 if warning:
3487 repo.ui.warn("%s\n" % msg)
3487 repo.ui.warn("%s\n" % msg)
3488 else:
3488 else:
3489 repo.ui.note("%s\n" % msg)
3489 repo.ui.note("%s\n" % msg)
3490
3490
3491 def wrongtooltocontinue(repo, task):
3491 def wrongtooltocontinue(repo, task):
3492 '''Raise an abort suggesting how to properly continue if there is an
3492 '''Raise an abort suggesting how to properly continue if there is an
3493 active task.
3493 active task.
3494
3494
3495 Uses howtocontinue() to find the active task.
3495 Uses howtocontinue() to find the active task.
3496
3496
3497 If there's no task (repo.ui.note for 'hg commit'), it does not offer
3497 If there's no task (repo.ui.note for 'hg commit'), it does not offer
3498 a hint.
3498 a hint.
3499 '''
3499 '''
3500 after = howtocontinue(repo)
3500 after = howtocontinue(repo)
3501 hint = None
3501 hint = None
3502 if after[1]:
3502 if after[1]:
3503 hint = after[0]
3503 hint = after[0]
3504 raise error.Abort(_('no %s in progress') % task, hint=hint)
3504 raise error.Abort(_('no %s in progress') % task, hint=hint)
3505
3505
3506 class dirstateguard(object):
3506 class dirstateguard(object):
3507 '''Restore dirstate at unexpected failure.
3507 '''Restore dirstate at unexpected failure.
3508
3508
3509 At the construction, this class does:
3509 At the construction, this class does:
3510
3510
3511 - write current ``repo.dirstate`` out, and
3511 - write current ``repo.dirstate`` out, and
3512 - save ``.hg/dirstate`` into the backup file
3512 - save ``.hg/dirstate`` into the backup file
3513
3513
3514 This restores ``.hg/dirstate`` from backup file, if ``release()``
3514 This restores ``.hg/dirstate`` from backup file, if ``release()``
3515 is invoked before ``close()``.
3515 is invoked before ``close()``.
3516
3516
3517 This just removes the backup file at ``close()`` before ``release()``.
3517 This just removes the backup file at ``close()`` before ``release()``.
3518 '''
3518 '''
3519
3519
3520 def __init__(self, repo, name):
3520 def __init__(self, repo, name):
3521 self._repo = repo
3521 self._repo = repo
3522 self._suffix = '.backup.%s.%d' % (name, id(self))
3522 self._suffix = '.backup.%s.%d' % (name, id(self))
3523 repo.dirstate.savebackup(repo.currenttransaction(), self._suffix)
3523 repo.dirstate.savebackup(repo.currenttransaction(), self._suffix)
3524 self._active = True
3524 self._active = True
3525 self._closed = False
3525 self._closed = False
3526
3526
3527 def __del__(self):
3527 def __del__(self):
3528 if self._active: # still active
3528 if self._active: # still active
3529 # this may occur, even if this class is used correctly:
3529 # this may occur, even if this class is used correctly:
3530 # for example, releasing other resources like transaction
3530 # for example, releasing other resources like transaction
3531 # may raise exception before ``dirstateguard.release`` in
3531 # may raise exception before ``dirstateguard.release`` in
3532 # ``release(tr, ....)``.
3532 # ``release(tr, ....)``.
3533 self._abort()
3533 self._abort()
3534
3534
3535 def close(self):
3535 def close(self):
3536 if not self._active: # already inactivated
3536 if not self._active: # already inactivated
3537 msg = (_("can't close already inactivated backup: dirstate%s")
3537 msg = (_("can't close already inactivated backup: dirstate%s")
3538 % self._suffix)
3538 % self._suffix)
3539 raise error.Abort(msg)
3539 raise error.Abort(msg)
3540
3540
3541 self._repo.dirstate.clearbackup(self._repo.currenttransaction(),
3541 self._repo.dirstate.clearbackup(self._repo.currenttransaction(),
3542 self._suffix)
3542 self._suffix)
3543 self._active = False
3543 self._active = False
3544 self._closed = True
3544 self._closed = True
3545
3545
3546 def _abort(self):
3546 def _abort(self):
3547 self._repo.dirstate.restorebackup(self._repo.currenttransaction(),
3547 self._repo.dirstate.restorebackup(self._repo.currenttransaction(),
3548 self._suffix)
3548 self._suffix)
3549 self._active = False
3549 self._active = False
3550
3550
3551 def release(self):
3551 def release(self):
3552 if not self._closed:
3552 if not self._closed:
3553 if not self._active: # already inactivated
3553 if not self._active: # already inactivated
3554 msg = (_("can't release already inactivated backup:"
3554 msg = (_("can't release already inactivated backup:"
3555 " dirstate%s")
3555 " dirstate%s")
3556 % self._suffix)
3556 % self._suffix)
3557 raise error.Abort(msg)
3557 raise error.Abort(msg)
3558 self._abort()
3558 self._abort()
@@ -1,1645 +1,1642 b''
1 # stuff related specifically to patch manipulation / parsing
1 # stuff related specifically to patch manipulation / parsing
2 #
2 #
3 # Copyright 2008 Mark Edgington <edgimar@gmail.com>
3 # Copyright 2008 Mark Edgington <edgimar@gmail.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 #
7 #
8 # This code is based on the Mark Edgington's crecord extension.
8 # This code is based on the Mark Edgington's crecord extension.
9 # (Itself based on Bryan O'Sullivan's record extension.)
9 # (Itself based on Bryan O'Sullivan's record extension.)
10
10
11 from __future__ import absolute_import
11 from __future__ import absolute_import
12
12
13 import locale
13 import locale
14 import os
14 import os
15 import re
15 import re
16 import signal
16 import signal
17 import struct
17 import struct
18 import sys
18 import sys
19
19
20 from .i18n import _
20 from .i18n import _
21 from . import (
21 from . import (
22 encoding,
22 encoding,
23 error,
23 error,
24 patch as patchmod,
24 patch as patchmod,
25 util,
25 util,
26 )
26 )
27 stringio = util.stringio
27 stringio = util.stringio
28
28
29 # This is required for ncurses to display non-ASCII characters in default user
29 # This is required for ncurses to display non-ASCII characters in default user
30 # locale encoding correctly. --immerrr
30 # locale encoding correctly. --immerrr
31 locale.setlocale(locale.LC_ALL, '')
31 locale.setlocale(locale.LC_ALL, '')
32
32
33 # patch comments based on the git one
33 # patch comments based on the git one
34 diffhelptext = _("""# To remove '-' lines, make them ' ' lines (context).
34 diffhelptext = _("""# To remove '-' lines, make them ' ' lines (context).
35 # To remove '+' lines, delete them.
35 # To remove '+' lines, delete them.
36 # Lines starting with # will be removed from the patch.
36 # Lines starting with # will be removed from the patch.
37 """)
37 """)
38
38
39 hunkhelptext = _("""#
39 hunkhelptext = _("""#
40 # If the patch applies cleanly, the edited hunk will immediately be
40 # If the patch applies cleanly, the edited hunk will immediately be
41 # added to the record list. If it does not apply cleanly, a rejects file
41 # added to the record list. If it does not apply cleanly, a rejects file
42 # will be generated. You can use that when you try again. If all lines
42 # will be generated. You can use that when you try again. If all lines
43 # of the hunk are removed, then the edit is aborted and the hunk is left
43 # of the hunk are removed, then the edit is aborted and the hunk is left
44 # unchanged.
44 # unchanged.
45 """)
45 """)
46
46
47 patchhelptext = _("""#
47 patchhelptext = _("""#
48 # If the patch applies cleanly, the edited patch will immediately
48 # If the patch applies cleanly, the edited patch will immediately
49 # be finalised. If it does not apply cleanly, rejects files will be
49 # be finalised. If it does not apply cleanly, rejects files will be
50 # generated. You can use those when you try again.
50 # generated. You can use those when you try again.
51 """)
51 """)
52
52
53 try:
53 try:
54 import curses
54 import curses
55 import fcntl
55 import fcntl
56 import termios
56 import termios
57 curses.error
57 curses.error
58 fcntl.ioctl
58 fcntl.ioctl
59 termios.TIOCGWINSZ
59 termios.TIOCGWINSZ
60 except ImportError:
60 except ImportError:
61 # I have no idea if wcurses works with crecord...
61 # I have no idea if wcurses works with crecord...
62 try:
62 try:
63 import wcurses as curses
63 import wcurses as curses
64 curses.error
64 curses.error
65 except ImportError:
65 except ImportError:
66 # wcurses is not shipped on Windows by default, or python is not
66 # wcurses is not shipped on Windows by default, or python is not
67 # compiled with curses
67 # compiled with curses
68 curses = False
68 curses = False
69
69
70 def checkcurses(ui):
70 def checkcurses(ui):
71 """Return True if the user wants to use curses
71 """Return True if the user wants to use curses
72
72
73 This method returns True if curses is found (and that python is built with
73 This method returns True if curses is found (and that python is built with
74 it) and that the user has the correct flag for the ui.
74 it) and that the user has the correct flag for the ui.
75 """
75 """
76 return curses and ui.interface("chunkselector") == "curses"
76 return curses and ui.interface("chunkselector") == "curses"
77
77
78 _origstdout = sys.__stdout__ # used by gethw()
78 _origstdout = sys.__stdout__ # used by gethw()
79
79
80 class patchnode(object):
80 class patchnode(object):
81 """abstract class for patch graph nodes
81 """abstract class for patch graph nodes
82 (i.e. patchroot, header, hunk, hunkline)
82 (i.e. patchroot, header, hunk, hunkline)
83 """
83 """
84
84
85 def firstchild(self):
85 def firstchild(self):
86 raise NotImplementedError("method must be implemented by subclass")
86 raise NotImplementedError("method must be implemented by subclass")
87
87
88 def lastchild(self):
88 def lastchild(self):
89 raise NotImplementedError("method must be implemented by subclass")
89 raise NotImplementedError("method must be implemented by subclass")
90
90
91 def allchildren(self):
91 def allchildren(self):
92 "Return a list of all of the direct children of this node"
92 "Return a list of all of the direct children of this node"
93 raise NotImplementedError("method must be implemented by subclass")
93 raise NotImplementedError("method must be implemented by subclass")
94
94
95 def nextsibling(self):
95 def nextsibling(self):
96 """
96 """
97 Return the closest next item of the same type where there are no items
97 Return the closest next item of the same type where there are no items
98 of different types between the current item and this closest item.
98 of different types between the current item and this closest item.
99 If no such item exists, return None.
99 If no such item exists, return None.
100 """
100 """
101 raise NotImplementedError("method must be implemented by subclass")
101 raise NotImplementedError("method must be implemented by subclass")
102
102
103 def prevsibling(self):
103 def prevsibling(self):
104 """
104 """
105 Return the closest previous item of the same type where there are no
105 Return the closest previous item of the same type where there are no
106 items of different types between the current item and this closest item.
106 items of different types between the current item and this closest item.
107 If no such item exists, return None.
107 If no such item exists, return None.
108 """
108 """
109 raise NotImplementedError("method must be implemented by subclass")
109 raise NotImplementedError("method must be implemented by subclass")
110
110
111 def parentitem(self):
111 def parentitem(self):
112 raise NotImplementedError("method must be implemented by subclass")
112 raise NotImplementedError("method must be implemented by subclass")
113
113
114 def nextitem(self, skipfolded=True):
114 def nextitem(self, skipfolded=True):
115 """
115 """
116 Try to return the next item closest to this item, regardless of item's
116 Try to return the next item closest to this item, regardless of item's
117 type (header, hunk, or hunkline).
117 type (header, hunk, or hunkline).
118
118
119 If skipfolded == True, and the current item is folded, then the child
119 If skipfolded == True, and the current item is folded, then the child
120 items that are hidden due to folding will be skipped when determining
120 items that are hidden due to folding will be skipped when determining
121 the next item.
121 the next item.
122
122
123 If it is not possible to get the next item, return None.
123 If it is not possible to get the next item, return None.
124 """
124 """
125 try:
125 try:
126 itemfolded = self.folded
126 itemfolded = self.folded
127 except AttributeError:
127 except AttributeError:
128 itemfolded = False
128 itemfolded = False
129 if skipfolded and itemfolded:
129 if skipfolded and itemfolded:
130 nextitem = self.nextsibling()
130 nextitem = self.nextsibling()
131 if nextitem is None:
131 if nextitem is None:
132 try:
132 try:
133 nextitem = self.parentitem().nextsibling()
133 nextitem = self.parentitem().nextsibling()
134 except AttributeError:
134 except AttributeError:
135 nextitem = None
135 nextitem = None
136 return nextitem
136 return nextitem
137 else:
137 else:
138 # try child
138 # try child
139 item = self.firstchild()
139 item = self.firstchild()
140 if item is not None:
140 if item is not None:
141 return item
141 return item
142
142
143 # else try next sibling
143 # else try next sibling
144 item = self.nextsibling()
144 item = self.nextsibling()
145 if item is not None:
145 if item is not None:
146 return item
146 return item
147
147
148 try:
148 try:
149 # else try parent's next sibling
149 # else try parent's next sibling
150 item = self.parentitem().nextsibling()
150 item = self.parentitem().nextsibling()
151 if item is not None:
151 if item is not None:
152 return item
152 return item
153
153
154 # else return grandparent's next sibling (or None)
154 # else return grandparent's next sibling (or None)
155 return self.parentitem().parentitem().nextsibling()
155 return self.parentitem().parentitem().nextsibling()
156
156
157 except AttributeError: # parent and/or grandparent was None
157 except AttributeError: # parent and/or grandparent was None
158 return None
158 return None
159
159
160 def previtem(self):
160 def previtem(self):
161 """
161 """
162 Try to return the previous item closest to this item, regardless of
162 Try to return the previous item closest to this item, regardless of
163 item's type (header, hunk, or hunkline).
163 item's type (header, hunk, or hunkline).
164
164
165 If it is not possible to get the previous item, return None.
165 If it is not possible to get the previous item, return None.
166 """
166 """
167 # try previous sibling's last child's last child,
167 # try previous sibling's last child's last child,
168 # else try previous sibling's last child, else try previous sibling
168 # else try previous sibling's last child, else try previous sibling
169 prevsibling = self.prevsibling()
169 prevsibling = self.prevsibling()
170 if prevsibling is not None:
170 if prevsibling is not None:
171 prevsiblinglastchild = prevsibling.lastchild()
171 prevsiblinglastchild = prevsibling.lastchild()
172 if ((prevsiblinglastchild is not None) and
172 if ((prevsiblinglastchild is not None) and
173 not prevsibling.folded):
173 not prevsibling.folded):
174 prevsiblinglclc = prevsiblinglastchild.lastchild()
174 prevsiblinglclc = prevsiblinglastchild.lastchild()
175 if ((prevsiblinglclc is not None) and
175 if ((prevsiblinglclc is not None) and
176 not prevsiblinglastchild.folded):
176 not prevsiblinglastchild.folded):
177 return prevsiblinglclc
177 return prevsiblinglclc
178 else:
178 else:
179 return prevsiblinglastchild
179 return prevsiblinglastchild
180 else:
180 else:
181 return prevsibling
181 return prevsibling
182
182
183 # try parent (or None)
183 # try parent (or None)
184 return self.parentitem()
184 return self.parentitem()
185
185
186 class patch(patchnode, list): # todo: rename patchroot
186 class patch(patchnode, list): # todo: rename patchroot
187 """
187 """
188 list of header objects representing the patch.
188 list of header objects representing the patch.
189 """
189 """
190 def __init__(self, headerlist):
190 def __init__(self, headerlist):
191 self.extend(headerlist)
191 self.extend(headerlist)
192 # add parent patch object reference to each header
192 # add parent patch object reference to each header
193 for header in self:
193 for header in self:
194 header.patch = self
194 header.patch = self
195
195
196 class uiheader(patchnode):
196 class uiheader(patchnode):
197 """patch header
197 """patch header
198
198
199 xxx shouldn't we move this to mercurial/patch.py ?
199 xxx shouldn't we move this to mercurial/patch.py ?
200 """
200 """
201
201
202 def __init__(self, header):
202 def __init__(self, header):
203 self.nonuiheader = header
203 self.nonuiheader = header
204 # flag to indicate whether to apply this chunk
204 # flag to indicate whether to apply this chunk
205 self.applied = True
205 self.applied = True
206 # flag which only affects the status display indicating if a node's
206 # flag which only affects the status display indicating if a node's
207 # children are partially applied (i.e. some applied, some not).
207 # children are partially applied (i.e. some applied, some not).
208 self.partial = False
208 self.partial = False
209
209
210 # flag to indicate whether to display as folded/unfolded to user
210 # flag to indicate whether to display as folded/unfolded to user
211 self.folded = True
211 self.folded = True
212
212
213 # list of all headers in patch
213 # list of all headers in patch
214 self.patch = None
214 self.patch = None
215
215
216 # flag is False if this header was ever unfolded from initial state
216 # flag is False if this header was ever unfolded from initial state
217 self.neverunfolded = True
217 self.neverunfolded = True
218 self.hunks = [uihunk(h, self) for h in self.hunks]
218 self.hunks = [uihunk(h, self) for h in self.hunks]
219
219
220 def prettystr(self):
220 def prettystr(self):
221 x = stringio()
221 x = stringio()
222 self.pretty(x)
222 self.pretty(x)
223 return x.getvalue()
223 return x.getvalue()
224
224
225 def nextsibling(self):
225 def nextsibling(self):
226 numheadersinpatch = len(self.patch)
226 numheadersinpatch = len(self.patch)
227 indexofthisheader = self.patch.index(self)
227 indexofthisheader = self.patch.index(self)
228
228
229 if indexofthisheader < numheadersinpatch - 1:
229 if indexofthisheader < numheadersinpatch - 1:
230 nextheader = self.patch[indexofthisheader + 1]
230 nextheader = self.patch[indexofthisheader + 1]
231 return nextheader
231 return nextheader
232 else:
232 else:
233 return None
233 return None
234
234
235 def prevsibling(self):
235 def prevsibling(self):
236 indexofthisheader = self.patch.index(self)
236 indexofthisheader = self.patch.index(self)
237 if indexofthisheader > 0:
237 if indexofthisheader > 0:
238 previousheader = self.patch[indexofthisheader - 1]
238 previousheader = self.patch[indexofthisheader - 1]
239 return previousheader
239 return previousheader
240 else:
240 else:
241 return None
241 return None
242
242
243 def parentitem(self):
243 def parentitem(self):
244 """
244 """
245 there is no 'real' parent item of a header that can be selected,
245 there is no 'real' parent item of a header that can be selected,
246 so return None.
246 so return None.
247 """
247 """
248 return None
248 return None
249
249
250 def firstchild(self):
250 def firstchild(self):
251 "return the first child of this item, if one exists. otherwise None."
251 "return the first child of this item, if one exists. otherwise None."
252 if len(self.hunks) > 0:
252 if len(self.hunks) > 0:
253 return self.hunks[0]
253 return self.hunks[0]
254 else:
254 else:
255 return None
255 return None
256
256
257 def lastchild(self):
257 def lastchild(self):
258 "return the last child of this item, if one exists. otherwise None."
258 "return the last child of this item, if one exists. otherwise None."
259 if len(self.hunks) > 0:
259 if len(self.hunks) > 0:
260 return self.hunks[-1]
260 return self.hunks[-1]
261 else:
261 else:
262 return None
262 return None
263
263
264 def allchildren(self):
264 def allchildren(self):
265 "return a list of all of the direct children of this node"
265 "return a list of all of the direct children of this node"
266 return self.hunks
266 return self.hunks
267
267
268 def __getattr__(self, name):
268 def __getattr__(self, name):
269 return getattr(self.nonuiheader, name)
269 return getattr(self.nonuiheader, name)
270
270
271 class uihunkline(patchnode):
271 class uihunkline(patchnode):
272 "represents a changed line in a hunk"
272 "represents a changed line in a hunk"
273 def __init__(self, linetext, hunk):
273 def __init__(self, linetext, hunk):
274 self.linetext = linetext
274 self.linetext = linetext
275 self.applied = True
275 self.applied = True
276 # the parent hunk to which this line belongs
276 # the parent hunk to which this line belongs
277 self.hunk = hunk
277 self.hunk = hunk
278 # folding lines currently is not used/needed, but this flag is needed
278 # folding lines currently is not used/needed, but this flag is needed
279 # in the previtem method.
279 # in the previtem method.
280 self.folded = False
280 self.folded = False
281
281
282 def prettystr(self):
282 def prettystr(self):
283 return self.linetext
283 return self.linetext
284
284
285 def nextsibling(self):
285 def nextsibling(self):
286 numlinesinhunk = len(self.hunk.changedlines)
286 numlinesinhunk = len(self.hunk.changedlines)
287 indexofthisline = self.hunk.changedlines.index(self)
287 indexofthisline = self.hunk.changedlines.index(self)
288
288
289 if (indexofthisline < numlinesinhunk - 1):
289 if (indexofthisline < numlinesinhunk - 1):
290 nextline = self.hunk.changedlines[indexofthisline + 1]
290 nextline = self.hunk.changedlines[indexofthisline + 1]
291 return nextline
291 return nextline
292 else:
292 else:
293 return None
293 return None
294
294
295 def prevsibling(self):
295 def prevsibling(self):
296 indexofthisline = self.hunk.changedlines.index(self)
296 indexofthisline = self.hunk.changedlines.index(self)
297 if indexofthisline > 0:
297 if indexofthisline > 0:
298 previousline = self.hunk.changedlines[indexofthisline - 1]
298 previousline = self.hunk.changedlines[indexofthisline - 1]
299 return previousline
299 return previousline
300 else:
300 else:
301 return None
301 return None
302
302
303 def parentitem(self):
303 def parentitem(self):
304 "return the parent to the current item"
304 "return the parent to the current item"
305 return self.hunk
305 return self.hunk
306
306
307 def firstchild(self):
307 def firstchild(self):
308 "return the first child of this item, if one exists. otherwise None."
308 "return the first child of this item, if one exists. otherwise None."
309 # hunk-lines don't have children
309 # hunk-lines don't have children
310 return None
310 return None
311
311
312 def lastchild(self):
312 def lastchild(self):
313 "return the last child of this item, if one exists. otherwise None."
313 "return the last child of this item, if one exists. otherwise None."
314 # hunk-lines don't have children
314 # hunk-lines don't have children
315 return None
315 return None
316
316
317 class uihunk(patchnode):
317 class uihunk(patchnode):
318 """ui patch hunk, wraps a hunk and keep track of ui behavior """
318 """ui patch hunk, wraps a hunk and keep track of ui behavior """
319 maxcontext = 3
319 maxcontext = 3
320
320
321 def __init__(self, hunk, header):
321 def __init__(self, hunk, header):
322 self._hunk = hunk
322 self._hunk = hunk
323 self.changedlines = [uihunkline(line, self) for line in hunk.hunk]
323 self.changedlines = [uihunkline(line, self) for line in hunk.hunk]
324 self.header = header
324 self.header = header
325 # used at end for detecting how many removed lines were un-applied
325 # used at end for detecting how many removed lines were un-applied
326 self.originalremoved = self.removed
326 self.originalremoved = self.removed
327
327
328 # flag to indicate whether to display as folded/unfolded to user
328 # flag to indicate whether to display as folded/unfolded to user
329 self.folded = True
329 self.folded = True
330 # flag to indicate whether to apply this chunk
330 # flag to indicate whether to apply this chunk
331 self.applied = True
331 self.applied = True
332 # flag which only affects the status display indicating if a node's
332 # flag which only affects the status display indicating if a node's
333 # children are partially applied (i.e. some applied, some not).
333 # children are partially applied (i.e. some applied, some not).
334 self.partial = False
334 self.partial = False
335
335
336 def nextsibling(self):
336 def nextsibling(self):
337 numhunksinheader = len(self.header.hunks)
337 numhunksinheader = len(self.header.hunks)
338 indexofthishunk = self.header.hunks.index(self)
338 indexofthishunk = self.header.hunks.index(self)
339
339
340 if (indexofthishunk < numhunksinheader - 1):
340 if (indexofthishunk < numhunksinheader - 1):
341 nexthunk = self.header.hunks[indexofthishunk + 1]
341 nexthunk = self.header.hunks[indexofthishunk + 1]
342 return nexthunk
342 return nexthunk
343 else:
343 else:
344 return None
344 return None
345
345
346 def prevsibling(self):
346 def prevsibling(self):
347 indexofthishunk = self.header.hunks.index(self)
347 indexofthishunk = self.header.hunks.index(self)
348 if indexofthishunk > 0:
348 if indexofthishunk > 0:
349 previoushunk = self.header.hunks[indexofthishunk - 1]
349 previoushunk = self.header.hunks[indexofthishunk - 1]
350 return previoushunk
350 return previoushunk
351 else:
351 else:
352 return None
352 return None
353
353
354 def parentitem(self):
354 def parentitem(self):
355 "return the parent to the current item"
355 "return the parent to the current item"
356 return self.header
356 return self.header
357
357
358 def firstchild(self):
358 def firstchild(self):
359 "return the first child of this item, if one exists. otherwise None."
359 "return the first child of this item, if one exists. otherwise None."
360 if len(self.changedlines) > 0:
360 if len(self.changedlines) > 0:
361 return self.changedlines[0]
361 return self.changedlines[0]
362 else:
362 else:
363 return None
363 return None
364
364
365 def lastchild(self):
365 def lastchild(self):
366 "return the last child of this item, if one exists. otherwise None."
366 "return the last child of this item, if one exists. otherwise None."
367 if len(self.changedlines) > 0:
367 if len(self.changedlines) > 0:
368 return self.changedlines[-1]
368 return self.changedlines[-1]
369 else:
369 else:
370 return None
370 return None
371
371
372 def allchildren(self):
372 def allchildren(self):
373 "return a list of all of the direct children of this node"
373 "return a list of all of the direct children of this node"
374 return self.changedlines
374 return self.changedlines
375
375
376 def countchanges(self):
376 def countchanges(self):
377 """changedlines -> (n+,n-)"""
377 """changedlines -> (n+,n-)"""
378 add = len([l for l in self.changedlines if l.applied
378 add = len([l for l in self.changedlines if l.applied
379 and l.prettystr()[0] == '+'])
379 and l.prettystr()[0] == '+'])
380 rem = len([l for l in self.changedlines if l.applied
380 rem = len([l for l in self.changedlines if l.applied
381 and l.prettystr()[0] == '-'])
381 and l.prettystr()[0] == '-'])
382 return add, rem
382 return add, rem
383
383
384 def getfromtoline(self):
384 def getfromtoline(self):
385 # calculate the number of removed lines converted to context lines
385 # calculate the number of removed lines converted to context lines
386 removedconvertedtocontext = self.originalremoved - self.removed
386 removedconvertedtocontext = self.originalremoved - self.removed
387
387
388 contextlen = (len(self.before) + len(self.after) +
388 contextlen = (len(self.before) + len(self.after) +
389 removedconvertedtocontext)
389 removedconvertedtocontext)
390 if self.after and self.after[-1] == '\\ no newline at end of file\n':
390 if self.after and self.after[-1] == '\\ no newline at end of file\n':
391 contextlen -= 1
391 contextlen -= 1
392 fromlen = contextlen + self.removed
392 fromlen = contextlen + self.removed
393 tolen = contextlen + self.added
393 tolen = contextlen + self.added
394
394
395 # diffutils manual, section "2.2.2.2 detailed description of unified
395 # diffutils manual, section "2.2.2.2 detailed description of unified
396 # format": "an empty hunk is considered to end at the line that
396 # format": "an empty hunk is considered to end at the line that
397 # precedes the hunk."
397 # precedes the hunk."
398 #
398 #
399 # so, if either of hunks is empty, decrease its line start. --immerrr
399 # so, if either of hunks is empty, decrease its line start. --immerrr
400 # but only do this if fromline > 0, to avoid having, e.g fromline=-1.
400 # but only do this if fromline > 0, to avoid having, e.g fromline=-1.
401 fromline, toline = self.fromline, self.toline
401 fromline, toline = self.fromline, self.toline
402 if fromline != 0:
402 if fromline != 0:
403 if fromlen == 0:
403 if fromlen == 0:
404 fromline -= 1
404 fromline -= 1
405 if tolen == 0:
405 if tolen == 0:
406 toline -= 1
406 toline -= 1
407
407
408 fromtoline = '@@ -%d,%d +%d,%d @@%s\n' % (
408 fromtoline = '@@ -%d,%d +%d,%d @@%s\n' % (
409 fromline, fromlen, toline, tolen,
409 fromline, fromlen, toline, tolen,
410 self.proc and (' ' + self.proc))
410 self.proc and (' ' + self.proc))
411 return fromtoline
411 return fromtoline
412
412
413 def write(self, fp):
413 def write(self, fp):
414 # updated self.added/removed, which are used by getfromtoline()
414 # updated self.added/removed, which are used by getfromtoline()
415 self.added, self.removed = self.countchanges()
415 self.added, self.removed = self.countchanges()
416 fp.write(self.getfromtoline())
416 fp.write(self.getfromtoline())
417
417
418 hunklinelist = []
418 hunklinelist = []
419 # add the following to the list: (1) all applied lines, and
419 # add the following to the list: (1) all applied lines, and
420 # (2) all unapplied removal lines (convert these to context lines)
420 # (2) all unapplied removal lines (convert these to context lines)
421 for changedline in self.changedlines:
421 for changedline in self.changedlines:
422 changedlinestr = changedline.prettystr()
422 changedlinestr = changedline.prettystr()
423 if changedline.applied:
423 if changedline.applied:
424 hunklinelist.append(changedlinestr)
424 hunklinelist.append(changedlinestr)
425 elif changedlinestr[0] == "-":
425 elif changedlinestr[0] == "-":
426 hunklinelist.append(" " + changedlinestr[1:])
426 hunklinelist.append(" " + changedlinestr[1:])
427
427
428 fp.write(''.join(self.before + hunklinelist + self.after))
428 fp.write(''.join(self.before + hunklinelist + self.after))
429
429
430 pretty = write
430 pretty = write
431
431
432 def prettystr(self):
432 def prettystr(self):
433 x = stringio()
433 x = stringio()
434 self.pretty(x)
434 self.pretty(x)
435 return x.getvalue()
435 return x.getvalue()
436
436
437 def __getattr__(self, name):
437 def __getattr__(self, name):
438 return getattr(self._hunk, name)
438 return getattr(self._hunk, name)
439
439
440 def __repr__(self):
440 def __repr__(self):
441 return '<hunk %r@%d>' % (self.filename(), self.fromline)
441 return '<hunk %r@%d>' % (self.filename(), self.fromline)
442
442
443 def filterpatch(ui, chunks, chunkselector, operation=None):
443 def filterpatch(ui, chunks, chunkselector):
444 """interactively filter patch chunks into applied-only chunks"""
444 """interactively filter patch chunks into applied-only chunks"""
445
446 if operation is None:
447 operation = _('confirm')
448 chunks = list(chunks)
445 chunks = list(chunks)
449 # convert chunks list into structure suitable for displaying/modifying
446 # convert chunks list into structure suitable for displaying/modifying
450 # with curses. create a list of headers only.
447 # with curses. create a list of headers only.
451 headers = [c for c in chunks if isinstance(c, patchmod.header)]
448 headers = [c for c in chunks if isinstance(c, patchmod.header)]
452
449
453 # if there are no changed files
450 # if there are no changed files
454 if len(headers) == 0:
451 if len(headers) == 0:
455 return [], {}
452 return [], {}
456 uiheaders = [uiheader(h) for h in headers]
453 uiheaders = [uiheader(h) for h in headers]
457 # let user choose headers/hunks/lines, and mark their applied flags
454 # let user choose headers/hunks/lines, and mark their applied flags
458 # accordingly
455 # accordingly
459 ret = chunkselector(ui, uiheaders)
456 ret = chunkselector(ui, uiheaders)
460 appliedhunklist = []
457 appliedhunklist = []
461 for hdr in uiheaders:
458 for hdr in uiheaders:
462 if (hdr.applied and
459 if (hdr.applied and
463 (hdr.special() or len([h for h in hdr.hunks if h.applied]) > 0)):
460 (hdr.special() or len([h for h in hdr.hunks if h.applied]) > 0)):
464 appliedhunklist.append(hdr)
461 appliedhunklist.append(hdr)
465 fixoffset = 0
462 fixoffset = 0
466 for hnk in hdr.hunks:
463 for hnk in hdr.hunks:
467 if hnk.applied:
464 if hnk.applied:
468 appliedhunklist.append(hnk)
465 appliedhunklist.append(hnk)
469 # adjust the 'to'-line offset of the hunk to be correct
466 # adjust the 'to'-line offset of the hunk to be correct
470 # after de-activating some of the other hunks for this file
467 # after de-activating some of the other hunks for this file
471 if fixoffset:
468 if fixoffset:
472 #hnk = copy.copy(hnk) # necessary??
469 #hnk = copy.copy(hnk) # necessary??
473 hnk.toline += fixoffset
470 hnk.toline += fixoffset
474 else:
471 else:
475 fixoffset += hnk.removed - hnk.added
472 fixoffset += hnk.removed - hnk.added
476
473
477 return (appliedhunklist, ret)
474 return (appliedhunklist, ret)
478
475
479 def gethw():
476 def gethw():
480 """
477 """
481 magically get the current height and width of the window (without initscr)
478 magically get the current height and width of the window (without initscr)
482
479
483 this is a rip-off of a rip-off - taken from the bpython code. it is
480 this is a rip-off of a rip-off - taken from the bpython code. it is
484 useful / necessary because otherwise curses.initscr() must be called,
481 useful / necessary because otherwise curses.initscr() must be called,
485 which can leave the terminal in a nasty state after exiting.
482 which can leave the terminal in a nasty state after exiting.
486 """
483 """
487 h, w = struct.unpack(
484 h, w = struct.unpack(
488 "hhhh", fcntl.ioctl(_origstdout, termios.TIOCGWINSZ, "\000"*8))[0:2]
485 "hhhh", fcntl.ioctl(_origstdout, termios.TIOCGWINSZ, "\000"*8))[0:2]
489 return h, w
486 return h, w
490
487
491 def chunkselector(ui, headerlist):
488 def chunkselector(ui, headerlist):
492 """
489 """
493 curses interface to get selection of chunks, and mark the applied flags
490 curses interface to get selection of chunks, and mark the applied flags
494 of the chosen chunks.
491 of the chosen chunks.
495 """
492 """
496 ui.write(_('starting interactive selection\n'))
493 ui.write(_('starting interactive selection\n'))
497 chunkselector = curseschunkselector(headerlist, ui)
494 chunkselector = curseschunkselector(headerlist, ui)
498 f = signal.getsignal(signal.SIGTSTP)
495 f = signal.getsignal(signal.SIGTSTP)
499 curses.wrapper(chunkselector.main)
496 curses.wrapper(chunkselector.main)
500 if chunkselector.initerr is not None:
497 if chunkselector.initerr is not None:
501 raise error.Abort(chunkselector.initerr)
498 raise error.Abort(chunkselector.initerr)
502 # ncurses does not restore signal handler for SIGTSTP
499 # ncurses does not restore signal handler for SIGTSTP
503 signal.signal(signal.SIGTSTP, f)
500 signal.signal(signal.SIGTSTP, f)
504 return chunkselector.opts
501 return chunkselector.opts
505
502
506 def testdecorator(testfn, f):
503 def testdecorator(testfn, f):
507 def u(*args, **kwargs):
504 def u(*args, **kwargs):
508 return f(testfn, *args, **kwargs)
505 return f(testfn, *args, **kwargs)
509 return u
506 return u
510
507
511 def testchunkselector(testfn, ui, headerlist):
508 def testchunkselector(testfn, ui, headerlist):
512 """
509 """
513 test interface to get selection of chunks, and mark the applied flags
510 test interface to get selection of chunks, and mark the applied flags
514 of the chosen chunks.
511 of the chosen chunks.
515 """
512 """
516 chunkselector = curseschunkselector(headerlist, ui)
513 chunkselector = curseschunkselector(headerlist, ui)
517 if testfn and os.path.exists(testfn):
514 if testfn and os.path.exists(testfn):
518 testf = open(testfn)
515 testf = open(testfn)
519 testcommands = map(lambda x: x.rstrip('\n'), testf.readlines())
516 testcommands = map(lambda x: x.rstrip('\n'), testf.readlines())
520 testf.close()
517 testf.close()
521 while True:
518 while True:
522 if chunkselector.handlekeypressed(testcommands.pop(0), test=True):
519 if chunkselector.handlekeypressed(testcommands.pop(0), test=True):
523 break
520 break
524 return chunkselector.opts
521 return chunkselector.opts
525
522
526 class curseschunkselector(object):
523 class curseschunkselector(object):
527 def __init__(self, headerlist, ui):
524 def __init__(self, headerlist, ui):
528 # put the headers into a patch object
525 # put the headers into a patch object
529 self.headerlist = patch(headerlist)
526 self.headerlist = patch(headerlist)
530
527
531 self.ui = ui
528 self.ui = ui
532 self.opts = {}
529 self.opts = {}
533
530
534 self.errorstr = None
531 self.errorstr = None
535 # list of all chunks
532 # list of all chunks
536 self.chunklist = []
533 self.chunklist = []
537 for h in headerlist:
534 for h in headerlist:
538 self.chunklist.append(h)
535 self.chunklist.append(h)
539 self.chunklist.extend(h.hunks)
536 self.chunklist.extend(h.hunks)
540
537
541 # dictionary mapping (fgcolor, bgcolor) pairs to the
538 # dictionary mapping (fgcolor, bgcolor) pairs to the
542 # corresponding curses color-pair value.
539 # corresponding curses color-pair value.
543 self.colorpairs = {}
540 self.colorpairs = {}
544 # maps custom nicknames of color-pairs to curses color-pair values
541 # maps custom nicknames of color-pairs to curses color-pair values
545 self.colorpairnames = {}
542 self.colorpairnames = {}
546
543
547 # the currently selected header, hunk, or hunk-line
544 # the currently selected header, hunk, or hunk-line
548 self.currentselecteditem = self.headerlist[0]
545 self.currentselecteditem = self.headerlist[0]
549
546
550 # updated when printing out patch-display -- the 'lines' here are the
547 # updated when printing out patch-display -- the 'lines' here are the
551 # line positions *in the pad*, not on the screen.
548 # line positions *in the pad*, not on the screen.
552 self.selecteditemstartline = 0
549 self.selecteditemstartline = 0
553 self.selecteditemendline = None
550 self.selecteditemendline = None
554
551
555 # define indentation levels
552 # define indentation levels
556 self.headerindentnumchars = 0
553 self.headerindentnumchars = 0
557 self.hunkindentnumchars = 3
554 self.hunkindentnumchars = 3
558 self.hunklineindentnumchars = 6
555 self.hunklineindentnumchars = 6
559
556
560 # the first line of the pad to print to the screen
557 # the first line of the pad to print to the screen
561 self.firstlineofpadtoprint = 0
558 self.firstlineofpadtoprint = 0
562
559
563 # keeps track of the number of lines in the pad
560 # keeps track of the number of lines in the pad
564 self.numpadlines = None
561 self.numpadlines = None
565
562
566 self.numstatuslines = 2
563 self.numstatuslines = 2
567
564
568 # keep a running count of the number of lines printed to the pad
565 # keep a running count of the number of lines printed to the pad
569 # (used for determining when the selected item begins/ends)
566 # (used for determining when the selected item begins/ends)
570 self.linesprintedtopadsofar = 0
567 self.linesprintedtopadsofar = 0
571
568
572 # the first line of the pad which is visible on the screen
569 # the first line of the pad which is visible on the screen
573 self.firstlineofpadtoprint = 0
570 self.firstlineofpadtoprint = 0
574
571
575 # stores optional text for a commit comment provided by the user
572 # stores optional text for a commit comment provided by the user
576 self.commenttext = ""
573 self.commenttext = ""
577
574
578 # if the last 'toggle all' command caused all changes to be applied
575 # if the last 'toggle all' command caused all changes to be applied
579 self.waslasttoggleallapplied = True
576 self.waslasttoggleallapplied = True
580
577
581 def uparrowevent(self):
578 def uparrowevent(self):
582 """
579 """
583 try to select the previous item to the current item that has the
580 try to select the previous item to the current item that has the
584 most-indented level. for example, if a hunk is selected, try to select
581 most-indented level. for example, if a hunk is selected, try to select
585 the last hunkline of the hunk prior to the selected hunk. or, if
582 the last hunkline of the hunk prior to the selected hunk. or, if
586 the first hunkline of a hunk is currently selected, then select the
583 the first hunkline of a hunk is currently selected, then select the
587 hunk itself.
584 hunk itself.
588 """
585 """
589 currentitem = self.currentselecteditem
586 currentitem = self.currentselecteditem
590
587
591 nextitem = currentitem.previtem()
588 nextitem = currentitem.previtem()
592
589
593 if nextitem is None:
590 if nextitem is None:
594 # if no parent item (i.e. currentitem is the first header), then
591 # if no parent item (i.e. currentitem is the first header), then
595 # no change...
592 # no change...
596 nextitem = currentitem
593 nextitem = currentitem
597
594
598 self.currentselecteditem = nextitem
595 self.currentselecteditem = nextitem
599
596
600 def uparrowshiftevent(self):
597 def uparrowshiftevent(self):
601 """
598 """
602 select (if possible) the previous item on the same level as the
599 select (if possible) the previous item on the same level as the
603 currently selected item. otherwise, select (if possible) the
600 currently selected item. otherwise, select (if possible) the
604 parent-item of the currently selected item.
601 parent-item of the currently selected item.
605 """
602 """
606 currentitem = self.currentselecteditem
603 currentitem = self.currentselecteditem
607 nextitem = currentitem.prevsibling()
604 nextitem = currentitem.prevsibling()
608 # if there's no previous sibling, try choosing the parent
605 # if there's no previous sibling, try choosing the parent
609 if nextitem is None:
606 if nextitem is None:
610 nextitem = currentitem.parentitem()
607 nextitem = currentitem.parentitem()
611 if nextitem is None:
608 if nextitem is None:
612 # if no parent item (i.e. currentitem is the first header), then
609 # if no parent item (i.e. currentitem is the first header), then
613 # no change...
610 # no change...
614 nextitem = currentitem
611 nextitem = currentitem
615
612
616 self.currentselecteditem = nextitem
613 self.currentselecteditem = nextitem
617
614
618 def downarrowevent(self):
615 def downarrowevent(self):
619 """
616 """
620 try to select the next item to the current item that has the
617 try to select the next item to the current item that has the
621 most-indented level. for example, if a hunk is selected, select
618 most-indented level. for example, if a hunk is selected, select
622 the first hunkline of the selected hunk. or, if the last hunkline of
619 the first hunkline of the selected hunk. or, if the last hunkline of
623 a hunk is currently selected, then select the next hunk, if one exists,
620 a hunk is currently selected, then select the next hunk, if one exists,
624 or if not, the next header if one exists.
621 or if not, the next header if one exists.
625 """
622 """
626 #self.startprintline += 1 #debug
623 #self.startprintline += 1 #debug
627 currentitem = self.currentselecteditem
624 currentitem = self.currentselecteditem
628
625
629 nextitem = currentitem.nextitem()
626 nextitem = currentitem.nextitem()
630 # if there's no next item, keep the selection as-is
627 # if there's no next item, keep the selection as-is
631 if nextitem is None:
628 if nextitem is None:
632 nextitem = currentitem
629 nextitem = currentitem
633
630
634 self.currentselecteditem = nextitem
631 self.currentselecteditem = nextitem
635
632
636 def downarrowshiftevent(self):
633 def downarrowshiftevent(self):
637 """
634 """
638 select (if possible) the next item on the same level as the currently
635 select (if possible) the next item on the same level as the currently
639 selected item. otherwise, select (if possible) the next item on the
636 selected item. otherwise, select (if possible) the next item on the
640 same level as the parent item of the currently selected item.
637 same level as the parent item of the currently selected item.
641 """
638 """
642 currentitem = self.currentselecteditem
639 currentitem = self.currentselecteditem
643 nextitem = currentitem.nextsibling()
640 nextitem = currentitem.nextsibling()
644 # if there's no next sibling, try choosing the parent's nextsibling
641 # if there's no next sibling, try choosing the parent's nextsibling
645 if nextitem is None:
642 if nextitem is None:
646 try:
643 try:
647 nextitem = currentitem.parentitem().nextsibling()
644 nextitem = currentitem.parentitem().nextsibling()
648 except AttributeError:
645 except AttributeError:
649 # parentitem returned None, so nextsibling() can't be called
646 # parentitem returned None, so nextsibling() can't be called
650 nextitem = None
647 nextitem = None
651 if nextitem is None:
648 if nextitem is None:
652 # if parent has no next sibling, then no change...
649 # if parent has no next sibling, then no change...
653 nextitem = currentitem
650 nextitem = currentitem
654
651
655 self.currentselecteditem = nextitem
652 self.currentselecteditem = nextitem
656
653
657 def rightarrowevent(self):
654 def rightarrowevent(self):
658 """
655 """
659 select (if possible) the first of this item's child-items.
656 select (if possible) the first of this item's child-items.
660 """
657 """
661 currentitem = self.currentselecteditem
658 currentitem = self.currentselecteditem
662 nextitem = currentitem.firstchild()
659 nextitem = currentitem.firstchild()
663
660
664 # turn off folding if we want to show a child-item
661 # turn off folding if we want to show a child-item
665 if currentitem.folded:
662 if currentitem.folded:
666 self.togglefolded(currentitem)
663 self.togglefolded(currentitem)
667
664
668 if nextitem is None:
665 if nextitem is None:
669 # if no next item on parent-level, then no change...
666 # if no next item on parent-level, then no change...
670 nextitem = currentitem
667 nextitem = currentitem
671
668
672 self.currentselecteditem = nextitem
669 self.currentselecteditem = nextitem
673
670
674 def leftarrowevent(self):
671 def leftarrowevent(self):
675 """
672 """
676 if the current item can be folded (i.e. it is an unfolded header or
673 if the current item can be folded (i.e. it is an unfolded header or
677 hunk), then fold it. otherwise try select (if possible) the parent
674 hunk), then fold it. otherwise try select (if possible) the parent
678 of this item.
675 of this item.
679 """
676 """
680 currentitem = self.currentselecteditem
677 currentitem = self.currentselecteditem
681
678
682 # try to fold the item
679 # try to fold the item
683 if not isinstance(currentitem, uihunkline):
680 if not isinstance(currentitem, uihunkline):
684 if not currentitem.folded:
681 if not currentitem.folded:
685 self.togglefolded(item=currentitem)
682 self.togglefolded(item=currentitem)
686 return
683 return
687
684
688 # if it can't be folded, try to select the parent item
685 # if it can't be folded, try to select the parent item
689 nextitem = currentitem.parentitem()
686 nextitem = currentitem.parentitem()
690
687
691 if nextitem is None:
688 if nextitem is None:
692 # if no item on parent-level, then no change...
689 # if no item on parent-level, then no change...
693 nextitem = currentitem
690 nextitem = currentitem
694 if not nextitem.folded:
691 if not nextitem.folded:
695 self.togglefolded(item=nextitem)
692 self.togglefolded(item=nextitem)
696
693
697 self.currentselecteditem = nextitem
694 self.currentselecteditem = nextitem
698
695
699 def leftarrowshiftevent(self):
696 def leftarrowshiftevent(self):
700 """
697 """
701 select the header of the current item (or fold current item if the
698 select the header of the current item (or fold current item if the
702 current item is already a header).
699 current item is already a header).
703 """
700 """
704 currentitem = self.currentselecteditem
701 currentitem = self.currentselecteditem
705
702
706 if isinstance(currentitem, uiheader):
703 if isinstance(currentitem, uiheader):
707 if not currentitem.folded:
704 if not currentitem.folded:
708 self.togglefolded(item=currentitem)
705 self.togglefolded(item=currentitem)
709 return
706 return
710
707
711 # select the parent item recursively until we're at a header
708 # select the parent item recursively until we're at a header
712 while True:
709 while True:
713 nextitem = currentitem.parentitem()
710 nextitem = currentitem.parentitem()
714 if nextitem is None:
711 if nextitem is None:
715 break
712 break
716 else:
713 else:
717 currentitem = nextitem
714 currentitem = nextitem
718
715
719 self.currentselecteditem = currentitem
716 self.currentselecteditem = currentitem
720
717
721 def updatescroll(self):
718 def updatescroll(self):
722 "scroll the screen to fully show the currently-selected"
719 "scroll the screen to fully show the currently-selected"
723 selstart = self.selecteditemstartline
720 selstart = self.selecteditemstartline
724 selend = self.selecteditemendline
721 selend = self.selecteditemendline
725 #selnumlines = selend - selstart
722 #selnumlines = selend - selstart
726 padstart = self.firstlineofpadtoprint
723 padstart = self.firstlineofpadtoprint
727 padend = padstart + self.yscreensize - self.numstatuslines - 1
724 padend = padstart + self.yscreensize - self.numstatuslines - 1
728 # 'buffered' pad start/end values which scroll with a certain
725 # 'buffered' pad start/end values which scroll with a certain
729 # top/bottom context margin
726 # top/bottom context margin
730 padstartbuffered = padstart + 3
727 padstartbuffered = padstart + 3
731 padendbuffered = padend - 3
728 padendbuffered = padend - 3
732
729
733 if selend > padendbuffered:
730 if selend > padendbuffered:
734 self.scrolllines(selend - padendbuffered)
731 self.scrolllines(selend - padendbuffered)
735 elif selstart < padstartbuffered:
732 elif selstart < padstartbuffered:
736 # negative values scroll in pgup direction
733 # negative values scroll in pgup direction
737 self.scrolllines(selstart - padstartbuffered)
734 self.scrolllines(selstart - padstartbuffered)
738
735
739 def scrolllines(self, numlines):
736 def scrolllines(self, numlines):
740 "scroll the screen up (down) by numlines when numlines >0 (<0)."
737 "scroll the screen up (down) by numlines when numlines >0 (<0)."
741 self.firstlineofpadtoprint += numlines
738 self.firstlineofpadtoprint += numlines
742 if self.firstlineofpadtoprint < 0:
739 if self.firstlineofpadtoprint < 0:
743 self.firstlineofpadtoprint = 0
740 self.firstlineofpadtoprint = 0
744 if self.firstlineofpadtoprint > self.numpadlines - 1:
741 if self.firstlineofpadtoprint > self.numpadlines - 1:
745 self.firstlineofpadtoprint = self.numpadlines - 1
742 self.firstlineofpadtoprint = self.numpadlines - 1
746
743
747 def toggleapply(self, item=None):
744 def toggleapply(self, item=None):
748 """
745 """
749 toggle the applied flag of the specified item. if no item is specified,
746 toggle the applied flag of the specified item. if no item is specified,
750 toggle the flag of the currently selected item.
747 toggle the flag of the currently selected item.
751 """
748 """
752 if item is None:
749 if item is None:
753 item = self.currentselecteditem
750 item = self.currentselecteditem
754
751
755 item.applied = not item.applied
752 item.applied = not item.applied
756
753
757 if isinstance(item, uiheader):
754 if isinstance(item, uiheader):
758 item.partial = False
755 item.partial = False
759 if item.applied:
756 if item.applied:
760 # apply all its hunks
757 # apply all its hunks
761 for hnk in item.hunks:
758 for hnk in item.hunks:
762 hnk.applied = True
759 hnk.applied = True
763 # apply all their hunklines
760 # apply all their hunklines
764 for hunkline in hnk.changedlines:
761 for hunkline in hnk.changedlines:
765 hunkline.applied = True
762 hunkline.applied = True
766 else:
763 else:
767 # un-apply all its hunks
764 # un-apply all its hunks
768 for hnk in item.hunks:
765 for hnk in item.hunks:
769 hnk.applied = False
766 hnk.applied = False
770 hnk.partial = False
767 hnk.partial = False
771 # un-apply all their hunklines
768 # un-apply all their hunklines
772 for hunkline in hnk.changedlines:
769 for hunkline in hnk.changedlines:
773 hunkline.applied = False
770 hunkline.applied = False
774 elif isinstance(item, uihunk):
771 elif isinstance(item, uihunk):
775 item.partial = False
772 item.partial = False
776 # apply all it's hunklines
773 # apply all it's hunklines
777 for hunkline in item.changedlines:
774 for hunkline in item.changedlines:
778 hunkline.applied = item.applied
775 hunkline.applied = item.applied
779
776
780 siblingappliedstatus = [hnk.applied for hnk in item.header.hunks]
777 siblingappliedstatus = [hnk.applied for hnk in item.header.hunks]
781 allsiblingsapplied = not (False in siblingappliedstatus)
778 allsiblingsapplied = not (False in siblingappliedstatus)
782 nosiblingsapplied = not (True in siblingappliedstatus)
779 nosiblingsapplied = not (True in siblingappliedstatus)
783
780
784 siblingspartialstatus = [hnk.partial for hnk in item.header.hunks]
781 siblingspartialstatus = [hnk.partial for hnk in item.header.hunks]
785 somesiblingspartial = (True in siblingspartialstatus)
782 somesiblingspartial = (True in siblingspartialstatus)
786
783
787 #cases where applied or partial should be removed from header
784 #cases where applied or partial should be removed from header
788
785
789 # if no 'sibling' hunks are applied (including this hunk)
786 # if no 'sibling' hunks are applied (including this hunk)
790 if nosiblingsapplied:
787 if nosiblingsapplied:
791 if not item.header.special():
788 if not item.header.special():
792 item.header.applied = False
789 item.header.applied = False
793 item.header.partial = False
790 item.header.partial = False
794 else: # some/all parent siblings are applied
791 else: # some/all parent siblings are applied
795 item.header.applied = True
792 item.header.applied = True
796 item.header.partial = (somesiblingspartial or
793 item.header.partial = (somesiblingspartial or
797 not allsiblingsapplied)
794 not allsiblingsapplied)
798
795
799 elif isinstance(item, uihunkline):
796 elif isinstance(item, uihunkline):
800 siblingappliedstatus = [ln.applied for ln in item.hunk.changedlines]
797 siblingappliedstatus = [ln.applied for ln in item.hunk.changedlines]
801 allsiblingsapplied = not (False in siblingappliedstatus)
798 allsiblingsapplied = not (False in siblingappliedstatus)
802 nosiblingsapplied = not (True in siblingappliedstatus)
799 nosiblingsapplied = not (True in siblingappliedstatus)
803
800
804 # if no 'sibling' lines are applied
801 # if no 'sibling' lines are applied
805 if nosiblingsapplied:
802 if nosiblingsapplied:
806 item.hunk.applied = False
803 item.hunk.applied = False
807 item.hunk.partial = False
804 item.hunk.partial = False
808 elif allsiblingsapplied:
805 elif allsiblingsapplied:
809 item.hunk.applied = True
806 item.hunk.applied = True
810 item.hunk.partial = False
807 item.hunk.partial = False
811 else: # some siblings applied
808 else: # some siblings applied
812 item.hunk.applied = True
809 item.hunk.applied = True
813 item.hunk.partial = True
810 item.hunk.partial = True
814
811
815 parentsiblingsapplied = [hnk.applied for hnk
812 parentsiblingsapplied = [hnk.applied for hnk
816 in item.hunk.header.hunks]
813 in item.hunk.header.hunks]
817 noparentsiblingsapplied = not (True in parentsiblingsapplied)
814 noparentsiblingsapplied = not (True in parentsiblingsapplied)
818 allparentsiblingsapplied = not (False in parentsiblingsapplied)
815 allparentsiblingsapplied = not (False in parentsiblingsapplied)
819
816
820 parentsiblingspartial = [hnk.partial for hnk
817 parentsiblingspartial = [hnk.partial for hnk
821 in item.hunk.header.hunks]
818 in item.hunk.header.hunks]
822 someparentsiblingspartial = (True in parentsiblingspartial)
819 someparentsiblingspartial = (True in parentsiblingspartial)
823
820
824 # if all parent hunks are not applied, un-apply header
821 # if all parent hunks are not applied, un-apply header
825 if noparentsiblingsapplied:
822 if noparentsiblingsapplied:
826 if not item.hunk.header.special():
823 if not item.hunk.header.special():
827 item.hunk.header.applied = False
824 item.hunk.header.applied = False
828 item.hunk.header.partial = False
825 item.hunk.header.partial = False
829 # set the applied and partial status of the header if needed
826 # set the applied and partial status of the header if needed
830 else: # some/all parent siblings are applied
827 else: # some/all parent siblings are applied
831 item.hunk.header.applied = True
828 item.hunk.header.applied = True
832 item.hunk.header.partial = (someparentsiblingspartial or
829 item.hunk.header.partial = (someparentsiblingspartial or
833 not allparentsiblingsapplied)
830 not allparentsiblingsapplied)
834
831
835 def toggleall(self):
832 def toggleall(self):
836 "toggle the applied flag of all items."
833 "toggle the applied flag of all items."
837 if self.waslasttoggleallapplied: # then unapply them this time
834 if self.waslasttoggleallapplied: # then unapply them this time
838 for item in self.headerlist:
835 for item in self.headerlist:
839 if item.applied:
836 if item.applied:
840 self.toggleapply(item)
837 self.toggleapply(item)
841 else:
838 else:
842 for item in self.headerlist:
839 for item in self.headerlist:
843 if not item.applied:
840 if not item.applied:
844 self.toggleapply(item)
841 self.toggleapply(item)
845 self.waslasttoggleallapplied = not self.waslasttoggleallapplied
842 self.waslasttoggleallapplied = not self.waslasttoggleallapplied
846
843
847 def togglefolded(self, item=None, foldparent=False):
844 def togglefolded(self, item=None, foldparent=False):
848 "toggle folded flag of specified item (defaults to currently selected)"
845 "toggle folded flag of specified item (defaults to currently selected)"
849 if item is None:
846 if item is None:
850 item = self.currentselecteditem
847 item = self.currentselecteditem
851 if foldparent or (isinstance(item, uiheader) and item.neverunfolded):
848 if foldparent or (isinstance(item, uiheader) and item.neverunfolded):
852 if not isinstance(item, uiheader):
849 if not isinstance(item, uiheader):
853 # we need to select the parent item in this case
850 # we need to select the parent item in this case
854 self.currentselecteditem = item = item.parentitem()
851 self.currentselecteditem = item = item.parentitem()
855 elif item.neverunfolded:
852 elif item.neverunfolded:
856 item.neverunfolded = False
853 item.neverunfolded = False
857
854
858 # also fold any foldable children of the parent/current item
855 # also fold any foldable children of the parent/current item
859 if isinstance(item, uiheader): # the original or 'new' item
856 if isinstance(item, uiheader): # the original or 'new' item
860 for child in item.allchildren():
857 for child in item.allchildren():
861 child.folded = not item.folded
858 child.folded = not item.folded
862
859
863 if isinstance(item, (uiheader, uihunk)):
860 if isinstance(item, (uiheader, uihunk)):
864 item.folded = not item.folded
861 item.folded = not item.folded
865
862
866 def alignstring(self, instr, window):
863 def alignstring(self, instr, window):
867 """
864 """
868 add whitespace to the end of a string in order to make it fill
865 add whitespace to the end of a string in order to make it fill
869 the screen in the x direction. the current cursor position is
866 the screen in the x direction. the current cursor position is
870 taken into account when making this calculation. the string can span
867 taken into account when making this calculation. the string can span
871 multiple lines.
868 multiple lines.
872 """
869 """
873 y, xstart = window.getyx()
870 y, xstart = window.getyx()
874 width = self.xscreensize
871 width = self.xscreensize
875 # turn tabs into spaces
872 # turn tabs into spaces
876 instr = instr.expandtabs(4)
873 instr = instr.expandtabs(4)
877 strwidth = encoding.colwidth(instr)
874 strwidth = encoding.colwidth(instr)
878 numspaces = (width - ((strwidth + xstart) % width) - 1)
875 numspaces = (width - ((strwidth + xstart) % width) - 1)
879 return instr + " " * numspaces + "\n"
876 return instr + " " * numspaces + "\n"
880
877
881 def printstring(self, window, text, fgcolor=None, bgcolor=None, pair=None,
878 def printstring(self, window, text, fgcolor=None, bgcolor=None, pair=None,
882 pairname=None, attrlist=None, towin=True, align=True, showwhtspc=False):
879 pairname=None, attrlist=None, towin=True, align=True, showwhtspc=False):
883 """
880 """
884 print the string, text, with the specified colors and attributes, to
881 print the string, text, with the specified colors and attributes, to
885 the specified curses window object.
882 the specified curses window object.
886
883
887 the foreground and background colors are of the form
884 the foreground and background colors are of the form
888 curses.color_xxxx, where xxxx is one of: [black, blue, cyan, green,
885 curses.color_xxxx, where xxxx is one of: [black, blue, cyan, green,
889 magenta, red, white, yellow]. if pairname is provided, a color
886 magenta, red, white, yellow]. if pairname is provided, a color
890 pair will be looked up in the self.colorpairnames dictionary.
887 pair will be looked up in the self.colorpairnames dictionary.
891
888
892 attrlist is a list containing text attributes in the form of
889 attrlist is a list containing text attributes in the form of
893 curses.a_xxxx, where xxxx can be: [bold, dim, normal, standout,
890 curses.a_xxxx, where xxxx can be: [bold, dim, normal, standout,
894 underline].
891 underline].
895
892
896 if align == True, whitespace is added to the printed string such that
893 if align == True, whitespace is added to the printed string such that
897 the string stretches to the right border of the window.
894 the string stretches to the right border of the window.
898
895
899 if showwhtspc == True, trailing whitespace of a string is highlighted.
896 if showwhtspc == True, trailing whitespace of a string is highlighted.
900 """
897 """
901 # preprocess the text, converting tabs to spaces
898 # preprocess the text, converting tabs to spaces
902 text = text.expandtabs(4)
899 text = text.expandtabs(4)
903 # strip \n, and convert control characters to ^[char] representation
900 # strip \n, and convert control characters to ^[char] representation
904 text = re.sub(r'[\x00-\x08\x0a-\x1f]',
901 text = re.sub(r'[\x00-\x08\x0a-\x1f]',
905 lambda m:'^' + chr(ord(m.group()) + 64), text.strip('\n'))
902 lambda m:'^' + chr(ord(m.group()) + 64), text.strip('\n'))
906
903
907 if pair is not None:
904 if pair is not None:
908 colorpair = pair
905 colorpair = pair
909 elif pairname is not None:
906 elif pairname is not None:
910 colorpair = self.colorpairnames[pairname]
907 colorpair = self.colorpairnames[pairname]
911 else:
908 else:
912 if fgcolor is None:
909 if fgcolor is None:
913 fgcolor = -1
910 fgcolor = -1
914 if bgcolor is None:
911 if bgcolor is None:
915 bgcolor = -1
912 bgcolor = -1
916 if (fgcolor, bgcolor) in self.colorpairs:
913 if (fgcolor, bgcolor) in self.colorpairs:
917 colorpair = self.colorpairs[(fgcolor, bgcolor)]
914 colorpair = self.colorpairs[(fgcolor, bgcolor)]
918 else:
915 else:
919 colorpair = self.getcolorpair(fgcolor, bgcolor)
916 colorpair = self.getcolorpair(fgcolor, bgcolor)
920 # add attributes if possible
917 # add attributes if possible
921 if attrlist is None:
918 if attrlist is None:
922 attrlist = []
919 attrlist = []
923 if colorpair < 256:
920 if colorpair < 256:
924 # then it is safe to apply all attributes
921 # then it is safe to apply all attributes
925 for textattr in attrlist:
922 for textattr in attrlist:
926 colorpair |= textattr
923 colorpair |= textattr
927 else:
924 else:
928 # just apply a select few (safe?) attributes
925 # just apply a select few (safe?) attributes
929 for textattr in (curses.A_UNDERLINE, curses.A_BOLD):
926 for textattr in (curses.A_UNDERLINE, curses.A_BOLD):
930 if textattr in attrlist:
927 if textattr in attrlist:
931 colorpair |= textattr
928 colorpair |= textattr
932
929
933 y, xstart = self.chunkpad.getyx()
930 y, xstart = self.chunkpad.getyx()
934 t = "" # variable for counting lines printed
931 t = "" # variable for counting lines printed
935 # if requested, show trailing whitespace
932 # if requested, show trailing whitespace
936 if showwhtspc:
933 if showwhtspc:
937 origlen = len(text)
934 origlen = len(text)
938 text = text.rstrip(' \n') # tabs have already been expanded
935 text = text.rstrip(' \n') # tabs have already been expanded
939 strippedlen = len(text)
936 strippedlen = len(text)
940 numtrailingspaces = origlen - strippedlen
937 numtrailingspaces = origlen - strippedlen
941
938
942 if towin:
939 if towin:
943 window.addstr(text, colorpair)
940 window.addstr(text, colorpair)
944 t += text
941 t += text
945
942
946 if showwhtspc:
943 if showwhtspc:
947 wscolorpair = colorpair | curses.A_REVERSE
944 wscolorpair = colorpair | curses.A_REVERSE
948 if towin:
945 if towin:
949 for i in range(numtrailingspaces):
946 for i in range(numtrailingspaces):
950 window.addch(curses.ACS_CKBOARD, wscolorpair)
947 window.addch(curses.ACS_CKBOARD, wscolorpair)
951 t += " " * numtrailingspaces
948 t += " " * numtrailingspaces
952
949
953 if align:
950 if align:
954 if towin:
951 if towin:
955 extrawhitespace = self.alignstring("", window)
952 extrawhitespace = self.alignstring("", window)
956 window.addstr(extrawhitespace, colorpair)
953 window.addstr(extrawhitespace, colorpair)
957 else:
954 else:
958 # need to use t, since the x position hasn't incremented
955 # need to use t, since the x position hasn't incremented
959 extrawhitespace = self.alignstring(t, window)
956 extrawhitespace = self.alignstring(t, window)
960 t += extrawhitespace
957 t += extrawhitespace
961
958
962 # is reset to 0 at the beginning of printitem()
959 # is reset to 0 at the beginning of printitem()
963
960
964 linesprinted = (xstart + len(t)) / self.xscreensize
961 linesprinted = (xstart + len(t)) / self.xscreensize
965 self.linesprintedtopadsofar += linesprinted
962 self.linesprintedtopadsofar += linesprinted
966 return t
963 return t
967
964
968 def updatescreen(self):
965 def updatescreen(self):
969 self.statuswin.erase()
966 self.statuswin.erase()
970 self.chunkpad.erase()
967 self.chunkpad.erase()
971
968
972 printstring = self.printstring
969 printstring = self.printstring
973
970
974 # print out the status lines at the top
971 # print out the status lines at the top
975 try:
972 try:
976 if self.errorstr is not None:
973 if self.errorstr is not None:
977 printstring(self.statuswin, self.errorstr, pairname='legend')
974 printstring(self.statuswin, self.errorstr, pairname='legend')
978 printstring(self.statuswin, 'Press any key to continue',
975 printstring(self.statuswin, 'Press any key to continue',
979 pairname='legend')
976 pairname='legend')
980 self.statuswin.refresh()
977 self.statuswin.refresh()
981 return
978 return
982 line1 = ("SELECT CHUNKS: (j/k/up/dn/pgup/pgdn) move cursor; "
979 line1 = ("SELECT CHUNKS: (j/k/up/dn/pgup/pgdn) move cursor; "
983 "(space/A) toggle hunk/all; (e)dit hunk;")
980 "(space/A) toggle hunk/all; (e)dit hunk;")
984 line2 = (" (f)old/unfold; (c)onfirm applied; (q)uit; (?) help "
981 line2 = (" (f)old/unfold; (c)onfirm applied; (q)uit; (?) help "
985 "| [X]=hunk applied **=folded, toggle [a]mend mode")
982 "| [X]=hunk applied **=folded, toggle [a]mend mode")
986
983
987 printstring(self.statuswin,
984 printstring(self.statuswin,
988 util.ellipsis(line1, self.xscreensize - 1),
985 util.ellipsis(line1, self.xscreensize - 1),
989 pairname="legend")
986 pairname="legend")
990 printstring(self.statuswin,
987 printstring(self.statuswin,
991 util.ellipsis(line2, self.xscreensize - 1),
988 util.ellipsis(line2, self.xscreensize - 1),
992 pairname="legend")
989 pairname="legend")
993 except curses.error:
990 except curses.error:
994 pass
991 pass
995
992
996 # print out the patch in the remaining part of the window
993 # print out the patch in the remaining part of the window
997 try:
994 try:
998 self.printitem()
995 self.printitem()
999 self.updatescroll()
996 self.updatescroll()
1000 self.chunkpad.refresh(self.firstlineofpadtoprint, 0,
997 self.chunkpad.refresh(self.firstlineofpadtoprint, 0,
1001 self.numstatuslines, 0,
998 self.numstatuslines, 0,
1002 self.yscreensize + 1 - self.numstatuslines,
999 self.yscreensize + 1 - self.numstatuslines,
1003 self.xscreensize)
1000 self.xscreensize)
1004 except curses.error:
1001 except curses.error:
1005 pass
1002 pass
1006
1003
1007 # refresh([pminrow, pmincol, sminrow, smincol, smaxrow, smaxcol])
1004 # refresh([pminrow, pmincol, sminrow, smincol, smaxrow, smaxcol])
1008 self.statuswin.refresh()
1005 self.statuswin.refresh()
1009
1006
1010 def getstatusprefixstring(self, item):
1007 def getstatusprefixstring(self, item):
1011 """
1008 """
1012 create a string to prefix a line with which indicates whether 'item'
1009 create a string to prefix a line with which indicates whether 'item'
1013 is applied and/or folded.
1010 is applied and/or folded.
1014 """
1011 """
1015
1012
1016 # create checkbox string
1013 # create checkbox string
1017 if item.applied:
1014 if item.applied:
1018 if not isinstance(item, uihunkline) and item.partial:
1015 if not isinstance(item, uihunkline) and item.partial:
1019 checkbox = "[~]"
1016 checkbox = "[~]"
1020 else:
1017 else:
1021 checkbox = "[x]"
1018 checkbox = "[x]"
1022 else:
1019 else:
1023 checkbox = "[ ]"
1020 checkbox = "[ ]"
1024
1021
1025 try:
1022 try:
1026 if item.folded:
1023 if item.folded:
1027 checkbox += "**"
1024 checkbox += "**"
1028 if isinstance(item, uiheader):
1025 if isinstance(item, uiheader):
1029 # one of "m", "a", or "d" (modified, added, deleted)
1026 # one of "m", "a", or "d" (modified, added, deleted)
1030 filestatus = item.changetype
1027 filestatus = item.changetype
1031
1028
1032 checkbox += filestatus + " "
1029 checkbox += filestatus + " "
1033 else:
1030 else:
1034 checkbox += " "
1031 checkbox += " "
1035 if isinstance(item, uiheader):
1032 if isinstance(item, uiheader):
1036 # add two more spaces for headers
1033 # add two more spaces for headers
1037 checkbox += " "
1034 checkbox += " "
1038 except AttributeError: # not foldable
1035 except AttributeError: # not foldable
1039 checkbox += " "
1036 checkbox += " "
1040
1037
1041 return checkbox
1038 return checkbox
1042
1039
1043 def printheader(self, header, selected=False, towin=True,
1040 def printheader(self, header, selected=False, towin=True,
1044 ignorefolding=False):
1041 ignorefolding=False):
1045 """
1042 """
1046 print the header to the pad. if countlines is True, don't print
1043 print the header to the pad. if countlines is True, don't print
1047 anything, but just count the number of lines which would be printed.
1044 anything, but just count the number of lines which would be printed.
1048 """
1045 """
1049
1046
1050 outstr = ""
1047 outstr = ""
1051 text = header.prettystr()
1048 text = header.prettystr()
1052 chunkindex = self.chunklist.index(header)
1049 chunkindex = self.chunklist.index(header)
1053
1050
1054 if chunkindex != 0 and not header.folded:
1051 if chunkindex != 0 and not header.folded:
1055 # add separating line before headers
1052 # add separating line before headers
1056 outstr += self.printstring(self.chunkpad, '_' * self.xscreensize,
1053 outstr += self.printstring(self.chunkpad, '_' * self.xscreensize,
1057 towin=towin, align=False)
1054 towin=towin, align=False)
1058 # select color-pair based on if the header is selected
1055 # select color-pair based on if the header is selected
1059 colorpair = self.getcolorpair(name=selected and "selected" or "normal",
1056 colorpair = self.getcolorpair(name=selected and "selected" or "normal",
1060 attrlist=[curses.A_BOLD])
1057 attrlist=[curses.A_BOLD])
1061
1058
1062 # print out each line of the chunk, expanding it to screen width
1059 # print out each line of the chunk, expanding it to screen width
1063
1060
1064 # number of characters to indent lines on this level by
1061 # number of characters to indent lines on this level by
1065 indentnumchars = 0
1062 indentnumchars = 0
1066 checkbox = self.getstatusprefixstring(header)
1063 checkbox = self.getstatusprefixstring(header)
1067 if not header.folded or ignorefolding:
1064 if not header.folded or ignorefolding:
1068 textlist = text.split("\n")
1065 textlist = text.split("\n")
1069 linestr = checkbox + textlist[0]
1066 linestr = checkbox + textlist[0]
1070 else:
1067 else:
1071 linestr = checkbox + header.filename()
1068 linestr = checkbox + header.filename()
1072 outstr += self.printstring(self.chunkpad, linestr, pair=colorpair,
1069 outstr += self.printstring(self.chunkpad, linestr, pair=colorpair,
1073 towin=towin)
1070 towin=towin)
1074 if not header.folded or ignorefolding:
1071 if not header.folded or ignorefolding:
1075 if len(textlist) > 1:
1072 if len(textlist) > 1:
1076 for line in textlist[1:]:
1073 for line in textlist[1:]:
1077 linestr = " "*(indentnumchars + len(checkbox)) + line
1074 linestr = " "*(indentnumchars + len(checkbox)) + line
1078 outstr += self.printstring(self.chunkpad, linestr,
1075 outstr += self.printstring(self.chunkpad, linestr,
1079 pair=colorpair, towin=towin)
1076 pair=colorpair, towin=towin)
1080
1077
1081 return outstr
1078 return outstr
1082
1079
1083 def printhunklinesbefore(self, hunk, selected=False, towin=True,
1080 def printhunklinesbefore(self, hunk, selected=False, towin=True,
1084 ignorefolding=False):
1081 ignorefolding=False):
1085 "includes start/end line indicator"
1082 "includes start/end line indicator"
1086 outstr = ""
1083 outstr = ""
1087 # where hunk is in list of siblings
1084 # where hunk is in list of siblings
1088 hunkindex = hunk.header.hunks.index(hunk)
1085 hunkindex = hunk.header.hunks.index(hunk)
1089
1086
1090 if hunkindex != 0:
1087 if hunkindex != 0:
1091 # add separating line before headers
1088 # add separating line before headers
1092 outstr += self.printstring(self.chunkpad, ' '*self.xscreensize,
1089 outstr += self.printstring(self.chunkpad, ' '*self.xscreensize,
1093 towin=towin, align=False)
1090 towin=towin, align=False)
1094
1091
1095 colorpair = self.getcolorpair(name=selected and "selected" or "normal",
1092 colorpair = self.getcolorpair(name=selected and "selected" or "normal",
1096 attrlist=[curses.A_BOLD])
1093 attrlist=[curses.A_BOLD])
1097
1094
1098 # print out from-to line with checkbox
1095 # print out from-to line with checkbox
1099 checkbox = self.getstatusprefixstring(hunk)
1096 checkbox = self.getstatusprefixstring(hunk)
1100
1097
1101 lineprefix = " "*self.hunkindentnumchars + checkbox
1098 lineprefix = " "*self.hunkindentnumchars + checkbox
1102 frtoline = " " + hunk.getfromtoline().strip("\n")
1099 frtoline = " " + hunk.getfromtoline().strip("\n")
1103
1100
1104 outstr += self.printstring(self.chunkpad, lineprefix, towin=towin,
1101 outstr += self.printstring(self.chunkpad, lineprefix, towin=towin,
1105 align=False) # add uncolored checkbox/indent
1102 align=False) # add uncolored checkbox/indent
1106 outstr += self.printstring(self.chunkpad, frtoline, pair=colorpair,
1103 outstr += self.printstring(self.chunkpad, frtoline, pair=colorpair,
1107 towin=towin)
1104 towin=towin)
1108
1105
1109 if hunk.folded and not ignorefolding:
1106 if hunk.folded and not ignorefolding:
1110 # skip remainder of output
1107 # skip remainder of output
1111 return outstr
1108 return outstr
1112
1109
1113 # print out lines of the chunk preceeding changed-lines
1110 # print out lines of the chunk preceeding changed-lines
1114 for line in hunk.before:
1111 for line in hunk.before:
1115 linestr = " "*(self.hunklineindentnumchars + len(checkbox)) + line
1112 linestr = " "*(self.hunklineindentnumchars + len(checkbox)) + line
1116 outstr += self.printstring(self.chunkpad, linestr, towin=towin)
1113 outstr += self.printstring(self.chunkpad, linestr, towin=towin)
1117
1114
1118 return outstr
1115 return outstr
1119
1116
1120 def printhunklinesafter(self, hunk, towin=True, ignorefolding=False):
1117 def printhunklinesafter(self, hunk, towin=True, ignorefolding=False):
1121 outstr = ""
1118 outstr = ""
1122 if hunk.folded and not ignorefolding:
1119 if hunk.folded and not ignorefolding:
1123 return outstr
1120 return outstr
1124
1121
1125 # a bit superfluous, but to avoid hard-coding indent amount
1122 # a bit superfluous, but to avoid hard-coding indent amount
1126 checkbox = self.getstatusprefixstring(hunk)
1123 checkbox = self.getstatusprefixstring(hunk)
1127 for line in hunk.after:
1124 for line in hunk.after:
1128 linestr = " "*(self.hunklineindentnumchars + len(checkbox)) + line
1125 linestr = " "*(self.hunklineindentnumchars + len(checkbox)) + line
1129 outstr += self.printstring(self.chunkpad, linestr, towin=towin)
1126 outstr += self.printstring(self.chunkpad, linestr, towin=towin)
1130
1127
1131 return outstr
1128 return outstr
1132
1129
1133 def printhunkchangedline(self, hunkline, selected=False, towin=True):
1130 def printhunkchangedline(self, hunkline, selected=False, towin=True):
1134 outstr = ""
1131 outstr = ""
1135 checkbox = self.getstatusprefixstring(hunkline)
1132 checkbox = self.getstatusprefixstring(hunkline)
1136
1133
1137 linestr = hunkline.prettystr().strip("\n")
1134 linestr = hunkline.prettystr().strip("\n")
1138
1135
1139 # select color-pair based on whether line is an addition/removal
1136 # select color-pair based on whether line is an addition/removal
1140 if selected:
1137 if selected:
1141 colorpair = self.getcolorpair(name="selected")
1138 colorpair = self.getcolorpair(name="selected")
1142 elif linestr.startswith("+"):
1139 elif linestr.startswith("+"):
1143 colorpair = self.getcolorpair(name="addition")
1140 colorpair = self.getcolorpair(name="addition")
1144 elif linestr.startswith("-"):
1141 elif linestr.startswith("-"):
1145 colorpair = self.getcolorpair(name="deletion")
1142 colorpair = self.getcolorpair(name="deletion")
1146 elif linestr.startswith("\\"):
1143 elif linestr.startswith("\\"):
1147 colorpair = self.getcolorpair(name="normal")
1144 colorpair = self.getcolorpair(name="normal")
1148
1145
1149 lineprefix = " "*self.hunklineindentnumchars + checkbox
1146 lineprefix = " "*self.hunklineindentnumchars + checkbox
1150 outstr += self.printstring(self.chunkpad, lineprefix, towin=towin,
1147 outstr += self.printstring(self.chunkpad, lineprefix, towin=towin,
1151 align=False) # add uncolored checkbox/indent
1148 align=False) # add uncolored checkbox/indent
1152 outstr += self.printstring(self.chunkpad, linestr, pair=colorpair,
1149 outstr += self.printstring(self.chunkpad, linestr, pair=colorpair,
1153 towin=towin, showwhtspc=True)
1150 towin=towin, showwhtspc=True)
1154 return outstr
1151 return outstr
1155
1152
1156 def printitem(self, item=None, ignorefolding=False, recursechildren=True,
1153 def printitem(self, item=None, ignorefolding=False, recursechildren=True,
1157 towin=True):
1154 towin=True):
1158 """
1155 """
1159 use __printitem() to print the the specified item.applied.
1156 use __printitem() to print the the specified item.applied.
1160 if item is not specified, then print the entire patch.
1157 if item is not specified, then print the entire patch.
1161 (hiding folded elements, etc. -- see __printitem() docstring)
1158 (hiding folded elements, etc. -- see __printitem() docstring)
1162 """
1159 """
1163
1160
1164 if item is None:
1161 if item is None:
1165 item = self.headerlist
1162 item = self.headerlist
1166 if recursechildren:
1163 if recursechildren:
1167 self.linesprintedtopadsofar = 0
1164 self.linesprintedtopadsofar = 0
1168
1165
1169 outstr = []
1166 outstr = []
1170 self.__printitem(item, ignorefolding, recursechildren, outstr,
1167 self.__printitem(item, ignorefolding, recursechildren, outstr,
1171 towin=towin)
1168 towin=towin)
1172 return ''.join(outstr)
1169 return ''.join(outstr)
1173
1170
1174 def outofdisplayedarea(self):
1171 def outofdisplayedarea(self):
1175 y, _ = self.chunkpad.getyx() # cursor location
1172 y, _ = self.chunkpad.getyx() # cursor location
1176 # * 2 here works but an optimization would be the max number of
1173 # * 2 here works but an optimization would be the max number of
1177 # consecutive non selectable lines
1174 # consecutive non selectable lines
1178 # i.e the max number of context line for any hunk in the patch
1175 # i.e the max number of context line for any hunk in the patch
1179 miny = min(0, self.firstlineofpadtoprint - self.yscreensize)
1176 miny = min(0, self.firstlineofpadtoprint - self.yscreensize)
1180 maxy = self.firstlineofpadtoprint + self.yscreensize * 2
1177 maxy = self.firstlineofpadtoprint + self.yscreensize * 2
1181 return y < miny or y > maxy
1178 return y < miny or y > maxy
1182
1179
1183 def handleselection(self, item, recursechildren):
1180 def handleselection(self, item, recursechildren):
1184 selected = (item is self.currentselecteditem)
1181 selected = (item is self.currentselecteditem)
1185 if selected and recursechildren:
1182 if selected and recursechildren:
1186 # assumes line numbering starting from line 0
1183 # assumes line numbering starting from line 0
1187 self.selecteditemstartline = self.linesprintedtopadsofar
1184 self.selecteditemstartline = self.linesprintedtopadsofar
1188 selecteditemlines = self.getnumlinesdisplayed(item,
1185 selecteditemlines = self.getnumlinesdisplayed(item,
1189 recursechildren=False)
1186 recursechildren=False)
1190 self.selecteditemendline = (self.selecteditemstartline +
1187 self.selecteditemendline = (self.selecteditemstartline +
1191 selecteditemlines - 1)
1188 selecteditemlines - 1)
1192 return selected
1189 return selected
1193
1190
1194 def __printitem(self, item, ignorefolding, recursechildren, outstr,
1191 def __printitem(self, item, ignorefolding, recursechildren, outstr,
1195 towin=True):
1192 towin=True):
1196 """
1193 """
1197 recursive method for printing out patch/header/hunk/hunk-line data to
1194 recursive method for printing out patch/header/hunk/hunk-line data to
1198 screen. also returns a string with all of the content of the displayed
1195 screen. also returns a string with all of the content of the displayed
1199 patch (not including coloring, etc.).
1196 patch (not including coloring, etc.).
1200
1197
1201 if ignorefolding is True, then folded items are printed out.
1198 if ignorefolding is True, then folded items are printed out.
1202
1199
1203 if recursechildren is False, then only print the item without its
1200 if recursechildren is False, then only print the item without its
1204 child items.
1201 child items.
1205 """
1202 """
1206
1203
1207 if towin and self.outofdisplayedarea():
1204 if towin and self.outofdisplayedarea():
1208 return
1205 return
1209
1206
1210 selected = self.handleselection(item, recursechildren)
1207 selected = self.handleselection(item, recursechildren)
1211
1208
1212 # patch object is a list of headers
1209 # patch object is a list of headers
1213 if isinstance(item, patch):
1210 if isinstance(item, patch):
1214 if recursechildren:
1211 if recursechildren:
1215 for hdr in item:
1212 for hdr in item:
1216 self.__printitem(hdr, ignorefolding,
1213 self.__printitem(hdr, ignorefolding,
1217 recursechildren, outstr, towin)
1214 recursechildren, outstr, towin)
1218 # todo: eliminate all isinstance() calls
1215 # todo: eliminate all isinstance() calls
1219 if isinstance(item, uiheader):
1216 if isinstance(item, uiheader):
1220 outstr.append(self.printheader(item, selected, towin=towin,
1217 outstr.append(self.printheader(item, selected, towin=towin,
1221 ignorefolding=ignorefolding))
1218 ignorefolding=ignorefolding))
1222 if recursechildren:
1219 if recursechildren:
1223 for hnk in item.hunks:
1220 for hnk in item.hunks:
1224 self.__printitem(hnk, ignorefolding,
1221 self.__printitem(hnk, ignorefolding,
1225 recursechildren, outstr, towin)
1222 recursechildren, outstr, towin)
1226 elif (isinstance(item, uihunk) and
1223 elif (isinstance(item, uihunk) and
1227 ((not item.header.folded) or ignorefolding)):
1224 ((not item.header.folded) or ignorefolding)):
1228 # print the hunk data which comes before the changed-lines
1225 # print the hunk data which comes before the changed-lines
1229 outstr.append(self.printhunklinesbefore(item, selected, towin=towin,
1226 outstr.append(self.printhunklinesbefore(item, selected, towin=towin,
1230 ignorefolding=ignorefolding))
1227 ignorefolding=ignorefolding))
1231 if recursechildren:
1228 if recursechildren:
1232 for l in item.changedlines:
1229 for l in item.changedlines:
1233 self.__printitem(l, ignorefolding,
1230 self.__printitem(l, ignorefolding,
1234 recursechildren, outstr, towin)
1231 recursechildren, outstr, towin)
1235 outstr.append(self.printhunklinesafter(item, towin=towin,
1232 outstr.append(self.printhunklinesafter(item, towin=towin,
1236 ignorefolding=ignorefolding))
1233 ignorefolding=ignorefolding))
1237 elif (isinstance(item, uihunkline) and
1234 elif (isinstance(item, uihunkline) and
1238 ((not item.hunk.folded) or ignorefolding)):
1235 ((not item.hunk.folded) or ignorefolding)):
1239 outstr.append(self.printhunkchangedline(item, selected,
1236 outstr.append(self.printhunkchangedline(item, selected,
1240 towin=towin))
1237 towin=towin))
1241
1238
1242 return outstr
1239 return outstr
1243
1240
1244 def getnumlinesdisplayed(self, item=None, ignorefolding=False,
1241 def getnumlinesdisplayed(self, item=None, ignorefolding=False,
1245 recursechildren=True):
1242 recursechildren=True):
1246 """
1243 """
1247 return the number of lines which would be displayed if the item were
1244 return the number of lines which would be displayed if the item were
1248 to be printed to the display. the item will not be printed to the
1245 to be printed to the display. the item will not be printed to the
1249 display (pad).
1246 display (pad).
1250 if no item is given, assume the entire patch.
1247 if no item is given, assume the entire patch.
1251 if ignorefolding is True, folded items will be unfolded when counting
1248 if ignorefolding is True, folded items will be unfolded when counting
1252 the number of lines.
1249 the number of lines.
1253 """
1250 """
1254
1251
1255 # temporarily disable printing to windows by printstring
1252 # temporarily disable printing to windows by printstring
1256 patchdisplaystring = self.printitem(item, ignorefolding,
1253 patchdisplaystring = self.printitem(item, ignorefolding,
1257 recursechildren, towin=False)
1254 recursechildren, towin=False)
1258 numlines = len(patchdisplaystring) / self.xscreensize
1255 numlines = len(patchdisplaystring) / self.xscreensize
1259 return numlines
1256 return numlines
1260
1257
1261 def sigwinchhandler(self, n, frame):
1258 def sigwinchhandler(self, n, frame):
1262 "handle window resizing"
1259 "handle window resizing"
1263 try:
1260 try:
1264 curses.endwin()
1261 curses.endwin()
1265 self.yscreensize, self.xscreensize = gethw()
1262 self.yscreensize, self.xscreensize = gethw()
1266 self.statuswin.resize(self.numstatuslines, self.xscreensize)
1263 self.statuswin.resize(self.numstatuslines, self.xscreensize)
1267 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1264 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1268 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1265 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1269 # todo: try to resize commit message window if possible
1266 # todo: try to resize commit message window if possible
1270 except curses.error:
1267 except curses.error:
1271 pass
1268 pass
1272
1269
1273 def getcolorpair(self, fgcolor=None, bgcolor=None, name=None,
1270 def getcolorpair(self, fgcolor=None, bgcolor=None, name=None,
1274 attrlist=None):
1271 attrlist=None):
1275 """
1272 """
1276 get a curses color pair, adding it to self.colorpairs if it is not
1273 get a curses color pair, adding it to self.colorpairs if it is not
1277 already defined. an optional string, name, can be passed as a shortcut
1274 already defined. an optional string, name, can be passed as a shortcut
1278 for referring to the color-pair. by default, if no arguments are
1275 for referring to the color-pair. by default, if no arguments are
1279 specified, the white foreground / black background color-pair is
1276 specified, the white foreground / black background color-pair is
1280 returned.
1277 returned.
1281
1278
1282 it is expected that this function will be used exclusively for
1279 it is expected that this function will be used exclusively for
1283 initializing color pairs, and not curses.init_pair().
1280 initializing color pairs, and not curses.init_pair().
1284
1281
1285 attrlist is used to 'flavor' the returned color-pair. this information
1282 attrlist is used to 'flavor' the returned color-pair. this information
1286 is not stored in self.colorpairs. it contains attribute values like
1283 is not stored in self.colorpairs. it contains attribute values like
1287 curses.A_BOLD.
1284 curses.A_BOLD.
1288 """
1285 """
1289
1286
1290 if (name is not None) and name in self.colorpairnames:
1287 if (name is not None) and name in self.colorpairnames:
1291 # then get the associated color pair and return it
1288 # then get the associated color pair and return it
1292 colorpair = self.colorpairnames[name]
1289 colorpair = self.colorpairnames[name]
1293 else:
1290 else:
1294 if fgcolor is None:
1291 if fgcolor is None:
1295 fgcolor = -1
1292 fgcolor = -1
1296 if bgcolor is None:
1293 if bgcolor is None:
1297 bgcolor = -1
1294 bgcolor = -1
1298 if (fgcolor, bgcolor) in self.colorpairs:
1295 if (fgcolor, bgcolor) in self.colorpairs:
1299 colorpair = self.colorpairs[(fgcolor, bgcolor)]
1296 colorpair = self.colorpairs[(fgcolor, bgcolor)]
1300 else:
1297 else:
1301 pairindex = len(self.colorpairs) + 1
1298 pairindex = len(self.colorpairs) + 1
1302 curses.init_pair(pairindex, fgcolor, bgcolor)
1299 curses.init_pair(pairindex, fgcolor, bgcolor)
1303 colorpair = self.colorpairs[(fgcolor, bgcolor)] = (
1300 colorpair = self.colorpairs[(fgcolor, bgcolor)] = (
1304 curses.color_pair(pairindex))
1301 curses.color_pair(pairindex))
1305 if name is not None:
1302 if name is not None:
1306 self.colorpairnames[name] = curses.color_pair(pairindex)
1303 self.colorpairnames[name] = curses.color_pair(pairindex)
1307
1304
1308 # add attributes if possible
1305 # add attributes if possible
1309 if attrlist is None:
1306 if attrlist is None:
1310 attrlist = []
1307 attrlist = []
1311 if colorpair < 256:
1308 if colorpair < 256:
1312 # then it is safe to apply all attributes
1309 # then it is safe to apply all attributes
1313 for textattr in attrlist:
1310 for textattr in attrlist:
1314 colorpair |= textattr
1311 colorpair |= textattr
1315 else:
1312 else:
1316 # just apply a select few (safe?) attributes
1313 # just apply a select few (safe?) attributes
1317 for textattrib in (curses.A_UNDERLINE, curses.A_BOLD):
1314 for textattrib in (curses.A_UNDERLINE, curses.A_BOLD):
1318 if textattrib in attrlist:
1315 if textattrib in attrlist:
1319 colorpair |= textattrib
1316 colorpair |= textattrib
1320 return colorpair
1317 return colorpair
1321
1318
1322 def initcolorpair(self, *args, **kwargs):
1319 def initcolorpair(self, *args, **kwargs):
1323 "same as getcolorpair."
1320 "same as getcolorpair."
1324 self.getcolorpair(*args, **kwargs)
1321 self.getcolorpair(*args, **kwargs)
1325
1322
1326 def helpwindow(self):
1323 def helpwindow(self):
1327 "print a help window to the screen. exit after any keypress."
1324 "print a help window to the screen. exit after any keypress."
1328 helptext = """ [press any key to return to the patch-display]
1325 helptext = """ [press any key to return to the patch-display]
1329
1326
1330 crecord allows you to interactively choose among the changes you have made,
1327 crecord allows you to interactively choose among the changes you have made,
1331 and confirm only those changes you select for further processing by the command
1328 and confirm only those changes you select for further processing by the command
1332 you are running (commit/shelve/revert), after confirming the selected
1329 you are running (commit/shelve/revert), after confirming the selected
1333 changes, the unselected changes are still present in your working copy, so you
1330 changes, the unselected changes are still present in your working copy, so you
1334 can use crecord multiple times to split large changes into smaller changesets.
1331 can use crecord multiple times to split large changes into smaller changesets.
1335 the following are valid keystrokes:
1332 the following are valid keystrokes:
1336
1333
1337 [space] : (un-)select item ([~]/[x] = partly/fully applied)
1334 [space] : (un-)select item ([~]/[x] = partly/fully applied)
1338 A : (un-)select all items
1335 A : (un-)select all items
1339 up/down-arrow [k/j] : go to previous/next unfolded item
1336 up/down-arrow [k/j] : go to previous/next unfolded item
1340 pgup/pgdn [K/J] : go to previous/next item of same type
1337 pgup/pgdn [K/J] : go to previous/next item of same type
1341 right/left-arrow [l/h] : go to child item / parent item
1338 right/left-arrow [l/h] : go to child item / parent item
1342 shift-left-arrow [H] : go to parent header / fold selected header
1339 shift-left-arrow [H] : go to parent header / fold selected header
1343 f : fold / unfold item, hiding/revealing its children
1340 f : fold / unfold item, hiding/revealing its children
1344 F : fold / unfold parent item and all of its ancestors
1341 F : fold / unfold parent item and all of its ancestors
1345 m : edit / resume editing the commit message
1342 m : edit / resume editing the commit message
1346 e : edit the currently selected hunk
1343 e : edit the currently selected hunk
1347 a : toggle amend mode, only with commit -i
1344 a : toggle amend mode, only with commit -i
1348 c : confirm selected changes
1345 c : confirm selected changes
1349 r : review/edit and confirm selected changes
1346 r : review/edit and confirm selected changes
1350 q : quit without confirming (no changes will be made)
1347 q : quit without confirming (no changes will be made)
1351 ? : help (what you're currently reading)"""
1348 ? : help (what you're currently reading)"""
1352
1349
1353 helpwin = curses.newwin(self.yscreensize, 0, 0, 0)
1350 helpwin = curses.newwin(self.yscreensize, 0, 0, 0)
1354 helplines = helptext.split("\n")
1351 helplines = helptext.split("\n")
1355 helplines = helplines + [" "]*(
1352 helplines = helplines + [" "]*(
1356 self.yscreensize - self.numstatuslines - len(helplines) - 1)
1353 self.yscreensize - self.numstatuslines - len(helplines) - 1)
1357 try:
1354 try:
1358 for line in helplines:
1355 for line in helplines:
1359 self.printstring(helpwin, line, pairname="legend")
1356 self.printstring(helpwin, line, pairname="legend")
1360 except curses.error:
1357 except curses.error:
1361 pass
1358 pass
1362 helpwin.refresh()
1359 helpwin.refresh()
1363 try:
1360 try:
1364 helpwin.getkey()
1361 helpwin.getkey()
1365 except curses.error:
1362 except curses.error:
1366 pass
1363 pass
1367
1364
1368 def confirmationwindow(self, windowtext):
1365 def confirmationwindow(self, windowtext):
1369 "display an informational window, then wait for and return a keypress."
1366 "display an informational window, then wait for and return a keypress."
1370
1367
1371 confirmwin = curses.newwin(self.yscreensize, 0, 0, 0)
1368 confirmwin = curses.newwin(self.yscreensize, 0, 0, 0)
1372 try:
1369 try:
1373 lines = windowtext.split("\n")
1370 lines = windowtext.split("\n")
1374 for line in lines:
1371 for line in lines:
1375 self.printstring(confirmwin, line, pairname="selected")
1372 self.printstring(confirmwin, line, pairname="selected")
1376 except curses.error:
1373 except curses.error:
1377 pass
1374 pass
1378 self.stdscr.refresh()
1375 self.stdscr.refresh()
1379 confirmwin.refresh()
1376 confirmwin.refresh()
1380 try:
1377 try:
1381 response = chr(self.stdscr.getch())
1378 response = chr(self.stdscr.getch())
1382 except ValueError:
1379 except ValueError:
1383 response = None
1380 response = None
1384
1381
1385 return response
1382 return response
1386
1383
1387 def reviewcommit(self):
1384 def reviewcommit(self):
1388 """ask for 'y' to be pressed to confirm selected. return True if
1385 """ask for 'y' to be pressed to confirm selected. return True if
1389 confirmed."""
1386 confirmed."""
1390 confirmtext = (
1387 confirmtext = (
1391 """if you answer yes to the following, the your currently chosen patch chunks
1388 """if you answer yes to the following, the your currently chosen patch chunks
1392 will be loaded into an editor. you may modify the patch from the editor, and
1389 will be loaded into an editor. you may modify the patch from the editor, and
1393 save the changes if you wish to change the patch. otherwise, you can just
1390 save the changes if you wish to change the patch. otherwise, you can just
1394 close the editor without saving to accept the current patch as-is.
1391 close the editor without saving to accept the current patch as-is.
1395
1392
1396 note: don't add/remove lines unless you also modify the range information.
1393 note: don't add/remove lines unless you also modify the range information.
1397 failing to follow this rule will result in the commit aborting.
1394 failing to follow this rule will result in the commit aborting.
1398
1395
1399 are you sure you want to review/edit and confirm the selected changes [yn]?
1396 are you sure you want to review/edit and confirm the selected changes [yn]?
1400 """)
1397 """)
1401 response = self.confirmationwindow(confirmtext)
1398 response = self.confirmationwindow(confirmtext)
1402 if response is None:
1399 if response is None:
1403 response = "n"
1400 response = "n"
1404 if response.lower().startswith("y"):
1401 if response.lower().startswith("y"):
1405 return True
1402 return True
1406 else:
1403 else:
1407 return False
1404 return False
1408
1405
1409 def toggleamend(self, opts, test):
1406 def toggleamend(self, opts, test):
1410 """Toggle the amend flag.
1407 """Toggle the amend flag.
1411
1408
1412 When the amend flag is set, a commit will modify the most recently
1409 When the amend flag is set, a commit will modify the most recently
1413 committed changeset, instead of creating a new changeset. Otherwise, a
1410 committed changeset, instead of creating a new changeset. Otherwise, a
1414 new changeset will be created (the normal commit behavior).
1411 new changeset will be created (the normal commit behavior).
1415 """
1412 """
1416
1413
1417 try:
1414 try:
1418 ver = float(util.version()[:3])
1415 ver = float(util.version()[:3])
1419 except ValueError:
1416 except ValueError:
1420 ver = 1
1417 ver = 1
1421 if ver < 2.19:
1418 if ver < 2.19:
1422 msg = ("The amend option is unavailable with hg versions < 2.2\n\n"
1419 msg = ("The amend option is unavailable with hg versions < 2.2\n\n"
1423 "Press any key to continue.")
1420 "Press any key to continue.")
1424 elif opts.get('amend') is None:
1421 elif opts.get('amend') is None:
1425 opts['amend'] = True
1422 opts['amend'] = True
1426 msg = ("Amend option is turned on -- commiting the currently "
1423 msg = ("Amend option is turned on -- commiting the currently "
1427 "selected changes will not create a new changeset, but "
1424 "selected changes will not create a new changeset, but "
1428 "instead update the most recently committed changeset.\n\n"
1425 "instead update the most recently committed changeset.\n\n"
1429 "Press any key to continue.")
1426 "Press any key to continue.")
1430 elif opts.get('amend') is True:
1427 elif opts.get('amend') is True:
1431 opts['amend'] = None
1428 opts['amend'] = None
1432 msg = ("Amend option is turned off -- commiting the currently "
1429 msg = ("Amend option is turned off -- commiting the currently "
1433 "selected changes will create a new changeset.\n\n"
1430 "selected changes will create a new changeset.\n\n"
1434 "Press any key to continue.")
1431 "Press any key to continue.")
1435 if not test:
1432 if not test:
1436 self.confirmationwindow(msg)
1433 self.confirmationwindow(msg)
1437
1434
1438 def recenterdisplayedarea(self):
1435 def recenterdisplayedarea(self):
1439 """
1436 """
1440 once we scrolled with pg up pg down we can be pointing outside of the
1437 once we scrolled with pg up pg down we can be pointing outside of the
1441 display zone. we print the patch with towin=False to compute the
1438 display zone. we print the patch with towin=False to compute the
1442 location of the selected item even though it is outside of the displayed
1439 location of the selected item even though it is outside of the displayed
1443 zone and then update the scroll.
1440 zone and then update the scroll.
1444 """
1441 """
1445 self.printitem(towin=False)
1442 self.printitem(towin=False)
1446 self.updatescroll()
1443 self.updatescroll()
1447
1444
1448 def toggleedit(self, item=None, test=False):
1445 def toggleedit(self, item=None, test=False):
1449 """
1446 """
1450 edit the currently selected chunk
1447 edit the currently selected chunk
1451 """
1448 """
1452 def updateui(self):
1449 def updateui(self):
1453 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1450 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1454 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1451 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1455 self.updatescroll()
1452 self.updatescroll()
1456 self.stdscr.refresh()
1453 self.stdscr.refresh()
1457 self.statuswin.refresh()
1454 self.statuswin.refresh()
1458 self.stdscr.keypad(1)
1455 self.stdscr.keypad(1)
1459
1456
1460 def editpatchwitheditor(self, chunk):
1457 def editpatchwitheditor(self, chunk):
1461 if chunk is None:
1458 if chunk is None:
1462 self.ui.write(_('cannot edit patch for whole file'))
1459 self.ui.write(_('cannot edit patch for whole file'))
1463 self.ui.write("\n")
1460 self.ui.write("\n")
1464 return None
1461 return None
1465 if chunk.header.binary():
1462 if chunk.header.binary():
1466 self.ui.write(_('cannot edit patch for binary file'))
1463 self.ui.write(_('cannot edit patch for binary file'))
1467 self.ui.write("\n")
1464 self.ui.write("\n")
1468 return None
1465 return None
1469
1466
1470 # write the initial patch
1467 # write the initial patch
1471 patch = stringio()
1468 patch = stringio()
1472 patch.write(diffhelptext + hunkhelptext)
1469 patch.write(diffhelptext + hunkhelptext)
1473 chunk.header.write(patch)
1470 chunk.header.write(patch)
1474 chunk.write(patch)
1471 chunk.write(patch)
1475
1472
1476 # start the editor and wait for it to complete
1473 # start the editor and wait for it to complete
1477 try:
1474 try:
1478 patch = self.ui.edit(patch.getvalue(), "",
1475 patch = self.ui.edit(patch.getvalue(), "",
1479 extra={"suffix": ".diff"})
1476 extra={"suffix": ".diff"})
1480 except error.Abort as exc:
1477 except error.Abort as exc:
1481 self.errorstr = str(exc)
1478 self.errorstr = str(exc)
1482 return None
1479 return None
1483
1480
1484 # remove comment lines
1481 # remove comment lines
1485 patch = [line + '\n' for line in patch.splitlines()
1482 patch = [line + '\n' for line in patch.splitlines()
1486 if not line.startswith('#')]
1483 if not line.startswith('#')]
1487 return patchmod.parsepatch(patch)
1484 return patchmod.parsepatch(patch)
1488
1485
1489 if item is None:
1486 if item is None:
1490 item = self.currentselecteditem
1487 item = self.currentselecteditem
1491 if isinstance(item, uiheader):
1488 if isinstance(item, uiheader):
1492 return
1489 return
1493 if isinstance(item, uihunkline):
1490 if isinstance(item, uihunkline):
1494 item = item.parentitem()
1491 item = item.parentitem()
1495 if not isinstance(item, uihunk):
1492 if not isinstance(item, uihunk):
1496 return
1493 return
1497
1494
1498 # To go back to that hunk or its replacement at the end of the edit
1495 # To go back to that hunk or its replacement at the end of the edit
1499 itemindex = item.parentitem().hunks.index(item)
1496 itemindex = item.parentitem().hunks.index(item)
1500
1497
1501 beforeadded, beforeremoved = item.added, item.removed
1498 beforeadded, beforeremoved = item.added, item.removed
1502 newpatches = editpatchwitheditor(self, item)
1499 newpatches = editpatchwitheditor(self, item)
1503 if newpatches is None:
1500 if newpatches is None:
1504 if not test:
1501 if not test:
1505 updateui(self)
1502 updateui(self)
1506 return
1503 return
1507 header = item.header
1504 header = item.header
1508 editedhunkindex = header.hunks.index(item)
1505 editedhunkindex = header.hunks.index(item)
1509 hunksbefore = header.hunks[:editedhunkindex]
1506 hunksbefore = header.hunks[:editedhunkindex]
1510 hunksafter = header.hunks[editedhunkindex + 1:]
1507 hunksafter = header.hunks[editedhunkindex + 1:]
1511 newpatchheader = newpatches[0]
1508 newpatchheader = newpatches[0]
1512 newhunks = [uihunk(h, header) for h in newpatchheader.hunks]
1509 newhunks = [uihunk(h, header) for h in newpatchheader.hunks]
1513 newadded = sum([h.added for h in newhunks])
1510 newadded = sum([h.added for h in newhunks])
1514 newremoved = sum([h.removed for h in newhunks])
1511 newremoved = sum([h.removed for h in newhunks])
1515 offset = (newadded - beforeadded) - (newremoved - beforeremoved)
1512 offset = (newadded - beforeadded) - (newremoved - beforeremoved)
1516
1513
1517 for h in hunksafter:
1514 for h in hunksafter:
1518 h.toline += offset
1515 h.toline += offset
1519 for h in newhunks:
1516 for h in newhunks:
1520 h.folded = False
1517 h.folded = False
1521 header.hunks = hunksbefore + newhunks + hunksafter
1518 header.hunks = hunksbefore + newhunks + hunksafter
1522 if self.emptypatch():
1519 if self.emptypatch():
1523 header.hunks = hunksbefore + [item] + hunksafter
1520 header.hunks = hunksbefore + [item] + hunksafter
1524 self.currentselecteditem = header
1521 self.currentselecteditem = header
1525 if len(header.hunks) > itemindex:
1522 if len(header.hunks) > itemindex:
1526 self.currentselecteditem = header.hunks[itemindex]
1523 self.currentselecteditem = header.hunks[itemindex]
1527
1524
1528 if not test:
1525 if not test:
1529 updateui(self)
1526 updateui(self)
1530
1527
1531 def emptypatch(self):
1528 def emptypatch(self):
1532 item = self.headerlist
1529 item = self.headerlist
1533 if not item:
1530 if not item:
1534 return True
1531 return True
1535 for header in item:
1532 for header in item:
1536 if header.hunks:
1533 if header.hunks:
1537 return False
1534 return False
1538 return True
1535 return True
1539
1536
1540 def handlekeypressed(self, keypressed, test=False):
1537 def handlekeypressed(self, keypressed, test=False):
1541 """
1538 """
1542 Perform actions based on pressed keys.
1539 Perform actions based on pressed keys.
1543
1540
1544 Return true to exit the main loop.
1541 Return true to exit the main loop.
1545 """
1542 """
1546 if keypressed in ["k", "KEY_UP"]:
1543 if keypressed in ["k", "KEY_UP"]:
1547 self.uparrowevent()
1544 self.uparrowevent()
1548 if keypressed in ["K", "KEY_PPAGE"]:
1545 if keypressed in ["K", "KEY_PPAGE"]:
1549 self.uparrowshiftevent()
1546 self.uparrowshiftevent()
1550 elif keypressed in ["j", "KEY_DOWN"]:
1547 elif keypressed in ["j", "KEY_DOWN"]:
1551 self.downarrowevent()
1548 self.downarrowevent()
1552 elif keypressed in ["J", "KEY_NPAGE"]:
1549 elif keypressed in ["J", "KEY_NPAGE"]:
1553 self.downarrowshiftevent()
1550 self.downarrowshiftevent()
1554 elif keypressed in ["l", "KEY_RIGHT"]:
1551 elif keypressed in ["l", "KEY_RIGHT"]:
1555 self.rightarrowevent()
1552 self.rightarrowevent()
1556 elif keypressed in ["h", "KEY_LEFT"]:
1553 elif keypressed in ["h", "KEY_LEFT"]:
1557 self.leftarrowevent()
1554 self.leftarrowevent()
1558 elif keypressed in ["H", "KEY_SLEFT"]:
1555 elif keypressed in ["H", "KEY_SLEFT"]:
1559 self.leftarrowshiftevent()
1556 self.leftarrowshiftevent()
1560 elif keypressed in ["q"]:
1557 elif keypressed in ["q"]:
1561 raise error.Abort(_('user quit'))
1558 raise error.Abort(_('user quit'))
1562 elif keypressed in ['a']:
1559 elif keypressed in ['a']:
1563 self.toggleamend(self.opts, test)
1560 self.toggleamend(self.opts, test)
1564 elif keypressed in ["c"]:
1561 elif keypressed in ["c"]:
1565 return True
1562 return True
1566 elif test and keypressed in ['X']:
1563 elif test and keypressed in ['X']:
1567 return True
1564 return True
1568 elif keypressed in ["r"]:
1565 elif keypressed in ["r"]:
1569 if self.reviewcommit():
1566 if self.reviewcommit():
1570 self.opts['review'] = True
1567 self.opts['review'] = True
1571 return True
1568 return True
1572 elif test and keypressed in ['R']:
1569 elif test and keypressed in ['R']:
1573 self.opts['review'] = True
1570 self.opts['review'] = True
1574 return True
1571 return True
1575 elif keypressed in [' '] or (test and keypressed in ["TOGGLE"]):
1572 elif keypressed in [' '] or (test and keypressed in ["TOGGLE"]):
1576 self.toggleapply()
1573 self.toggleapply()
1577 elif keypressed in ['A']:
1574 elif keypressed in ['A']:
1578 self.toggleall()
1575 self.toggleall()
1579 elif keypressed in ['e']:
1576 elif keypressed in ['e']:
1580 self.toggleedit(test=test)
1577 self.toggleedit(test=test)
1581 elif keypressed in ["f"]:
1578 elif keypressed in ["f"]:
1582 self.togglefolded()
1579 self.togglefolded()
1583 elif keypressed in ["F"]:
1580 elif keypressed in ["F"]:
1584 self.togglefolded(foldparent=True)
1581 self.togglefolded(foldparent=True)
1585 elif keypressed in ["?"]:
1582 elif keypressed in ["?"]:
1586 self.helpwindow()
1583 self.helpwindow()
1587 self.stdscr.clear()
1584 self.stdscr.clear()
1588 self.stdscr.refresh()
1585 self.stdscr.refresh()
1589
1586
1590 def main(self, stdscr):
1587 def main(self, stdscr):
1591 """
1588 """
1592 method to be wrapped by curses.wrapper() for selecting chunks.
1589 method to be wrapped by curses.wrapper() for selecting chunks.
1593 """
1590 """
1594
1591
1595 signal.signal(signal.SIGWINCH, self.sigwinchhandler)
1592 signal.signal(signal.SIGWINCH, self.sigwinchhandler)
1596 self.stdscr = stdscr
1593 self.stdscr = stdscr
1597 # error during initialization, cannot be printed in the curses
1594 # error during initialization, cannot be printed in the curses
1598 # interface, it should be printed by the calling code
1595 # interface, it should be printed by the calling code
1599 self.initerr = None
1596 self.initerr = None
1600 self.yscreensize, self.xscreensize = self.stdscr.getmaxyx()
1597 self.yscreensize, self.xscreensize = self.stdscr.getmaxyx()
1601
1598
1602 curses.start_color()
1599 curses.start_color()
1603 curses.use_default_colors()
1600 curses.use_default_colors()
1604
1601
1605 # available colors: black, blue, cyan, green, magenta, white, yellow
1602 # available colors: black, blue, cyan, green, magenta, white, yellow
1606 # init_pair(color_id, foreground_color, background_color)
1603 # init_pair(color_id, foreground_color, background_color)
1607 self.initcolorpair(None, None, name="normal")
1604 self.initcolorpair(None, None, name="normal")
1608 self.initcolorpair(curses.COLOR_WHITE, curses.COLOR_MAGENTA,
1605 self.initcolorpair(curses.COLOR_WHITE, curses.COLOR_MAGENTA,
1609 name="selected")
1606 name="selected")
1610 self.initcolorpair(curses.COLOR_RED, None, name="deletion")
1607 self.initcolorpair(curses.COLOR_RED, None, name="deletion")
1611 self.initcolorpair(curses.COLOR_GREEN, None, name="addition")
1608 self.initcolorpair(curses.COLOR_GREEN, None, name="addition")
1612 self.initcolorpair(curses.COLOR_WHITE, curses.COLOR_BLUE, name="legend")
1609 self.initcolorpair(curses.COLOR_WHITE, curses.COLOR_BLUE, name="legend")
1613 # newwin([height, width,] begin_y, begin_x)
1610 # newwin([height, width,] begin_y, begin_x)
1614 self.statuswin = curses.newwin(self.numstatuslines, 0, 0, 0)
1611 self.statuswin = curses.newwin(self.numstatuslines, 0, 0, 0)
1615 self.statuswin.keypad(1) # interpret arrow-key, etc. esc sequences
1612 self.statuswin.keypad(1) # interpret arrow-key, etc. esc sequences
1616
1613
1617 # figure out how much space to allocate for the chunk-pad which is
1614 # figure out how much space to allocate for the chunk-pad which is
1618 # used for displaying the patch
1615 # used for displaying the patch
1619
1616
1620 # stupid hack to prevent getnumlinesdisplayed from failing
1617 # stupid hack to prevent getnumlinesdisplayed from failing
1621 self.chunkpad = curses.newpad(1, self.xscreensize)
1618 self.chunkpad = curses.newpad(1, self.xscreensize)
1622
1619
1623 # add 1 so to account for last line text reaching end of line
1620 # add 1 so to account for last line text reaching end of line
1624 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1621 self.numpadlines = self.getnumlinesdisplayed(ignorefolding=True) + 1
1625
1622
1626 try:
1623 try:
1627 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1624 self.chunkpad = curses.newpad(self.numpadlines, self.xscreensize)
1628 except curses.error:
1625 except curses.error:
1629 self.initerr = _('this diff is too large to be displayed')
1626 self.initerr = _('this diff is too large to be displayed')
1630 return
1627 return
1631 # initialize selecteitemendline (initial start-line is 0)
1628 # initialize selecteitemendline (initial start-line is 0)
1632 self.selecteditemendline = self.getnumlinesdisplayed(
1629 self.selecteditemendline = self.getnumlinesdisplayed(
1633 self.currentselecteditem, recursechildren=False)
1630 self.currentselecteditem, recursechildren=False)
1634
1631
1635 while True:
1632 while True:
1636 self.updatescreen()
1633 self.updatescreen()
1637 try:
1634 try:
1638 keypressed = self.statuswin.getkey()
1635 keypressed = self.statuswin.getkey()
1639 if self.errorstr is not None:
1636 if self.errorstr is not None:
1640 self.errorstr = None
1637 self.errorstr = None
1641 continue
1638 continue
1642 except curses.error:
1639 except curses.error:
1643 keypressed = "foobar"
1640 keypressed = "foobar"
1644 if self.handlekeypressed(keypressed):
1641 if self.handlekeypressed(keypressed):
1645 break
1642 break
General Comments 0
You need to be logged in to leave comments. Login now