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