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