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