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