##// END OF EJS Templates
filectx: correct docstring about "changeid"...
Martin von Zweigbergk -
r40098:b6c2543e default
parent child Browse files
Show More
@@ -1,2440 +1,2440 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 __future__ import absolute_import
9 9
10 10 import errno
11 11 import filecmp
12 12 import os
13 13 import stat
14 14
15 15 from .i18n import _
16 16 from .node import (
17 17 addednodeid,
18 18 hex,
19 19 modifiednodeid,
20 20 nullid,
21 21 nullrev,
22 22 short,
23 23 wdirfilenodeids,
24 24 wdirid,
25 25 )
26 26 from . import (
27 27 dagop,
28 28 encoding,
29 29 error,
30 30 fileset,
31 31 match as matchmod,
32 32 obsolete as obsmod,
33 33 patch,
34 34 pathutil,
35 35 phases,
36 36 pycompat,
37 37 repoview,
38 38 scmutil,
39 39 sparse,
40 40 subrepo,
41 41 subrepoutil,
42 42 util,
43 43 )
44 44 from .utils import (
45 45 dateutil,
46 46 stringutil,
47 47 )
48 48
49 49 propertycache = util.propertycache
50 50
51 51 class basectx(object):
52 52 """A basectx object represents the common logic for its children:
53 53 changectx: read-only context that is already present in the repo,
54 54 workingctx: a context that represents the working directory and can
55 55 be committed,
56 56 memctx: a context that represents changes in-memory and can also
57 57 be committed."""
58 58
59 59 def __init__(self, repo):
60 60 self._repo = repo
61 61
62 62 def __bytes__(self):
63 63 return short(self.node())
64 64
65 65 __str__ = encoding.strmethod(__bytes__)
66 66
67 67 def __repr__(self):
68 68 return r"<%s %s>" % (type(self).__name__, str(self))
69 69
70 70 def __eq__(self, other):
71 71 try:
72 72 return type(self) == type(other) and self._rev == other._rev
73 73 except AttributeError:
74 74 return False
75 75
76 76 def __ne__(self, other):
77 77 return not (self == other)
78 78
79 79 def __contains__(self, key):
80 80 return key in self._manifest
81 81
82 82 def __getitem__(self, key):
83 83 return self.filectx(key)
84 84
85 85 def __iter__(self):
86 86 return iter(self._manifest)
87 87
88 88 def _buildstatusmanifest(self, status):
89 89 """Builds a manifest that includes the given status results, if this is
90 90 a working copy context. For non-working copy contexts, it just returns
91 91 the normal manifest."""
92 92 return self.manifest()
93 93
94 94 def _matchstatus(self, other, match):
95 95 """This internal method provides a way for child objects to override the
96 96 match operator.
97 97 """
98 98 return match
99 99
100 100 def _buildstatus(self, other, s, match, listignored, listclean,
101 101 listunknown):
102 102 """build a status with respect to another context"""
103 103 # Load earliest manifest first for caching reasons. More specifically,
104 104 # if you have revisions 1000 and 1001, 1001 is probably stored as a
105 105 # delta against 1000. Thus, if you read 1000 first, we'll reconstruct
106 106 # 1000 and cache it so that when you read 1001, we just need to apply a
107 107 # delta to what's in the cache. So that's one full reconstruction + one
108 108 # delta application.
109 109 mf2 = None
110 110 if self.rev() is not None and self.rev() < other.rev():
111 111 mf2 = self._buildstatusmanifest(s)
112 112 mf1 = other._buildstatusmanifest(s)
113 113 if mf2 is None:
114 114 mf2 = self._buildstatusmanifest(s)
115 115
116 116 modified, added = [], []
117 117 removed = []
118 118 clean = []
119 119 deleted, unknown, ignored = s.deleted, s.unknown, s.ignored
120 120 deletedset = set(deleted)
121 121 d = mf1.diff(mf2, match=match, clean=listclean)
122 122 for fn, value in d.iteritems():
123 123 if fn in deletedset:
124 124 continue
125 125 if value is None:
126 126 clean.append(fn)
127 127 continue
128 128 (node1, flag1), (node2, flag2) = value
129 129 if node1 is None:
130 130 added.append(fn)
131 131 elif node2 is None:
132 132 removed.append(fn)
133 133 elif flag1 != flag2:
134 134 modified.append(fn)
135 135 elif node2 not in wdirfilenodeids:
136 136 # When comparing files between two commits, we save time by
137 137 # not comparing the file contents when the nodeids differ.
138 138 # Note that this means we incorrectly report a reverted change
139 139 # to a file as a modification.
140 140 modified.append(fn)
141 141 elif self[fn].cmp(other[fn]):
142 142 modified.append(fn)
143 143 else:
144 144 clean.append(fn)
145 145
146 146 if removed:
147 147 # need to filter files if they are already reported as removed
148 148 unknown = [fn for fn in unknown if fn not in mf1 and
149 149 (not match or match(fn))]
150 150 ignored = [fn for fn in ignored if fn not in mf1 and
151 151 (not match or match(fn))]
152 152 # if they're deleted, don't report them as removed
153 153 removed = [fn for fn in removed if fn not in deletedset]
154 154
155 155 return scmutil.status(modified, added, removed, deleted, unknown,
156 156 ignored, clean)
157 157
158 158 @propertycache
159 159 def substate(self):
160 160 return subrepoutil.state(self, self._repo.ui)
161 161
162 162 def subrev(self, subpath):
163 163 return self.substate[subpath][1]
164 164
165 165 def rev(self):
166 166 return self._rev
167 167 def node(self):
168 168 return self._node
169 169 def hex(self):
170 170 return hex(self.node())
171 171 def manifest(self):
172 172 return self._manifest
173 173 def manifestctx(self):
174 174 return self._manifestctx
175 175 def repo(self):
176 176 return self._repo
177 177 def phasestr(self):
178 178 return phases.phasenames[self.phase()]
179 179 def mutable(self):
180 180 return self.phase() > phases.public
181 181
182 182 def matchfileset(self, expr, badfn=None):
183 183 return fileset.match(self, expr, badfn=badfn)
184 184
185 185 def obsolete(self):
186 186 """True if the changeset is obsolete"""
187 187 return self.rev() in obsmod.getrevs(self._repo, 'obsolete')
188 188
189 189 def extinct(self):
190 190 """True if the changeset is extinct"""
191 191 return self.rev() in obsmod.getrevs(self._repo, 'extinct')
192 192
193 193 def orphan(self):
194 194 """True if the changeset is not obsolete, but its ancestor is"""
195 195 return self.rev() in obsmod.getrevs(self._repo, 'orphan')
196 196
197 197 def phasedivergent(self):
198 198 """True if the changeset tries to be a successor of a public changeset
199 199
200 200 Only non-public and non-obsolete changesets may be phase-divergent.
201 201 """
202 202 return self.rev() in obsmod.getrevs(self._repo, 'phasedivergent')
203 203
204 204 def contentdivergent(self):
205 205 """Is a successor of a changeset with multiple possible successor sets
206 206
207 207 Only non-public and non-obsolete changesets may be content-divergent.
208 208 """
209 209 return self.rev() in obsmod.getrevs(self._repo, 'contentdivergent')
210 210
211 211 def isunstable(self):
212 212 """True if the changeset is either orphan, phase-divergent or
213 213 content-divergent"""
214 214 return self.orphan() or self.phasedivergent() or self.contentdivergent()
215 215
216 216 def instabilities(self):
217 217 """return the list of instabilities affecting this changeset.
218 218
219 219 Instabilities are returned as strings. possible values are:
220 220 - orphan,
221 221 - phase-divergent,
222 222 - content-divergent.
223 223 """
224 224 instabilities = []
225 225 if self.orphan():
226 226 instabilities.append('orphan')
227 227 if self.phasedivergent():
228 228 instabilities.append('phase-divergent')
229 229 if self.contentdivergent():
230 230 instabilities.append('content-divergent')
231 231 return instabilities
232 232
233 233 def parents(self):
234 234 """return contexts for each parent changeset"""
235 235 return self._parents
236 236
237 237 def p1(self):
238 238 return self._parents[0]
239 239
240 240 def p2(self):
241 241 parents = self._parents
242 242 if len(parents) == 2:
243 243 return parents[1]
244 244 return self._repo[nullrev]
245 245
246 246 def _fileinfo(self, path):
247 247 if r'_manifest' in self.__dict__:
248 248 try:
249 249 return self._manifest[path], self._manifest.flags(path)
250 250 except KeyError:
251 251 raise error.ManifestLookupError(self._node, path,
252 252 _('not found in manifest'))
253 253 if r'_manifestdelta' in self.__dict__ or path in self.files():
254 254 if path in self._manifestdelta:
255 255 return (self._manifestdelta[path],
256 256 self._manifestdelta.flags(path))
257 257 mfl = self._repo.manifestlog
258 258 try:
259 259 node, flag = mfl[self._changeset.manifest].find(path)
260 260 except KeyError:
261 261 raise error.ManifestLookupError(self._node, path,
262 262 _('not found in manifest'))
263 263
264 264 return node, flag
265 265
266 266 def filenode(self, path):
267 267 return self._fileinfo(path)[0]
268 268
269 269 def flags(self, path):
270 270 try:
271 271 return self._fileinfo(path)[1]
272 272 except error.LookupError:
273 273 return ''
274 274
275 275 def sub(self, path, allowcreate=True):
276 276 '''return a subrepo for the stored revision of path, never wdir()'''
277 277 return subrepo.subrepo(self, path, allowcreate=allowcreate)
278 278
279 279 def nullsub(self, path, pctx):
280 280 return subrepo.nullsubrepo(self, path, pctx)
281 281
282 282 def workingsub(self, path):
283 283 '''return a subrepo for the stored revision, or wdir if this is a wdir
284 284 context.
285 285 '''
286 286 return subrepo.subrepo(self, path, allowwdir=True)
287 287
288 288 def match(self, pats=None, include=None, exclude=None, default='glob',
289 289 listsubrepos=False, badfn=None):
290 290 r = self._repo
291 291 return matchmod.match(r.root, r.getcwd(), pats,
292 292 include, exclude, default,
293 293 auditor=r.nofsauditor, ctx=self,
294 294 listsubrepos=listsubrepos, badfn=badfn)
295 295
296 296 def diff(self, ctx2=None, match=None, changes=None, opts=None,
297 297 losedatafn=None, prefix='', relroot='', copy=None,
298 298 hunksfilterfn=None):
299 299 """Returns a diff generator for the given contexts and matcher"""
300 300 if ctx2 is None:
301 301 ctx2 = self.p1()
302 302 if ctx2 is not None:
303 303 ctx2 = self._repo[ctx2]
304 304 return patch.diff(self._repo, ctx2, self, match=match, changes=changes,
305 305 opts=opts, losedatafn=losedatafn, prefix=prefix,
306 306 relroot=relroot, copy=copy,
307 307 hunksfilterfn=hunksfilterfn)
308 308
309 309 def dirs(self):
310 310 return self._manifest.dirs()
311 311
312 312 def hasdir(self, dir):
313 313 return self._manifest.hasdir(dir)
314 314
315 315 def status(self, other=None, match=None, listignored=False,
316 316 listclean=False, listunknown=False, listsubrepos=False):
317 317 """return status of files between two nodes or node and working
318 318 directory.
319 319
320 320 If other is None, compare this node with working directory.
321 321
322 322 returns (modified, added, removed, deleted, unknown, ignored, clean)
323 323 """
324 324
325 325 ctx1 = self
326 326 ctx2 = self._repo[other]
327 327
328 328 # This next code block is, admittedly, fragile logic that tests for
329 329 # reversing the contexts and wouldn't need to exist if it weren't for
330 330 # the fast (and common) code path of comparing the working directory
331 331 # with its first parent.
332 332 #
333 333 # What we're aiming for here is the ability to call:
334 334 #
335 335 # workingctx.status(parentctx)
336 336 #
337 337 # If we always built the manifest for each context and compared those,
338 338 # then we'd be done. But the special case of the above call means we
339 339 # just copy the manifest of the parent.
340 340 reversed = False
341 341 if (not isinstance(ctx1, changectx)
342 342 and isinstance(ctx2, changectx)):
343 343 reversed = True
344 344 ctx1, ctx2 = ctx2, ctx1
345 345
346 346 match = match or matchmod.always(self._repo.root, self._repo.getcwd())
347 347 match = ctx2._matchstatus(ctx1, match)
348 348 r = scmutil.status([], [], [], [], [], [], [])
349 349 r = ctx2._buildstatus(ctx1, r, match, listignored, listclean,
350 350 listunknown)
351 351
352 352 if reversed:
353 353 # Reverse added and removed. Clear deleted, unknown and ignored as
354 354 # these make no sense to reverse.
355 355 r = scmutil.status(r.modified, r.removed, r.added, [], [], [],
356 356 r.clean)
357 357
358 358 if listsubrepos:
359 359 for subpath, sub in scmutil.itersubrepos(ctx1, ctx2):
360 360 try:
361 361 rev2 = ctx2.subrev(subpath)
362 362 except KeyError:
363 363 # A subrepo that existed in node1 was deleted between
364 364 # node1 and node2 (inclusive). Thus, ctx2's substate
365 365 # won't contain that subpath. The best we can do ignore it.
366 366 rev2 = None
367 367 submatch = matchmod.subdirmatcher(subpath, match)
368 368 s = sub.status(rev2, match=submatch, ignored=listignored,
369 369 clean=listclean, unknown=listunknown,
370 370 listsubrepos=True)
371 371 for rfiles, sfiles in zip(r, s):
372 372 rfiles.extend("%s/%s" % (subpath, f) for f in sfiles)
373 373
374 374 narrowmatch = self._repo.narrowmatch()
375 375 if not narrowmatch.always():
376 376 for l in r:
377 377 l[:] = list(filter(narrowmatch, l))
378 378 for l in r:
379 379 l.sort()
380 380
381 381 return r
382 382
383 383 class changectx(basectx):
384 384 """A changecontext object makes access to data related to a particular
385 385 changeset convenient. It represents a read-only context already present in
386 386 the repo."""
387 387 def __init__(self, repo, rev, node):
388 388 super(changectx, self).__init__(repo)
389 389 self._rev = rev
390 390 self._node = node
391 391
392 392 def __hash__(self):
393 393 try:
394 394 return hash(self._rev)
395 395 except AttributeError:
396 396 return id(self)
397 397
398 398 def __nonzero__(self):
399 399 return self._rev != nullrev
400 400
401 401 __bool__ = __nonzero__
402 402
403 403 @propertycache
404 404 def _changeset(self):
405 405 return self._repo.changelog.changelogrevision(self.rev())
406 406
407 407 @propertycache
408 408 def _manifest(self):
409 409 return self._manifestctx.read()
410 410
411 411 @property
412 412 def _manifestctx(self):
413 413 return self._repo.manifestlog[self._changeset.manifest]
414 414
415 415 @propertycache
416 416 def _manifestdelta(self):
417 417 return self._manifestctx.readdelta()
418 418
419 419 @propertycache
420 420 def _parents(self):
421 421 repo = self._repo
422 422 p1, p2 = repo.changelog.parentrevs(self._rev)
423 423 if p2 == nullrev:
424 424 return [repo[p1]]
425 425 return [repo[p1], repo[p2]]
426 426
427 427 def changeset(self):
428 428 c = self._changeset
429 429 return (
430 430 c.manifest,
431 431 c.user,
432 432 c.date,
433 433 c.files,
434 434 c.description,
435 435 c.extra,
436 436 )
437 437 def manifestnode(self):
438 438 return self._changeset.manifest
439 439
440 440 def user(self):
441 441 return self._changeset.user
442 442 def date(self):
443 443 return self._changeset.date
444 444 def files(self):
445 445 return self._changeset.files
446 446 def description(self):
447 447 return self._changeset.description
448 448 def branch(self):
449 449 return encoding.tolocal(self._changeset.extra.get("branch"))
450 450 def closesbranch(self):
451 451 return 'close' in self._changeset.extra
452 452 def extra(self):
453 453 """Return a dict of extra information."""
454 454 return self._changeset.extra
455 455 def tags(self):
456 456 """Return a list of byte tag names"""
457 457 return self._repo.nodetags(self._node)
458 458 def bookmarks(self):
459 459 """Return a list of byte bookmark names."""
460 460 return self._repo.nodebookmarks(self._node)
461 461 def phase(self):
462 462 return self._repo._phasecache.phase(self._repo, self._rev)
463 463 def hidden(self):
464 464 return self._rev in repoview.filterrevs(self._repo, 'visible')
465 465
466 466 def isinmemory(self):
467 467 return False
468 468
469 469 def children(self):
470 470 """return list of changectx contexts for each child changeset.
471 471
472 472 This returns only the immediate child changesets. Use descendants() to
473 473 recursively walk children.
474 474 """
475 475 c = self._repo.changelog.children(self._node)
476 476 return [self._repo[x] for x in c]
477 477
478 478 def ancestors(self):
479 479 for a in self._repo.changelog.ancestors([self._rev]):
480 480 yield self._repo[a]
481 481
482 482 def descendants(self):
483 483 """Recursively yield all children of the changeset.
484 484
485 485 For just the immediate children, use children()
486 486 """
487 487 for d in self._repo.changelog.descendants([self._rev]):
488 488 yield self._repo[d]
489 489
490 490 def filectx(self, path, fileid=None, filelog=None):
491 491 """get a file context from this changeset"""
492 492 if fileid is None:
493 493 fileid = self.filenode(path)
494 494 return filectx(self._repo, path, fileid=fileid,
495 495 changectx=self, filelog=filelog)
496 496
497 497 def ancestor(self, c2, warn=False):
498 498 """return the "best" ancestor context of self and c2
499 499
500 500 If there are multiple candidates, it will show a message and check
501 501 merge.preferancestor configuration before falling back to the
502 502 revlog ancestor."""
503 503 # deal with workingctxs
504 504 n2 = c2._node
505 505 if n2 is None:
506 506 n2 = c2._parents[0]._node
507 507 cahs = self._repo.changelog.commonancestorsheads(self._node, n2)
508 508 if not cahs:
509 509 anc = nullid
510 510 elif len(cahs) == 1:
511 511 anc = cahs[0]
512 512 else:
513 513 # experimental config: merge.preferancestor
514 514 for r in self._repo.ui.configlist('merge', 'preferancestor'):
515 515 try:
516 516 ctx = scmutil.revsymbol(self._repo, r)
517 517 except error.RepoLookupError:
518 518 continue
519 519 anc = ctx.node()
520 520 if anc in cahs:
521 521 break
522 522 else:
523 523 anc = self._repo.changelog.ancestor(self._node, n2)
524 524 if warn:
525 525 self._repo.ui.status(
526 526 (_("note: using %s as ancestor of %s and %s\n") %
527 527 (short(anc), short(self._node), short(n2))) +
528 528 ''.join(_(" alternatively, use --config "
529 529 "merge.preferancestor=%s\n") %
530 530 short(n) for n in sorted(cahs) if n != anc))
531 531 return self._repo[anc]
532 532
533 533 def isancestorof(self, other):
534 534 """True if this changeset is an ancestor of other"""
535 535 return self._repo.changelog.isancestorrev(self._rev, other._rev)
536 536
537 537 def walk(self, match):
538 538 '''Generates matching file names.'''
539 539
540 540 # Wrap match.bad method to have message with nodeid
541 541 def bad(fn, msg):
542 542 # The manifest doesn't know about subrepos, so don't complain about
543 543 # paths into valid subrepos.
544 544 if any(fn == s or fn.startswith(s + '/')
545 545 for s in self.substate):
546 546 return
547 547 match.bad(fn, _('no such file in rev %s') % self)
548 548
549 549 m = matchmod.badmatch(match, bad)
550 550 return self._manifest.walk(m)
551 551
552 552 def matches(self, match):
553 553 return self.walk(match)
554 554
555 555 class basefilectx(object):
556 556 """A filecontext object represents the common logic for its children:
557 557 filectx: read-only access to a filerevision that is already present
558 558 in the repo,
559 559 workingfilectx: a filecontext that represents files from the working
560 560 directory,
561 561 memfilectx: a filecontext that represents files in-memory,
562 562 """
563 563 @propertycache
564 564 def _filelog(self):
565 565 return self._repo.file(self._path)
566 566
567 567 @propertycache
568 568 def _changeid(self):
569 569 if r'_changeid' in self.__dict__:
570 570 return self._changeid
571 571 elif r'_changectx' in self.__dict__:
572 572 return self._changectx.rev()
573 573 elif r'_descendantrev' in self.__dict__:
574 574 # this file context was created from a revision with a known
575 575 # descendant, we can (lazily) correct for linkrev aliases
576 576 return self._adjustlinkrev(self._descendantrev)
577 577 else:
578 578 return self._filelog.linkrev(self._filerev)
579 579
580 580 @propertycache
581 581 def _filenode(self):
582 582 if r'_fileid' in self.__dict__:
583 583 return self._filelog.lookup(self._fileid)
584 584 else:
585 585 return self._changectx.filenode(self._path)
586 586
587 587 @propertycache
588 588 def _filerev(self):
589 589 return self._filelog.rev(self._filenode)
590 590
591 591 @propertycache
592 592 def _repopath(self):
593 593 return self._path
594 594
595 595 def __nonzero__(self):
596 596 try:
597 597 self._filenode
598 598 return True
599 599 except error.LookupError:
600 600 # file is missing
601 601 return False
602 602
603 603 __bool__ = __nonzero__
604 604
605 605 def __bytes__(self):
606 606 try:
607 607 return "%s@%s" % (self.path(), self._changectx)
608 608 except error.LookupError:
609 609 return "%s@???" % self.path()
610 610
611 611 __str__ = encoding.strmethod(__bytes__)
612 612
613 613 def __repr__(self):
614 614 return r"<%s %s>" % (type(self).__name__, str(self))
615 615
616 616 def __hash__(self):
617 617 try:
618 618 return hash((self._path, self._filenode))
619 619 except AttributeError:
620 620 return id(self)
621 621
622 622 def __eq__(self, other):
623 623 try:
624 624 return (type(self) == type(other) and self._path == other._path
625 625 and self._filenode == other._filenode)
626 626 except AttributeError:
627 627 return False
628 628
629 629 def __ne__(self, other):
630 630 return not (self == other)
631 631
632 632 def filerev(self):
633 633 return self._filerev
634 634 def filenode(self):
635 635 return self._filenode
636 636 @propertycache
637 637 def _flags(self):
638 638 return self._changectx.flags(self._path)
639 639 def flags(self):
640 640 return self._flags
641 641 def filelog(self):
642 642 return self._filelog
643 643 def rev(self):
644 644 return self._changeid
645 645 def linkrev(self):
646 646 return self._filelog.linkrev(self._filerev)
647 647 def node(self):
648 648 return self._changectx.node()
649 649 def hex(self):
650 650 return self._changectx.hex()
651 651 def user(self):
652 652 return self._changectx.user()
653 653 def date(self):
654 654 return self._changectx.date()
655 655 def files(self):
656 656 return self._changectx.files()
657 657 def description(self):
658 658 return self._changectx.description()
659 659 def branch(self):
660 660 return self._changectx.branch()
661 661 def extra(self):
662 662 return self._changectx.extra()
663 663 def phase(self):
664 664 return self._changectx.phase()
665 665 def phasestr(self):
666 666 return self._changectx.phasestr()
667 667 def obsolete(self):
668 668 return self._changectx.obsolete()
669 669 def instabilities(self):
670 670 return self._changectx.instabilities()
671 671 def manifest(self):
672 672 return self._changectx.manifest()
673 673 def changectx(self):
674 674 return self._changectx
675 675 def renamed(self):
676 676 return self._copied
677 677 def repo(self):
678 678 return self._repo
679 679 def size(self):
680 680 return len(self.data())
681 681
682 682 def path(self):
683 683 return self._path
684 684
685 685 def isbinary(self):
686 686 try:
687 687 return stringutil.binary(self.data())
688 688 except IOError:
689 689 return False
690 690 def isexec(self):
691 691 return 'x' in self.flags()
692 692 def islink(self):
693 693 return 'l' in self.flags()
694 694
695 695 def isabsent(self):
696 696 """whether this filectx represents a file not in self._changectx
697 697
698 698 This is mainly for merge code to detect change/delete conflicts. This is
699 699 expected to be True for all subclasses of basectx."""
700 700 return False
701 701
702 702 _customcmp = False
703 703 def cmp(self, fctx):
704 704 """compare with other file context
705 705
706 706 returns True if different than fctx.
707 707 """
708 708 if fctx._customcmp:
709 709 return fctx.cmp(self)
710 710
711 711 if (fctx._filenode is None
712 712 and (self._repo._encodefilterpats
713 713 # if file data starts with '\1\n', empty metadata block is
714 714 # prepended, which adds 4 bytes to filelog.size().
715 715 or self.size() - 4 == fctx.size())
716 716 or self.size() == fctx.size()):
717 717 return self._filelog.cmp(self._filenode, fctx.data())
718 718
719 719 return True
720 720
721 721 def _adjustlinkrev(self, srcrev, inclusive=False):
722 722 """return the first ancestor of <srcrev> introducing <fnode>
723 723
724 724 If the linkrev of the file revision does not point to an ancestor of
725 725 srcrev, we'll walk down the ancestors until we find one introducing
726 726 this file revision.
727 727
728 728 :srcrev: the changeset revision we search ancestors from
729 729 :inclusive: if true, the src revision will also be checked
730 730 """
731 731 repo = self._repo
732 732 cl = repo.unfiltered().changelog
733 733 mfl = repo.manifestlog
734 734 # fetch the linkrev
735 735 lkr = self.linkrev()
736 736 if srcrev == lkr:
737 737 return lkr
738 738 # hack to reuse ancestor computation when searching for renames
739 739 memberanc = getattr(self, '_ancestrycontext', None)
740 740 iteranc = None
741 741 if srcrev is None:
742 742 # wctx case, used by workingfilectx during mergecopy
743 743 revs = [p.rev() for p in self._repo[None].parents()]
744 744 inclusive = True # we skipped the real (revless) source
745 745 else:
746 746 revs = [srcrev]
747 747 if memberanc is None:
748 748 memberanc = iteranc = cl.ancestors(revs, lkr,
749 749 inclusive=inclusive)
750 750 # check if this linkrev is an ancestor of srcrev
751 751 if lkr not in memberanc:
752 752 if iteranc is None:
753 753 iteranc = cl.ancestors(revs, lkr, inclusive=inclusive)
754 754 fnode = self._filenode
755 755 path = self._path
756 756 for a in iteranc:
757 757 ac = cl.read(a) # get changeset data (we avoid object creation)
758 758 if path in ac[3]: # checking the 'files' field.
759 759 # The file has been touched, check if the content is
760 760 # similar to the one we search for.
761 761 if fnode == mfl[ac[0]].readfast().get(path):
762 762 return a
763 763 # In theory, we should never get out of that loop without a result.
764 764 # But if manifest uses a buggy file revision (not children of the
765 765 # one it replaces) we could. Such a buggy situation will likely
766 766 # result is crash somewhere else at to some point.
767 767 return lkr
768 768
769 769 def introrev(self):
770 770 """return the rev of the changeset which introduced this file revision
771 771
772 772 This method is different from linkrev because it take into account the
773 773 changeset the filectx was created from. It ensures the returned
774 774 revision is one of its ancestors. This prevents bugs from
775 775 'linkrev-shadowing' when a file revision is used by multiple
776 776 changesets.
777 777 """
778 778 attrs = vars(self)
779 779 hastoprev = (r'_changeid' in attrs or r'_changectx' in attrs)
780 780 if hastoprev:
781 781 return self._adjustlinkrev(self.rev(), inclusive=True)
782 782 else:
783 783 return self.linkrev()
784 784
785 785 def introfilectx(self):
786 786 """Return filectx having identical contents, but pointing to the
787 787 changeset revision where this filectx was introduced"""
788 788 introrev = self.introrev()
789 789 if self.rev() == introrev:
790 790 return self
791 791 return self.filectx(self.filenode(), changeid=introrev)
792 792
793 793 def _parentfilectx(self, path, fileid, filelog):
794 794 """create parent filectx keeping ancestry info for _adjustlinkrev()"""
795 795 fctx = filectx(self._repo, path, fileid=fileid, filelog=filelog)
796 796 if r'_changeid' in vars(self) or r'_changectx' in vars(self):
797 797 # If self is associated with a changeset (probably explicitly
798 798 # fed), ensure the created filectx is associated with a
799 799 # changeset that is an ancestor of self.changectx.
800 800 # This lets us later use _adjustlinkrev to get a correct link.
801 801 fctx._descendantrev = self.rev()
802 802 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
803 803 elif r'_descendantrev' in vars(self):
804 804 # Otherwise propagate _descendantrev if we have one associated.
805 805 fctx._descendantrev = self._descendantrev
806 806 fctx._ancestrycontext = getattr(self, '_ancestrycontext', None)
807 807 return fctx
808 808
809 809 def parents(self):
810 810 _path = self._path
811 811 fl = self._filelog
812 812 parents = self._filelog.parents(self._filenode)
813 813 pl = [(_path, node, fl) for node in parents if node != nullid]
814 814
815 815 r = fl.renamed(self._filenode)
816 816 if r:
817 817 # - In the simple rename case, both parent are nullid, pl is empty.
818 818 # - In case of merge, only one of the parent is null id and should
819 819 # be replaced with the rename information. This parent is -always-
820 820 # the first one.
821 821 #
822 822 # As null id have always been filtered out in the previous list
823 823 # comprehension, inserting to 0 will always result in "replacing
824 824 # first nullid parent with rename information.
825 825 pl.insert(0, (r[0], r[1], self._repo.file(r[0])))
826 826
827 827 return [self._parentfilectx(path, fnode, l) for path, fnode, l in pl]
828 828
829 829 def p1(self):
830 830 return self.parents()[0]
831 831
832 832 def p2(self):
833 833 p = self.parents()
834 834 if len(p) == 2:
835 835 return p[1]
836 836 return filectx(self._repo, self._path, fileid=-1, filelog=self._filelog)
837 837
838 838 def annotate(self, follow=False, skiprevs=None, diffopts=None):
839 839 """Returns a list of annotateline objects for each line in the file
840 840
841 841 - line.fctx is the filectx of the node where that line was last changed
842 842 - line.lineno is the line number at the first appearance in the managed
843 843 file
844 844 - line.text is the data on that line (including newline character)
845 845 """
846 846 getlog = util.lrucachefunc(lambda x: self._repo.file(x))
847 847
848 848 def parents(f):
849 849 # Cut _descendantrev here to mitigate the penalty of lazy linkrev
850 850 # adjustment. Otherwise, p._adjustlinkrev() would walk changelog
851 851 # from the topmost introrev (= srcrev) down to p.linkrev() if it
852 852 # isn't an ancestor of the srcrev.
853 853 f._changeid
854 854 pl = f.parents()
855 855
856 856 # Don't return renamed parents if we aren't following.
857 857 if not follow:
858 858 pl = [p for p in pl if p.path() == f.path()]
859 859
860 860 # renamed filectx won't have a filelog yet, so set it
861 861 # from the cache to save time
862 862 for p in pl:
863 863 if not r'_filelog' in p.__dict__:
864 864 p._filelog = getlog(p.path())
865 865
866 866 return pl
867 867
868 868 # use linkrev to find the first changeset where self appeared
869 869 base = self.introfilectx()
870 870 if getattr(base, '_ancestrycontext', None) is None:
871 871 cl = self._repo.changelog
872 872 if base.rev() is None:
873 873 # wctx is not inclusive, but works because _ancestrycontext
874 874 # is used to test filelog revisions
875 875 ac = cl.ancestors([p.rev() for p in base.parents()],
876 876 inclusive=True)
877 877 else:
878 878 ac = cl.ancestors([base.rev()], inclusive=True)
879 879 base._ancestrycontext = ac
880 880
881 881 return dagop.annotate(base, parents, skiprevs=skiprevs,
882 882 diffopts=diffopts)
883 883
884 884 def ancestors(self, followfirst=False):
885 885 visit = {}
886 886 c = self
887 887 if followfirst:
888 888 cut = 1
889 889 else:
890 890 cut = None
891 891
892 892 while True:
893 893 for parent in c.parents()[:cut]:
894 894 visit[(parent.linkrev(), parent.filenode())] = parent
895 895 if not visit:
896 896 break
897 897 c = visit.pop(max(visit))
898 898 yield c
899 899
900 900 def decodeddata(self):
901 901 """Returns `data()` after running repository decoding filters.
902 902
903 903 This is often equivalent to how the data would be expressed on disk.
904 904 """
905 905 return self._repo.wwritedata(self.path(), self.data())
906 906
907 907 class filectx(basefilectx):
908 908 """A filecontext object makes access to data related to a particular
909 909 filerevision convenient."""
910 910 def __init__(self, repo, path, changeid=None, fileid=None,
911 911 filelog=None, changectx=None):
912 """changeid can be a changeset revision, node, or tag.
912 """changeid must be a revision number, if specified.
913 913 fileid can be a file revision or node."""
914 914 self._repo = repo
915 915 self._path = path
916 916
917 917 assert (changeid is not None
918 918 or fileid is not None
919 919 or changectx is not None), \
920 920 ("bad args: changeid=%r, fileid=%r, changectx=%r"
921 921 % (changeid, fileid, changectx))
922 922
923 923 if filelog is not None:
924 924 self._filelog = filelog
925 925
926 926 if changeid is not None:
927 927 self._changeid = changeid
928 928 if changectx is not None:
929 929 self._changectx = changectx
930 930 if fileid is not None:
931 931 self._fileid = fileid
932 932
933 933 @propertycache
934 934 def _changectx(self):
935 935 try:
936 936 return self._repo[self._changeid]
937 937 except error.FilteredRepoLookupError:
938 938 # Linkrev may point to any revision in the repository. When the
939 939 # repository is filtered this may lead to `filectx` trying to build
940 940 # `changectx` for filtered revision. In such case we fallback to
941 941 # creating `changectx` on the unfiltered version of the reposition.
942 942 # This fallback should not be an issue because `changectx` from
943 943 # `filectx` are not used in complex operations that care about
944 944 # filtering.
945 945 #
946 946 # This fallback is a cheap and dirty fix that prevent several
947 947 # crashes. It does not ensure the behavior is correct. However the
948 948 # behavior was not correct before filtering either and "incorrect
949 949 # behavior" is seen as better as "crash"
950 950 #
951 951 # Linkrevs have several serious troubles with filtering that are
952 952 # complicated to solve. Proper handling of the issue here should be
953 953 # considered when solving linkrev issue are on the table.
954 954 return self._repo.unfiltered()[self._changeid]
955 955
956 956 def filectx(self, fileid, changeid=None):
957 957 '''opens an arbitrary revision of the file without
958 958 opening a new filelog'''
959 959 return filectx(self._repo, self._path, fileid=fileid,
960 960 filelog=self._filelog, changeid=changeid)
961 961
962 962 def rawdata(self):
963 963 return self._filelog.revision(self._filenode, raw=True)
964 964
965 965 def rawflags(self):
966 966 """low-level revlog flags"""
967 967 return self._filelog.flags(self._filerev)
968 968
969 969 def data(self):
970 970 try:
971 971 return self._filelog.read(self._filenode)
972 972 except error.CensoredNodeError:
973 973 if self._repo.ui.config("censor", "policy") == "ignore":
974 974 return ""
975 975 raise error.Abort(_("censored node: %s") % short(self._filenode),
976 976 hint=_("set censor.policy to ignore errors"))
977 977
978 978 def size(self):
979 979 return self._filelog.size(self._filerev)
980 980
981 981 @propertycache
982 982 def _copied(self):
983 983 """check if file was actually renamed in this changeset revision
984 984
985 985 If rename logged in file revision, we report copy for changeset only
986 986 if file revisions linkrev points back to the changeset in question
987 987 or both changeset parents contain different file revisions.
988 988 """
989 989
990 990 renamed = self._filelog.renamed(self._filenode)
991 991 if not renamed:
992 992 return None
993 993
994 994 if self.rev() == self.linkrev():
995 995 return renamed
996 996
997 997 name = self.path()
998 998 fnode = self._filenode
999 999 for p in self._changectx.parents():
1000 1000 try:
1001 1001 if fnode == p.filenode(name):
1002 1002 return None
1003 1003 except error.LookupError:
1004 1004 pass
1005 1005 return renamed
1006 1006
1007 1007 def children(self):
1008 1008 # hard for renames
1009 1009 c = self._filelog.children(self._filenode)
1010 1010 return [filectx(self._repo, self._path, fileid=x,
1011 1011 filelog=self._filelog) for x in c]
1012 1012
1013 1013 class committablectx(basectx):
1014 1014 """A committablectx object provides common functionality for a context that
1015 1015 wants the ability to commit, e.g. workingctx or memctx."""
1016 1016 def __init__(self, repo, text="", user=None, date=None, extra=None,
1017 1017 changes=None):
1018 1018 super(committablectx, self).__init__(repo)
1019 1019 self._rev = None
1020 1020 self._node = None
1021 1021 self._text = text
1022 1022 if date:
1023 1023 self._date = dateutil.parsedate(date)
1024 1024 if user:
1025 1025 self._user = user
1026 1026 if changes:
1027 1027 self._status = changes
1028 1028
1029 1029 self._extra = {}
1030 1030 if extra:
1031 1031 self._extra = extra.copy()
1032 1032 if 'branch' not in self._extra:
1033 1033 try:
1034 1034 branch = encoding.fromlocal(self._repo.dirstate.branch())
1035 1035 except UnicodeDecodeError:
1036 1036 raise error.Abort(_('branch name not in UTF-8!'))
1037 1037 self._extra['branch'] = branch
1038 1038 if self._extra['branch'] == '':
1039 1039 self._extra['branch'] = 'default'
1040 1040
1041 1041 def __bytes__(self):
1042 1042 return bytes(self._parents[0]) + "+"
1043 1043
1044 1044 __str__ = encoding.strmethod(__bytes__)
1045 1045
1046 1046 def __nonzero__(self):
1047 1047 return True
1048 1048
1049 1049 __bool__ = __nonzero__
1050 1050
1051 1051 def _buildflagfunc(self):
1052 1052 # Create a fallback function for getting file flags when the
1053 1053 # filesystem doesn't support them
1054 1054
1055 1055 copiesget = self._repo.dirstate.copies().get
1056 1056 parents = self.parents()
1057 1057 if len(parents) < 2:
1058 1058 # when we have one parent, it's easy: copy from parent
1059 1059 man = parents[0].manifest()
1060 1060 def func(f):
1061 1061 f = copiesget(f, f)
1062 1062 return man.flags(f)
1063 1063 else:
1064 1064 # merges are tricky: we try to reconstruct the unstored
1065 1065 # result from the merge (issue1802)
1066 1066 p1, p2 = parents
1067 1067 pa = p1.ancestor(p2)
1068 1068 m1, m2, ma = p1.manifest(), p2.manifest(), pa.manifest()
1069 1069
1070 1070 def func(f):
1071 1071 f = copiesget(f, f) # may be wrong for merges with copies
1072 1072 fl1, fl2, fla = m1.flags(f), m2.flags(f), ma.flags(f)
1073 1073 if fl1 == fl2:
1074 1074 return fl1
1075 1075 if fl1 == fla:
1076 1076 return fl2
1077 1077 if fl2 == fla:
1078 1078 return fl1
1079 1079 return '' # punt for conflicts
1080 1080
1081 1081 return func
1082 1082
1083 1083 @propertycache
1084 1084 def _flagfunc(self):
1085 1085 return self._repo.dirstate.flagfunc(self._buildflagfunc)
1086 1086
1087 1087 @propertycache
1088 1088 def _status(self):
1089 1089 return self._repo.status()
1090 1090
1091 1091 @propertycache
1092 1092 def _user(self):
1093 1093 return self._repo.ui.username()
1094 1094
1095 1095 @propertycache
1096 1096 def _date(self):
1097 1097 ui = self._repo.ui
1098 1098 date = ui.configdate('devel', 'default-date')
1099 1099 if date is None:
1100 1100 date = dateutil.makedate()
1101 1101 return date
1102 1102
1103 1103 def subrev(self, subpath):
1104 1104 return None
1105 1105
1106 1106 def manifestnode(self):
1107 1107 return None
1108 1108 def user(self):
1109 1109 return self._user or self._repo.ui.username()
1110 1110 def date(self):
1111 1111 return self._date
1112 1112 def description(self):
1113 1113 return self._text
1114 1114 def files(self):
1115 1115 return sorted(self._status.modified + self._status.added +
1116 1116 self._status.removed)
1117 1117
1118 1118 def modified(self):
1119 1119 return self._status.modified
1120 1120 def added(self):
1121 1121 return self._status.added
1122 1122 def removed(self):
1123 1123 return self._status.removed
1124 1124 def deleted(self):
1125 1125 return self._status.deleted
1126 1126 def branch(self):
1127 1127 return encoding.tolocal(self._extra['branch'])
1128 1128 def closesbranch(self):
1129 1129 return 'close' in self._extra
1130 1130 def extra(self):
1131 1131 return self._extra
1132 1132
1133 1133 def isinmemory(self):
1134 1134 return False
1135 1135
1136 1136 def tags(self):
1137 1137 return []
1138 1138
1139 1139 def bookmarks(self):
1140 1140 b = []
1141 1141 for p in self.parents():
1142 1142 b.extend(p.bookmarks())
1143 1143 return b
1144 1144
1145 1145 def phase(self):
1146 1146 phase = phases.draft # default phase to draft
1147 1147 for p in self.parents():
1148 1148 phase = max(phase, p.phase())
1149 1149 return phase
1150 1150
1151 1151 def hidden(self):
1152 1152 return False
1153 1153
1154 1154 def children(self):
1155 1155 return []
1156 1156
1157 1157 def flags(self, path):
1158 1158 if r'_manifest' in self.__dict__:
1159 1159 try:
1160 1160 return self._manifest.flags(path)
1161 1161 except KeyError:
1162 1162 return ''
1163 1163
1164 1164 try:
1165 1165 return self._flagfunc(path)
1166 1166 except OSError:
1167 1167 return ''
1168 1168
1169 1169 def ancestor(self, c2):
1170 1170 """return the "best" ancestor context of self and c2"""
1171 1171 return self._parents[0].ancestor(c2) # punt on two parents for now
1172 1172
1173 1173 def walk(self, match):
1174 1174 '''Generates matching file names.'''
1175 1175 return sorted(self._repo.dirstate.walk(match,
1176 1176 subrepos=sorted(self.substate),
1177 1177 unknown=True, ignored=False))
1178 1178
1179 1179 def matches(self, match):
1180 1180 ds = self._repo.dirstate
1181 1181 return sorted(f for f in ds.matches(match) if ds[f] != 'r')
1182 1182
1183 1183 def ancestors(self):
1184 1184 for p in self._parents:
1185 1185 yield p
1186 1186 for a in self._repo.changelog.ancestors(
1187 1187 [p.rev() for p in self._parents]):
1188 1188 yield self._repo[a]
1189 1189
1190 1190 def markcommitted(self, node):
1191 1191 """Perform post-commit cleanup necessary after committing this ctx
1192 1192
1193 1193 Specifically, this updates backing stores this working context
1194 1194 wraps to reflect the fact that the changes reflected by this
1195 1195 workingctx have been committed. For example, it marks
1196 1196 modified and added files as normal in the dirstate.
1197 1197
1198 1198 """
1199 1199
1200 1200 with self._repo.dirstate.parentchange():
1201 1201 for f in self.modified() + self.added():
1202 1202 self._repo.dirstate.normal(f)
1203 1203 for f in self.removed():
1204 1204 self._repo.dirstate.drop(f)
1205 1205 self._repo.dirstate.setparents(node)
1206 1206
1207 1207 # write changes out explicitly, because nesting wlock at
1208 1208 # runtime may prevent 'wlock.release()' in 'repo.commit()'
1209 1209 # from immediately doing so for subsequent changing files
1210 1210 self._repo.dirstate.write(self._repo.currenttransaction())
1211 1211
1212 1212 def dirty(self, missing=False, merge=True, branch=True):
1213 1213 return False
1214 1214
1215 1215 class workingctx(committablectx):
1216 1216 """A workingctx object makes access to data related to
1217 1217 the current working directory convenient.
1218 1218 date - any valid date string or (unixtime, offset), or None.
1219 1219 user - username string, or None.
1220 1220 extra - a dictionary of extra values, or None.
1221 1221 changes - a list of file lists as returned by localrepo.status()
1222 1222 or None to use the repository status.
1223 1223 """
1224 1224 def __init__(self, repo, text="", user=None, date=None, extra=None,
1225 1225 changes=None):
1226 1226 super(workingctx, self).__init__(repo, text, user, date, extra, changes)
1227 1227
1228 1228 def __iter__(self):
1229 1229 d = self._repo.dirstate
1230 1230 for f in d:
1231 1231 if d[f] != 'r':
1232 1232 yield f
1233 1233
1234 1234 def __contains__(self, key):
1235 1235 return self._repo.dirstate[key] not in "?r"
1236 1236
1237 1237 def hex(self):
1238 1238 return hex(wdirid)
1239 1239
1240 1240 @propertycache
1241 1241 def _parents(self):
1242 1242 p = self._repo.dirstate.parents()
1243 1243 if p[1] == nullid:
1244 1244 p = p[:-1]
1245 1245 # use unfiltered repo to delay/avoid loading obsmarkers
1246 1246 unfi = self._repo.unfiltered()
1247 1247 return [changectx(self._repo, unfi.changelog.rev(n), n) for n in p]
1248 1248
1249 1249 def _fileinfo(self, path):
1250 1250 # populate __dict__['_manifest'] as workingctx has no _manifestdelta
1251 1251 self._manifest
1252 1252 return super(workingctx, self)._fileinfo(path)
1253 1253
1254 1254 def filectx(self, path, filelog=None):
1255 1255 """get a file context from the working directory"""
1256 1256 return workingfilectx(self._repo, path, workingctx=self,
1257 1257 filelog=filelog)
1258 1258
1259 1259 def dirty(self, missing=False, merge=True, branch=True):
1260 1260 "check whether a working directory is modified"
1261 1261 # check subrepos first
1262 1262 for s in sorted(self.substate):
1263 1263 if self.sub(s).dirty(missing=missing):
1264 1264 return True
1265 1265 # check current working dir
1266 1266 return ((merge and self.p2()) or
1267 1267 (branch and self.branch() != self.p1().branch()) or
1268 1268 self.modified() or self.added() or self.removed() or
1269 1269 (missing and self.deleted()))
1270 1270
1271 1271 def add(self, list, prefix=""):
1272 1272 with self._repo.wlock():
1273 1273 ui, ds = self._repo.ui, self._repo.dirstate
1274 1274 uipath = lambda f: ds.pathto(pathutil.join(prefix, f))
1275 1275 rejected = []
1276 1276 lstat = self._repo.wvfs.lstat
1277 1277 for f in list:
1278 1278 # ds.pathto() returns an absolute file when this is invoked from
1279 1279 # the keyword extension. That gets flagged as non-portable on
1280 1280 # Windows, since it contains the drive letter and colon.
1281 1281 scmutil.checkportable(ui, os.path.join(prefix, f))
1282 1282 try:
1283 1283 st = lstat(f)
1284 1284 except OSError:
1285 1285 ui.warn(_("%s does not exist!\n") % uipath(f))
1286 1286 rejected.append(f)
1287 1287 continue
1288 1288 limit = ui.configbytes('ui', 'large-file-limit')
1289 1289 if limit != 0 and st.st_size > limit:
1290 1290 ui.warn(_("%s: up to %d MB of RAM may be required "
1291 1291 "to manage this file\n"
1292 1292 "(use 'hg revert %s' to cancel the "
1293 1293 "pending addition)\n")
1294 1294 % (f, 3 * st.st_size // 1000000, uipath(f)))
1295 1295 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1296 1296 ui.warn(_("%s not added: only files and symlinks "
1297 1297 "supported currently\n") % uipath(f))
1298 1298 rejected.append(f)
1299 1299 elif ds[f] in 'amn':
1300 1300 ui.warn(_("%s already tracked!\n") % uipath(f))
1301 1301 elif ds[f] == 'r':
1302 1302 ds.normallookup(f)
1303 1303 else:
1304 1304 ds.add(f)
1305 1305 return rejected
1306 1306
1307 1307 def forget(self, files, prefix=""):
1308 1308 with self._repo.wlock():
1309 1309 ds = self._repo.dirstate
1310 1310 uipath = lambda f: ds.pathto(pathutil.join(prefix, f))
1311 1311 rejected = []
1312 1312 for f in files:
1313 1313 if f not in self._repo.dirstate:
1314 1314 self._repo.ui.warn(_("%s not tracked!\n") % uipath(f))
1315 1315 rejected.append(f)
1316 1316 elif self._repo.dirstate[f] != 'a':
1317 1317 self._repo.dirstate.remove(f)
1318 1318 else:
1319 1319 self._repo.dirstate.drop(f)
1320 1320 return rejected
1321 1321
1322 1322 def undelete(self, list):
1323 1323 pctxs = self.parents()
1324 1324 with self._repo.wlock():
1325 1325 ds = self._repo.dirstate
1326 1326 for f in list:
1327 1327 if self._repo.dirstate[f] != 'r':
1328 1328 self._repo.ui.warn(_("%s not removed!\n") % ds.pathto(f))
1329 1329 else:
1330 1330 fctx = f in pctxs[0] and pctxs[0][f] or pctxs[1][f]
1331 1331 t = fctx.data()
1332 1332 self._repo.wwrite(f, t, fctx.flags())
1333 1333 self._repo.dirstate.normal(f)
1334 1334
1335 1335 def copy(self, source, dest):
1336 1336 try:
1337 1337 st = self._repo.wvfs.lstat(dest)
1338 1338 except OSError as err:
1339 1339 if err.errno != errno.ENOENT:
1340 1340 raise
1341 1341 self._repo.ui.warn(_("%s does not exist!\n")
1342 1342 % self._repo.dirstate.pathto(dest))
1343 1343 return
1344 1344 if not (stat.S_ISREG(st.st_mode) or stat.S_ISLNK(st.st_mode)):
1345 1345 self._repo.ui.warn(_("copy failed: %s is not a file or a "
1346 1346 "symbolic link\n")
1347 1347 % self._repo.dirstate.pathto(dest))
1348 1348 else:
1349 1349 with self._repo.wlock():
1350 1350 if self._repo.dirstate[dest] in '?':
1351 1351 self._repo.dirstate.add(dest)
1352 1352 elif self._repo.dirstate[dest] in 'r':
1353 1353 self._repo.dirstate.normallookup(dest)
1354 1354 self._repo.dirstate.copy(source, dest)
1355 1355
1356 1356 def match(self, pats=None, include=None, exclude=None, default='glob',
1357 1357 listsubrepos=False, badfn=None):
1358 1358 r = self._repo
1359 1359
1360 1360 # Only a case insensitive filesystem needs magic to translate user input
1361 1361 # to actual case in the filesystem.
1362 1362 icasefs = not util.fscasesensitive(r.root)
1363 1363 return matchmod.match(r.root, r.getcwd(), pats, include, exclude,
1364 1364 default, auditor=r.auditor, ctx=self,
1365 1365 listsubrepos=listsubrepos, badfn=badfn,
1366 1366 icasefs=icasefs)
1367 1367
1368 1368 def _filtersuspectsymlink(self, files):
1369 1369 if not files or self._repo.dirstate._checklink:
1370 1370 return files
1371 1371
1372 1372 # Symlink placeholders may get non-symlink-like contents
1373 1373 # via user error or dereferencing by NFS or Samba servers,
1374 1374 # so we filter out any placeholders that don't look like a
1375 1375 # symlink
1376 1376 sane = []
1377 1377 for f in files:
1378 1378 if self.flags(f) == 'l':
1379 1379 d = self[f].data()
1380 1380 if (d == '' or len(d) >= 1024 or '\n' in d
1381 1381 or stringutil.binary(d)):
1382 1382 self._repo.ui.debug('ignoring suspect symlink placeholder'
1383 1383 ' "%s"\n' % f)
1384 1384 continue
1385 1385 sane.append(f)
1386 1386 return sane
1387 1387
1388 1388 def _checklookup(self, files):
1389 1389 # check for any possibly clean files
1390 1390 if not files:
1391 1391 return [], [], []
1392 1392
1393 1393 modified = []
1394 1394 deleted = []
1395 1395 fixup = []
1396 1396 pctx = self._parents[0]
1397 1397 # do a full compare of any files that might have changed
1398 1398 for f in sorted(files):
1399 1399 try:
1400 1400 # This will return True for a file that got replaced by a
1401 1401 # directory in the interim, but fixing that is pretty hard.
1402 1402 if (f not in pctx or self.flags(f) != pctx.flags(f)
1403 1403 or pctx[f].cmp(self[f])):
1404 1404 modified.append(f)
1405 1405 else:
1406 1406 fixup.append(f)
1407 1407 except (IOError, OSError):
1408 1408 # A file become inaccessible in between? Mark it as deleted,
1409 1409 # matching dirstate behavior (issue5584).
1410 1410 # The dirstate has more complex behavior around whether a
1411 1411 # missing file matches a directory, etc, but we don't need to
1412 1412 # bother with that: if f has made it to this point, we're sure
1413 1413 # it's in the dirstate.
1414 1414 deleted.append(f)
1415 1415
1416 1416 return modified, deleted, fixup
1417 1417
1418 1418 def _poststatusfixup(self, status, fixup):
1419 1419 """update dirstate for files that are actually clean"""
1420 1420 poststatus = self._repo.postdsstatus()
1421 1421 if fixup or poststatus:
1422 1422 try:
1423 1423 oldid = self._repo.dirstate.identity()
1424 1424
1425 1425 # updating the dirstate is optional
1426 1426 # so we don't wait on the lock
1427 1427 # wlock can invalidate the dirstate, so cache normal _after_
1428 1428 # taking the lock
1429 1429 with self._repo.wlock(False):
1430 1430 if self._repo.dirstate.identity() == oldid:
1431 1431 if fixup:
1432 1432 normal = self._repo.dirstate.normal
1433 1433 for f in fixup:
1434 1434 normal(f)
1435 1435 # write changes out explicitly, because nesting
1436 1436 # wlock at runtime may prevent 'wlock.release()'
1437 1437 # after this block from doing so for subsequent
1438 1438 # changing files
1439 1439 tr = self._repo.currenttransaction()
1440 1440 self._repo.dirstate.write(tr)
1441 1441
1442 1442 if poststatus:
1443 1443 for ps in poststatus:
1444 1444 ps(self, status)
1445 1445 else:
1446 1446 # in this case, writing changes out breaks
1447 1447 # consistency, because .hg/dirstate was
1448 1448 # already changed simultaneously after last
1449 1449 # caching (see also issue5584 for detail)
1450 1450 self._repo.ui.debug('skip updating dirstate: '
1451 1451 'identity mismatch\n')
1452 1452 except error.LockError:
1453 1453 pass
1454 1454 finally:
1455 1455 # Even if the wlock couldn't be grabbed, clear out the list.
1456 1456 self._repo.clearpostdsstatus()
1457 1457
1458 1458 def _dirstatestatus(self, match, ignored=False, clean=False, unknown=False):
1459 1459 '''Gets the status from the dirstate -- internal use only.'''
1460 1460 subrepos = []
1461 1461 if '.hgsub' in self:
1462 1462 subrepos = sorted(self.substate)
1463 1463 cmp, s = self._repo.dirstate.status(match, subrepos, ignored=ignored,
1464 1464 clean=clean, unknown=unknown)
1465 1465
1466 1466 # check for any possibly clean files
1467 1467 fixup = []
1468 1468 if cmp:
1469 1469 modified2, deleted2, fixup = self._checklookup(cmp)
1470 1470 s.modified.extend(modified2)
1471 1471 s.deleted.extend(deleted2)
1472 1472
1473 1473 if fixup and clean:
1474 1474 s.clean.extend(fixup)
1475 1475
1476 1476 self._poststatusfixup(s, fixup)
1477 1477
1478 1478 if match.always():
1479 1479 # cache for performance
1480 1480 if s.unknown or s.ignored or s.clean:
1481 1481 # "_status" is cached with list*=False in the normal route
1482 1482 self._status = scmutil.status(s.modified, s.added, s.removed,
1483 1483 s.deleted, [], [], [])
1484 1484 else:
1485 1485 self._status = s
1486 1486
1487 1487 return s
1488 1488
1489 1489 @propertycache
1490 1490 def _manifest(self):
1491 1491 """generate a manifest corresponding to the values in self._status
1492 1492
1493 1493 This reuse the file nodeid from parent, but we use special node
1494 1494 identifiers for added and modified files. This is used by manifests
1495 1495 merge to see that files are different and by update logic to avoid
1496 1496 deleting newly added files.
1497 1497 """
1498 1498 return self._buildstatusmanifest(self._status)
1499 1499
1500 1500 def _buildstatusmanifest(self, status):
1501 1501 """Builds a manifest that includes the given status results."""
1502 1502 parents = self.parents()
1503 1503
1504 1504 man = parents[0].manifest().copy()
1505 1505
1506 1506 ff = self._flagfunc
1507 1507 for i, l in ((addednodeid, status.added),
1508 1508 (modifiednodeid, status.modified)):
1509 1509 for f in l:
1510 1510 man[f] = i
1511 1511 try:
1512 1512 man.setflag(f, ff(f))
1513 1513 except OSError:
1514 1514 pass
1515 1515
1516 1516 for f in status.deleted + status.removed:
1517 1517 if f in man:
1518 1518 del man[f]
1519 1519
1520 1520 return man
1521 1521
1522 1522 def _buildstatus(self, other, s, match, listignored, listclean,
1523 1523 listunknown):
1524 1524 """build a status with respect to another context
1525 1525
1526 1526 This includes logic for maintaining the fast path of status when
1527 1527 comparing the working directory against its parent, which is to skip
1528 1528 building a new manifest if self (working directory) is not comparing
1529 1529 against its parent (repo['.']).
1530 1530 """
1531 1531 s = self._dirstatestatus(match, listignored, listclean, listunknown)
1532 1532 # Filter out symlinks that, in the case of FAT32 and NTFS filesystems,
1533 1533 # might have accidentally ended up with the entire contents of the file
1534 1534 # they are supposed to be linking to.
1535 1535 s.modified[:] = self._filtersuspectsymlink(s.modified)
1536 1536 if other != self._repo['.']:
1537 1537 s = super(workingctx, self)._buildstatus(other, s, match,
1538 1538 listignored, listclean,
1539 1539 listunknown)
1540 1540 return s
1541 1541
1542 1542 def _matchstatus(self, other, match):
1543 1543 """override the match method with a filter for directory patterns
1544 1544
1545 1545 We use inheritance to customize the match.bad method only in cases of
1546 1546 workingctx since it belongs only to the working directory when
1547 1547 comparing against the parent changeset.
1548 1548
1549 1549 If we aren't comparing against the working directory's parent, then we
1550 1550 just use the default match object sent to us.
1551 1551 """
1552 1552 if other != self._repo['.']:
1553 1553 def bad(f, msg):
1554 1554 # 'f' may be a directory pattern from 'match.files()',
1555 1555 # so 'f not in ctx1' is not enough
1556 1556 if f not in other and not other.hasdir(f):
1557 1557 self._repo.ui.warn('%s: %s\n' %
1558 1558 (self._repo.dirstate.pathto(f), msg))
1559 1559 match.bad = bad
1560 1560 return match
1561 1561
1562 1562 def markcommitted(self, node):
1563 1563 super(workingctx, self).markcommitted(node)
1564 1564
1565 1565 sparse.aftercommit(self._repo, node)
1566 1566
1567 1567 class committablefilectx(basefilectx):
1568 1568 """A committablefilectx provides common functionality for a file context
1569 1569 that wants the ability to commit, e.g. workingfilectx or memfilectx."""
1570 1570 def __init__(self, repo, path, filelog=None, ctx=None):
1571 1571 self._repo = repo
1572 1572 self._path = path
1573 1573 self._changeid = None
1574 1574 self._filerev = self._filenode = None
1575 1575
1576 1576 if filelog is not None:
1577 1577 self._filelog = filelog
1578 1578 if ctx:
1579 1579 self._changectx = ctx
1580 1580
1581 1581 def __nonzero__(self):
1582 1582 return True
1583 1583
1584 1584 __bool__ = __nonzero__
1585 1585
1586 1586 def linkrev(self):
1587 1587 # linked to self._changectx no matter if file is modified or not
1588 1588 return self.rev()
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 [self._parentfilectx(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 (self._repo.wvfs.lstat(self._path)[stat.ST_MTIME], tz)
1638 1638 except OSError as err:
1639 1639 if err.errno != errno.ENOENT:
1640 1640 raise
1641 1641 return (t, tz)
1642 1642
1643 1643 def exists(self):
1644 1644 return self._repo.wvfs.exists(self._path)
1645 1645
1646 1646 def lexists(self):
1647 1647 return self._repo.wvfs.lexists(self._path)
1648 1648
1649 1649 def audit(self):
1650 1650 return self._repo.wvfs.audit(self._path)
1651 1651
1652 1652 def cmp(self, fctx):
1653 1653 """compare with other file context
1654 1654
1655 1655 returns True if different than fctx.
1656 1656 """
1657 1657 # fctx should be a filectx (not a workingfilectx)
1658 1658 # invert comparison to reuse the same code path
1659 1659 return fctx.cmp(self)
1660 1660
1661 1661 def remove(self, ignoremissing=False):
1662 1662 """wraps unlink for a repo's working directory"""
1663 1663 rmdir = self._repo.ui.configbool('experimental', 'removeemptydirs')
1664 1664 self._repo.wvfs.unlinkpath(self._path, ignoremissing=ignoremissing,
1665 1665 rmdir=rmdir)
1666 1666
1667 1667 def write(self, data, flags, backgroundclose=False, **kwargs):
1668 1668 """wraps repo.wwrite"""
1669 1669 self._repo.wwrite(self._path, data, flags,
1670 1670 backgroundclose=backgroundclose,
1671 1671 **kwargs)
1672 1672
1673 1673 def markcopied(self, src):
1674 1674 """marks this file a copy of `src`"""
1675 1675 if self._repo.dirstate[self._path] in "nma":
1676 1676 self._repo.dirstate.copy(src, self._path)
1677 1677
1678 1678 def clearunknown(self):
1679 1679 """Removes conflicting items in the working directory so that
1680 1680 ``write()`` can be called successfully.
1681 1681 """
1682 1682 wvfs = self._repo.wvfs
1683 1683 f = self._path
1684 1684 wvfs.audit(f)
1685 1685 if self._repo.ui.configbool('experimental', 'merge.checkpathconflicts'):
1686 1686 # remove files under the directory as they should already be
1687 1687 # warned and backed up
1688 1688 if wvfs.isdir(f) and not wvfs.islink(f):
1689 1689 wvfs.rmtree(f, forcibly=True)
1690 1690 for p in reversed(list(util.finddirs(f))):
1691 1691 if wvfs.isfileorlink(p):
1692 1692 wvfs.unlink(p)
1693 1693 break
1694 1694 else:
1695 1695 # don't remove files if path conflicts are not processed
1696 1696 if wvfs.isdir(f) and not wvfs.islink(f):
1697 1697 wvfs.removedirs(f)
1698 1698
1699 1699 def setflags(self, l, x):
1700 1700 self._repo.wvfs.setflags(self._path, l, x)
1701 1701
1702 1702 class overlayworkingctx(committablectx):
1703 1703 """Wraps another mutable context with a write-back cache that can be
1704 1704 converted into a commit context.
1705 1705
1706 1706 self._cache[path] maps to a dict with keys: {
1707 1707 'exists': bool?
1708 1708 'date': date?
1709 1709 'data': str?
1710 1710 'flags': str?
1711 1711 'copied': str? (path or None)
1712 1712 }
1713 1713 If `exists` is True, `flags` must be non-None and 'date' is non-None. If it
1714 1714 is `False`, the file was deleted.
1715 1715 """
1716 1716
1717 1717 def __init__(self, repo):
1718 1718 super(overlayworkingctx, self).__init__(repo)
1719 1719 self.clean()
1720 1720
1721 1721 def setbase(self, wrappedctx):
1722 1722 self._wrappedctx = wrappedctx
1723 1723 self._parents = [wrappedctx]
1724 1724 # Drop old manifest cache as it is now out of date.
1725 1725 # This is necessary when, e.g., rebasing several nodes with one
1726 1726 # ``overlayworkingctx`` (e.g. with --collapse).
1727 1727 util.clearcachedproperty(self, '_manifest')
1728 1728
1729 1729 def data(self, path):
1730 1730 if self.isdirty(path):
1731 1731 if self._cache[path]['exists']:
1732 1732 if self._cache[path]['data']:
1733 1733 return self._cache[path]['data']
1734 1734 else:
1735 1735 # Must fallback here, too, because we only set flags.
1736 1736 return self._wrappedctx[path].data()
1737 1737 else:
1738 1738 raise error.ProgrammingError("No such file or directory: %s" %
1739 1739 path)
1740 1740 else:
1741 1741 return self._wrappedctx[path].data()
1742 1742
1743 1743 @propertycache
1744 1744 def _manifest(self):
1745 1745 parents = self.parents()
1746 1746 man = parents[0].manifest().copy()
1747 1747
1748 1748 flag = self._flagfunc
1749 1749 for path in self.added():
1750 1750 man[path] = addednodeid
1751 1751 man.setflag(path, flag(path))
1752 1752 for path in self.modified():
1753 1753 man[path] = modifiednodeid
1754 1754 man.setflag(path, flag(path))
1755 1755 for path in self.removed():
1756 1756 del man[path]
1757 1757 return man
1758 1758
1759 1759 @propertycache
1760 1760 def _flagfunc(self):
1761 1761 def f(path):
1762 1762 return self._cache[path]['flags']
1763 1763 return f
1764 1764
1765 1765 def files(self):
1766 1766 return sorted(self.added() + self.modified() + self.removed())
1767 1767
1768 1768 def modified(self):
1769 1769 return [f for f in self._cache.keys() if self._cache[f]['exists'] and
1770 1770 self._existsinparent(f)]
1771 1771
1772 1772 def added(self):
1773 1773 return [f for f in self._cache.keys() if self._cache[f]['exists'] and
1774 1774 not self._existsinparent(f)]
1775 1775
1776 1776 def removed(self):
1777 1777 return [f for f in self._cache.keys() if
1778 1778 not self._cache[f]['exists'] and self._existsinparent(f)]
1779 1779
1780 1780 def isinmemory(self):
1781 1781 return True
1782 1782
1783 1783 def filedate(self, path):
1784 1784 if self.isdirty(path):
1785 1785 return self._cache[path]['date']
1786 1786 else:
1787 1787 return self._wrappedctx[path].date()
1788 1788
1789 1789 def markcopied(self, path, origin):
1790 1790 if self.isdirty(path):
1791 1791 self._cache[path]['copied'] = origin
1792 1792 else:
1793 1793 raise error.ProgrammingError('markcopied() called on clean context')
1794 1794
1795 1795 def copydata(self, path):
1796 1796 if self.isdirty(path):
1797 1797 return self._cache[path]['copied']
1798 1798 else:
1799 1799 raise error.ProgrammingError('copydata() called on clean context')
1800 1800
1801 1801 def flags(self, path):
1802 1802 if self.isdirty(path):
1803 1803 if self._cache[path]['exists']:
1804 1804 return self._cache[path]['flags']
1805 1805 else:
1806 1806 raise error.ProgrammingError("No such file or directory: %s" %
1807 1807 self._path)
1808 1808 else:
1809 1809 return self._wrappedctx[path].flags()
1810 1810
1811 1811 def _existsinparent(self, path):
1812 1812 try:
1813 1813 # ``commitctx` raises a ``ManifestLookupError`` if a path does not
1814 1814 # exist, unlike ``workingctx``, which returns a ``workingfilectx``
1815 1815 # with an ``exists()`` function.
1816 1816 self._wrappedctx[path]
1817 1817 return True
1818 1818 except error.ManifestLookupError:
1819 1819 return False
1820 1820
1821 1821 def _auditconflicts(self, path):
1822 1822 """Replicates conflict checks done by wvfs.write().
1823 1823
1824 1824 Since we never write to the filesystem and never call `applyupdates` in
1825 1825 IMM, we'll never check that a path is actually writable -- e.g., because
1826 1826 it adds `a/foo`, but `a` is actually a file in the other commit.
1827 1827 """
1828 1828 def fail(path, component):
1829 1829 # p1() is the base and we're receiving "writes" for p2()'s
1830 1830 # files.
1831 1831 if 'l' in self.p1()[component].flags():
1832 1832 raise error.Abort("error: %s conflicts with symlink %s "
1833 1833 "in %s." % (path, component,
1834 1834 self.p1().rev()))
1835 1835 else:
1836 1836 raise error.Abort("error: '%s' conflicts with file '%s' in "
1837 1837 "%s." % (path, component,
1838 1838 self.p1().rev()))
1839 1839
1840 1840 # Test that each new directory to be created to write this path from p2
1841 1841 # is not a file in p1.
1842 1842 components = path.split('/')
1843 1843 for i in pycompat.xrange(len(components)):
1844 1844 component = "/".join(components[0:i])
1845 1845 if component in self.p1() and self._cache[component]['exists']:
1846 1846 fail(path, component)
1847 1847
1848 1848 # Test the other direction -- that this path from p2 isn't a directory
1849 1849 # in p1 (test that p1 doesn't any paths matching `path/*`).
1850 1850 match = matchmod.match('/', '', [path + '/'], default=b'relpath')
1851 1851 matches = self.p1().manifest().matches(match)
1852 1852 mfiles = matches.keys()
1853 1853 if len(mfiles) > 0:
1854 1854 if len(mfiles) == 1 and mfiles[0] == path:
1855 1855 return
1856 1856 # omit the files which are deleted in current IMM wctx
1857 1857 mfiles = [m for m in mfiles if self._cache[m]['exists']]
1858 1858 if not mfiles:
1859 1859 return
1860 1860 raise error.Abort("error: file '%s' cannot be written because "
1861 1861 " '%s/' is a folder in %s (containing %d "
1862 1862 "entries: %s)"
1863 1863 % (path, path, self.p1(), len(mfiles),
1864 1864 ', '.join(mfiles)))
1865 1865
1866 1866 def write(self, path, data, flags='', **kwargs):
1867 1867 if data is None:
1868 1868 raise error.ProgrammingError("data must be non-None")
1869 1869 self._auditconflicts(path)
1870 1870 self._markdirty(path, exists=True, data=data, date=dateutil.makedate(),
1871 1871 flags=flags)
1872 1872
1873 1873 def setflags(self, path, l, x):
1874 1874 flag = ''
1875 1875 if l:
1876 1876 flag = 'l'
1877 1877 elif x:
1878 1878 flag = 'x'
1879 1879 self._markdirty(path, exists=True, date=dateutil.makedate(),
1880 1880 flags=flag)
1881 1881
1882 1882 def remove(self, path):
1883 1883 self._markdirty(path, exists=False)
1884 1884
1885 1885 def exists(self, path):
1886 1886 """exists behaves like `lexists`, but needs to follow symlinks and
1887 1887 return False if they are broken.
1888 1888 """
1889 1889 if self.isdirty(path):
1890 1890 # If this path exists and is a symlink, "follow" it by calling
1891 1891 # exists on the destination path.
1892 1892 if (self._cache[path]['exists'] and
1893 1893 'l' in self._cache[path]['flags']):
1894 1894 return self.exists(self._cache[path]['data'].strip())
1895 1895 else:
1896 1896 return self._cache[path]['exists']
1897 1897
1898 1898 return self._existsinparent(path)
1899 1899
1900 1900 def lexists(self, path):
1901 1901 """lexists returns True if the path exists"""
1902 1902 if self.isdirty(path):
1903 1903 return self._cache[path]['exists']
1904 1904
1905 1905 return self._existsinparent(path)
1906 1906
1907 1907 def size(self, path):
1908 1908 if self.isdirty(path):
1909 1909 if self._cache[path]['exists']:
1910 1910 return len(self._cache[path]['data'])
1911 1911 else:
1912 1912 raise error.ProgrammingError("No such file or directory: %s" %
1913 1913 self._path)
1914 1914 return self._wrappedctx[path].size()
1915 1915
1916 1916 def tomemctx(self, text, branch=None, extra=None, date=None, parents=None,
1917 1917 user=None, editor=None):
1918 1918 """Converts this ``overlayworkingctx`` into a ``memctx`` ready to be
1919 1919 committed.
1920 1920
1921 1921 ``text`` is the commit message.
1922 1922 ``parents`` (optional) are rev numbers.
1923 1923 """
1924 1924 # Default parents to the wrapped contexts' if not passed.
1925 1925 if parents is None:
1926 1926 parents = self._wrappedctx.parents()
1927 1927 if len(parents) == 1:
1928 1928 parents = (parents[0], None)
1929 1929
1930 1930 # ``parents`` is passed as rev numbers; convert to ``commitctxs``.
1931 1931 if parents[1] is None:
1932 1932 parents = (self._repo[parents[0]], None)
1933 1933 else:
1934 1934 parents = (self._repo[parents[0]], self._repo[parents[1]])
1935 1935
1936 1936 files = self._cache.keys()
1937 1937 def getfile(repo, memctx, path):
1938 1938 if self._cache[path]['exists']:
1939 1939 return memfilectx(repo, memctx, path,
1940 1940 self._cache[path]['data'],
1941 1941 'l' in self._cache[path]['flags'],
1942 1942 'x' in self._cache[path]['flags'],
1943 1943 self._cache[path]['copied'])
1944 1944 else:
1945 1945 # Returning None, but including the path in `files`, is
1946 1946 # necessary for memctx to register a deletion.
1947 1947 return None
1948 1948 return memctx(self._repo, parents, text, files, getfile, date=date,
1949 1949 extra=extra, user=user, branch=branch, editor=editor)
1950 1950
1951 1951 def isdirty(self, path):
1952 1952 return path in self._cache
1953 1953
1954 1954 def isempty(self):
1955 1955 # We need to discard any keys that are actually clean before the empty
1956 1956 # commit check.
1957 1957 self._compact()
1958 1958 return len(self._cache) == 0
1959 1959
1960 1960 def clean(self):
1961 1961 self._cache = {}
1962 1962
1963 1963 def _compact(self):
1964 1964 """Removes keys from the cache that are actually clean, by comparing
1965 1965 them with the underlying context.
1966 1966
1967 1967 This can occur during the merge process, e.g. by passing --tool :local
1968 1968 to resolve a conflict.
1969 1969 """
1970 1970 keys = []
1971 1971 for path in self._cache.keys():
1972 1972 cache = self._cache[path]
1973 1973 try:
1974 1974 underlying = self._wrappedctx[path]
1975 1975 if (underlying.data() == cache['data'] and
1976 1976 underlying.flags() == cache['flags']):
1977 1977 keys.append(path)
1978 1978 except error.ManifestLookupError:
1979 1979 # Path not in the underlying manifest (created).
1980 1980 continue
1981 1981
1982 1982 for path in keys:
1983 1983 del self._cache[path]
1984 1984 return keys
1985 1985
1986 1986 def _markdirty(self, path, exists, data=None, date=None, flags=''):
1987 1987 # data not provided, let's see if we already have some; if not, let's
1988 1988 # grab it from our underlying context, so that we always have data if
1989 1989 # the file is marked as existing.
1990 1990 if exists and data is None:
1991 1991 oldentry = self._cache.get(path) or {}
1992 1992 data = oldentry.get('data') or self._wrappedctx[path].data()
1993 1993
1994 1994 self._cache[path] = {
1995 1995 'exists': exists,
1996 1996 'data': data,
1997 1997 'date': date,
1998 1998 'flags': flags,
1999 1999 'copied': None,
2000 2000 }
2001 2001
2002 2002 def filectx(self, path, filelog=None):
2003 2003 return overlayworkingfilectx(self._repo, path, parent=self,
2004 2004 filelog=filelog)
2005 2005
2006 2006 class overlayworkingfilectx(committablefilectx):
2007 2007 """Wrap a ``workingfilectx`` but intercepts all writes into an in-memory
2008 2008 cache, which can be flushed through later by calling ``flush()``."""
2009 2009
2010 2010 def __init__(self, repo, path, filelog=None, parent=None):
2011 2011 super(overlayworkingfilectx, self).__init__(repo, path, filelog,
2012 2012 parent)
2013 2013 self._repo = repo
2014 2014 self._parent = parent
2015 2015 self._path = path
2016 2016
2017 2017 def cmp(self, fctx):
2018 2018 return self.data() != fctx.data()
2019 2019
2020 2020 def changectx(self):
2021 2021 return self._parent
2022 2022
2023 2023 def data(self):
2024 2024 return self._parent.data(self._path)
2025 2025
2026 2026 def date(self):
2027 2027 return self._parent.filedate(self._path)
2028 2028
2029 2029 def exists(self):
2030 2030 return self.lexists()
2031 2031
2032 2032 def lexists(self):
2033 2033 return self._parent.exists(self._path)
2034 2034
2035 2035 def renamed(self):
2036 2036 path = self._parent.copydata(self._path)
2037 2037 if not path:
2038 2038 return None
2039 2039 return path, self._changectx._parents[0]._manifest.get(path, nullid)
2040 2040
2041 2041 def size(self):
2042 2042 return self._parent.size(self._path)
2043 2043
2044 2044 def markcopied(self, origin):
2045 2045 self._parent.markcopied(self._path, origin)
2046 2046
2047 2047 def audit(self):
2048 2048 pass
2049 2049
2050 2050 def flags(self):
2051 2051 return self._parent.flags(self._path)
2052 2052
2053 2053 def setflags(self, islink, isexec):
2054 2054 return self._parent.setflags(self._path, islink, isexec)
2055 2055
2056 2056 def write(self, data, flags, backgroundclose=False, **kwargs):
2057 2057 return self._parent.write(self._path, data, flags, **kwargs)
2058 2058
2059 2059 def remove(self, ignoremissing=False):
2060 2060 return self._parent.remove(self._path)
2061 2061
2062 2062 def clearunknown(self):
2063 2063 pass
2064 2064
2065 2065 class workingcommitctx(workingctx):
2066 2066 """A workingcommitctx object makes access to data related to
2067 2067 the revision being committed convenient.
2068 2068
2069 2069 This hides changes in the working directory, if they aren't
2070 2070 committed in this context.
2071 2071 """
2072 2072 def __init__(self, repo, changes,
2073 2073 text="", user=None, date=None, extra=None):
2074 2074 super(workingcommitctx, self).__init__(repo, text, user, date, extra,
2075 2075 changes)
2076 2076
2077 2077 def _dirstatestatus(self, match, ignored=False, clean=False, unknown=False):
2078 2078 """Return matched files only in ``self._status``
2079 2079
2080 2080 Uncommitted files appear "clean" via this context, even if
2081 2081 they aren't actually so in the working directory.
2082 2082 """
2083 2083 if clean:
2084 2084 clean = [f for f in self._manifest if f not in self._changedset]
2085 2085 else:
2086 2086 clean = []
2087 2087 return scmutil.status([f for f in self._status.modified if match(f)],
2088 2088 [f for f in self._status.added if match(f)],
2089 2089 [f for f in self._status.removed if match(f)],
2090 2090 [], [], [], clean)
2091 2091
2092 2092 @propertycache
2093 2093 def _changedset(self):
2094 2094 """Return the set of files changed in this context
2095 2095 """
2096 2096 changed = set(self._status.modified)
2097 2097 changed.update(self._status.added)
2098 2098 changed.update(self._status.removed)
2099 2099 return changed
2100 2100
2101 2101 def makecachingfilectxfn(func):
2102 2102 """Create a filectxfn that caches based on the path.
2103 2103
2104 2104 We can't use util.cachefunc because it uses all arguments as the cache
2105 2105 key and this creates a cycle since the arguments include the repo and
2106 2106 memctx.
2107 2107 """
2108 2108 cache = {}
2109 2109
2110 2110 def getfilectx(repo, memctx, path):
2111 2111 if path not in cache:
2112 2112 cache[path] = func(repo, memctx, path)
2113 2113 return cache[path]
2114 2114
2115 2115 return getfilectx
2116 2116
2117 2117 def memfilefromctx(ctx):
2118 2118 """Given a context return a memfilectx for ctx[path]
2119 2119
2120 2120 This is a convenience method for building a memctx based on another
2121 2121 context.
2122 2122 """
2123 2123 def getfilectx(repo, memctx, path):
2124 2124 fctx = ctx[path]
2125 2125 # this is weird but apparently we only keep track of one parent
2126 2126 # (why not only store that instead of a tuple?)
2127 2127 copied = fctx.renamed()
2128 2128 if copied:
2129 2129 copied = copied[0]
2130 2130 return memfilectx(repo, memctx, path, fctx.data(),
2131 2131 islink=fctx.islink(), isexec=fctx.isexec(),
2132 2132 copied=copied)
2133 2133
2134 2134 return getfilectx
2135 2135
2136 2136 def memfilefrompatch(patchstore):
2137 2137 """Given a patch (e.g. patchstore object) return a memfilectx
2138 2138
2139 2139 This is a convenience method for building a memctx based on a patchstore.
2140 2140 """
2141 2141 def getfilectx(repo, memctx, path):
2142 2142 data, mode, copied = patchstore.getfile(path)
2143 2143 if data is None:
2144 2144 return None
2145 2145 islink, isexec = mode
2146 2146 return memfilectx(repo, memctx, path, data, islink=islink,
2147 2147 isexec=isexec, copied=copied)
2148 2148
2149 2149 return getfilectx
2150 2150
2151 2151 class memctx(committablectx):
2152 2152 """Use memctx to perform in-memory commits via localrepo.commitctx().
2153 2153
2154 2154 Revision information is supplied at initialization time while
2155 2155 related files data and is made available through a callback
2156 2156 mechanism. 'repo' is the current localrepo, 'parents' is a
2157 2157 sequence of two parent revisions identifiers (pass None for every
2158 2158 missing parent), 'text' is the commit message and 'files' lists
2159 2159 names of files touched by the revision (normalized and relative to
2160 2160 repository root).
2161 2161
2162 2162 filectxfn(repo, memctx, path) is a callable receiving the
2163 2163 repository, the current memctx object and the normalized path of
2164 2164 requested file, relative to repository root. It is fired by the
2165 2165 commit function for every file in 'files', but calls order is
2166 2166 undefined. If the file is available in the revision being
2167 2167 committed (updated or added), filectxfn returns a memfilectx
2168 2168 object. If the file was removed, filectxfn return None for recent
2169 2169 Mercurial. Moved files are represented by marking the source file
2170 2170 removed and the new file added with copy information (see
2171 2171 memfilectx).
2172 2172
2173 2173 user receives the committer name and defaults to current
2174 2174 repository username, date is the commit date in any format
2175 2175 supported by dateutil.parsedate() and defaults to current date, extra
2176 2176 is a dictionary of metadata or is left empty.
2177 2177 """
2178 2178
2179 2179 # Mercurial <= 3.1 expects the filectxfn to raise IOError for missing files.
2180 2180 # Extensions that need to retain compatibility across Mercurial 3.1 can use
2181 2181 # this field to determine what to do in filectxfn.
2182 2182 _returnnoneformissingfiles = True
2183 2183
2184 2184 def __init__(self, repo, parents, text, files, filectxfn, user=None,
2185 2185 date=None, extra=None, branch=None, editor=False):
2186 2186 super(memctx, self).__init__(repo, text, user, date, extra)
2187 2187 self._rev = None
2188 2188 self._node = None
2189 2189 parents = [(p or nullid) for p in parents]
2190 2190 p1, p2 = parents
2191 2191 self._parents = [self._repo[p] for p in (p1, p2)]
2192 2192 files = sorted(set(files))
2193 2193 self._files = files
2194 2194 if branch is not None:
2195 2195 self._extra['branch'] = encoding.fromlocal(branch)
2196 2196 self.substate = {}
2197 2197
2198 2198 if isinstance(filectxfn, patch.filestore):
2199 2199 filectxfn = memfilefrompatch(filectxfn)
2200 2200 elif not callable(filectxfn):
2201 2201 # if store is not callable, wrap it in a function
2202 2202 filectxfn = memfilefromctx(filectxfn)
2203 2203
2204 2204 # memoizing increases performance for e.g. vcs convert scenarios.
2205 2205 self._filectxfn = makecachingfilectxfn(filectxfn)
2206 2206
2207 2207 if editor:
2208 2208 self._text = editor(self._repo, self, [])
2209 2209 self._repo.savecommitmessage(self._text)
2210 2210
2211 2211 def filectx(self, path, filelog=None):
2212 2212 """get a file context from the working directory
2213 2213
2214 2214 Returns None if file doesn't exist and should be removed."""
2215 2215 return self._filectxfn(self._repo, self, path)
2216 2216
2217 2217 def commit(self):
2218 2218 """commit context to the repo"""
2219 2219 return self._repo.commitctx(self)
2220 2220
2221 2221 @propertycache
2222 2222 def _manifest(self):
2223 2223 """generate a manifest based on the return values of filectxfn"""
2224 2224
2225 2225 # keep this simple for now; just worry about p1
2226 2226 pctx = self._parents[0]
2227 2227 man = pctx.manifest().copy()
2228 2228
2229 2229 for f in self._status.modified:
2230 2230 man[f] = modifiednodeid
2231 2231
2232 2232 for f in self._status.added:
2233 2233 man[f] = addednodeid
2234 2234
2235 2235 for f in self._status.removed:
2236 2236 if f in man:
2237 2237 del man[f]
2238 2238
2239 2239 return man
2240 2240
2241 2241 @propertycache
2242 2242 def _status(self):
2243 2243 """Calculate exact status from ``files`` specified at construction
2244 2244 """
2245 2245 man1 = self.p1().manifest()
2246 2246 p2 = self._parents[1]
2247 2247 # "1 < len(self._parents)" can't be used for checking
2248 2248 # existence of the 2nd parent, because "memctx._parents" is
2249 2249 # explicitly initialized by the list, of which length is 2.
2250 2250 if p2.node() != nullid:
2251 2251 man2 = p2.manifest()
2252 2252 managing = lambda f: f in man1 or f in man2
2253 2253 else:
2254 2254 managing = lambda f: f in man1
2255 2255
2256 2256 modified, added, removed = [], [], []
2257 2257 for f in self._files:
2258 2258 if not managing(f):
2259 2259 added.append(f)
2260 2260 elif self[f]:
2261 2261 modified.append(f)
2262 2262 else:
2263 2263 removed.append(f)
2264 2264
2265 2265 return scmutil.status(modified, added, removed, [], [], [], [])
2266 2266
2267 2267 class memfilectx(committablefilectx):
2268 2268 """memfilectx represents an in-memory file to commit.
2269 2269
2270 2270 See memctx and committablefilectx for more details.
2271 2271 """
2272 2272 def __init__(self, repo, changectx, path, data, islink=False,
2273 2273 isexec=False, copied=None):
2274 2274 """
2275 2275 path is the normalized file path relative to repository root.
2276 2276 data is the file content as a string.
2277 2277 islink is True if the file is a symbolic link.
2278 2278 isexec is True if the file is executable.
2279 2279 copied is the source file path if current file was copied in the
2280 2280 revision being committed, or None."""
2281 2281 super(memfilectx, self).__init__(repo, path, None, changectx)
2282 2282 self._data = data
2283 2283 if islink:
2284 2284 self._flags = 'l'
2285 2285 elif isexec:
2286 2286 self._flags = 'x'
2287 2287 else:
2288 2288 self._flags = ''
2289 2289 self._copied = None
2290 2290 if copied:
2291 2291 self._copied = (copied, nullid)
2292 2292
2293 2293 def data(self):
2294 2294 return self._data
2295 2295
2296 2296 def remove(self, ignoremissing=False):
2297 2297 """wraps unlink for a repo's working directory"""
2298 2298 # need to figure out what to do here
2299 2299 del self._changectx[self._path]
2300 2300
2301 2301 def write(self, data, flags, **kwargs):
2302 2302 """wraps repo.wwrite"""
2303 2303 self._data = data
2304 2304
2305 2305
2306 2306 class metadataonlyctx(committablectx):
2307 2307 """Like memctx but it's reusing the manifest of different commit.
2308 2308 Intended to be used by lightweight operations that are creating
2309 2309 metadata-only changes.
2310 2310
2311 2311 Revision information is supplied at initialization time. 'repo' is the
2312 2312 current localrepo, 'ctx' is original revision which manifest we're reuisng
2313 2313 'parents' is a sequence of two parent revisions identifiers (pass None for
2314 2314 every missing parent), 'text' is the commit.
2315 2315
2316 2316 user receives the committer name and defaults to current repository
2317 2317 username, date is the commit date in any format supported by
2318 2318 dateutil.parsedate() and defaults to current date, extra is a dictionary of
2319 2319 metadata or is left empty.
2320 2320 """
2321 2321 def __init__(self, repo, originalctx, parents=None, text=None, user=None,
2322 2322 date=None, extra=None, editor=False):
2323 2323 if text is None:
2324 2324 text = originalctx.description()
2325 2325 super(metadataonlyctx, self).__init__(repo, text, user, date, extra)
2326 2326 self._rev = None
2327 2327 self._node = None
2328 2328 self._originalctx = originalctx
2329 2329 self._manifestnode = originalctx.manifestnode()
2330 2330 if parents is None:
2331 2331 parents = originalctx.parents()
2332 2332 else:
2333 2333 parents = [repo[p] for p in parents if p is not None]
2334 2334 parents = parents[:]
2335 2335 while len(parents) < 2:
2336 2336 parents.append(repo[nullid])
2337 2337 p1, p2 = self._parents = parents
2338 2338
2339 2339 # sanity check to ensure that the reused manifest parents are
2340 2340 # manifests of our commit parents
2341 2341 mp1, mp2 = self.manifestctx().parents
2342 2342 if p1 != nullid and p1.manifestnode() != mp1:
2343 2343 raise RuntimeError('can\'t reuse the manifest: '
2344 2344 'its p1 doesn\'t match the new ctx p1')
2345 2345 if p2 != nullid and p2.manifestnode() != mp2:
2346 2346 raise RuntimeError('can\'t reuse the manifest: '
2347 2347 'its p2 doesn\'t match the new ctx p2')
2348 2348
2349 2349 self._files = originalctx.files()
2350 2350 self.substate = {}
2351 2351
2352 2352 if editor:
2353 2353 self._text = editor(self._repo, self, [])
2354 2354 self._repo.savecommitmessage(self._text)
2355 2355
2356 2356 def manifestnode(self):
2357 2357 return self._manifestnode
2358 2358
2359 2359 @property
2360 2360 def _manifestctx(self):
2361 2361 return self._repo.manifestlog[self._manifestnode]
2362 2362
2363 2363 def filectx(self, path, filelog=None):
2364 2364 return self._originalctx.filectx(path, filelog=filelog)
2365 2365
2366 2366 def commit(self):
2367 2367 """commit context to the repo"""
2368 2368 return self._repo.commitctx(self)
2369 2369
2370 2370 @property
2371 2371 def _manifest(self):
2372 2372 return self._originalctx.manifest()
2373 2373
2374 2374 @propertycache
2375 2375 def _status(self):
2376 2376 """Calculate exact status from ``files`` specified in the ``origctx``
2377 2377 and parents manifests.
2378 2378 """
2379 2379 man1 = self.p1().manifest()
2380 2380 p2 = self._parents[1]
2381 2381 # "1 < len(self._parents)" can't be used for checking
2382 2382 # existence of the 2nd parent, because "metadataonlyctx._parents" is
2383 2383 # explicitly initialized by the list, of which length is 2.
2384 2384 if p2.node() != nullid:
2385 2385 man2 = p2.manifest()
2386 2386 managing = lambda f: f in man1 or f in man2
2387 2387 else:
2388 2388 managing = lambda f: f in man1
2389 2389
2390 2390 modified, added, removed = [], [], []
2391 2391 for f in self._files:
2392 2392 if not managing(f):
2393 2393 added.append(f)
2394 2394 elif f in self:
2395 2395 modified.append(f)
2396 2396 else:
2397 2397 removed.append(f)
2398 2398
2399 2399 return scmutil.status(modified, added, removed, [], [], [], [])
2400 2400
2401 2401 class arbitraryfilectx(object):
2402 2402 """Allows you to use filectx-like functions on a file in an arbitrary
2403 2403 location on disk, possibly not in the working directory.
2404 2404 """
2405 2405 def __init__(self, path, repo=None):
2406 2406 # Repo is optional because contrib/simplemerge uses this class.
2407 2407 self._repo = repo
2408 2408 self._path = path
2409 2409
2410 2410 def cmp(self, fctx):
2411 2411 # filecmp follows symlinks whereas `cmp` should not, so skip the fast
2412 2412 # path if either side is a symlink.
2413 2413 symlinks = ('l' in self.flags() or 'l' in fctx.flags())
2414 2414 if not symlinks and isinstance(fctx, workingfilectx) and self._repo:
2415 2415 # Add a fast-path for merge if both sides are disk-backed.
2416 2416 # Note that filecmp uses the opposite return values (True if same)
2417 2417 # from our cmp functions (True if different).
2418 2418 return not filecmp.cmp(self.path(), self._repo.wjoin(fctx.path()))
2419 2419 return self.data() != fctx.data()
2420 2420
2421 2421 def path(self):
2422 2422 return self._path
2423 2423
2424 2424 def flags(self):
2425 2425 return ''
2426 2426
2427 2427 def data(self):
2428 2428 return util.readfile(self._path)
2429 2429
2430 2430 def decodeddata(self):
2431 2431 with open(self._path, "rb") as f:
2432 2432 return f.read()
2433 2433
2434 2434 def remove(self):
2435 2435 util.unlink(self._path)
2436 2436
2437 2437 def write(self, data, flags, **kwargs):
2438 2438 assert not flags
2439 2439 with open(self._path, "w") as f:
2440 2440 f.write(data)
General Comments 0
You need to be logged in to leave comments. Login now