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