##// END OF EJS Templates
localrepo: fastpath access to "."...
marmoute -
r44564:d86dede1 default
parent child Browse files
Show More
@@ -1,3769 +1,3772 b''
1 1 # localrepo.py - read/write repository class for mercurial
2 2 #
3 3 # Copyright 2005-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 random
13 13 import sys
14 14 import time
15 15 import weakref
16 16
17 17 from .i18n import _
18 18 from .node import (
19 19 bin,
20 20 hex,
21 21 nullid,
22 22 nullrev,
23 23 short,
24 24 )
25 25 from .pycompat import (
26 26 delattr,
27 27 getattr,
28 28 )
29 29 from . import (
30 30 bookmarks,
31 31 branchmap,
32 32 bundle2,
33 33 changegroup,
34 34 color,
35 35 context,
36 36 dirstate,
37 37 dirstateguard,
38 38 discovery,
39 39 encoding,
40 40 error,
41 41 exchange,
42 42 extensions,
43 43 filelog,
44 44 hook,
45 45 lock as lockmod,
46 46 match as matchmod,
47 47 merge as mergemod,
48 48 mergeutil,
49 49 namespaces,
50 50 narrowspec,
51 51 obsolete,
52 52 pathutil,
53 53 phases,
54 54 pushkey,
55 55 pycompat,
56 56 repoview,
57 57 revset,
58 58 revsetlang,
59 59 scmutil,
60 60 sparse,
61 61 store as storemod,
62 62 subrepoutil,
63 63 tags as tagsmod,
64 64 transaction,
65 65 txnutil,
66 66 util,
67 67 vfs as vfsmod,
68 68 )
69 69
70 70 from .interfaces import (
71 71 repository,
72 72 util as interfaceutil,
73 73 )
74 74
75 75 from .utils import (
76 76 hashutil,
77 77 procutil,
78 78 stringutil,
79 79 )
80 80
81 81 from .revlogutils import constants as revlogconst
82 82
83 83 release = lockmod.release
84 84 urlerr = util.urlerr
85 85 urlreq = util.urlreq
86 86
87 87 # set of (path, vfs-location) tuples. vfs-location is:
88 88 # - 'plain for vfs relative paths
89 89 # - '' for svfs relative paths
90 90 _cachedfiles = set()
91 91
92 92
93 93 class _basefilecache(scmutil.filecache):
94 94 """All filecache usage on repo are done for logic that should be unfiltered
95 95 """
96 96
97 97 def __get__(self, repo, type=None):
98 98 if repo is None:
99 99 return self
100 100 # proxy to unfiltered __dict__ since filtered repo has no entry
101 101 unfi = repo.unfiltered()
102 102 try:
103 103 return unfi.__dict__[self.sname]
104 104 except KeyError:
105 105 pass
106 106 return super(_basefilecache, self).__get__(unfi, type)
107 107
108 108 def set(self, repo, value):
109 109 return super(_basefilecache, self).set(repo.unfiltered(), value)
110 110
111 111
112 112 class repofilecache(_basefilecache):
113 113 """filecache for files in .hg but outside of .hg/store"""
114 114
115 115 def __init__(self, *paths):
116 116 super(repofilecache, self).__init__(*paths)
117 117 for path in paths:
118 118 _cachedfiles.add((path, b'plain'))
119 119
120 120 def join(self, obj, fname):
121 121 return obj.vfs.join(fname)
122 122
123 123
124 124 class storecache(_basefilecache):
125 125 """filecache for files in the store"""
126 126
127 127 def __init__(self, *paths):
128 128 super(storecache, self).__init__(*paths)
129 129 for path in paths:
130 130 _cachedfiles.add((path, b''))
131 131
132 132 def join(self, obj, fname):
133 133 return obj.sjoin(fname)
134 134
135 135
136 136 class mixedrepostorecache(_basefilecache):
137 137 """filecache for a mix files in .hg/store and outside"""
138 138
139 139 def __init__(self, *pathsandlocations):
140 140 # scmutil.filecache only uses the path for passing back into our
141 141 # join(), so we can safely pass a list of paths and locations
142 142 super(mixedrepostorecache, self).__init__(*pathsandlocations)
143 143 _cachedfiles.update(pathsandlocations)
144 144
145 145 def join(self, obj, fnameandlocation):
146 146 fname, location = fnameandlocation
147 147 if location == b'plain':
148 148 return obj.vfs.join(fname)
149 149 else:
150 150 if location != b'':
151 151 raise error.ProgrammingError(
152 152 b'unexpected location: %s' % location
153 153 )
154 154 return obj.sjoin(fname)
155 155
156 156
157 157 def isfilecached(repo, name):
158 158 """check if a repo has already cached "name" filecache-ed property
159 159
160 160 This returns (cachedobj-or-None, iscached) tuple.
161 161 """
162 162 cacheentry = repo.unfiltered()._filecache.get(name, None)
163 163 if not cacheentry:
164 164 return None, False
165 165 return cacheentry.obj, True
166 166
167 167
168 168 class unfilteredpropertycache(util.propertycache):
169 169 """propertycache that apply to unfiltered repo only"""
170 170
171 171 def __get__(self, repo, type=None):
172 172 unfi = repo.unfiltered()
173 173 if unfi is repo:
174 174 return super(unfilteredpropertycache, self).__get__(unfi)
175 175 return getattr(unfi, self.name)
176 176
177 177
178 178 class filteredpropertycache(util.propertycache):
179 179 """propertycache that must take filtering in account"""
180 180
181 181 def cachevalue(self, obj, value):
182 182 object.__setattr__(obj, self.name, value)
183 183
184 184
185 185 def hasunfilteredcache(repo, name):
186 186 """check if a repo has an unfilteredpropertycache value for <name>"""
187 187 return name in vars(repo.unfiltered())
188 188
189 189
190 190 def unfilteredmethod(orig):
191 191 """decorate method that always need to be run on unfiltered version"""
192 192
193 193 def wrapper(repo, *args, **kwargs):
194 194 return orig(repo.unfiltered(), *args, **kwargs)
195 195
196 196 return wrapper
197 197
198 198
199 199 moderncaps = {
200 200 b'lookup',
201 201 b'branchmap',
202 202 b'pushkey',
203 203 b'known',
204 204 b'getbundle',
205 205 b'unbundle',
206 206 }
207 207 legacycaps = moderncaps.union({b'changegroupsubset'})
208 208
209 209
210 210 @interfaceutil.implementer(repository.ipeercommandexecutor)
211 211 class localcommandexecutor(object):
212 212 def __init__(self, peer):
213 213 self._peer = peer
214 214 self._sent = False
215 215 self._closed = False
216 216
217 217 def __enter__(self):
218 218 return self
219 219
220 220 def __exit__(self, exctype, excvalue, exctb):
221 221 self.close()
222 222
223 223 def callcommand(self, command, args):
224 224 if self._sent:
225 225 raise error.ProgrammingError(
226 226 b'callcommand() cannot be used after sendcommands()'
227 227 )
228 228
229 229 if self._closed:
230 230 raise error.ProgrammingError(
231 231 b'callcommand() cannot be used after close()'
232 232 )
233 233
234 234 # We don't need to support anything fancy. Just call the named
235 235 # method on the peer and return a resolved future.
236 236 fn = getattr(self._peer, pycompat.sysstr(command))
237 237
238 238 f = pycompat.futures.Future()
239 239
240 240 try:
241 241 result = fn(**pycompat.strkwargs(args))
242 242 except Exception:
243 243 pycompat.future_set_exception_info(f, sys.exc_info()[1:])
244 244 else:
245 245 f.set_result(result)
246 246
247 247 return f
248 248
249 249 def sendcommands(self):
250 250 self._sent = True
251 251
252 252 def close(self):
253 253 self._closed = True
254 254
255 255
256 256 @interfaceutil.implementer(repository.ipeercommands)
257 257 class localpeer(repository.peer):
258 258 '''peer for a local repo; reflects only the most recent API'''
259 259
260 260 def __init__(self, repo, caps=None):
261 261 super(localpeer, self).__init__()
262 262
263 263 if caps is None:
264 264 caps = moderncaps.copy()
265 265 self._repo = repo.filtered(b'served')
266 266 self.ui = repo.ui
267 267 self._caps = repo._restrictcapabilities(caps)
268 268
269 269 # Begin of _basepeer interface.
270 270
271 271 def url(self):
272 272 return self._repo.url()
273 273
274 274 def local(self):
275 275 return self._repo
276 276
277 277 def peer(self):
278 278 return self
279 279
280 280 def canpush(self):
281 281 return True
282 282
283 283 def close(self):
284 284 self._repo.close()
285 285
286 286 # End of _basepeer interface.
287 287
288 288 # Begin of _basewirecommands interface.
289 289
290 290 def branchmap(self):
291 291 return self._repo.branchmap()
292 292
293 293 def capabilities(self):
294 294 return self._caps
295 295
296 296 def clonebundles(self):
297 297 return self._repo.tryread(b'clonebundles.manifest')
298 298
299 299 def debugwireargs(self, one, two, three=None, four=None, five=None):
300 300 """Used to test argument passing over the wire"""
301 301 return b"%s %s %s %s %s" % (
302 302 one,
303 303 two,
304 304 pycompat.bytestr(three),
305 305 pycompat.bytestr(four),
306 306 pycompat.bytestr(five),
307 307 )
308 308
309 309 def getbundle(
310 310 self, source, heads=None, common=None, bundlecaps=None, **kwargs
311 311 ):
312 312 chunks = exchange.getbundlechunks(
313 313 self._repo,
314 314 source,
315 315 heads=heads,
316 316 common=common,
317 317 bundlecaps=bundlecaps,
318 318 **kwargs
319 319 )[1]
320 320 cb = util.chunkbuffer(chunks)
321 321
322 322 if exchange.bundle2requested(bundlecaps):
323 323 # When requesting a bundle2, getbundle returns a stream to make the
324 324 # wire level function happier. We need to build a proper object
325 325 # from it in local peer.
326 326 return bundle2.getunbundler(self.ui, cb)
327 327 else:
328 328 return changegroup.getunbundler(b'01', cb, None)
329 329
330 330 def heads(self):
331 331 return self._repo.heads()
332 332
333 333 def known(self, nodes):
334 334 return self._repo.known(nodes)
335 335
336 336 def listkeys(self, namespace):
337 337 return self._repo.listkeys(namespace)
338 338
339 339 def lookup(self, key):
340 340 return self._repo.lookup(key)
341 341
342 342 def pushkey(self, namespace, key, old, new):
343 343 return self._repo.pushkey(namespace, key, old, new)
344 344
345 345 def stream_out(self):
346 346 raise error.Abort(_(b'cannot perform stream clone against local peer'))
347 347
348 348 def unbundle(self, bundle, heads, url):
349 349 """apply a bundle on a repo
350 350
351 351 This function handles the repo locking itself."""
352 352 try:
353 353 try:
354 354 bundle = exchange.readbundle(self.ui, bundle, None)
355 355 ret = exchange.unbundle(self._repo, bundle, heads, b'push', url)
356 356 if util.safehasattr(ret, b'getchunks'):
357 357 # This is a bundle20 object, turn it into an unbundler.
358 358 # This little dance should be dropped eventually when the
359 359 # API is finally improved.
360 360 stream = util.chunkbuffer(ret.getchunks())
361 361 ret = bundle2.getunbundler(self.ui, stream)
362 362 return ret
363 363 except Exception as exc:
364 364 # If the exception contains output salvaged from a bundle2
365 365 # reply, we need to make sure it is printed before continuing
366 366 # to fail. So we build a bundle2 with such output and consume
367 367 # it directly.
368 368 #
369 369 # This is not very elegant but allows a "simple" solution for
370 370 # issue4594
371 371 output = getattr(exc, '_bundle2salvagedoutput', ())
372 372 if output:
373 373 bundler = bundle2.bundle20(self._repo.ui)
374 374 for out in output:
375 375 bundler.addpart(out)
376 376 stream = util.chunkbuffer(bundler.getchunks())
377 377 b = bundle2.getunbundler(self.ui, stream)
378 378 bundle2.processbundle(self._repo, b)
379 379 raise
380 380 except error.PushRaced as exc:
381 381 raise error.ResponseError(
382 382 _(b'push failed:'), stringutil.forcebytestr(exc)
383 383 )
384 384
385 385 # End of _basewirecommands interface.
386 386
387 387 # Begin of peer interface.
388 388
389 389 def commandexecutor(self):
390 390 return localcommandexecutor(self)
391 391
392 392 # End of peer interface.
393 393
394 394
395 395 @interfaceutil.implementer(repository.ipeerlegacycommands)
396 396 class locallegacypeer(localpeer):
397 397 '''peer extension which implements legacy methods too; used for tests with
398 398 restricted capabilities'''
399 399
400 400 def __init__(self, repo):
401 401 super(locallegacypeer, self).__init__(repo, caps=legacycaps)
402 402
403 403 # Begin of baselegacywirecommands interface.
404 404
405 405 def between(self, pairs):
406 406 return self._repo.between(pairs)
407 407
408 408 def branches(self, nodes):
409 409 return self._repo.branches(nodes)
410 410
411 411 def changegroup(self, nodes, source):
412 412 outgoing = discovery.outgoing(
413 413 self._repo, missingroots=nodes, missingheads=self._repo.heads()
414 414 )
415 415 return changegroup.makechangegroup(self._repo, outgoing, b'01', source)
416 416
417 417 def changegroupsubset(self, bases, heads, source):
418 418 outgoing = discovery.outgoing(
419 419 self._repo, missingroots=bases, missingheads=heads
420 420 )
421 421 return changegroup.makechangegroup(self._repo, outgoing, b'01', source)
422 422
423 423 # End of baselegacywirecommands interface.
424 424
425 425
426 426 # Increment the sub-version when the revlog v2 format changes to lock out old
427 427 # clients.
428 428 REVLOGV2_REQUIREMENT = b'exp-revlogv2.1'
429 429
430 430 # A repository with the sparserevlog feature will have delta chains that
431 431 # can spread over a larger span. Sparse reading cuts these large spans into
432 432 # pieces, so that each piece isn't too big.
433 433 # Without the sparserevlog capability, reading from the repository could use
434 434 # huge amounts of memory, because the whole span would be read at once,
435 435 # including all the intermediate revisions that aren't pertinent for the chain.
436 436 # This is why once a repository has enabled sparse-read, it becomes required.
437 437 SPARSEREVLOG_REQUIREMENT = b'sparserevlog'
438 438
439 439 # A repository with the sidedataflag requirement will allow to store extra
440 440 # information for revision without altering their original hashes.
441 441 SIDEDATA_REQUIREMENT = b'exp-sidedata-flag'
442 442
443 443 # A repository with the the copies-sidedata-changeset requirement will store
444 444 # copies related information in changeset's sidedata.
445 445 COPIESSDC_REQUIREMENT = b'exp-copies-sidedata-changeset'
446 446
447 447 # Functions receiving (ui, features) that extensions can register to impact
448 448 # the ability to load repositories with custom requirements. Only
449 449 # functions defined in loaded extensions are called.
450 450 #
451 451 # The function receives a set of requirement strings that the repository
452 452 # is capable of opening. Functions will typically add elements to the
453 453 # set to reflect that the extension knows how to handle that requirements.
454 454 featuresetupfuncs = set()
455 455
456 456
457 457 def makelocalrepository(baseui, path, intents=None):
458 458 """Create a local repository object.
459 459
460 460 Given arguments needed to construct a local repository, this function
461 461 performs various early repository loading functionality (such as
462 462 reading the ``.hg/requires`` and ``.hg/hgrc`` files), validates that
463 463 the repository can be opened, derives a type suitable for representing
464 464 that repository, and returns an instance of it.
465 465
466 466 The returned object conforms to the ``repository.completelocalrepository``
467 467 interface.
468 468
469 469 The repository type is derived by calling a series of factory functions
470 470 for each aspect/interface of the final repository. These are defined by
471 471 ``REPO_INTERFACES``.
472 472
473 473 Each factory function is called to produce a type implementing a specific
474 474 interface. The cumulative list of returned types will be combined into a
475 475 new type and that type will be instantiated to represent the local
476 476 repository.
477 477
478 478 The factory functions each receive various state that may be consulted
479 479 as part of deriving a type.
480 480
481 481 Extensions should wrap these factory functions to customize repository type
482 482 creation. Note that an extension's wrapped function may be called even if
483 483 that extension is not loaded for the repo being constructed. Extensions
484 484 should check if their ``__name__`` appears in the
485 485 ``extensionmodulenames`` set passed to the factory function and no-op if
486 486 not.
487 487 """
488 488 ui = baseui.copy()
489 489 # Prevent copying repo configuration.
490 490 ui.copy = baseui.copy
491 491
492 492 # Working directory VFS rooted at repository root.
493 493 wdirvfs = vfsmod.vfs(path, expandpath=True, realpath=True)
494 494
495 495 # Main VFS for .hg/ directory.
496 496 hgpath = wdirvfs.join(b'.hg')
497 497 hgvfs = vfsmod.vfs(hgpath, cacheaudited=True)
498 498
499 499 # The .hg/ path should exist and should be a directory. All other
500 500 # cases are errors.
501 501 if not hgvfs.isdir():
502 502 try:
503 503 hgvfs.stat()
504 504 except OSError as e:
505 505 if e.errno != errno.ENOENT:
506 506 raise
507 507
508 508 raise error.RepoError(_(b'repository %s not found') % path)
509 509
510 510 # .hg/requires file contains a newline-delimited list of
511 511 # features/capabilities the opener (us) must have in order to use
512 512 # the repository. This file was introduced in Mercurial 0.9.2,
513 513 # which means very old repositories may not have one. We assume
514 514 # a missing file translates to no requirements.
515 515 try:
516 516 requirements = set(hgvfs.read(b'requires').splitlines())
517 517 except IOError as e:
518 518 if e.errno != errno.ENOENT:
519 519 raise
520 520 requirements = set()
521 521
522 522 # The .hg/hgrc file may load extensions or contain config options
523 523 # that influence repository construction. Attempt to load it and
524 524 # process any new extensions that it may have pulled in.
525 525 if loadhgrc(ui, wdirvfs, hgvfs, requirements):
526 526 afterhgrcload(ui, wdirvfs, hgvfs, requirements)
527 527 extensions.loadall(ui)
528 528 extensions.populateui(ui)
529 529
530 530 # Set of module names of extensions loaded for this repository.
531 531 extensionmodulenames = {m.__name__ for n, m in extensions.extensions(ui)}
532 532
533 533 supportedrequirements = gathersupportedrequirements(ui)
534 534
535 535 # We first validate the requirements are known.
536 536 ensurerequirementsrecognized(requirements, supportedrequirements)
537 537
538 538 # Then we validate that the known set is reasonable to use together.
539 539 ensurerequirementscompatible(ui, requirements)
540 540
541 541 # TODO there are unhandled edge cases related to opening repositories with
542 542 # shared storage. If storage is shared, we should also test for requirements
543 543 # compatibility in the pointed-to repo. This entails loading the .hg/hgrc in
544 544 # that repo, as that repo may load extensions needed to open it. This is a
545 545 # bit complicated because we don't want the other hgrc to overwrite settings
546 546 # in this hgrc.
547 547 #
548 548 # This bug is somewhat mitigated by the fact that we copy the .hg/requires
549 549 # file when sharing repos. But if a requirement is added after the share is
550 550 # performed, thereby introducing a new requirement for the opener, we may
551 551 # will not see that and could encounter a run-time error interacting with
552 552 # that shared store since it has an unknown-to-us requirement.
553 553
554 554 # At this point, we know we should be capable of opening the repository.
555 555 # Now get on with doing that.
556 556
557 557 features = set()
558 558
559 559 # The "store" part of the repository holds versioned data. How it is
560 560 # accessed is determined by various requirements. The ``shared`` or
561 561 # ``relshared`` requirements indicate the store lives in the path contained
562 562 # in the ``.hg/sharedpath`` file. This is an absolute path for
563 563 # ``shared`` and relative to ``.hg/`` for ``relshared``.
564 564 if b'shared' in requirements or b'relshared' in requirements:
565 565 sharedpath = hgvfs.read(b'sharedpath').rstrip(b'\n')
566 566 if b'relshared' in requirements:
567 567 sharedpath = hgvfs.join(sharedpath)
568 568
569 569 sharedvfs = vfsmod.vfs(sharedpath, realpath=True)
570 570
571 571 if not sharedvfs.exists():
572 572 raise error.RepoError(
573 573 _(b'.hg/sharedpath points to nonexistent directory %s')
574 574 % sharedvfs.base
575 575 )
576 576
577 577 features.add(repository.REPO_FEATURE_SHARED_STORAGE)
578 578
579 579 storebasepath = sharedvfs.base
580 580 cachepath = sharedvfs.join(b'cache')
581 581 else:
582 582 storebasepath = hgvfs.base
583 583 cachepath = hgvfs.join(b'cache')
584 584 wcachepath = hgvfs.join(b'wcache')
585 585
586 586 # The store has changed over time and the exact layout is dictated by
587 587 # requirements. The store interface abstracts differences across all
588 588 # of them.
589 589 store = makestore(
590 590 requirements,
591 591 storebasepath,
592 592 lambda base: vfsmod.vfs(base, cacheaudited=True),
593 593 )
594 594 hgvfs.createmode = store.createmode
595 595
596 596 storevfs = store.vfs
597 597 storevfs.options = resolvestorevfsoptions(ui, requirements, features)
598 598
599 599 # The cache vfs is used to manage cache files.
600 600 cachevfs = vfsmod.vfs(cachepath, cacheaudited=True)
601 601 cachevfs.createmode = store.createmode
602 602 # The cache vfs is used to manage cache files related to the working copy
603 603 wcachevfs = vfsmod.vfs(wcachepath, cacheaudited=True)
604 604 wcachevfs.createmode = store.createmode
605 605
606 606 # Now resolve the type for the repository object. We do this by repeatedly
607 607 # calling a factory function to produces types for specific aspects of the
608 608 # repo's operation. The aggregate returned types are used as base classes
609 609 # for a dynamically-derived type, which will represent our new repository.
610 610
611 611 bases = []
612 612 extrastate = {}
613 613
614 614 for iface, fn in REPO_INTERFACES:
615 615 # We pass all potentially useful state to give extensions tons of
616 616 # flexibility.
617 617 typ = fn()(
618 618 ui=ui,
619 619 intents=intents,
620 620 requirements=requirements,
621 621 features=features,
622 622 wdirvfs=wdirvfs,
623 623 hgvfs=hgvfs,
624 624 store=store,
625 625 storevfs=storevfs,
626 626 storeoptions=storevfs.options,
627 627 cachevfs=cachevfs,
628 628 wcachevfs=wcachevfs,
629 629 extensionmodulenames=extensionmodulenames,
630 630 extrastate=extrastate,
631 631 baseclasses=bases,
632 632 )
633 633
634 634 if not isinstance(typ, type):
635 635 raise error.ProgrammingError(
636 636 b'unable to construct type for %s' % iface
637 637 )
638 638
639 639 bases.append(typ)
640 640
641 641 # type() allows you to use characters in type names that wouldn't be
642 642 # recognized as Python symbols in source code. We abuse that to add
643 643 # rich information about our constructed repo.
644 644 name = pycompat.sysstr(
645 645 b'derivedrepo:%s<%s>' % (wdirvfs.base, b','.join(sorted(requirements)))
646 646 )
647 647
648 648 cls = type(name, tuple(bases), {})
649 649
650 650 return cls(
651 651 baseui=baseui,
652 652 ui=ui,
653 653 origroot=path,
654 654 wdirvfs=wdirvfs,
655 655 hgvfs=hgvfs,
656 656 requirements=requirements,
657 657 supportedrequirements=supportedrequirements,
658 658 sharedpath=storebasepath,
659 659 store=store,
660 660 cachevfs=cachevfs,
661 661 wcachevfs=wcachevfs,
662 662 features=features,
663 663 intents=intents,
664 664 )
665 665
666 666
667 667 def loadhgrc(ui, wdirvfs, hgvfs, requirements):
668 668 """Load hgrc files/content into a ui instance.
669 669
670 670 This is called during repository opening to load any additional
671 671 config files or settings relevant to the current repository.
672 672
673 673 Returns a bool indicating whether any additional configs were loaded.
674 674
675 675 Extensions should monkeypatch this function to modify how per-repo
676 676 configs are loaded. For example, an extension may wish to pull in
677 677 configs from alternate files or sources.
678 678 """
679 679 try:
680 680 ui.readconfig(hgvfs.join(b'hgrc'), root=wdirvfs.base)
681 681 return True
682 682 except IOError:
683 683 return False
684 684
685 685
686 686 def afterhgrcload(ui, wdirvfs, hgvfs, requirements):
687 687 """Perform additional actions after .hg/hgrc is loaded.
688 688
689 689 This function is called during repository loading immediately after
690 690 the .hg/hgrc file is loaded and before per-repo extensions are loaded.
691 691
692 692 The function can be used to validate configs, automatically add
693 693 options (including extensions) based on requirements, etc.
694 694 """
695 695
696 696 # Map of requirements to list of extensions to load automatically when
697 697 # requirement is present.
698 698 autoextensions = {
699 699 b'largefiles': [b'largefiles'],
700 700 b'lfs': [b'lfs'],
701 701 }
702 702
703 703 for requirement, names in sorted(autoextensions.items()):
704 704 if requirement not in requirements:
705 705 continue
706 706
707 707 for name in names:
708 708 if not ui.hasconfig(b'extensions', name):
709 709 ui.setconfig(b'extensions', name, b'', source=b'autoload')
710 710
711 711
712 712 def gathersupportedrequirements(ui):
713 713 """Determine the complete set of recognized requirements."""
714 714 # Start with all requirements supported by this file.
715 715 supported = set(localrepository._basesupported)
716 716
717 717 # Execute ``featuresetupfuncs`` entries if they belong to an extension
718 718 # relevant to this ui instance.
719 719 modules = {m.__name__ for n, m in extensions.extensions(ui)}
720 720
721 721 for fn in featuresetupfuncs:
722 722 if fn.__module__ in modules:
723 723 fn(ui, supported)
724 724
725 725 # Add derived requirements from registered compression engines.
726 726 for name in util.compengines:
727 727 engine = util.compengines[name]
728 728 if engine.available() and engine.revlogheader():
729 729 supported.add(b'exp-compression-%s' % name)
730 730 if engine.name() == b'zstd':
731 731 supported.add(b'revlog-compression-zstd')
732 732
733 733 return supported
734 734
735 735
736 736 def ensurerequirementsrecognized(requirements, supported):
737 737 """Validate that a set of local requirements is recognized.
738 738
739 739 Receives a set of requirements. Raises an ``error.RepoError`` if there
740 740 exists any requirement in that set that currently loaded code doesn't
741 741 recognize.
742 742
743 743 Returns a set of supported requirements.
744 744 """
745 745 missing = set()
746 746
747 747 for requirement in requirements:
748 748 if requirement in supported:
749 749 continue
750 750
751 751 if not requirement or not requirement[0:1].isalnum():
752 752 raise error.RequirementError(_(b'.hg/requires file is corrupt'))
753 753
754 754 missing.add(requirement)
755 755
756 756 if missing:
757 757 raise error.RequirementError(
758 758 _(b'repository requires features unknown to this Mercurial: %s')
759 759 % b' '.join(sorted(missing)),
760 760 hint=_(
761 761 b'see https://mercurial-scm.org/wiki/MissingRequirement '
762 762 b'for more information'
763 763 ),
764 764 )
765 765
766 766
767 767 def ensurerequirementscompatible(ui, requirements):
768 768 """Validates that a set of recognized requirements is mutually compatible.
769 769
770 770 Some requirements may not be compatible with others or require
771 771 config options that aren't enabled. This function is called during
772 772 repository opening to ensure that the set of requirements needed
773 773 to open a repository is sane and compatible with config options.
774 774
775 775 Extensions can monkeypatch this function to perform additional
776 776 checking.
777 777
778 778 ``error.RepoError`` should be raised on failure.
779 779 """
780 780 if b'exp-sparse' in requirements and not sparse.enabled:
781 781 raise error.RepoError(
782 782 _(
783 783 b'repository is using sparse feature but '
784 784 b'sparse is not enabled; enable the '
785 785 b'"sparse" extensions to access'
786 786 )
787 787 )
788 788
789 789
790 790 def makestore(requirements, path, vfstype):
791 791 """Construct a storage object for a repository."""
792 792 if b'store' in requirements:
793 793 if b'fncache' in requirements:
794 794 return storemod.fncachestore(
795 795 path, vfstype, b'dotencode' in requirements
796 796 )
797 797
798 798 return storemod.encodedstore(path, vfstype)
799 799
800 800 return storemod.basicstore(path, vfstype)
801 801
802 802
803 803 def resolvestorevfsoptions(ui, requirements, features):
804 804 """Resolve the options to pass to the store vfs opener.
805 805
806 806 The returned dict is used to influence behavior of the storage layer.
807 807 """
808 808 options = {}
809 809
810 810 if b'treemanifest' in requirements:
811 811 options[b'treemanifest'] = True
812 812
813 813 # experimental config: format.manifestcachesize
814 814 manifestcachesize = ui.configint(b'format', b'manifestcachesize')
815 815 if manifestcachesize is not None:
816 816 options[b'manifestcachesize'] = manifestcachesize
817 817
818 818 # In the absence of another requirement superseding a revlog-related
819 819 # requirement, we have to assume the repo is using revlog version 0.
820 820 # This revlog format is super old and we don't bother trying to parse
821 821 # opener options for it because those options wouldn't do anything
822 822 # meaningful on such old repos.
823 823 if b'revlogv1' in requirements or REVLOGV2_REQUIREMENT in requirements:
824 824 options.update(resolverevlogstorevfsoptions(ui, requirements, features))
825 825 else: # explicitly mark repo as using revlogv0
826 826 options[b'revlogv0'] = True
827 827
828 828 if COPIESSDC_REQUIREMENT in requirements:
829 829 options[b'copies-storage'] = b'changeset-sidedata'
830 830 else:
831 831 writecopiesto = ui.config(b'experimental', b'copies.write-to')
832 832 copiesextramode = (b'changeset-only', b'compatibility')
833 833 if writecopiesto in copiesextramode:
834 834 options[b'copies-storage'] = b'extra'
835 835
836 836 return options
837 837
838 838
839 839 def resolverevlogstorevfsoptions(ui, requirements, features):
840 840 """Resolve opener options specific to revlogs."""
841 841
842 842 options = {}
843 843 options[b'flagprocessors'] = {}
844 844
845 845 if b'revlogv1' in requirements:
846 846 options[b'revlogv1'] = True
847 847 if REVLOGV2_REQUIREMENT in requirements:
848 848 options[b'revlogv2'] = True
849 849
850 850 if b'generaldelta' in requirements:
851 851 options[b'generaldelta'] = True
852 852
853 853 # experimental config: format.chunkcachesize
854 854 chunkcachesize = ui.configint(b'format', b'chunkcachesize')
855 855 if chunkcachesize is not None:
856 856 options[b'chunkcachesize'] = chunkcachesize
857 857
858 858 deltabothparents = ui.configbool(
859 859 b'storage', b'revlog.optimize-delta-parent-choice'
860 860 )
861 861 options[b'deltabothparents'] = deltabothparents
862 862
863 863 lazydelta = ui.configbool(b'storage', b'revlog.reuse-external-delta')
864 864 lazydeltabase = False
865 865 if lazydelta:
866 866 lazydeltabase = ui.configbool(
867 867 b'storage', b'revlog.reuse-external-delta-parent'
868 868 )
869 869 if lazydeltabase is None:
870 870 lazydeltabase = not scmutil.gddeltaconfig(ui)
871 871 options[b'lazydelta'] = lazydelta
872 872 options[b'lazydeltabase'] = lazydeltabase
873 873
874 874 chainspan = ui.configbytes(b'experimental', b'maxdeltachainspan')
875 875 if 0 <= chainspan:
876 876 options[b'maxdeltachainspan'] = chainspan
877 877
878 878 mmapindexthreshold = ui.configbytes(b'experimental', b'mmapindexthreshold')
879 879 if mmapindexthreshold is not None:
880 880 options[b'mmapindexthreshold'] = mmapindexthreshold
881 881
882 882 withsparseread = ui.configbool(b'experimental', b'sparse-read')
883 883 srdensitythres = float(
884 884 ui.config(b'experimental', b'sparse-read.density-threshold')
885 885 )
886 886 srmingapsize = ui.configbytes(b'experimental', b'sparse-read.min-gap-size')
887 887 options[b'with-sparse-read'] = withsparseread
888 888 options[b'sparse-read-density-threshold'] = srdensitythres
889 889 options[b'sparse-read-min-gap-size'] = srmingapsize
890 890
891 891 sparserevlog = SPARSEREVLOG_REQUIREMENT in requirements
892 892 options[b'sparse-revlog'] = sparserevlog
893 893 if sparserevlog:
894 894 options[b'generaldelta'] = True
895 895
896 896 sidedata = SIDEDATA_REQUIREMENT in requirements
897 897 options[b'side-data'] = sidedata
898 898
899 899 maxchainlen = None
900 900 if sparserevlog:
901 901 maxchainlen = revlogconst.SPARSE_REVLOG_MAX_CHAIN_LENGTH
902 902 # experimental config: format.maxchainlen
903 903 maxchainlen = ui.configint(b'format', b'maxchainlen', maxchainlen)
904 904 if maxchainlen is not None:
905 905 options[b'maxchainlen'] = maxchainlen
906 906
907 907 for r in requirements:
908 908 # we allow multiple compression engine requirement to co-exist because
909 909 # strickly speaking, revlog seems to support mixed compression style.
910 910 #
911 911 # The compression used for new entries will be "the last one"
912 912 prefix = r.startswith
913 913 if prefix(b'revlog-compression-') or prefix(b'exp-compression-'):
914 914 options[b'compengine'] = r.split(b'-', 2)[2]
915 915
916 916 options[b'zlib.level'] = ui.configint(b'storage', b'revlog.zlib.level')
917 917 if options[b'zlib.level'] is not None:
918 918 if not (0 <= options[b'zlib.level'] <= 9):
919 919 msg = _(b'invalid value for `storage.revlog.zlib.level` config: %d')
920 920 raise error.Abort(msg % options[b'zlib.level'])
921 921 options[b'zstd.level'] = ui.configint(b'storage', b'revlog.zstd.level')
922 922 if options[b'zstd.level'] is not None:
923 923 if not (0 <= options[b'zstd.level'] <= 22):
924 924 msg = _(b'invalid value for `storage.revlog.zstd.level` config: %d')
925 925 raise error.Abort(msg % options[b'zstd.level'])
926 926
927 927 if repository.NARROW_REQUIREMENT in requirements:
928 928 options[b'enableellipsis'] = True
929 929
930 930 if ui.configbool(b'experimental', b'rust.index'):
931 931 options[b'rust.index'] = True
932 932
933 933 return options
934 934
935 935
936 936 def makemain(**kwargs):
937 937 """Produce a type conforming to ``ilocalrepositorymain``."""
938 938 return localrepository
939 939
940 940
941 941 @interfaceutil.implementer(repository.ilocalrepositoryfilestorage)
942 942 class revlogfilestorage(object):
943 943 """File storage when using revlogs."""
944 944
945 945 def file(self, path):
946 946 if path[0] == b'/':
947 947 path = path[1:]
948 948
949 949 return filelog.filelog(self.svfs, path)
950 950
951 951
952 952 @interfaceutil.implementer(repository.ilocalrepositoryfilestorage)
953 953 class revlognarrowfilestorage(object):
954 954 """File storage when using revlogs and narrow files."""
955 955
956 956 def file(self, path):
957 957 if path[0] == b'/':
958 958 path = path[1:]
959 959
960 960 return filelog.narrowfilelog(self.svfs, path, self._storenarrowmatch)
961 961
962 962
963 963 def makefilestorage(requirements, features, **kwargs):
964 964 """Produce a type conforming to ``ilocalrepositoryfilestorage``."""
965 965 features.add(repository.REPO_FEATURE_REVLOG_FILE_STORAGE)
966 966 features.add(repository.REPO_FEATURE_STREAM_CLONE)
967 967
968 968 if repository.NARROW_REQUIREMENT in requirements:
969 969 return revlognarrowfilestorage
970 970 else:
971 971 return revlogfilestorage
972 972
973 973
974 974 # List of repository interfaces and factory functions for them. Each
975 975 # will be called in order during ``makelocalrepository()`` to iteratively
976 976 # derive the final type for a local repository instance. We capture the
977 977 # function as a lambda so we don't hold a reference and the module-level
978 978 # functions can be wrapped.
979 979 REPO_INTERFACES = [
980 980 (repository.ilocalrepositorymain, lambda: makemain),
981 981 (repository.ilocalrepositoryfilestorage, lambda: makefilestorage),
982 982 ]
983 983
984 984
985 985 @interfaceutil.implementer(repository.ilocalrepositorymain)
986 986 class localrepository(object):
987 987 """Main class for representing local repositories.
988 988
989 989 All local repositories are instances of this class.
990 990
991 991 Constructed on its own, instances of this class are not usable as
992 992 repository objects. To obtain a usable repository object, call
993 993 ``hg.repository()``, ``localrepo.instance()``, or
994 994 ``localrepo.makelocalrepository()``. The latter is the lowest-level.
995 995 ``instance()`` adds support for creating new repositories.
996 996 ``hg.repository()`` adds more extension integration, including calling
997 997 ``reposetup()``. Generally speaking, ``hg.repository()`` should be
998 998 used.
999 999 """
1000 1000
1001 1001 # obsolete experimental requirements:
1002 1002 # - manifestv2: An experimental new manifest format that allowed
1003 1003 # for stem compression of long paths. Experiment ended up not
1004 1004 # being successful (repository sizes went up due to worse delta
1005 1005 # chains), and the code was deleted in 4.6.
1006 1006 supportedformats = {
1007 1007 b'revlogv1',
1008 1008 b'generaldelta',
1009 1009 b'treemanifest',
1010 1010 COPIESSDC_REQUIREMENT,
1011 1011 REVLOGV2_REQUIREMENT,
1012 1012 SIDEDATA_REQUIREMENT,
1013 1013 SPARSEREVLOG_REQUIREMENT,
1014 1014 bookmarks.BOOKMARKS_IN_STORE_REQUIREMENT,
1015 1015 }
1016 1016 _basesupported = supportedformats | {
1017 1017 b'store',
1018 1018 b'fncache',
1019 1019 b'shared',
1020 1020 b'relshared',
1021 1021 b'dotencode',
1022 1022 b'exp-sparse',
1023 1023 b'internal-phase',
1024 1024 }
1025 1025
1026 1026 # list of prefix for file which can be written without 'wlock'
1027 1027 # Extensions should extend this list when needed
1028 1028 _wlockfreeprefix = {
1029 1029 # We migh consider requiring 'wlock' for the next
1030 1030 # two, but pretty much all the existing code assume
1031 1031 # wlock is not needed so we keep them excluded for
1032 1032 # now.
1033 1033 b'hgrc',
1034 1034 b'requires',
1035 1035 # XXX cache is a complicatged business someone
1036 1036 # should investigate this in depth at some point
1037 1037 b'cache/',
1038 1038 # XXX shouldn't be dirstate covered by the wlock?
1039 1039 b'dirstate',
1040 1040 # XXX bisect was still a bit too messy at the time
1041 1041 # this changeset was introduced. Someone should fix
1042 1042 # the remainig bit and drop this line
1043 1043 b'bisect.state',
1044 1044 }
1045 1045
1046 1046 def __init__(
1047 1047 self,
1048 1048 baseui,
1049 1049 ui,
1050 1050 origroot,
1051 1051 wdirvfs,
1052 1052 hgvfs,
1053 1053 requirements,
1054 1054 supportedrequirements,
1055 1055 sharedpath,
1056 1056 store,
1057 1057 cachevfs,
1058 1058 wcachevfs,
1059 1059 features,
1060 1060 intents=None,
1061 1061 ):
1062 1062 """Create a new local repository instance.
1063 1063
1064 1064 Most callers should use ``hg.repository()``, ``localrepo.instance()``,
1065 1065 or ``localrepo.makelocalrepository()`` for obtaining a new repository
1066 1066 object.
1067 1067
1068 1068 Arguments:
1069 1069
1070 1070 baseui
1071 1071 ``ui.ui`` instance that ``ui`` argument was based off of.
1072 1072
1073 1073 ui
1074 1074 ``ui.ui`` instance for use by the repository.
1075 1075
1076 1076 origroot
1077 1077 ``bytes`` path to working directory root of this repository.
1078 1078
1079 1079 wdirvfs
1080 1080 ``vfs.vfs`` rooted at the working directory.
1081 1081
1082 1082 hgvfs
1083 1083 ``vfs.vfs`` rooted at .hg/
1084 1084
1085 1085 requirements
1086 1086 ``set`` of bytestrings representing repository opening requirements.
1087 1087
1088 1088 supportedrequirements
1089 1089 ``set`` of bytestrings representing repository requirements that we
1090 1090 know how to open. May be a supetset of ``requirements``.
1091 1091
1092 1092 sharedpath
1093 1093 ``bytes`` Defining path to storage base directory. Points to a
1094 1094 ``.hg/`` directory somewhere.
1095 1095
1096 1096 store
1097 1097 ``store.basicstore`` (or derived) instance providing access to
1098 1098 versioned storage.
1099 1099
1100 1100 cachevfs
1101 1101 ``vfs.vfs`` used for cache files.
1102 1102
1103 1103 wcachevfs
1104 1104 ``vfs.vfs`` used for cache files related to the working copy.
1105 1105
1106 1106 features
1107 1107 ``set`` of bytestrings defining features/capabilities of this
1108 1108 instance.
1109 1109
1110 1110 intents
1111 1111 ``set`` of system strings indicating what this repo will be used
1112 1112 for.
1113 1113 """
1114 1114 self.baseui = baseui
1115 1115 self.ui = ui
1116 1116 self.origroot = origroot
1117 1117 # vfs rooted at working directory.
1118 1118 self.wvfs = wdirvfs
1119 1119 self.root = wdirvfs.base
1120 1120 # vfs rooted at .hg/. Used to access most non-store paths.
1121 1121 self.vfs = hgvfs
1122 1122 self.path = hgvfs.base
1123 1123 self.requirements = requirements
1124 1124 self.supported = supportedrequirements
1125 1125 self.sharedpath = sharedpath
1126 1126 self.store = store
1127 1127 self.cachevfs = cachevfs
1128 1128 self.wcachevfs = wcachevfs
1129 1129 self.features = features
1130 1130
1131 1131 self.filtername = None
1132 1132
1133 1133 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
1134 1134 b'devel', b'check-locks'
1135 1135 ):
1136 1136 self.vfs.audit = self._getvfsward(self.vfs.audit)
1137 1137 # A list of callback to shape the phase if no data were found.
1138 1138 # Callback are in the form: func(repo, roots) --> processed root.
1139 1139 # This list it to be filled by extension during repo setup
1140 1140 self._phasedefaults = []
1141 1141
1142 1142 color.setup(self.ui)
1143 1143
1144 1144 self.spath = self.store.path
1145 1145 self.svfs = self.store.vfs
1146 1146 self.sjoin = self.store.join
1147 1147 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
1148 1148 b'devel', b'check-locks'
1149 1149 ):
1150 1150 if util.safehasattr(self.svfs, b'vfs'): # this is filtervfs
1151 1151 self.svfs.vfs.audit = self._getsvfsward(self.svfs.vfs.audit)
1152 1152 else: # standard vfs
1153 1153 self.svfs.audit = self._getsvfsward(self.svfs.audit)
1154 1154
1155 1155 self._dirstatevalidatewarned = False
1156 1156
1157 1157 self._branchcaches = branchmap.BranchMapCache()
1158 1158 self._revbranchcache = None
1159 1159 self._filterpats = {}
1160 1160 self._datafilters = {}
1161 1161 self._transref = self._lockref = self._wlockref = None
1162 1162
1163 1163 # A cache for various files under .hg/ that tracks file changes,
1164 1164 # (used by the filecache decorator)
1165 1165 #
1166 1166 # Maps a property name to its util.filecacheentry
1167 1167 self._filecache = {}
1168 1168
1169 1169 # hold sets of revision to be filtered
1170 1170 # should be cleared when something might have changed the filter value:
1171 1171 # - new changesets,
1172 1172 # - phase change,
1173 1173 # - new obsolescence marker,
1174 1174 # - working directory parent change,
1175 1175 # - bookmark changes
1176 1176 self.filteredrevcache = {}
1177 1177
1178 1178 # post-dirstate-status hooks
1179 1179 self._postdsstatus = []
1180 1180
1181 1181 # generic mapping between names and nodes
1182 1182 self.names = namespaces.namespaces()
1183 1183
1184 1184 # Key to signature value.
1185 1185 self._sparsesignaturecache = {}
1186 1186 # Signature to cached matcher instance.
1187 1187 self._sparsematchercache = {}
1188 1188
1189 1189 self._extrafilterid = repoview.extrafilter(ui)
1190 1190
1191 1191 self.filecopiesmode = None
1192 1192 if COPIESSDC_REQUIREMENT in self.requirements:
1193 1193 self.filecopiesmode = b'changeset-sidedata'
1194 1194
1195 1195 def _getvfsward(self, origfunc):
1196 1196 """build a ward for self.vfs"""
1197 1197 rref = weakref.ref(self)
1198 1198
1199 1199 def checkvfs(path, mode=None):
1200 1200 ret = origfunc(path, mode=mode)
1201 1201 repo = rref()
1202 1202 if (
1203 1203 repo is None
1204 1204 or not util.safehasattr(repo, b'_wlockref')
1205 1205 or not util.safehasattr(repo, b'_lockref')
1206 1206 ):
1207 1207 return
1208 1208 if mode in (None, b'r', b'rb'):
1209 1209 return
1210 1210 if path.startswith(repo.path):
1211 1211 # truncate name relative to the repository (.hg)
1212 1212 path = path[len(repo.path) + 1 :]
1213 1213 if path.startswith(b'cache/'):
1214 1214 msg = b'accessing cache with vfs instead of cachevfs: "%s"'
1215 1215 repo.ui.develwarn(msg % path, stacklevel=3, config=b"cache-vfs")
1216 1216 if path.startswith(b'journal.') or path.startswith(b'undo.'):
1217 1217 # journal is covered by 'lock'
1218 1218 if repo._currentlock(repo._lockref) is None:
1219 1219 repo.ui.develwarn(
1220 1220 b'write with no lock: "%s"' % path,
1221 1221 stacklevel=3,
1222 1222 config=b'check-locks',
1223 1223 )
1224 1224 elif repo._currentlock(repo._wlockref) is None:
1225 1225 # rest of vfs files are covered by 'wlock'
1226 1226 #
1227 1227 # exclude special files
1228 1228 for prefix in self._wlockfreeprefix:
1229 1229 if path.startswith(prefix):
1230 1230 return
1231 1231 repo.ui.develwarn(
1232 1232 b'write with no wlock: "%s"' % path,
1233 1233 stacklevel=3,
1234 1234 config=b'check-locks',
1235 1235 )
1236 1236 return ret
1237 1237
1238 1238 return checkvfs
1239 1239
1240 1240 def _getsvfsward(self, origfunc):
1241 1241 """build a ward for self.svfs"""
1242 1242 rref = weakref.ref(self)
1243 1243
1244 1244 def checksvfs(path, mode=None):
1245 1245 ret = origfunc(path, mode=mode)
1246 1246 repo = rref()
1247 1247 if repo is None or not util.safehasattr(repo, b'_lockref'):
1248 1248 return
1249 1249 if mode in (None, b'r', b'rb'):
1250 1250 return
1251 1251 if path.startswith(repo.sharedpath):
1252 1252 # truncate name relative to the repository (.hg)
1253 1253 path = path[len(repo.sharedpath) + 1 :]
1254 1254 if repo._currentlock(repo._lockref) is None:
1255 1255 repo.ui.develwarn(
1256 1256 b'write with no lock: "%s"' % path, stacklevel=4
1257 1257 )
1258 1258 return ret
1259 1259
1260 1260 return checksvfs
1261 1261
1262 1262 def close(self):
1263 1263 self._writecaches()
1264 1264
1265 1265 def _writecaches(self):
1266 1266 if self._revbranchcache:
1267 1267 self._revbranchcache.write()
1268 1268
1269 1269 def _restrictcapabilities(self, caps):
1270 1270 if self.ui.configbool(b'experimental', b'bundle2-advertise'):
1271 1271 caps = set(caps)
1272 1272 capsblob = bundle2.encodecaps(
1273 1273 bundle2.getrepocaps(self, role=b'client')
1274 1274 )
1275 1275 caps.add(b'bundle2=' + urlreq.quote(capsblob))
1276 1276 return caps
1277 1277
1278 1278 def _writerequirements(self):
1279 1279 scmutil.writerequires(self.vfs, self.requirements)
1280 1280
1281 1281 # Don't cache auditor/nofsauditor, or you'll end up with reference cycle:
1282 1282 # self -> auditor -> self._checknested -> self
1283 1283
1284 1284 @property
1285 1285 def auditor(self):
1286 1286 # This is only used by context.workingctx.match in order to
1287 1287 # detect files in subrepos.
1288 1288 return pathutil.pathauditor(self.root, callback=self._checknested)
1289 1289
1290 1290 @property
1291 1291 def nofsauditor(self):
1292 1292 # This is only used by context.basectx.match in order to detect
1293 1293 # files in subrepos.
1294 1294 return pathutil.pathauditor(
1295 1295 self.root, callback=self._checknested, realfs=False, cached=True
1296 1296 )
1297 1297
1298 1298 def _checknested(self, path):
1299 1299 """Determine if path is a legal nested repository."""
1300 1300 if not path.startswith(self.root):
1301 1301 return False
1302 1302 subpath = path[len(self.root) + 1 :]
1303 1303 normsubpath = util.pconvert(subpath)
1304 1304
1305 1305 # XXX: Checking against the current working copy is wrong in
1306 1306 # the sense that it can reject things like
1307 1307 #
1308 1308 # $ hg cat -r 10 sub/x.txt
1309 1309 #
1310 1310 # if sub/ is no longer a subrepository in the working copy
1311 1311 # parent revision.
1312 1312 #
1313 1313 # However, it can of course also allow things that would have
1314 1314 # been rejected before, such as the above cat command if sub/
1315 1315 # is a subrepository now, but was a normal directory before.
1316 1316 # The old path auditor would have rejected by mistake since it
1317 1317 # panics when it sees sub/.hg/.
1318 1318 #
1319 1319 # All in all, checking against the working copy seems sensible
1320 1320 # since we want to prevent access to nested repositories on
1321 1321 # the filesystem *now*.
1322 1322 ctx = self[None]
1323 1323 parts = util.splitpath(subpath)
1324 1324 while parts:
1325 1325 prefix = b'/'.join(parts)
1326 1326 if prefix in ctx.substate:
1327 1327 if prefix == normsubpath:
1328 1328 return True
1329 1329 else:
1330 1330 sub = ctx.sub(prefix)
1331 1331 return sub.checknested(subpath[len(prefix) + 1 :])
1332 1332 else:
1333 1333 parts.pop()
1334 1334 return False
1335 1335
1336 1336 def peer(self):
1337 1337 return localpeer(self) # not cached to avoid reference cycle
1338 1338
1339 1339 def unfiltered(self):
1340 1340 """Return unfiltered version of the repository
1341 1341
1342 1342 Intended to be overwritten by filtered repo."""
1343 1343 return self
1344 1344
1345 1345 def filtered(self, name, visibilityexceptions=None):
1346 1346 """Return a filtered version of a repository
1347 1347
1348 1348 The `name` parameter is the identifier of the requested view. This
1349 1349 will return a repoview object set "exactly" to the specified view.
1350 1350
1351 1351 This function does not apply recursive filtering to a repository. For
1352 1352 example calling `repo.filtered("served")` will return a repoview using
1353 1353 the "served" view, regardless of the initial view used by `repo`.
1354 1354
1355 1355 In other word, there is always only one level of `repoview` "filtering".
1356 1356 """
1357 1357 if self._extrafilterid is not None and b'%' not in name:
1358 1358 name = name + b'%' + self._extrafilterid
1359 1359
1360 1360 cls = repoview.newtype(self.unfiltered().__class__)
1361 1361 return cls(self, name, visibilityexceptions)
1362 1362
1363 1363 @mixedrepostorecache(
1364 1364 (b'bookmarks', b'plain'),
1365 1365 (b'bookmarks.current', b'plain'),
1366 1366 (b'bookmarks', b''),
1367 1367 (b'00changelog.i', b''),
1368 1368 )
1369 1369 def _bookmarks(self):
1370 1370 # Since the multiple files involved in the transaction cannot be
1371 1371 # written atomically (with current repository format), there is a race
1372 1372 # condition here.
1373 1373 #
1374 1374 # 1) changelog content A is read
1375 1375 # 2) outside transaction update changelog to content B
1376 1376 # 3) outside transaction update bookmark file referring to content B
1377 1377 # 4) bookmarks file content is read and filtered against changelog-A
1378 1378 #
1379 1379 # When this happens, bookmarks against nodes missing from A are dropped.
1380 1380 #
1381 1381 # Having this happening during read is not great, but it become worse
1382 1382 # when this happen during write because the bookmarks to the "unknown"
1383 1383 # nodes will be dropped for good. However, writes happen within locks.
1384 1384 # This locking makes it possible to have a race free consistent read.
1385 1385 # For this purpose data read from disc before locking are
1386 1386 # "invalidated" right after the locks are taken. This invalidations are
1387 1387 # "light", the `filecache` mechanism keep the data in memory and will
1388 1388 # reuse them if the underlying files did not changed. Not parsing the
1389 1389 # same data multiple times helps performances.
1390 1390 #
1391 1391 # Unfortunately in the case describe above, the files tracked by the
1392 1392 # bookmarks file cache might not have changed, but the in-memory
1393 1393 # content is still "wrong" because we used an older changelog content
1394 1394 # to process the on-disk data. So after locking, the changelog would be
1395 1395 # refreshed but `_bookmarks` would be preserved.
1396 1396 # Adding `00changelog.i` to the list of tracked file is not
1397 1397 # enough, because at the time we build the content for `_bookmarks` in
1398 1398 # (4), the changelog file has already diverged from the content used
1399 1399 # for loading `changelog` in (1)
1400 1400 #
1401 1401 # To prevent the issue, we force the changelog to be explicitly
1402 1402 # reloaded while computing `_bookmarks`. The data race can still happen
1403 1403 # without the lock (with a narrower window), but it would no longer go
1404 1404 # undetected during the lock time refresh.
1405 1405 #
1406 1406 # The new schedule is as follow
1407 1407 #
1408 1408 # 1) filecache logic detect that `_bookmarks` needs to be computed
1409 1409 # 2) cachestat for `bookmarks` and `changelog` are captured (for book)
1410 1410 # 3) We force `changelog` filecache to be tested
1411 1411 # 4) cachestat for `changelog` are captured (for changelog)
1412 1412 # 5) `_bookmarks` is computed and cached
1413 1413 #
1414 1414 # The step in (3) ensure we have a changelog at least as recent as the
1415 1415 # cache stat computed in (1). As a result at locking time:
1416 1416 # * if the changelog did not changed since (1) -> we can reuse the data
1417 1417 # * otherwise -> the bookmarks get refreshed.
1418 1418 self._refreshchangelog()
1419 1419 return bookmarks.bmstore(self)
1420 1420
1421 1421 def _refreshchangelog(self):
1422 1422 """make sure the in memory changelog match the on-disk one"""
1423 1423 if 'changelog' in vars(self) and self.currenttransaction() is None:
1424 1424 del self.changelog
1425 1425
1426 1426 @property
1427 1427 def _activebookmark(self):
1428 1428 return self._bookmarks.active
1429 1429
1430 1430 # _phasesets depend on changelog. what we need is to call
1431 1431 # _phasecache.invalidate() if '00changelog.i' was changed, but it
1432 1432 # can't be easily expressed in filecache mechanism.
1433 1433 @storecache(b'phaseroots', b'00changelog.i')
1434 1434 def _phasecache(self):
1435 1435 return phases.phasecache(self, self._phasedefaults)
1436 1436
1437 1437 @storecache(b'obsstore')
1438 1438 def obsstore(self):
1439 1439 return obsolete.makestore(self.ui, self)
1440 1440
1441 1441 @storecache(b'00changelog.i')
1442 1442 def changelog(self):
1443 1443 return self.store.changelog(txnutil.mayhavepending(self.root))
1444 1444
1445 1445 @storecache(b'00manifest.i')
1446 1446 def manifestlog(self):
1447 1447 return self.store.manifestlog(self, self._storenarrowmatch)
1448 1448
1449 1449 @repofilecache(b'dirstate')
1450 1450 def dirstate(self):
1451 1451 return self._makedirstate()
1452 1452
1453 1453 def _makedirstate(self):
1454 1454 """Extension point for wrapping the dirstate per-repo."""
1455 1455 sparsematchfn = lambda: sparse.matcher(self)
1456 1456
1457 1457 return dirstate.dirstate(
1458 1458 self.vfs, self.ui, self.root, self._dirstatevalidate, sparsematchfn
1459 1459 )
1460 1460
1461 1461 def _dirstatevalidate(self, node):
1462 1462 try:
1463 1463 self.changelog.rev(node)
1464 1464 return node
1465 1465 except error.LookupError:
1466 1466 if not self._dirstatevalidatewarned:
1467 1467 self._dirstatevalidatewarned = True
1468 1468 self.ui.warn(
1469 1469 _(b"warning: ignoring unknown working parent %s!\n")
1470 1470 % short(node)
1471 1471 )
1472 1472 return nullid
1473 1473
1474 1474 @storecache(narrowspec.FILENAME)
1475 1475 def narrowpats(self):
1476 1476 """matcher patterns for this repository's narrowspec
1477 1477
1478 1478 A tuple of (includes, excludes).
1479 1479 """
1480 1480 return narrowspec.load(self)
1481 1481
1482 1482 @storecache(narrowspec.FILENAME)
1483 1483 def _storenarrowmatch(self):
1484 1484 if repository.NARROW_REQUIREMENT not in self.requirements:
1485 1485 return matchmod.always()
1486 1486 include, exclude = self.narrowpats
1487 1487 return narrowspec.match(self.root, include=include, exclude=exclude)
1488 1488
1489 1489 @storecache(narrowspec.FILENAME)
1490 1490 def _narrowmatch(self):
1491 1491 if repository.NARROW_REQUIREMENT not in self.requirements:
1492 1492 return matchmod.always()
1493 1493 narrowspec.checkworkingcopynarrowspec(self)
1494 1494 include, exclude = self.narrowpats
1495 1495 return narrowspec.match(self.root, include=include, exclude=exclude)
1496 1496
1497 1497 def narrowmatch(self, match=None, includeexact=False):
1498 1498 """matcher corresponding the the repo's narrowspec
1499 1499
1500 1500 If `match` is given, then that will be intersected with the narrow
1501 1501 matcher.
1502 1502
1503 1503 If `includeexact` is True, then any exact matches from `match` will
1504 1504 be included even if they're outside the narrowspec.
1505 1505 """
1506 1506 if match:
1507 1507 if includeexact and not self._narrowmatch.always():
1508 1508 # do not exclude explicitly-specified paths so that they can
1509 1509 # be warned later on
1510 1510 em = matchmod.exact(match.files())
1511 1511 nm = matchmod.unionmatcher([self._narrowmatch, em])
1512 1512 return matchmod.intersectmatchers(match, nm)
1513 1513 return matchmod.intersectmatchers(match, self._narrowmatch)
1514 1514 return self._narrowmatch
1515 1515
1516 1516 def setnarrowpats(self, newincludes, newexcludes):
1517 1517 narrowspec.save(self, newincludes, newexcludes)
1518 1518 self.invalidate(clearfilecache=True)
1519 1519
1520 1520 @unfilteredpropertycache
1521 1521 def _quick_access_changeid_null(self):
1522 1522 return {
1523 1523 b'null': (nullrev, nullid),
1524 1524 nullrev: (nullrev, nullid),
1525 1525 nullid: (nullrev, nullid),
1526 1526 }
1527 1527
1528 1528 @unfilteredpropertycache
1529 1529 def _quick_access_changeid_wc(self):
1530 1530 # also fast path access to the working copy parents
1531 1531 # however, only do it for filter that ensure wc is visible.
1532 1532 quick = {}
1533 1533 cl = self.unfiltered().changelog
1534 1534 for node in self.dirstate.parents():
1535 1535 if node == nullid:
1536 1536 continue
1537 1537 rev = cl.index.get_rev(node)
1538 1538 if rev is None:
1539 1539 # unknown working copy parent case:
1540 1540 #
1541 1541 # skip the fast path and let higher code deal with it
1542 1542 continue
1543 1543 pair = (rev, node)
1544 1544 quick[rev] = pair
1545 1545 quick[node] = pair
1546 p1node = self.dirstate.p1()
1547 if p1node != nullid:
1548 quick[b'.'] = quick[p1node]
1546 1549 return quick
1547 1550
1548 1551 @unfilteredmethod
1549 1552 def _quick_access_changeid_invalidate(self):
1550 1553 if '_quick_access_changeid_wc' in vars(self):
1551 1554 del self.__dict__['_quick_access_changeid_wc']
1552 1555
1553 1556 @property
1554 1557 def _quick_access_changeid(self):
1555 1558 """an helper dictionnary for __getitem__ calls
1556 1559
1557 1560 This contains a list of symbol we can recognise right away without
1558 1561 further processing.
1559 1562 """
1560 1563 mapping = self._quick_access_changeid_null
1561 1564 if self.filtername in repoview.filter_has_wc:
1562 1565 mapping = mapping.copy()
1563 1566 mapping.update(self._quick_access_changeid_wc)
1564 1567 return mapping
1565 1568
1566 1569 def __getitem__(self, changeid):
1567 1570 # dealing with special cases
1568 1571 if changeid is None:
1569 1572 return context.workingctx(self)
1570 1573 if isinstance(changeid, context.basectx):
1571 1574 return changeid
1572 1575
1573 1576 # dealing with multiple revisions
1574 1577 if isinstance(changeid, slice):
1575 1578 # wdirrev isn't contiguous so the slice shouldn't include it
1576 1579 return [
1577 1580 self[i]
1578 1581 for i in pycompat.xrange(*changeid.indices(len(self)))
1579 1582 if i not in self.changelog.filteredrevs
1580 1583 ]
1581 1584
1582 1585 # dealing with some special values
1583 1586 quick_access = self._quick_access_changeid.get(changeid)
1584 1587 if quick_access is not None:
1585 1588 rev, node = quick_access
1586 1589 return context.changectx(self, rev, node, maybe_filtered=False)
1587 1590 if changeid == b'tip':
1588 1591 node = self.changelog.tip()
1589 1592 rev = self.changelog.rev(node)
1590 1593 return context.changectx(self, rev, node)
1591 1594
1592 1595 # dealing with arbitrary values
1593 1596 try:
1594 1597 if isinstance(changeid, int):
1595 1598 node = self.changelog.node(changeid)
1596 1599 rev = changeid
1597 1600 elif changeid == b'.':
1598 1601 # this is a hack to delay/avoid loading obsmarkers
1599 1602 # when we know that '.' won't be hidden
1600 1603 node = self.dirstate.p1()
1601 1604 rev = self.unfiltered().changelog.rev(node)
1602 1605 elif len(changeid) == 20:
1603 1606 try:
1604 1607 node = changeid
1605 1608 rev = self.changelog.rev(changeid)
1606 1609 except error.FilteredLookupError:
1607 1610 changeid = hex(changeid) # for the error message
1608 1611 raise
1609 1612 except LookupError:
1610 1613 # check if it might have come from damaged dirstate
1611 1614 #
1612 1615 # XXX we could avoid the unfiltered if we had a recognizable
1613 1616 # exception for filtered changeset access
1614 1617 if (
1615 1618 self.local()
1616 1619 and changeid in self.unfiltered().dirstate.parents()
1617 1620 ):
1618 1621 msg = _(b"working directory has unknown parent '%s'!")
1619 1622 raise error.Abort(msg % short(changeid))
1620 1623 changeid = hex(changeid) # for the error message
1621 1624 raise
1622 1625
1623 1626 elif len(changeid) == 40:
1624 1627 node = bin(changeid)
1625 1628 rev = self.changelog.rev(node)
1626 1629 else:
1627 1630 raise error.ProgrammingError(
1628 1631 b"unsupported changeid '%s' of type %s"
1629 1632 % (changeid, pycompat.bytestr(type(changeid)))
1630 1633 )
1631 1634
1632 1635 return context.changectx(self, rev, node)
1633 1636
1634 1637 except (error.FilteredIndexError, error.FilteredLookupError):
1635 1638 raise error.FilteredRepoLookupError(
1636 1639 _(b"filtered revision '%s'") % pycompat.bytestr(changeid)
1637 1640 )
1638 1641 except (IndexError, LookupError):
1639 1642 raise error.RepoLookupError(
1640 1643 _(b"unknown revision '%s'") % pycompat.bytestr(changeid)
1641 1644 )
1642 1645 except error.WdirUnsupported:
1643 1646 return context.workingctx(self)
1644 1647
1645 1648 def __contains__(self, changeid):
1646 1649 """True if the given changeid exists
1647 1650
1648 1651 error.AmbiguousPrefixLookupError is raised if an ambiguous node
1649 1652 specified.
1650 1653 """
1651 1654 try:
1652 1655 self[changeid]
1653 1656 return True
1654 1657 except error.RepoLookupError:
1655 1658 return False
1656 1659
1657 1660 def __nonzero__(self):
1658 1661 return True
1659 1662
1660 1663 __bool__ = __nonzero__
1661 1664
1662 1665 def __len__(self):
1663 1666 # no need to pay the cost of repoview.changelog
1664 1667 unfi = self.unfiltered()
1665 1668 return len(unfi.changelog)
1666 1669
1667 1670 def __iter__(self):
1668 1671 return iter(self.changelog)
1669 1672
1670 1673 def revs(self, expr, *args):
1671 1674 '''Find revisions matching a revset.
1672 1675
1673 1676 The revset is specified as a string ``expr`` that may contain
1674 1677 %-formatting to escape certain types. See ``revsetlang.formatspec``.
1675 1678
1676 1679 Revset aliases from the configuration are not expanded. To expand
1677 1680 user aliases, consider calling ``scmutil.revrange()`` or
1678 1681 ``repo.anyrevs([expr], user=True)``.
1679 1682
1680 1683 Returns a smartset.abstractsmartset, which is a list-like interface
1681 1684 that contains integer revisions.
1682 1685 '''
1683 1686 tree = revsetlang.spectree(expr, *args)
1684 1687 return revset.makematcher(tree)(self)
1685 1688
1686 1689 def set(self, expr, *args):
1687 1690 '''Find revisions matching a revset and emit changectx instances.
1688 1691
1689 1692 This is a convenience wrapper around ``revs()`` that iterates the
1690 1693 result and is a generator of changectx instances.
1691 1694
1692 1695 Revset aliases from the configuration are not expanded. To expand
1693 1696 user aliases, consider calling ``scmutil.revrange()``.
1694 1697 '''
1695 1698 for r in self.revs(expr, *args):
1696 1699 yield self[r]
1697 1700
1698 1701 def anyrevs(self, specs, user=False, localalias=None):
1699 1702 '''Find revisions matching one of the given revsets.
1700 1703
1701 1704 Revset aliases from the configuration are not expanded by default. To
1702 1705 expand user aliases, specify ``user=True``. To provide some local
1703 1706 definitions overriding user aliases, set ``localalias`` to
1704 1707 ``{name: definitionstring}``.
1705 1708 '''
1706 1709 if specs == [b'null']:
1707 1710 return revset.baseset([nullrev])
1708 1711 if user:
1709 1712 m = revset.matchany(
1710 1713 self.ui,
1711 1714 specs,
1712 1715 lookup=revset.lookupfn(self),
1713 1716 localalias=localalias,
1714 1717 )
1715 1718 else:
1716 1719 m = revset.matchany(None, specs, localalias=localalias)
1717 1720 return m(self)
1718 1721
1719 1722 def url(self):
1720 1723 return b'file:' + self.root
1721 1724
1722 1725 def hook(self, name, throw=False, **args):
1723 1726 """Call a hook, passing this repo instance.
1724 1727
1725 1728 This a convenience method to aid invoking hooks. Extensions likely
1726 1729 won't call this unless they have registered a custom hook or are
1727 1730 replacing code that is expected to call a hook.
1728 1731 """
1729 1732 return hook.hook(self.ui, self, name, throw, **args)
1730 1733
1731 1734 @filteredpropertycache
1732 1735 def _tagscache(self):
1733 1736 '''Returns a tagscache object that contains various tags related
1734 1737 caches.'''
1735 1738
1736 1739 # This simplifies its cache management by having one decorated
1737 1740 # function (this one) and the rest simply fetch things from it.
1738 1741 class tagscache(object):
1739 1742 def __init__(self):
1740 1743 # These two define the set of tags for this repository. tags
1741 1744 # maps tag name to node; tagtypes maps tag name to 'global' or
1742 1745 # 'local'. (Global tags are defined by .hgtags across all
1743 1746 # heads, and local tags are defined in .hg/localtags.)
1744 1747 # They constitute the in-memory cache of tags.
1745 1748 self.tags = self.tagtypes = None
1746 1749
1747 1750 self.nodetagscache = self.tagslist = None
1748 1751
1749 1752 cache = tagscache()
1750 1753 cache.tags, cache.tagtypes = self._findtags()
1751 1754
1752 1755 return cache
1753 1756
1754 1757 def tags(self):
1755 1758 '''return a mapping of tag to node'''
1756 1759 t = {}
1757 1760 if self.changelog.filteredrevs:
1758 1761 tags, tt = self._findtags()
1759 1762 else:
1760 1763 tags = self._tagscache.tags
1761 1764 rev = self.changelog.rev
1762 1765 for k, v in pycompat.iteritems(tags):
1763 1766 try:
1764 1767 # ignore tags to unknown nodes
1765 1768 rev(v)
1766 1769 t[k] = v
1767 1770 except (error.LookupError, ValueError):
1768 1771 pass
1769 1772 return t
1770 1773
1771 1774 def _findtags(self):
1772 1775 '''Do the hard work of finding tags. Return a pair of dicts
1773 1776 (tags, tagtypes) where tags maps tag name to node, and tagtypes
1774 1777 maps tag name to a string like \'global\' or \'local\'.
1775 1778 Subclasses or extensions are free to add their own tags, but
1776 1779 should be aware that the returned dicts will be retained for the
1777 1780 duration of the localrepo object.'''
1778 1781
1779 1782 # XXX what tagtype should subclasses/extensions use? Currently
1780 1783 # mq and bookmarks add tags, but do not set the tagtype at all.
1781 1784 # Should each extension invent its own tag type? Should there
1782 1785 # be one tagtype for all such "virtual" tags? Or is the status
1783 1786 # quo fine?
1784 1787
1785 1788 # map tag name to (node, hist)
1786 1789 alltags = tagsmod.findglobaltags(self.ui, self)
1787 1790 # map tag name to tag type
1788 1791 tagtypes = dict((tag, b'global') for tag in alltags)
1789 1792
1790 1793 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
1791 1794
1792 1795 # Build the return dicts. Have to re-encode tag names because
1793 1796 # the tags module always uses UTF-8 (in order not to lose info
1794 1797 # writing to the cache), but the rest of Mercurial wants them in
1795 1798 # local encoding.
1796 1799 tags = {}
1797 1800 for (name, (node, hist)) in pycompat.iteritems(alltags):
1798 1801 if node != nullid:
1799 1802 tags[encoding.tolocal(name)] = node
1800 1803 tags[b'tip'] = self.changelog.tip()
1801 1804 tagtypes = dict(
1802 1805 [
1803 1806 (encoding.tolocal(name), value)
1804 1807 for (name, value) in pycompat.iteritems(tagtypes)
1805 1808 ]
1806 1809 )
1807 1810 return (tags, tagtypes)
1808 1811
1809 1812 def tagtype(self, tagname):
1810 1813 '''
1811 1814 return the type of the given tag. result can be:
1812 1815
1813 1816 'local' : a local tag
1814 1817 'global' : a global tag
1815 1818 None : tag does not exist
1816 1819 '''
1817 1820
1818 1821 return self._tagscache.tagtypes.get(tagname)
1819 1822
1820 1823 def tagslist(self):
1821 1824 '''return a list of tags ordered by revision'''
1822 1825 if not self._tagscache.tagslist:
1823 1826 l = []
1824 1827 for t, n in pycompat.iteritems(self.tags()):
1825 1828 l.append((self.changelog.rev(n), t, n))
1826 1829 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
1827 1830
1828 1831 return self._tagscache.tagslist
1829 1832
1830 1833 def nodetags(self, node):
1831 1834 '''return the tags associated with a node'''
1832 1835 if not self._tagscache.nodetagscache:
1833 1836 nodetagscache = {}
1834 1837 for t, n in pycompat.iteritems(self._tagscache.tags):
1835 1838 nodetagscache.setdefault(n, []).append(t)
1836 1839 for tags in pycompat.itervalues(nodetagscache):
1837 1840 tags.sort()
1838 1841 self._tagscache.nodetagscache = nodetagscache
1839 1842 return self._tagscache.nodetagscache.get(node, [])
1840 1843
1841 1844 def nodebookmarks(self, node):
1842 1845 """return the list of bookmarks pointing to the specified node"""
1843 1846 return self._bookmarks.names(node)
1844 1847
1845 1848 def branchmap(self):
1846 1849 '''returns a dictionary {branch: [branchheads]} with branchheads
1847 1850 ordered by increasing revision number'''
1848 1851 return self._branchcaches[self]
1849 1852
1850 1853 @unfilteredmethod
1851 1854 def revbranchcache(self):
1852 1855 if not self._revbranchcache:
1853 1856 self._revbranchcache = branchmap.revbranchcache(self.unfiltered())
1854 1857 return self._revbranchcache
1855 1858
1856 1859 def branchtip(self, branch, ignoremissing=False):
1857 1860 '''return the tip node for a given branch
1858 1861
1859 1862 If ignoremissing is True, then this method will not raise an error.
1860 1863 This is helpful for callers that only expect None for a missing branch
1861 1864 (e.g. namespace).
1862 1865
1863 1866 '''
1864 1867 try:
1865 1868 return self.branchmap().branchtip(branch)
1866 1869 except KeyError:
1867 1870 if not ignoremissing:
1868 1871 raise error.RepoLookupError(_(b"unknown branch '%s'") % branch)
1869 1872 else:
1870 1873 pass
1871 1874
1872 1875 def lookup(self, key):
1873 1876 node = scmutil.revsymbol(self, key).node()
1874 1877 if node is None:
1875 1878 raise error.RepoLookupError(_(b"unknown revision '%s'") % key)
1876 1879 return node
1877 1880
1878 1881 def lookupbranch(self, key):
1879 1882 if self.branchmap().hasbranch(key):
1880 1883 return key
1881 1884
1882 1885 return scmutil.revsymbol(self, key).branch()
1883 1886
1884 1887 def known(self, nodes):
1885 1888 cl = self.changelog
1886 1889 get_rev = cl.index.get_rev
1887 1890 filtered = cl.filteredrevs
1888 1891 result = []
1889 1892 for n in nodes:
1890 1893 r = get_rev(n)
1891 1894 resp = not (r is None or r in filtered)
1892 1895 result.append(resp)
1893 1896 return result
1894 1897
1895 1898 def local(self):
1896 1899 return self
1897 1900
1898 1901 def publishing(self):
1899 1902 # it's safe (and desirable) to trust the publish flag unconditionally
1900 1903 # so that we don't finalize changes shared between users via ssh or nfs
1901 1904 return self.ui.configbool(b'phases', b'publish', untrusted=True)
1902 1905
1903 1906 def cancopy(self):
1904 1907 # so statichttprepo's override of local() works
1905 1908 if not self.local():
1906 1909 return False
1907 1910 if not self.publishing():
1908 1911 return True
1909 1912 # if publishing we can't copy if there is filtered content
1910 1913 return not self.filtered(b'visible').changelog.filteredrevs
1911 1914
1912 1915 def shared(self):
1913 1916 '''the type of shared repository (None if not shared)'''
1914 1917 if self.sharedpath != self.path:
1915 1918 return b'store'
1916 1919 return None
1917 1920
1918 1921 def wjoin(self, f, *insidef):
1919 1922 return self.vfs.reljoin(self.root, f, *insidef)
1920 1923
1921 1924 def setparents(self, p1, p2=nullid):
1922 1925 self[None].setparents(p1, p2)
1923 1926 self._quick_access_changeid_invalidate()
1924 1927
1925 1928 def filectx(self, path, changeid=None, fileid=None, changectx=None):
1926 1929 """changeid must be a changeset revision, if specified.
1927 1930 fileid can be a file revision or node."""
1928 1931 return context.filectx(
1929 1932 self, path, changeid, fileid, changectx=changectx
1930 1933 )
1931 1934
1932 1935 def getcwd(self):
1933 1936 return self.dirstate.getcwd()
1934 1937
1935 1938 def pathto(self, f, cwd=None):
1936 1939 return self.dirstate.pathto(f, cwd)
1937 1940
1938 1941 def _loadfilter(self, filter):
1939 1942 if filter not in self._filterpats:
1940 1943 l = []
1941 1944 for pat, cmd in self.ui.configitems(filter):
1942 1945 if cmd == b'!':
1943 1946 continue
1944 1947 mf = matchmod.match(self.root, b'', [pat])
1945 1948 fn = None
1946 1949 params = cmd
1947 1950 for name, filterfn in pycompat.iteritems(self._datafilters):
1948 1951 if cmd.startswith(name):
1949 1952 fn = filterfn
1950 1953 params = cmd[len(name) :].lstrip()
1951 1954 break
1952 1955 if not fn:
1953 1956 fn = lambda s, c, **kwargs: procutil.filter(s, c)
1954 1957 fn.__name__ = 'commandfilter'
1955 1958 # Wrap old filters not supporting keyword arguments
1956 1959 if not pycompat.getargspec(fn)[2]:
1957 1960 oldfn = fn
1958 1961 fn = lambda s, c, oldfn=oldfn, **kwargs: oldfn(s, c)
1959 1962 fn.__name__ = 'compat-' + oldfn.__name__
1960 1963 l.append((mf, fn, params))
1961 1964 self._filterpats[filter] = l
1962 1965 return self._filterpats[filter]
1963 1966
1964 1967 def _filter(self, filterpats, filename, data):
1965 1968 for mf, fn, cmd in filterpats:
1966 1969 if mf(filename):
1967 1970 self.ui.debug(
1968 1971 b"filtering %s through %s\n"
1969 1972 % (filename, cmd or pycompat.sysbytes(fn.__name__))
1970 1973 )
1971 1974 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
1972 1975 break
1973 1976
1974 1977 return data
1975 1978
1976 1979 @unfilteredpropertycache
1977 1980 def _encodefilterpats(self):
1978 1981 return self._loadfilter(b'encode')
1979 1982
1980 1983 @unfilteredpropertycache
1981 1984 def _decodefilterpats(self):
1982 1985 return self._loadfilter(b'decode')
1983 1986
1984 1987 def adddatafilter(self, name, filter):
1985 1988 self._datafilters[name] = filter
1986 1989
1987 1990 def wread(self, filename):
1988 1991 if self.wvfs.islink(filename):
1989 1992 data = self.wvfs.readlink(filename)
1990 1993 else:
1991 1994 data = self.wvfs.read(filename)
1992 1995 return self._filter(self._encodefilterpats, filename, data)
1993 1996
1994 1997 def wwrite(self, filename, data, flags, backgroundclose=False, **kwargs):
1995 1998 """write ``data`` into ``filename`` in the working directory
1996 1999
1997 2000 This returns length of written (maybe decoded) data.
1998 2001 """
1999 2002 data = self._filter(self._decodefilterpats, filename, data)
2000 2003 if b'l' in flags:
2001 2004 self.wvfs.symlink(data, filename)
2002 2005 else:
2003 2006 self.wvfs.write(
2004 2007 filename, data, backgroundclose=backgroundclose, **kwargs
2005 2008 )
2006 2009 if b'x' in flags:
2007 2010 self.wvfs.setflags(filename, False, True)
2008 2011 else:
2009 2012 self.wvfs.setflags(filename, False, False)
2010 2013 return len(data)
2011 2014
2012 2015 def wwritedata(self, filename, data):
2013 2016 return self._filter(self._decodefilterpats, filename, data)
2014 2017
2015 2018 def currenttransaction(self):
2016 2019 """return the current transaction or None if non exists"""
2017 2020 if self._transref:
2018 2021 tr = self._transref()
2019 2022 else:
2020 2023 tr = None
2021 2024
2022 2025 if tr and tr.running():
2023 2026 return tr
2024 2027 return None
2025 2028
2026 2029 def transaction(self, desc, report=None):
2027 2030 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
2028 2031 b'devel', b'check-locks'
2029 2032 ):
2030 2033 if self._currentlock(self._lockref) is None:
2031 2034 raise error.ProgrammingError(b'transaction requires locking')
2032 2035 tr = self.currenttransaction()
2033 2036 if tr is not None:
2034 2037 return tr.nest(name=desc)
2035 2038
2036 2039 # abort here if the journal already exists
2037 2040 if self.svfs.exists(b"journal"):
2038 2041 raise error.RepoError(
2039 2042 _(b"abandoned transaction found"),
2040 2043 hint=_(b"run 'hg recover' to clean up transaction"),
2041 2044 )
2042 2045
2043 2046 idbase = b"%.40f#%f" % (random.random(), time.time())
2044 2047 ha = hex(hashutil.sha1(idbase).digest())
2045 2048 txnid = b'TXN:' + ha
2046 2049 self.hook(b'pretxnopen', throw=True, txnname=desc, txnid=txnid)
2047 2050
2048 2051 self._writejournal(desc)
2049 2052 renames = [(vfs, x, undoname(x)) for vfs, x in self._journalfiles()]
2050 2053 if report:
2051 2054 rp = report
2052 2055 else:
2053 2056 rp = self.ui.warn
2054 2057 vfsmap = {b'plain': self.vfs, b'store': self.svfs} # root of .hg/
2055 2058 # we must avoid cyclic reference between repo and transaction.
2056 2059 reporef = weakref.ref(self)
2057 2060 # Code to track tag movement
2058 2061 #
2059 2062 # Since tags are all handled as file content, it is actually quite hard
2060 2063 # to track these movement from a code perspective. So we fallback to a
2061 2064 # tracking at the repository level. One could envision to track changes
2062 2065 # to the '.hgtags' file through changegroup apply but that fails to
2063 2066 # cope with case where transaction expose new heads without changegroup
2064 2067 # being involved (eg: phase movement).
2065 2068 #
2066 2069 # For now, We gate the feature behind a flag since this likely comes
2067 2070 # with performance impacts. The current code run more often than needed
2068 2071 # and do not use caches as much as it could. The current focus is on
2069 2072 # the behavior of the feature so we disable it by default. The flag
2070 2073 # will be removed when we are happy with the performance impact.
2071 2074 #
2072 2075 # Once this feature is no longer experimental move the following
2073 2076 # documentation to the appropriate help section:
2074 2077 #
2075 2078 # The ``HG_TAG_MOVED`` variable will be set if the transaction touched
2076 2079 # tags (new or changed or deleted tags). In addition the details of
2077 2080 # these changes are made available in a file at:
2078 2081 # ``REPOROOT/.hg/changes/tags.changes``.
2079 2082 # Make sure you check for HG_TAG_MOVED before reading that file as it
2080 2083 # might exist from a previous transaction even if no tag were touched
2081 2084 # in this one. Changes are recorded in a line base format::
2082 2085 #
2083 2086 # <action> <hex-node> <tag-name>\n
2084 2087 #
2085 2088 # Actions are defined as follow:
2086 2089 # "-R": tag is removed,
2087 2090 # "+A": tag is added,
2088 2091 # "-M": tag is moved (old value),
2089 2092 # "+M": tag is moved (new value),
2090 2093 tracktags = lambda x: None
2091 2094 # experimental config: experimental.hook-track-tags
2092 2095 shouldtracktags = self.ui.configbool(
2093 2096 b'experimental', b'hook-track-tags'
2094 2097 )
2095 2098 if desc != b'strip' and shouldtracktags:
2096 2099 oldheads = self.changelog.headrevs()
2097 2100
2098 2101 def tracktags(tr2):
2099 2102 repo = reporef()
2100 2103 oldfnodes = tagsmod.fnoderevs(repo.ui, repo, oldheads)
2101 2104 newheads = repo.changelog.headrevs()
2102 2105 newfnodes = tagsmod.fnoderevs(repo.ui, repo, newheads)
2103 2106 # notes: we compare lists here.
2104 2107 # As we do it only once buiding set would not be cheaper
2105 2108 changes = tagsmod.difftags(repo.ui, repo, oldfnodes, newfnodes)
2106 2109 if changes:
2107 2110 tr2.hookargs[b'tag_moved'] = b'1'
2108 2111 with repo.vfs(
2109 2112 b'changes/tags.changes', b'w', atomictemp=True
2110 2113 ) as changesfile:
2111 2114 # note: we do not register the file to the transaction
2112 2115 # because we needs it to still exist on the transaction
2113 2116 # is close (for txnclose hooks)
2114 2117 tagsmod.writediff(changesfile, changes)
2115 2118
2116 2119 def validate(tr2):
2117 2120 """will run pre-closing hooks"""
2118 2121 # XXX the transaction API is a bit lacking here so we take a hacky
2119 2122 # path for now
2120 2123 #
2121 2124 # We cannot add this as a "pending" hooks since the 'tr.hookargs'
2122 2125 # dict is copied before these run. In addition we needs the data
2123 2126 # available to in memory hooks too.
2124 2127 #
2125 2128 # Moreover, we also need to make sure this runs before txnclose
2126 2129 # hooks and there is no "pending" mechanism that would execute
2127 2130 # logic only if hooks are about to run.
2128 2131 #
2129 2132 # Fixing this limitation of the transaction is also needed to track
2130 2133 # other families of changes (bookmarks, phases, obsolescence).
2131 2134 #
2132 2135 # This will have to be fixed before we remove the experimental
2133 2136 # gating.
2134 2137 tracktags(tr2)
2135 2138 repo = reporef()
2136 2139
2137 2140 singleheadopt = (b'experimental', b'single-head-per-branch')
2138 2141 singlehead = repo.ui.configbool(*singleheadopt)
2139 2142 if singlehead:
2140 2143 singleheadsub = repo.ui.configsuboptions(*singleheadopt)[1]
2141 2144 accountclosed = singleheadsub.get(
2142 2145 b"account-closed-heads", False
2143 2146 )
2144 2147 scmutil.enforcesinglehead(repo, tr2, desc, accountclosed)
2145 2148 if hook.hashook(repo.ui, b'pretxnclose-bookmark'):
2146 2149 for name, (old, new) in sorted(
2147 2150 tr.changes[b'bookmarks'].items()
2148 2151 ):
2149 2152 args = tr.hookargs.copy()
2150 2153 args.update(bookmarks.preparehookargs(name, old, new))
2151 2154 repo.hook(
2152 2155 b'pretxnclose-bookmark',
2153 2156 throw=True,
2154 2157 **pycompat.strkwargs(args)
2155 2158 )
2156 2159 if hook.hashook(repo.ui, b'pretxnclose-phase'):
2157 2160 cl = repo.unfiltered().changelog
2158 2161 for rev, (old, new) in tr.changes[b'phases'].items():
2159 2162 args = tr.hookargs.copy()
2160 2163 node = hex(cl.node(rev))
2161 2164 args.update(phases.preparehookargs(node, old, new))
2162 2165 repo.hook(
2163 2166 b'pretxnclose-phase',
2164 2167 throw=True,
2165 2168 **pycompat.strkwargs(args)
2166 2169 )
2167 2170
2168 2171 repo.hook(
2169 2172 b'pretxnclose', throw=True, **pycompat.strkwargs(tr.hookargs)
2170 2173 )
2171 2174
2172 2175 def releasefn(tr, success):
2173 2176 repo = reporef()
2174 2177 if repo is None:
2175 2178 # If the repo has been GC'd (and this release function is being
2176 2179 # called from transaction.__del__), there's not much we can do,
2177 2180 # so just leave the unfinished transaction there and let the
2178 2181 # user run `hg recover`.
2179 2182 return
2180 2183 if success:
2181 2184 # this should be explicitly invoked here, because
2182 2185 # in-memory changes aren't written out at closing
2183 2186 # transaction, if tr.addfilegenerator (via
2184 2187 # dirstate.write or so) isn't invoked while
2185 2188 # transaction running
2186 2189 repo.dirstate.write(None)
2187 2190 else:
2188 2191 # discard all changes (including ones already written
2189 2192 # out) in this transaction
2190 2193 narrowspec.restorebackup(self, b'journal.narrowspec')
2191 2194 narrowspec.restorewcbackup(self, b'journal.narrowspec.dirstate')
2192 2195 repo.dirstate.restorebackup(None, b'journal.dirstate')
2193 2196
2194 2197 repo.invalidate(clearfilecache=True)
2195 2198
2196 2199 tr = transaction.transaction(
2197 2200 rp,
2198 2201 self.svfs,
2199 2202 vfsmap,
2200 2203 b"journal",
2201 2204 b"undo",
2202 2205 aftertrans(renames),
2203 2206 self.store.createmode,
2204 2207 validator=validate,
2205 2208 releasefn=releasefn,
2206 2209 checkambigfiles=_cachedfiles,
2207 2210 name=desc,
2208 2211 )
2209 2212 tr.changes[b'origrepolen'] = len(self)
2210 2213 tr.changes[b'obsmarkers'] = set()
2211 2214 tr.changes[b'phases'] = {}
2212 2215 tr.changes[b'bookmarks'] = {}
2213 2216
2214 2217 tr.hookargs[b'txnid'] = txnid
2215 2218 tr.hookargs[b'txnname'] = desc
2216 2219 # note: writing the fncache only during finalize mean that the file is
2217 2220 # outdated when running hooks. As fncache is used for streaming clone,
2218 2221 # this is not expected to break anything that happen during the hooks.
2219 2222 tr.addfinalize(b'flush-fncache', self.store.write)
2220 2223
2221 2224 def txnclosehook(tr2):
2222 2225 """To be run if transaction is successful, will schedule a hook run
2223 2226 """
2224 2227 # Don't reference tr2 in hook() so we don't hold a reference.
2225 2228 # This reduces memory consumption when there are multiple
2226 2229 # transactions per lock. This can likely go away if issue5045
2227 2230 # fixes the function accumulation.
2228 2231 hookargs = tr2.hookargs
2229 2232
2230 2233 def hookfunc(unused_success):
2231 2234 repo = reporef()
2232 2235 if hook.hashook(repo.ui, b'txnclose-bookmark'):
2233 2236 bmchanges = sorted(tr.changes[b'bookmarks'].items())
2234 2237 for name, (old, new) in bmchanges:
2235 2238 args = tr.hookargs.copy()
2236 2239 args.update(bookmarks.preparehookargs(name, old, new))
2237 2240 repo.hook(
2238 2241 b'txnclose-bookmark',
2239 2242 throw=False,
2240 2243 **pycompat.strkwargs(args)
2241 2244 )
2242 2245
2243 2246 if hook.hashook(repo.ui, b'txnclose-phase'):
2244 2247 cl = repo.unfiltered().changelog
2245 2248 phasemv = sorted(tr.changes[b'phases'].items())
2246 2249 for rev, (old, new) in phasemv:
2247 2250 args = tr.hookargs.copy()
2248 2251 node = hex(cl.node(rev))
2249 2252 args.update(phases.preparehookargs(node, old, new))
2250 2253 repo.hook(
2251 2254 b'txnclose-phase',
2252 2255 throw=False,
2253 2256 **pycompat.strkwargs(args)
2254 2257 )
2255 2258
2256 2259 repo.hook(
2257 2260 b'txnclose', throw=False, **pycompat.strkwargs(hookargs)
2258 2261 )
2259 2262
2260 2263 reporef()._afterlock(hookfunc)
2261 2264
2262 2265 tr.addfinalize(b'txnclose-hook', txnclosehook)
2263 2266 # Include a leading "-" to make it happen before the transaction summary
2264 2267 # reports registered via scmutil.registersummarycallback() whose names
2265 2268 # are 00-txnreport etc. That way, the caches will be warm when the
2266 2269 # callbacks run.
2267 2270 tr.addpostclose(b'-warm-cache', self._buildcacheupdater(tr))
2268 2271
2269 2272 def txnaborthook(tr2):
2270 2273 """To be run if transaction is aborted
2271 2274 """
2272 2275 reporef().hook(
2273 2276 b'txnabort', throw=False, **pycompat.strkwargs(tr2.hookargs)
2274 2277 )
2275 2278
2276 2279 tr.addabort(b'txnabort-hook', txnaborthook)
2277 2280 # avoid eager cache invalidation. in-memory data should be identical
2278 2281 # to stored data if transaction has no error.
2279 2282 tr.addpostclose(b'refresh-filecachestats', self._refreshfilecachestats)
2280 2283 self._transref = weakref.ref(tr)
2281 2284 scmutil.registersummarycallback(self, tr, desc)
2282 2285 return tr
2283 2286
2284 2287 def _journalfiles(self):
2285 2288 return (
2286 2289 (self.svfs, b'journal'),
2287 2290 (self.svfs, b'journal.narrowspec'),
2288 2291 (self.vfs, b'journal.narrowspec.dirstate'),
2289 2292 (self.vfs, b'journal.dirstate'),
2290 2293 (self.vfs, b'journal.branch'),
2291 2294 (self.vfs, b'journal.desc'),
2292 2295 (bookmarks.bookmarksvfs(self), b'journal.bookmarks'),
2293 2296 (self.svfs, b'journal.phaseroots'),
2294 2297 )
2295 2298
2296 2299 def undofiles(self):
2297 2300 return [(vfs, undoname(x)) for vfs, x in self._journalfiles()]
2298 2301
2299 2302 @unfilteredmethod
2300 2303 def _writejournal(self, desc):
2301 2304 self.dirstate.savebackup(None, b'journal.dirstate')
2302 2305 narrowspec.savewcbackup(self, b'journal.narrowspec.dirstate')
2303 2306 narrowspec.savebackup(self, b'journal.narrowspec')
2304 2307 self.vfs.write(
2305 2308 b"journal.branch", encoding.fromlocal(self.dirstate.branch())
2306 2309 )
2307 2310 self.vfs.write(b"journal.desc", b"%d\n%s\n" % (len(self), desc))
2308 2311 bookmarksvfs = bookmarks.bookmarksvfs(self)
2309 2312 bookmarksvfs.write(
2310 2313 b"journal.bookmarks", bookmarksvfs.tryread(b"bookmarks")
2311 2314 )
2312 2315 self.svfs.write(b"journal.phaseroots", self.svfs.tryread(b"phaseroots"))
2313 2316
2314 2317 def recover(self):
2315 2318 with self.lock():
2316 2319 if self.svfs.exists(b"journal"):
2317 2320 self.ui.status(_(b"rolling back interrupted transaction\n"))
2318 2321 vfsmap = {
2319 2322 b'': self.svfs,
2320 2323 b'plain': self.vfs,
2321 2324 }
2322 2325 transaction.rollback(
2323 2326 self.svfs,
2324 2327 vfsmap,
2325 2328 b"journal",
2326 2329 self.ui.warn,
2327 2330 checkambigfiles=_cachedfiles,
2328 2331 )
2329 2332 self.invalidate()
2330 2333 return True
2331 2334 else:
2332 2335 self.ui.warn(_(b"no interrupted transaction available\n"))
2333 2336 return False
2334 2337
2335 2338 def rollback(self, dryrun=False, force=False):
2336 2339 wlock = lock = dsguard = None
2337 2340 try:
2338 2341 wlock = self.wlock()
2339 2342 lock = self.lock()
2340 2343 if self.svfs.exists(b"undo"):
2341 2344 dsguard = dirstateguard.dirstateguard(self, b'rollback')
2342 2345
2343 2346 return self._rollback(dryrun, force, dsguard)
2344 2347 else:
2345 2348 self.ui.warn(_(b"no rollback information available\n"))
2346 2349 return 1
2347 2350 finally:
2348 2351 release(dsguard, lock, wlock)
2349 2352
2350 2353 @unfilteredmethod # Until we get smarter cache management
2351 2354 def _rollback(self, dryrun, force, dsguard):
2352 2355 ui = self.ui
2353 2356 try:
2354 2357 args = self.vfs.read(b'undo.desc').splitlines()
2355 2358 (oldlen, desc, detail) = (int(args[0]), args[1], None)
2356 2359 if len(args) >= 3:
2357 2360 detail = args[2]
2358 2361 oldtip = oldlen - 1
2359 2362
2360 2363 if detail and ui.verbose:
2361 2364 msg = _(
2362 2365 b'repository tip rolled back to revision %d'
2363 2366 b' (undo %s: %s)\n'
2364 2367 ) % (oldtip, desc, detail)
2365 2368 else:
2366 2369 msg = _(
2367 2370 b'repository tip rolled back to revision %d (undo %s)\n'
2368 2371 ) % (oldtip, desc)
2369 2372 except IOError:
2370 2373 msg = _(b'rolling back unknown transaction\n')
2371 2374 desc = None
2372 2375
2373 2376 if not force and self[b'.'] != self[b'tip'] and desc == b'commit':
2374 2377 raise error.Abort(
2375 2378 _(
2376 2379 b'rollback of last commit while not checked out '
2377 2380 b'may lose data'
2378 2381 ),
2379 2382 hint=_(b'use -f to force'),
2380 2383 )
2381 2384
2382 2385 ui.status(msg)
2383 2386 if dryrun:
2384 2387 return 0
2385 2388
2386 2389 parents = self.dirstate.parents()
2387 2390 self.destroying()
2388 2391 vfsmap = {b'plain': self.vfs, b'': self.svfs}
2389 2392 transaction.rollback(
2390 2393 self.svfs, vfsmap, b'undo', ui.warn, checkambigfiles=_cachedfiles
2391 2394 )
2392 2395 bookmarksvfs = bookmarks.bookmarksvfs(self)
2393 2396 if bookmarksvfs.exists(b'undo.bookmarks'):
2394 2397 bookmarksvfs.rename(
2395 2398 b'undo.bookmarks', b'bookmarks', checkambig=True
2396 2399 )
2397 2400 if self.svfs.exists(b'undo.phaseroots'):
2398 2401 self.svfs.rename(b'undo.phaseroots', b'phaseroots', checkambig=True)
2399 2402 self.invalidate()
2400 2403
2401 2404 has_node = self.changelog.index.has_node
2402 2405 parentgone = any(not has_node(p) for p in parents)
2403 2406 if parentgone:
2404 2407 # prevent dirstateguard from overwriting already restored one
2405 2408 dsguard.close()
2406 2409
2407 2410 narrowspec.restorebackup(self, b'undo.narrowspec')
2408 2411 narrowspec.restorewcbackup(self, b'undo.narrowspec.dirstate')
2409 2412 self.dirstate.restorebackup(None, b'undo.dirstate')
2410 2413 try:
2411 2414 branch = self.vfs.read(b'undo.branch')
2412 2415 self.dirstate.setbranch(encoding.tolocal(branch))
2413 2416 except IOError:
2414 2417 ui.warn(
2415 2418 _(
2416 2419 b'named branch could not be reset: '
2417 2420 b'current branch is still \'%s\'\n'
2418 2421 )
2419 2422 % self.dirstate.branch()
2420 2423 )
2421 2424
2422 2425 parents = tuple([p.rev() for p in self[None].parents()])
2423 2426 if len(parents) > 1:
2424 2427 ui.status(
2425 2428 _(
2426 2429 b'working directory now based on '
2427 2430 b'revisions %d and %d\n'
2428 2431 )
2429 2432 % parents
2430 2433 )
2431 2434 else:
2432 2435 ui.status(
2433 2436 _(b'working directory now based on revision %d\n') % parents
2434 2437 )
2435 2438 mergemod.mergestate.clean(self, self[b'.'].node())
2436 2439
2437 2440 # TODO: if we know which new heads may result from this rollback, pass
2438 2441 # them to destroy(), which will prevent the branchhead cache from being
2439 2442 # invalidated.
2440 2443 self.destroyed()
2441 2444 return 0
2442 2445
2443 2446 def _buildcacheupdater(self, newtransaction):
2444 2447 """called during transaction to build the callback updating cache
2445 2448
2446 2449 Lives on the repository to help extension who might want to augment
2447 2450 this logic. For this purpose, the created transaction is passed to the
2448 2451 method.
2449 2452 """
2450 2453 # we must avoid cyclic reference between repo and transaction.
2451 2454 reporef = weakref.ref(self)
2452 2455
2453 2456 def updater(tr):
2454 2457 repo = reporef()
2455 2458 repo.updatecaches(tr)
2456 2459
2457 2460 return updater
2458 2461
2459 2462 @unfilteredmethod
2460 2463 def updatecaches(self, tr=None, full=False):
2461 2464 """warm appropriate caches
2462 2465
2463 2466 If this function is called after a transaction closed. The transaction
2464 2467 will be available in the 'tr' argument. This can be used to selectively
2465 2468 update caches relevant to the changes in that transaction.
2466 2469
2467 2470 If 'full' is set, make sure all caches the function knows about have
2468 2471 up-to-date data. Even the ones usually loaded more lazily.
2469 2472 """
2470 2473 if tr is not None and tr.hookargs.get(b'source') == b'strip':
2471 2474 # During strip, many caches are invalid but
2472 2475 # later call to `destroyed` will refresh them.
2473 2476 return
2474 2477
2475 2478 if tr is None or tr.changes[b'origrepolen'] < len(self):
2476 2479 # accessing the 'ser ved' branchmap should refresh all the others,
2477 2480 self.ui.debug(b'updating the branch cache\n')
2478 2481 self.filtered(b'served').branchmap()
2479 2482 self.filtered(b'served.hidden').branchmap()
2480 2483
2481 2484 if full:
2482 2485 unfi = self.unfiltered()
2483 2486 rbc = unfi.revbranchcache()
2484 2487 for r in unfi.changelog:
2485 2488 rbc.branchinfo(r)
2486 2489 rbc.write()
2487 2490
2488 2491 # ensure the working copy parents are in the manifestfulltextcache
2489 2492 for ctx in self[b'.'].parents():
2490 2493 ctx.manifest() # accessing the manifest is enough
2491 2494
2492 2495 # accessing fnode cache warms the cache
2493 2496 tagsmod.fnoderevs(self.ui, unfi, unfi.changelog.revs())
2494 2497 # accessing tags warm the cache
2495 2498 self.tags()
2496 2499 self.filtered(b'served').tags()
2497 2500
2498 2501 # The `full` arg is documented as updating even the lazily-loaded
2499 2502 # caches immediately, so we're forcing a write to cause these caches
2500 2503 # to be warmed up even if they haven't explicitly been requested
2501 2504 # yet (if they've never been used by hg, they won't ever have been
2502 2505 # written, even if they're a subset of another kind of cache that
2503 2506 # *has* been used).
2504 2507 for filt in repoview.filtertable.keys():
2505 2508 filtered = self.filtered(filt)
2506 2509 filtered.branchmap().write(filtered)
2507 2510
2508 2511 def invalidatecaches(self):
2509 2512
2510 2513 if '_tagscache' in vars(self):
2511 2514 # can't use delattr on proxy
2512 2515 del self.__dict__['_tagscache']
2513 2516
2514 2517 self._branchcaches.clear()
2515 2518 self.invalidatevolatilesets()
2516 2519 self._sparsesignaturecache.clear()
2517 2520
2518 2521 def invalidatevolatilesets(self):
2519 2522 self.filteredrevcache.clear()
2520 2523 obsolete.clearobscaches(self)
2521 2524 self._quick_access_changeid_invalidate()
2522 2525
2523 2526 def invalidatedirstate(self):
2524 2527 '''Invalidates the dirstate, causing the next call to dirstate
2525 2528 to check if it was modified since the last time it was read,
2526 2529 rereading it if it has.
2527 2530
2528 2531 This is different to dirstate.invalidate() that it doesn't always
2529 2532 rereads the dirstate. Use dirstate.invalidate() if you want to
2530 2533 explicitly read the dirstate again (i.e. restoring it to a previous
2531 2534 known good state).'''
2532 2535 if hasunfilteredcache(self, 'dirstate'):
2533 2536 for k in self.dirstate._filecache:
2534 2537 try:
2535 2538 delattr(self.dirstate, k)
2536 2539 except AttributeError:
2537 2540 pass
2538 2541 delattr(self.unfiltered(), 'dirstate')
2539 2542
2540 2543 def invalidate(self, clearfilecache=False):
2541 2544 '''Invalidates both store and non-store parts other than dirstate
2542 2545
2543 2546 If a transaction is running, invalidation of store is omitted,
2544 2547 because discarding in-memory changes might cause inconsistency
2545 2548 (e.g. incomplete fncache causes unintentional failure, but
2546 2549 redundant one doesn't).
2547 2550 '''
2548 2551 unfiltered = self.unfiltered() # all file caches are stored unfiltered
2549 2552 for k in list(self._filecache.keys()):
2550 2553 # dirstate is invalidated separately in invalidatedirstate()
2551 2554 if k == b'dirstate':
2552 2555 continue
2553 2556 if (
2554 2557 k == b'changelog'
2555 2558 and self.currenttransaction()
2556 2559 and self.changelog._delayed
2557 2560 ):
2558 2561 # The changelog object may store unwritten revisions. We don't
2559 2562 # want to lose them.
2560 2563 # TODO: Solve the problem instead of working around it.
2561 2564 continue
2562 2565
2563 2566 if clearfilecache:
2564 2567 del self._filecache[k]
2565 2568 try:
2566 2569 delattr(unfiltered, k)
2567 2570 except AttributeError:
2568 2571 pass
2569 2572 self.invalidatecaches()
2570 2573 if not self.currenttransaction():
2571 2574 # TODO: Changing contents of store outside transaction
2572 2575 # causes inconsistency. We should make in-memory store
2573 2576 # changes detectable, and abort if changed.
2574 2577 self.store.invalidatecaches()
2575 2578
2576 2579 def invalidateall(self):
2577 2580 '''Fully invalidates both store and non-store parts, causing the
2578 2581 subsequent operation to reread any outside changes.'''
2579 2582 # extension should hook this to invalidate its caches
2580 2583 self.invalidate()
2581 2584 self.invalidatedirstate()
2582 2585
2583 2586 @unfilteredmethod
2584 2587 def _refreshfilecachestats(self, tr):
2585 2588 """Reload stats of cached files so that they are flagged as valid"""
2586 2589 for k, ce in self._filecache.items():
2587 2590 k = pycompat.sysstr(k)
2588 2591 if k == 'dirstate' or k not in self.__dict__:
2589 2592 continue
2590 2593 ce.refresh()
2591 2594
2592 2595 def _lock(
2593 2596 self,
2594 2597 vfs,
2595 2598 lockname,
2596 2599 wait,
2597 2600 releasefn,
2598 2601 acquirefn,
2599 2602 desc,
2600 2603 inheritchecker=None,
2601 2604 parentenvvar=None,
2602 2605 ):
2603 2606 parentlock = None
2604 2607 # the contents of parentenvvar are used by the underlying lock to
2605 2608 # determine whether it can be inherited
2606 2609 if parentenvvar is not None:
2607 2610 parentlock = encoding.environ.get(parentenvvar)
2608 2611
2609 2612 timeout = 0
2610 2613 warntimeout = 0
2611 2614 if wait:
2612 2615 timeout = self.ui.configint(b"ui", b"timeout")
2613 2616 warntimeout = self.ui.configint(b"ui", b"timeout.warn")
2614 2617 # internal config: ui.signal-safe-lock
2615 2618 signalsafe = self.ui.configbool(b'ui', b'signal-safe-lock')
2616 2619
2617 2620 l = lockmod.trylock(
2618 2621 self.ui,
2619 2622 vfs,
2620 2623 lockname,
2621 2624 timeout,
2622 2625 warntimeout,
2623 2626 releasefn=releasefn,
2624 2627 acquirefn=acquirefn,
2625 2628 desc=desc,
2626 2629 inheritchecker=inheritchecker,
2627 2630 parentlock=parentlock,
2628 2631 signalsafe=signalsafe,
2629 2632 )
2630 2633 return l
2631 2634
2632 2635 def _afterlock(self, callback):
2633 2636 """add a callback to be run when the repository is fully unlocked
2634 2637
2635 2638 The callback will be executed when the outermost lock is released
2636 2639 (with wlock being higher level than 'lock')."""
2637 2640 for ref in (self._wlockref, self._lockref):
2638 2641 l = ref and ref()
2639 2642 if l and l.held:
2640 2643 l.postrelease.append(callback)
2641 2644 break
2642 2645 else: # no lock have been found.
2643 2646 callback(True)
2644 2647
2645 2648 def lock(self, wait=True):
2646 2649 '''Lock the repository store (.hg/store) and return a weak reference
2647 2650 to the lock. Use this before modifying the store (e.g. committing or
2648 2651 stripping). If you are opening a transaction, get a lock as well.)
2649 2652
2650 2653 If both 'lock' and 'wlock' must be acquired, ensure you always acquires
2651 2654 'wlock' first to avoid a dead-lock hazard.'''
2652 2655 l = self._currentlock(self._lockref)
2653 2656 if l is not None:
2654 2657 l.lock()
2655 2658 return l
2656 2659
2657 2660 l = self._lock(
2658 2661 vfs=self.svfs,
2659 2662 lockname=b"lock",
2660 2663 wait=wait,
2661 2664 releasefn=None,
2662 2665 acquirefn=self.invalidate,
2663 2666 desc=_(b'repository %s') % self.origroot,
2664 2667 )
2665 2668 self._lockref = weakref.ref(l)
2666 2669 return l
2667 2670
2668 2671 def _wlockchecktransaction(self):
2669 2672 if self.currenttransaction() is not None:
2670 2673 raise error.LockInheritanceContractViolation(
2671 2674 b'wlock cannot be inherited in the middle of a transaction'
2672 2675 )
2673 2676
2674 2677 def wlock(self, wait=True):
2675 2678 '''Lock the non-store parts of the repository (everything under
2676 2679 .hg except .hg/store) and return a weak reference to the lock.
2677 2680
2678 2681 Use this before modifying files in .hg.
2679 2682
2680 2683 If both 'lock' and 'wlock' must be acquired, ensure you always acquires
2681 2684 'wlock' first to avoid a dead-lock hazard.'''
2682 2685 l = self._wlockref and self._wlockref()
2683 2686 if l is not None and l.held:
2684 2687 l.lock()
2685 2688 return l
2686 2689
2687 2690 # We do not need to check for non-waiting lock acquisition. Such
2688 2691 # acquisition would not cause dead-lock as they would just fail.
2689 2692 if wait and (
2690 2693 self.ui.configbool(b'devel', b'all-warnings')
2691 2694 or self.ui.configbool(b'devel', b'check-locks')
2692 2695 ):
2693 2696 if self._currentlock(self._lockref) is not None:
2694 2697 self.ui.develwarn(b'"wlock" acquired after "lock"')
2695 2698
2696 2699 def unlock():
2697 2700 if self.dirstate.pendingparentchange():
2698 2701 self.dirstate.invalidate()
2699 2702 else:
2700 2703 self.dirstate.write(None)
2701 2704
2702 2705 self._filecache[b'dirstate'].refresh()
2703 2706
2704 2707 l = self._lock(
2705 2708 self.vfs,
2706 2709 b"wlock",
2707 2710 wait,
2708 2711 unlock,
2709 2712 self.invalidatedirstate,
2710 2713 _(b'working directory of %s') % self.origroot,
2711 2714 inheritchecker=self._wlockchecktransaction,
2712 2715 parentenvvar=b'HG_WLOCK_LOCKER',
2713 2716 )
2714 2717 self._wlockref = weakref.ref(l)
2715 2718 return l
2716 2719
2717 2720 def _currentlock(self, lockref):
2718 2721 """Returns the lock if it's held, or None if it's not."""
2719 2722 if lockref is None:
2720 2723 return None
2721 2724 l = lockref()
2722 2725 if l is None or not l.held:
2723 2726 return None
2724 2727 return l
2725 2728
2726 2729 def currentwlock(self):
2727 2730 """Returns the wlock if it's held, or None if it's not."""
2728 2731 return self._currentlock(self._wlockref)
2729 2732
2730 2733 def _filecommit(
2731 2734 self,
2732 2735 fctx,
2733 2736 manifest1,
2734 2737 manifest2,
2735 2738 linkrev,
2736 2739 tr,
2737 2740 changelist,
2738 2741 includecopymeta,
2739 2742 ):
2740 2743 """
2741 2744 commit an individual file as part of a larger transaction
2742 2745 """
2743 2746
2744 2747 fname = fctx.path()
2745 2748 fparent1 = manifest1.get(fname, nullid)
2746 2749 fparent2 = manifest2.get(fname, nullid)
2747 2750 if isinstance(fctx, context.filectx):
2748 2751 node = fctx.filenode()
2749 2752 if node in [fparent1, fparent2]:
2750 2753 self.ui.debug(b'reusing %s filelog entry\n' % fname)
2751 2754 if (
2752 2755 fparent1 != nullid
2753 2756 and manifest1.flags(fname) != fctx.flags()
2754 2757 ) or (
2755 2758 fparent2 != nullid
2756 2759 and manifest2.flags(fname) != fctx.flags()
2757 2760 ):
2758 2761 changelist.append(fname)
2759 2762 return node
2760 2763
2761 2764 flog = self.file(fname)
2762 2765 meta = {}
2763 2766 cfname = fctx.copysource()
2764 2767 if cfname and cfname != fname:
2765 2768 # Mark the new revision of this file as a copy of another
2766 2769 # file. This copy data will effectively act as a parent
2767 2770 # of this new revision. If this is a merge, the first
2768 2771 # parent will be the nullid (meaning "look up the copy data")
2769 2772 # and the second one will be the other parent. For example:
2770 2773 #
2771 2774 # 0 --- 1 --- 3 rev1 changes file foo
2772 2775 # \ / rev2 renames foo to bar and changes it
2773 2776 # \- 2 -/ rev3 should have bar with all changes and
2774 2777 # should record that bar descends from
2775 2778 # bar in rev2 and foo in rev1
2776 2779 #
2777 2780 # this allows this merge to succeed:
2778 2781 #
2779 2782 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
2780 2783 # \ / merging rev3 and rev4 should use bar@rev2
2781 2784 # \- 2 --- 4 as the merge base
2782 2785 #
2783 2786
2784 2787 cnode = manifest1.get(cfname)
2785 2788 newfparent = fparent2
2786 2789
2787 2790 if manifest2: # branch merge
2788 2791 if fparent2 == nullid or cnode is None: # copied on remote side
2789 2792 if cfname in manifest2:
2790 2793 cnode = manifest2[cfname]
2791 2794 newfparent = fparent1
2792 2795
2793 2796 # Here, we used to search backwards through history to try to find
2794 2797 # where the file copy came from if the source of a copy was not in
2795 2798 # the parent directory. However, this doesn't actually make sense to
2796 2799 # do (what does a copy from something not in your working copy even
2797 2800 # mean?) and it causes bugs (eg, issue4476). Instead, we will warn
2798 2801 # the user that copy information was dropped, so if they didn't
2799 2802 # expect this outcome it can be fixed, but this is the correct
2800 2803 # behavior in this circumstance.
2801 2804
2802 2805 if cnode:
2803 2806 self.ui.debug(
2804 2807 b" %s: copy %s:%s\n" % (fname, cfname, hex(cnode))
2805 2808 )
2806 2809 if includecopymeta:
2807 2810 meta[b"copy"] = cfname
2808 2811 meta[b"copyrev"] = hex(cnode)
2809 2812 fparent1, fparent2 = nullid, newfparent
2810 2813 else:
2811 2814 self.ui.warn(
2812 2815 _(
2813 2816 b"warning: can't find ancestor for '%s' "
2814 2817 b"copied from '%s'!\n"
2815 2818 )
2816 2819 % (fname, cfname)
2817 2820 )
2818 2821
2819 2822 elif fparent1 == nullid:
2820 2823 fparent1, fparent2 = fparent2, nullid
2821 2824 elif fparent2 != nullid:
2822 2825 # is one parent an ancestor of the other?
2823 2826 fparentancestors = flog.commonancestorsheads(fparent1, fparent2)
2824 2827 if fparent1 in fparentancestors:
2825 2828 fparent1, fparent2 = fparent2, nullid
2826 2829 elif fparent2 in fparentancestors:
2827 2830 fparent2 = nullid
2828 2831
2829 2832 # is the file changed?
2830 2833 text = fctx.data()
2831 2834 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
2832 2835 changelist.append(fname)
2833 2836 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
2834 2837 # are just the flags changed during merge?
2835 2838 elif fname in manifest1 and manifest1.flags(fname) != fctx.flags():
2836 2839 changelist.append(fname)
2837 2840
2838 2841 return fparent1
2839 2842
2840 2843 def checkcommitpatterns(self, wctx, match, status, fail):
2841 2844 """check for commit arguments that aren't committable"""
2842 2845 if match.isexact() or match.prefix():
2843 2846 matched = set(status.modified + status.added + status.removed)
2844 2847
2845 2848 for f in match.files():
2846 2849 f = self.dirstate.normalize(f)
2847 2850 if f == b'.' or f in matched or f in wctx.substate:
2848 2851 continue
2849 2852 if f in status.deleted:
2850 2853 fail(f, _(b'file not found!'))
2851 2854 # Is it a directory that exists or used to exist?
2852 2855 if self.wvfs.isdir(f) or wctx.p1().hasdir(f):
2853 2856 d = f + b'/'
2854 2857 for mf in matched:
2855 2858 if mf.startswith(d):
2856 2859 break
2857 2860 else:
2858 2861 fail(f, _(b"no match under directory!"))
2859 2862 elif f not in self.dirstate:
2860 2863 fail(f, _(b"file not tracked!"))
2861 2864
2862 2865 @unfilteredmethod
2863 2866 def commit(
2864 2867 self,
2865 2868 text=b"",
2866 2869 user=None,
2867 2870 date=None,
2868 2871 match=None,
2869 2872 force=False,
2870 2873 editor=None,
2871 2874 extra=None,
2872 2875 ):
2873 2876 """Add a new revision to current repository.
2874 2877
2875 2878 Revision information is gathered from the working directory,
2876 2879 match can be used to filter the committed files. If editor is
2877 2880 supplied, it is called to get a commit message.
2878 2881 """
2879 2882 if extra is None:
2880 2883 extra = {}
2881 2884
2882 2885 def fail(f, msg):
2883 2886 raise error.Abort(b'%s: %s' % (f, msg))
2884 2887
2885 2888 if not match:
2886 2889 match = matchmod.always()
2887 2890
2888 2891 if not force:
2889 2892 match.bad = fail
2890 2893
2891 2894 # lock() for recent changelog (see issue4368)
2892 2895 with self.wlock(), self.lock():
2893 2896 wctx = self[None]
2894 2897 merge = len(wctx.parents()) > 1
2895 2898
2896 2899 if not force and merge and not match.always():
2897 2900 raise error.Abort(
2898 2901 _(
2899 2902 b'cannot partially commit a merge '
2900 2903 b'(do not specify files or patterns)'
2901 2904 )
2902 2905 )
2903 2906
2904 2907 status = self.status(match=match, clean=force)
2905 2908 if force:
2906 2909 status.modified.extend(
2907 2910 status.clean
2908 2911 ) # mq may commit clean files
2909 2912
2910 2913 # check subrepos
2911 2914 subs, commitsubs, newstate = subrepoutil.precommit(
2912 2915 self.ui, wctx, status, match, force=force
2913 2916 )
2914 2917
2915 2918 # make sure all explicit patterns are matched
2916 2919 if not force:
2917 2920 self.checkcommitpatterns(wctx, match, status, fail)
2918 2921
2919 2922 cctx = context.workingcommitctx(
2920 2923 self, status, text, user, date, extra
2921 2924 )
2922 2925
2923 2926 # internal config: ui.allowemptycommit
2924 2927 allowemptycommit = (
2925 2928 wctx.branch() != wctx.p1().branch()
2926 2929 or extra.get(b'close')
2927 2930 or merge
2928 2931 or cctx.files()
2929 2932 or self.ui.configbool(b'ui', b'allowemptycommit')
2930 2933 )
2931 2934 if not allowemptycommit:
2932 2935 return None
2933 2936
2934 2937 if merge and cctx.deleted():
2935 2938 raise error.Abort(_(b"cannot commit merge with missing files"))
2936 2939
2937 2940 ms = mergemod.mergestate.read(self)
2938 2941 mergeutil.checkunresolved(ms)
2939 2942
2940 2943 if editor:
2941 2944 cctx._text = editor(self, cctx, subs)
2942 2945 edited = text != cctx._text
2943 2946
2944 2947 # Save commit message in case this transaction gets rolled back
2945 2948 # (e.g. by a pretxncommit hook). Leave the content alone on
2946 2949 # the assumption that the user will use the same editor again.
2947 2950 msgfn = self.savecommitmessage(cctx._text)
2948 2951
2949 2952 # commit subs and write new state
2950 2953 if subs:
2951 2954 uipathfn = scmutil.getuipathfn(self)
2952 2955 for s in sorted(commitsubs):
2953 2956 sub = wctx.sub(s)
2954 2957 self.ui.status(
2955 2958 _(b'committing subrepository %s\n')
2956 2959 % uipathfn(subrepoutil.subrelpath(sub))
2957 2960 )
2958 2961 sr = sub.commit(cctx._text, user, date)
2959 2962 newstate[s] = (newstate[s][0], sr)
2960 2963 subrepoutil.writestate(self, newstate)
2961 2964
2962 2965 p1, p2 = self.dirstate.parents()
2963 2966 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or b'')
2964 2967 try:
2965 2968 self.hook(
2966 2969 b"precommit", throw=True, parent1=hookp1, parent2=hookp2
2967 2970 )
2968 2971 with self.transaction(b'commit'):
2969 2972 ret = self.commitctx(cctx, True)
2970 2973 # update bookmarks, dirstate and mergestate
2971 2974 bookmarks.update(self, [p1, p2], ret)
2972 2975 cctx.markcommitted(ret)
2973 2976 ms.reset()
2974 2977 except: # re-raises
2975 2978 if edited:
2976 2979 self.ui.write(
2977 2980 _(b'note: commit message saved in %s\n') % msgfn
2978 2981 )
2979 2982 raise
2980 2983
2981 2984 def commithook(unused_success):
2982 2985 # hack for command that use a temporary commit (eg: histedit)
2983 2986 # temporary commit got stripped before hook release
2984 2987 if self.changelog.hasnode(ret):
2985 2988 self.hook(
2986 2989 b"commit", node=hex(ret), parent1=hookp1, parent2=hookp2
2987 2990 )
2988 2991
2989 2992 self._afterlock(commithook)
2990 2993 return ret
2991 2994
2992 2995 @unfilteredmethod
2993 2996 def commitctx(self, ctx, error=False, origctx=None):
2994 2997 """Add a new revision to current repository.
2995 2998 Revision information is passed via the context argument.
2996 2999
2997 3000 ctx.files() should list all files involved in this commit, i.e.
2998 3001 modified/added/removed files. On merge, it may be wider than the
2999 3002 ctx.files() to be committed, since any file nodes derived directly
3000 3003 from p1 or p2 are excluded from the committed ctx.files().
3001 3004
3002 3005 origctx is for convert to work around the problem that bug
3003 3006 fixes to the files list in changesets change hashes. For
3004 3007 convert to be the identity, it can pass an origctx and this
3005 3008 function will use the same files list when it makes sense to
3006 3009 do so.
3007 3010 """
3008 3011
3009 3012 p1, p2 = ctx.p1(), ctx.p2()
3010 3013 user = ctx.user()
3011 3014
3012 3015 if self.filecopiesmode == b'changeset-sidedata':
3013 3016 writechangesetcopy = True
3014 3017 writefilecopymeta = True
3015 3018 writecopiesto = None
3016 3019 else:
3017 3020 writecopiesto = self.ui.config(b'experimental', b'copies.write-to')
3018 3021 writefilecopymeta = writecopiesto != b'changeset-only'
3019 3022 writechangesetcopy = writecopiesto in (
3020 3023 b'changeset-only',
3021 3024 b'compatibility',
3022 3025 )
3023 3026 p1copies, p2copies = None, None
3024 3027 if writechangesetcopy:
3025 3028 p1copies = ctx.p1copies()
3026 3029 p2copies = ctx.p2copies()
3027 3030 filesadded, filesremoved = None, None
3028 3031 with self.lock(), self.transaction(b"commit") as tr:
3029 3032 trp = weakref.proxy(tr)
3030 3033
3031 3034 if ctx.manifestnode():
3032 3035 # reuse an existing manifest revision
3033 3036 self.ui.debug(b'reusing known manifest\n')
3034 3037 mn = ctx.manifestnode()
3035 3038 files = ctx.files()
3036 3039 if writechangesetcopy:
3037 3040 filesadded = ctx.filesadded()
3038 3041 filesremoved = ctx.filesremoved()
3039 3042 elif ctx.files():
3040 3043 m1ctx = p1.manifestctx()
3041 3044 m2ctx = p2.manifestctx()
3042 3045 mctx = m1ctx.copy()
3043 3046
3044 3047 m = mctx.read()
3045 3048 m1 = m1ctx.read()
3046 3049 m2 = m2ctx.read()
3047 3050
3048 3051 # check in files
3049 3052 added = []
3050 3053 changed = []
3051 3054 removed = list(ctx.removed())
3052 3055 linkrev = len(self)
3053 3056 self.ui.note(_(b"committing files:\n"))
3054 3057 uipathfn = scmutil.getuipathfn(self)
3055 3058 for f in sorted(ctx.modified() + ctx.added()):
3056 3059 self.ui.note(uipathfn(f) + b"\n")
3057 3060 try:
3058 3061 fctx = ctx[f]
3059 3062 if fctx is None:
3060 3063 removed.append(f)
3061 3064 else:
3062 3065 added.append(f)
3063 3066 m[f] = self._filecommit(
3064 3067 fctx,
3065 3068 m1,
3066 3069 m2,
3067 3070 linkrev,
3068 3071 trp,
3069 3072 changed,
3070 3073 writefilecopymeta,
3071 3074 )
3072 3075 m.setflag(f, fctx.flags())
3073 3076 except OSError:
3074 3077 self.ui.warn(
3075 3078 _(b"trouble committing %s!\n") % uipathfn(f)
3076 3079 )
3077 3080 raise
3078 3081 except IOError as inst:
3079 3082 errcode = getattr(inst, 'errno', errno.ENOENT)
3080 3083 if error or errcode and errcode != errno.ENOENT:
3081 3084 self.ui.warn(
3082 3085 _(b"trouble committing %s!\n") % uipathfn(f)
3083 3086 )
3084 3087 raise
3085 3088
3086 3089 # update manifest
3087 3090 removed = [f for f in removed if f in m1 or f in m2]
3088 3091 drop = sorted([f for f in removed if f in m])
3089 3092 for f in drop:
3090 3093 del m[f]
3091 3094 if p2.rev() != nullrev:
3092 3095
3093 3096 @util.cachefunc
3094 3097 def mas():
3095 3098 p1n = p1.node()
3096 3099 p2n = p2.node()
3097 3100 cahs = self.changelog.commonancestorsheads(p1n, p2n)
3098 3101 if not cahs:
3099 3102 cahs = [nullrev]
3100 3103 return [self[r].manifest() for r in cahs]
3101 3104
3102 3105 def deletionfromparent(f):
3103 3106 # When a file is removed relative to p1 in a merge, this
3104 3107 # function determines whether the absence is due to a
3105 3108 # deletion from a parent, or whether the merge commit
3106 3109 # itself deletes the file. We decide this by doing a
3107 3110 # simplified three way merge of the manifest entry for
3108 3111 # the file. There are two ways we decide the merge
3109 3112 # itself didn't delete a file:
3110 3113 # - neither parent (nor the merge) contain the file
3111 3114 # - exactly one parent contains the file, and that
3112 3115 # parent has the same filelog entry as the merge
3113 3116 # ancestor (or all of them if there two). In other
3114 3117 # words, that parent left the file unchanged while the
3115 3118 # other one deleted it.
3116 3119 # One way to think about this is that deleting a file is
3117 3120 # similar to emptying it, so the list of changed files
3118 3121 # should be similar either way. The computation
3119 3122 # described above is not done directly in _filecommit
3120 3123 # when creating the list of changed files, however
3121 3124 # it does something very similar by comparing filelog
3122 3125 # nodes.
3123 3126 if f in m1:
3124 3127 return f not in m2 and all(
3125 3128 f in ma and ma.find(f) == m1.find(f)
3126 3129 for ma in mas()
3127 3130 )
3128 3131 elif f in m2:
3129 3132 return all(
3130 3133 f in ma and ma.find(f) == m2.find(f)
3131 3134 for ma in mas()
3132 3135 )
3133 3136 else:
3134 3137 return True
3135 3138
3136 3139 removed = [f for f in removed if not deletionfromparent(f)]
3137 3140
3138 3141 files = changed + removed
3139 3142 md = None
3140 3143 if not files:
3141 3144 # if no "files" actually changed in terms of the changelog,
3142 3145 # try hard to detect unmodified manifest entry so that the
3143 3146 # exact same commit can be reproduced later on convert.
3144 3147 md = m1.diff(m, scmutil.matchfiles(self, ctx.files()))
3145 3148 if not files and md:
3146 3149 self.ui.debug(
3147 3150 b'not reusing manifest (no file change in '
3148 3151 b'changelog, but manifest differs)\n'
3149 3152 )
3150 3153 if files or md:
3151 3154 self.ui.note(_(b"committing manifest\n"))
3152 3155 # we're using narrowmatch here since it's already applied at
3153 3156 # other stages (such as dirstate.walk), so we're already
3154 3157 # ignoring things outside of narrowspec in most cases. The
3155 3158 # one case where we might have files outside the narrowspec
3156 3159 # at this point is merges, and we already error out in the
3157 3160 # case where the merge has files outside of the narrowspec,
3158 3161 # so this is safe.
3159 3162 mn = mctx.write(
3160 3163 trp,
3161 3164 linkrev,
3162 3165 p1.manifestnode(),
3163 3166 p2.manifestnode(),
3164 3167 added,
3165 3168 drop,
3166 3169 match=self.narrowmatch(),
3167 3170 )
3168 3171
3169 3172 if writechangesetcopy:
3170 3173 filesadded = [
3171 3174 f for f in changed if not (f in m1 or f in m2)
3172 3175 ]
3173 3176 filesremoved = removed
3174 3177 else:
3175 3178 self.ui.debug(
3176 3179 b'reusing manifest from p1 (listed files '
3177 3180 b'actually unchanged)\n'
3178 3181 )
3179 3182 mn = p1.manifestnode()
3180 3183 else:
3181 3184 self.ui.debug(b'reusing manifest from p1 (no file change)\n')
3182 3185 mn = p1.manifestnode()
3183 3186 files = []
3184 3187
3185 3188 if writecopiesto == b'changeset-only':
3186 3189 # If writing only to changeset extras, use None to indicate that
3187 3190 # no entry should be written. If writing to both, write an empty
3188 3191 # entry to prevent the reader from falling back to reading
3189 3192 # filelogs.
3190 3193 p1copies = p1copies or None
3191 3194 p2copies = p2copies or None
3192 3195 filesadded = filesadded or None
3193 3196 filesremoved = filesremoved or None
3194 3197
3195 3198 if origctx and origctx.manifestnode() == mn:
3196 3199 files = origctx.files()
3197 3200
3198 3201 # update changelog
3199 3202 self.ui.note(_(b"committing changelog\n"))
3200 3203 self.changelog.delayupdate(tr)
3201 3204 n = self.changelog.add(
3202 3205 mn,
3203 3206 files,
3204 3207 ctx.description(),
3205 3208 trp,
3206 3209 p1.node(),
3207 3210 p2.node(),
3208 3211 user,
3209 3212 ctx.date(),
3210 3213 ctx.extra().copy(),
3211 3214 p1copies,
3212 3215 p2copies,
3213 3216 filesadded,
3214 3217 filesremoved,
3215 3218 )
3216 3219 xp1, xp2 = p1.hex(), p2 and p2.hex() or b''
3217 3220 self.hook(
3218 3221 b'pretxncommit',
3219 3222 throw=True,
3220 3223 node=hex(n),
3221 3224 parent1=xp1,
3222 3225 parent2=xp2,
3223 3226 )
3224 3227 # set the new commit is proper phase
3225 3228 targetphase = subrepoutil.newcommitphase(self.ui, ctx)
3226 3229 if targetphase:
3227 3230 # retract boundary do not alter parent changeset.
3228 3231 # if a parent have higher the resulting phase will
3229 3232 # be compliant anyway
3230 3233 #
3231 3234 # if minimal phase was 0 we don't need to retract anything
3232 3235 phases.registernew(self, tr, targetphase, [n])
3233 3236 return n
3234 3237
3235 3238 @unfilteredmethod
3236 3239 def destroying(self):
3237 3240 '''Inform the repository that nodes are about to be destroyed.
3238 3241 Intended for use by strip and rollback, so there's a common
3239 3242 place for anything that has to be done before destroying history.
3240 3243
3241 3244 This is mostly useful for saving state that is in memory and waiting
3242 3245 to be flushed when the current lock is released. Because a call to
3243 3246 destroyed is imminent, the repo will be invalidated causing those
3244 3247 changes to stay in memory (waiting for the next unlock), or vanish
3245 3248 completely.
3246 3249 '''
3247 3250 # When using the same lock to commit and strip, the phasecache is left
3248 3251 # dirty after committing. Then when we strip, the repo is invalidated,
3249 3252 # causing those changes to disappear.
3250 3253 if '_phasecache' in vars(self):
3251 3254 self._phasecache.write()
3252 3255
3253 3256 @unfilteredmethod
3254 3257 def destroyed(self):
3255 3258 '''Inform the repository that nodes have been destroyed.
3256 3259 Intended for use by strip and rollback, so there's a common
3257 3260 place for anything that has to be done after destroying history.
3258 3261 '''
3259 3262 # When one tries to:
3260 3263 # 1) destroy nodes thus calling this method (e.g. strip)
3261 3264 # 2) use phasecache somewhere (e.g. commit)
3262 3265 #
3263 3266 # then 2) will fail because the phasecache contains nodes that were
3264 3267 # removed. We can either remove phasecache from the filecache,
3265 3268 # causing it to reload next time it is accessed, or simply filter
3266 3269 # the removed nodes now and write the updated cache.
3267 3270 self._phasecache.filterunknown(self)
3268 3271 self._phasecache.write()
3269 3272
3270 3273 # refresh all repository caches
3271 3274 self.updatecaches()
3272 3275
3273 3276 # Ensure the persistent tag cache is updated. Doing it now
3274 3277 # means that the tag cache only has to worry about destroyed
3275 3278 # heads immediately after a strip/rollback. That in turn
3276 3279 # guarantees that "cachetip == currenttip" (comparing both rev
3277 3280 # and node) always means no nodes have been added or destroyed.
3278 3281
3279 3282 # XXX this is suboptimal when qrefresh'ing: we strip the current
3280 3283 # head, refresh the tag cache, then immediately add a new head.
3281 3284 # But I think doing it this way is necessary for the "instant
3282 3285 # tag cache retrieval" case to work.
3283 3286 self.invalidate()
3284 3287
3285 3288 def status(
3286 3289 self,
3287 3290 node1=b'.',
3288 3291 node2=None,
3289 3292 match=None,
3290 3293 ignored=False,
3291 3294 clean=False,
3292 3295 unknown=False,
3293 3296 listsubrepos=False,
3294 3297 ):
3295 3298 '''a convenience method that calls node1.status(node2)'''
3296 3299 return self[node1].status(
3297 3300 node2, match, ignored, clean, unknown, listsubrepos
3298 3301 )
3299 3302
3300 3303 def addpostdsstatus(self, ps):
3301 3304 """Add a callback to run within the wlock, at the point at which status
3302 3305 fixups happen.
3303 3306
3304 3307 On status completion, callback(wctx, status) will be called with the
3305 3308 wlock held, unless the dirstate has changed from underneath or the wlock
3306 3309 couldn't be grabbed.
3307 3310
3308 3311 Callbacks should not capture and use a cached copy of the dirstate --
3309 3312 it might change in the meanwhile. Instead, they should access the
3310 3313 dirstate via wctx.repo().dirstate.
3311 3314
3312 3315 This list is emptied out after each status run -- extensions should
3313 3316 make sure it adds to this list each time dirstate.status is called.
3314 3317 Extensions should also make sure they don't call this for statuses
3315 3318 that don't involve the dirstate.
3316 3319 """
3317 3320
3318 3321 # The list is located here for uniqueness reasons -- it is actually
3319 3322 # managed by the workingctx, but that isn't unique per-repo.
3320 3323 self._postdsstatus.append(ps)
3321 3324
3322 3325 def postdsstatus(self):
3323 3326 """Used by workingctx to get the list of post-dirstate-status hooks."""
3324 3327 return self._postdsstatus
3325 3328
3326 3329 def clearpostdsstatus(self):
3327 3330 """Used by workingctx to clear post-dirstate-status hooks."""
3328 3331 del self._postdsstatus[:]
3329 3332
3330 3333 def heads(self, start=None):
3331 3334 if start is None:
3332 3335 cl = self.changelog
3333 3336 headrevs = reversed(cl.headrevs())
3334 3337 return [cl.node(rev) for rev in headrevs]
3335 3338
3336 3339 heads = self.changelog.heads(start)
3337 3340 # sort the output in rev descending order
3338 3341 return sorted(heads, key=self.changelog.rev, reverse=True)
3339 3342
3340 3343 def branchheads(self, branch=None, start=None, closed=False):
3341 3344 '''return a (possibly filtered) list of heads for the given branch
3342 3345
3343 3346 Heads are returned in topological order, from newest to oldest.
3344 3347 If branch is None, use the dirstate branch.
3345 3348 If start is not None, return only heads reachable from start.
3346 3349 If closed is True, return heads that are marked as closed as well.
3347 3350 '''
3348 3351 if branch is None:
3349 3352 branch = self[None].branch()
3350 3353 branches = self.branchmap()
3351 3354 if not branches.hasbranch(branch):
3352 3355 return []
3353 3356 # the cache returns heads ordered lowest to highest
3354 3357 bheads = list(reversed(branches.branchheads(branch, closed=closed)))
3355 3358 if start is not None:
3356 3359 # filter out the heads that cannot be reached from startrev
3357 3360 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
3358 3361 bheads = [h for h in bheads if h in fbheads]
3359 3362 return bheads
3360 3363
3361 3364 def branches(self, nodes):
3362 3365 if not nodes:
3363 3366 nodes = [self.changelog.tip()]
3364 3367 b = []
3365 3368 for n in nodes:
3366 3369 t = n
3367 3370 while True:
3368 3371 p = self.changelog.parents(n)
3369 3372 if p[1] != nullid or p[0] == nullid:
3370 3373 b.append((t, n, p[0], p[1]))
3371 3374 break
3372 3375 n = p[0]
3373 3376 return b
3374 3377
3375 3378 def between(self, pairs):
3376 3379 r = []
3377 3380
3378 3381 for top, bottom in pairs:
3379 3382 n, l, i = top, [], 0
3380 3383 f = 1
3381 3384
3382 3385 while n != bottom and n != nullid:
3383 3386 p = self.changelog.parents(n)[0]
3384 3387 if i == f:
3385 3388 l.append(n)
3386 3389 f = f * 2
3387 3390 n = p
3388 3391 i += 1
3389 3392
3390 3393 r.append(l)
3391 3394
3392 3395 return r
3393 3396
3394 3397 def checkpush(self, pushop):
3395 3398 """Extensions can override this function if additional checks have
3396 3399 to be performed before pushing, or call it if they override push
3397 3400 command.
3398 3401 """
3399 3402
3400 3403 @unfilteredpropertycache
3401 3404 def prepushoutgoinghooks(self):
3402 3405 """Return util.hooks consists of a pushop with repo, remote, outgoing
3403 3406 methods, which are called before pushing changesets.
3404 3407 """
3405 3408 return util.hooks()
3406 3409
3407 3410 def pushkey(self, namespace, key, old, new):
3408 3411 try:
3409 3412 tr = self.currenttransaction()
3410 3413 hookargs = {}
3411 3414 if tr is not None:
3412 3415 hookargs.update(tr.hookargs)
3413 3416 hookargs = pycompat.strkwargs(hookargs)
3414 3417 hookargs['namespace'] = namespace
3415 3418 hookargs['key'] = key
3416 3419 hookargs['old'] = old
3417 3420 hookargs['new'] = new
3418 3421 self.hook(b'prepushkey', throw=True, **hookargs)
3419 3422 except error.HookAbort as exc:
3420 3423 self.ui.write_err(_(b"pushkey-abort: %s\n") % exc)
3421 3424 if exc.hint:
3422 3425 self.ui.write_err(_(b"(%s)\n") % exc.hint)
3423 3426 return False
3424 3427 self.ui.debug(b'pushing key for "%s:%s"\n' % (namespace, key))
3425 3428 ret = pushkey.push(self, namespace, key, old, new)
3426 3429
3427 3430 def runhook(unused_success):
3428 3431 self.hook(
3429 3432 b'pushkey',
3430 3433 namespace=namespace,
3431 3434 key=key,
3432 3435 old=old,
3433 3436 new=new,
3434 3437 ret=ret,
3435 3438 )
3436 3439
3437 3440 self._afterlock(runhook)
3438 3441 return ret
3439 3442
3440 3443 def listkeys(self, namespace):
3441 3444 self.hook(b'prelistkeys', throw=True, namespace=namespace)
3442 3445 self.ui.debug(b'listing keys for "%s"\n' % namespace)
3443 3446 values = pushkey.list(self, namespace)
3444 3447 self.hook(b'listkeys', namespace=namespace, values=values)
3445 3448 return values
3446 3449
3447 3450 def debugwireargs(self, one, two, three=None, four=None, five=None):
3448 3451 '''used to test argument passing over the wire'''
3449 3452 return b"%s %s %s %s %s" % (
3450 3453 one,
3451 3454 two,
3452 3455 pycompat.bytestr(three),
3453 3456 pycompat.bytestr(four),
3454 3457 pycompat.bytestr(five),
3455 3458 )
3456 3459
3457 3460 def savecommitmessage(self, text):
3458 3461 fp = self.vfs(b'last-message.txt', b'wb')
3459 3462 try:
3460 3463 fp.write(text)
3461 3464 finally:
3462 3465 fp.close()
3463 3466 return self.pathto(fp.name[len(self.root) + 1 :])
3464 3467
3465 3468
3466 3469 # used to avoid circular references so destructors work
3467 3470 def aftertrans(files):
3468 3471 renamefiles = [tuple(t) for t in files]
3469 3472
3470 3473 def a():
3471 3474 for vfs, src, dest in renamefiles:
3472 3475 # if src and dest refer to a same file, vfs.rename is a no-op,
3473 3476 # leaving both src and dest on disk. delete dest to make sure
3474 3477 # the rename couldn't be such a no-op.
3475 3478 vfs.tryunlink(dest)
3476 3479 try:
3477 3480 vfs.rename(src, dest)
3478 3481 except OSError: # journal file does not yet exist
3479 3482 pass
3480 3483
3481 3484 return a
3482 3485
3483 3486
3484 3487 def undoname(fn):
3485 3488 base, name = os.path.split(fn)
3486 3489 assert name.startswith(b'journal')
3487 3490 return os.path.join(base, name.replace(b'journal', b'undo', 1))
3488 3491
3489 3492
3490 3493 def instance(ui, path, create, intents=None, createopts=None):
3491 3494 localpath = util.urllocalpath(path)
3492 3495 if create:
3493 3496 createrepository(ui, localpath, createopts=createopts)
3494 3497
3495 3498 return makelocalrepository(ui, localpath, intents=intents)
3496 3499
3497 3500
3498 3501 def islocal(path):
3499 3502 return True
3500 3503
3501 3504
3502 3505 def defaultcreateopts(ui, createopts=None):
3503 3506 """Populate the default creation options for a repository.
3504 3507
3505 3508 A dictionary of explicitly requested creation options can be passed
3506 3509 in. Missing keys will be populated.
3507 3510 """
3508 3511 createopts = dict(createopts or {})
3509 3512
3510 3513 if b'backend' not in createopts:
3511 3514 # experimental config: storage.new-repo-backend
3512 3515 createopts[b'backend'] = ui.config(b'storage', b'new-repo-backend')
3513 3516
3514 3517 return createopts
3515 3518
3516 3519
3517 3520 def newreporequirements(ui, createopts):
3518 3521 """Determine the set of requirements for a new local repository.
3519 3522
3520 3523 Extensions can wrap this function to specify custom requirements for
3521 3524 new repositories.
3522 3525 """
3523 3526 # If the repo is being created from a shared repository, we copy
3524 3527 # its requirements.
3525 3528 if b'sharedrepo' in createopts:
3526 3529 requirements = set(createopts[b'sharedrepo'].requirements)
3527 3530 if createopts.get(b'sharedrelative'):
3528 3531 requirements.add(b'relshared')
3529 3532 else:
3530 3533 requirements.add(b'shared')
3531 3534
3532 3535 return requirements
3533 3536
3534 3537 if b'backend' not in createopts:
3535 3538 raise error.ProgrammingError(
3536 3539 b'backend key not present in createopts; '
3537 3540 b'was defaultcreateopts() called?'
3538 3541 )
3539 3542
3540 3543 if createopts[b'backend'] != b'revlogv1':
3541 3544 raise error.Abort(
3542 3545 _(
3543 3546 b'unable to determine repository requirements for '
3544 3547 b'storage backend: %s'
3545 3548 )
3546 3549 % createopts[b'backend']
3547 3550 )
3548 3551
3549 3552 requirements = {b'revlogv1'}
3550 3553 if ui.configbool(b'format', b'usestore'):
3551 3554 requirements.add(b'store')
3552 3555 if ui.configbool(b'format', b'usefncache'):
3553 3556 requirements.add(b'fncache')
3554 3557 if ui.configbool(b'format', b'dotencode'):
3555 3558 requirements.add(b'dotencode')
3556 3559
3557 3560 compengine = ui.config(b'format', b'revlog-compression')
3558 3561 if compengine not in util.compengines:
3559 3562 raise error.Abort(
3560 3563 _(
3561 3564 b'compression engine %s defined by '
3562 3565 b'format.revlog-compression not available'
3563 3566 )
3564 3567 % compengine,
3565 3568 hint=_(
3566 3569 b'run "hg debuginstall" to list available '
3567 3570 b'compression engines'
3568 3571 ),
3569 3572 )
3570 3573
3571 3574 # zlib is the historical default and doesn't need an explicit requirement.
3572 3575 elif compengine == b'zstd':
3573 3576 requirements.add(b'revlog-compression-zstd')
3574 3577 elif compengine != b'zlib':
3575 3578 requirements.add(b'exp-compression-%s' % compengine)
3576 3579
3577 3580 if scmutil.gdinitconfig(ui):
3578 3581 requirements.add(b'generaldelta')
3579 3582 if ui.configbool(b'format', b'sparse-revlog'):
3580 3583 requirements.add(SPARSEREVLOG_REQUIREMENT)
3581 3584
3582 3585 # experimental config: format.exp-use-side-data
3583 3586 if ui.configbool(b'format', b'exp-use-side-data'):
3584 3587 requirements.add(SIDEDATA_REQUIREMENT)
3585 3588 # experimental config: format.exp-use-copies-side-data-changeset
3586 3589 if ui.configbool(b'format', b'exp-use-copies-side-data-changeset'):
3587 3590 requirements.add(SIDEDATA_REQUIREMENT)
3588 3591 requirements.add(COPIESSDC_REQUIREMENT)
3589 3592 if ui.configbool(b'experimental', b'treemanifest'):
3590 3593 requirements.add(b'treemanifest')
3591 3594
3592 3595 revlogv2 = ui.config(b'experimental', b'revlogv2')
3593 3596 if revlogv2 == b'enable-unstable-format-and-corrupt-my-data':
3594 3597 requirements.remove(b'revlogv1')
3595 3598 # generaldelta is implied by revlogv2.
3596 3599 requirements.discard(b'generaldelta')
3597 3600 requirements.add(REVLOGV2_REQUIREMENT)
3598 3601 # experimental config: format.internal-phase
3599 3602 if ui.configbool(b'format', b'internal-phase'):
3600 3603 requirements.add(b'internal-phase')
3601 3604
3602 3605 if createopts.get(b'narrowfiles'):
3603 3606 requirements.add(repository.NARROW_REQUIREMENT)
3604 3607
3605 3608 if createopts.get(b'lfs'):
3606 3609 requirements.add(b'lfs')
3607 3610
3608 3611 if ui.configbool(b'format', b'bookmarks-in-store'):
3609 3612 requirements.add(bookmarks.BOOKMARKS_IN_STORE_REQUIREMENT)
3610 3613
3611 3614 return requirements
3612 3615
3613 3616
3614 3617 def filterknowncreateopts(ui, createopts):
3615 3618 """Filters a dict of repo creation options against options that are known.
3616 3619
3617 3620 Receives a dict of repo creation options and returns a dict of those
3618 3621 options that we don't know how to handle.
3619 3622
3620 3623 This function is called as part of repository creation. If the
3621 3624 returned dict contains any items, repository creation will not
3622 3625 be allowed, as it means there was a request to create a repository
3623 3626 with options not recognized by loaded code.
3624 3627
3625 3628 Extensions can wrap this function to filter out creation options
3626 3629 they know how to handle.
3627 3630 """
3628 3631 known = {
3629 3632 b'backend',
3630 3633 b'lfs',
3631 3634 b'narrowfiles',
3632 3635 b'sharedrepo',
3633 3636 b'sharedrelative',
3634 3637 b'shareditems',
3635 3638 b'shallowfilestore',
3636 3639 }
3637 3640
3638 3641 return {k: v for k, v in createopts.items() if k not in known}
3639 3642
3640 3643
3641 3644 def createrepository(ui, path, createopts=None):
3642 3645 """Create a new repository in a vfs.
3643 3646
3644 3647 ``path`` path to the new repo's working directory.
3645 3648 ``createopts`` options for the new repository.
3646 3649
3647 3650 The following keys for ``createopts`` are recognized:
3648 3651
3649 3652 backend
3650 3653 The storage backend to use.
3651 3654 lfs
3652 3655 Repository will be created with ``lfs`` requirement. The lfs extension
3653 3656 will automatically be loaded when the repository is accessed.
3654 3657 narrowfiles
3655 3658 Set up repository to support narrow file storage.
3656 3659 sharedrepo
3657 3660 Repository object from which storage should be shared.
3658 3661 sharedrelative
3659 3662 Boolean indicating if the path to the shared repo should be
3660 3663 stored as relative. By default, the pointer to the "parent" repo
3661 3664 is stored as an absolute path.
3662 3665 shareditems
3663 3666 Set of items to share to the new repository (in addition to storage).
3664 3667 shallowfilestore
3665 3668 Indicates that storage for files should be shallow (not all ancestor
3666 3669 revisions are known).
3667 3670 """
3668 3671 createopts = defaultcreateopts(ui, createopts=createopts)
3669 3672
3670 3673 unknownopts = filterknowncreateopts(ui, createopts)
3671 3674
3672 3675 if not isinstance(unknownopts, dict):
3673 3676 raise error.ProgrammingError(
3674 3677 b'filterknowncreateopts() did not return a dict'
3675 3678 )
3676 3679
3677 3680 if unknownopts:
3678 3681 raise error.Abort(
3679 3682 _(
3680 3683 b'unable to create repository because of unknown '
3681 3684 b'creation option: %s'
3682 3685 )
3683 3686 % b', '.join(sorted(unknownopts)),
3684 3687 hint=_(b'is a required extension not loaded?'),
3685 3688 )
3686 3689
3687 3690 requirements = newreporequirements(ui, createopts=createopts)
3688 3691
3689 3692 wdirvfs = vfsmod.vfs(path, expandpath=True, realpath=True)
3690 3693
3691 3694 hgvfs = vfsmod.vfs(wdirvfs.join(b'.hg'))
3692 3695 if hgvfs.exists():
3693 3696 raise error.RepoError(_(b'repository %s already exists') % path)
3694 3697
3695 3698 if b'sharedrepo' in createopts:
3696 3699 sharedpath = createopts[b'sharedrepo'].sharedpath
3697 3700
3698 3701 if createopts.get(b'sharedrelative'):
3699 3702 try:
3700 3703 sharedpath = os.path.relpath(sharedpath, hgvfs.base)
3701 3704 except (IOError, ValueError) as e:
3702 3705 # ValueError is raised on Windows if the drive letters differ
3703 3706 # on each path.
3704 3707 raise error.Abort(
3705 3708 _(b'cannot calculate relative path'),
3706 3709 hint=stringutil.forcebytestr(e),
3707 3710 )
3708 3711
3709 3712 if not wdirvfs.exists():
3710 3713 wdirvfs.makedirs()
3711 3714
3712 3715 hgvfs.makedir(notindexed=True)
3713 3716 if b'sharedrepo' not in createopts:
3714 3717 hgvfs.mkdir(b'cache')
3715 3718 hgvfs.mkdir(b'wcache')
3716 3719
3717 3720 if b'store' in requirements and b'sharedrepo' not in createopts:
3718 3721 hgvfs.mkdir(b'store')
3719 3722
3720 3723 # We create an invalid changelog outside the store so very old
3721 3724 # Mercurial versions (which didn't know about the requirements
3722 3725 # file) encounter an error on reading the changelog. This
3723 3726 # effectively locks out old clients and prevents them from
3724 3727 # mucking with a repo in an unknown format.
3725 3728 #
3726 3729 # The revlog header has version 2, which won't be recognized by
3727 3730 # such old clients.
3728 3731 hgvfs.append(
3729 3732 b'00changelog.i',
3730 3733 b'\0\0\0\2 dummy changelog to prevent using the old repo '
3731 3734 b'layout',
3732 3735 )
3733 3736
3734 3737 scmutil.writerequires(hgvfs, requirements)
3735 3738
3736 3739 # Write out file telling readers where to find the shared store.
3737 3740 if b'sharedrepo' in createopts:
3738 3741 hgvfs.write(b'sharedpath', sharedpath)
3739 3742
3740 3743 if createopts.get(b'shareditems'):
3741 3744 shared = b'\n'.join(sorted(createopts[b'shareditems'])) + b'\n'
3742 3745 hgvfs.write(b'shared', shared)
3743 3746
3744 3747
3745 3748 def poisonrepository(repo):
3746 3749 """Poison a repository instance so it can no longer be used."""
3747 3750 # Perform any cleanup on the instance.
3748 3751 repo.close()
3749 3752
3750 3753 # Our strategy is to replace the type of the object with one that
3751 3754 # has all attribute lookups result in error.
3752 3755 #
3753 3756 # But we have to allow the close() method because some constructors
3754 3757 # of repos call close() on repo references.
3755 3758 class poisonedrepository(object):
3756 3759 def __getattribute__(self, item):
3757 3760 if item == 'close':
3758 3761 return object.__getattribute__(self, item)
3759 3762
3760 3763 raise error.ProgrammingError(
3761 3764 b'repo instances should not be used after unshare'
3762 3765 )
3763 3766
3764 3767 def close(self):
3765 3768 pass
3766 3769
3767 3770 # We may have a repoview, which intercepts __setattr__. So be sure
3768 3771 # we operate at the lowest level possible.
3769 3772 object.__setattr__(repo, '__class__', poisonedrepository)
General Comments 0
You need to be logged in to leave comments. Login now