##// END OF EJS Templates
context.walk: call with util.all() a generator, not a list...
Martin von Zweigbergk -
r24380:dd3bccb4 default
parent child Browse files
Show More
@@ -1,1873 +1,1873 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 462 if repo.filtername == '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 fset = set(match.files())
591 591 # for dirstate.walk, files=['.'] means "walk the whole tree".
592 592 # follow that here, too
593 593 fset.discard('.')
594 594
595 595 # avoid the entire walk if we're only looking for specific files
596 596 if fset and not match.anypats():
597 if util.all([fn in self for fn in fset]):
597 if util.all(fn in self for fn in fset):
598 598 for fn in sorted(fset):
599 599 if match(fn):
600 600 yield fn
601 601 raise StopIteration
602 602
603 603 for fn in self:
604 604 if fn in fset:
605 605 # specified pattern is the exact name
606 606 fset.remove(fn)
607 607 if match(fn):
608 608 yield fn
609 609 for fn in sorted(fset):
610 610 if not self.hasdir(fn):
611 611 match.bad(fn, _('no such file in rev %s') % self)
612 612
613 613 def matches(self, match):
614 614 return self.walk(match)
615 615
616 616 class basefilectx(object):
617 617 """A filecontext object represents the common logic for its children:
618 618 filectx: read-only access to a filerevision that is already present
619 619 in the repo,
620 620 workingfilectx: a filecontext that represents files from the working
621 621 directory,
622 622 memfilectx: a filecontext that represents files in-memory."""
623 623 def __new__(cls, repo, path, *args, **kwargs):
624 624 return super(basefilectx, cls).__new__(cls)
625 625
626 626 @propertycache
627 627 def _filelog(self):
628 628 return self._repo.file(self._path)
629 629
630 630 @propertycache
631 631 def _changeid(self):
632 632 if '_changeid' in self.__dict__:
633 633 return self._changeid
634 634 elif '_changectx' in self.__dict__:
635 635 return self._changectx.rev()
636 636 elif '_descendantrev' in self.__dict__:
637 637 # this file context was created from a revision with a known
638 638 # descendant, we can (lazily) correct for linkrev aliases
639 639 return self._adjustlinkrev(self._path, self._filelog,
640 640 self._filenode, self._descendantrev)
641 641 else:
642 642 return self._filelog.linkrev(self._filerev)
643 643
644 644 @propertycache
645 645 def _filenode(self):
646 646 if '_fileid' in self.__dict__:
647 647 return self._filelog.lookup(self._fileid)
648 648 else:
649 649 return self._changectx.filenode(self._path)
650 650
651 651 @propertycache
652 652 def _filerev(self):
653 653 return self._filelog.rev(self._filenode)
654 654
655 655 @propertycache
656 656 def _repopath(self):
657 657 return self._path
658 658
659 659 def __nonzero__(self):
660 660 try:
661 661 self._filenode
662 662 return True
663 663 except error.LookupError:
664 664 # file is missing
665 665 return False
666 666
667 667 def __str__(self):
668 668 return "%s@%s" % (self.path(), self._changectx)
669 669
670 670 def __repr__(self):
671 671 return "<%s %s>" % (type(self).__name__, str(self))
672 672
673 673 def __hash__(self):
674 674 try:
675 675 return hash((self._path, self._filenode))
676 676 except AttributeError:
677 677 return id(self)
678 678
679 679 def __eq__(self, other):
680 680 try:
681 681 return (type(self) == type(other) and self._path == other._path
682 682 and self._filenode == other._filenode)
683 683 except AttributeError:
684 684 return False
685 685
686 686 def __ne__(self, other):
687 687 return not (self == other)
688 688
689 689 def filerev(self):
690 690 return self._filerev
691 691 def filenode(self):
692 692 return self._filenode
693 693 def flags(self):
694 694 return self._changectx.flags(self._path)
695 695 def filelog(self):
696 696 return self._filelog
697 697 def rev(self):
698 698 return self._changeid
699 699 def linkrev(self):
700 700 return self._filelog.linkrev(self._filerev)
701 701 def node(self):
702 702 return self._changectx.node()
703 703 def hex(self):
704 704 return self._changectx.hex()
705 705 def user(self):
706 706 return self._changectx.user()
707 707 def date(self):
708 708 return self._changectx.date()
709 709 def files(self):
710 710 return self._changectx.files()
711 711 def description(self):
712 712 return self._changectx.description()
713 713 def branch(self):
714 714 return self._changectx.branch()
715 715 def extra(self):
716 716 return self._changectx.extra()
717 717 def phase(self):
718 718 return self._changectx.phase()
719 719 def phasestr(self):
720 720 return self._changectx.phasestr()
721 721 def manifest(self):
722 722 return self._changectx.manifest()
723 723 def changectx(self):
724 724 return self._changectx
725 725 def repo(self):
726 726 return self._repo
727 727
728 728 def path(self):
729 729 return self._path
730 730
731 731 def isbinary(self):
732 732 try:
733 733 return util.binary(self.data())
734 734 except IOError:
735 735 return False
736 736 def isexec(self):
737 737 return 'x' in self.flags()
738 738 def islink(self):
739 739 return 'l' in self.flags()
740 740
741 741 def cmp(self, fctx):
742 742 """compare with other file context
743 743
744 744 returns True if different than fctx.
745 745 """
746 746 if (fctx._filerev is None
747 747 and (self._repo._encodefilterpats
748 748 # if file data starts with '\1\n', empty metadata block is
749 749 # prepended, which adds 4 bytes to filelog.size().
750 750 or self.size() - 4 == fctx.size())
751 751 or self.size() == fctx.size()):
752 752 return self._filelog.cmp(self._filenode, fctx.data())
753 753
754 754 return True
755 755
756 756 def _adjustlinkrev(self, path, filelog, fnode, srcrev, inclusive=False):
757 757 """return the first ancestor of <srcrev> introducing <fnode>
758 758
759 759 If the linkrev of the file revision does not point to an ancestor of
760 760 srcrev, we'll walk down the ancestors until we find one introducing
761 761 this file revision.
762 762
763 763 :repo: a localrepository object (used to access changelog and manifest)
764 764 :path: the file path
765 765 :fnode: the nodeid of the file revision
766 766 :filelog: the filelog of this path
767 767 :srcrev: the changeset revision we search ancestors from
768 768 :inclusive: if true, the src revision will also be checked
769 769 """
770 770 repo = self._repo
771 771 cl = repo.unfiltered().changelog
772 772 ma = repo.manifest
773 773 # fetch the linkrev
774 774 fr = filelog.rev(fnode)
775 775 lkr = filelog.linkrev(fr)
776 776 # hack to reuse ancestor computation when searching for renames
777 777 memberanc = getattr(self, '_ancestrycontext', None)
778 778 iteranc = None
779 779 if memberanc is None:
780 780 memberanc = iteranc = cl.ancestors([srcrev], lkr,
781 781 inclusive=inclusive)
782 782 # check if this linkrev is an ancestor of srcrev
783 783 if lkr not in memberanc:
784 784 if iteranc is None:
785 785 iteranc = cl.ancestors([srcrev], lkr, inclusive=inclusive)
786 786 for a in iteranc:
787 787 ac = cl.read(a) # get changeset data (we avoid object creation)
788 788 if path in ac[3]: # checking the 'files' field.
789 789 # The file has been touched, check if the content is
790 790 # similar to the one we search for.
791 791 if fnode == ma.readfast(ac[0]).get(path):
792 792 return a
793 793 # In theory, we should never get out of that loop without a result.
794 794 # But if manifest uses a buggy file revision (not children of the
795 795 # one it replaces) we could. Such a buggy situation will likely
796 796 # result is crash somewhere else at to some point.
797 797 return lkr
798 798
799 799 def introrev(self):
800 800 """return the rev of the changeset which introduced this file revision
801 801
802 802 This method is different from linkrev because it take into account the
803 803 changeset the filectx was created from. It ensures the returned
804 804 revision is one of its ancestors. This prevents bugs from
805 805 'linkrev-shadowing' when a file revision is used by multiple
806 806 changesets.
807 807 """
808 808 lkr = self.linkrev()
809 809 attrs = vars(self)
810 810 noctx = not ('_changeid' in attrs or '_changectx' in attrs)
811 811 if noctx or self.rev() == lkr:
812 812 return self.linkrev()
813 813 return self._adjustlinkrev(self._path, self._filelog, self._filenode,
814 814 self.rev(), inclusive=True)
815 815
816 816 def parents(self):
817 817 _path = self._path
818 818 fl = self._filelog
819 819 parents = self._filelog.parents(self._filenode)
820 820 pl = [(_path, node, fl) for node in parents if node != nullid]
821 821
822 822 r = fl.renamed(self._filenode)
823 823 if r:
824 824 # - In the simple rename case, both parent are nullid, pl is empty.
825 825 # - In case of merge, only one of the parent is null id and should
826 826 # be replaced with the rename information. This parent is -always-
827 827 # the first one.
828 828 #
829 829 # As null id have always been filtered out in the previous list
830 830 # comprehension, inserting to 0 will always result in "replacing
831 831 # first nullid parent with rename information.
832 832 pl.insert(0, (r[0], r[1], self._repo.file(r[0])))
833 833
834 834 ret = []
835 835 for path, fnode, l in pl:
836 836 if '_changeid' in vars(self) or '_changectx' in vars(self):
837 837 # If self is associated with a changeset (probably explicitly
838 838 # fed), ensure the created filectx is associated with a
839 839 # changeset that is an ancestor of self.changectx.
840 840 # This lets us later use _adjustlinkrev to get a correct link.
841 841 fctx = filectx(self._repo, path, fileid=fnode, filelog=l)
842 842 fctx._descendantrev = self.rev()
843 843 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
844 844 elif '_descendantrev' in vars(self):
845 845 # Otherwise propagate _descendantrev if we have one associated.
846 846 fctx = filectx(self._repo, path, fileid=fnode, filelog=l)
847 847 fctx._descendantrev = self._descendantrev
848 848 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
849 849 else:
850 850 fctx = filectx(self._repo, path, fileid=fnode, filelog=l)
851 851 ret.append(fctx)
852 852 return ret
853 853
854 854 def p1(self):
855 855 return self.parents()[0]
856 856
857 857 def p2(self):
858 858 p = self.parents()
859 859 if len(p) == 2:
860 860 return p[1]
861 861 return filectx(self._repo, self._path, fileid=-1, filelog=self._filelog)
862 862
863 863 def annotate(self, follow=False, linenumber=None, diffopts=None):
864 864 '''returns a list of tuples of (ctx, line) for each line
865 865 in the file, where ctx is the filectx of the node where
866 866 that line was last changed.
867 867 This returns tuples of ((ctx, linenumber), line) for each line,
868 868 if "linenumber" parameter is NOT "None".
869 869 In such tuples, linenumber means one at the first appearance
870 870 in the managed file.
871 871 To reduce annotation cost,
872 872 this returns fixed value(False is used) as linenumber,
873 873 if "linenumber" parameter is "False".'''
874 874
875 875 if linenumber is None:
876 876 def decorate(text, rev):
877 877 return ([rev] * len(text.splitlines()), text)
878 878 elif linenumber:
879 879 def decorate(text, rev):
880 880 size = len(text.splitlines())
881 881 return ([(rev, i) for i in xrange(1, size + 1)], text)
882 882 else:
883 883 def decorate(text, rev):
884 884 return ([(rev, False)] * len(text.splitlines()), text)
885 885
886 886 def pair(parent, child):
887 887 blocks = mdiff.allblocks(parent[1], child[1], opts=diffopts,
888 888 refine=True)
889 889 for (a1, a2, b1, b2), t in blocks:
890 890 # Changed blocks ('!') or blocks made only of blank lines ('~')
891 891 # belong to the child.
892 892 if t == '=':
893 893 child[0][b1:b2] = parent[0][a1:a2]
894 894 return child
895 895
896 896 getlog = util.lrucachefunc(lambda x: self._repo.file(x))
897 897
898 898 def parents(f):
899 899 pl = f.parents()
900 900
901 901 # Don't return renamed parents if we aren't following.
902 902 if not follow:
903 903 pl = [p for p in pl if p.path() == f.path()]
904 904
905 905 # renamed filectx won't have a filelog yet, so set it
906 906 # from the cache to save time
907 907 for p in pl:
908 908 if not '_filelog' in p.__dict__:
909 909 p._filelog = getlog(p.path())
910 910
911 911 return pl
912 912
913 913 # use linkrev to find the first changeset where self appeared
914 914 base = self
915 915 introrev = self.introrev()
916 916 if self.rev() != introrev:
917 917 base = self.filectx(self.filenode(), changeid=introrev)
918 918
919 919 # This algorithm would prefer to be recursive, but Python is a
920 920 # bit recursion-hostile. Instead we do an iterative
921 921 # depth-first search.
922 922
923 923 visit = [base]
924 924 hist = {}
925 925 pcache = {}
926 926 needed = {base: 1}
927 927 while visit:
928 928 f = visit[-1]
929 929 pcached = f in pcache
930 930 if not pcached:
931 931 pcache[f] = parents(f)
932 932
933 933 ready = True
934 934 pl = pcache[f]
935 935 for p in pl:
936 936 if p not in hist:
937 937 ready = False
938 938 visit.append(p)
939 939 if not pcached:
940 940 needed[p] = needed.get(p, 0) + 1
941 941 if ready:
942 942 visit.pop()
943 943 reusable = f in hist
944 944 if reusable:
945 945 curr = hist[f]
946 946 else:
947 947 curr = decorate(f.data(), f)
948 948 for p in pl:
949 949 if not reusable:
950 950 curr = pair(hist[p], curr)
951 951 if needed[p] == 1:
952 952 del hist[p]
953 953 del needed[p]
954 954 else:
955 955 needed[p] -= 1
956 956
957 957 hist[f] = curr
958 958 pcache[f] = []
959 959
960 960 return zip(hist[base][0], hist[base][1].splitlines(True))
961 961
962 962 def ancestors(self, followfirst=False):
963 963 visit = {}
964 964 c = self
965 965 if followfirst:
966 966 cut = 1
967 967 else:
968 968 cut = None
969 969
970 970 while True:
971 971 for parent in c.parents()[:cut]:
972 972 visit[(parent.linkrev(), parent.filenode())] = parent
973 973 if not visit:
974 974 break
975 975 c = visit.pop(max(visit))
976 976 yield c
977 977
978 978 class filectx(basefilectx):
979 979 """A filecontext object makes access to data related to a particular
980 980 filerevision convenient."""
981 981 def __init__(self, repo, path, changeid=None, fileid=None,
982 982 filelog=None, changectx=None):
983 983 """changeid can be a changeset revision, node, or tag.
984 984 fileid can be a file revision or node."""
985 985 self._repo = repo
986 986 self._path = path
987 987
988 988 assert (changeid is not None
989 989 or fileid is not None
990 990 or changectx is not None), \
991 991 ("bad args: changeid=%r, fileid=%r, changectx=%r"
992 992 % (changeid, fileid, changectx))
993 993
994 994 if filelog is not None:
995 995 self._filelog = filelog
996 996
997 997 if changeid is not None:
998 998 self._changeid = changeid
999 999 if changectx is not None:
1000 1000 self._changectx = changectx
1001 1001 if fileid is not None:
1002 1002 self._fileid = fileid
1003 1003
1004 1004 @propertycache
1005 1005 def _changectx(self):
1006 1006 try:
1007 1007 return changectx(self._repo, self._changeid)
1008 1008 except error.FilteredRepoLookupError:
1009 1009 # Linkrev may point to any revision in the repository. When the
1010 1010 # repository is filtered this may lead to `filectx` trying to build
1011 1011 # `changectx` for filtered revision. In such case we fallback to
1012 1012 # creating `changectx` on the unfiltered version of the reposition.
1013 1013 # This fallback should not be an issue because `changectx` from
1014 1014 # `filectx` are not used in complex operations that care about
1015 1015 # filtering.
1016 1016 #
1017 1017 # This fallback is a cheap and dirty fix that prevent several
1018 1018 # crashes. It does not ensure the behavior is correct. However the
1019 1019 # behavior was not correct before filtering either and "incorrect
1020 1020 # behavior" is seen as better as "crash"
1021 1021 #
1022 1022 # Linkrevs have several serious troubles with filtering that are
1023 1023 # complicated to solve. Proper handling of the issue here should be
1024 1024 # considered when solving linkrev issue are on the table.
1025 1025 return changectx(self._repo.unfiltered(), self._changeid)
1026 1026
1027 1027 def filectx(self, fileid, changeid=None):
1028 1028 '''opens an arbitrary revision of the file without
1029 1029 opening a new filelog'''
1030 1030 return filectx(self._repo, self._path, fileid=fileid,
1031 1031 filelog=self._filelog, changeid=changeid)
1032 1032
1033 1033 def data(self):
1034 1034 try:
1035 1035 return self._filelog.read(self._filenode)
1036 1036 except error.CensoredNodeError:
1037 1037 if self._repo.ui.config("censor", "policy", "abort") == "ignore":
1038 1038 return ""
1039 1039 raise util.Abort(_("censored node: %s") % short(self._filenode),
1040 1040 hint=_("set censor.policy to ignore errors"))
1041 1041
1042 1042 def size(self):
1043 1043 return self._filelog.size(self._filerev)
1044 1044
1045 1045 def renamed(self):
1046 1046 """check if file was actually renamed in this changeset revision
1047 1047
1048 1048 If rename logged in file revision, we report copy for changeset only
1049 1049 if file revisions linkrev points back to the changeset in question
1050 1050 or both changeset parents contain different file revisions.
1051 1051 """
1052 1052
1053 1053 renamed = self._filelog.renamed(self._filenode)
1054 1054 if not renamed:
1055 1055 return renamed
1056 1056
1057 1057 if self.rev() == self.linkrev():
1058 1058 return renamed
1059 1059
1060 1060 name = self.path()
1061 1061 fnode = self._filenode
1062 1062 for p in self._changectx.parents():
1063 1063 try:
1064 1064 if fnode == p.filenode(name):
1065 1065 return None
1066 1066 except error.LookupError:
1067 1067 pass
1068 1068 return renamed
1069 1069
1070 1070 def children(self):
1071 1071 # hard for renames
1072 1072 c = self._filelog.children(self._filenode)
1073 1073 return [filectx(self._repo, self._path, fileid=x,
1074 1074 filelog=self._filelog) for x in c]
1075 1075
1076 1076 class committablectx(basectx):
1077 1077 """A committablectx object provides common functionality for a context that
1078 1078 wants the ability to commit, e.g. workingctx or memctx."""
1079 1079 def __init__(self, repo, text="", user=None, date=None, extra=None,
1080 1080 changes=None):
1081 1081 self._repo = repo
1082 1082 self._rev = None
1083 1083 self._node = None
1084 1084 self._text = text
1085 1085 if date:
1086 1086 self._date = util.parsedate(date)
1087 1087 if user:
1088 1088 self._user = user
1089 1089 if changes:
1090 1090 self._status = changes
1091 1091
1092 1092 self._extra = {}
1093 1093 if extra:
1094 1094 self._extra = extra.copy()
1095 1095 if 'branch' not in self._extra:
1096 1096 try:
1097 1097 branch = encoding.fromlocal(self._repo.dirstate.branch())
1098 1098 except UnicodeDecodeError:
1099 1099 raise util.Abort(_('branch name not in UTF-8!'))
1100 1100 self._extra['branch'] = branch
1101 1101 if self._extra['branch'] == '':
1102 1102 self._extra['branch'] = 'default'
1103 1103
1104 1104 def __str__(self):
1105 1105 return str(self._parents[0]) + "+"
1106 1106
1107 1107 def __nonzero__(self):
1108 1108 return True
1109 1109
1110 1110 def _buildflagfunc(self):
1111 1111 # Create a fallback function for getting file flags when the
1112 1112 # filesystem doesn't support them
1113 1113
1114 1114 copiesget = self._repo.dirstate.copies().get
1115 1115
1116 1116 if len(self._parents) < 2:
1117 1117 # when we have one parent, it's easy: copy from parent
1118 1118 man = self._parents[0].manifest()
1119 1119 def func(f):
1120 1120 f = copiesget(f, f)
1121 1121 return man.flags(f)
1122 1122 else:
1123 1123 # merges are tricky: we try to reconstruct the unstored
1124 1124 # result from the merge (issue1802)
1125 1125 p1, p2 = self._parents
1126 1126 pa = p1.ancestor(p2)
1127 1127 m1, m2, ma = p1.manifest(), p2.manifest(), pa.manifest()
1128 1128
1129 1129 def func(f):
1130 1130 f = copiesget(f, f) # may be wrong for merges with copies
1131 1131 fl1, fl2, fla = m1.flags(f), m2.flags(f), ma.flags(f)
1132 1132 if fl1 == fl2:
1133 1133 return fl1
1134 1134 if fl1 == fla:
1135 1135 return fl2
1136 1136 if fl2 == fla:
1137 1137 return fl1
1138 1138 return '' # punt for conflicts
1139 1139
1140 1140 return func
1141 1141
1142 1142 @propertycache
1143 1143 def _flagfunc(self):
1144 1144 return self._repo.dirstate.flagfunc(self._buildflagfunc)
1145 1145
1146 1146 @propertycache
1147 1147 def _manifest(self):
1148 1148 """generate a manifest corresponding to the values in self._status
1149 1149
1150 1150 This reuse the file nodeid from parent, but we append an extra letter
1151 1151 when modified. Modified files get an extra 'm' while added files get
1152 1152 an extra 'a'. This is used by manifests merge to see that files
1153 1153 are different and by update logic to avoid deleting newly added files.
1154 1154 """
1155 1155
1156 1156 man1 = self._parents[0].manifest()
1157 1157 man = man1.copy()
1158 1158 if len(self._parents) > 1:
1159 1159 man2 = self.p2().manifest()
1160 1160 def getman(f):
1161 1161 if f in man1:
1162 1162 return man1
1163 1163 return man2
1164 1164 else:
1165 1165 getman = lambda f: man1
1166 1166
1167 1167 copied = self._repo.dirstate.copies()
1168 1168 ff = self._flagfunc
1169 1169 for i, l in (("a", self._status.added), ("m", self._status.modified)):
1170 1170 for f in l:
1171 1171 orig = copied.get(f, f)
1172 1172 man[f] = getman(orig).get(orig, nullid) + i
1173 1173 try:
1174 1174 man.setflag(f, ff(f))
1175 1175 except OSError:
1176 1176 pass
1177 1177
1178 1178 for f in self._status.deleted + self._status.removed:
1179 1179 if f in man:
1180 1180 del man[f]
1181 1181
1182 1182 return man
1183 1183
1184 1184 @propertycache
1185 1185 def _status(self):
1186 1186 return self._repo.status()
1187 1187
1188 1188 @propertycache
1189 1189 def _user(self):
1190 1190 return self._repo.ui.username()
1191 1191
1192 1192 @propertycache
1193 1193 def _date(self):
1194 1194 return util.makedate()
1195 1195
1196 1196 def subrev(self, subpath):
1197 1197 return None
1198 1198
1199 1199 def user(self):
1200 1200 return self._user or self._repo.ui.username()
1201 1201 def date(self):
1202 1202 return self._date
1203 1203 def description(self):
1204 1204 return self._text
1205 1205 def files(self):
1206 1206 return sorted(self._status.modified + self._status.added +
1207 1207 self._status.removed)
1208 1208
1209 1209 def modified(self):
1210 1210 return self._status.modified
1211 1211 def added(self):
1212 1212 return self._status.added
1213 1213 def removed(self):
1214 1214 return self._status.removed
1215 1215 def deleted(self):
1216 1216 return self._status.deleted
1217 1217 def branch(self):
1218 1218 return encoding.tolocal(self._extra['branch'])
1219 1219 def closesbranch(self):
1220 1220 return 'close' in self._extra
1221 1221 def extra(self):
1222 1222 return self._extra
1223 1223
1224 1224 def tags(self):
1225 1225 t = []
1226 1226 for p in self.parents():
1227 1227 t.extend(p.tags())
1228 1228 return t
1229 1229
1230 1230 def bookmarks(self):
1231 1231 b = []
1232 1232 for p in self.parents():
1233 1233 b.extend(p.bookmarks())
1234 1234 return b
1235 1235
1236 1236 def phase(self):
1237 1237 phase = phases.draft # default phase to draft
1238 1238 for p in self.parents():
1239 1239 phase = max(phase, p.phase())
1240 1240 return phase
1241 1241
1242 1242 def hidden(self):
1243 1243 return False
1244 1244
1245 1245 def children(self):
1246 1246 return []
1247 1247
1248 1248 def flags(self, path):
1249 1249 if '_manifest' in self.__dict__:
1250 1250 try:
1251 1251 return self._manifest.flags(path)
1252 1252 except KeyError:
1253 1253 return ''
1254 1254
1255 1255 try:
1256 1256 return self._flagfunc(path)
1257 1257 except OSError:
1258 1258 return ''
1259 1259
1260 1260 def ancestor(self, c2):
1261 1261 """return the "best" ancestor context of self and c2"""
1262 1262 return self._parents[0].ancestor(c2) # punt on two parents for now
1263 1263
1264 1264 def walk(self, match):
1265 1265 return sorted(self._repo.dirstate.walk(match, sorted(self.substate),
1266 1266 True, False))
1267 1267
1268 1268 def matches(self, match):
1269 1269 return sorted(self._repo.dirstate.matches(match))
1270 1270
1271 1271 def ancestors(self):
1272 1272 for p in self._parents:
1273 1273 yield p
1274 1274 for a in self._repo.changelog.ancestors(
1275 1275 [p.rev() for p in self._parents]):
1276 1276 yield changectx(self._repo, a)
1277 1277
1278 1278 def markcommitted(self, node):
1279 1279 """Perform post-commit cleanup necessary after committing this ctx
1280 1280
1281 1281 Specifically, this updates backing stores this working context
1282 1282 wraps to reflect the fact that the changes reflected by this
1283 1283 workingctx have been committed. For example, it marks
1284 1284 modified and added files as normal in the dirstate.
1285 1285
1286 1286 """
1287 1287
1288 1288 self._repo.dirstate.beginparentchange()
1289 1289 for f in self.modified() + self.added():
1290 1290 self._repo.dirstate.normal(f)
1291 1291 for f in self.removed():
1292 1292 self._repo.dirstate.drop(f)
1293 1293 self._repo.dirstate.setparents(node)
1294 1294 self._repo.dirstate.endparentchange()
1295 1295
1296 1296 class workingctx(committablectx):
1297 1297 """A workingctx object makes access to data related to
1298 1298 the current working directory convenient.
1299 1299 date - any valid date string or (unixtime, offset), or None.
1300 1300 user - username string, or None.
1301 1301 extra - a dictionary of extra values, or None.
1302 1302 changes - a list of file lists as returned by localrepo.status()
1303 1303 or None to use the repository status.
1304 1304 """
1305 1305 def __init__(self, repo, text="", user=None, date=None, extra=None,
1306 1306 changes=None):
1307 1307 super(workingctx, self).__init__(repo, text, user, date, extra, changes)
1308 1308
1309 1309 def __iter__(self):
1310 1310 d = self._repo.dirstate
1311 1311 for f in d:
1312 1312 if d[f] != 'r':
1313 1313 yield f
1314 1314
1315 1315 def __contains__(self, key):
1316 1316 return self._repo.dirstate[key] not in "?r"
1317 1317
1318 1318 @propertycache
1319 1319 def _parents(self):
1320 1320 p = self._repo.dirstate.parents()
1321 1321 if p[1] == nullid:
1322 1322 p = p[:-1]
1323 1323 return [changectx(self._repo, x) for x in p]
1324 1324
1325 1325 def filectx(self, path, filelog=None):
1326 1326 """get a file context from the working directory"""
1327 1327 return workingfilectx(self._repo, path, workingctx=self,
1328 1328 filelog=filelog)
1329 1329
1330 1330 def dirty(self, missing=False, merge=True, branch=True):
1331 1331 "check whether a working directory is modified"
1332 1332 # check subrepos first
1333 1333 for s in sorted(self.substate):
1334 1334 if self.sub(s).dirty():
1335 1335 return True
1336 1336 # check current working dir
1337 1337 return ((merge and self.p2()) or
1338 1338 (branch and self.branch() != self.p1().branch()) or
1339 1339 self.modified() or self.added() or self.removed() or
1340 1340 (missing and self.deleted()))
1341 1341
1342 1342 def add(self, list, prefix=""):
1343 1343 join = lambda f: os.path.join(prefix, f)
1344 1344 wlock = self._repo.wlock()
1345 1345 ui, ds = self._repo.ui, self._repo.dirstate
1346 1346 try:
1347 1347 rejected = []
1348 1348 lstat = self._repo.wvfs.lstat
1349 1349 for f in list:
1350 1350 scmutil.checkportable(ui, join(f))
1351 1351 try:
1352 1352 st = lstat(f)
1353 1353 except OSError:
1354 1354 ui.warn(_("%s does not exist!\n") % join(f))
1355 1355 rejected.append(f)
1356 1356 continue
1357 1357 if st.st_size > 10000000:
1358 1358 ui.warn(_("%s: up to %d MB of RAM may be required "
1359 1359 "to manage this file\n"
1360 1360 "(use 'hg revert %s' to cancel the "
1361 1361 "pending addition)\n")
1362 1362 % (f, 3 * st.st_size // 1000000, join(f)))
1363 1363 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1364 1364 ui.warn(_("%s not added: only files and symlinks "
1365 1365 "supported currently\n") % join(f))
1366 1366 rejected.append(f)
1367 1367 elif ds[f] in 'amn':
1368 1368 ui.warn(_("%s already tracked!\n") % join(f))
1369 1369 elif ds[f] == 'r':
1370 1370 ds.normallookup(f)
1371 1371 else:
1372 1372 ds.add(f)
1373 1373 return rejected
1374 1374 finally:
1375 1375 wlock.release()
1376 1376
1377 1377 def forget(self, files, prefix=""):
1378 1378 join = lambda f: os.path.join(prefix, f)
1379 1379 wlock = self._repo.wlock()
1380 1380 try:
1381 1381 rejected = []
1382 1382 for f in files:
1383 1383 if f not in self._repo.dirstate:
1384 1384 self._repo.ui.warn(_("%s not tracked!\n") % join(f))
1385 1385 rejected.append(f)
1386 1386 elif self._repo.dirstate[f] != 'a':
1387 1387 self._repo.dirstate.remove(f)
1388 1388 else:
1389 1389 self._repo.dirstate.drop(f)
1390 1390 return rejected
1391 1391 finally:
1392 1392 wlock.release()
1393 1393
1394 1394 def undelete(self, list):
1395 1395 pctxs = self.parents()
1396 1396 wlock = self._repo.wlock()
1397 1397 try:
1398 1398 for f in list:
1399 1399 if self._repo.dirstate[f] != 'r':
1400 1400 self._repo.ui.warn(_("%s not removed!\n") % f)
1401 1401 else:
1402 1402 fctx = f in pctxs[0] and pctxs[0][f] or pctxs[1][f]
1403 1403 t = fctx.data()
1404 1404 self._repo.wwrite(f, t, fctx.flags())
1405 1405 self._repo.dirstate.normal(f)
1406 1406 finally:
1407 1407 wlock.release()
1408 1408
1409 1409 def copy(self, source, dest):
1410 1410 try:
1411 1411 st = self._repo.wvfs.lstat(dest)
1412 1412 except OSError, err:
1413 1413 if err.errno != errno.ENOENT:
1414 1414 raise
1415 1415 self._repo.ui.warn(_("%s does not exist!\n") % dest)
1416 1416 return
1417 1417 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1418 1418 self._repo.ui.warn(_("copy failed: %s is not a file or a "
1419 1419 "symbolic link\n") % dest)
1420 1420 else:
1421 1421 wlock = self._repo.wlock()
1422 1422 try:
1423 1423 if self._repo.dirstate[dest] in '?':
1424 1424 self._repo.dirstate.add(dest)
1425 1425 elif self._repo.dirstate[dest] in 'r':
1426 1426 self._repo.dirstate.normallookup(dest)
1427 1427 self._repo.dirstate.copy(source, dest)
1428 1428 finally:
1429 1429 wlock.release()
1430 1430
1431 1431 def _filtersuspectsymlink(self, files):
1432 1432 if not files or self._repo.dirstate._checklink:
1433 1433 return files
1434 1434
1435 1435 # Symlink placeholders may get non-symlink-like contents
1436 1436 # via user error or dereferencing by NFS or Samba servers,
1437 1437 # so we filter out any placeholders that don't look like a
1438 1438 # symlink
1439 1439 sane = []
1440 1440 for f in files:
1441 1441 if self.flags(f) == 'l':
1442 1442 d = self[f].data()
1443 1443 if d == '' or len(d) >= 1024 or '\n' in d or util.binary(d):
1444 1444 self._repo.ui.debug('ignoring suspect symlink placeholder'
1445 1445 ' "%s"\n' % f)
1446 1446 continue
1447 1447 sane.append(f)
1448 1448 return sane
1449 1449
1450 1450 def _checklookup(self, files):
1451 1451 # check for any possibly clean files
1452 1452 if not files:
1453 1453 return [], []
1454 1454
1455 1455 modified = []
1456 1456 fixup = []
1457 1457 pctx = self._parents[0]
1458 1458 # do a full compare of any files that might have changed
1459 1459 for f in sorted(files):
1460 1460 if (f not in pctx or self.flags(f) != pctx.flags(f)
1461 1461 or pctx[f].cmp(self[f])):
1462 1462 modified.append(f)
1463 1463 else:
1464 1464 fixup.append(f)
1465 1465
1466 1466 # update dirstate for files that are actually clean
1467 1467 if fixup:
1468 1468 try:
1469 1469 # updating the dirstate is optional
1470 1470 # so we don't wait on the lock
1471 1471 # wlock can invalidate the dirstate, so cache normal _after_
1472 1472 # taking the lock
1473 1473 wlock = self._repo.wlock(False)
1474 1474 normal = self._repo.dirstate.normal
1475 1475 try:
1476 1476 for f in fixup:
1477 1477 normal(f)
1478 1478 finally:
1479 1479 wlock.release()
1480 1480 except error.LockError:
1481 1481 pass
1482 1482 return modified, fixup
1483 1483
1484 1484 def _manifestmatches(self, match, s):
1485 1485 """Slow path for workingctx
1486 1486
1487 1487 The fast path is when we compare the working directory to its parent
1488 1488 which means this function is comparing with a non-parent; therefore we
1489 1489 need to build a manifest and return what matches.
1490 1490 """
1491 1491 mf = self._repo['.']._manifestmatches(match, s)
1492 1492 for f in s.modified + s.added:
1493 1493 mf[f] = _newnode
1494 1494 mf.setflag(f, self.flags(f))
1495 1495 for f in s.removed:
1496 1496 if f in mf:
1497 1497 del mf[f]
1498 1498 return mf
1499 1499
1500 1500 def _dirstatestatus(self, match=None, ignored=False, clean=False,
1501 1501 unknown=False):
1502 1502 '''Gets the status from the dirstate -- internal use only.'''
1503 1503 listignored, listclean, listunknown = ignored, clean, unknown
1504 1504 match = match or matchmod.always(self._repo.root, self._repo.getcwd())
1505 1505 subrepos = []
1506 1506 if '.hgsub' in self:
1507 1507 subrepos = sorted(self.substate)
1508 1508 cmp, s = self._repo.dirstate.status(match, subrepos, listignored,
1509 1509 listclean, listunknown)
1510 1510
1511 1511 # check for any possibly clean files
1512 1512 if cmp:
1513 1513 modified2, fixup = self._checklookup(cmp)
1514 1514 s.modified.extend(modified2)
1515 1515
1516 1516 # update dirstate for files that are actually clean
1517 1517 if fixup and listclean:
1518 1518 s.clean.extend(fixup)
1519 1519
1520 1520 if match.always():
1521 1521 # cache for performance
1522 1522 if s.unknown or s.ignored or s.clean:
1523 1523 # "_status" is cached with list*=False in the normal route
1524 1524 self._status = scmutil.status(s.modified, s.added, s.removed,
1525 1525 s.deleted, [], [], [])
1526 1526 else:
1527 1527 self._status = s
1528 1528
1529 1529 return s
1530 1530
1531 1531 def _buildstatus(self, other, s, match, listignored, listclean,
1532 1532 listunknown):
1533 1533 """build a status with respect to another context
1534 1534
1535 1535 This includes logic for maintaining the fast path of status when
1536 1536 comparing the working directory against its parent, which is to skip
1537 1537 building a new manifest if self (working directory) is not comparing
1538 1538 against its parent (repo['.']).
1539 1539 """
1540 1540 s = self._dirstatestatus(match, listignored, listclean, listunknown)
1541 1541 # Filter out symlinks that, in the case of FAT32 and NTFS filesystems,
1542 1542 # might have accidentally ended up with the entire contents of the file
1543 1543 # they are supposed to be linking to.
1544 1544 s.modified[:] = self._filtersuspectsymlink(s.modified)
1545 1545 if other != self._repo['.']:
1546 1546 s = super(workingctx, self)._buildstatus(other, s, match,
1547 1547 listignored, listclean,
1548 1548 listunknown)
1549 1549 return s
1550 1550
1551 1551 def _matchstatus(self, other, match):
1552 1552 """override the match method with a filter for directory patterns
1553 1553
1554 1554 We use inheritance to customize the match.bad method only in cases of
1555 1555 workingctx since it belongs only to the working directory when
1556 1556 comparing against the parent changeset.
1557 1557
1558 1558 If we aren't comparing against the working directory's parent, then we
1559 1559 just use the default match object sent to us.
1560 1560 """
1561 1561 superself = super(workingctx, self)
1562 1562 match = superself._matchstatus(other, match)
1563 1563 if other != self._repo['.']:
1564 1564 def bad(f, msg):
1565 1565 # 'f' may be a directory pattern from 'match.files()',
1566 1566 # so 'f not in ctx1' is not enough
1567 1567 if f not in other and not other.hasdir(f):
1568 1568 self._repo.ui.warn('%s: %s\n' %
1569 1569 (self._repo.dirstate.pathto(f), msg))
1570 1570 match.bad = bad
1571 1571 return match
1572 1572
1573 1573 class committablefilectx(basefilectx):
1574 1574 """A committablefilectx provides common functionality for a file context
1575 1575 that wants the ability to commit, e.g. workingfilectx or memfilectx."""
1576 1576 def __init__(self, repo, path, filelog=None, ctx=None):
1577 1577 self._repo = repo
1578 1578 self._path = path
1579 1579 self._changeid = None
1580 1580 self._filerev = self._filenode = None
1581 1581
1582 1582 if filelog is not None:
1583 1583 self._filelog = filelog
1584 1584 if ctx:
1585 1585 self._changectx = ctx
1586 1586
1587 1587 def __nonzero__(self):
1588 1588 return True
1589 1589
1590 1590 def parents(self):
1591 1591 '''return parent filectxs, following copies if necessary'''
1592 1592 def filenode(ctx, path):
1593 1593 return ctx._manifest.get(path, nullid)
1594 1594
1595 1595 path = self._path
1596 1596 fl = self._filelog
1597 1597 pcl = self._changectx._parents
1598 1598 renamed = self.renamed()
1599 1599
1600 1600 if renamed:
1601 1601 pl = [renamed + (None,)]
1602 1602 else:
1603 1603 pl = [(path, filenode(pcl[0], path), fl)]
1604 1604
1605 1605 for pc in pcl[1:]:
1606 1606 pl.append((path, filenode(pc, path), fl))
1607 1607
1608 1608 return [filectx(self._repo, p, fileid=n, filelog=l)
1609 1609 for p, n, l in pl if n != nullid]
1610 1610
1611 1611 def children(self):
1612 1612 return []
1613 1613
1614 1614 class workingfilectx(committablefilectx):
1615 1615 """A workingfilectx object makes access to data related to a particular
1616 1616 file in the working directory convenient."""
1617 1617 def __init__(self, repo, path, filelog=None, workingctx=None):
1618 1618 super(workingfilectx, self).__init__(repo, path, filelog, workingctx)
1619 1619
1620 1620 @propertycache
1621 1621 def _changectx(self):
1622 1622 return workingctx(self._repo)
1623 1623
1624 1624 def data(self):
1625 1625 return self._repo.wread(self._path)
1626 1626 def renamed(self):
1627 1627 rp = self._repo.dirstate.copied(self._path)
1628 1628 if not rp:
1629 1629 return None
1630 1630 return rp, self._changectx._parents[0]._manifest.get(rp, nullid)
1631 1631
1632 1632 def size(self):
1633 1633 return self._repo.wvfs.lstat(self._path).st_size
1634 1634 def date(self):
1635 1635 t, tz = self._changectx.date()
1636 1636 try:
1637 1637 return (int(self._repo.wvfs.lstat(self._path).st_mtime), tz)
1638 1638 except OSError, err:
1639 1639 if err.errno != errno.ENOENT:
1640 1640 raise
1641 1641 return (t, tz)
1642 1642
1643 1643 def cmp(self, fctx):
1644 1644 """compare with other file context
1645 1645
1646 1646 returns True if different than fctx.
1647 1647 """
1648 1648 # fctx should be a filectx (not a workingfilectx)
1649 1649 # invert comparison to reuse the same code path
1650 1650 return fctx.cmp(self)
1651 1651
1652 1652 def remove(self, ignoremissing=False):
1653 1653 """wraps unlink for a repo's working directory"""
1654 1654 util.unlinkpath(self._repo.wjoin(self._path), ignoremissing)
1655 1655
1656 1656 def write(self, data, flags):
1657 1657 """wraps repo.wwrite"""
1658 1658 self._repo.wwrite(self._path, data, flags)
1659 1659
1660 1660 class workingcommitctx(workingctx):
1661 1661 """A workingcommitctx object makes access to data related to
1662 1662 the revision being committed convenient.
1663 1663
1664 1664 This hides changes in the working directory, if they aren't
1665 1665 committed in this context.
1666 1666 """
1667 1667 def __init__(self, repo, changes,
1668 1668 text="", user=None, date=None, extra=None):
1669 1669 super(workingctx, self).__init__(repo, text, user, date, extra,
1670 1670 changes)
1671 1671
1672 1672 def _dirstatestatus(self, match=None, ignored=False, clean=False,
1673 1673 unknown=False):
1674 1674 """Return matched files only in ``self._status``
1675 1675
1676 1676 Uncommitted files appear "clean" via this context, even if
1677 1677 they aren't actually so in the working directory.
1678 1678 """
1679 1679 match = match or matchmod.always(self._repo.root, self._repo.getcwd())
1680 1680 if clean:
1681 1681 clean = [f for f in self._manifest if f not in self._changedset]
1682 1682 else:
1683 1683 clean = []
1684 1684 return scmutil.status([f for f in self._status.modified if match(f)],
1685 1685 [f for f in self._status.added if match(f)],
1686 1686 [f for f in self._status.removed if match(f)],
1687 1687 [], [], [], clean)
1688 1688
1689 1689 @propertycache
1690 1690 def _changedset(self):
1691 1691 """Return the set of files changed in this context
1692 1692 """
1693 1693 changed = set(self._status.modified)
1694 1694 changed.update(self._status.added)
1695 1695 changed.update(self._status.removed)
1696 1696 return changed
1697 1697
1698 1698 class memctx(committablectx):
1699 1699 """Use memctx to perform in-memory commits via localrepo.commitctx().
1700 1700
1701 1701 Revision information is supplied at initialization time while
1702 1702 related files data and is made available through a callback
1703 1703 mechanism. 'repo' is the current localrepo, 'parents' is a
1704 1704 sequence of two parent revisions identifiers (pass None for every
1705 1705 missing parent), 'text' is the commit message and 'files' lists
1706 1706 names of files touched by the revision (normalized and relative to
1707 1707 repository root).
1708 1708
1709 1709 filectxfn(repo, memctx, path) is a callable receiving the
1710 1710 repository, the current memctx object and the normalized path of
1711 1711 requested file, relative to repository root. It is fired by the
1712 1712 commit function for every file in 'files', but calls order is
1713 1713 undefined. If the file is available in the revision being
1714 1714 committed (updated or added), filectxfn returns a memfilectx
1715 1715 object. If the file was removed, filectxfn raises an
1716 1716 IOError. Moved files are represented by marking the source file
1717 1717 removed and the new file added with copy information (see
1718 1718 memfilectx).
1719 1719
1720 1720 user receives the committer name and defaults to current
1721 1721 repository username, date is the commit date in any format
1722 1722 supported by util.parsedate() and defaults to current date, extra
1723 1723 is a dictionary of metadata or is left empty.
1724 1724 """
1725 1725
1726 1726 # Mercurial <= 3.1 expects the filectxfn to raise IOError for missing files.
1727 1727 # Extensions that need to retain compatibility across Mercurial 3.1 can use
1728 1728 # this field to determine what to do in filectxfn.
1729 1729 _returnnoneformissingfiles = True
1730 1730
1731 1731 def __init__(self, repo, parents, text, files, filectxfn, user=None,
1732 1732 date=None, extra=None, editor=False):
1733 1733 super(memctx, self).__init__(repo, text, user, date, extra)
1734 1734 self._rev = None
1735 1735 self._node = None
1736 1736 parents = [(p or nullid) for p in parents]
1737 1737 p1, p2 = parents
1738 1738 self._parents = [changectx(self._repo, p) for p in (p1, p2)]
1739 1739 files = sorted(set(files))
1740 1740 self._files = files
1741 1741 self.substate = {}
1742 1742
1743 1743 # if store is not callable, wrap it in a function
1744 1744 if not callable(filectxfn):
1745 1745 def getfilectx(repo, memctx, path):
1746 1746 fctx = filectxfn[path]
1747 1747 # this is weird but apparently we only keep track of one parent
1748 1748 # (why not only store that instead of a tuple?)
1749 1749 copied = fctx.renamed()
1750 1750 if copied:
1751 1751 copied = copied[0]
1752 1752 return memfilectx(repo, path, fctx.data(),
1753 1753 islink=fctx.islink(), isexec=fctx.isexec(),
1754 1754 copied=copied, memctx=memctx)
1755 1755 self._filectxfn = getfilectx
1756 1756 else:
1757 1757 # "util.cachefunc" reduces invocation of possibly expensive
1758 1758 # "filectxfn" for performance (e.g. converting from another VCS)
1759 1759 self._filectxfn = util.cachefunc(filectxfn)
1760 1760
1761 1761 if extra:
1762 1762 self._extra = extra.copy()
1763 1763 else:
1764 1764 self._extra = {}
1765 1765
1766 1766 if self._extra.get('branch', '') == '':
1767 1767 self._extra['branch'] = 'default'
1768 1768
1769 1769 if editor:
1770 1770 self._text = editor(self._repo, self, [])
1771 1771 self._repo.savecommitmessage(self._text)
1772 1772
1773 1773 def filectx(self, path, filelog=None):
1774 1774 """get a file context from the working directory
1775 1775
1776 1776 Returns None if file doesn't exist and should be removed."""
1777 1777 return self._filectxfn(self._repo, self, path)
1778 1778
1779 1779 def commit(self):
1780 1780 """commit context to the repo"""
1781 1781 return self._repo.commitctx(self)
1782 1782
1783 1783 @propertycache
1784 1784 def _manifest(self):
1785 1785 """generate a manifest based on the return values of filectxfn"""
1786 1786
1787 1787 # keep this simple for now; just worry about p1
1788 1788 pctx = self._parents[0]
1789 1789 man = pctx.manifest().copy()
1790 1790
1791 1791 for f in self._status.modified:
1792 1792 p1node = nullid
1793 1793 p2node = nullid
1794 1794 p = pctx[f].parents() # if file isn't in pctx, check p2?
1795 1795 if len(p) > 0:
1796 1796 p1node = p[0].node()
1797 1797 if len(p) > 1:
1798 1798 p2node = p[1].node()
1799 1799 man[f] = revlog.hash(self[f].data(), p1node, p2node)
1800 1800
1801 1801 for f in self._status.added:
1802 1802 man[f] = revlog.hash(self[f].data(), nullid, nullid)
1803 1803
1804 1804 for f in self._status.removed:
1805 1805 if f in man:
1806 1806 del man[f]
1807 1807
1808 1808 return man
1809 1809
1810 1810 @propertycache
1811 1811 def _status(self):
1812 1812 """Calculate exact status from ``files`` specified at construction
1813 1813 """
1814 1814 man1 = self.p1().manifest()
1815 1815 p2 = self._parents[1]
1816 1816 # "1 < len(self._parents)" can't be used for checking
1817 1817 # existence of the 2nd parent, because "memctx._parents" is
1818 1818 # explicitly initialized by the list, of which length is 2.
1819 1819 if p2.node() != nullid:
1820 1820 man2 = p2.manifest()
1821 1821 managing = lambda f: f in man1 or f in man2
1822 1822 else:
1823 1823 managing = lambda f: f in man1
1824 1824
1825 1825 modified, added, removed = [], [], []
1826 1826 for f in self._files:
1827 1827 if not managing(f):
1828 1828 added.append(f)
1829 1829 elif self[f]:
1830 1830 modified.append(f)
1831 1831 else:
1832 1832 removed.append(f)
1833 1833
1834 1834 return scmutil.status(modified, added, removed, [], [], [], [])
1835 1835
1836 1836 class memfilectx(committablefilectx):
1837 1837 """memfilectx represents an in-memory file to commit.
1838 1838
1839 1839 See memctx and committablefilectx for more details.
1840 1840 """
1841 1841 def __init__(self, repo, path, data, islink=False,
1842 1842 isexec=False, copied=None, memctx=None):
1843 1843 """
1844 1844 path is the normalized file path relative to repository root.
1845 1845 data is the file content as a string.
1846 1846 islink is True if the file is a symbolic link.
1847 1847 isexec is True if the file is executable.
1848 1848 copied is the source file path if current file was copied in the
1849 1849 revision being committed, or None."""
1850 1850 super(memfilectx, self).__init__(repo, path, None, memctx)
1851 1851 self._data = data
1852 1852 self._flags = (islink and 'l' or '') + (isexec and 'x' or '')
1853 1853 self._copied = None
1854 1854 if copied:
1855 1855 self._copied = (copied, nullid)
1856 1856
1857 1857 def data(self):
1858 1858 return self._data
1859 1859 def size(self):
1860 1860 return len(self.data())
1861 1861 def flags(self):
1862 1862 return self._flags
1863 1863 def renamed(self):
1864 1864 return self._copied
1865 1865
1866 1866 def remove(self, ignoremissing=False):
1867 1867 """wraps unlink for a repo's working directory"""
1868 1868 # need to figure out what to do here
1869 1869 del self._changectx[self._path]
1870 1870
1871 1871 def write(self, data, flags):
1872 1872 """wraps repo.wwrite"""
1873 1873 self._data = data
General Comments 0
You need to be logged in to leave comments. Login now