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