##// END OF EJS Templates
context: make warning message for hidden revision extensible...
Laurent Charignon -
r24922:d9fb88c0 default
parent child Browse files
Show More
@@ -1,1898 +1,1898 b''
1 1 # context.py - changeset and file context objects for mercurial
2 2 #
3 3 # Copyright 2006, 2007 Matt Mackall <mpm@selenic.com>
4 4 #
5 5 # This software may be used and distributed according to the terms of the
6 6 # GNU General Public License version 2 or any later version.
7 7
8 8 from node import nullid, nullrev, short, hex, bin
9 9 from i18n import _
10 10 import mdiff, error, util, scmutil, subrepo, patch, encoding, phases
11 11 import match as matchmod
12 12 import os, errno, stat
13 13 import obsolete as obsmod
14 14 import repoview
15 15 import fileset
16 16 import revlog
17 17
18 18 propertycache = util.propertycache
19 19
20 20 # Phony node value to stand-in for new files in some uses of
21 21 # manifests. Manifests support 21-byte hashes for nodes which are
22 22 # dirty in the working copy.
23 23 _newnode = '!' * 21
24 24
25 25 class basectx(object):
26 26 """A basectx object represents the common logic for its children:
27 27 changectx: read-only context that is already present in the repo,
28 28 workingctx: a context that represents the working directory and can
29 29 be committed,
30 30 memctx: a context that represents changes in-memory and can also
31 31 be committed."""
32 32 def __new__(cls, repo, changeid='', *args, **kwargs):
33 33 if isinstance(changeid, basectx):
34 34 return changeid
35 35
36 36 o = super(basectx, cls).__new__(cls)
37 37
38 38 o._repo = repo
39 39 o._rev = nullrev
40 40 o._node = nullid
41 41
42 42 return o
43 43
44 44 def __str__(self):
45 45 return short(self.node())
46 46
47 47 def __int__(self):
48 48 return self.rev()
49 49
50 50 def __repr__(self):
51 51 return "<%s %s>" % (type(self).__name__, str(self))
52 52
53 53 def __eq__(self, other):
54 54 try:
55 55 return type(self) == type(other) and self._rev == other._rev
56 56 except AttributeError:
57 57 return False
58 58
59 59 def __ne__(self, other):
60 60 return not (self == other)
61 61
62 62 def __contains__(self, key):
63 63 return key in self._manifest
64 64
65 65 def __getitem__(self, key):
66 66 return self.filectx(key)
67 67
68 68 def __iter__(self):
69 69 return iter(self._manifest)
70 70
71 71 def _manifestmatches(self, match, s):
72 72 """generate a new manifest filtered by the match argument
73 73
74 74 This method is for internal use only and mainly exists to provide an
75 75 object oriented way for other contexts to customize the manifest
76 76 generation.
77 77 """
78 78 return self.manifest().matches(match)
79 79
80 80 def _matchstatus(self, other, match):
81 81 """return match.always if match is none
82 82
83 83 This internal method provides a way for child objects to override the
84 84 match operator.
85 85 """
86 86 return match or matchmod.always(self._repo.root, self._repo.getcwd())
87 87
88 88 def _buildstatus(self, other, s, match, listignored, listclean,
89 89 listunknown):
90 90 """build a status with respect to another context"""
91 91 # Load earliest manifest first for caching reasons. More specifically,
92 92 # if you have revisions 1000 and 1001, 1001 is probably stored as a
93 93 # delta against 1000. Thus, if you read 1000 first, we'll reconstruct
94 94 # 1000 and cache it so that when you read 1001, we just need to apply a
95 95 # delta to what's in the cache. So that's one full reconstruction + one
96 96 # delta application.
97 97 if self.rev() is not None and self.rev() < other.rev():
98 98 self.manifest()
99 99 mf1 = other._manifestmatches(match, s)
100 100 mf2 = self._manifestmatches(match, s)
101 101
102 102 modified, added = [], []
103 103 removed = []
104 104 clean = []
105 105 deleted, unknown, ignored = s.deleted, s.unknown, s.ignored
106 106 deletedset = set(deleted)
107 107 d = mf1.diff(mf2, clean=listclean)
108 108 for fn, value in d.iteritems():
109 109 if fn in deletedset:
110 110 continue
111 111 if value is None:
112 112 clean.append(fn)
113 113 continue
114 114 (node1, flag1), (node2, flag2) = value
115 115 if node1 is None:
116 116 added.append(fn)
117 117 elif node2 is None:
118 118 removed.append(fn)
119 119 elif node2 != _newnode:
120 120 # The file was not a new file in mf2, so an entry
121 121 # from diff is really a difference.
122 122 modified.append(fn)
123 123 elif self[fn].cmp(other[fn]):
124 124 # node2 was newnode, but the working file doesn't
125 125 # match the one in mf1.
126 126 modified.append(fn)
127 127 else:
128 128 clean.append(fn)
129 129
130 130 if removed:
131 131 # need to filter files if they are already reported as removed
132 132 unknown = [fn for fn in unknown if fn not in mf1]
133 133 ignored = [fn for fn in ignored if fn not in mf1]
134 134 # if they're deleted, don't report them as removed
135 135 removed = [fn for fn in removed if fn not in deletedset]
136 136
137 137 return scmutil.status(modified, added, removed, deleted, unknown,
138 138 ignored, clean)
139 139
140 140 @propertycache
141 141 def substate(self):
142 142 return subrepo.state(self, self._repo.ui)
143 143
144 144 def subrev(self, subpath):
145 145 return self.substate[subpath][1]
146 146
147 147 def rev(self):
148 148 return self._rev
149 149 def node(self):
150 150 return self._node
151 151 def hex(self):
152 152 return hex(self.node())
153 153 def manifest(self):
154 154 return self._manifest
155 155 def repo(self):
156 156 return self._repo
157 157 def phasestr(self):
158 158 return phases.phasenames[self.phase()]
159 159 def mutable(self):
160 160 return self.phase() > phases.public
161 161
162 162 def getfileset(self, expr):
163 163 return fileset.getfileset(self, expr)
164 164
165 165 def obsolete(self):
166 166 """True if the changeset is obsolete"""
167 167 return self.rev() in obsmod.getrevs(self._repo, 'obsolete')
168 168
169 169 def extinct(self):
170 170 """True if the changeset is extinct"""
171 171 return self.rev() in obsmod.getrevs(self._repo, 'extinct')
172 172
173 173 def unstable(self):
174 174 """True if the changeset is not obsolete but it's ancestor are"""
175 175 return self.rev() in obsmod.getrevs(self._repo, 'unstable')
176 176
177 177 def bumped(self):
178 178 """True if the changeset try to be a successor of a public changeset
179 179
180 180 Only non-public and non-obsolete changesets may be bumped.
181 181 """
182 182 return self.rev() in obsmod.getrevs(self._repo, 'bumped')
183 183
184 184 def divergent(self):
185 185 """Is a successors of a changeset with multiple possible successors set
186 186
187 187 Only non-public and non-obsolete changesets may be divergent.
188 188 """
189 189 return self.rev() in obsmod.getrevs(self._repo, 'divergent')
190 190
191 191 def troubled(self):
192 192 """True if the changeset is either unstable, bumped or divergent"""
193 193 return self.unstable() or self.bumped() or self.divergent()
194 194
195 195 def troubles(self):
196 196 """return the list of troubles affecting this changesets.
197 197
198 198 Troubles are returned as strings. possible values are:
199 199 - unstable,
200 200 - bumped,
201 201 - divergent.
202 202 """
203 203 troubles = []
204 204 if self.unstable():
205 205 troubles.append('unstable')
206 206 if self.bumped():
207 207 troubles.append('bumped')
208 208 if self.divergent():
209 209 troubles.append('divergent')
210 210 return troubles
211 211
212 212 def parents(self):
213 213 """return contexts for each parent changeset"""
214 214 return self._parents
215 215
216 216 def p1(self):
217 217 return self._parents[0]
218 218
219 219 def p2(self):
220 220 if len(self._parents) == 2:
221 221 return self._parents[1]
222 222 return changectx(self._repo, -1)
223 223
224 224 def _fileinfo(self, path):
225 225 if '_manifest' in self.__dict__:
226 226 try:
227 227 return self._manifest[path], self._manifest.flags(path)
228 228 except KeyError:
229 229 raise error.ManifestLookupError(self._node, path,
230 230 _('not found in manifest'))
231 231 if '_manifestdelta' in self.__dict__ or path in self.files():
232 232 if path in self._manifestdelta:
233 233 return (self._manifestdelta[path],
234 234 self._manifestdelta.flags(path))
235 235 node, flag = self._repo.manifest.find(self._changeset[0], path)
236 236 if not node:
237 237 raise error.ManifestLookupError(self._node, path,
238 238 _('not found in manifest'))
239 239
240 240 return node, flag
241 241
242 242 def filenode(self, path):
243 243 return self._fileinfo(path)[0]
244 244
245 245 def flags(self, path):
246 246 try:
247 247 return self._fileinfo(path)[1]
248 248 except error.LookupError:
249 249 return ''
250 250
251 251 def sub(self, path):
252 252 return subrepo.subrepo(self, path)
253 253
254 254 def match(self, pats=[], include=None, exclude=None, default='glob'):
255 255 r = self._repo
256 256 return matchmod.match(r.root, r.getcwd(), pats,
257 257 include, exclude, default,
258 258 auditor=r.auditor, ctx=self)
259 259
260 260 def diff(self, ctx2=None, match=None, **opts):
261 261 """Returns a diff generator for the given contexts and matcher"""
262 262 if ctx2 is None:
263 263 ctx2 = self.p1()
264 264 if ctx2 is not None:
265 265 ctx2 = self._repo[ctx2]
266 266 diffopts = patch.diffopts(self._repo.ui, opts)
267 267 return patch.diff(self._repo, ctx2, self, match=match, opts=diffopts)
268 268
269 269 def dirs(self):
270 270 return self._manifest.dirs()
271 271
272 272 def hasdir(self, dir):
273 273 return self._manifest.hasdir(dir)
274 274
275 275 def dirty(self, missing=False, merge=True, branch=True):
276 276 return False
277 277
278 278 def status(self, other=None, match=None, listignored=False,
279 279 listclean=False, listunknown=False, listsubrepos=False):
280 280 """return status of files between two nodes or node and working
281 281 directory.
282 282
283 283 If other is None, compare this node with working directory.
284 284
285 285 returns (modified, added, removed, deleted, unknown, ignored, clean)
286 286 """
287 287
288 288 ctx1 = self
289 289 ctx2 = self._repo[other]
290 290
291 291 # This next code block is, admittedly, fragile logic that tests for
292 292 # reversing the contexts and wouldn't need to exist if it weren't for
293 293 # the fast (and common) code path of comparing the working directory
294 294 # with its first parent.
295 295 #
296 296 # What we're aiming for here is the ability to call:
297 297 #
298 298 # workingctx.status(parentctx)
299 299 #
300 300 # If we always built the manifest for each context and compared those,
301 301 # then we'd be done. But the special case of the above call means we
302 302 # just copy the manifest of the parent.
303 303 reversed = False
304 304 if (not isinstance(ctx1, changectx)
305 305 and isinstance(ctx2, changectx)):
306 306 reversed = True
307 307 ctx1, ctx2 = ctx2, ctx1
308 308
309 309 match = ctx2._matchstatus(ctx1, match)
310 310 r = scmutil.status([], [], [], [], [], [], [])
311 311 r = ctx2._buildstatus(ctx1, r, match, listignored, listclean,
312 312 listunknown)
313 313
314 314 if reversed:
315 315 # Reverse added and removed. Clear deleted, unknown and ignored as
316 316 # these make no sense to reverse.
317 317 r = scmutil.status(r.modified, r.removed, r.added, [], [], [],
318 318 r.clean)
319 319
320 320 if listsubrepos:
321 321 for subpath, sub in scmutil.itersubrepos(ctx1, ctx2):
322 322 rev2 = ctx2.subrev(subpath)
323 323 try:
324 324 submatch = matchmod.narrowmatcher(subpath, match)
325 325 s = sub.status(rev2, match=submatch, ignored=listignored,
326 326 clean=listclean, unknown=listunknown,
327 327 listsubrepos=True)
328 328 for rfiles, sfiles in zip(r, s):
329 329 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
330 330 except error.LookupError:
331 331 self._repo.ui.status(_("skipping missing "
332 332 "subrepository: %s\n") % subpath)
333 333
334 334 for l in r:
335 335 l.sort()
336 336
337 337 return r
338 338
339 339
340 340 def makememctx(repo, parents, text, user, date, branch, files, store,
341 341 editor=None):
342 342 def getfilectx(repo, memctx, path):
343 343 data, mode, copied = store.getfile(path)
344 344 if data is None:
345 345 return None
346 346 islink, isexec = mode
347 347 return memfilectx(repo, path, data, islink=islink, isexec=isexec,
348 348 copied=copied, memctx=memctx)
349 349 extra = {}
350 350 if branch:
351 351 extra['branch'] = encoding.fromlocal(branch)
352 352 ctx = memctx(repo, parents, text, files, getfilectx, user,
353 353 date, extra, editor)
354 354 return ctx
355 355
356 356 class changectx(basectx):
357 357 """A changecontext object makes access to data related to a particular
358 358 changeset convenient. It represents a read-only context already present in
359 359 the repo."""
360 360 def __init__(self, repo, changeid=''):
361 361 """changeid is a revision number, node, or tag"""
362 362
363 363 # since basectx.__new__ already took care of copying the object, we
364 364 # don't need to do anything in __init__, so we just exit here
365 365 if isinstance(changeid, basectx):
366 366 return
367 367
368 368 if changeid == '':
369 369 changeid = '.'
370 370 self._repo = repo
371 371
372 372 try:
373 373 if isinstance(changeid, int):
374 374 self._node = repo.changelog.node(changeid)
375 375 self._rev = changeid
376 376 return
377 377 if isinstance(changeid, long):
378 378 changeid = str(changeid)
379 379 if changeid == 'null':
380 380 self._node = nullid
381 381 self._rev = nullrev
382 382 return
383 383 if changeid == 'tip':
384 384 self._node = repo.changelog.tip()
385 385 self._rev = repo.changelog.rev(self._node)
386 386 return
387 387 if changeid == '.' or changeid == repo.dirstate.p1():
388 388 # this is a hack to delay/avoid loading obsmarkers
389 389 # when we know that '.' won't be hidden
390 390 self._node = repo.dirstate.p1()
391 391 self._rev = repo.unfiltered().changelog.rev(self._node)
392 392 return
393 393 if len(changeid) == 20:
394 394 try:
395 395 self._node = changeid
396 396 self._rev = repo.changelog.rev(changeid)
397 397 return
398 398 except error.FilteredRepoLookupError:
399 399 raise
400 400 except LookupError:
401 401 pass
402 402
403 403 try:
404 404 r = int(changeid)
405 405 if str(r) != changeid:
406 406 raise ValueError
407 407 l = len(repo.changelog)
408 408 if r < 0:
409 409 r += l
410 410 if r < 0 or r >= l:
411 411 raise ValueError
412 412 self._rev = r
413 413 self._node = repo.changelog.node(r)
414 414 return
415 415 except error.FilteredIndexError:
416 416 raise
417 417 except (ValueError, OverflowError, IndexError):
418 418 pass
419 419
420 420 if len(changeid) == 40:
421 421 try:
422 422 self._node = bin(changeid)
423 423 self._rev = repo.changelog.rev(self._node)
424 424 return
425 425 except error.FilteredLookupError:
426 426 raise
427 427 except (TypeError, LookupError):
428 428 pass
429 429
430 430 # lookup bookmarks through the name interface
431 431 try:
432 432 self._node = repo.names.singlenode(repo, changeid)
433 433 self._rev = repo.changelog.rev(self._node)
434 434 return
435 435 except KeyError:
436 436 pass
437 437 except error.FilteredRepoLookupError:
438 438 raise
439 439 except error.RepoLookupError:
440 440 pass
441 441
442 442 self._node = repo.unfiltered().changelog._partialmatch(changeid)
443 443 if self._node is not None:
444 444 self._rev = repo.changelog.rev(self._node)
445 445 return
446 446
447 447 # lookup failed
448 448 # check if it might have come from damaged dirstate
449 449 #
450 450 # XXX we could avoid the unfiltered if we had a recognizable
451 451 # exception for filtered changeset access
452 452 if changeid in repo.unfiltered().dirstate.parents():
453 453 msg = _("working directory has unknown parent '%s'!")
454 454 raise error.Abort(msg % short(changeid))
455 455 try:
456 456 if len(changeid) == 20:
457 457 changeid = hex(changeid)
458 458 except TypeError:
459 459 pass
460 460 except (error.FilteredIndexError, error.FilteredLookupError,
461 461 error.FilteredRepoLookupError):
462 if repo.filtername == 'visible':
462 if repo.filtername.startswith('visible'):
463 463 msg = _("hidden revision '%s'") % changeid
464 464 hint = _('use --hidden to access hidden revisions')
465 465 raise error.FilteredRepoLookupError(msg, hint=hint)
466 466 msg = _("filtered revision '%s' (not in '%s' subset)")
467 467 msg %= (changeid, repo.filtername)
468 468 raise error.FilteredRepoLookupError(msg)
469 469 except IndexError:
470 470 pass
471 471 raise error.RepoLookupError(
472 472 _("unknown revision '%s'") % changeid)
473 473
474 474 def __hash__(self):
475 475 try:
476 476 return hash(self._rev)
477 477 except AttributeError:
478 478 return id(self)
479 479
480 480 def __nonzero__(self):
481 481 return self._rev != nullrev
482 482
483 483 @propertycache
484 484 def _changeset(self):
485 485 return self._repo.changelog.read(self.rev())
486 486
487 487 @propertycache
488 488 def _manifest(self):
489 489 return self._repo.manifest.read(self._changeset[0])
490 490
491 491 @propertycache
492 492 def _manifestdelta(self):
493 493 return self._repo.manifest.readdelta(self._changeset[0])
494 494
495 495 @propertycache
496 496 def _parents(self):
497 497 p = self._repo.changelog.parentrevs(self._rev)
498 498 if p[1] == nullrev:
499 499 p = p[:-1]
500 500 return [changectx(self._repo, x) for x in p]
501 501
502 502 def changeset(self):
503 503 return self._changeset
504 504 def manifestnode(self):
505 505 return self._changeset[0]
506 506
507 507 def user(self):
508 508 return self._changeset[1]
509 509 def date(self):
510 510 return self._changeset[2]
511 511 def files(self):
512 512 return self._changeset[3]
513 513 def description(self):
514 514 return self._changeset[4]
515 515 def branch(self):
516 516 return encoding.tolocal(self._changeset[5].get("branch"))
517 517 def closesbranch(self):
518 518 return 'close' in self._changeset[5]
519 519 def extra(self):
520 520 return self._changeset[5]
521 521 def tags(self):
522 522 return self._repo.nodetags(self._node)
523 523 def bookmarks(self):
524 524 return self._repo.nodebookmarks(self._node)
525 525 def phase(self):
526 526 return self._repo._phasecache.phase(self._repo, self._rev)
527 527 def hidden(self):
528 528 return self._rev in repoview.filterrevs(self._repo, 'visible')
529 529
530 530 def children(self):
531 531 """return contexts for each child changeset"""
532 532 c = self._repo.changelog.children(self._node)
533 533 return [changectx(self._repo, x) for x in c]
534 534
535 535 def ancestors(self):
536 536 for a in self._repo.changelog.ancestors([self._rev]):
537 537 yield changectx(self._repo, a)
538 538
539 539 def descendants(self):
540 540 for d in self._repo.changelog.descendants([self._rev]):
541 541 yield changectx(self._repo, d)
542 542
543 543 def filectx(self, path, fileid=None, filelog=None):
544 544 """get a file context from this changeset"""
545 545 if fileid is None:
546 546 fileid = self.filenode(path)
547 547 return filectx(self._repo, path, fileid=fileid,
548 548 changectx=self, filelog=filelog)
549 549
550 550 def ancestor(self, c2, warn=False):
551 551 """return the "best" ancestor context of self and c2
552 552
553 553 If there are multiple candidates, it will show a message and check
554 554 merge.preferancestor configuration before falling back to the
555 555 revlog ancestor."""
556 556 # deal with workingctxs
557 557 n2 = c2._node
558 558 if n2 is None:
559 559 n2 = c2._parents[0]._node
560 560 cahs = self._repo.changelog.commonancestorsheads(self._node, n2)
561 561 if not cahs:
562 562 anc = nullid
563 563 elif len(cahs) == 1:
564 564 anc = cahs[0]
565 565 else:
566 566 for r in self._repo.ui.configlist('merge', 'preferancestor'):
567 567 try:
568 568 ctx = changectx(self._repo, r)
569 569 except error.RepoLookupError:
570 570 continue
571 571 anc = ctx.node()
572 572 if anc in cahs:
573 573 break
574 574 else:
575 575 anc = self._repo.changelog.ancestor(self._node, n2)
576 576 if warn:
577 577 self._repo.ui.status(
578 578 (_("note: using %s as ancestor of %s and %s\n") %
579 579 (short(anc), short(self._node), short(n2))) +
580 580 ''.join(_(" alternatively, use --config "
581 581 "merge.preferancestor=%s\n") %
582 582 short(n) for n in sorted(cahs) if n != anc))
583 583 return changectx(self._repo, anc)
584 584
585 585 def descendant(self, other):
586 586 """True if other is descendant of this changeset"""
587 587 return self._repo.changelog.descendant(self._rev, other._rev)
588 588
589 589 def walk(self, match):
590 590 '''Generates matching file names.'''
591 591
592 592 # Override match.bad method to have message with nodeid
593 593 oldbad = match.bad
594 594 def bad(fn, msg):
595 595 oldbad(fn, _('no such file in rev %s') % self)
596 596 match.bad = bad
597 597
598 598 return self._manifest.walk(match)
599 599
600 600 def matches(self, match):
601 601 return self.walk(match)
602 602
603 603 class basefilectx(object):
604 604 """A filecontext object represents the common logic for its children:
605 605 filectx: read-only access to a filerevision that is already present
606 606 in the repo,
607 607 workingfilectx: a filecontext that represents files from the working
608 608 directory,
609 609 memfilectx: a filecontext that represents files in-memory."""
610 610 def __new__(cls, repo, path, *args, **kwargs):
611 611 return super(basefilectx, cls).__new__(cls)
612 612
613 613 @propertycache
614 614 def _filelog(self):
615 615 return self._repo.file(self._path)
616 616
617 617 @propertycache
618 618 def _changeid(self):
619 619 if '_changeid' in self.__dict__:
620 620 return self._changeid
621 621 elif '_changectx' in self.__dict__:
622 622 return self._changectx.rev()
623 623 elif '_descendantrev' in self.__dict__:
624 624 # this file context was created from a revision with a known
625 625 # descendant, we can (lazily) correct for linkrev aliases
626 626 return self._adjustlinkrev(self._path, self._filelog,
627 627 self._filenode, self._descendantrev)
628 628 else:
629 629 return self._filelog.linkrev(self._filerev)
630 630
631 631 @propertycache
632 632 def _filenode(self):
633 633 if '_fileid' in self.__dict__:
634 634 return self._filelog.lookup(self._fileid)
635 635 else:
636 636 return self._changectx.filenode(self._path)
637 637
638 638 @propertycache
639 639 def _filerev(self):
640 640 return self._filelog.rev(self._filenode)
641 641
642 642 @propertycache
643 643 def _repopath(self):
644 644 return self._path
645 645
646 646 def __nonzero__(self):
647 647 try:
648 648 self._filenode
649 649 return True
650 650 except error.LookupError:
651 651 # file is missing
652 652 return False
653 653
654 654 def __str__(self):
655 655 return "%s@%s" % (self.path(), self._changectx)
656 656
657 657 def __repr__(self):
658 658 return "<%s %s>" % (type(self).__name__, str(self))
659 659
660 660 def __hash__(self):
661 661 try:
662 662 return hash((self._path, self._filenode))
663 663 except AttributeError:
664 664 return id(self)
665 665
666 666 def __eq__(self, other):
667 667 try:
668 668 return (type(self) == type(other) and self._path == other._path
669 669 and self._filenode == other._filenode)
670 670 except AttributeError:
671 671 return False
672 672
673 673 def __ne__(self, other):
674 674 return not (self == other)
675 675
676 676 def filerev(self):
677 677 return self._filerev
678 678 def filenode(self):
679 679 return self._filenode
680 680 def flags(self):
681 681 return self._changectx.flags(self._path)
682 682 def filelog(self):
683 683 return self._filelog
684 684 def rev(self):
685 685 return self._changeid
686 686 def linkrev(self):
687 687 return self._filelog.linkrev(self._filerev)
688 688 def node(self):
689 689 return self._changectx.node()
690 690 def hex(self):
691 691 return self._changectx.hex()
692 692 def user(self):
693 693 return self._changectx.user()
694 694 def date(self):
695 695 return self._changectx.date()
696 696 def files(self):
697 697 return self._changectx.files()
698 698 def description(self):
699 699 return self._changectx.description()
700 700 def branch(self):
701 701 return self._changectx.branch()
702 702 def extra(self):
703 703 return self._changectx.extra()
704 704 def phase(self):
705 705 return self._changectx.phase()
706 706 def phasestr(self):
707 707 return self._changectx.phasestr()
708 708 def manifest(self):
709 709 return self._changectx.manifest()
710 710 def changectx(self):
711 711 return self._changectx
712 712 def repo(self):
713 713 return self._repo
714 714
715 715 def path(self):
716 716 return self._path
717 717
718 718 def isbinary(self):
719 719 try:
720 720 return util.binary(self.data())
721 721 except IOError:
722 722 return False
723 723 def isexec(self):
724 724 return 'x' in self.flags()
725 725 def islink(self):
726 726 return 'l' in self.flags()
727 727
728 728 def cmp(self, fctx):
729 729 """compare with other file context
730 730
731 731 returns True if different than fctx.
732 732 """
733 733 if (fctx._filerev is None
734 734 and (self._repo._encodefilterpats
735 735 # if file data starts with '\1\n', empty metadata block is
736 736 # prepended, which adds 4 bytes to filelog.size().
737 737 or self.size() - 4 == fctx.size())
738 738 or self.size() == fctx.size()):
739 739 return self._filelog.cmp(self._filenode, fctx.data())
740 740
741 741 return True
742 742
743 743 def _adjustlinkrev(self, path, filelog, fnode, srcrev, inclusive=False):
744 744 """return the first ancestor of <srcrev> introducing <fnode>
745 745
746 746 If the linkrev of the file revision does not point to an ancestor of
747 747 srcrev, we'll walk down the ancestors until we find one introducing
748 748 this file revision.
749 749
750 750 :repo: a localrepository object (used to access changelog and manifest)
751 751 :path: the file path
752 752 :fnode: the nodeid of the file revision
753 753 :filelog: the filelog of this path
754 754 :srcrev: the changeset revision we search ancestors from
755 755 :inclusive: if true, the src revision will also be checked
756 756 """
757 757 repo = self._repo
758 758 cl = repo.unfiltered().changelog
759 759 ma = repo.manifest
760 760 # fetch the linkrev
761 761 fr = filelog.rev(fnode)
762 762 lkr = filelog.linkrev(fr)
763 763 # hack to reuse ancestor computation when searching for renames
764 764 memberanc = getattr(self, '_ancestrycontext', None)
765 765 iteranc = None
766 766 if srcrev is None:
767 767 # wctx case, used by workingfilectx during mergecopy
768 768 revs = [p.rev() for p in self._repo[None].parents()]
769 769 inclusive = True # we skipped the real (revless) source
770 770 else:
771 771 revs = [srcrev]
772 772 if memberanc is None:
773 773 memberanc = iteranc = cl.ancestors(revs, lkr,
774 774 inclusive=inclusive)
775 775 # check if this linkrev is an ancestor of srcrev
776 776 if lkr not in memberanc:
777 777 if iteranc is None:
778 778 iteranc = cl.ancestors(revs, lkr, inclusive=inclusive)
779 779 for a in iteranc:
780 780 ac = cl.read(a) # get changeset data (we avoid object creation)
781 781 if path in ac[3]: # checking the 'files' field.
782 782 # The file has been touched, check if the content is
783 783 # similar to the one we search for.
784 784 if fnode == ma.readfast(ac[0]).get(path):
785 785 return a
786 786 # In theory, we should never get out of that loop without a result.
787 787 # But if manifest uses a buggy file revision (not children of the
788 788 # one it replaces) we could. Such a buggy situation will likely
789 789 # result is crash somewhere else at to some point.
790 790 return lkr
791 791
792 792 def introrev(self):
793 793 """return the rev of the changeset which introduced this file revision
794 794
795 795 This method is different from linkrev because it take into account the
796 796 changeset the filectx was created from. It ensures the returned
797 797 revision is one of its ancestors. This prevents bugs from
798 798 'linkrev-shadowing' when a file revision is used by multiple
799 799 changesets.
800 800 """
801 801 lkr = self.linkrev()
802 802 attrs = vars(self)
803 803 noctx = not ('_changeid' in attrs or '_changectx' in attrs)
804 804 if noctx or self.rev() == lkr:
805 805 return self.linkrev()
806 806 return self._adjustlinkrev(self._path, self._filelog, self._filenode,
807 807 self.rev(), inclusive=True)
808 808
809 809 def _parentfilectx(self, path, fileid, filelog):
810 810 """create parent filectx keeping ancestry info for _adjustlinkrev()"""
811 811 fctx = filectx(self._repo, path, fileid=fileid, filelog=filelog)
812 812 if '_changeid' in vars(self) or '_changectx' in vars(self):
813 813 # If self is associated with a changeset (probably explicitly
814 814 # fed), ensure the created filectx is associated with a
815 815 # changeset that is an ancestor of self.changectx.
816 816 # This lets us later use _adjustlinkrev to get a correct link.
817 817 fctx._descendantrev = self.rev()
818 818 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
819 819 elif '_descendantrev' in vars(self):
820 820 # Otherwise propagate _descendantrev if we have one associated.
821 821 fctx._descendantrev = self._descendantrev
822 822 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
823 823 return fctx
824 824
825 825 def parents(self):
826 826 _path = self._path
827 827 fl = self._filelog
828 828 parents = self._filelog.parents(self._filenode)
829 829 pl = [(_path, node, fl) for node in parents if node != nullid]
830 830
831 831 r = fl.renamed(self._filenode)
832 832 if r:
833 833 # - In the simple rename case, both parent are nullid, pl is empty.
834 834 # - In case of merge, only one of the parent is null id and should
835 835 # be replaced with the rename information. This parent is -always-
836 836 # the first one.
837 837 #
838 838 # As null id have always been filtered out in the previous list
839 839 # comprehension, inserting to 0 will always result in "replacing
840 840 # first nullid parent with rename information.
841 841 pl.insert(0, (r[0], r[1], self._repo.file(r[0])))
842 842
843 843 return [self._parentfilectx(path, fnode, l) for path, fnode, l in pl]
844 844
845 845 def p1(self):
846 846 return self.parents()[0]
847 847
848 848 def p2(self):
849 849 p = self.parents()
850 850 if len(p) == 2:
851 851 return p[1]
852 852 return filectx(self._repo, self._path, fileid=-1, filelog=self._filelog)
853 853
854 854 def annotate(self, follow=False, linenumber=None, diffopts=None):
855 855 '''returns a list of tuples of (ctx, line) for each line
856 856 in the file, where ctx is the filectx of the node where
857 857 that line was last changed.
858 858 This returns tuples of ((ctx, linenumber), line) for each line,
859 859 if "linenumber" parameter is NOT "None".
860 860 In such tuples, linenumber means one at the first appearance
861 861 in the managed file.
862 862 To reduce annotation cost,
863 863 this returns fixed value(False is used) as linenumber,
864 864 if "linenumber" parameter is "False".'''
865 865
866 866 if linenumber is None:
867 867 def decorate(text, rev):
868 868 return ([rev] * len(text.splitlines()), text)
869 869 elif linenumber:
870 870 def decorate(text, rev):
871 871 size = len(text.splitlines())
872 872 return ([(rev, i) for i in xrange(1, size + 1)], text)
873 873 else:
874 874 def decorate(text, rev):
875 875 return ([(rev, False)] * len(text.splitlines()), text)
876 876
877 877 def pair(parent, child):
878 878 blocks = mdiff.allblocks(parent[1], child[1], opts=diffopts,
879 879 refine=True)
880 880 for (a1, a2, b1, b2), t in blocks:
881 881 # Changed blocks ('!') or blocks made only of blank lines ('~')
882 882 # belong to the child.
883 883 if t == '=':
884 884 child[0][b1:b2] = parent[0][a1:a2]
885 885 return child
886 886
887 887 getlog = util.lrucachefunc(lambda x: self._repo.file(x))
888 888
889 889 def parents(f):
890 890 # Cut _descendantrev here to mitigate the penalty of lazy linkrev
891 891 # adjustment. Otherwise, p._adjustlinkrev() would walk changelog
892 892 # from the topmost introrev (= srcrev) down to p.linkrev() if it
893 893 # isn't an ancestor of the srcrev.
894 894 f._changeid
895 895 pl = f.parents()
896 896
897 897 # Don't return renamed parents if we aren't following.
898 898 if not follow:
899 899 pl = [p for p in pl if p.path() == f.path()]
900 900
901 901 # renamed filectx won't have a filelog yet, so set it
902 902 # from the cache to save time
903 903 for p in pl:
904 904 if not '_filelog' in p.__dict__:
905 905 p._filelog = getlog(p.path())
906 906
907 907 return pl
908 908
909 909 # use linkrev to find the first changeset where self appeared
910 910 base = self
911 911 introrev = self.introrev()
912 912 if self.rev() != introrev:
913 913 base = self.filectx(self.filenode(), changeid=introrev)
914 914 if getattr(base, '_ancestrycontext', None) is None:
915 915 cl = self._repo.changelog
916 916 if introrev is None:
917 917 # wctx is not inclusive, but works because _ancestrycontext
918 918 # is used to test filelog revisions
919 919 ac = cl.ancestors([p.rev() for p in base.parents()],
920 920 inclusive=True)
921 921 else:
922 922 ac = cl.ancestors([introrev], inclusive=True)
923 923 base._ancestrycontext = ac
924 924
925 925 # This algorithm would prefer to be recursive, but Python is a
926 926 # bit recursion-hostile. Instead we do an iterative
927 927 # depth-first search.
928 928
929 929 visit = [base]
930 930 hist = {}
931 931 pcache = {}
932 932 needed = {base: 1}
933 933 while visit:
934 934 f = visit[-1]
935 935 pcached = f in pcache
936 936 if not pcached:
937 937 pcache[f] = parents(f)
938 938
939 939 ready = True
940 940 pl = pcache[f]
941 941 for p in pl:
942 942 if p not in hist:
943 943 ready = False
944 944 visit.append(p)
945 945 if not pcached:
946 946 needed[p] = needed.get(p, 0) + 1
947 947 if ready:
948 948 visit.pop()
949 949 reusable = f in hist
950 950 if reusable:
951 951 curr = hist[f]
952 952 else:
953 953 curr = decorate(f.data(), f)
954 954 for p in pl:
955 955 if not reusable:
956 956 curr = pair(hist[p], curr)
957 957 if needed[p] == 1:
958 958 del hist[p]
959 959 del needed[p]
960 960 else:
961 961 needed[p] -= 1
962 962
963 963 hist[f] = curr
964 964 pcache[f] = []
965 965
966 966 return zip(hist[base][0], hist[base][1].splitlines(True))
967 967
968 968 def ancestors(self, followfirst=False):
969 969 visit = {}
970 970 c = self
971 971 if followfirst:
972 972 cut = 1
973 973 else:
974 974 cut = None
975 975
976 976 while True:
977 977 for parent in c.parents()[:cut]:
978 978 visit[(parent.linkrev(), parent.filenode())] = parent
979 979 if not visit:
980 980 break
981 981 c = visit.pop(max(visit))
982 982 yield c
983 983
984 984 class filectx(basefilectx):
985 985 """A filecontext object makes access to data related to a particular
986 986 filerevision convenient."""
987 987 def __init__(self, repo, path, changeid=None, fileid=None,
988 988 filelog=None, changectx=None):
989 989 """changeid can be a changeset revision, node, or tag.
990 990 fileid can be a file revision or node."""
991 991 self._repo = repo
992 992 self._path = path
993 993
994 994 assert (changeid is not None
995 995 or fileid is not None
996 996 or changectx is not None), \
997 997 ("bad args: changeid=%r, fileid=%r, changectx=%r"
998 998 % (changeid, fileid, changectx))
999 999
1000 1000 if filelog is not None:
1001 1001 self._filelog = filelog
1002 1002
1003 1003 if changeid is not None:
1004 1004 self._changeid = changeid
1005 1005 if changectx is not None:
1006 1006 self._changectx = changectx
1007 1007 if fileid is not None:
1008 1008 self._fileid = fileid
1009 1009
1010 1010 @propertycache
1011 1011 def _changectx(self):
1012 1012 try:
1013 1013 return changectx(self._repo, self._changeid)
1014 1014 except error.FilteredRepoLookupError:
1015 1015 # Linkrev may point to any revision in the repository. When the
1016 1016 # repository is filtered this may lead to `filectx` trying to build
1017 1017 # `changectx` for filtered revision. In such case we fallback to
1018 1018 # creating `changectx` on the unfiltered version of the reposition.
1019 1019 # This fallback should not be an issue because `changectx` from
1020 1020 # `filectx` are not used in complex operations that care about
1021 1021 # filtering.
1022 1022 #
1023 1023 # This fallback is a cheap and dirty fix that prevent several
1024 1024 # crashes. It does not ensure the behavior is correct. However the
1025 1025 # behavior was not correct before filtering either and "incorrect
1026 1026 # behavior" is seen as better as "crash"
1027 1027 #
1028 1028 # Linkrevs have several serious troubles with filtering that are
1029 1029 # complicated to solve. Proper handling of the issue here should be
1030 1030 # considered when solving linkrev issue are on the table.
1031 1031 return changectx(self._repo.unfiltered(), self._changeid)
1032 1032
1033 1033 def filectx(self, fileid, changeid=None):
1034 1034 '''opens an arbitrary revision of the file without
1035 1035 opening a new filelog'''
1036 1036 return filectx(self._repo, self._path, fileid=fileid,
1037 1037 filelog=self._filelog, changeid=changeid)
1038 1038
1039 1039 def data(self):
1040 1040 try:
1041 1041 return self._filelog.read(self._filenode)
1042 1042 except error.CensoredNodeError:
1043 1043 if self._repo.ui.config("censor", "policy", "abort") == "ignore":
1044 1044 return ""
1045 1045 raise util.Abort(_("censored node: %s") % short(self._filenode),
1046 1046 hint=_("set censor.policy to ignore errors"))
1047 1047
1048 1048 def size(self):
1049 1049 return self._filelog.size(self._filerev)
1050 1050
1051 1051 def renamed(self):
1052 1052 """check if file was actually renamed in this changeset revision
1053 1053
1054 1054 If rename logged in file revision, we report copy for changeset only
1055 1055 if file revisions linkrev points back to the changeset in question
1056 1056 or both changeset parents contain different file revisions.
1057 1057 """
1058 1058
1059 1059 renamed = self._filelog.renamed(self._filenode)
1060 1060 if not renamed:
1061 1061 return renamed
1062 1062
1063 1063 if self.rev() == self.linkrev():
1064 1064 return renamed
1065 1065
1066 1066 name = self.path()
1067 1067 fnode = self._filenode
1068 1068 for p in self._changectx.parents():
1069 1069 try:
1070 1070 if fnode == p.filenode(name):
1071 1071 return None
1072 1072 except error.LookupError:
1073 1073 pass
1074 1074 return renamed
1075 1075
1076 1076 def children(self):
1077 1077 # hard for renames
1078 1078 c = self._filelog.children(self._filenode)
1079 1079 return [filectx(self._repo, self._path, fileid=x,
1080 1080 filelog=self._filelog) for x in c]
1081 1081
1082 1082 class committablectx(basectx):
1083 1083 """A committablectx object provides common functionality for a context that
1084 1084 wants the ability to commit, e.g. workingctx or memctx."""
1085 1085 def __init__(self, repo, text="", user=None, date=None, extra=None,
1086 1086 changes=None):
1087 1087 self._repo = repo
1088 1088 self._rev = None
1089 1089 self._node = None
1090 1090 self._text = text
1091 1091 if date:
1092 1092 self._date = util.parsedate(date)
1093 1093 if user:
1094 1094 self._user = user
1095 1095 if changes:
1096 1096 self._status = changes
1097 1097
1098 1098 self._extra = {}
1099 1099 if extra:
1100 1100 self._extra = extra.copy()
1101 1101 if 'branch' not in self._extra:
1102 1102 try:
1103 1103 branch = encoding.fromlocal(self._repo.dirstate.branch())
1104 1104 except UnicodeDecodeError:
1105 1105 raise util.Abort(_('branch name not in UTF-8!'))
1106 1106 self._extra['branch'] = branch
1107 1107 if self._extra['branch'] == '':
1108 1108 self._extra['branch'] = 'default'
1109 1109
1110 1110 def __str__(self):
1111 1111 return str(self._parents[0]) + "+"
1112 1112
1113 1113 def __nonzero__(self):
1114 1114 return True
1115 1115
1116 1116 def _buildflagfunc(self):
1117 1117 # Create a fallback function for getting file flags when the
1118 1118 # filesystem doesn't support them
1119 1119
1120 1120 copiesget = self._repo.dirstate.copies().get
1121 1121
1122 1122 if len(self._parents) < 2:
1123 1123 # when we have one parent, it's easy: copy from parent
1124 1124 man = self._parents[0].manifest()
1125 1125 def func(f):
1126 1126 f = copiesget(f, f)
1127 1127 return man.flags(f)
1128 1128 else:
1129 1129 # merges are tricky: we try to reconstruct the unstored
1130 1130 # result from the merge (issue1802)
1131 1131 p1, p2 = self._parents
1132 1132 pa = p1.ancestor(p2)
1133 1133 m1, m2, ma = p1.manifest(), p2.manifest(), pa.manifest()
1134 1134
1135 1135 def func(f):
1136 1136 f = copiesget(f, f) # may be wrong for merges with copies
1137 1137 fl1, fl2, fla = m1.flags(f), m2.flags(f), ma.flags(f)
1138 1138 if fl1 == fl2:
1139 1139 return fl1
1140 1140 if fl1 == fla:
1141 1141 return fl2
1142 1142 if fl2 == fla:
1143 1143 return fl1
1144 1144 return '' # punt for conflicts
1145 1145
1146 1146 return func
1147 1147
1148 1148 @propertycache
1149 1149 def _flagfunc(self):
1150 1150 return self._repo.dirstate.flagfunc(self._buildflagfunc)
1151 1151
1152 1152 @propertycache
1153 1153 def _manifest(self):
1154 1154 """generate a manifest corresponding to the values in self._status
1155 1155
1156 1156 This reuse the file nodeid from parent, but we append an extra letter
1157 1157 when modified. Modified files get an extra 'm' while added files get
1158 1158 an extra 'a'. This is used by manifests merge to see that files
1159 1159 are different and by update logic to avoid deleting newly added files.
1160 1160 """
1161 1161
1162 1162 man1 = self._parents[0].manifest()
1163 1163 man = man1.copy()
1164 1164 if len(self._parents) > 1:
1165 1165 man2 = self.p2().manifest()
1166 1166 def getman(f):
1167 1167 if f in man1:
1168 1168 return man1
1169 1169 return man2
1170 1170 else:
1171 1171 getman = lambda f: man1
1172 1172
1173 1173 copied = self._repo.dirstate.copies()
1174 1174 ff = self._flagfunc
1175 1175 for i, l in (("a", self._status.added), ("m", self._status.modified)):
1176 1176 for f in l:
1177 1177 orig = copied.get(f, f)
1178 1178 man[f] = getman(orig).get(orig, nullid) + i
1179 1179 try:
1180 1180 man.setflag(f, ff(f))
1181 1181 except OSError:
1182 1182 pass
1183 1183
1184 1184 for f in self._status.deleted + self._status.removed:
1185 1185 if f in man:
1186 1186 del man[f]
1187 1187
1188 1188 return man
1189 1189
1190 1190 @propertycache
1191 1191 def _status(self):
1192 1192 return self._repo.status()
1193 1193
1194 1194 @propertycache
1195 1195 def _user(self):
1196 1196 return self._repo.ui.username()
1197 1197
1198 1198 @propertycache
1199 1199 def _date(self):
1200 1200 return util.makedate()
1201 1201
1202 1202 def subrev(self, subpath):
1203 1203 return None
1204 1204
1205 1205 def manifestnode(self):
1206 1206 return None
1207 1207 def user(self):
1208 1208 return self._user or self._repo.ui.username()
1209 1209 def date(self):
1210 1210 return self._date
1211 1211 def description(self):
1212 1212 return self._text
1213 1213 def files(self):
1214 1214 return sorted(self._status.modified + self._status.added +
1215 1215 self._status.removed)
1216 1216
1217 1217 def modified(self):
1218 1218 return self._status.modified
1219 1219 def added(self):
1220 1220 return self._status.added
1221 1221 def removed(self):
1222 1222 return self._status.removed
1223 1223 def deleted(self):
1224 1224 return self._status.deleted
1225 1225 def branch(self):
1226 1226 return encoding.tolocal(self._extra['branch'])
1227 1227 def closesbranch(self):
1228 1228 return 'close' in self._extra
1229 1229 def extra(self):
1230 1230 return self._extra
1231 1231
1232 1232 def tags(self):
1233 1233 t = []
1234 1234 for p in self.parents():
1235 1235 t.extend(p.tags())
1236 1236 return t
1237 1237
1238 1238 def bookmarks(self):
1239 1239 b = []
1240 1240 for p in self.parents():
1241 1241 b.extend(p.bookmarks())
1242 1242 return b
1243 1243
1244 1244 def phase(self):
1245 1245 phase = phases.draft # default phase to draft
1246 1246 for p in self.parents():
1247 1247 phase = max(phase, p.phase())
1248 1248 return phase
1249 1249
1250 1250 def hidden(self):
1251 1251 return False
1252 1252
1253 1253 def children(self):
1254 1254 return []
1255 1255
1256 1256 def flags(self, path):
1257 1257 if '_manifest' in self.__dict__:
1258 1258 try:
1259 1259 return self._manifest.flags(path)
1260 1260 except KeyError:
1261 1261 return ''
1262 1262
1263 1263 try:
1264 1264 return self._flagfunc(path)
1265 1265 except OSError:
1266 1266 return ''
1267 1267
1268 1268 def ancestor(self, c2):
1269 1269 """return the "best" ancestor context of self and c2"""
1270 1270 return self._parents[0].ancestor(c2) # punt on two parents for now
1271 1271
1272 1272 def walk(self, match):
1273 1273 '''Generates matching file names.'''
1274 1274 return sorted(self._repo.dirstate.walk(match, sorted(self.substate),
1275 1275 True, False))
1276 1276
1277 1277 def matches(self, match):
1278 1278 return sorted(self._repo.dirstate.matches(match))
1279 1279
1280 1280 def ancestors(self):
1281 1281 for p in self._parents:
1282 1282 yield p
1283 1283 for a in self._repo.changelog.ancestors(
1284 1284 [p.rev() for p in self._parents]):
1285 1285 yield changectx(self._repo, a)
1286 1286
1287 1287 def markcommitted(self, node):
1288 1288 """Perform post-commit cleanup necessary after committing this ctx
1289 1289
1290 1290 Specifically, this updates backing stores this working context
1291 1291 wraps to reflect the fact that the changes reflected by this
1292 1292 workingctx have been committed. For example, it marks
1293 1293 modified and added files as normal in the dirstate.
1294 1294
1295 1295 """
1296 1296
1297 1297 self._repo.dirstate.beginparentchange()
1298 1298 for f in self.modified() + self.added():
1299 1299 self._repo.dirstate.normal(f)
1300 1300 for f in self.removed():
1301 1301 self._repo.dirstate.drop(f)
1302 1302 self._repo.dirstate.setparents(node)
1303 1303 self._repo.dirstate.endparentchange()
1304 1304
1305 1305 class workingctx(committablectx):
1306 1306 """A workingctx object makes access to data related to
1307 1307 the current working directory convenient.
1308 1308 date - any valid date string or (unixtime, offset), or None.
1309 1309 user - username string, or None.
1310 1310 extra - a dictionary of extra values, or None.
1311 1311 changes - a list of file lists as returned by localrepo.status()
1312 1312 or None to use the repository status.
1313 1313 """
1314 1314 def __init__(self, repo, text="", user=None, date=None, extra=None,
1315 1315 changes=None):
1316 1316 super(workingctx, self).__init__(repo, text, user, date, extra, changes)
1317 1317
1318 1318 def __iter__(self):
1319 1319 d = self._repo.dirstate
1320 1320 for f in d:
1321 1321 if d[f] != 'r':
1322 1322 yield f
1323 1323
1324 1324 def __contains__(self, key):
1325 1325 return self._repo.dirstate[key] not in "?r"
1326 1326
1327 1327 @propertycache
1328 1328 def _parents(self):
1329 1329 p = self._repo.dirstate.parents()
1330 1330 if p[1] == nullid:
1331 1331 p = p[:-1]
1332 1332 return [changectx(self._repo, x) for x in p]
1333 1333
1334 1334 def filectx(self, path, filelog=None):
1335 1335 """get a file context from the working directory"""
1336 1336 return workingfilectx(self._repo, path, workingctx=self,
1337 1337 filelog=filelog)
1338 1338
1339 1339 def dirty(self, missing=False, merge=True, branch=True):
1340 1340 "check whether a working directory is modified"
1341 1341 # check subrepos first
1342 1342 for s in sorted(self.substate):
1343 1343 if self.sub(s).dirty():
1344 1344 return True
1345 1345 # check current working dir
1346 1346 return ((merge and self.p2()) or
1347 1347 (branch and self.branch() != self.p1().branch()) or
1348 1348 self.modified() or self.added() or self.removed() or
1349 1349 (missing and self.deleted()))
1350 1350
1351 1351 def add(self, list, prefix=""):
1352 1352 join = lambda f: os.path.join(prefix, f)
1353 1353 wlock = self._repo.wlock()
1354 1354 ui, ds = self._repo.ui, self._repo.dirstate
1355 1355 try:
1356 1356 rejected = []
1357 1357 lstat = self._repo.wvfs.lstat
1358 1358 for f in list:
1359 1359 scmutil.checkportable(ui, join(f))
1360 1360 try:
1361 1361 st = lstat(f)
1362 1362 except OSError:
1363 1363 ui.warn(_("%s does not exist!\n") % join(f))
1364 1364 rejected.append(f)
1365 1365 continue
1366 1366 if st.st_size > 10000000:
1367 1367 ui.warn(_("%s: up to %d MB of RAM may be required "
1368 1368 "to manage this file\n"
1369 1369 "(use 'hg revert %s' to cancel the "
1370 1370 "pending addition)\n")
1371 1371 % (f, 3 * st.st_size // 1000000, join(f)))
1372 1372 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1373 1373 ui.warn(_("%s not added: only files and symlinks "
1374 1374 "supported currently\n") % join(f))
1375 1375 rejected.append(f)
1376 1376 elif ds[f] in 'amn':
1377 1377 ui.warn(_("%s already tracked!\n") % join(f))
1378 1378 elif ds[f] == 'r':
1379 1379 ds.normallookup(f)
1380 1380 else:
1381 1381 ds.add(f)
1382 1382 return rejected
1383 1383 finally:
1384 1384 wlock.release()
1385 1385
1386 1386 def forget(self, files, prefix=""):
1387 1387 join = lambda f: os.path.join(prefix, f)
1388 1388 wlock = self._repo.wlock()
1389 1389 try:
1390 1390 rejected = []
1391 1391 for f in files:
1392 1392 if f not in self._repo.dirstate:
1393 1393 self._repo.ui.warn(_("%s not tracked!\n") % join(f))
1394 1394 rejected.append(f)
1395 1395 elif self._repo.dirstate[f] != 'a':
1396 1396 self._repo.dirstate.remove(f)
1397 1397 else:
1398 1398 self._repo.dirstate.drop(f)
1399 1399 return rejected
1400 1400 finally:
1401 1401 wlock.release()
1402 1402
1403 1403 def undelete(self, list):
1404 1404 pctxs = self.parents()
1405 1405 wlock = self._repo.wlock()
1406 1406 try:
1407 1407 for f in list:
1408 1408 if self._repo.dirstate[f] != 'r':
1409 1409 self._repo.ui.warn(_("%s not removed!\n") % f)
1410 1410 else:
1411 1411 fctx = f in pctxs[0] and pctxs[0][f] or pctxs[1][f]
1412 1412 t = fctx.data()
1413 1413 self._repo.wwrite(f, t, fctx.flags())
1414 1414 self._repo.dirstate.normal(f)
1415 1415 finally:
1416 1416 wlock.release()
1417 1417
1418 1418 def copy(self, source, dest):
1419 1419 try:
1420 1420 st = self._repo.wvfs.lstat(dest)
1421 1421 except OSError, err:
1422 1422 if err.errno != errno.ENOENT:
1423 1423 raise
1424 1424 self._repo.ui.warn(_("%s does not exist!\n") % dest)
1425 1425 return
1426 1426 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1427 1427 self._repo.ui.warn(_("copy failed: %s is not a file or a "
1428 1428 "symbolic link\n") % dest)
1429 1429 else:
1430 1430 wlock = self._repo.wlock()
1431 1431 try:
1432 1432 if self._repo.dirstate[dest] in '?':
1433 1433 self._repo.dirstate.add(dest)
1434 1434 elif self._repo.dirstate[dest] in 'r':
1435 1435 self._repo.dirstate.normallookup(dest)
1436 1436 self._repo.dirstate.copy(source, dest)
1437 1437 finally:
1438 1438 wlock.release()
1439 1439
1440 1440 def match(self, pats=[], include=None, exclude=None, default='glob'):
1441 1441 r = self._repo
1442 1442
1443 1443 # Only a case insensitive filesystem needs magic to translate user input
1444 1444 # to actual case in the filesystem.
1445 1445 if not util.checkcase(r.root):
1446 1446 return matchmod.icasefsmatcher(r.root, r.getcwd(), pats, include,
1447 1447 exclude, default, r.auditor, self)
1448 1448 return matchmod.match(r.root, r.getcwd(), pats,
1449 1449 include, exclude, default,
1450 1450 auditor=r.auditor, ctx=self)
1451 1451
1452 1452 def _filtersuspectsymlink(self, files):
1453 1453 if not files or self._repo.dirstate._checklink:
1454 1454 return files
1455 1455
1456 1456 # Symlink placeholders may get non-symlink-like contents
1457 1457 # via user error or dereferencing by NFS or Samba servers,
1458 1458 # so we filter out any placeholders that don't look like a
1459 1459 # symlink
1460 1460 sane = []
1461 1461 for f in files:
1462 1462 if self.flags(f) == 'l':
1463 1463 d = self[f].data()
1464 1464 if d == '' or len(d) >= 1024 or '\n' in d or util.binary(d):
1465 1465 self._repo.ui.debug('ignoring suspect symlink placeholder'
1466 1466 ' "%s"\n' % f)
1467 1467 continue
1468 1468 sane.append(f)
1469 1469 return sane
1470 1470
1471 1471 def _checklookup(self, files):
1472 1472 # check for any possibly clean files
1473 1473 if not files:
1474 1474 return [], []
1475 1475
1476 1476 modified = []
1477 1477 fixup = []
1478 1478 pctx = self._parents[0]
1479 1479 # do a full compare of any files that might have changed
1480 1480 for f in sorted(files):
1481 1481 if (f not in pctx or self.flags(f) != pctx.flags(f)
1482 1482 or pctx[f].cmp(self[f])):
1483 1483 modified.append(f)
1484 1484 else:
1485 1485 fixup.append(f)
1486 1486
1487 1487 # update dirstate for files that are actually clean
1488 1488 if fixup:
1489 1489 try:
1490 1490 # updating the dirstate is optional
1491 1491 # so we don't wait on the lock
1492 1492 # wlock can invalidate the dirstate, so cache normal _after_
1493 1493 # taking the lock
1494 1494 wlock = self._repo.wlock(False)
1495 1495 normal = self._repo.dirstate.normal
1496 1496 try:
1497 1497 for f in fixup:
1498 1498 normal(f)
1499 1499 finally:
1500 1500 wlock.release()
1501 1501 except error.LockError:
1502 1502 pass
1503 1503 return modified, fixup
1504 1504
1505 1505 def _manifestmatches(self, match, s):
1506 1506 """Slow path for workingctx
1507 1507
1508 1508 The fast path is when we compare the working directory to its parent
1509 1509 which means this function is comparing with a non-parent; therefore we
1510 1510 need to build a manifest and return what matches.
1511 1511 """
1512 1512 mf = self._repo['.']._manifestmatches(match, s)
1513 1513 for f in s.modified + s.added:
1514 1514 mf[f] = _newnode
1515 1515 mf.setflag(f, self.flags(f))
1516 1516 for f in s.removed:
1517 1517 if f in mf:
1518 1518 del mf[f]
1519 1519 return mf
1520 1520
1521 1521 def _dirstatestatus(self, match=None, ignored=False, clean=False,
1522 1522 unknown=False):
1523 1523 '''Gets the status from the dirstate -- internal use only.'''
1524 1524 listignored, listclean, listunknown = ignored, clean, unknown
1525 1525 match = match or matchmod.always(self._repo.root, self._repo.getcwd())
1526 1526 subrepos = []
1527 1527 if '.hgsub' in self:
1528 1528 subrepos = sorted(self.substate)
1529 1529 cmp, s = self._repo.dirstate.status(match, subrepos, listignored,
1530 1530 listclean, listunknown)
1531 1531
1532 1532 # check for any possibly clean files
1533 1533 if cmp:
1534 1534 modified2, fixup = self._checklookup(cmp)
1535 1535 s.modified.extend(modified2)
1536 1536
1537 1537 # update dirstate for files that are actually clean
1538 1538 if fixup and listclean:
1539 1539 s.clean.extend(fixup)
1540 1540
1541 1541 if match.always():
1542 1542 # cache for performance
1543 1543 if s.unknown or s.ignored or s.clean:
1544 1544 # "_status" is cached with list*=False in the normal route
1545 1545 self._status = scmutil.status(s.modified, s.added, s.removed,
1546 1546 s.deleted, [], [], [])
1547 1547 else:
1548 1548 self._status = s
1549 1549
1550 1550 return s
1551 1551
1552 1552 def _buildstatus(self, other, s, match, listignored, listclean,
1553 1553 listunknown):
1554 1554 """build a status with respect to another context
1555 1555
1556 1556 This includes logic for maintaining the fast path of status when
1557 1557 comparing the working directory against its parent, which is to skip
1558 1558 building a new manifest if self (working directory) is not comparing
1559 1559 against its parent (repo['.']).
1560 1560 """
1561 1561 s = self._dirstatestatus(match, listignored, listclean, listunknown)
1562 1562 # Filter out symlinks that, in the case of FAT32 and NTFS filesystems,
1563 1563 # might have accidentally ended up with the entire contents of the file
1564 1564 # they are supposed to be linking to.
1565 1565 s.modified[:] = self._filtersuspectsymlink(s.modified)
1566 1566 if other != self._repo['.']:
1567 1567 s = super(workingctx, self)._buildstatus(other, s, match,
1568 1568 listignored, listclean,
1569 1569 listunknown)
1570 1570 return s
1571 1571
1572 1572 def _matchstatus(self, other, match):
1573 1573 """override the match method with a filter for directory patterns
1574 1574
1575 1575 We use inheritance to customize the match.bad method only in cases of
1576 1576 workingctx since it belongs only to the working directory when
1577 1577 comparing against the parent changeset.
1578 1578
1579 1579 If we aren't comparing against the working directory's parent, then we
1580 1580 just use the default match object sent to us.
1581 1581 """
1582 1582 superself = super(workingctx, self)
1583 1583 match = superself._matchstatus(other, match)
1584 1584 if other != self._repo['.']:
1585 1585 def bad(f, msg):
1586 1586 # 'f' may be a directory pattern from 'match.files()',
1587 1587 # so 'f not in ctx1' is not enough
1588 1588 if f not in other and not other.hasdir(f):
1589 1589 self._repo.ui.warn('%s: %s\n' %
1590 1590 (self._repo.dirstate.pathto(f), msg))
1591 1591 match.bad = bad
1592 1592 return match
1593 1593
1594 1594 class committablefilectx(basefilectx):
1595 1595 """A committablefilectx provides common functionality for a file context
1596 1596 that wants the ability to commit, e.g. workingfilectx or memfilectx."""
1597 1597 def __init__(self, repo, path, filelog=None, ctx=None):
1598 1598 self._repo = repo
1599 1599 self._path = path
1600 1600 self._changeid = None
1601 1601 self._filerev = self._filenode = None
1602 1602
1603 1603 if filelog is not None:
1604 1604 self._filelog = filelog
1605 1605 if ctx:
1606 1606 self._changectx = ctx
1607 1607
1608 1608 def __nonzero__(self):
1609 1609 return True
1610 1610
1611 1611 def linkrev(self):
1612 1612 # linked to self._changectx no matter if file is modified or not
1613 1613 return self.rev()
1614 1614
1615 1615 def parents(self):
1616 1616 '''return parent filectxs, following copies if necessary'''
1617 1617 def filenode(ctx, path):
1618 1618 return ctx._manifest.get(path, nullid)
1619 1619
1620 1620 path = self._path
1621 1621 fl = self._filelog
1622 1622 pcl = self._changectx._parents
1623 1623 renamed = self.renamed()
1624 1624
1625 1625 if renamed:
1626 1626 pl = [renamed + (None,)]
1627 1627 else:
1628 1628 pl = [(path, filenode(pcl[0], path), fl)]
1629 1629
1630 1630 for pc in pcl[1:]:
1631 1631 pl.append((path, filenode(pc, path), fl))
1632 1632
1633 1633 return [self._parentfilectx(p, fileid=n, filelog=l)
1634 1634 for p, n, l in pl if n != nullid]
1635 1635
1636 1636 def children(self):
1637 1637 return []
1638 1638
1639 1639 class workingfilectx(committablefilectx):
1640 1640 """A workingfilectx object makes access to data related to a particular
1641 1641 file in the working directory convenient."""
1642 1642 def __init__(self, repo, path, filelog=None, workingctx=None):
1643 1643 super(workingfilectx, self).__init__(repo, path, filelog, workingctx)
1644 1644
1645 1645 @propertycache
1646 1646 def _changectx(self):
1647 1647 return workingctx(self._repo)
1648 1648
1649 1649 def data(self):
1650 1650 return self._repo.wread(self._path)
1651 1651 def renamed(self):
1652 1652 rp = self._repo.dirstate.copied(self._path)
1653 1653 if not rp:
1654 1654 return None
1655 1655 return rp, self._changectx._parents[0]._manifest.get(rp, nullid)
1656 1656
1657 1657 def size(self):
1658 1658 return self._repo.wvfs.lstat(self._path).st_size
1659 1659 def date(self):
1660 1660 t, tz = self._changectx.date()
1661 1661 try:
1662 1662 return (int(self._repo.wvfs.lstat(self._path).st_mtime), tz)
1663 1663 except OSError, err:
1664 1664 if err.errno != errno.ENOENT:
1665 1665 raise
1666 1666 return (t, tz)
1667 1667
1668 1668 def cmp(self, fctx):
1669 1669 """compare with other file context
1670 1670
1671 1671 returns True if different than fctx.
1672 1672 """
1673 1673 # fctx should be a filectx (not a workingfilectx)
1674 1674 # invert comparison to reuse the same code path
1675 1675 return fctx.cmp(self)
1676 1676
1677 1677 def remove(self, ignoremissing=False):
1678 1678 """wraps unlink for a repo's working directory"""
1679 1679 util.unlinkpath(self._repo.wjoin(self._path), ignoremissing)
1680 1680
1681 1681 def write(self, data, flags):
1682 1682 """wraps repo.wwrite"""
1683 1683 self._repo.wwrite(self._path, data, flags)
1684 1684
1685 1685 class workingcommitctx(workingctx):
1686 1686 """A workingcommitctx object makes access to data related to
1687 1687 the revision being committed convenient.
1688 1688
1689 1689 This hides changes in the working directory, if they aren't
1690 1690 committed in this context.
1691 1691 """
1692 1692 def __init__(self, repo, changes,
1693 1693 text="", user=None, date=None, extra=None):
1694 1694 super(workingctx, self).__init__(repo, text, user, date, extra,
1695 1695 changes)
1696 1696
1697 1697 def _dirstatestatus(self, match=None, ignored=False, clean=False,
1698 1698 unknown=False):
1699 1699 """Return matched files only in ``self._status``
1700 1700
1701 1701 Uncommitted files appear "clean" via this context, even if
1702 1702 they aren't actually so in the working directory.
1703 1703 """
1704 1704 match = match or matchmod.always(self._repo.root, self._repo.getcwd())
1705 1705 if clean:
1706 1706 clean = [f for f in self._manifest if f not in self._changedset]
1707 1707 else:
1708 1708 clean = []
1709 1709 return scmutil.status([f for f in self._status.modified if match(f)],
1710 1710 [f for f in self._status.added if match(f)],
1711 1711 [f for f in self._status.removed if match(f)],
1712 1712 [], [], [], clean)
1713 1713
1714 1714 @propertycache
1715 1715 def _changedset(self):
1716 1716 """Return the set of files changed in this context
1717 1717 """
1718 1718 changed = set(self._status.modified)
1719 1719 changed.update(self._status.added)
1720 1720 changed.update(self._status.removed)
1721 1721 return changed
1722 1722
1723 1723 class memctx(committablectx):
1724 1724 """Use memctx to perform in-memory commits via localrepo.commitctx().
1725 1725
1726 1726 Revision information is supplied at initialization time while
1727 1727 related files data and is made available through a callback
1728 1728 mechanism. 'repo' is the current localrepo, 'parents' is a
1729 1729 sequence of two parent revisions identifiers (pass None for every
1730 1730 missing parent), 'text' is the commit message and 'files' lists
1731 1731 names of files touched by the revision (normalized and relative to
1732 1732 repository root).
1733 1733
1734 1734 filectxfn(repo, memctx, path) is a callable receiving the
1735 1735 repository, the current memctx object and the normalized path of
1736 1736 requested file, relative to repository root. It is fired by the
1737 1737 commit function for every file in 'files', but calls order is
1738 1738 undefined. If the file is available in the revision being
1739 1739 committed (updated or added), filectxfn returns a memfilectx
1740 1740 object. If the file was removed, filectxfn raises an
1741 1741 IOError. Moved files are represented by marking the source file
1742 1742 removed and the new file added with copy information (see
1743 1743 memfilectx).
1744 1744
1745 1745 user receives the committer name and defaults to current
1746 1746 repository username, date is the commit date in any format
1747 1747 supported by util.parsedate() and defaults to current date, extra
1748 1748 is a dictionary of metadata or is left empty.
1749 1749 """
1750 1750
1751 1751 # Mercurial <= 3.1 expects the filectxfn to raise IOError for missing files.
1752 1752 # Extensions that need to retain compatibility across Mercurial 3.1 can use
1753 1753 # this field to determine what to do in filectxfn.
1754 1754 _returnnoneformissingfiles = True
1755 1755
1756 1756 def __init__(self, repo, parents, text, files, filectxfn, user=None,
1757 1757 date=None, extra=None, editor=False):
1758 1758 super(memctx, self).__init__(repo, text, user, date, extra)
1759 1759 self._rev = None
1760 1760 self._node = None
1761 1761 parents = [(p or nullid) for p in parents]
1762 1762 p1, p2 = parents
1763 1763 self._parents = [changectx(self._repo, p) for p in (p1, p2)]
1764 1764 files = sorted(set(files))
1765 1765 self._files = files
1766 1766 self.substate = {}
1767 1767
1768 1768 # if store is not callable, wrap it in a function
1769 1769 if not callable(filectxfn):
1770 1770 def getfilectx(repo, memctx, path):
1771 1771 fctx = filectxfn[path]
1772 1772 # this is weird but apparently we only keep track of one parent
1773 1773 # (why not only store that instead of a tuple?)
1774 1774 copied = fctx.renamed()
1775 1775 if copied:
1776 1776 copied = copied[0]
1777 1777 return memfilectx(repo, path, fctx.data(),
1778 1778 islink=fctx.islink(), isexec=fctx.isexec(),
1779 1779 copied=copied, memctx=memctx)
1780 1780 self._filectxfn = getfilectx
1781 1781 else:
1782 1782 # "util.cachefunc" reduces invocation of possibly expensive
1783 1783 # "filectxfn" for performance (e.g. converting from another VCS)
1784 1784 self._filectxfn = util.cachefunc(filectxfn)
1785 1785
1786 1786 if extra:
1787 1787 self._extra = extra.copy()
1788 1788 else:
1789 1789 self._extra = {}
1790 1790
1791 1791 if self._extra.get('branch', '') == '':
1792 1792 self._extra['branch'] = 'default'
1793 1793
1794 1794 if editor:
1795 1795 self._text = editor(self._repo, self, [])
1796 1796 self._repo.savecommitmessage(self._text)
1797 1797
1798 1798 def filectx(self, path, filelog=None):
1799 1799 """get a file context from the working directory
1800 1800
1801 1801 Returns None if file doesn't exist and should be removed."""
1802 1802 return self._filectxfn(self._repo, self, path)
1803 1803
1804 1804 def commit(self):
1805 1805 """commit context to the repo"""
1806 1806 return self._repo.commitctx(self)
1807 1807
1808 1808 @propertycache
1809 1809 def _manifest(self):
1810 1810 """generate a manifest based on the return values of filectxfn"""
1811 1811
1812 1812 # keep this simple for now; just worry about p1
1813 1813 pctx = self._parents[0]
1814 1814 man = pctx.manifest().copy()
1815 1815
1816 1816 for f in self._status.modified:
1817 1817 p1node = nullid
1818 1818 p2node = nullid
1819 1819 p = pctx[f].parents() # if file isn't in pctx, check p2?
1820 1820 if len(p) > 0:
1821 1821 p1node = p[0].node()
1822 1822 if len(p) > 1:
1823 1823 p2node = p[1].node()
1824 1824 man[f] = revlog.hash(self[f].data(), p1node, p2node)
1825 1825
1826 1826 for f in self._status.added:
1827 1827 man[f] = revlog.hash(self[f].data(), nullid, nullid)
1828 1828
1829 1829 for f in self._status.removed:
1830 1830 if f in man:
1831 1831 del man[f]
1832 1832
1833 1833 return man
1834 1834
1835 1835 @propertycache
1836 1836 def _status(self):
1837 1837 """Calculate exact status from ``files`` specified at construction
1838 1838 """
1839 1839 man1 = self.p1().manifest()
1840 1840 p2 = self._parents[1]
1841 1841 # "1 < len(self._parents)" can't be used for checking
1842 1842 # existence of the 2nd parent, because "memctx._parents" is
1843 1843 # explicitly initialized by the list, of which length is 2.
1844 1844 if p2.node() != nullid:
1845 1845 man2 = p2.manifest()
1846 1846 managing = lambda f: f in man1 or f in man2
1847 1847 else:
1848 1848 managing = lambda f: f in man1
1849 1849
1850 1850 modified, added, removed = [], [], []
1851 1851 for f in self._files:
1852 1852 if not managing(f):
1853 1853 added.append(f)
1854 1854 elif self[f]:
1855 1855 modified.append(f)
1856 1856 else:
1857 1857 removed.append(f)
1858 1858
1859 1859 return scmutil.status(modified, added, removed, [], [], [], [])
1860 1860
1861 1861 class memfilectx(committablefilectx):
1862 1862 """memfilectx represents an in-memory file to commit.
1863 1863
1864 1864 See memctx and committablefilectx for more details.
1865 1865 """
1866 1866 def __init__(self, repo, path, data, islink=False,
1867 1867 isexec=False, copied=None, memctx=None):
1868 1868 """
1869 1869 path is the normalized file path relative to repository root.
1870 1870 data is the file content as a string.
1871 1871 islink is True if the file is a symbolic link.
1872 1872 isexec is True if the file is executable.
1873 1873 copied is the source file path if current file was copied in the
1874 1874 revision being committed, or None."""
1875 1875 super(memfilectx, self).__init__(repo, path, None, memctx)
1876 1876 self._data = data
1877 1877 self._flags = (islink and 'l' or '') + (isexec and 'x' or '')
1878 1878 self._copied = None
1879 1879 if copied:
1880 1880 self._copied = (copied, nullid)
1881 1881
1882 1882 def data(self):
1883 1883 return self._data
1884 1884 def size(self):
1885 1885 return len(self.data())
1886 1886 def flags(self):
1887 1887 return self._flags
1888 1888 def renamed(self):
1889 1889 return self._copied
1890 1890
1891 1891 def remove(self, ignoremissing=False):
1892 1892 """wraps unlink for a repo's working directory"""
1893 1893 # need to figure out what to do here
1894 1894 del self._changectx[self._path]
1895 1895
1896 1896 def write(self, data, flags):
1897 1897 """wraps repo.wwrite"""
1898 1898 self._data = data
General Comments 0
You need to be logged in to leave comments. Login now