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