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