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