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