##// END OF EJS Templates
import: pass 'editform' argument to 'cmdutil.getcommiteditor'...
FUJIWARA Katsunori -
r22011:97acb450 default
parent child Browse files
Show More
@@ -1,2636 +1,2637 b''
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 editor = getcommiteditor(**opts)
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 editor = getcommiteditor(editform='import.bypass')
690 691 memctx = context.makememctx(repo, (p1.node(), p2.node()),
691 692 message,
692 693 opts.get('user') or user,
693 694 opts.get('date') or date,
694 695 branch, files, store,
695 editor=getcommiteditor())
696 editor=editor)
696 697 n = memctx.commit()
697 698 finally:
698 699 store.close()
699 700 if opts.get('exact') and hex(n) != nodeid:
700 701 raise util.Abort(_('patch is damaged or loses information'))
701 702 if n:
702 703 # i18n: refers to a short changeset id
703 704 msg = _('created %s') % short(n)
704 705 return (msg, n, rejects)
705 706 finally:
706 707 os.unlink(tmpname)
707 708
708 709 def export(repo, revs, template='hg-%h.patch', fp=None, switch_parent=False,
709 710 opts=None):
710 711 '''export changesets as hg patches.'''
711 712
712 713 total = len(revs)
713 714 revwidth = max([len(str(rev)) for rev in revs])
714 715 filemode = {}
715 716
716 717 def single(rev, seqno, fp):
717 718 ctx = repo[rev]
718 719 node = ctx.node()
719 720 parents = [p.node() for p in ctx.parents() if p]
720 721 branch = ctx.branch()
721 722 if switch_parent:
722 723 parents.reverse()
723 724 prev = (parents and parents[0]) or nullid
724 725
725 726 shouldclose = False
726 727 if not fp and len(template) > 0:
727 728 desc_lines = ctx.description().rstrip().split('\n')
728 729 desc = desc_lines[0] #Commit always has a first line.
729 730 fp = makefileobj(repo, template, node, desc=desc, total=total,
730 731 seqno=seqno, revwidth=revwidth, mode='wb',
731 732 modemap=filemode)
732 733 if fp != template:
733 734 shouldclose = True
734 735 if fp and fp != sys.stdout and util.safehasattr(fp, 'name'):
735 736 repo.ui.note("%s\n" % fp.name)
736 737
737 738 if not fp:
738 739 write = repo.ui.write
739 740 else:
740 741 def write(s, **kw):
741 742 fp.write(s)
742 743
743 744
744 745 write("# HG changeset patch\n")
745 746 write("# User %s\n" % ctx.user())
746 747 write("# Date %d %d\n" % ctx.date())
747 748 write("# %s\n" % util.datestr(ctx.date()))
748 749 if branch and branch != 'default':
749 750 write("# Branch %s\n" % branch)
750 751 write("# Node ID %s\n" % hex(node))
751 752 write("# Parent %s\n" % hex(prev))
752 753 if len(parents) > 1:
753 754 write("# Parent %s\n" % hex(parents[1]))
754 755 write(ctx.description().rstrip())
755 756 write("\n\n")
756 757
757 758 for chunk, label in patch.diffui(repo, prev, node, opts=opts):
758 759 write(chunk, label=label)
759 760
760 761 if shouldclose:
761 762 fp.close()
762 763
763 764 for seqno, rev in enumerate(revs):
764 765 single(rev, seqno + 1, fp)
765 766
766 767 def diffordiffstat(ui, repo, diffopts, node1, node2, match,
767 768 changes=None, stat=False, fp=None, prefix='',
768 769 listsubrepos=False):
769 770 '''show diff or diffstat.'''
770 771 if fp is None:
771 772 write = ui.write
772 773 else:
773 774 def write(s, **kw):
774 775 fp.write(s)
775 776
776 777 if stat:
777 778 diffopts = diffopts.copy(context=0)
778 779 width = 80
779 780 if not ui.plain():
780 781 width = ui.termwidth()
781 782 chunks = patch.diff(repo, node1, node2, match, changes, diffopts,
782 783 prefix=prefix)
783 784 for chunk, label in patch.diffstatui(util.iterlines(chunks),
784 785 width=width,
785 786 git=diffopts.git):
786 787 write(chunk, label=label)
787 788 else:
788 789 for chunk, label in patch.diffui(repo, node1, node2, match,
789 790 changes, diffopts, prefix=prefix):
790 791 write(chunk, label=label)
791 792
792 793 if listsubrepos:
793 794 ctx1 = repo[node1]
794 795 ctx2 = repo[node2]
795 796 for subpath, sub in scmutil.itersubrepos(ctx1, ctx2):
796 797 tempnode2 = node2
797 798 try:
798 799 if node2 is not None:
799 800 tempnode2 = ctx2.substate[subpath][1]
800 801 except KeyError:
801 802 # A subrepo that existed in node1 was deleted between node1 and
802 803 # node2 (inclusive). Thus, ctx2's substate won't contain that
803 804 # subpath. The best we can do is to ignore it.
804 805 tempnode2 = None
805 806 submatch = matchmod.narrowmatcher(subpath, match)
806 807 sub.diff(ui, diffopts, tempnode2, submatch, changes=changes,
807 808 stat=stat, fp=fp, prefix=prefix)
808 809
809 810 class changeset_printer(object):
810 811 '''show changeset information when templating not requested.'''
811 812
812 813 def __init__(self, ui, repo, patch, diffopts, buffered):
813 814 self.ui = ui
814 815 self.repo = repo
815 816 self.buffered = buffered
816 817 self.patch = patch
817 818 self.diffopts = diffopts
818 819 self.header = {}
819 820 self.hunk = {}
820 821 self.lastheader = None
821 822 self.footer = None
822 823
823 824 def flush(self, rev):
824 825 if rev in self.header:
825 826 h = self.header[rev]
826 827 if h != self.lastheader:
827 828 self.lastheader = h
828 829 self.ui.write(h)
829 830 del self.header[rev]
830 831 if rev in self.hunk:
831 832 self.ui.write(self.hunk[rev])
832 833 del self.hunk[rev]
833 834 return 1
834 835 return 0
835 836
836 837 def close(self):
837 838 if self.footer:
838 839 self.ui.write(self.footer)
839 840
840 841 def show(self, ctx, copies=None, matchfn=None, **props):
841 842 if self.buffered:
842 843 self.ui.pushbuffer()
843 844 self._show(ctx, copies, matchfn, props)
844 845 self.hunk[ctx.rev()] = self.ui.popbuffer(labeled=True)
845 846 else:
846 847 self._show(ctx, copies, matchfn, props)
847 848
848 849 def _show(self, ctx, copies, matchfn, props):
849 850 '''show a single changeset or file revision'''
850 851 changenode = ctx.node()
851 852 rev = ctx.rev()
852 853
853 854 if self.ui.quiet:
854 855 self.ui.write("%d:%s\n" % (rev, short(changenode)),
855 856 label='log.node')
856 857 return
857 858
858 859 log = self.repo.changelog
859 860 date = util.datestr(ctx.date())
860 861
861 862 hexfunc = self.ui.debugflag and hex or short
862 863
863 864 parents = [(p, hexfunc(log.node(p)))
864 865 for p in self._meaningful_parentrevs(log, rev)]
865 866
866 867 # i18n: column positioning for "hg log"
867 868 self.ui.write(_("changeset: %d:%s\n") % (rev, hexfunc(changenode)),
868 869 label='log.changeset changeset.%s' % ctx.phasestr())
869 870
870 871 branch = ctx.branch()
871 872 # don't show the default branch name
872 873 if branch != 'default':
873 874 # i18n: column positioning for "hg log"
874 875 self.ui.write(_("branch: %s\n") % branch,
875 876 label='log.branch')
876 877 for bookmark in self.repo.nodebookmarks(changenode):
877 878 # i18n: column positioning for "hg log"
878 879 self.ui.write(_("bookmark: %s\n") % bookmark,
879 880 label='log.bookmark')
880 881 for tag in self.repo.nodetags(changenode):
881 882 # i18n: column positioning for "hg log"
882 883 self.ui.write(_("tag: %s\n") % tag,
883 884 label='log.tag')
884 885 if self.ui.debugflag and ctx.phase():
885 886 # i18n: column positioning for "hg log"
886 887 self.ui.write(_("phase: %s\n") % _(ctx.phasestr()),
887 888 label='log.phase')
888 889 for parent in parents:
889 890 # i18n: column positioning for "hg log"
890 891 self.ui.write(_("parent: %d:%s\n") % parent,
891 892 label='log.parent changeset.%s' % ctx.phasestr())
892 893
893 894 if self.ui.debugflag:
894 895 mnode = ctx.manifestnode()
895 896 # i18n: column positioning for "hg log"
896 897 self.ui.write(_("manifest: %d:%s\n") %
897 898 (self.repo.manifest.rev(mnode), hex(mnode)),
898 899 label='ui.debug log.manifest')
899 900 # i18n: column positioning for "hg log"
900 901 self.ui.write(_("user: %s\n") % ctx.user(),
901 902 label='log.user')
902 903 # i18n: column positioning for "hg log"
903 904 self.ui.write(_("date: %s\n") % date,
904 905 label='log.date')
905 906
906 907 if self.ui.debugflag:
907 908 files = self.repo.status(log.parents(changenode)[0], changenode)[:3]
908 909 for key, value in zip([# i18n: column positioning for "hg log"
909 910 _("files:"),
910 911 # i18n: column positioning for "hg log"
911 912 _("files+:"),
912 913 # i18n: column positioning for "hg log"
913 914 _("files-:")], files):
914 915 if value:
915 916 self.ui.write("%-12s %s\n" % (key, " ".join(value)),
916 917 label='ui.debug log.files')
917 918 elif ctx.files() and self.ui.verbose:
918 919 # i18n: column positioning for "hg log"
919 920 self.ui.write(_("files: %s\n") % " ".join(ctx.files()),
920 921 label='ui.note log.files')
921 922 if copies and self.ui.verbose:
922 923 copies = ['%s (%s)' % c for c in copies]
923 924 # i18n: column positioning for "hg log"
924 925 self.ui.write(_("copies: %s\n") % ' '.join(copies),
925 926 label='ui.note log.copies')
926 927
927 928 extra = ctx.extra()
928 929 if extra and self.ui.debugflag:
929 930 for key, value in sorted(extra.items()):
930 931 # i18n: column positioning for "hg log"
931 932 self.ui.write(_("extra: %s=%s\n")
932 933 % (key, value.encode('string_escape')),
933 934 label='ui.debug log.extra')
934 935
935 936 description = ctx.description().strip()
936 937 if description:
937 938 if self.ui.verbose:
938 939 self.ui.write(_("description:\n"),
939 940 label='ui.note log.description')
940 941 self.ui.write(description,
941 942 label='ui.note log.description')
942 943 self.ui.write("\n\n")
943 944 else:
944 945 # i18n: column positioning for "hg log"
945 946 self.ui.write(_("summary: %s\n") %
946 947 description.splitlines()[0],
947 948 label='log.summary')
948 949 self.ui.write("\n")
949 950
950 951 self.showpatch(changenode, matchfn)
951 952
952 953 def showpatch(self, node, matchfn):
953 954 if not matchfn:
954 955 matchfn = self.patch
955 956 if matchfn:
956 957 stat = self.diffopts.get('stat')
957 958 diff = self.diffopts.get('patch')
958 959 diffopts = patch.diffopts(self.ui, self.diffopts)
959 960 prev = self.repo.changelog.parents(node)[0]
960 961 if stat:
961 962 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
962 963 match=matchfn, stat=True)
963 964 if diff:
964 965 if stat:
965 966 self.ui.write("\n")
966 967 diffordiffstat(self.ui, self.repo, diffopts, prev, node,
967 968 match=matchfn, stat=False)
968 969 self.ui.write("\n")
969 970
970 971 def _meaningful_parentrevs(self, log, rev):
971 972 """Return list of meaningful (or all if debug) parentrevs for rev.
972 973
973 974 For merges (two non-nullrev revisions) both parents are meaningful.
974 975 Otherwise the first parent revision is considered meaningful if it
975 976 is not the preceding revision.
976 977 """
977 978 parents = log.parentrevs(rev)
978 979 if not self.ui.debugflag and parents[1] == nullrev:
979 980 if parents[0] >= rev - 1:
980 981 parents = []
981 982 else:
982 983 parents = [parents[0]]
983 984 return parents
984 985
985 986
986 987 class changeset_templater(changeset_printer):
987 988 '''format changeset information.'''
988 989
989 990 def __init__(self, ui, repo, patch, diffopts, tmpl, mapfile, buffered):
990 991 changeset_printer.__init__(self, ui, repo, patch, diffopts, buffered)
991 992 formatnode = ui.debugflag and (lambda x: x) or (lambda x: x[:12])
992 993 defaulttempl = {
993 994 'parent': '{rev}:{node|formatnode} ',
994 995 'manifest': '{rev}:{node|formatnode}',
995 996 'file_copy': '{name} ({source})',
996 997 'extra': '{key}={value|stringescape}'
997 998 }
998 999 # filecopy is preserved for compatibility reasons
999 1000 defaulttempl['filecopy'] = defaulttempl['file_copy']
1000 1001 self.t = templater.templater(mapfile, {'formatnode': formatnode},
1001 1002 cache=defaulttempl)
1002 1003 if tmpl:
1003 1004 self.t.cache['changeset'] = tmpl
1004 1005
1005 1006 self.cache = {}
1006 1007
1007 1008 def _meaningful_parentrevs(self, ctx):
1008 1009 """Return list of meaningful (or all if debug) parentrevs for rev.
1009 1010 """
1010 1011 parents = ctx.parents()
1011 1012 if len(parents) > 1:
1012 1013 return parents
1013 1014 if self.ui.debugflag:
1014 1015 return [parents[0], self.repo['null']]
1015 1016 if parents[0].rev() >= ctx.rev() - 1:
1016 1017 return []
1017 1018 return parents
1018 1019
1019 1020 def _show(self, ctx, copies, matchfn, props):
1020 1021 '''show a single changeset or file revision'''
1021 1022
1022 1023 showlist = templatekw.showlist
1023 1024
1024 1025 # showparents() behaviour depends on ui trace level which
1025 1026 # causes unexpected behaviours at templating level and makes
1026 1027 # it harder to extract it in a standalone function. Its
1027 1028 # behaviour cannot be changed so leave it here for now.
1028 1029 def showparents(**args):
1029 1030 ctx = args['ctx']
1030 1031 parents = [[('rev', p.rev()), ('node', p.hex())]
1031 1032 for p in self._meaningful_parentrevs(ctx)]
1032 1033 return showlist('parent', parents, **args)
1033 1034
1034 1035 props = props.copy()
1035 1036 props.update(templatekw.keywords)
1036 1037 props['parents'] = showparents
1037 1038 props['templ'] = self.t
1038 1039 props['ctx'] = ctx
1039 1040 props['repo'] = self.repo
1040 1041 props['revcache'] = {'copies': copies}
1041 1042 props['cache'] = self.cache
1042 1043
1043 1044 # find correct templates for current mode
1044 1045
1045 1046 tmplmodes = [
1046 1047 (True, None),
1047 1048 (self.ui.verbose, 'verbose'),
1048 1049 (self.ui.quiet, 'quiet'),
1049 1050 (self.ui.debugflag, 'debug'),
1050 1051 ]
1051 1052
1052 1053 types = {'header': '', 'footer':'', 'changeset': 'changeset'}
1053 1054 for mode, postfix in tmplmodes:
1054 1055 for type in types:
1055 1056 cur = postfix and ('%s_%s' % (type, postfix)) or type
1056 1057 if mode and cur in self.t:
1057 1058 types[type] = cur
1058 1059
1059 1060 try:
1060 1061
1061 1062 # write header
1062 1063 if types['header']:
1063 1064 h = templater.stringify(self.t(types['header'], **props))
1064 1065 if self.buffered:
1065 1066 self.header[ctx.rev()] = h
1066 1067 else:
1067 1068 if self.lastheader != h:
1068 1069 self.lastheader = h
1069 1070 self.ui.write(h)
1070 1071
1071 1072 # write changeset metadata, then patch if requested
1072 1073 key = types['changeset']
1073 1074 self.ui.write(templater.stringify(self.t(key, **props)))
1074 1075 self.showpatch(ctx.node(), matchfn)
1075 1076
1076 1077 if types['footer']:
1077 1078 if not self.footer:
1078 1079 self.footer = templater.stringify(self.t(types['footer'],
1079 1080 **props))
1080 1081
1081 1082 except KeyError, inst:
1082 1083 msg = _("%s: no key named '%s'")
1083 1084 raise util.Abort(msg % (self.t.mapfile, inst.args[0]))
1084 1085 except SyntaxError, inst:
1085 1086 raise util.Abort('%s: %s' % (self.t.mapfile, inst.args[0]))
1086 1087
1087 1088 def gettemplate(ui, tmpl, style):
1088 1089 """
1089 1090 Find the template matching the given template spec or style.
1090 1091 """
1091 1092
1092 1093 # ui settings
1093 1094 if not tmpl and not style:
1094 1095 tmpl = ui.config('ui', 'logtemplate')
1095 1096 if tmpl:
1096 1097 try:
1097 1098 tmpl = templater.parsestring(tmpl)
1098 1099 except SyntaxError:
1099 1100 tmpl = templater.parsestring(tmpl, quoted=False)
1100 1101 return tmpl, None
1101 1102 else:
1102 1103 style = util.expandpath(ui.config('ui', 'style', ''))
1103 1104
1104 1105 if style:
1105 1106 mapfile = style
1106 1107 if not os.path.split(mapfile)[0]:
1107 1108 mapname = (templater.templatepath('map-cmdline.' + mapfile)
1108 1109 or templater.templatepath(mapfile))
1109 1110 if mapname:
1110 1111 mapfile = mapname
1111 1112 return None, mapfile
1112 1113
1113 1114 if not tmpl:
1114 1115 return None, None
1115 1116
1116 1117 # looks like a literal template?
1117 1118 if '{' in tmpl:
1118 1119 return tmpl, None
1119 1120
1120 1121 # perhaps a stock style?
1121 1122 if not os.path.split(tmpl)[0]:
1122 1123 mapname = (templater.templatepath('map-cmdline.' + tmpl)
1123 1124 or templater.templatepath(tmpl))
1124 1125 if mapname and os.path.isfile(mapname):
1125 1126 return None, mapname
1126 1127
1127 1128 # perhaps it's a reference to [templates]
1128 1129 t = ui.config('templates', tmpl)
1129 1130 if t:
1130 1131 try:
1131 1132 tmpl = templater.parsestring(t)
1132 1133 except SyntaxError:
1133 1134 tmpl = templater.parsestring(t, quoted=False)
1134 1135 return tmpl, None
1135 1136
1136 1137 if tmpl == 'list':
1137 1138 ui.write(_("available styles: %s\n") % templater.stylelist())
1138 1139 raise util.Abort(_("specify a template"))
1139 1140
1140 1141 # perhaps it's a path to a map or a template
1141 1142 if ('/' in tmpl or '\\' in tmpl) and os.path.isfile(tmpl):
1142 1143 # is it a mapfile for a style?
1143 1144 if os.path.basename(tmpl).startswith("map-"):
1144 1145 return None, os.path.realpath(tmpl)
1145 1146 tmpl = open(tmpl).read()
1146 1147 return tmpl, None
1147 1148
1148 1149 # constant string?
1149 1150 return tmpl, None
1150 1151
1151 1152 def show_changeset(ui, repo, opts, buffered=False):
1152 1153 """show one changeset using template or regular display.
1153 1154
1154 1155 Display format will be the first non-empty hit of:
1155 1156 1. option 'template'
1156 1157 2. option 'style'
1157 1158 3. [ui] setting 'logtemplate'
1158 1159 4. [ui] setting 'style'
1159 1160 If all of these values are either the unset or the empty string,
1160 1161 regular display via changeset_printer() is done.
1161 1162 """
1162 1163 # options
1163 1164 patch = None
1164 1165 if opts.get('patch') or opts.get('stat'):
1165 1166 patch = scmutil.matchall(repo)
1166 1167
1167 1168 tmpl, mapfile = gettemplate(ui, opts.get('template'), opts.get('style'))
1168 1169
1169 1170 if not tmpl and not mapfile:
1170 1171 return changeset_printer(ui, repo, patch, opts, buffered)
1171 1172
1172 1173 try:
1173 1174 t = changeset_templater(ui, repo, patch, opts, tmpl, mapfile, buffered)
1174 1175 except SyntaxError, inst:
1175 1176 raise util.Abort(inst.args[0])
1176 1177 return t
1177 1178
1178 1179 def showmarker(ui, marker):
1179 1180 """utility function to display obsolescence marker in a readable way
1180 1181
1181 1182 To be used by debug function."""
1182 1183 ui.write(hex(marker.precnode()))
1183 1184 for repl in marker.succnodes():
1184 1185 ui.write(' ')
1185 1186 ui.write(hex(repl))
1186 1187 ui.write(' %X ' % marker._data[2])
1187 1188 ui.write('{%s}' % (', '.join('%r: %r' % t for t in
1188 1189 sorted(marker.metadata().items()))))
1189 1190 ui.write('\n')
1190 1191
1191 1192 def finddate(ui, repo, date):
1192 1193 """Find the tipmost changeset that matches the given date spec"""
1193 1194
1194 1195 df = util.matchdate(date)
1195 1196 m = scmutil.matchall(repo)
1196 1197 results = {}
1197 1198
1198 1199 def prep(ctx, fns):
1199 1200 d = ctx.date()
1200 1201 if df(d[0]):
1201 1202 results[ctx.rev()] = d
1202 1203
1203 1204 for ctx in walkchangerevs(repo, m, {'rev': None}, prep):
1204 1205 rev = ctx.rev()
1205 1206 if rev in results:
1206 1207 ui.status(_("found revision %s from %s\n") %
1207 1208 (rev, util.datestr(results[rev])))
1208 1209 return str(rev)
1209 1210
1210 1211 raise util.Abort(_("revision matching date not found"))
1211 1212
1212 1213 def increasingwindows(windowsize=8, sizelimit=512):
1213 1214 while True:
1214 1215 yield windowsize
1215 1216 if windowsize < sizelimit:
1216 1217 windowsize *= 2
1217 1218
1218 1219 class FileWalkError(Exception):
1219 1220 pass
1220 1221
1221 1222 def walkfilerevs(repo, match, follow, revs, fncache):
1222 1223 '''Walks the file history for the matched files.
1223 1224
1224 1225 Returns the changeset revs that are involved in the file history.
1225 1226
1226 1227 Throws FileWalkError if the file history can't be walked using
1227 1228 filelogs alone.
1228 1229 '''
1229 1230 wanted = set()
1230 1231 copies = []
1231 1232 minrev, maxrev = min(revs), max(revs)
1232 1233 def filerevgen(filelog, last):
1233 1234 """
1234 1235 Only files, no patterns. Check the history of each file.
1235 1236
1236 1237 Examines filelog entries within minrev, maxrev linkrev range
1237 1238 Returns an iterator yielding (linkrev, parentlinkrevs, copied)
1238 1239 tuples in backwards order
1239 1240 """
1240 1241 cl_count = len(repo)
1241 1242 revs = []
1242 1243 for j in xrange(0, last + 1):
1243 1244 linkrev = filelog.linkrev(j)
1244 1245 if linkrev < minrev:
1245 1246 continue
1246 1247 # only yield rev for which we have the changelog, it can
1247 1248 # happen while doing "hg log" during a pull or commit
1248 1249 if linkrev >= cl_count:
1249 1250 break
1250 1251
1251 1252 parentlinkrevs = []
1252 1253 for p in filelog.parentrevs(j):
1253 1254 if p != nullrev:
1254 1255 parentlinkrevs.append(filelog.linkrev(p))
1255 1256 n = filelog.node(j)
1256 1257 revs.append((linkrev, parentlinkrevs,
1257 1258 follow and filelog.renamed(n)))
1258 1259
1259 1260 return reversed(revs)
1260 1261 def iterfiles():
1261 1262 pctx = repo['.']
1262 1263 for filename in match.files():
1263 1264 if follow:
1264 1265 if filename not in pctx:
1265 1266 raise util.Abort(_('cannot follow file not in parent '
1266 1267 'revision: "%s"') % filename)
1267 1268 yield filename, pctx[filename].filenode()
1268 1269 else:
1269 1270 yield filename, None
1270 1271 for filename_node in copies:
1271 1272 yield filename_node
1272 1273
1273 1274 for file_, node in iterfiles():
1274 1275 filelog = repo.file(file_)
1275 1276 if not len(filelog):
1276 1277 if node is None:
1277 1278 # A zero count may be a directory or deleted file, so
1278 1279 # try to find matching entries on the slow path.
1279 1280 if follow:
1280 1281 raise util.Abort(
1281 1282 _('cannot follow nonexistent file: "%s"') % file_)
1282 1283 raise FileWalkError("Cannot walk via filelog")
1283 1284 else:
1284 1285 continue
1285 1286
1286 1287 if node is None:
1287 1288 last = len(filelog) - 1
1288 1289 else:
1289 1290 last = filelog.rev(node)
1290 1291
1291 1292
1292 1293 # keep track of all ancestors of the file
1293 1294 ancestors = set([filelog.linkrev(last)])
1294 1295
1295 1296 # iterate from latest to oldest revision
1296 1297 for rev, flparentlinkrevs, copied in filerevgen(filelog, last):
1297 1298 if not follow:
1298 1299 if rev > maxrev:
1299 1300 continue
1300 1301 else:
1301 1302 # Note that last might not be the first interesting
1302 1303 # rev to us:
1303 1304 # if the file has been changed after maxrev, we'll
1304 1305 # have linkrev(last) > maxrev, and we still need
1305 1306 # to explore the file graph
1306 1307 if rev not in ancestors:
1307 1308 continue
1308 1309 # XXX insert 1327 fix here
1309 1310 if flparentlinkrevs:
1310 1311 ancestors.update(flparentlinkrevs)
1311 1312
1312 1313 fncache.setdefault(rev, []).append(file_)
1313 1314 wanted.add(rev)
1314 1315 if copied:
1315 1316 copies.append(copied)
1316 1317
1317 1318 return wanted
1318 1319
1319 1320 def walkchangerevs(repo, match, opts, prepare):
1320 1321 '''Iterate over files and the revs in which they changed.
1321 1322
1322 1323 Callers most commonly need to iterate backwards over the history
1323 1324 in which they are interested. Doing so has awful (quadratic-looking)
1324 1325 performance, so we use iterators in a "windowed" way.
1325 1326
1326 1327 We walk a window of revisions in the desired order. Within the
1327 1328 window, we first walk forwards to gather data, then in the desired
1328 1329 order (usually backwards) to display it.
1329 1330
1330 1331 This function returns an iterator yielding contexts. Before
1331 1332 yielding each context, the iterator will first call the prepare
1332 1333 function on each context in the window in forward order.'''
1333 1334
1334 1335 follow = opts.get('follow') or opts.get('follow_first')
1335 1336
1336 1337 if opts.get('rev'):
1337 1338 revs = scmutil.revrange(repo, opts.get('rev'))
1338 1339 elif follow:
1339 1340 revs = repo.revs('reverse(:.)')
1340 1341 else:
1341 1342 revs = revset.spanset(repo)
1342 1343 revs.reverse()
1343 1344 if not revs:
1344 1345 return []
1345 1346 wanted = set()
1346 1347 slowpath = match.anypats() or (match.files() and opts.get('removed'))
1347 1348 fncache = {}
1348 1349 change = repo.changectx
1349 1350
1350 1351 # First step is to fill wanted, the set of revisions that we want to yield.
1351 1352 # When it does not induce extra cost, we also fill fncache for revisions in
1352 1353 # wanted: a cache of filenames that were changed (ctx.files()) and that
1353 1354 # match the file filtering conditions.
1354 1355
1355 1356 if not slowpath and not match.files():
1356 1357 # No files, no patterns. Display all revs.
1357 1358 wanted = revs
1358 1359
1359 1360 if not slowpath and match.files():
1360 1361 # We only have to read through the filelog to find wanted revisions
1361 1362
1362 1363 try:
1363 1364 wanted = walkfilerevs(repo, match, follow, revs, fncache)
1364 1365 except FileWalkError:
1365 1366 slowpath = True
1366 1367
1367 1368 # We decided to fall back to the slowpath because at least one
1368 1369 # of the paths was not a file. Check to see if at least one of them
1369 1370 # existed in history, otherwise simply return
1370 1371 for path in match.files():
1371 1372 if path == '.' or path in repo.store:
1372 1373 break
1373 1374 else:
1374 1375 return []
1375 1376
1376 1377 if slowpath:
1377 1378 # We have to read the changelog to match filenames against
1378 1379 # changed files
1379 1380
1380 1381 if follow:
1381 1382 raise util.Abort(_('can only follow copies/renames for explicit '
1382 1383 'filenames'))
1383 1384
1384 1385 # The slow path checks files modified in every changeset.
1385 1386 # This is really slow on large repos, so compute the set lazily.
1386 1387 class lazywantedset(object):
1387 1388 def __init__(self):
1388 1389 self.set = set()
1389 1390 self.revs = set(revs)
1390 1391
1391 1392 # No need to worry about locality here because it will be accessed
1392 1393 # in the same order as the increasing window below.
1393 1394 def __contains__(self, value):
1394 1395 if value in self.set:
1395 1396 return True
1396 1397 elif not value in self.revs:
1397 1398 return False
1398 1399 else:
1399 1400 self.revs.discard(value)
1400 1401 ctx = change(value)
1401 1402 matches = filter(match, ctx.files())
1402 1403 if matches:
1403 1404 fncache[value] = matches
1404 1405 self.set.add(value)
1405 1406 return True
1406 1407 return False
1407 1408
1408 1409 def discard(self, value):
1409 1410 self.revs.discard(value)
1410 1411 self.set.discard(value)
1411 1412
1412 1413 wanted = lazywantedset()
1413 1414
1414 1415 class followfilter(object):
1415 1416 def __init__(self, onlyfirst=False):
1416 1417 self.startrev = nullrev
1417 1418 self.roots = set()
1418 1419 self.onlyfirst = onlyfirst
1419 1420
1420 1421 def match(self, rev):
1421 1422 def realparents(rev):
1422 1423 if self.onlyfirst:
1423 1424 return repo.changelog.parentrevs(rev)[0:1]
1424 1425 else:
1425 1426 return filter(lambda x: x != nullrev,
1426 1427 repo.changelog.parentrevs(rev))
1427 1428
1428 1429 if self.startrev == nullrev:
1429 1430 self.startrev = rev
1430 1431 return True
1431 1432
1432 1433 if rev > self.startrev:
1433 1434 # forward: all descendants
1434 1435 if not self.roots:
1435 1436 self.roots.add(self.startrev)
1436 1437 for parent in realparents(rev):
1437 1438 if parent in self.roots:
1438 1439 self.roots.add(rev)
1439 1440 return True
1440 1441 else:
1441 1442 # backwards: all parents
1442 1443 if not self.roots:
1443 1444 self.roots.update(realparents(self.startrev))
1444 1445 if rev in self.roots:
1445 1446 self.roots.remove(rev)
1446 1447 self.roots.update(realparents(rev))
1447 1448 return True
1448 1449
1449 1450 return False
1450 1451
1451 1452 # it might be worthwhile to do this in the iterator if the rev range
1452 1453 # is descending and the prune args are all within that range
1453 1454 for rev in opts.get('prune', ()):
1454 1455 rev = repo[rev].rev()
1455 1456 ff = followfilter()
1456 1457 stop = min(revs[0], revs[-1])
1457 1458 for x in xrange(rev, stop - 1, -1):
1458 1459 if ff.match(x):
1459 1460 wanted = wanted - [x]
1460 1461
1461 1462 # Now that wanted is correctly initialized, we can iterate over the
1462 1463 # revision range, yielding only revisions in wanted.
1463 1464 def iterate():
1464 1465 if follow and not match.files():
1465 1466 ff = followfilter(onlyfirst=opts.get('follow_first'))
1466 1467 def want(rev):
1467 1468 return ff.match(rev) and rev in wanted
1468 1469 else:
1469 1470 def want(rev):
1470 1471 return rev in wanted
1471 1472
1472 1473 it = iter(revs)
1473 1474 stopiteration = False
1474 1475 for windowsize in increasingwindows():
1475 1476 nrevs = []
1476 1477 for i in xrange(windowsize):
1477 1478 try:
1478 1479 rev = it.next()
1479 1480 if want(rev):
1480 1481 nrevs.append(rev)
1481 1482 except (StopIteration):
1482 1483 stopiteration = True
1483 1484 break
1484 1485 for rev in sorted(nrevs):
1485 1486 fns = fncache.get(rev)
1486 1487 ctx = change(rev)
1487 1488 if not fns:
1488 1489 def fns_generator():
1489 1490 for f in ctx.files():
1490 1491 if match(f):
1491 1492 yield f
1492 1493 fns = fns_generator()
1493 1494 prepare(ctx, fns)
1494 1495 for rev in nrevs:
1495 1496 yield change(rev)
1496 1497
1497 1498 if stopiteration:
1498 1499 break
1499 1500
1500 1501 return iterate()
1501 1502
1502 1503 def _makelogfilematcher(repo, files, followfirst):
1503 1504 # When displaying a revision with --patch --follow FILE, we have
1504 1505 # to know which file of the revision must be diffed. With
1505 1506 # --follow, we want the names of the ancestors of FILE in the
1506 1507 # revision, stored in "fcache". "fcache" is populated by
1507 1508 # reproducing the graph traversal already done by --follow revset
1508 1509 # and relating linkrevs to file names (which is not "correct" but
1509 1510 # good enough).
1510 1511 fcache = {}
1511 1512 fcacheready = [False]
1512 1513 pctx = repo['.']
1513 1514
1514 1515 def populate():
1515 1516 for fn in files:
1516 1517 for i in ((pctx[fn],), pctx[fn].ancestors(followfirst=followfirst)):
1517 1518 for c in i:
1518 1519 fcache.setdefault(c.linkrev(), set()).add(c.path())
1519 1520
1520 1521 def filematcher(rev):
1521 1522 if not fcacheready[0]:
1522 1523 # Lazy initialization
1523 1524 fcacheready[0] = True
1524 1525 populate()
1525 1526 return scmutil.matchfiles(repo, fcache.get(rev, []))
1526 1527
1527 1528 return filematcher
1528 1529
1529 1530 def _makelogrevset(repo, pats, opts, revs):
1530 1531 """Return (expr, filematcher) where expr is a revset string built
1531 1532 from log options and file patterns or None. If --stat or --patch
1532 1533 are not passed filematcher is None. Otherwise it is a callable
1533 1534 taking a revision number and returning a match objects filtering
1534 1535 the files to be detailed when displaying the revision.
1535 1536 """
1536 1537 opt2revset = {
1537 1538 'no_merges': ('not merge()', None),
1538 1539 'only_merges': ('merge()', None),
1539 1540 '_ancestors': ('ancestors(%(val)s)', None),
1540 1541 '_fancestors': ('_firstancestors(%(val)s)', None),
1541 1542 '_descendants': ('descendants(%(val)s)', None),
1542 1543 '_fdescendants': ('_firstdescendants(%(val)s)', None),
1543 1544 '_matchfiles': ('_matchfiles(%(val)s)', None),
1544 1545 'date': ('date(%(val)r)', None),
1545 1546 'branch': ('branch(%(val)r)', ' or '),
1546 1547 '_patslog': ('filelog(%(val)r)', ' or '),
1547 1548 '_patsfollow': ('follow(%(val)r)', ' or '),
1548 1549 '_patsfollowfirst': ('_followfirst(%(val)r)', ' or '),
1549 1550 'keyword': ('keyword(%(val)r)', ' or '),
1550 1551 'prune': ('not (%(val)r or ancestors(%(val)r))', ' and '),
1551 1552 'user': ('user(%(val)r)', ' or '),
1552 1553 }
1553 1554
1554 1555 opts = dict(opts)
1555 1556 # follow or not follow?
1556 1557 follow = opts.get('follow') or opts.get('follow_first')
1557 1558 followfirst = opts.get('follow_first') and 1 or 0
1558 1559 # --follow with FILE behaviour depends on revs...
1559 1560 it = iter(revs)
1560 1561 startrev = it.next()
1561 1562 try:
1562 1563 followdescendants = startrev < it.next()
1563 1564 except (StopIteration):
1564 1565 followdescendants = False
1565 1566
1566 1567 # branch and only_branch are really aliases and must be handled at
1567 1568 # the same time
1568 1569 opts['branch'] = opts.get('branch', []) + opts.get('only_branch', [])
1569 1570 opts['branch'] = [repo.lookupbranch(b) for b in opts['branch']]
1570 1571 # pats/include/exclude are passed to match.match() directly in
1571 1572 # _matchfiles() revset but walkchangerevs() builds its matcher with
1572 1573 # scmutil.match(). The difference is input pats are globbed on
1573 1574 # platforms without shell expansion (windows).
1574 1575 pctx = repo[None]
1575 1576 match, pats = scmutil.matchandpats(pctx, pats, opts)
1576 1577 slowpath = match.anypats() or (match.files() and opts.get('removed'))
1577 1578 if not slowpath:
1578 1579 for f in match.files():
1579 1580 if follow and f not in pctx:
1580 1581 # If the file exists, it may be a directory, so let it
1581 1582 # take the slow path.
1582 1583 if os.path.exists(repo.wjoin(f)):
1583 1584 slowpath = True
1584 1585 continue
1585 1586 else:
1586 1587 raise util.Abort(_('cannot follow file not in parent '
1587 1588 'revision: "%s"') % f)
1588 1589 filelog = repo.file(f)
1589 1590 if not filelog:
1590 1591 # A zero count may be a directory or deleted file, so
1591 1592 # try to find matching entries on the slow path.
1592 1593 if follow:
1593 1594 raise util.Abort(
1594 1595 _('cannot follow nonexistent file: "%s"') % f)
1595 1596 slowpath = True
1596 1597
1597 1598 # We decided to fall back to the slowpath because at least one
1598 1599 # of the paths was not a file. Check to see if at least one of them
1599 1600 # existed in history - in that case, we'll continue down the
1600 1601 # slowpath; otherwise, we can turn off the slowpath
1601 1602 if slowpath:
1602 1603 for path in match.files():
1603 1604 if path == '.' or path in repo.store:
1604 1605 break
1605 1606 else:
1606 1607 slowpath = False
1607 1608
1608 1609 if slowpath:
1609 1610 # See walkchangerevs() slow path.
1610 1611 #
1611 1612 # pats/include/exclude cannot be represented as separate
1612 1613 # revset expressions as their filtering logic applies at file
1613 1614 # level. For instance "-I a -X a" matches a revision touching
1614 1615 # "a" and "b" while "file(a) and not file(b)" does
1615 1616 # not. Besides, filesets are evaluated against the working
1616 1617 # directory.
1617 1618 matchargs = ['r:', 'd:relpath']
1618 1619 for p in pats:
1619 1620 matchargs.append('p:' + p)
1620 1621 for p in opts.get('include', []):
1621 1622 matchargs.append('i:' + p)
1622 1623 for p in opts.get('exclude', []):
1623 1624 matchargs.append('x:' + p)
1624 1625 matchargs = ','.join(('%r' % p) for p in matchargs)
1625 1626 opts['_matchfiles'] = matchargs
1626 1627 else:
1627 1628 if follow:
1628 1629 fpats = ('_patsfollow', '_patsfollowfirst')
1629 1630 fnopats = (('_ancestors', '_fancestors'),
1630 1631 ('_descendants', '_fdescendants'))
1631 1632 if pats:
1632 1633 # follow() revset interprets its file argument as a
1633 1634 # manifest entry, so use match.files(), not pats.
1634 1635 opts[fpats[followfirst]] = list(match.files())
1635 1636 else:
1636 1637 opts[fnopats[followdescendants][followfirst]] = str(startrev)
1637 1638 else:
1638 1639 opts['_patslog'] = list(pats)
1639 1640
1640 1641 filematcher = None
1641 1642 if opts.get('patch') or opts.get('stat'):
1642 1643 # When following files, track renames via a special matcher.
1643 1644 # If we're forced to take the slowpath it means we're following
1644 1645 # at least one pattern/directory, so don't bother with rename tracking.
1645 1646 if follow and not match.always() and not slowpath:
1646 1647 # _makelogfilematcher expects its files argument to be relative to
1647 1648 # the repo root, so use match.files(), not pats.
1648 1649 filematcher = _makelogfilematcher(repo, match.files(), followfirst)
1649 1650 else:
1650 1651 filematcher = lambda rev: match
1651 1652
1652 1653 expr = []
1653 1654 for op, val in opts.iteritems():
1654 1655 if not val:
1655 1656 continue
1656 1657 if op not in opt2revset:
1657 1658 continue
1658 1659 revop, andor = opt2revset[op]
1659 1660 if '%(val)' not in revop:
1660 1661 expr.append(revop)
1661 1662 else:
1662 1663 if not isinstance(val, list):
1663 1664 e = revop % {'val': val}
1664 1665 else:
1665 1666 e = '(' + andor.join((revop % {'val': v}) for v in val) + ')'
1666 1667 expr.append(e)
1667 1668
1668 1669 if expr:
1669 1670 expr = '(' + ' and '.join(expr) + ')'
1670 1671 else:
1671 1672 expr = None
1672 1673 return expr, filematcher
1673 1674
1674 1675 def getgraphlogrevs(repo, pats, opts):
1675 1676 """Return (revs, expr, filematcher) where revs is an iterable of
1676 1677 revision numbers, expr is a revset string built from log options
1677 1678 and file patterns or None, and used to filter 'revs'. If --stat or
1678 1679 --patch are not passed filematcher is None. Otherwise it is a
1679 1680 callable taking a revision number and returning a match objects
1680 1681 filtering the files to be detailed when displaying the revision.
1681 1682 """
1682 1683 if not len(repo):
1683 1684 return [], None, None
1684 1685 limit = loglimit(opts)
1685 1686 # Default --rev value depends on --follow but --follow behaviour
1686 1687 # depends on revisions resolved from --rev...
1687 1688 follow = opts.get('follow') or opts.get('follow_first')
1688 1689 possiblyunsorted = False # whether revs might need sorting
1689 1690 if opts.get('rev'):
1690 1691 revs = scmutil.revrange(repo, opts['rev'])
1691 1692 # Don't sort here because _makelogrevset might depend on the
1692 1693 # order of revs
1693 1694 possiblyunsorted = True
1694 1695 else:
1695 1696 if follow and len(repo) > 0:
1696 1697 revs = repo.revs('reverse(:.)')
1697 1698 else:
1698 1699 revs = revset.spanset(repo)
1699 1700 revs.reverse()
1700 1701 if not revs:
1701 1702 return revset.baseset(), None, None
1702 1703 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
1703 1704 if possiblyunsorted:
1704 1705 revs.sort(reverse=True)
1705 1706 if expr:
1706 1707 # Revset matchers often operate faster on revisions in changelog
1707 1708 # order, because most filters deal with the changelog.
1708 1709 revs.reverse()
1709 1710 matcher = revset.match(repo.ui, expr)
1710 1711 # Revset matches can reorder revisions. "A or B" typically returns
1711 1712 # returns the revision matching A then the revision matching B. Sort
1712 1713 # again to fix that.
1713 1714 revs = matcher(repo, revs)
1714 1715 revs.sort(reverse=True)
1715 1716 if limit is not None:
1716 1717 limitedrevs = revset.baseset()
1717 1718 for idx, rev in enumerate(revs):
1718 1719 if idx >= limit:
1719 1720 break
1720 1721 limitedrevs.append(rev)
1721 1722 revs = limitedrevs
1722 1723
1723 1724 return revs, expr, filematcher
1724 1725
1725 1726 def getlogrevs(repo, pats, opts):
1726 1727 """Return (revs, expr, filematcher) where revs is an iterable of
1727 1728 revision numbers, expr is a revset string built from log options
1728 1729 and file patterns or None, and used to filter 'revs'. If --stat or
1729 1730 --patch are not passed filematcher is None. Otherwise it is a
1730 1731 callable taking a revision number and returning a match objects
1731 1732 filtering the files to be detailed when displaying the revision.
1732 1733 """
1733 1734 limit = loglimit(opts)
1734 1735 # Default --rev value depends on --follow but --follow behaviour
1735 1736 # depends on revisions resolved from --rev...
1736 1737 follow = opts.get('follow') or opts.get('follow_first')
1737 1738 if opts.get('rev'):
1738 1739 revs = scmutil.revrange(repo, opts['rev'])
1739 1740 elif follow:
1740 1741 revs = repo.revs('reverse(:.)')
1741 1742 else:
1742 1743 revs = revset.spanset(repo)
1743 1744 revs.reverse()
1744 1745 if not revs:
1745 1746 return revset.baseset([]), None, None
1746 1747 expr, filematcher = _makelogrevset(repo, pats, opts, revs)
1747 1748 if expr:
1748 1749 # Revset matchers often operate faster on revisions in changelog
1749 1750 # order, because most filters deal with the changelog.
1750 1751 if not opts.get('rev'):
1751 1752 revs.reverse()
1752 1753 matcher = revset.match(repo.ui, expr)
1753 1754 # Revset matches can reorder revisions. "A or B" typically returns
1754 1755 # returns the revision matching A then the revision matching B. Sort
1755 1756 # again to fix that.
1756 1757 revs = matcher(repo, revs)
1757 1758 if not opts.get('rev'):
1758 1759 revs.sort(reverse=True)
1759 1760 if limit is not None:
1760 1761 count = 0
1761 1762 limitedrevs = revset.baseset([])
1762 1763 it = iter(revs)
1763 1764 while count < limit:
1764 1765 try:
1765 1766 limitedrevs.append(it.next())
1766 1767 except (StopIteration):
1767 1768 break
1768 1769 count += 1
1769 1770 revs = limitedrevs
1770 1771
1771 1772 return revs, expr, filematcher
1772 1773
1773 1774 def displaygraph(ui, dag, displayer, showparents, edgefn, getrenamed=None,
1774 1775 filematcher=None):
1775 1776 seen, state = [], graphmod.asciistate()
1776 1777 for rev, type, ctx, parents in dag:
1777 1778 char = 'o'
1778 1779 if ctx.node() in showparents:
1779 1780 char = '@'
1780 1781 elif ctx.obsolete():
1781 1782 char = 'x'
1782 1783 copies = None
1783 1784 if getrenamed and ctx.rev():
1784 1785 copies = []
1785 1786 for fn in ctx.files():
1786 1787 rename = getrenamed(fn, ctx.rev())
1787 1788 if rename:
1788 1789 copies.append((fn, rename[0]))
1789 1790 revmatchfn = None
1790 1791 if filematcher is not None:
1791 1792 revmatchfn = filematcher(ctx.rev())
1792 1793 displayer.show(ctx, copies=copies, matchfn=revmatchfn)
1793 1794 lines = displayer.hunk.pop(rev).split('\n')
1794 1795 if not lines[-1]:
1795 1796 del lines[-1]
1796 1797 displayer.flush(rev)
1797 1798 edges = edgefn(type, char, lines, seen, rev, parents)
1798 1799 for type, char, lines, coldata in edges:
1799 1800 graphmod.ascii(ui, state, type, char, lines, coldata)
1800 1801 displayer.close()
1801 1802
1802 1803 def graphlog(ui, repo, *pats, **opts):
1803 1804 # Parameters are identical to log command ones
1804 1805 revs, expr, filematcher = getgraphlogrevs(repo, pats, opts)
1805 1806 revdag = graphmod.dagwalker(repo, revs)
1806 1807
1807 1808 getrenamed = None
1808 1809 if opts.get('copies'):
1809 1810 endrev = None
1810 1811 if opts.get('rev'):
1811 1812 endrev = scmutil.revrange(repo, opts.get('rev')).max() + 1
1812 1813 getrenamed = templatekw.getrenamedfn(repo, endrev=endrev)
1813 1814 displayer = show_changeset(ui, repo, opts, buffered=True)
1814 1815 showparents = [ctx.node() for ctx in repo[None].parents()]
1815 1816 displaygraph(ui, revdag, displayer, showparents,
1816 1817 graphmod.asciiedges, getrenamed, filematcher)
1817 1818
1818 1819 def checkunsupportedgraphflags(pats, opts):
1819 1820 for op in ["newest_first"]:
1820 1821 if op in opts and opts[op]:
1821 1822 raise util.Abort(_("-G/--graph option is incompatible with --%s")
1822 1823 % op.replace("_", "-"))
1823 1824
1824 1825 def graphrevs(repo, nodes, opts):
1825 1826 limit = loglimit(opts)
1826 1827 nodes.reverse()
1827 1828 if limit is not None:
1828 1829 nodes = nodes[:limit]
1829 1830 return graphmod.nodes(repo, nodes)
1830 1831
1831 1832 def add(ui, repo, match, dryrun, listsubrepos, prefix, explicitonly):
1832 1833 join = lambda f: os.path.join(prefix, f)
1833 1834 bad = []
1834 1835 oldbad = match.bad
1835 1836 match.bad = lambda x, y: bad.append(x) or oldbad(x, y)
1836 1837 names = []
1837 1838 wctx = repo[None]
1838 1839 cca = None
1839 1840 abort, warn = scmutil.checkportabilityalert(ui)
1840 1841 if abort or warn:
1841 1842 cca = scmutil.casecollisionauditor(ui, abort, repo.dirstate)
1842 1843 for f in repo.walk(match):
1843 1844 exact = match.exact(f)
1844 1845 if exact or not explicitonly and f not in repo.dirstate:
1845 1846 if cca:
1846 1847 cca(f)
1847 1848 names.append(f)
1848 1849 if ui.verbose or not exact:
1849 1850 ui.status(_('adding %s\n') % match.rel(join(f)))
1850 1851
1851 1852 for subpath in sorted(wctx.substate):
1852 1853 sub = wctx.sub(subpath)
1853 1854 try:
1854 1855 submatch = matchmod.narrowmatcher(subpath, match)
1855 1856 if listsubrepos:
1856 1857 bad.extend(sub.add(ui, submatch, dryrun, listsubrepos, prefix,
1857 1858 False))
1858 1859 else:
1859 1860 bad.extend(sub.add(ui, submatch, dryrun, listsubrepos, prefix,
1860 1861 True))
1861 1862 except error.LookupError:
1862 1863 ui.status(_("skipping missing subrepository: %s\n")
1863 1864 % join(subpath))
1864 1865
1865 1866 if not dryrun:
1866 1867 rejected = wctx.add(names, prefix)
1867 1868 bad.extend(f for f in rejected if f in match.files())
1868 1869 return bad
1869 1870
1870 1871 def forget(ui, repo, match, prefix, explicitonly):
1871 1872 join = lambda f: os.path.join(prefix, f)
1872 1873 bad = []
1873 1874 oldbad = match.bad
1874 1875 match.bad = lambda x, y: bad.append(x) or oldbad(x, y)
1875 1876 wctx = repo[None]
1876 1877 forgot = []
1877 1878 s = repo.status(match=match, clean=True)
1878 1879 forget = sorted(s[0] + s[1] + s[3] + s[6])
1879 1880 if explicitonly:
1880 1881 forget = [f for f in forget if match.exact(f)]
1881 1882
1882 1883 for subpath in sorted(wctx.substate):
1883 1884 sub = wctx.sub(subpath)
1884 1885 try:
1885 1886 submatch = matchmod.narrowmatcher(subpath, match)
1886 1887 subbad, subforgot = sub.forget(ui, submatch, prefix)
1887 1888 bad.extend([subpath + '/' + f for f in subbad])
1888 1889 forgot.extend([subpath + '/' + f for f in subforgot])
1889 1890 except error.LookupError:
1890 1891 ui.status(_("skipping missing subrepository: %s\n")
1891 1892 % join(subpath))
1892 1893
1893 1894 if not explicitonly:
1894 1895 for f in match.files():
1895 1896 if f not in repo.dirstate and not os.path.isdir(match.rel(join(f))):
1896 1897 if f not in forgot:
1897 1898 if os.path.exists(match.rel(join(f))):
1898 1899 ui.warn(_('not removing %s: '
1899 1900 'file is already untracked\n')
1900 1901 % match.rel(join(f)))
1901 1902 bad.append(f)
1902 1903
1903 1904 for f in forget:
1904 1905 if ui.verbose or not match.exact(f):
1905 1906 ui.status(_('removing %s\n') % match.rel(join(f)))
1906 1907
1907 1908 rejected = wctx.forget(forget, prefix)
1908 1909 bad.extend(f for f in rejected if f in match.files())
1909 1910 forgot.extend(forget)
1910 1911 return bad, forgot
1911 1912
1912 1913 def cat(ui, repo, ctx, matcher, prefix, **opts):
1913 1914 err = 1
1914 1915
1915 1916 def write(path):
1916 1917 fp = makefileobj(repo, opts.get('output'), ctx.node(),
1917 1918 pathname=os.path.join(prefix, path))
1918 1919 data = ctx[path].data()
1919 1920 if opts.get('decode'):
1920 1921 data = repo.wwritedata(path, data)
1921 1922 fp.write(data)
1922 1923 fp.close()
1923 1924
1924 1925 # Automation often uses hg cat on single files, so special case it
1925 1926 # for performance to avoid the cost of parsing the manifest.
1926 1927 if len(matcher.files()) == 1 and not matcher.anypats():
1927 1928 file = matcher.files()[0]
1928 1929 mf = repo.manifest
1929 1930 mfnode = ctx._changeset[0]
1930 1931 if mf.find(mfnode, file)[0]:
1931 1932 write(file)
1932 1933 return 0
1933 1934
1934 1935 # Don't warn about "missing" files that are really in subrepos
1935 1936 bad = matcher.bad
1936 1937
1937 1938 def badfn(path, msg):
1938 1939 for subpath in ctx.substate:
1939 1940 if path.startswith(subpath):
1940 1941 return
1941 1942 bad(path, msg)
1942 1943
1943 1944 matcher.bad = badfn
1944 1945
1945 1946 for abs in ctx.walk(matcher):
1946 1947 write(abs)
1947 1948 err = 0
1948 1949
1949 1950 matcher.bad = bad
1950 1951
1951 1952 for subpath in sorted(ctx.substate):
1952 1953 sub = ctx.sub(subpath)
1953 1954 try:
1954 1955 submatch = matchmod.narrowmatcher(subpath, matcher)
1955 1956
1956 1957 if not sub.cat(ui, submatch, os.path.join(prefix, sub._path),
1957 1958 **opts):
1958 1959 err = 0
1959 1960 except error.RepoLookupError:
1960 1961 ui.status(_("skipping missing subrepository: %s\n")
1961 1962 % os.path.join(prefix, subpath))
1962 1963
1963 1964 return err
1964 1965
1965 1966 def duplicatecopies(repo, rev, fromrev, skiprev=None):
1966 1967 '''reproduce copies from fromrev to rev in the dirstate
1967 1968
1968 1969 If skiprev is specified, it's a revision that should be used to
1969 1970 filter copy records. Any copies that occur between fromrev and
1970 1971 skiprev will not be duplicated, even if they appear in the set of
1971 1972 copies between fromrev and rev.
1972 1973 '''
1973 1974 exclude = {}
1974 1975 if skiprev is not None:
1975 1976 exclude = copies.pathcopies(repo[fromrev], repo[skiprev])
1976 1977 for dst, src in copies.pathcopies(repo[fromrev], repo[rev]).iteritems():
1977 1978 # copies.pathcopies returns backward renames, so dst might not
1978 1979 # actually be in the dirstate
1979 1980 if dst in exclude:
1980 1981 continue
1981 1982 if repo.dirstate[dst] in "nma":
1982 1983 repo.dirstate.copy(src, dst)
1983 1984
1984 1985 def commit(ui, repo, commitfunc, pats, opts):
1985 1986 '''commit the specified files or all outstanding changes'''
1986 1987 date = opts.get('date')
1987 1988 if date:
1988 1989 opts['date'] = util.parsedate(date)
1989 1990 message = logmessage(ui, opts)
1990 1991
1991 1992 # extract addremove carefully -- this function can be called from a command
1992 1993 # that doesn't support addremove
1993 1994 if opts.get('addremove'):
1994 1995 scmutil.addremove(repo, pats, opts)
1995 1996
1996 1997 return commitfunc(ui, repo, message,
1997 1998 scmutil.match(repo[None], pats, opts), opts)
1998 1999
1999 2000 def amend(ui, repo, commitfunc, old, extra, pats, opts):
2000 2001 ui.note(_('amending changeset %s\n') % old)
2001 2002 base = old.p1()
2002 2003
2003 2004 wlock = lock = newid = None
2004 2005 try:
2005 2006 wlock = repo.wlock()
2006 2007 lock = repo.lock()
2007 2008 tr = repo.transaction('amend')
2008 2009 try:
2009 2010 # See if we got a message from -m or -l, if not, open the editor
2010 2011 # with the message of the changeset to amend
2011 2012 message = logmessage(ui, opts)
2012 2013 # ensure logfile does not conflict with later enforcement of the
2013 2014 # message. potential logfile content has been processed by
2014 2015 # `logmessage` anyway.
2015 2016 opts.pop('logfile')
2016 2017 # First, do a regular commit to record all changes in the working
2017 2018 # directory (if there are any)
2018 2019 ui.callhooks = False
2019 2020 currentbookmark = repo._bookmarkcurrent
2020 2021 try:
2021 2022 repo._bookmarkcurrent = None
2022 2023 opts['message'] = 'temporary amend commit for %s' % old
2023 2024 node = commit(ui, repo, commitfunc, pats, opts)
2024 2025 finally:
2025 2026 repo._bookmarkcurrent = currentbookmark
2026 2027 ui.callhooks = True
2027 2028 ctx = repo[node]
2028 2029
2029 2030 # Participating changesets:
2030 2031 #
2031 2032 # node/ctx o - new (intermediate) commit that contains changes
2032 2033 # | from working dir to go into amending commit
2033 2034 # | (or a workingctx if there were no changes)
2034 2035 # |
2035 2036 # old o - changeset to amend
2036 2037 # |
2037 2038 # base o - parent of amending changeset
2038 2039
2039 2040 # Update extra dict from amended commit (e.g. to preserve graft
2040 2041 # source)
2041 2042 extra.update(old.extra())
2042 2043
2043 2044 # Also update it from the intermediate commit or from the wctx
2044 2045 extra.update(ctx.extra())
2045 2046
2046 2047 if len(old.parents()) > 1:
2047 2048 # ctx.files() isn't reliable for merges, so fall back to the
2048 2049 # slower repo.status() method
2049 2050 files = set([fn for st in repo.status(base, old)[:3]
2050 2051 for fn in st])
2051 2052 else:
2052 2053 files = set(old.files())
2053 2054
2054 2055 # Second, we use either the commit we just did, or if there were no
2055 2056 # changes the parent of the working directory as the version of the
2056 2057 # files in the final amend commit
2057 2058 if node:
2058 2059 ui.note(_('copying changeset %s to %s\n') % (ctx, base))
2059 2060
2060 2061 user = ctx.user()
2061 2062 date = ctx.date()
2062 2063 # Recompute copies (avoid recording a -> b -> a)
2063 2064 copied = copies.pathcopies(base, ctx)
2064 2065
2065 2066 # Prune files which were reverted by the updates: if old
2066 2067 # introduced file X and our intermediate commit, node,
2067 2068 # renamed that file, then those two files are the same and
2068 2069 # we can discard X from our list of files. Likewise if X
2069 2070 # was deleted, it's no longer relevant
2070 2071 files.update(ctx.files())
2071 2072
2072 2073 def samefile(f):
2073 2074 if f in ctx.manifest():
2074 2075 a = ctx.filectx(f)
2075 2076 if f in base.manifest():
2076 2077 b = base.filectx(f)
2077 2078 return (not a.cmp(b)
2078 2079 and a.flags() == b.flags())
2079 2080 else:
2080 2081 return False
2081 2082 else:
2082 2083 return f not in base.manifest()
2083 2084 files = [f for f in files if not samefile(f)]
2084 2085
2085 2086 def filectxfn(repo, ctx_, path):
2086 2087 try:
2087 2088 fctx = ctx[path]
2088 2089 flags = fctx.flags()
2089 2090 mctx = context.memfilectx(repo,
2090 2091 fctx.path(), fctx.data(),
2091 2092 islink='l' in flags,
2092 2093 isexec='x' in flags,
2093 2094 copied=copied.get(path))
2094 2095 return mctx
2095 2096 except KeyError:
2096 2097 raise IOError
2097 2098 else:
2098 2099 ui.note(_('copying changeset %s to %s\n') % (old, base))
2099 2100
2100 2101 # Use version of files as in the old cset
2101 2102 def filectxfn(repo, ctx_, path):
2102 2103 try:
2103 2104 return old.filectx(path)
2104 2105 except KeyError:
2105 2106 raise IOError
2106 2107
2107 2108 user = opts.get('user') or old.user()
2108 2109 date = opts.get('date') or old.date()
2109 2110 editform = 'commit.amend'
2110 2111 editor = getcommiteditor(editform=editform, **opts)
2111 2112 if not message:
2112 2113 editor = getcommiteditor(edit=True, editform=editform)
2113 2114 message = old.description()
2114 2115
2115 2116 pureextra = extra.copy()
2116 2117 extra['amend_source'] = old.hex()
2117 2118
2118 2119 new = context.memctx(repo,
2119 2120 parents=[base.node(), old.p2().node()],
2120 2121 text=message,
2121 2122 files=files,
2122 2123 filectxfn=filectxfn,
2123 2124 user=user,
2124 2125 date=date,
2125 2126 extra=extra,
2126 2127 editor=editor)
2127 2128
2128 2129 newdesc = changelog.stripdesc(new.description())
2129 2130 if ((not node)
2130 2131 and newdesc == old.description()
2131 2132 and user == old.user()
2132 2133 and date == old.date()
2133 2134 and pureextra == old.extra()):
2134 2135 # nothing changed. continuing here would create a new node
2135 2136 # anyway because of the amend_source noise.
2136 2137 #
2137 2138 # This not what we expect from amend.
2138 2139 return old.node()
2139 2140
2140 2141 ph = repo.ui.config('phases', 'new-commit', phases.draft)
2141 2142 try:
2142 2143 if opts.get('secret'):
2143 2144 commitphase = 'secret'
2144 2145 else:
2145 2146 commitphase = old.phase()
2146 2147 repo.ui.setconfig('phases', 'new-commit', commitphase, 'amend')
2147 2148 newid = repo.commitctx(new)
2148 2149 finally:
2149 2150 repo.ui.setconfig('phases', 'new-commit', ph, 'amend')
2150 2151 if newid != old.node():
2151 2152 # Reroute the working copy parent to the new changeset
2152 2153 repo.setparents(newid, nullid)
2153 2154
2154 2155 # Move bookmarks from old parent to amend commit
2155 2156 bms = repo.nodebookmarks(old.node())
2156 2157 if bms:
2157 2158 marks = repo._bookmarks
2158 2159 for bm in bms:
2159 2160 marks[bm] = newid
2160 2161 marks.write()
2161 2162 #commit the whole amend process
2162 2163 if obsolete._enabled and newid != old.node():
2163 2164 # mark the new changeset as successor of the rewritten one
2164 2165 new = repo[newid]
2165 2166 obs = [(old, (new,))]
2166 2167 if node:
2167 2168 obs.append((ctx, ()))
2168 2169
2169 2170 obsolete.createmarkers(repo, obs)
2170 2171 tr.close()
2171 2172 finally:
2172 2173 tr.release()
2173 2174 if (not obsolete._enabled) and newid != old.node():
2174 2175 # Strip the intermediate commit (if there was one) and the amended
2175 2176 # commit
2176 2177 if node:
2177 2178 ui.note(_('stripping intermediate changeset %s\n') % ctx)
2178 2179 ui.note(_('stripping amended changeset %s\n') % old)
2179 2180 repair.strip(ui, repo, old.node(), topic='amend-backup')
2180 2181 finally:
2181 2182 if newid is None:
2182 2183 repo.dirstate.invalidate()
2183 2184 lockmod.release(lock, wlock)
2184 2185 return newid
2185 2186
2186 2187 def commiteditor(repo, ctx, subs, editform=''):
2187 2188 if ctx.description():
2188 2189 return ctx.description()
2189 2190 return commitforceeditor(repo, ctx, subs, editform=editform)
2190 2191
2191 2192 def commitforceeditor(repo, ctx, subs, finishdesc=None, extramsg=None,
2192 2193 editform=''):
2193 2194 if not extramsg:
2194 2195 extramsg = _("Leave message empty to abort commit.")
2195 2196 tmpl = repo.ui.config('committemplate', 'changeset', '').strip()
2196 2197 if tmpl:
2197 2198 committext = buildcommittemplate(repo, ctx, subs, extramsg, tmpl)
2198 2199 else:
2199 2200 committext = buildcommittext(repo, ctx, subs, extramsg)
2200 2201
2201 2202 # run editor in the repository root
2202 2203 olddir = os.getcwd()
2203 2204 os.chdir(repo.root)
2204 2205 text = repo.ui.edit(committext, ctx.user(), ctx.extra())
2205 2206 text = re.sub("(?m)^HG:.*(\n|$)", "", text)
2206 2207 os.chdir(olddir)
2207 2208
2208 2209 if finishdesc:
2209 2210 text = finishdesc(text)
2210 2211 if not text.strip():
2211 2212 raise util.Abort(_("empty commit message"))
2212 2213
2213 2214 return text
2214 2215
2215 2216 def buildcommittemplate(repo, ctx, subs, extramsg, tmpl):
2216 2217 ui = repo.ui
2217 2218 tmpl, mapfile = gettemplate(ui, tmpl, None)
2218 2219
2219 2220 try:
2220 2221 t = changeset_templater(ui, repo, None, {}, tmpl, mapfile, False)
2221 2222 except SyntaxError, inst:
2222 2223 raise util.Abort(inst.args[0])
2223 2224
2224 2225 if not extramsg:
2225 2226 extramsg = '' # ensure that extramsg is string
2226 2227
2227 2228 ui.pushbuffer()
2228 2229 t.show(ctx, extramsg=extramsg)
2229 2230 return ui.popbuffer()
2230 2231
2231 2232 def buildcommittext(repo, ctx, subs, extramsg):
2232 2233 edittext = []
2233 2234 modified, added, removed = ctx.modified(), ctx.added(), ctx.removed()
2234 2235 if ctx.description():
2235 2236 edittext.append(ctx.description())
2236 2237 edittext.append("")
2237 2238 edittext.append("") # Empty line between message and comments.
2238 2239 edittext.append(_("HG: Enter commit message."
2239 2240 " Lines beginning with 'HG:' are removed."))
2240 2241 edittext.append("HG: %s" % extramsg)
2241 2242 edittext.append("HG: --")
2242 2243 edittext.append(_("HG: user: %s") % ctx.user())
2243 2244 if ctx.p2():
2244 2245 edittext.append(_("HG: branch merge"))
2245 2246 if ctx.branch():
2246 2247 edittext.append(_("HG: branch '%s'") % ctx.branch())
2247 2248 if bookmarks.iscurrent(repo):
2248 2249 edittext.append(_("HG: bookmark '%s'") % repo._bookmarkcurrent)
2249 2250 edittext.extend([_("HG: subrepo %s") % s for s in subs])
2250 2251 edittext.extend([_("HG: added %s") % f for f in added])
2251 2252 edittext.extend([_("HG: changed %s") % f for f in modified])
2252 2253 edittext.extend([_("HG: removed %s") % f for f in removed])
2253 2254 if not added and not modified and not removed:
2254 2255 edittext.append(_("HG: no files changed"))
2255 2256 edittext.append("")
2256 2257
2257 2258 return "\n".join(edittext)
2258 2259
2259 2260 def commitstatus(repo, node, branch, bheads=None, opts={}):
2260 2261 ctx = repo[node]
2261 2262 parents = ctx.parents()
2262 2263
2263 2264 if (not opts.get('amend') and bheads and node not in bheads and not
2264 2265 [x for x in parents if x.node() in bheads and x.branch() == branch]):
2265 2266 repo.ui.status(_('created new head\n'))
2266 2267 # The message is not printed for initial roots. For the other
2267 2268 # changesets, it is printed in the following situations:
2268 2269 #
2269 2270 # Par column: for the 2 parents with ...
2270 2271 # N: null or no parent
2271 2272 # B: parent is on another named branch
2272 2273 # C: parent is a regular non head changeset
2273 2274 # H: parent was a branch head of the current branch
2274 2275 # Msg column: whether we print "created new head" message
2275 2276 # In the following, it is assumed that there already exists some
2276 2277 # initial branch heads of the current branch, otherwise nothing is
2277 2278 # printed anyway.
2278 2279 #
2279 2280 # Par Msg Comment
2280 2281 # N N y additional topo root
2281 2282 #
2282 2283 # B N y additional branch root
2283 2284 # C N y additional topo head
2284 2285 # H N n usual case
2285 2286 #
2286 2287 # B B y weird additional branch root
2287 2288 # C B y branch merge
2288 2289 # H B n merge with named branch
2289 2290 #
2290 2291 # C C y additional head from merge
2291 2292 # C H n merge with a head
2292 2293 #
2293 2294 # H H n head merge: head count decreases
2294 2295
2295 2296 if not opts.get('close_branch'):
2296 2297 for r in parents:
2297 2298 if r.closesbranch() and r.branch() == branch:
2298 2299 repo.ui.status(_('reopening closed branch head %d\n') % r)
2299 2300
2300 2301 if repo.ui.debugflag:
2301 2302 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx.hex()))
2302 2303 elif repo.ui.verbose:
2303 2304 repo.ui.write(_('committed changeset %d:%s\n') % (int(ctx), ctx))
2304 2305
2305 2306 def revert(ui, repo, ctx, parents, *pats, **opts):
2306 2307 parent, p2 = parents
2307 2308 node = ctx.node()
2308 2309
2309 2310 mf = ctx.manifest()
2310 2311 if node == p2:
2311 2312 parent = p2
2312 2313 if node == parent:
2313 2314 pmf = mf
2314 2315 else:
2315 2316 pmf = None
2316 2317
2317 2318 # need all matching names in dirstate and manifest of target rev,
2318 2319 # so have to walk both. do not print errors if files exist in one
2319 2320 # but not other.
2320 2321
2321 2322 # `names` is a mapping for all elements in working copy and target revision
2322 2323 # The mapping is in the form:
2323 2324 # <asb path in repo> -> (<path from CWD>, <exactly specified by matcher?>)
2324 2325 names = {}
2325 2326
2326 2327 wlock = repo.wlock()
2327 2328 try:
2328 2329 ## filling of the `names` mapping
2329 2330 # walk dirstate to fill `names`
2330 2331
2331 2332 m = scmutil.match(repo[None], pats, opts)
2332 2333 m.bad = lambda x, y: False
2333 2334 for abs in repo.walk(m):
2334 2335 names[abs] = m.rel(abs), m.exact(abs)
2335 2336
2336 2337 # walk target manifest to fill `names`
2337 2338
2338 2339 def badfn(path, msg):
2339 2340 if path in names:
2340 2341 return
2341 2342 if path in ctx.substate:
2342 2343 return
2343 2344 path_ = path + '/'
2344 2345 for f in names:
2345 2346 if f.startswith(path_):
2346 2347 return
2347 2348 ui.warn("%s: %s\n" % (m.rel(path), msg))
2348 2349
2349 2350 m = scmutil.match(ctx, pats, opts)
2350 2351 m.bad = badfn
2351 2352 for abs in ctx.walk(m):
2352 2353 if abs not in names:
2353 2354 names[abs] = m.rel(abs), m.exact(abs)
2354 2355
2355 2356 # get the list of subrepos that must be reverted
2356 2357 targetsubs = sorted(s for s in ctx.substate if m(s))
2357 2358
2358 2359 # Find status of all file in `names`. (Against working directory parent)
2359 2360 m = scmutil.matchfiles(repo, names)
2360 2361 changes = repo.status(node1=parent, match=m)[:4]
2361 2362 modified, added, removed, deleted = map(set, changes)
2362 2363
2363 2364 # if f is a rename, update `names` to also revert the source
2364 2365 cwd = repo.getcwd()
2365 2366 for f in added:
2366 2367 src = repo.dirstate.copied(f)
2367 2368 if src and src not in names and repo.dirstate[src] == 'r':
2368 2369 removed.add(src)
2369 2370 names[src] = (repo.pathto(src, cwd), True)
2370 2371
2371 2372 ## computation of the action to performs on `names` content.
2372 2373
2373 2374 def removeforget(abs):
2374 2375 if repo.dirstate[abs] == 'a':
2375 2376 return _('forgetting %s\n')
2376 2377 return _('removing %s\n')
2377 2378
2378 2379 # action to be actually performed by revert
2379 2380 # (<list of file>, message>) tuple
2380 2381 actions = {'revert': ([], _('reverting %s\n')),
2381 2382 'add': ([], _('adding %s\n')),
2382 2383 'remove': ([], removeforget),
2383 2384 'undelete': ([], _('undeleting %s\n'))}
2384 2385
2385 2386 disptable = (
2386 2387 # dispatch table:
2387 2388 # file state
2388 2389 # action if in target manifest
2389 2390 # action if not in target manifest
2390 2391 # make backup if in target manifest
2391 2392 # make backup if not in target manifest
2392 2393 (modified, (actions['revert'], True),
2393 2394 (actions['remove'], True)),
2394 2395 (added, (actions['revert'], True),
2395 2396 (actions['remove'], False)),
2396 2397 (removed, (actions['undelete'], True),
2397 2398 (None, False)),
2398 2399 (deleted, (actions['revert'], False),
2399 2400 (actions['remove'], False)),
2400 2401 )
2401 2402
2402 2403 for abs, (rel, exact) in sorted(names.items()):
2403 2404 # hash on file in target manifest (or None if missing from target)
2404 2405 mfentry = mf.get(abs)
2405 2406 # target file to be touch on disk (relative to cwd)
2406 2407 target = repo.wjoin(abs)
2407 2408 def handle(xlist, dobackup):
2408 2409 xlist[0].append(abs)
2409 2410 if (dobackup and not opts.get('no_backup') and
2410 2411 os.path.lexists(target) and
2411 2412 abs in ctx and repo[None][abs].cmp(ctx[abs])):
2412 2413 bakname = "%s.orig" % rel
2413 2414 ui.note(_('saving current version of %s as %s\n') %
2414 2415 (rel, bakname))
2415 2416 if not opts.get('dry_run'):
2416 2417 util.rename(target, bakname)
2417 2418 if ui.verbose or not exact:
2418 2419 msg = xlist[1]
2419 2420 if not isinstance(msg, basestring):
2420 2421 msg = msg(abs)
2421 2422 ui.status(msg % rel)
2422 2423 # search the entry in the dispatch table.
2423 2424 # if the file is in any of this sets, it was touched in the working
2424 2425 # directory parent and we are sure it needs to be reverted.
2425 2426 for table, hit, miss in disptable:
2426 2427 if abs not in table:
2427 2428 continue
2428 2429 # file has changed in dirstate
2429 2430 if mfentry:
2430 2431 handle(*hit)
2431 2432 elif miss[0] is not None:
2432 2433 handle(*miss)
2433 2434 break
2434 2435 else:
2435 2436 # Not touched in current dirstate.
2436 2437
2437 2438 # file is unknown in parent, restore older version or ignore.
2438 2439 if abs not in repo.dirstate:
2439 2440 if mfentry:
2440 2441 handle(actions['add'], True)
2441 2442 elif exact:
2442 2443 ui.warn(_('file not managed: %s\n') % rel)
2443 2444 continue
2444 2445
2445 2446 # parent is target, no changes mean no changes
2446 2447 if node == parent:
2447 2448 if exact:
2448 2449 ui.warn(_('no changes needed to %s\n') % rel)
2449 2450 continue
2450 2451 # no change in dirstate but parent and target may differ
2451 2452 if pmf is None:
2452 2453 # only need parent manifest in this unlikely case,
2453 2454 # so do not read by default
2454 2455 pmf = repo[parent].manifest()
2455 2456 if abs in pmf and mfentry:
2456 2457 # if version of file is same in parent and target
2457 2458 # manifests, do nothing
2458 2459 if (pmf[abs] != mfentry or
2459 2460 pmf.flags(abs) != mf.flags(abs)):
2460 2461 handle(actions['revert'], False)
2461 2462 else:
2462 2463 handle(actions['remove'], False)
2463 2464
2464 2465 if not opts.get('dry_run'):
2465 2466 _performrevert(repo, parents, ctx, actions)
2466 2467
2467 2468 if targetsubs:
2468 2469 # Revert the subrepos on the revert list
2469 2470 for sub in targetsubs:
2470 2471 ctx.sub(sub).revert(ui, ctx.substate[sub], *pats, **opts)
2471 2472 finally:
2472 2473 wlock.release()
2473 2474
2474 2475 def _performrevert(repo, parents, ctx, actions):
2475 2476 """function that actually perform all the actions computed for revert
2476 2477
2477 2478 This is an independent function to let extension to plug in and react to
2478 2479 the imminent revert.
2479 2480
2480 2481 Make sure you have the working directory locked when calling this function.
2481 2482 """
2482 2483 parent, p2 = parents
2483 2484 node = ctx.node()
2484 2485 def checkout(f):
2485 2486 fc = ctx[f]
2486 2487 repo.wwrite(f, fc.data(), fc.flags())
2487 2488
2488 2489 audit_path = pathutil.pathauditor(repo.root)
2489 2490 for f in actions['remove'][0]:
2490 2491 if repo.dirstate[f] == 'a':
2491 2492 repo.dirstate.drop(f)
2492 2493 continue
2493 2494 audit_path(f)
2494 2495 try:
2495 2496 util.unlinkpath(repo.wjoin(f))
2496 2497 except OSError:
2497 2498 pass
2498 2499 repo.dirstate.remove(f)
2499 2500
2500 2501 normal = None
2501 2502 if node == parent:
2502 2503 # We're reverting to our parent. If possible, we'd like status
2503 2504 # to report the file as clean. We have to use normallookup for
2504 2505 # merges to avoid losing information about merged/dirty files.
2505 2506 if p2 != nullid:
2506 2507 normal = repo.dirstate.normallookup
2507 2508 else:
2508 2509 normal = repo.dirstate.normal
2509 2510 for f in actions['revert'][0]:
2510 2511 checkout(f)
2511 2512 if normal:
2512 2513 normal(f)
2513 2514
2514 2515 for f in actions['add'][0]:
2515 2516 checkout(f)
2516 2517 repo.dirstate.add(f)
2517 2518
2518 2519 normal = repo.dirstate.normallookup
2519 2520 if node == parent and p2 == nullid:
2520 2521 normal = repo.dirstate.normal
2521 2522 for f in actions['undelete'][0]:
2522 2523 checkout(f)
2523 2524 normal(f)
2524 2525
2525 2526 copied = copies.pathcopies(repo[parent], ctx)
2526 2527
2527 2528 for f in actions['add'][0] + actions['undelete'][0] + actions['revert'][0]:
2528 2529 if f in copied:
2529 2530 repo.dirstate.copy(copied[f], f)
2530 2531
2531 2532 def command(table):
2532 2533 """Returns a function object to be used as a decorator for making commands.
2533 2534
2534 2535 This function receives a command table as its argument. The table should
2535 2536 be a dict.
2536 2537
2537 2538 The returned function can be used as a decorator for adding commands
2538 2539 to that command table. This function accepts multiple arguments to define
2539 2540 a command.
2540 2541
2541 2542 The first argument is the command name.
2542 2543
2543 2544 The options argument is an iterable of tuples defining command arguments.
2544 2545 See ``mercurial.fancyopts.fancyopts()`` for the format of each tuple.
2545 2546
2546 2547 The synopsis argument defines a short, one line summary of how to use the
2547 2548 command. This shows up in the help output.
2548 2549
2549 2550 The norepo argument defines whether the command does not require a
2550 2551 local repository. Most commands operate against a repository, thus the
2551 2552 default is False.
2552 2553
2553 2554 The optionalrepo argument defines whether the command optionally requires
2554 2555 a local repository.
2555 2556
2556 2557 The inferrepo argument defines whether to try to find a repository from the
2557 2558 command line arguments. If True, arguments will be examined for potential
2558 2559 repository locations. See ``findrepo()``. If a repository is found, it
2559 2560 will be used.
2560 2561 """
2561 2562 def cmd(name, options=(), synopsis=None, norepo=False, optionalrepo=False,
2562 2563 inferrepo=False):
2563 2564 def decorator(func):
2564 2565 if synopsis:
2565 2566 table[name] = func, list(options), synopsis
2566 2567 else:
2567 2568 table[name] = func, list(options)
2568 2569
2569 2570 if norepo:
2570 2571 # Avoid import cycle.
2571 2572 import commands
2572 2573 commands.norepo += ' %s' % ' '.join(parsealiases(name))
2573 2574
2574 2575 if optionalrepo:
2575 2576 import commands
2576 2577 commands.optionalrepo += ' %s' % ' '.join(parsealiases(name))
2577 2578
2578 2579 if inferrepo:
2579 2580 import commands
2580 2581 commands.inferrepo += ' %s' % ' '.join(parsealiases(name))
2581 2582
2582 2583 return func
2583 2584 return decorator
2584 2585
2585 2586 return cmd
2586 2587
2587 2588 # a list of (ui, repo, otherpeer, opts, missing) functions called by
2588 2589 # commands.outgoing. "missing" is "missing" of the result of
2589 2590 # "findcommonoutgoing()"
2590 2591 outgoinghooks = util.hooks()
2591 2592
2592 2593 # a list of (ui, repo) functions called by commands.summary
2593 2594 summaryhooks = util.hooks()
2594 2595
2595 2596 # a list of (ui, repo, opts, changes) functions called by commands.summary.
2596 2597 #
2597 2598 # functions should return tuple of booleans below, if 'changes' is None:
2598 2599 # (whether-incomings-are-needed, whether-outgoings-are-needed)
2599 2600 #
2600 2601 # otherwise, 'changes' is a tuple of tuples below:
2601 2602 # - (sourceurl, sourcebranch, sourcepeer, incoming)
2602 2603 # - (desturl, destbranch, destpeer, outgoing)
2603 2604 summaryremotehooks = util.hooks()
2604 2605
2605 2606 # A list of state files kept by multistep operations like graft.
2606 2607 # Since graft cannot be aborted, it is considered 'clearable' by update.
2607 2608 # note: bisect is intentionally excluded
2608 2609 # (state file, clearable, allowcommit, error, hint)
2609 2610 unfinishedstates = [
2610 2611 ('graftstate', True, False, _('graft in progress'),
2611 2612 _("use 'hg graft --continue' or 'hg update' to abort")),
2612 2613 ('updatestate', True, False, _('last update was interrupted'),
2613 2614 _("use 'hg update' to get a consistent checkout"))
2614 2615 ]
2615 2616
2616 2617 def checkunfinished(repo, commit=False):
2617 2618 '''Look for an unfinished multistep operation, like graft, and abort
2618 2619 if found. It's probably good to check this right before
2619 2620 bailifchanged().
2620 2621 '''
2621 2622 for f, clearable, allowcommit, msg, hint in unfinishedstates:
2622 2623 if commit and allowcommit:
2623 2624 continue
2624 2625 if repo.vfs.exists(f):
2625 2626 raise util.Abort(msg, hint=hint)
2626 2627
2627 2628 def clearunfinished(repo):
2628 2629 '''Check for unfinished operations (as above), and clear the ones
2629 2630 that are clearable.
2630 2631 '''
2631 2632 for f, clearable, allowcommit, msg, hint in unfinishedstates:
2632 2633 if not clearable and repo.vfs.exists(f):
2633 2634 raise util.Abort(msg, hint=hint)
2634 2635 for f, clearable, allowcommit, msg, hint in unfinishedstates:
2635 2636 if clearable and repo.vfs.exists(f):
2636 2637 util.unlink(repo.join(f))
General Comments 0
You need to be logged in to leave comments. Login now