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