##// END OF EJS Templates
graphlog: preserve topo sort even if additional filter options specified...
Yuya Nishihara -
r30003:46825334 default
parent child Browse files
Show More
@@ -1,3553 +1,3550 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 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 or dir:
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.manifest.dirlog(dir)
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"')
1467 1467 % j(self.ui.popbuffer()))
1468 1468 if diff:
1469 1469 self.ui.pushbuffer()
1470 1470 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
1471 1471 match=matchfn, stat=False)
1472 1472 self.ui.write((',\n "diff": "%s"') % j(self.ui.popbuffer()))
1473 1473
1474 1474 self.ui.write("\n }")
1475 1475
1476 1476 class changeset_templater(changeset_printer):
1477 1477 '''format changeset information.'''
1478 1478
1479 1479 def __init__(self, ui, repo, matchfn, diffopts, tmpl, mapfile, buffered):
1480 1480 changeset_printer.__init__(self, ui, repo, matchfn, diffopts, buffered)
1481 1481 formatnode = ui.debugflag and (lambda x: x) or (lambda x: x[:12])
1482 1482 filters = {'formatnode': formatnode}
1483 1483 defaulttempl = {
1484 1484 'parent': '{rev}:{node|formatnode} ',
1485 1485 'manifest': '{rev}:{node|formatnode}',
1486 1486 'file_copy': '{name} ({source})',
1487 1487 'extra': '{key}={value|stringescape}'
1488 1488 }
1489 1489 # filecopy is preserved for compatibility reasons
1490 1490 defaulttempl['filecopy'] = defaulttempl['file_copy']
1491 1491 assert not (tmpl and mapfile)
1492 1492 if mapfile:
1493 1493 self.t = templater.templater.frommapfile(mapfile, filters=filters,
1494 1494 cache=defaulttempl)
1495 1495 else:
1496 1496 self.t = formatter.maketemplater(ui, 'changeset', tmpl,
1497 1497 filters=filters,
1498 1498 cache=defaulttempl)
1499 1499
1500 1500 self.cache = {}
1501 1501
1502 1502 # find correct templates for current mode
1503 1503 tmplmodes = [
1504 1504 (True, None),
1505 1505 (self.ui.verbose, 'verbose'),
1506 1506 (self.ui.quiet, 'quiet'),
1507 1507 (self.ui.debugflag, 'debug'),
1508 1508 ]
1509 1509
1510 1510 self._parts = {'header': '', 'footer': '', 'changeset': 'changeset',
1511 1511 'docheader': '', 'docfooter': ''}
1512 1512 for mode, postfix in tmplmodes:
1513 1513 for t in self._parts:
1514 1514 cur = t
1515 1515 if postfix:
1516 1516 cur += "_" + postfix
1517 1517 if mode and cur in self.t:
1518 1518 self._parts[t] = cur
1519 1519
1520 1520 if self._parts['docheader']:
1521 1521 self.ui.write(templater.stringify(self.t(self._parts['docheader'])))
1522 1522
1523 1523 def close(self):
1524 1524 if self._parts['docfooter']:
1525 1525 if not self.footer:
1526 1526 self.footer = ""
1527 1527 self.footer += templater.stringify(self.t(self._parts['docfooter']))
1528 1528 return super(changeset_templater, self).close()
1529 1529
1530 1530 def _show(self, ctx, copies, matchfn, props):
1531 1531 '''show a single changeset or file revision'''
1532 1532 props = props.copy()
1533 1533 props.update(templatekw.keywords)
1534 1534 props['templ'] = self.t
1535 1535 props['ctx'] = ctx
1536 1536 props['repo'] = self.repo
1537 1537 props['ui'] = self.repo.ui
1538 1538 props['revcache'] = {'copies': copies}
1539 1539 props['cache'] = self.cache
1540 1540
1541 1541 # write header
1542 1542 if self._parts['header']:
1543 1543 h = templater.stringify(self.t(self._parts['header'], **props))
1544 1544 if self.buffered:
1545 1545 self.header[ctx.rev()] = h
1546 1546 else:
1547 1547 if self.lastheader != h:
1548 1548 self.lastheader = h
1549 1549 self.ui.write(h)
1550 1550
1551 1551 # write changeset metadata, then patch if requested
1552 1552 key = self._parts['changeset']
1553 1553 self.ui.write(templater.stringify(self.t(key, **props)))
1554 1554 self.showpatch(ctx, matchfn)
1555 1555
1556 1556 if self._parts['footer']:
1557 1557 if not self.footer:
1558 1558 self.footer = templater.stringify(
1559 1559 self.t(self._parts['footer'], **props))
1560 1560
1561 1561 def gettemplate(ui, tmpl, style):
1562 1562 """
1563 1563 Find the template matching the given template spec or style.
1564 1564 """
1565 1565
1566 1566 # ui settings
1567 1567 if not tmpl and not style: # template are stronger than style
1568 1568 tmpl = ui.config('ui', 'logtemplate')
1569 1569 if tmpl:
1570 1570 return templater.unquotestring(tmpl), None
1571 1571 else:
1572 1572 style = util.expandpath(ui.config('ui', 'style', ''))
1573 1573
1574 1574 if not tmpl and style:
1575 1575 mapfile = style
1576 1576 if not os.path.split(mapfile)[0]:
1577 1577 mapname = (templater.templatepath('map-cmdline.' + mapfile)
1578 1578 or templater.templatepath(mapfile))
1579 1579 if mapname:
1580 1580 mapfile = mapname
1581 1581 return None, mapfile
1582 1582
1583 1583 if not tmpl:
1584 1584 return None, None
1585 1585
1586 1586 return formatter.lookuptemplate(ui, 'changeset', tmpl)
1587 1587
1588 1588 def show_changeset(ui, repo, opts, buffered=False):
1589 1589 """show one changeset using template or regular display.
1590 1590
1591 1591 Display format will be the first non-empty hit of:
1592 1592 1. option 'template'
1593 1593 2. option 'style'
1594 1594 3. [ui] setting 'logtemplate'
1595 1595 4. [ui] setting 'style'
1596 1596 If all of these values are either the unset or the empty string,
1597 1597 regular display via changeset_printer() is done.
1598 1598 """
1599 1599 # options
1600 1600 matchfn = None
1601 1601 if opts.get('patch') or opts.get('stat'):
1602 1602 matchfn = scmutil.matchall(repo)
1603 1603
1604 1604 if opts.get('template') == 'json':
1605 1605 return jsonchangeset(ui, repo, matchfn, opts, buffered)
1606 1606
1607 1607 tmpl, mapfile = gettemplate(ui, opts.get('template'), opts.get('style'))
1608 1608
1609 1609 if not tmpl and not mapfile:
1610 1610 return changeset_printer(ui, repo, matchfn, opts, buffered)
1611 1611
1612 1612 return changeset_templater(ui, repo, matchfn, opts, tmpl, mapfile, buffered)
1613 1613
1614 1614 def showmarker(fm, marker, index=None):
1615 1615 """utility function to display obsolescence marker in a readable way
1616 1616
1617 1617 To be used by debug function."""
1618 1618 if index is not None:
1619 1619 fm.write('index', '%i ', index)
1620 1620 fm.write('precnode', '%s ', hex(marker.precnode()))
1621 1621 succs = marker.succnodes()
1622 1622 fm.condwrite(succs, 'succnodes', '%s ',
1623 1623 fm.formatlist(map(hex, succs), name='node'))
1624 1624 fm.write('flag', '%X ', marker.flags())
1625 1625 parents = marker.parentnodes()
1626 1626 if parents is not None:
1627 1627 fm.write('parentnodes', '{%s} ',
1628 1628 fm.formatlist(map(hex, parents), name='node', sep=', '))
1629 1629 fm.write('date', '(%s) ', fm.formatdate(marker.date()))
1630 1630 meta = marker.metadata().copy()
1631 1631 meta.pop('date', None)
1632 1632 fm.write('metadata', '{%s}', fm.formatdict(meta, fmt='%r: %r', sep=', '))
1633 1633 fm.plain('\n')
1634 1634
1635 1635 def finddate(ui, repo, date):
1636 1636 """Find the tipmost changeset that matches the given date spec"""
1637 1637
1638 1638 df = util.matchdate(date)
1639 1639 m = scmutil.matchall(repo)
1640 1640 results = {}
1641 1641
1642 1642 def prep(ctx, fns):
1643 1643 d = ctx.date()
1644 1644 if df(d[0]):
1645 1645 results[ctx.rev()] = d
1646 1646
1647 1647 for ctx in walkchangerevs(repo, m, {'rev': None}, prep):
1648 1648 rev = ctx.rev()
1649 1649 if rev in results:
1650 1650 ui.status(_("found revision %s from %s\n") %
1651 1651 (rev, util.datestr(results[rev])))
1652 1652 return str(rev)
1653 1653
1654 1654 raise error.Abort(_("revision matching date not found"))
1655 1655
1656 1656 def increasingwindows(windowsize=8, sizelimit=512):
1657 1657 while True:
1658 1658 yield windowsize
1659 1659 if windowsize < sizelimit:
1660 1660 windowsize *= 2
1661 1661
1662 1662 class FileWalkError(Exception):
1663 1663 pass
1664 1664
1665 1665 def walkfilerevs(repo, match, follow, revs, fncache):
1666 1666 '''Walks the file history for the matched files.
1667 1667
1668 1668 Returns the changeset revs that are involved in the file history.
1669 1669
1670 1670 Throws FileWalkError if the file history can't be walked using
1671 1671 filelogs alone.
1672 1672 '''
1673 1673 wanted = set()
1674 1674 copies = []
1675 1675 minrev, maxrev = min(revs), max(revs)
1676 1676 def filerevgen(filelog, last):
1677 1677 """
1678 1678 Only files, no patterns. Check the history of each file.
1679 1679
1680 1680 Examines filelog entries within minrev, maxrev linkrev range
1681 1681 Returns an iterator yielding (linkrev, parentlinkrevs, copied)
1682 1682 tuples in backwards order
1683 1683 """
1684 1684 cl_count = len(repo)
1685 1685 revs = []
1686 1686 for j in xrange(0, last + 1):
1687 1687 linkrev = filelog.linkrev(j)
1688 1688 if linkrev < minrev:
1689 1689 continue
1690 1690 # only yield rev for which we have the changelog, it can
1691 1691 # happen while doing "hg log" during a pull or commit
1692 1692 if linkrev >= cl_count:
1693 1693 break
1694 1694
1695 1695 parentlinkrevs = []
1696 1696 for p in filelog.parentrevs(j):
1697 1697 if p != nullrev:
1698 1698 parentlinkrevs.append(filelog.linkrev(p))
1699 1699 n = filelog.node(j)
1700 1700 revs.append((linkrev, parentlinkrevs,
1701 1701 follow and filelog.renamed(n)))
1702 1702
1703 1703 return reversed(revs)
1704 1704 def iterfiles():
1705 1705 pctx = repo['.']
1706 1706 for filename in match.files():
1707 1707 if follow:
1708 1708 if filename not in pctx:
1709 1709 raise error.Abort(_('cannot follow file not in parent '
1710 1710 'revision: "%s"') % filename)
1711 1711 yield filename, pctx[filename].filenode()
1712 1712 else:
1713 1713 yield filename, None
1714 1714 for filename_node in copies:
1715 1715 yield filename_node
1716 1716
1717 1717 for file_, node in iterfiles():
1718 1718 filelog = repo.file(file_)
1719 1719 if not len(filelog):
1720 1720 if node is None:
1721 1721 # A zero count may be a directory or deleted file, so
1722 1722 # try to find matching entries on the slow path.
1723 1723 if follow:
1724 1724 raise error.Abort(
1725 1725 _('cannot follow nonexistent file: "%s"') % file_)
1726 1726 raise FileWalkError("Cannot walk via filelog")
1727 1727 else:
1728 1728 continue
1729 1729
1730 1730 if node is None:
1731 1731 last = len(filelog) - 1
1732 1732 else:
1733 1733 last = filelog.rev(node)
1734 1734
1735 1735 # keep track of all ancestors of the file
1736 1736 ancestors = set([filelog.linkrev(last)])
1737 1737
1738 1738 # iterate from latest to oldest revision
1739 1739 for rev, flparentlinkrevs, copied in filerevgen(filelog, last):
1740 1740 if not follow:
1741 1741 if rev > maxrev:
1742 1742 continue
1743 1743 else:
1744 1744 # Note that last might not be the first interesting
1745 1745 # rev to us:
1746 1746 # if the file has been changed after maxrev, we'll
1747 1747 # have linkrev(last) > maxrev, and we still need
1748 1748 # to explore the file graph
1749 1749 if rev not in ancestors:
1750 1750 continue
1751 1751 # XXX insert 1327 fix here
1752 1752 if flparentlinkrevs:
1753 1753 ancestors.update(flparentlinkrevs)
1754 1754
1755 1755 fncache.setdefault(rev, []).append(file_)
1756 1756 wanted.add(rev)
1757 1757 if copied:
1758 1758 copies.append(copied)
1759 1759
1760 1760 return wanted
1761 1761
1762 1762 class _followfilter(object):
1763 1763 def __init__(self, repo, onlyfirst=False):
1764 1764 self.repo = repo
1765 1765 self.startrev = nullrev
1766 1766 self.roots = set()
1767 1767 self.onlyfirst = onlyfirst
1768 1768
1769 1769 def match(self, rev):
1770 1770 def realparents(rev):
1771 1771 if self.onlyfirst:
1772 1772 return self.repo.changelog.parentrevs(rev)[0:1]
1773 1773 else:
1774 1774 return filter(lambda x: x != nullrev,
1775 1775 self.repo.changelog.parentrevs(rev))
1776 1776
1777 1777 if self.startrev == nullrev:
1778 1778 self.startrev = rev
1779 1779 return True
1780 1780
1781 1781 if rev > self.startrev:
1782 1782 # forward: all descendants
1783 1783 if not self.roots:
1784 1784 self.roots.add(self.startrev)
1785 1785 for parent in realparents(rev):
1786 1786 if parent in self.roots:
1787 1787 self.roots.add(rev)
1788 1788 return True
1789 1789 else:
1790 1790 # backwards: all parents
1791 1791 if not self.roots:
1792 1792 self.roots.update(realparents(self.startrev))
1793 1793 if rev in self.roots:
1794 1794 self.roots.remove(rev)
1795 1795 self.roots.update(realparents(rev))
1796 1796 return True
1797 1797
1798 1798 return False
1799 1799
1800 1800 def walkchangerevs(repo, match, opts, prepare):
1801 1801 '''Iterate over files and the revs in which they changed.
1802 1802
1803 1803 Callers most commonly need to iterate backwards over the history
1804 1804 in which they are interested. Doing so has awful (quadratic-looking)
1805 1805 performance, so we use iterators in a "windowed" way.
1806 1806
1807 1807 We walk a window of revisions in the desired order. Within the
1808 1808 window, we first walk forwards to gather data, then in the desired
1809 1809 order (usually backwards) to display it.
1810 1810
1811 1811 This function returns an iterator yielding contexts. Before
1812 1812 yielding each context, the iterator will first call the prepare
1813 1813 function on each context in the window in forward order.'''
1814 1814
1815 1815 follow = opts.get('follow') or opts.get('follow_first')
1816 1816 revs = _logrevs(repo, opts)
1817 1817 if not revs:
1818 1818 return []
1819 1819 wanted = set()
1820 1820 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
1821 1821 opts.get('removed'))
1822 1822 fncache = {}
1823 1823 change = repo.changectx
1824 1824
1825 1825 # First step is to fill wanted, the set of revisions that we want to yield.
1826 1826 # When it does not induce extra cost, we also fill fncache for revisions in
1827 1827 # wanted: a cache of filenames that were changed (ctx.files()) and that
1828 1828 # match the file filtering conditions.
1829 1829
1830 1830 if match.always():
1831 1831 # No files, no patterns. Display all revs.
1832 1832 wanted = revs
1833 1833 elif not slowpath:
1834 1834 # We only have to read through the filelog to find wanted revisions
1835 1835
1836 1836 try:
1837 1837 wanted = walkfilerevs(repo, match, follow, revs, fncache)
1838 1838 except FileWalkError:
1839 1839 slowpath = True
1840 1840
1841 1841 # We decided to fall back to the slowpath because at least one
1842 1842 # of the paths was not a file. Check to see if at least one of them
1843 1843 # existed in history, otherwise simply return
1844 1844 for path in match.files():
1845 1845 if path == '.' or path in repo.store:
1846 1846 break
1847 1847 else:
1848 1848 return []
1849 1849
1850 1850 if slowpath:
1851 1851 # We have to read the changelog to match filenames against
1852 1852 # changed files
1853 1853
1854 1854 if follow:
1855 1855 raise error.Abort(_('can only follow copies/renames for explicit '
1856 1856 'filenames'))
1857 1857
1858 1858 # The slow path checks files modified in every changeset.
1859 1859 # This is really slow on large repos, so compute the set lazily.
1860 1860 class lazywantedset(object):
1861 1861 def __init__(self):
1862 1862 self.set = set()
1863 1863 self.revs = set(revs)
1864 1864
1865 1865 # No need to worry about locality here because it will be accessed
1866 1866 # in the same order as the increasing window below.
1867 1867 def __contains__(self, value):
1868 1868 if value in self.set:
1869 1869 return True
1870 1870 elif not value in self.revs:
1871 1871 return False
1872 1872 else:
1873 1873 self.revs.discard(value)
1874 1874 ctx = change(value)
1875 1875 matches = filter(match, ctx.files())
1876 1876 if matches:
1877 1877 fncache[value] = matches
1878 1878 self.set.add(value)
1879 1879 return True
1880 1880 return False
1881 1881
1882 1882 def discard(self, value):
1883 1883 self.revs.discard(value)
1884 1884 self.set.discard(value)
1885 1885
1886 1886 wanted = lazywantedset()
1887 1887
1888 1888 # it might be worthwhile to do this in the iterator if the rev range
1889 1889 # is descending and the prune args are all within that range
1890 1890 for rev in opts.get('prune', ()):
1891 1891 rev = repo[rev].rev()
1892 1892 ff = _followfilter(repo)
1893 1893 stop = min(revs[0], revs[-1])
1894 1894 for x in xrange(rev, stop - 1, -1):
1895 1895 if ff.match(x):
1896 1896 wanted = wanted - [x]
1897 1897
1898 1898 # Now that wanted is correctly initialized, we can iterate over the
1899 1899 # revision range, yielding only revisions in wanted.
1900 1900 def iterate():
1901 1901 if follow and match.always():
1902 1902 ff = _followfilter(repo, onlyfirst=opts.get('follow_first'))
1903 1903 def want(rev):
1904 1904 return ff.match(rev) and rev in wanted
1905 1905 else:
1906 1906 def want(rev):
1907 1907 return rev in wanted
1908 1908
1909 1909 it = iter(revs)
1910 1910 stopiteration = False
1911 1911 for windowsize in increasingwindows():
1912 1912 nrevs = []
1913 1913 for i in xrange(windowsize):
1914 1914 rev = next(it, None)
1915 1915 if rev is None:
1916 1916 stopiteration = True
1917 1917 break
1918 1918 elif want(rev):
1919 1919 nrevs.append(rev)
1920 1920 for rev in sorted(nrevs):
1921 1921 fns = fncache.get(rev)
1922 1922 ctx = change(rev)
1923 1923 if not fns:
1924 1924 def fns_generator():
1925 1925 for f in ctx.files():
1926 1926 if match(f):
1927 1927 yield f
1928 1928 fns = fns_generator()
1929 1929 prepare(ctx, fns)
1930 1930 for rev in nrevs:
1931 1931 yield change(rev)
1932 1932
1933 1933 if stopiteration:
1934 1934 break
1935 1935
1936 1936 return iterate()
1937 1937
1938 1938 def _makefollowlogfilematcher(repo, files, followfirst):
1939 1939 # When displaying a revision with --patch --follow FILE, we have
1940 1940 # to know which file of the revision must be diffed. With
1941 1941 # --follow, we want the names of the ancestors of FILE in the
1942 1942 # revision, stored in "fcache". "fcache" is populated by
1943 1943 # reproducing the graph traversal already done by --follow revset
1944 1944 # and relating linkrevs to file names (which is not "correct" but
1945 1945 # good enough).
1946 1946 fcache = {}
1947 1947 fcacheready = [False]
1948 1948 pctx = repo['.']
1949 1949
1950 1950 def populate():
1951 1951 for fn in files:
1952 1952 for i in ((pctx[fn],), pctx[fn].ancestors(followfirst=followfirst)):
1953 1953 for c in i:
1954 1954 fcache.setdefault(c.linkrev(), set()).add(c.path())
1955 1955
1956 1956 def filematcher(rev):
1957 1957 if not fcacheready[0]:
1958 1958 # Lazy initialization
1959 1959 fcacheready[0] = True
1960 1960 populate()
1961 1961 return scmutil.matchfiles(repo, fcache.get(rev, []))
1962 1962
1963 1963 return filematcher
1964 1964
1965 1965 def _makenofollowlogfilematcher(repo, pats, opts):
1966 1966 '''hook for extensions to override the filematcher for non-follow cases'''
1967 1967 return None
1968 1968
1969 1969 def _makelogrevset(repo, pats, opts, revs):
1970 1970 """Return (expr, filematcher) where expr is a revset string built
1971 1971 from log options and file patterns or None. If --stat or --patch
1972 1972 are not passed filematcher is None. Otherwise it is a callable
1973 1973 taking a revision number and returning a match objects filtering
1974 1974 the files to be detailed when displaying the revision.
1975 1975 """
1976 1976 opt2revset = {
1977 1977 'no_merges': ('not merge()', None),
1978 1978 'only_merges': ('merge()', None),
1979 1979 '_ancestors': ('ancestors(%(val)s)', None),
1980 1980 '_fancestors': ('_firstancestors(%(val)s)', None),
1981 1981 '_descendants': ('descendants(%(val)s)', None),
1982 1982 '_fdescendants': ('_firstdescendants(%(val)s)', None),
1983 1983 '_matchfiles': ('_matchfiles(%(val)s)', None),
1984 1984 'date': ('date(%(val)r)', None),
1985 1985 'branch': ('branch(%(val)r)', ' or '),
1986 1986 '_patslog': ('filelog(%(val)r)', ' or '),
1987 1987 '_patsfollow': ('follow(%(val)r)', ' or '),
1988 1988 '_patsfollowfirst': ('_followfirst(%(val)r)', ' or '),
1989 1989 'keyword': ('keyword(%(val)r)', ' or '),
1990 1990 'prune': ('not (%(val)r or ancestors(%(val)r))', ' and '),
1991 1991 'user': ('user(%(val)r)', ' or '),
1992 1992 }
1993 1993
1994 1994 opts = dict(opts)
1995 1995 # follow or not follow?
1996 1996 follow = opts.get('follow') or opts.get('follow_first')
1997 1997 if opts.get('follow_first'):
1998 1998 followfirst = 1
1999 1999 else:
2000 2000 followfirst = 0
2001 2001 # --follow with FILE behavior depends on revs...
2002 2002 it = iter(revs)
2003 2003 startrev = next(it)
2004 2004 followdescendants = startrev < next(it, startrev)
2005 2005
2006 2006 # branch and only_branch are really aliases and must be handled at
2007 2007 # the same time
2008 2008 opts['branch'] = opts.get('branch', []) + opts.get('only_branch', [])
2009 2009 opts['branch'] = [repo.lookupbranch(b) for b in opts['branch']]
2010 2010 # pats/include/exclude are passed to match.match() directly in
2011 2011 # _matchfiles() revset but walkchangerevs() builds its matcher with
2012 2012 # scmutil.match(). The difference is input pats are globbed on
2013 2013 # platforms without shell expansion (windows).
2014 2014 wctx = repo[None]
2015 2015 match, pats = scmutil.matchandpats(wctx, pats, opts)
2016 2016 slowpath = match.anypats() or ((match.isexact() or match.prefix()) and
2017 2017 opts.get('removed'))
2018 2018 if not slowpath:
2019 2019 for f in match.files():
2020 2020 if follow and f not in wctx:
2021 2021 # If the file exists, it may be a directory, so let it
2022 2022 # take the slow path.
2023 2023 if os.path.exists(repo.wjoin(f)):
2024 2024 slowpath = True
2025 2025 continue
2026 2026 else:
2027 2027 raise error.Abort(_('cannot follow file not in parent '
2028 2028 'revision: "%s"') % f)
2029 2029 filelog = repo.file(f)
2030 2030 if not filelog:
2031 2031 # A zero count may be a directory or deleted file, so
2032 2032 # try to find matching entries on the slow path.
2033 2033 if follow:
2034 2034 raise error.Abort(
2035 2035 _('cannot follow nonexistent file: "%s"') % f)
2036 2036 slowpath = True
2037 2037
2038 2038 # We decided to fall back to the slowpath because at least one
2039 2039 # of the paths was not a file. Check to see if at least one of them
2040 2040 # existed in history - in that case, we'll continue down the
2041 2041 # slowpath; otherwise, we can turn off the slowpath
2042 2042 if slowpath:
2043 2043 for path in match.files():
2044 2044 if path == '.' or path in repo.store:
2045 2045 break
2046 2046 else:
2047 2047 slowpath = False
2048 2048
2049 2049 fpats = ('_patsfollow', '_patsfollowfirst')
2050 2050 fnopats = (('_ancestors', '_fancestors'),
2051 2051 ('_descendants', '_fdescendants'))
2052 2052 if slowpath:
2053 2053 # See walkchangerevs() slow path.
2054 2054 #
2055 2055 # pats/include/exclude cannot be represented as separate
2056 2056 # revset expressions as their filtering logic applies at file
2057 2057 # level. For instance "-I a -X a" matches a revision touching
2058 2058 # "a" and "b" while "file(a) and not file(b)" does
2059 2059 # not. Besides, filesets are evaluated against the working
2060 2060 # directory.
2061 2061 matchargs = ['r:', 'd:relpath']
2062 2062 for p in pats:
2063 2063 matchargs.append('p:' + p)
2064 2064 for p in opts.get('include', []):
2065 2065 matchargs.append('i:' + p)
2066 2066 for p in opts.get('exclude', []):
2067 2067 matchargs.append('x:' + p)
2068 2068 matchargs = ','.join(('%r' % p) for p in matchargs)
2069 2069 opts['_matchfiles'] = matchargs
2070 2070 if follow:
2071 2071 opts[fnopats[0][followfirst]] = '.'
2072 2072 else:
2073 2073 if follow:
2074 2074 if pats:
2075 2075 # follow() revset interprets its file argument as a
2076 2076 # manifest entry, so use match.files(), not pats.
2077 2077 opts[fpats[followfirst]] = list(match.files())
2078 2078 else:
2079 2079 op = fnopats[followdescendants][followfirst]
2080 2080 opts[op] = 'rev(%d)' % startrev
2081 2081 else:
2082 2082 opts['_patslog'] = list(pats)
2083 2083
2084 2084 filematcher = None
2085 2085 if opts.get('patch') or opts.get('stat'):
2086 2086 # When following files, track renames via a special matcher.
2087 2087 # If we're forced to take the slowpath it means we're following
2088 2088 # at least one pattern/directory, so don't bother with rename tracking.
2089 2089 if follow and not match.always() and not slowpath:
2090 2090 # _makefollowlogfilematcher expects its files argument to be
2091 2091 # relative to the repo root, so use match.files(), not pats.
2092 2092 filematcher = _makefollowlogfilematcher(repo, match.files(),
2093 2093 followfirst)
2094 2094 else:
2095 2095 filematcher = _makenofollowlogfilematcher(repo, pats, opts)
2096 2096 if filematcher is None:
2097 2097 filematcher = lambda rev: match
2098 2098
2099 2099 expr = []
2100 2100 for op, val in sorted(opts.iteritems()):
2101 2101 if not val:
2102 2102 continue
2103 2103 if op not in opt2revset:
2104 2104 continue
2105 2105 revop, andor = opt2revset[op]
2106 2106 if '%(val)' not in revop:
2107 2107 expr.append(revop)
2108 2108 else:
2109 2109 if not isinstance(val, list):
2110 2110 e = revop % {'val': val}
2111 2111 else:
2112 2112 e = '(' + andor.join((revop % {'val': v}) for v in val) + ')'
2113 2113 expr.append(e)
2114 2114
2115 2115 if expr:
2116 2116 expr = '(' + ' and '.join(expr) + ')'
2117 2117 else:
2118 2118 expr = None
2119 2119 return expr, filematcher
2120 2120
2121 2121 def _logrevs(repo, opts):
2122 2122 # Default --rev value depends on --follow but --follow behavior
2123 2123 # depends on revisions resolved from --rev...
2124 2124 follow = opts.get('follow') or opts.get('follow_first')
2125 2125 if opts.get('rev'):
2126 2126 revs = scmutil.revrange(repo, opts['rev'])
2127 2127 elif follow and repo.dirstate.p1() == nullid:
2128 2128 revs = revset.baseset()
2129 2129 elif follow:
2130 2130 revs = repo.revs('reverse(:.)')
2131 2131 else:
2132 2132 revs = revset.spanset(repo)
2133 2133 revs.reverse()
2134 2134 return revs
2135 2135
2136 2136 def getgraphlogrevs(repo, pats, opts):
2137 2137 """Return (revs, expr, filematcher) where revs is an iterable of
2138 2138 revision numbers, expr is a revset string built from log options
2139 2139 and file patterns or None, and used to filter 'revs'. If --stat or
2140 2140 --patch are not passed filematcher is None. Otherwise it is a
2141 2141 callable taking a revision number and returning a match objects
2142 2142 filtering the files to be detailed when displaying the revision.
2143 2143 """
2144 2144 limit = loglimit(opts)
2145 2145 revs = _logrevs(repo, opts)
2146 2146 if not revs:
2147 2147 return revset.baseset(), None, None
2148 2148 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2149 2149 if opts.get('rev'):
2150 2150 # User-specified revs might be unsorted, but don't sort before
2151 2151 # _makelogrevset because it might depend on the order of revs
2152 2152 if not (revs.isdescending() or revs.istopo()):
2153 2153 revs.sort(reverse=True)
2154 2154 if expr:
2155 2155 # Revset matchers often operate faster on revisions in changelog
2156 2156 # order, because most filters deal with the changelog.
2157 2157 revs.reverse()
2158 matcher = revset.match(repo.ui, expr)
2159 # Revset matches can reorder revisions. "A or B" typically returns
2160 # returns the revision matching A then the revision matching B. Sort
2161 # again to fix that.
2158 matcher = revset.match(repo.ui, expr, order=revset.followorder)
2162 2159 revs = matcher(repo, revs)
2163 revs.sort(reverse=True)
2160 revs.reverse()
2164 2161 if limit is not None:
2165 2162 limitedrevs = []
2166 2163 for idx, rev in enumerate(revs):
2167 2164 if idx >= limit:
2168 2165 break
2169 2166 limitedrevs.append(rev)
2170 2167 revs = revset.baseset(limitedrevs)
2171 2168
2172 2169 return revs, expr, filematcher
2173 2170
2174 2171 def getlogrevs(repo, pats, opts):
2175 2172 """Return (revs, expr, filematcher) where revs is an iterable of
2176 2173 revision numbers, expr is a revset string built from log options
2177 2174 and file patterns or None, and used to filter 'revs'. If --stat or
2178 2175 --patch are not passed filematcher is None. Otherwise it is a
2179 2176 callable taking a revision number and returning a match objects
2180 2177 filtering the files to be detailed when displaying the revision.
2181 2178 """
2182 2179 limit = loglimit(opts)
2183 2180 revs = _logrevs(repo, opts)
2184 2181 if not revs:
2185 2182 return revset.baseset([]), None, None
2186 2183 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
2187 2184 if expr:
2188 2185 # Revset matchers often operate faster on revisions in changelog
2189 2186 # order, because most filters deal with the changelog.
2190 2187 if not opts.get('rev'):
2191 2188 revs.reverse()
2192 2189 matcher = revset.match(repo.ui, expr, order=revset.followorder)
2193 2190 revs = matcher(repo, revs)
2194 2191 if not opts.get('rev'):
2195 2192 revs.reverse()
2196 2193 if limit is not None:
2197 2194 limitedrevs = []
2198 2195 for idx, r in enumerate(revs):
2199 2196 if limit <= idx:
2200 2197 break
2201 2198 limitedrevs.append(r)
2202 2199 revs = revset.baseset(limitedrevs)
2203 2200
2204 2201 return revs, expr, filematcher
2205 2202
2206 2203 def _graphnodeformatter(ui, displayer):
2207 2204 spec = ui.config('ui', 'graphnodetemplate')
2208 2205 if not spec:
2209 2206 return templatekw.showgraphnode # fast path for "{graphnode}"
2210 2207
2211 2208 templ = formatter.gettemplater(ui, 'graphnode', spec)
2212 2209 cache = {}
2213 2210 if isinstance(displayer, changeset_templater):
2214 2211 cache = displayer.cache # reuse cache of slow templates
2215 2212 props = templatekw.keywords.copy()
2216 2213 props['templ'] = templ
2217 2214 props['cache'] = cache
2218 2215 def formatnode(repo, ctx):
2219 2216 props['ctx'] = ctx
2220 2217 props['repo'] = repo
2221 2218 props['ui'] = repo.ui
2222 2219 props['revcache'] = {}
2223 2220 return templater.stringify(templ('graphnode', **props))
2224 2221 return formatnode
2225 2222
2226 2223 def displaygraph(ui, repo, dag, displayer, edgefn, getrenamed=None,
2227 2224 filematcher=None):
2228 2225 formatnode = _graphnodeformatter(ui, displayer)
2229 2226 state = graphmod.asciistate()
2230 2227 styles = state['styles']
2231 2228
2232 2229 # only set graph styling if HGPLAIN is not set.
2233 2230 if ui.plain('graph'):
2234 2231 # set all edge styles to |, the default pre-3.8 behaviour
2235 2232 styles.update(dict.fromkeys(styles, '|'))
2236 2233 else:
2237 2234 edgetypes = {
2238 2235 'parent': graphmod.PARENT,
2239 2236 'grandparent': graphmod.GRANDPARENT,
2240 2237 'missing': graphmod.MISSINGPARENT
2241 2238 }
2242 2239 for name, key in edgetypes.items():
2243 2240 # experimental config: experimental.graphstyle.*
2244 2241 styles[key] = ui.config('experimental', 'graphstyle.%s' % name,
2245 2242 styles[key])
2246 2243 if not styles[key]:
2247 2244 styles[key] = None
2248 2245
2249 2246 # experimental config: experimental.graphshorten
2250 2247 state['graphshorten'] = ui.configbool('experimental', 'graphshorten')
2251 2248
2252 2249 for rev, type, ctx, parents in dag:
2253 2250 char = formatnode(repo, ctx)
2254 2251 copies = None
2255 2252 if getrenamed and ctx.rev():
2256 2253 copies = []
2257 2254 for fn in ctx.files():
2258 2255 rename = getrenamed(fn, ctx.rev())
2259 2256 if rename:
2260 2257 copies.append((fn, rename[0]))
2261 2258 revmatchfn = None
2262 2259 if filematcher is not None:
2263 2260 revmatchfn = filematcher(ctx.rev())
2264 2261 displayer.show(ctx, copies=copies, matchfn=revmatchfn)
2265 2262 lines = displayer.hunk.pop(rev).split('\n')
2266 2263 if not lines[-1]:
2267 2264 del lines[-1]
2268 2265 displayer.flush(ctx)
2269 2266 edges = edgefn(type, char, lines, state, rev, parents)
2270 2267 for type, char, lines, coldata in edges:
2271 2268 graphmod.ascii(ui, state, type, char, lines, coldata)
2272 2269 displayer.close()
2273 2270
2274 2271 def graphlog(ui, repo, *pats, **opts):
2275 2272 # Parameters are identical to log command ones
2276 2273 revs, expr, filematcher = getgraphlogrevs(repo, pats, opts)
2277 2274 revdag = graphmod.dagwalker(repo, revs)
2278 2275
2279 2276 getrenamed = None
2280 2277 if opts.get('copies'):
2281 2278 endrev = None
2282 2279 if opts.get('rev'):
2283 2280 endrev = scmutil.revrange(repo, opts.get('rev')).max() + 1
2284 2281 getrenamed = templatekw.getrenamedfn(repo, endrev=endrev)
2285 2282 displayer = show_changeset(ui, repo, opts, buffered=True)
2286 2283 displaygraph(ui, repo, revdag, displayer, graphmod.asciiedges, getrenamed,
2287 2284 filematcher)
2288 2285
2289 2286 def checkunsupportedgraphflags(pats, opts):
2290 2287 for op in ["newest_first"]:
2291 2288 if op in opts and opts[op]:
2292 2289 raise error.Abort(_("-G/--graph option is incompatible with --%s")
2293 2290 % op.replace("_", "-"))
2294 2291
2295 2292 def graphrevs(repo, nodes, opts):
2296 2293 limit = loglimit(opts)
2297 2294 nodes.reverse()
2298 2295 if limit is not None:
2299 2296 nodes = nodes[:limit]
2300 2297 return graphmod.nodes(repo, nodes)
2301 2298
2302 2299 def add(ui, repo, match, prefix, explicitonly, **opts):
2303 2300 join = lambda f: os.path.join(prefix, f)
2304 2301 bad = []
2305 2302
2306 2303 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2307 2304 names = []
2308 2305 wctx = repo[None]
2309 2306 cca = None
2310 2307 abort, warn = scmutil.checkportabilityalert(ui)
2311 2308 if abort or warn:
2312 2309 cca = scmutil.casecollisionauditor(ui, abort, repo.dirstate)
2313 2310
2314 2311 badmatch = matchmod.badmatch(match, badfn)
2315 2312 dirstate = repo.dirstate
2316 2313 # We don't want to just call wctx.walk here, since it would return a lot of
2317 2314 # clean files, which we aren't interested in and takes time.
2318 2315 for f in sorted(dirstate.walk(badmatch, sorted(wctx.substate),
2319 2316 True, False, full=False)):
2320 2317 exact = match.exact(f)
2321 2318 if exact or not explicitonly and f not in wctx and repo.wvfs.lexists(f):
2322 2319 if cca:
2323 2320 cca(f)
2324 2321 names.append(f)
2325 2322 if ui.verbose or not exact:
2326 2323 ui.status(_('adding %s\n') % match.rel(f))
2327 2324
2328 2325 for subpath in sorted(wctx.substate):
2329 2326 sub = wctx.sub(subpath)
2330 2327 try:
2331 2328 submatch = matchmod.subdirmatcher(subpath, match)
2332 2329 if opts.get('subrepos'):
2333 2330 bad.extend(sub.add(ui, submatch, prefix, False, **opts))
2334 2331 else:
2335 2332 bad.extend(sub.add(ui, submatch, prefix, True, **opts))
2336 2333 except error.LookupError:
2337 2334 ui.status(_("skipping missing subrepository: %s\n")
2338 2335 % join(subpath))
2339 2336
2340 2337 if not opts.get('dry_run'):
2341 2338 rejected = wctx.add(names, prefix)
2342 2339 bad.extend(f for f in rejected if f in match.files())
2343 2340 return bad
2344 2341
2345 2342 def forget(ui, repo, match, prefix, explicitonly):
2346 2343 join = lambda f: os.path.join(prefix, f)
2347 2344 bad = []
2348 2345 badfn = lambda x, y: bad.append(x) or match.bad(x, y)
2349 2346 wctx = repo[None]
2350 2347 forgot = []
2351 2348
2352 2349 s = repo.status(match=matchmod.badmatch(match, badfn), clean=True)
2353 2350 forget = sorted(s[0] + s[1] + s[3] + s[6])
2354 2351 if explicitonly:
2355 2352 forget = [f for f in forget if match.exact(f)]
2356 2353
2357 2354 for subpath in sorted(wctx.substate):
2358 2355 sub = wctx.sub(subpath)
2359 2356 try:
2360 2357 submatch = matchmod.subdirmatcher(subpath, match)
2361 2358 subbad, subforgot = sub.forget(submatch, prefix)
2362 2359 bad.extend([subpath + '/' + f for f in subbad])
2363 2360 forgot.extend([subpath + '/' + f for f in subforgot])
2364 2361 except error.LookupError:
2365 2362 ui.status(_("skipping missing subrepository: %s\n")
2366 2363 % join(subpath))
2367 2364
2368 2365 if not explicitonly:
2369 2366 for f in match.files():
2370 2367 if f not in repo.dirstate and not repo.wvfs.isdir(f):
2371 2368 if f not in forgot:
2372 2369 if repo.wvfs.exists(f):
2373 2370 # Don't complain if the exact case match wasn't given.
2374 2371 # But don't do this until after checking 'forgot', so
2375 2372 # that subrepo files aren't normalized, and this op is
2376 2373 # purely from data cached by the status walk above.
2377 2374 if repo.dirstate.normalize(f) in repo.dirstate:
2378 2375 continue
2379 2376 ui.warn(_('not removing %s: '
2380 2377 'file is already untracked\n')
2381 2378 % match.rel(f))
2382 2379 bad.append(f)
2383 2380
2384 2381 for f in forget:
2385 2382 if ui.verbose or not match.exact(f):
2386 2383 ui.status(_('removing %s\n') % match.rel(f))
2387 2384
2388 2385 rejected = wctx.forget(forget, prefix)
2389 2386 bad.extend(f for f in rejected if f in match.files())
2390 2387 forgot.extend(f for f in forget if f not in rejected)
2391 2388 return bad, forgot
2392 2389
2393 2390 def files(ui, ctx, m, fm, fmt, subrepos):
2394 2391 rev = ctx.rev()
2395 2392 ret = 1
2396 2393 ds = ctx.repo().dirstate
2397 2394
2398 2395 for f in ctx.matches(m):
2399 2396 if rev is None and ds[f] == 'r':
2400 2397 continue
2401 2398 fm.startitem()
2402 2399 if ui.verbose:
2403 2400 fc = ctx[f]
2404 2401 fm.write('size flags', '% 10d % 1s ', fc.size(), fc.flags())
2405 2402 fm.data(abspath=f)
2406 2403 fm.write('path', fmt, m.rel(f))
2407 2404 ret = 0
2408 2405
2409 2406 for subpath in sorted(ctx.substate):
2410 2407 submatch = matchmod.subdirmatcher(subpath, m)
2411 2408 if (subrepos or m.exact(subpath) or any(submatch.files())):
2412 2409 sub = ctx.sub(subpath)
2413 2410 try:
2414 2411 recurse = m.exact(subpath) or subrepos
2415 2412 if sub.printfiles(ui, submatch, fm, fmt, recurse) == 0:
2416 2413 ret = 0
2417 2414 except error.LookupError:
2418 2415 ui.status(_("skipping missing subrepository: %s\n")
2419 2416 % m.abs(subpath))
2420 2417
2421 2418 return ret
2422 2419
2423 2420 def remove(ui, repo, m, prefix, after, force, subrepos, warnings=None):
2424 2421 join = lambda f: os.path.join(prefix, f)
2425 2422 ret = 0
2426 2423 s = repo.status(match=m, clean=True)
2427 2424 modified, added, deleted, clean = s[0], s[1], s[3], s[6]
2428 2425
2429 2426 wctx = repo[None]
2430 2427
2431 2428 if warnings is None:
2432 2429 warnings = []
2433 2430 warn = True
2434 2431 else:
2435 2432 warn = False
2436 2433
2437 2434 subs = sorted(wctx.substate)
2438 2435 total = len(subs)
2439 2436 count = 0
2440 2437 for subpath in subs:
2441 2438 count += 1
2442 2439 submatch = matchmod.subdirmatcher(subpath, m)
2443 2440 if subrepos or m.exact(subpath) or any(submatch.files()):
2444 2441 ui.progress(_('searching'), count, total=total, unit=_('subrepos'))
2445 2442 sub = wctx.sub(subpath)
2446 2443 try:
2447 2444 if sub.removefiles(submatch, prefix, after, force, subrepos,
2448 2445 warnings):
2449 2446 ret = 1
2450 2447 except error.LookupError:
2451 2448 warnings.append(_("skipping missing subrepository: %s\n")
2452 2449 % join(subpath))
2453 2450 ui.progress(_('searching'), None)
2454 2451
2455 2452 # warn about failure to delete explicit files/dirs
2456 2453 deleteddirs = util.dirs(deleted)
2457 2454 files = m.files()
2458 2455 total = len(files)
2459 2456 count = 0
2460 2457 for f in files:
2461 2458 def insubrepo():
2462 2459 for subpath in wctx.substate:
2463 2460 if f.startswith(subpath + '/'):
2464 2461 return True
2465 2462 return False
2466 2463
2467 2464 count += 1
2468 2465 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2469 2466 isdir = f in deleteddirs or wctx.hasdir(f)
2470 2467 if (f in repo.dirstate or isdir or f == '.'
2471 2468 or insubrepo() or f in subs):
2472 2469 continue
2473 2470
2474 2471 if repo.wvfs.exists(f):
2475 2472 if repo.wvfs.isdir(f):
2476 2473 warnings.append(_('not removing %s: no tracked files\n')
2477 2474 % m.rel(f))
2478 2475 else:
2479 2476 warnings.append(_('not removing %s: file is untracked\n')
2480 2477 % m.rel(f))
2481 2478 # missing files will generate a warning elsewhere
2482 2479 ret = 1
2483 2480 ui.progress(_('deleting'), None)
2484 2481
2485 2482 if force:
2486 2483 list = modified + deleted + clean + added
2487 2484 elif after:
2488 2485 list = deleted
2489 2486 remaining = modified + added + clean
2490 2487 total = len(remaining)
2491 2488 count = 0
2492 2489 for f in remaining:
2493 2490 count += 1
2494 2491 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2495 2492 warnings.append(_('not removing %s: file still exists\n')
2496 2493 % m.rel(f))
2497 2494 ret = 1
2498 2495 ui.progress(_('skipping'), None)
2499 2496 else:
2500 2497 list = deleted + clean
2501 2498 total = len(modified) + len(added)
2502 2499 count = 0
2503 2500 for f in modified:
2504 2501 count += 1
2505 2502 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2506 2503 warnings.append(_('not removing %s: file is modified (use -f'
2507 2504 ' to force removal)\n') % m.rel(f))
2508 2505 ret = 1
2509 2506 for f in added:
2510 2507 count += 1
2511 2508 ui.progress(_('skipping'), count, total=total, unit=_('files'))
2512 2509 warnings.append(_("not removing %s: file has been marked for add"
2513 2510 " (use 'hg forget' to undo add)\n") % m.rel(f))
2514 2511 ret = 1
2515 2512 ui.progress(_('skipping'), None)
2516 2513
2517 2514 list = sorted(list)
2518 2515 total = len(list)
2519 2516 count = 0
2520 2517 for f in list:
2521 2518 count += 1
2522 2519 if ui.verbose or not m.exact(f):
2523 2520 ui.progress(_('deleting'), count, total=total, unit=_('files'))
2524 2521 ui.status(_('removing %s\n') % m.rel(f))
2525 2522 ui.progress(_('deleting'), None)
2526 2523
2527 2524 with repo.wlock():
2528 2525 if not after:
2529 2526 for f in list:
2530 2527 if f in added:
2531 2528 continue # we never unlink added files on remove
2532 2529 util.unlinkpath(repo.wjoin(f), ignoremissing=True)
2533 2530 repo[None].forget(list)
2534 2531
2535 2532 if warn:
2536 2533 for warning in warnings:
2537 2534 ui.warn(warning)
2538 2535
2539 2536 return ret
2540 2537
2541 2538 def cat(ui, repo, ctx, matcher, prefix, **opts):
2542 2539 err = 1
2543 2540
2544 2541 def write(path):
2545 2542 fp = makefileobj(repo, opts.get('output'), ctx.node(),
2546 2543 pathname=os.path.join(prefix, path))
2547 2544 data = ctx[path].data()
2548 2545 if opts.get('decode'):
2549 2546 data = repo.wwritedata(path, data)
2550 2547 fp.write(data)
2551 2548 fp.close()
2552 2549
2553 2550 # Automation often uses hg cat on single files, so special case it
2554 2551 # for performance to avoid the cost of parsing the manifest.
2555 2552 if len(matcher.files()) == 1 and not matcher.anypats():
2556 2553 file = matcher.files()[0]
2557 2554 mf = repo.manifest
2558 2555 mfnode = ctx.manifestnode()
2559 2556 if mfnode and mf.find(mfnode, file)[0]:
2560 2557 write(file)
2561 2558 return 0
2562 2559
2563 2560 for abs in ctx.walk(matcher):
2564 2561 write(abs)
2565 2562 err = 0
2566 2563
2567 2564 for subpath in sorted(ctx.substate):
2568 2565 sub = ctx.sub(subpath)
2569 2566 try:
2570 2567 submatch = matchmod.subdirmatcher(subpath, matcher)
2571 2568
2572 2569 if not sub.cat(submatch, os.path.join(prefix, sub._path),
2573 2570 **opts):
2574 2571 err = 0
2575 2572 except error.RepoLookupError:
2576 2573 ui.status(_("skipping missing subrepository: %s\n")
2577 2574 % os.path.join(prefix, subpath))
2578 2575
2579 2576 return err
2580 2577
2581 2578 def commit(ui, repo, commitfunc, pats, opts):
2582 2579 '''commit the specified files or all outstanding changes'''
2583 2580 date = opts.get('date')
2584 2581 if date:
2585 2582 opts['date'] = util.parsedate(date)
2586 2583 message = logmessage(ui, opts)
2587 2584 matcher = scmutil.match(repo[None], pats, opts)
2588 2585
2589 2586 # extract addremove carefully -- this function can be called from a command
2590 2587 # that doesn't support addremove
2591 2588 if opts.get('addremove'):
2592 2589 if scmutil.addremove(repo, matcher, "", opts) != 0:
2593 2590 raise error.Abort(
2594 2591 _("failed to mark all new/missing files as added/removed"))
2595 2592
2596 2593 return commitfunc(ui, repo, message, matcher, opts)
2597 2594
2598 2595 def samefile(f, ctx1, ctx2):
2599 2596 if f in ctx1.manifest():
2600 2597 a = ctx1.filectx(f)
2601 2598 if f in ctx2.manifest():
2602 2599 b = ctx2.filectx(f)
2603 2600 return (not a.cmp(b)
2604 2601 and a.flags() == b.flags())
2605 2602 else:
2606 2603 return False
2607 2604 else:
2608 2605 return f not in ctx2.manifest()
2609 2606
2610 2607 def amend(ui, repo, commitfunc, old, extra, pats, opts):
2611 2608 # avoid cycle context -> subrepo -> cmdutil
2612 2609 from . import context
2613 2610
2614 2611 # amend will reuse the existing user if not specified, but the obsolete
2615 2612 # marker creation requires that the current user's name is specified.
2616 2613 if obsolete.isenabled(repo, obsolete.createmarkersopt):
2617 2614 ui.username() # raise exception if username not set
2618 2615
2619 2616 ui.note(_('amending changeset %s\n') % old)
2620 2617 base = old.p1()
2621 2618 createmarkers = obsolete.isenabled(repo, obsolete.createmarkersopt)
2622 2619
2623 2620 wlock = lock = newid = None
2624 2621 try:
2625 2622 wlock = repo.wlock()
2626 2623 lock = repo.lock()
2627 2624 with repo.transaction('amend') as tr:
2628 2625 # See if we got a message from -m or -l, if not, open the editor
2629 2626 # with the message of the changeset to amend
2630 2627 message = logmessage(ui, opts)
2631 2628 # ensure logfile does not conflict with later enforcement of the
2632 2629 # message. potential logfile content has been processed by
2633 2630 # `logmessage` anyway.
2634 2631 opts.pop('logfile')
2635 2632 # First, do a regular commit to record all changes in the working
2636 2633 # directory (if there are any)
2637 2634 ui.callhooks = False
2638 2635 activebookmark = repo._bookmarks.active
2639 2636 try:
2640 2637 repo._bookmarks.active = None
2641 2638 opts['message'] = 'temporary amend commit for %s' % old
2642 2639 node = commit(ui, repo, commitfunc, pats, opts)
2643 2640 finally:
2644 2641 repo._bookmarks.active = activebookmark
2645 2642 repo._bookmarks.recordchange(tr)
2646 2643 ui.callhooks = True
2647 2644 ctx = repo[node]
2648 2645
2649 2646 # Participating changesets:
2650 2647 #
2651 2648 # node/ctx o - new (intermediate) commit that contains changes
2652 2649 # | from working dir to go into amending commit
2653 2650 # | (or a workingctx if there were no changes)
2654 2651 # |
2655 2652 # old o - changeset to amend
2656 2653 # |
2657 2654 # base o - parent of amending changeset
2658 2655
2659 2656 # Update extra dict from amended commit (e.g. to preserve graft
2660 2657 # source)
2661 2658 extra.update(old.extra())
2662 2659
2663 2660 # Also update it from the intermediate commit or from the wctx
2664 2661 extra.update(ctx.extra())
2665 2662
2666 2663 if len(old.parents()) > 1:
2667 2664 # ctx.files() isn't reliable for merges, so fall back to the
2668 2665 # slower repo.status() method
2669 2666 files = set([fn for st in repo.status(base, old)[:3]
2670 2667 for fn in st])
2671 2668 else:
2672 2669 files = set(old.files())
2673 2670
2674 2671 # Second, we use either the commit we just did, or if there were no
2675 2672 # changes the parent of the working directory as the version of the
2676 2673 # files in the final amend commit
2677 2674 if node:
2678 2675 ui.note(_('copying changeset %s to %s\n') % (ctx, base))
2679 2676
2680 2677 user = ctx.user()
2681 2678 date = ctx.date()
2682 2679 # Recompute copies (avoid recording a -> b -> a)
2683 2680 copied = copies.pathcopies(base, ctx)
2684 2681 if old.p2:
2685 2682 copied.update(copies.pathcopies(old.p2(), ctx))
2686 2683
2687 2684 # Prune files which were reverted by the updates: if old
2688 2685 # introduced file X and our intermediate commit, node,
2689 2686 # renamed that file, then those two files are the same and
2690 2687 # we can discard X from our list of files. Likewise if X
2691 2688 # was deleted, it's no longer relevant
2692 2689 files.update(ctx.files())
2693 2690 files = [f for f in files if not samefile(f, ctx, base)]
2694 2691
2695 2692 def filectxfn(repo, ctx_, path):
2696 2693 try:
2697 2694 fctx = ctx[path]
2698 2695 flags = fctx.flags()
2699 2696 mctx = context.memfilectx(repo,
2700 2697 fctx.path(), fctx.data(),
2701 2698 islink='l' in flags,
2702 2699 isexec='x' in flags,
2703 2700 copied=copied.get(path))
2704 2701 return mctx
2705 2702 except KeyError:
2706 2703 return None
2707 2704 else:
2708 2705 ui.note(_('copying changeset %s to %s\n') % (old, base))
2709 2706
2710 2707 # Use version of files as in the old cset
2711 2708 def filectxfn(repo, ctx_, path):
2712 2709 try:
2713 2710 return old.filectx(path)
2714 2711 except KeyError:
2715 2712 return None
2716 2713
2717 2714 user = opts.get('user') or old.user()
2718 2715 date = opts.get('date') or old.date()
2719 2716 editform = mergeeditform(old, 'commit.amend')
2720 2717 editor = getcommiteditor(editform=editform, **opts)
2721 2718 if not message:
2722 2719 editor = getcommiteditor(edit=True, editform=editform)
2723 2720 message = old.description()
2724 2721
2725 2722 pureextra = extra.copy()
2726 2723 extra['amend_source'] = old.hex()
2727 2724
2728 2725 new = context.memctx(repo,
2729 2726 parents=[base.node(), old.p2().node()],
2730 2727 text=message,
2731 2728 files=files,
2732 2729 filectxfn=filectxfn,
2733 2730 user=user,
2734 2731 date=date,
2735 2732 extra=extra,
2736 2733 editor=editor)
2737 2734
2738 2735 newdesc = changelog.stripdesc(new.description())
2739 2736 if ((not node)
2740 2737 and newdesc == old.description()
2741 2738 and user == old.user()
2742 2739 and date == old.date()
2743 2740 and pureextra == old.extra()):
2744 2741 # nothing changed. continuing here would create a new node
2745 2742 # anyway because of the amend_source noise.
2746 2743 #
2747 2744 # This not what we expect from amend.
2748 2745 return old.node()
2749 2746
2750 2747 ph = repo.ui.config('phases', 'new-commit', phases.draft)
2751 2748 try:
2752 2749 if opts.get('secret'):
2753 2750 commitphase = 'secret'
2754 2751 else:
2755 2752 commitphase = old.phase()
2756 2753 repo.ui.setconfig('phases', 'new-commit', commitphase, 'amend')
2757 2754 newid = repo.commitctx(new)
2758 2755 finally:
2759 2756 repo.ui.setconfig('phases', 'new-commit', ph, 'amend')
2760 2757 if newid != old.node():
2761 2758 # Reroute the working copy parent to the new changeset
2762 2759 repo.setparents(newid, nullid)
2763 2760
2764 2761 # Move bookmarks from old parent to amend commit
2765 2762 bms = repo.nodebookmarks(old.node())
2766 2763 if bms:
2767 2764 marks = repo._bookmarks
2768 2765 for bm in bms:
2769 2766 ui.debug('moving bookmarks %r from %s to %s\n' %
2770 2767 (marks, old.hex(), hex(newid)))
2771 2768 marks[bm] = newid
2772 2769 marks.recordchange(tr)
2773 2770 #commit the whole amend process
2774 2771 if createmarkers:
2775 2772 # mark the new changeset as successor of the rewritten one
2776 2773 new = repo[newid]
2777 2774 obs = [(old, (new,))]
2778 2775 if node:
2779 2776 obs.append((ctx, ()))
2780 2777
2781 2778 obsolete.createmarkers(repo, obs)
2782 2779 if not createmarkers and newid != old.node():
2783 2780 # Strip the intermediate commit (if there was one) and the amended
2784 2781 # commit
2785 2782 if node:
2786 2783 ui.note(_('stripping intermediate changeset %s\n') % ctx)
2787 2784 ui.note(_('stripping amended changeset %s\n') % old)
2788 2785 repair.strip(ui, repo, old.node(), topic='amend-backup')
2789 2786 finally:
2790 2787 lockmod.release(lock, wlock)
2791 2788 return newid
2792 2789
2793 2790 def commiteditor(repo, ctx, subs, editform=''):
2794 2791 if ctx.description():
2795 2792 return ctx.description()
2796 2793 return commitforceeditor(repo, ctx, subs, editform=editform,
2797 2794 unchangedmessagedetection=True)
2798 2795
2799 2796 def commitforceeditor(repo, ctx, subs, finishdesc=None, extramsg=None,
2800 2797 editform='', unchangedmessagedetection=False):
2801 2798 if not extramsg:
2802 2799 extramsg = _("Leave message empty to abort commit.")
2803 2800
2804 2801 forms = [e for e in editform.split('.') if e]
2805 2802 forms.insert(0, 'changeset')
2806 2803 templatetext = None
2807 2804 while forms:
2808 2805 tmpl = repo.ui.config('committemplate', '.'.join(forms))
2809 2806 if tmpl:
2810 2807 templatetext = committext = buildcommittemplate(
2811 2808 repo, ctx, subs, extramsg, tmpl)
2812 2809 break
2813 2810 forms.pop()
2814 2811 else:
2815 2812 committext = buildcommittext(repo, ctx, subs, extramsg)
2816 2813
2817 2814 # run editor in the repository root
2818 2815 olddir = os.getcwd()
2819 2816 os.chdir(repo.root)
2820 2817
2821 2818 # make in-memory changes visible to external process
2822 2819 tr = repo.currenttransaction()
2823 2820 repo.dirstate.write(tr)
2824 2821 pending = tr and tr.writepending() and repo.root
2825 2822
2826 2823 editortext = repo.ui.edit(committext, ctx.user(), ctx.extra(),
2827 2824 editform=editform, pending=pending)
2828 2825 text = re.sub("(?m)^HG:.*(\n|$)", "", editortext)
2829 2826 os.chdir(olddir)
2830 2827
2831 2828 if finishdesc:
2832 2829 text = finishdesc(text)
2833 2830 if not text.strip():
2834 2831 raise error.Abort(_("empty commit message"))
2835 2832 if unchangedmessagedetection and editortext == templatetext:
2836 2833 raise error.Abort(_("commit message unchanged"))
2837 2834
2838 2835 return text
2839 2836
2840 2837 def buildcommittemplate(repo, ctx, subs, extramsg, tmpl):
2841 2838 ui = repo.ui
2842 2839 tmpl, mapfile = gettemplate(ui, tmpl, None)
2843 2840
2844 2841 t = changeset_templater(ui, repo, None, {}, tmpl, mapfile, False)
2845 2842
2846 2843 for k, v in repo.ui.configitems('committemplate'):
2847 2844 if k != 'changeset':
2848 2845 t.t.cache[k] = v
2849 2846
2850 2847 if not extramsg:
2851 2848 extramsg = '' # ensure that extramsg is string
2852 2849
2853 2850 ui.pushbuffer()
2854 2851 t.show(ctx, extramsg=extramsg)
2855 2852 return ui.popbuffer()
2856 2853
2857 2854 def hgprefix(msg):
2858 2855 return "\n".join(["HG: %s" % a for a in msg.split("\n") if a])
2859 2856
2860 2857 def buildcommittext(repo, ctx, subs, extramsg):
2861 2858 edittext = []
2862 2859 modified, added, removed = ctx.modified(), ctx.added(), ctx.removed()
2863 2860 if ctx.description():
2864 2861 edittext.append(ctx.description())
2865 2862 edittext.append("")
2866 2863 edittext.append("") # Empty line between message and comments.
2867 2864 edittext.append(hgprefix(_("Enter commit message."
2868 2865 " Lines beginning with 'HG:' are removed.")))
2869 2866 edittext.append(hgprefix(extramsg))
2870 2867 edittext.append("HG: --")
2871 2868 edittext.append(hgprefix(_("user: %s") % ctx.user()))
2872 2869 if ctx.p2():
2873 2870 edittext.append(hgprefix(_("branch merge")))
2874 2871 if ctx.branch():
2875 2872 edittext.append(hgprefix(_("branch '%s'") % ctx.branch()))
2876 2873 if bookmarks.isactivewdirparent(repo):
2877 2874 edittext.append(hgprefix(_("bookmark '%s'") % repo._activebookmark))
2878 2875 edittext.extend([hgprefix(_("subrepo %s") % s) for s in subs])
2879 2876 edittext.extend([hgprefix(_("added %s") % f) for f in added])
2880 2877 edittext.extend([hgprefix(_("changed %s") % f) for f in modified])
2881 2878 edittext.extend([hgprefix(_("removed %s") % f) for f in removed])
2882 2879 if not added and not modified and not removed:
2883 2880 edittext.append(hgprefix(_("no files changed")))
2884 2881 edittext.append("")
2885 2882
2886 2883 return "\n".join(edittext)
2887 2884
2888 2885 def commitstatus(repo, node, branch, bheads=None, opts=None):
2889 2886 if opts is None:
2890 2887 opts = {}
2891 2888 ctx = repo[node]
2892 2889 parents = ctx.parents()
2893 2890
2894 2891 if (not opts.get('amend') and bheads and node not in bheads and not
2895 2892 [x for x in parents if x.node() in bheads and x.branch() == branch]):
2896 2893 repo.ui.status(_('created new head\n'))
2897 2894 # The message is not printed for initial roots. For the other
2898 2895 # changesets, it is printed in the following situations:
2899 2896 #
2900 2897 # Par column: for the 2 parents with ...
2901 2898 # N: null or no parent
2902 2899 # B: parent is on another named branch
2903 2900 # C: parent is a regular non head changeset
2904 2901 # H: parent was a branch head of the current branch
2905 2902 # Msg column: whether we print "created new head" message
2906 2903 # In the following, it is assumed that there already exists some
2907 2904 # initial branch heads of the current branch, otherwise nothing is
2908 2905 # printed anyway.
2909 2906 #
2910 2907 # Par Msg Comment
2911 2908 # N N y additional topo root
2912 2909 #
2913 2910 # B N y additional branch root
2914 2911 # C N y additional topo head
2915 2912 # H N n usual case
2916 2913 #
2917 2914 # B B y weird additional branch root
2918 2915 # C B y branch merge
2919 2916 # H B n merge with named branch
2920 2917 #
2921 2918 # C C y additional head from merge
2922 2919 # C H n merge with a head
2923 2920 #
2924 2921 # H H n head merge: head count decreases
2925 2922
2926 2923 if not opts.get('close_branch'):
2927 2924 for r in parents:
2928 2925 if r.closesbranch() and r.branch() == branch:
2929 2926 repo.ui.status(_('reopening closed branch head %d\n') % r)
2930 2927
2931 2928 if repo.ui.debugflag:
2932 2929 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx.hex()))
2933 2930 elif repo.ui.verbose:
2934 2931 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx))
2935 2932
2936 2933 def postcommitstatus(repo, pats, opts):
2937 2934 return repo.status(match=scmutil.match(repo[None], pats, opts))
2938 2935
2939 2936 def revert(ui, repo, ctx, parents, *pats, **opts):
2940 2937 parent, p2 = parents
2941 2938 node = ctx.node()
2942 2939
2943 2940 mf = ctx.manifest()
2944 2941 if node == p2:
2945 2942 parent = p2
2946 2943
2947 2944 # need all matching names in dirstate and manifest of target rev,
2948 2945 # so have to walk both. do not print errors if files exist in one
2949 2946 # but not other. in both cases, filesets should be evaluated against
2950 2947 # workingctx to get consistent result (issue4497). this means 'set:**'
2951 2948 # cannot be used to select missing files from target rev.
2952 2949
2953 2950 # `names` is a mapping for all elements in working copy and target revision
2954 2951 # The mapping is in the form:
2955 2952 # <asb path in repo> -> (<path from CWD>, <exactly specified by matcher?>)
2956 2953 names = {}
2957 2954
2958 2955 with repo.wlock():
2959 2956 ## filling of the `names` mapping
2960 2957 # walk dirstate to fill `names`
2961 2958
2962 2959 interactive = opts.get('interactive', False)
2963 2960 wctx = repo[None]
2964 2961 m = scmutil.match(wctx, pats, opts)
2965 2962
2966 2963 # we'll need this later
2967 2964 targetsubs = sorted(s for s in wctx.substate if m(s))
2968 2965
2969 2966 if not m.always():
2970 2967 for abs in repo.walk(matchmod.badmatch(m, lambda x, y: False)):
2971 2968 names[abs] = m.rel(abs), m.exact(abs)
2972 2969
2973 2970 # walk target manifest to fill `names`
2974 2971
2975 2972 def badfn(path, msg):
2976 2973 if path in names:
2977 2974 return
2978 2975 if path in ctx.substate:
2979 2976 return
2980 2977 path_ = path + '/'
2981 2978 for f in names:
2982 2979 if f.startswith(path_):
2983 2980 return
2984 2981 ui.warn("%s: %s\n" % (m.rel(path), msg))
2985 2982
2986 2983 for abs in ctx.walk(matchmod.badmatch(m, badfn)):
2987 2984 if abs not in names:
2988 2985 names[abs] = m.rel(abs), m.exact(abs)
2989 2986
2990 2987 # Find status of all file in `names`.
2991 2988 m = scmutil.matchfiles(repo, names)
2992 2989
2993 2990 changes = repo.status(node1=node, match=m,
2994 2991 unknown=True, ignored=True, clean=True)
2995 2992 else:
2996 2993 changes = repo.status(node1=node, match=m)
2997 2994 for kind in changes:
2998 2995 for abs in kind:
2999 2996 names[abs] = m.rel(abs), m.exact(abs)
3000 2997
3001 2998 m = scmutil.matchfiles(repo, names)
3002 2999
3003 3000 modified = set(changes.modified)
3004 3001 added = set(changes.added)
3005 3002 removed = set(changes.removed)
3006 3003 _deleted = set(changes.deleted)
3007 3004 unknown = set(changes.unknown)
3008 3005 unknown.update(changes.ignored)
3009 3006 clean = set(changes.clean)
3010 3007 modadded = set()
3011 3008
3012 3009 # split between files known in target manifest and the others
3013 3010 smf = set(mf)
3014 3011
3015 3012 # determine the exact nature of the deleted changesets
3016 3013 deladded = _deleted - smf
3017 3014 deleted = _deleted - deladded
3018 3015
3019 3016 # We need to account for the state of the file in the dirstate,
3020 3017 # even when we revert against something else than parent. This will
3021 3018 # slightly alter the behavior of revert (doing back up or not, delete
3022 3019 # or just forget etc).
3023 3020 if parent == node:
3024 3021 dsmodified = modified
3025 3022 dsadded = added
3026 3023 dsremoved = removed
3027 3024 # store all local modifications, useful later for rename detection
3028 3025 localchanges = dsmodified | dsadded
3029 3026 modified, added, removed = set(), set(), set()
3030 3027 else:
3031 3028 changes = repo.status(node1=parent, match=m)
3032 3029 dsmodified = set(changes.modified)
3033 3030 dsadded = set(changes.added)
3034 3031 dsremoved = set(changes.removed)
3035 3032 # store all local modifications, useful later for rename detection
3036 3033 localchanges = dsmodified | dsadded
3037 3034
3038 3035 # only take into account for removes between wc and target
3039 3036 clean |= dsremoved - removed
3040 3037 dsremoved &= removed
3041 3038 # distinct between dirstate remove and other
3042 3039 removed -= dsremoved
3043 3040
3044 3041 modadded = added & dsmodified
3045 3042 added -= modadded
3046 3043
3047 3044 # tell newly modified apart.
3048 3045 dsmodified &= modified
3049 3046 dsmodified |= modified & dsadded # dirstate added may need backup
3050 3047 modified -= dsmodified
3051 3048
3052 3049 # We need to wait for some post-processing to update this set
3053 3050 # before making the distinction. The dirstate will be used for
3054 3051 # that purpose.
3055 3052 dsadded = added
3056 3053
3057 3054 # in case of merge, files that are actually added can be reported as
3058 3055 # modified, we need to post process the result
3059 3056 if p2 != nullid:
3060 3057 mergeadd = dsmodified - smf
3061 3058 dsadded |= mergeadd
3062 3059 dsmodified -= mergeadd
3063 3060
3064 3061 # if f is a rename, update `names` to also revert the source
3065 3062 cwd = repo.getcwd()
3066 3063 for f in localchanges:
3067 3064 src = repo.dirstate.copied(f)
3068 3065 # XXX should we check for rename down to target node?
3069 3066 if src and src not in names and repo.dirstate[src] == 'r':
3070 3067 dsremoved.add(src)
3071 3068 names[src] = (repo.pathto(src, cwd), True)
3072 3069
3073 3070 # distinguish between file to forget and the other
3074 3071 added = set()
3075 3072 for abs in dsadded:
3076 3073 if repo.dirstate[abs] != 'a':
3077 3074 added.add(abs)
3078 3075 dsadded -= added
3079 3076
3080 3077 for abs in deladded:
3081 3078 if repo.dirstate[abs] == 'a':
3082 3079 dsadded.add(abs)
3083 3080 deladded -= dsadded
3084 3081
3085 3082 # For files marked as removed, we check if an unknown file is present at
3086 3083 # the same path. If a such file exists it may need to be backed up.
3087 3084 # Making the distinction at this stage helps have simpler backup
3088 3085 # logic.
3089 3086 removunk = set()
3090 3087 for abs in removed:
3091 3088 target = repo.wjoin(abs)
3092 3089 if os.path.lexists(target):
3093 3090 removunk.add(abs)
3094 3091 removed -= removunk
3095 3092
3096 3093 dsremovunk = set()
3097 3094 for abs in dsremoved:
3098 3095 target = repo.wjoin(abs)
3099 3096 if os.path.lexists(target):
3100 3097 dsremovunk.add(abs)
3101 3098 dsremoved -= dsremovunk
3102 3099
3103 3100 # action to be actually performed by revert
3104 3101 # (<list of file>, message>) tuple
3105 3102 actions = {'revert': ([], _('reverting %s\n')),
3106 3103 'add': ([], _('adding %s\n')),
3107 3104 'remove': ([], _('removing %s\n')),
3108 3105 'drop': ([], _('removing %s\n')),
3109 3106 'forget': ([], _('forgetting %s\n')),
3110 3107 'undelete': ([], _('undeleting %s\n')),
3111 3108 'noop': (None, _('no changes needed to %s\n')),
3112 3109 'unknown': (None, _('file not managed: %s\n')),
3113 3110 }
3114 3111
3115 3112 # "constant" that convey the backup strategy.
3116 3113 # All set to `discard` if `no-backup` is set do avoid checking
3117 3114 # no_backup lower in the code.
3118 3115 # These values are ordered for comparison purposes
3119 3116 backupinteractive = 3 # do backup if interactively modified
3120 3117 backup = 2 # unconditionally do backup
3121 3118 check = 1 # check if the existing file differs from target
3122 3119 discard = 0 # never do backup
3123 3120 if opts.get('no_backup'):
3124 3121 backupinteractive = backup = check = discard
3125 3122 if interactive:
3126 3123 dsmodifiedbackup = backupinteractive
3127 3124 else:
3128 3125 dsmodifiedbackup = backup
3129 3126 tobackup = set()
3130 3127
3131 3128 backupanddel = actions['remove']
3132 3129 if not opts.get('no_backup'):
3133 3130 backupanddel = actions['drop']
3134 3131
3135 3132 disptable = (
3136 3133 # dispatch table:
3137 3134 # file state
3138 3135 # action
3139 3136 # make backup
3140 3137
3141 3138 ## Sets that results that will change file on disk
3142 3139 # Modified compared to target, no local change
3143 3140 (modified, actions['revert'], discard),
3144 3141 # Modified compared to target, but local file is deleted
3145 3142 (deleted, actions['revert'], discard),
3146 3143 # Modified compared to target, local change
3147 3144 (dsmodified, actions['revert'], dsmodifiedbackup),
3148 3145 # Added since target
3149 3146 (added, actions['remove'], discard),
3150 3147 # Added in working directory
3151 3148 (dsadded, actions['forget'], discard),
3152 3149 # Added since target, have local modification
3153 3150 (modadded, backupanddel, backup),
3154 3151 # Added since target but file is missing in working directory
3155 3152 (deladded, actions['drop'], discard),
3156 3153 # Removed since target, before working copy parent
3157 3154 (removed, actions['add'], discard),
3158 3155 # Same as `removed` but an unknown file exists at the same path
3159 3156 (removunk, actions['add'], check),
3160 3157 # Removed since targe, marked as such in working copy parent
3161 3158 (dsremoved, actions['undelete'], discard),
3162 3159 # Same as `dsremoved` but an unknown file exists at the same path
3163 3160 (dsremovunk, actions['undelete'], check),
3164 3161 ## the following sets does not result in any file changes
3165 3162 # File with no modification
3166 3163 (clean, actions['noop'], discard),
3167 3164 # Existing file, not tracked anywhere
3168 3165 (unknown, actions['unknown'], discard),
3169 3166 )
3170 3167
3171 3168 for abs, (rel, exact) in sorted(names.items()):
3172 3169 # target file to be touch on disk (relative to cwd)
3173 3170 target = repo.wjoin(abs)
3174 3171 # search the entry in the dispatch table.
3175 3172 # if the file is in any of these sets, it was touched in the working
3176 3173 # directory parent and we are sure it needs to be reverted.
3177 3174 for table, (xlist, msg), dobackup in disptable:
3178 3175 if abs not in table:
3179 3176 continue
3180 3177 if xlist is not None:
3181 3178 xlist.append(abs)
3182 3179 if dobackup:
3183 3180 # If in interactive mode, don't automatically create
3184 3181 # .orig files (issue4793)
3185 3182 if dobackup == backupinteractive:
3186 3183 tobackup.add(abs)
3187 3184 elif (backup <= dobackup or wctx[abs].cmp(ctx[abs])):
3188 3185 bakname = scmutil.origpath(ui, repo, rel)
3189 3186 ui.note(_('saving current version of %s as %s\n') %
3190 3187 (rel, bakname))
3191 3188 if not opts.get('dry_run'):
3192 3189 if interactive:
3193 3190 util.copyfile(target, bakname)
3194 3191 else:
3195 3192 util.rename(target, bakname)
3196 3193 if ui.verbose or not exact:
3197 3194 if not isinstance(msg, basestring):
3198 3195 msg = msg(abs)
3199 3196 ui.status(msg % rel)
3200 3197 elif exact:
3201 3198 ui.warn(msg % rel)
3202 3199 break
3203 3200
3204 3201 if not opts.get('dry_run'):
3205 3202 needdata = ('revert', 'add', 'undelete')
3206 3203 _revertprefetch(repo, ctx, *[actions[name][0] for name in needdata])
3207 3204 _performrevert(repo, parents, ctx, actions, interactive, tobackup)
3208 3205
3209 3206 if targetsubs:
3210 3207 # Revert the subrepos on the revert list
3211 3208 for sub in targetsubs:
3212 3209 try:
3213 3210 wctx.sub(sub).revert(ctx.substate[sub], *pats, **opts)
3214 3211 except KeyError:
3215 3212 raise error.Abort("subrepository '%s' does not exist in %s!"
3216 3213 % (sub, short(ctx.node())))
3217 3214
3218 3215 def _revertprefetch(repo, ctx, *files):
3219 3216 """Let extension changing the storage layer prefetch content"""
3220 3217 pass
3221 3218
3222 3219 def _performrevert(repo, parents, ctx, actions, interactive=False,
3223 3220 tobackup=None):
3224 3221 """function that actually perform all the actions computed for revert
3225 3222
3226 3223 This is an independent function to let extension to plug in and react to
3227 3224 the imminent revert.
3228 3225
3229 3226 Make sure you have the working directory locked when calling this function.
3230 3227 """
3231 3228 parent, p2 = parents
3232 3229 node = ctx.node()
3233 3230 excluded_files = []
3234 3231 matcher_opts = {"exclude": excluded_files}
3235 3232
3236 3233 def checkout(f):
3237 3234 fc = ctx[f]
3238 3235 repo.wwrite(f, fc.data(), fc.flags())
3239 3236
3240 3237 audit_path = pathutil.pathauditor(repo.root)
3241 3238 for f in actions['forget'][0]:
3242 3239 if interactive:
3243 3240 choice = \
3244 3241 repo.ui.promptchoice(
3245 3242 _("forget added file %s (yn)?$$ &Yes $$ &No")
3246 3243 % f)
3247 3244 if choice == 0:
3248 3245 repo.dirstate.drop(f)
3249 3246 else:
3250 3247 excluded_files.append(repo.wjoin(f))
3251 3248 else:
3252 3249 repo.dirstate.drop(f)
3253 3250 for f in actions['remove'][0]:
3254 3251 audit_path(f)
3255 3252 try:
3256 3253 util.unlinkpath(repo.wjoin(f))
3257 3254 except OSError:
3258 3255 pass
3259 3256 repo.dirstate.remove(f)
3260 3257 for f in actions['drop'][0]:
3261 3258 audit_path(f)
3262 3259 repo.dirstate.remove(f)
3263 3260
3264 3261 normal = None
3265 3262 if node == parent:
3266 3263 # We're reverting to our parent. If possible, we'd like status
3267 3264 # to report the file as clean. We have to use normallookup for
3268 3265 # merges to avoid losing information about merged/dirty files.
3269 3266 if p2 != nullid:
3270 3267 normal = repo.dirstate.normallookup
3271 3268 else:
3272 3269 normal = repo.dirstate.normal
3273 3270
3274 3271 newlyaddedandmodifiedfiles = set()
3275 3272 if interactive:
3276 3273 # Prompt the user for changes to revert
3277 3274 torevert = [repo.wjoin(f) for f in actions['revert'][0]]
3278 3275 m = scmutil.match(ctx, torevert, matcher_opts)
3279 3276 diffopts = patch.difffeatureopts(repo.ui, whitespace=True)
3280 3277 diffopts.nodates = True
3281 3278 diffopts.git = True
3282 3279 reversehunks = repo.ui.configbool('experimental',
3283 3280 'revertalternateinteractivemode',
3284 3281 True)
3285 3282 if reversehunks:
3286 3283 diff = patch.diff(repo, ctx.node(), None, m, opts=diffopts)
3287 3284 else:
3288 3285 diff = patch.diff(repo, None, ctx.node(), m, opts=diffopts)
3289 3286 originalchunks = patch.parsepatch(diff)
3290 3287 operation = 'discard' if node == parent else 'revert'
3291 3288
3292 3289 try:
3293 3290
3294 3291 chunks, opts = recordfilter(repo.ui, originalchunks,
3295 3292 operation=operation)
3296 3293 if reversehunks:
3297 3294 chunks = patch.reversehunks(chunks)
3298 3295
3299 3296 except patch.PatchError as err:
3300 3297 raise error.Abort(_('error parsing patch: %s') % err)
3301 3298
3302 3299 newlyaddedandmodifiedfiles = newandmodified(chunks, originalchunks)
3303 3300 if tobackup is None:
3304 3301 tobackup = set()
3305 3302 # Apply changes
3306 3303 fp = stringio()
3307 3304 for c in chunks:
3308 3305 # Create a backup file only if this hunk should be backed up
3309 3306 if ishunk(c) and c.header.filename() in tobackup:
3310 3307 abs = c.header.filename()
3311 3308 target = repo.wjoin(abs)
3312 3309 bakname = scmutil.origpath(repo.ui, repo, m.rel(abs))
3313 3310 util.copyfile(target, bakname)
3314 3311 tobackup.remove(abs)
3315 3312 c.write(fp)
3316 3313 dopatch = fp.tell()
3317 3314 fp.seek(0)
3318 3315 if dopatch:
3319 3316 try:
3320 3317 patch.internalpatch(repo.ui, repo, fp, 1, eolmode=None)
3321 3318 except patch.PatchError as err:
3322 3319 raise error.Abort(str(err))
3323 3320 del fp
3324 3321 else:
3325 3322 for f in actions['revert'][0]:
3326 3323 checkout(f)
3327 3324 if normal:
3328 3325 normal(f)
3329 3326
3330 3327 for f in actions['add'][0]:
3331 3328 # Don't checkout modified files, they are already created by the diff
3332 3329 if f not in newlyaddedandmodifiedfiles:
3333 3330 checkout(f)
3334 3331 repo.dirstate.add(f)
3335 3332
3336 3333 normal = repo.dirstate.normallookup
3337 3334 if node == parent and p2 == nullid:
3338 3335 normal = repo.dirstate.normal
3339 3336 for f in actions['undelete'][0]:
3340 3337 checkout(f)
3341 3338 normal(f)
3342 3339
3343 3340 copied = copies.pathcopies(repo[parent], ctx)
3344 3341
3345 3342 for f in actions['add'][0] + actions['undelete'][0] + actions['revert'][0]:
3346 3343 if f in copied:
3347 3344 repo.dirstate.copy(copied[f], f)
3348 3345
3349 3346 def command(table):
3350 3347 """Returns a function object to be used as a decorator for making commands.
3351 3348
3352 3349 This function receives a command table as its argument. The table should
3353 3350 be a dict.
3354 3351
3355 3352 The returned function can be used as a decorator for adding commands
3356 3353 to that command table. This function accepts multiple arguments to define
3357 3354 a command.
3358 3355
3359 3356 The first argument is the command name.
3360 3357
3361 3358 The options argument is an iterable of tuples defining command arguments.
3362 3359 See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.
3363 3360
3364 3361 The synopsis argument defines a short, one line summary of how to use the
3365 3362 command. This shows up in the help output.
3366 3363
3367 3364 The norepo argument defines whether the command does not require a
3368 3365 local repository. Most commands operate against a repository, thus the
3369 3366 default is False.
3370 3367
3371 3368 The optionalrepo argument defines whether the command optionally requires
3372 3369 a local repository.
3373 3370
3374 3371 The inferrepo argument defines whether to try to find a repository from the
3375 3372 command line arguments. If True, arguments will be examined for potential
3376 3373 repository locations. See ``findrepo()``. If a repository is found, it
3377 3374 will be used.
3378 3375 """
3379 3376 def cmd(name, options=(), synopsis=None, norepo=False, optionalrepo=False,
3380 3377 inferrepo=False):
3381 3378 def decorator(func):
3382 3379 func.norepo = norepo
3383 3380 func.optionalrepo = optionalrepo
3384 3381 func.inferrepo = inferrepo
3385 3382 if synopsis:
3386 3383 table[name] = func, list(options), synopsis
3387 3384 else:
3388 3385 table[name] = func, list(options)
3389 3386 return func
3390 3387 return decorator
3391 3388
3392 3389 return cmd
3393 3390
3394 3391 # a list of (ui, repo, otherpeer, opts, missing) functions called by
3395 3392 # commands.outgoing. "missing" is "missing" of the result of
3396 3393 # "findcommonoutgoing()"
3397 3394 outgoinghooks = util.hooks()
3398 3395
3399 3396 # a list of (ui, repo) functions called by commands.summary
3400 3397 summaryhooks = util.hooks()
3401 3398
3402 3399 # a list of (ui, repo, opts, changes) functions called by commands.summary.
3403 3400 #
3404 3401 # functions should return tuple of booleans below, if 'changes' is None:
3405 3402 # (whether-incomings-are-needed, whether-outgoings-are-needed)
3406 3403 #
3407 3404 # otherwise, 'changes' is a tuple of tuples below:
3408 3405 # - (sourceurl, sourcebranch, sourcepeer, incoming)
3409 3406 # - (desturl, destbranch, destpeer, outgoing)
3410 3407 summaryremotehooks = util.hooks()
3411 3408
3412 3409 # A list of state files kept by multistep operations like graft.
3413 3410 # Since graft cannot be aborted, it is considered 'clearable' by update.
3414 3411 # note: bisect is intentionally excluded
3415 3412 # (state file, clearable, allowcommit, error, hint)
3416 3413 unfinishedstates = [
3417 3414 ('graftstate', True, False, _('graft in progress'),
3418 3415 _("use 'hg graft --continue' or 'hg update' to abort")),
3419 3416 ('updatestate', True, False, _('last update was interrupted'),
3420 3417 _("use 'hg update' to get a consistent checkout"))
3421 3418 ]
3422 3419
3423 3420 def checkunfinished(repo, commit=False):
3424 3421 '''Look for an unfinished multistep operation, like graft, and abort
3425 3422 if found. It's probably good to check this right before
3426 3423 bailifchanged().
3427 3424 '''
3428 3425 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3429 3426 if commit and allowcommit:
3430 3427 continue
3431 3428 if repo.vfs.exists(f):
3432 3429 raise error.Abort(msg, hint=hint)
3433 3430
3434 3431 def clearunfinished(repo):
3435 3432 '''Check for unfinished operations (as above), and clear the ones
3436 3433 that are clearable.
3437 3434 '''
3438 3435 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3439 3436 if not clearable and repo.vfs.exists(f):
3440 3437 raise error.Abort(msg, hint=hint)
3441 3438 for f, clearable, allowcommit, msg, hint in unfinishedstates:
3442 3439 if clearable and repo.vfs.exists(f):
3443 3440 util.unlink(repo.join(f))
3444 3441
3445 3442 afterresolvedstates = [
3446 3443 ('graftstate',
3447 3444 _('hg graft --continue')),
3448 3445 ]
3449 3446
3450 3447 def howtocontinue(repo):
3451 3448 '''Check for an unfinished operation and return the command to finish
3452 3449 it.
3453 3450
3454 3451 afterresolvedstates tupples define a .hg/{file} and the corresponding
3455 3452 command needed to finish it.
3456 3453
3457 3454 Returns a (msg, warning) tuple. 'msg' is a string and 'warning' is
3458 3455 a boolean.
3459 3456 '''
3460 3457 contmsg = _("continue: %s")
3461 3458 for f, msg in afterresolvedstates:
3462 3459 if repo.vfs.exists(f):
3463 3460 return contmsg % msg, True
3464 3461 workingctx = repo[None]
3465 3462 dirty = any(repo.status()) or any(workingctx.sub(s).dirty()
3466 3463 for s in workingctx.substate)
3467 3464 if dirty:
3468 3465 return contmsg % _("hg commit"), False
3469 3466 return None, None
3470 3467
3471 3468 def checkafterresolved(repo):
3472 3469 '''Inform the user about the next action after completing hg resolve
3473 3470
3474 3471 If there's a matching afterresolvedstates, howtocontinue will yield
3475 3472 repo.ui.warn as the reporter.
3476 3473
3477 3474 Otherwise, it will yield repo.ui.note.
3478 3475 '''
3479 3476 msg, warning = howtocontinue(repo)
3480 3477 if msg is not None:
3481 3478 if warning:
3482 3479 repo.ui.warn("%s\n" % msg)
3483 3480 else:
3484 3481 repo.ui.note("%s\n" % msg)
3485 3482
3486 3483 def wrongtooltocontinue(repo, task):
3487 3484 '''Raise an abort suggesting how to properly continue if there is an
3488 3485 active task.
3489 3486
3490 3487 Uses howtocontinue() to find the active task.
3491 3488
3492 3489 If there's no task (repo.ui.note for 'hg commit'), it does not offer
3493 3490 a hint.
3494 3491 '''
3495 3492 after = howtocontinue(repo)
3496 3493 hint = None
3497 3494 if after[1]:
3498 3495 hint = after[0]
3499 3496 raise error.Abort(_('no %s in progress') % task, hint=hint)
3500 3497
3501 3498 class dirstateguard(object):
3502 3499 '''Restore dirstate at unexpected failure.
3503 3500
3504 3501 At the construction, this class does:
3505 3502
3506 3503 - write current ``repo.dirstate`` out, and
3507 3504 - save ``.hg/dirstate`` into the backup file
3508 3505
3509 3506 This restores ``.hg/dirstate`` from backup file, if ``release()``
3510 3507 is invoked before ``close()``.
3511 3508
3512 3509 This just removes the backup file at ``close()`` before ``release()``.
3513 3510 '''
3514 3511
3515 3512 def __init__(self, repo, name):
3516 3513 self._repo = repo
3517 3514 self._suffix = '.backup.%s.%d' % (name, id(self))
3518 3515 repo.dirstate.savebackup(repo.currenttransaction(), self._suffix)
3519 3516 self._active = True
3520 3517 self._closed = False
3521 3518
3522 3519 def __del__(self):
3523 3520 if self._active: # still active
3524 3521 # this may occur, even if this class is used correctly:
3525 3522 # for example, releasing other resources like transaction
3526 3523 # may raise exception before ``dirstateguard.release`` in
3527 3524 # ``release(tr, ....)``.
3528 3525 self._abort()
3529 3526
3530 3527 def close(self):
3531 3528 if not self._active: # already inactivated
3532 3529 msg = (_("can't close already inactivated backup: dirstate%s")
3533 3530 % self._suffix)
3534 3531 raise error.Abort(msg)
3535 3532
3536 3533 self._repo.dirstate.clearbackup(self._repo.currenttransaction(),
3537 3534 self._suffix)
3538 3535 self._active = False
3539 3536 self._closed = True
3540 3537
3541 3538 def _abort(self):
3542 3539 self._repo.dirstate.restorebackup(self._repo.currenttransaction(),
3543 3540 self._suffix)
3544 3541 self._active = False
3545 3542
3546 3543 def release(self):
3547 3544 if not self._closed:
3548 3545 if not self._active: # already inactivated
3549 3546 msg = (_("can't release already inactivated backup:"
3550 3547 " dirstate%s")
3551 3548 % self._suffix)
3552 3549 raise error.Abort(msg)
3553 3550 self._abort()
@@ -1,101 +1,115 b''
1 1 This test file aims at test topological iteration and the various configuration it can has.
2 2
3 3 $ cat >> $HGRCPATH << EOF
4 4 > [ui]
5 5 > logtemplate={rev}\n
6 6 > EOF
7 7
8 8 On this simple example, all topological branch are displayed in turn until we
9 9 can finally display 0. this implies skipping from 8 to 3 and coming back to 7
10 10 later.
11 11
12 12 $ hg init test01
13 13 $ cd test01
14 14 $ hg unbundle $TESTDIR/bundles/remote.hg
15 15 adding changesets
16 16 adding manifests
17 17 adding file changes
18 18 added 9 changesets with 7 changes to 4 files (+1 heads)
19 19 (run 'hg heads' to see heads, 'hg merge' to merge)
20 20
21 21 $ hg log -G
22 22 o 8
23 23 |
24 24 | o 7
25 25 | |
26 26 | o 6
27 27 | |
28 28 | o 5
29 29 | |
30 30 | o 4
31 31 | |
32 32 o | 3
33 33 | |
34 34 o | 2
35 35 | |
36 36 o | 1
37 37 |/
38 38 o 0
39 39
40 40
41 41 (display all nodes)
42 42
43 43 $ hg log -G -r 'sort(all(), topo)'
44 44 o 8
45 45 |
46 46 o 3
47 47 |
48 48 o 2
49 49 |
50 50 o 1
51 51 |
52 52 | o 7
53 53 | |
54 54 | o 6
55 55 | |
56 56 | o 5
57 57 | |
58 58 | o 4
59 59 |/
60 60 o 0
61 61
62 62
63 (display nodes filtered by log options)
64
65 $ hg log -G -r 'sort(all(), topo)' -k '.3'
66 o 8
67 |
68 o 3
69 |
70 ~
71 o 7
72 |
73 o 6
74 |
75 ~
76
63 77 (revset skipping nodes)
64 78
65 79 $ hg log -G --rev 'sort(not (2+6), topo)'
66 80 o 8
67 81 |
68 82 o 3
69 83 :
70 84 o 1
71 85 |
72 86 | o 7
73 87 | :
74 88 | o 5
75 89 | |
76 90 | o 4
77 91 |/
78 92 o 0
79 93
80 94
81 95 (begin) from the other branch
82 96
83 97 $ hg log -G -r 'sort(all(), topo, topo.firstbranch=5)'
84 98 o 7
85 99 |
86 100 o 6
87 101 |
88 102 o 5
89 103 |
90 104 o 4
91 105 |
92 106 | o 8
93 107 | |
94 108 | o 3
95 109 | |
96 110 | o 2
97 111 | |
98 112 | o 1
99 113 |/
100 114 o 0
101 115
General Comments 0
You need to be logged in to leave comments. Login now