##// END OF EJS Templates
localrepo: recognize trivial "null" queries in `anyrev`...
marmoute -
r44146:c0e04f07 default draft
parent child Browse files
Show More
@@ -1,3727 +1,3729 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 hashlib
12 12 import os
13 13 import random
14 14 import sys
15 15 import time
16 16 import weakref
17 17
18 18 from .i18n import _
19 19 from .node import (
20 20 bin,
21 21 hex,
22 22 nullid,
23 23 nullrev,
24 24 short,
25 25 )
26 26 from .pycompat import (
27 27 delattr,
28 28 getattr,
29 29 )
30 30 from . import (
31 31 bookmarks,
32 32 branchmap,
33 33 bundle2,
34 34 changegroup,
35 35 color,
36 36 context,
37 37 dirstate,
38 38 dirstateguard,
39 39 discovery,
40 40 encoding,
41 41 error,
42 42 exchange,
43 43 extensions,
44 44 filelog,
45 45 hook,
46 46 lock as lockmod,
47 47 match as matchmod,
48 48 merge as mergemod,
49 49 mergeutil,
50 50 namespaces,
51 51 narrowspec,
52 52 obsolete,
53 53 pathutil,
54 54 phases,
55 55 pushkey,
56 56 pycompat,
57 57 repoview,
58 58 revset,
59 59 revsetlang,
60 60 scmutil,
61 61 sparse,
62 62 store as storemod,
63 63 subrepoutil,
64 64 tags as tagsmod,
65 65 transaction,
66 66 txnutil,
67 67 util,
68 68 vfs as vfsmod,
69 69 )
70 70
71 71 from .interfaces import (
72 72 repository,
73 73 util as interfaceutil,
74 74 )
75 75
76 76 from .utils import (
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 return options
931 931
932 932
933 933 def makemain(**kwargs):
934 934 """Produce a type conforming to ``ilocalrepositorymain``."""
935 935 return localrepository
936 936
937 937
938 938 @interfaceutil.implementer(repository.ilocalrepositoryfilestorage)
939 939 class revlogfilestorage(object):
940 940 """File storage when using revlogs."""
941 941
942 942 def file(self, path):
943 943 if path[0] == b'/':
944 944 path = path[1:]
945 945
946 946 return filelog.filelog(self.svfs, path)
947 947
948 948
949 949 @interfaceutil.implementer(repository.ilocalrepositoryfilestorage)
950 950 class revlognarrowfilestorage(object):
951 951 """File storage when using revlogs and narrow files."""
952 952
953 953 def file(self, path):
954 954 if path[0] == b'/':
955 955 path = path[1:]
956 956
957 957 return filelog.narrowfilelog(self.svfs, path, self._storenarrowmatch)
958 958
959 959
960 960 def makefilestorage(requirements, features, **kwargs):
961 961 """Produce a type conforming to ``ilocalrepositoryfilestorage``."""
962 962 features.add(repository.REPO_FEATURE_REVLOG_FILE_STORAGE)
963 963 features.add(repository.REPO_FEATURE_STREAM_CLONE)
964 964
965 965 if repository.NARROW_REQUIREMENT in requirements:
966 966 return revlognarrowfilestorage
967 967 else:
968 968 return revlogfilestorage
969 969
970 970
971 971 # List of repository interfaces and factory functions for them. Each
972 972 # will be called in order during ``makelocalrepository()`` to iteratively
973 973 # derive the final type for a local repository instance. We capture the
974 974 # function as a lambda so we don't hold a reference and the module-level
975 975 # functions can be wrapped.
976 976 REPO_INTERFACES = [
977 977 (repository.ilocalrepositorymain, lambda: makemain),
978 978 (repository.ilocalrepositoryfilestorage, lambda: makefilestorage),
979 979 ]
980 980
981 981
982 982 @interfaceutil.implementer(repository.ilocalrepositorymain)
983 983 class localrepository(object):
984 984 """Main class for representing local repositories.
985 985
986 986 All local repositories are instances of this class.
987 987
988 988 Constructed on its own, instances of this class are not usable as
989 989 repository objects. To obtain a usable repository object, call
990 990 ``hg.repository()``, ``localrepo.instance()``, or
991 991 ``localrepo.makelocalrepository()``. The latter is the lowest-level.
992 992 ``instance()`` adds support for creating new repositories.
993 993 ``hg.repository()`` adds more extension integration, including calling
994 994 ``reposetup()``. Generally speaking, ``hg.repository()`` should be
995 995 used.
996 996 """
997 997
998 998 # obsolete experimental requirements:
999 999 # - manifestv2: An experimental new manifest format that allowed
1000 1000 # for stem compression of long paths. Experiment ended up not
1001 1001 # being successful (repository sizes went up due to worse delta
1002 1002 # chains), and the code was deleted in 4.6.
1003 1003 supportedformats = {
1004 1004 b'revlogv1',
1005 1005 b'generaldelta',
1006 1006 b'treemanifest',
1007 1007 COPIESSDC_REQUIREMENT,
1008 1008 REVLOGV2_REQUIREMENT,
1009 1009 SIDEDATA_REQUIREMENT,
1010 1010 SPARSEREVLOG_REQUIREMENT,
1011 1011 bookmarks.BOOKMARKS_IN_STORE_REQUIREMENT,
1012 1012 }
1013 1013 _basesupported = supportedformats | {
1014 1014 b'store',
1015 1015 b'fncache',
1016 1016 b'shared',
1017 1017 b'relshared',
1018 1018 b'dotencode',
1019 1019 b'exp-sparse',
1020 1020 b'internal-phase',
1021 1021 }
1022 1022
1023 1023 # list of prefix for file which can be written without 'wlock'
1024 1024 # Extensions should extend this list when needed
1025 1025 _wlockfreeprefix = {
1026 1026 # We migh consider requiring 'wlock' for the next
1027 1027 # two, but pretty much all the existing code assume
1028 1028 # wlock is not needed so we keep them excluded for
1029 1029 # now.
1030 1030 b'hgrc',
1031 1031 b'requires',
1032 1032 # XXX cache is a complicatged business someone
1033 1033 # should investigate this in depth at some point
1034 1034 b'cache/',
1035 1035 # XXX shouldn't be dirstate covered by the wlock?
1036 1036 b'dirstate',
1037 1037 # XXX bisect was still a bit too messy at the time
1038 1038 # this changeset was introduced. Someone should fix
1039 1039 # the remainig bit and drop this line
1040 1040 b'bisect.state',
1041 1041 }
1042 1042
1043 1043 def __init__(
1044 1044 self,
1045 1045 baseui,
1046 1046 ui,
1047 1047 origroot,
1048 1048 wdirvfs,
1049 1049 hgvfs,
1050 1050 requirements,
1051 1051 supportedrequirements,
1052 1052 sharedpath,
1053 1053 store,
1054 1054 cachevfs,
1055 1055 wcachevfs,
1056 1056 features,
1057 1057 intents=None,
1058 1058 ):
1059 1059 """Create a new local repository instance.
1060 1060
1061 1061 Most callers should use ``hg.repository()``, ``localrepo.instance()``,
1062 1062 or ``localrepo.makelocalrepository()`` for obtaining a new repository
1063 1063 object.
1064 1064
1065 1065 Arguments:
1066 1066
1067 1067 baseui
1068 1068 ``ui.ui`` instance that ``ui`` argument was based off of.
1069 1069
1070 1070 ui
1071 1071 ``ui.ui`` instance for use by the repository.
1072 1072
1073 1073 origroot
1074 1074 ``bytes`` path to working directory root of this repository.
1075 1075
1076 1076 wdirvfs
1077 1077 ``vfs.vfs`` rooted at the working directory.
1078 1078
1079 1079 hgvfs
1080 1080 ``vfs.vfs`` rooted at .hg/
1081 1081
1082 1082 requirements
1083 1083 ``set`` of bytestrings representing repository opening requirements.
1084 1084
1085 1085 supportedrequirements
1086 1086 ``set`` of bytestrings representing repository requirements that we
1087 1087 know how to open. May be a supetset of ``requirements``.
1088 1088
1089 1089 sharedpath
1090 1090 ``bytes`` Defining path to storage base directory. Points to a
1091 1091 ``.hg/`` directory somewhere.
1092 1092
1093 1093 store
1094 1094 ``store.basicstore`` (or derived) instance providing access to
1095 1095 versioned storage.
1096 1096
1097 1097 cachevfs
1098 1098 ``vfs.vfs`` used for cache files.
1099 1099
1100 1100 wcachevfs
1101 1101 ``vfs.vfs`` used for cache files related to the working copy.
1102 1102
1103 1103 features
1104 1104 ``set`` of bytestrings defining features/capabilities of this
1105 1105 instance.
1106 1106
1107 1107 intents
1108 1108 ``set`` of system strings indicating what this repo will be used
1109 1109 for.
1110 1110 """
1111 1111 self.baseui = baseui
1112 1112 self.ui = ui
1113 1113 self.origroot = origroot
1114 1114 # vfs rooted at working directory.
1115 1115 self.wvfs = wdirvfs
1116 1116 self.root = wdirvfs.base
1117 1117 # vfs rooted at .hg/. Used to access most non-store paths.
1118 1118 self.vfs = hgvfs
1119 1119 self.path = hgvfs.base
1120 1120 self.requirements = requirements
1121 1121 self.supported = supportedrequirements
1122 1122 self.sharedpath = sharedpath
1123 1123 self.store = store
1124 1124 self.cachevfs = cachevfs
1125 1125 self.wcachevfs = wcachevfs
1126 1126 self.features = features
1127 1127
1128 1128 self.filtername = None
1129 1129
1130 1130 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
1131 1131 b'devel', b'check-locks'
1132 1132 ):
1133 1133 self.vfs.audit = self._getvfsward(self.vfs.audit)
1134 1134 # A list of callback to shape the phase if no data were found.
1135 1135 # Callback are in the form: func(repo, roots) --> processed root.
1136 1136 # This list it to be filled by extension during repo setup
1137 1137 self._phasedefaults = []
1138 1138
1139 1139 color.setup(self.ui)
1140 1140
1141 1141 self.spath = self.store.path
1142 1142 self.svfs = self.store.vfs
1143 1143 self.sjoin = self.store.join
1144 1144 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
1145 1145 b'devel', b'check-locks'
1146 1146 ):
1147 1147 if util.safehasattr(self.svfs, b'vfs'): # this is filtervfs
1148 1148 self.svfs.vfs.audit = self._getsvfsward(self.svfs.vfs.audit)
1149 1149 else: # standard vfs
1150 1150 self.svfs.audit = self._getsvfsward(self.svfs.audit)
1151 1151
1152 1152 self._dirstatevalidatewarned = False
1153 1153
1154 1154 self._branchcaches = branchmap.BranchMapCache()
1155 1155 self._revbranchcache = None
1156 1156 self._filterpats = {}
1157 1157 self._datafilters = {}
1158 1158 self._transref = self._lockref = self._wlockref = None
1159 1159
1160 1160 # A cache for various files under .hg/ that tracks file changes,
1161 1161 # (used by the filecache decorator)
1162 1162 #
1163 1163 # Maps a property name to its util.filecacheentry
1164 1164 self._filecache = {}
1165 1165
1166 1166 # hold sets of revision to be filtered
1167 1167 # should be cleared when something might have changed the filter value:
1168 1168 # - new changesets,
1169 1169 # - phase change,
1170 1170 # - new obsolescence marker,
1171 1171 # - working directory parent change,
1172 1172 # - bookmark changes
1173 1173 self.filteredrevcache = {}
1174 1174
1175 1175 # post-dirstate-status hooks
1176 1176 self._postdsstatus = []
1177 1177
1178 1178 # generic mapping between names and nodes
1179 1179 self.names = namespaces.namespaces()
1180 1180
1181 1181 # Key to signature value.
1182 1182 self._sparsesignaturecache = {}
1183 1183 # Signature to cached matcher instance.
1184 1184 self._sparsematchercache = {}
1185 1185
1186 1186 self._extrafilterid = repoview.extrafilter(ui)
1187 1187
1188 1188 self.filecopiesmode = None
1189 1189 if COPIESSDC_REQUIREMENT in self.requirements:
1190 1190 self.filecopiesmode = b'changeset-sidedata'
1191 1191
1192 1192 def _getvfsward(self, origfunc):
1193 1193 """build a ward for self.vfs"""
1194 1194 rref = weakref.ref(self)
1195 1195
1196 1196 def checkvfs(path, mode=None):
1197 1197 ret = origfunc(path, mode=mode)
1198 1198 repo = rref()
1199 1199 if (
1200 1200 repo is None
1201 1201 or not util.safehasattr(repo, b'_wlockref')
1202 1202 or not util.safehasattr(repo, b'_lockref')
1203 1203 ):
1204 1204 return
1205 1205 if mode in (None, b'r', b'rb'):
1206 1206 return
1207 1207 if path.startswith(repo.path):
1208 1208 # truncate name relative to the repository (.hg)
1209 1209 path = path[len(repo.path) + 1 :]
1210 1210 if path.startswith(b'cache/'):
1211 1211 msg = b'accessing cache with vfs instead of cachevfs: "%s"'
1212 1212 repo.ui.develwarn(msg % path, stacklevel=3, config=b"cache-vfs")
1213 1213 if path.startswith(b'journal.') or path.startswith(b'undo.'):
1214 1214 # journal is covered by 'lock'
1215 1215 if repo._currentlock(repo._lockref) is None:
1216 1216 repo.ui.develwarn(
1217 1217 b'write with no lock: "%s"' % path,
1218 1218 stacklevel=3,
1219 1219 config=b'check-locks',
1220 1220 )
1221 1221 elif repo._currentlock(repo._wlockref) is None:
1222 1222 # rest of vfs files are covered by 'wlock'
1223 1223 #
1224 1224 # exclude special files
1225 1225 for prefix in self._wlockfreeprefix:
1226 1226 if path.startswith(prefix):
1227 1227 return
1228 1228 repo.ui.develwarn(
1229 1229 b'write with no wlock: "%s"' % path,
1230 1230 stacklevel=3,
1231 1231 config=b'check-locks',
1232 1232 )
1233 1233 return ret
1234 1234
1235 1235 return checkvfs
1236 1236
1237 1237 def _getsvfsward(self, origfunc):
1238 1238 """build a ward for self.svfs"""
1239 1239 rref = weakref.ref(self)
1240 1240
1241 1241 def checksvfs(path, mode=None):
1242 1242 ret = origfunc(path, mode=mode)
1243 1243 repo = rref()
1244 1244 if repo is None or not util.safehasattr(repo, b'_lockref'):
1245 1245 return
1246 1246 if mode in (None, b'r', b'rb'):
1247 1247 return
1248 1248 if path.startswith(repo.sharedpath):
1249 1249 # truncate name relative to the repository (.hg)
1250 1250 path = path[len(repo.sharedpath) + 1 :]
1251 1251 if repo._currentlock(repo._lockref) is None:
1252 1252 repo.ui.develwarn(
1253 1253 b'write with no lock: "%s"' % path, stacklevel=4
1254 1254 )
1255 1255 return ret
1256 1256
1257 1257 return checksvfs
1258 1258
1259 1259 def close(self):
1260 1260 self._writecaches()
1261 1261
1262 1262 def _writecaches(self):
1263 1263 if self._revbranchcache:
1264 1264 self._revbranchcache.write()
1265 1265
1266 1266 def _restrictcapabilities(self, caps):
1267 1267 if self.ui.configbool(b'experimental', b'bundle2-advertise'):
1268 1268 caps = set(caps)
1269 1269 capsblob = bundle2.encodecaps(
1270 1270 bundle2.getrepocaps(self, role=b'client')
1271 1271 )
1272 1272 caps.add(b'bundle2=' + urlreq.quote(capsblob))
1273 1273 return caps
1274 1274
1275 1275 def _writerequirements(self):
1276 1276 scmutil.writerequires(self.vfs, self.requirements)
1277 1277
1278 1278 # Don't cache auditor/nofsauditor, or you'll end up with reference cycle:
1279 1279 # self -> auditor -> self._checknested -> self
1280 1280
1281 1281 @property
1282 1282 def auditor(self):
1283 1283 # This is only used by context.workingctx.match in order to
1284 1284 # detect files in subrepos.
1285 1285 return pathutil.pathauditor(self.root, callback=self._checknested)
1286 1286
1287 1287 @property
1288 1288 def nofsauditor(self):
1289 1289 # This is only used by context.basectx.match in order to detect
1290 1290 # files in subrepos.
1291 1291 return pathutil.pathauditor(
1292 1292 self.root, callback=self._checknested, realfs=False, cached=True
1293 1293 )
1294 1294
1295 1295 def _checknested(self, path):
1296 1296 """Determine if path is a legal nested repository."""
1297 1297 if not path.startswith(self.root):
1298 1298 return False
1299 1299 subpath = path[len(self.root) + 1 :]
1300 1300 normsubpath = util.pconvert(subpath)
1301 1301
1302 1302 # XXX: Checking against the current working copy is wrong in
1303 1303 # the sense that it can reject things like
1304 1304 #
1305 1305 # $ hg cat -r 10 sub/x.txt
1306 1306 #
1307 1307 # if sub/ is no longer a subrepository in the working copy
1308 1308 # parent revision.
1309 1309 #
1310 1310 # However, it can of course also allow things that would have
1311 1311 # been rejected before, such as the above cat command if sub/
1312 1312 # is a subrepository now, but was a normal directory before.
1313 1313 # The old path auditor would have rejected by mistake since it
1314 1314 # panics when it sees sub/.hg/.
1315 1315 #
1316 1316 # All in all, checking against the working copy seems sensible
1317 1317 # since we want to prevent access to nested repositories on
1318 1318 # the filesystem *now*.
1319 1319 ctx = self[None]
1320 1320 parts = util.splitpath(subpath)
1321 1321 while parts:
1322 1322 prefix = b'/'.join(parts)
1323 1323 if prefix in ctx.substate:
1324 1324 if prefix == normsubpath:
1325 1325 return True
1326 1326 else:
1327 1327 sub = ctx.sub(prefix)
1328 1328 return sub.checknested(subpath[len(prefix) + 1 :])
1329 1329 else:
1330 1330 parts.pop()
1331 1331 return False
1332 1332
1333 1333 def peer(self):
1334 1334 return localpeer(self) # not cached to avoid reference cycle
1335 1335
1336 1336 def unfiltered(self):
1337 1337 """Return unfiltered version of the repository
1338 1338
1339 1339 Intended to be overwritten by filtered repo."""
1340 1340 return self
1341 1341
1342 1342 def filtered(self, name, visibilityexceptions=None):
1343 1343 """Return a filtered version of a repository
1344 1344
1345 1345 The `name` parameter is the identifier of the requested view. This
1346 1346 will return a repoview object set "exactly" to the specified view.
1347 1347
1348 1348 This function does not apply recursive filtering to a repository. For
1349 1349 example calling `repo.filtered("served")` will return a repoview using
1350 1350 the "served" view, regardless of the initial view used by `repo`.
1351 1351
1352 1352 In other word, there is always only one level of `repoview` "filtering".
1353 1353 """
1354 1354 if self._extrafilterid is not None and b'%' not in name:
1355 1355 name = name + b'%' + self._extrafilterid
1356 1356
1357 1357 cls = repoview.newtype(self.unfiltered().__class__)
1358 1358 return cls(self, name, visibilityexceptions)
1359 1359
1360 1360 @mixedrepostorecache(
1361 1361 (b'bookmarks', b'plain'),
1362 1362 (b'bookmarks.current', b'plain'),
1363 1363 (b'bookmarks', b''),
1364 1364 (b'00changelog.i', b''),
1365 1365 )
1366 1366 def _bookmarks(self):
1367 1367 # Since the multiple files involved in the transaction cannot be
1368 1368 # written atomically (with current repository format), there is a race
1369 1369 # condition here.
1370 1370 #
1371 1371 # 1) changelog content A is read
1372 1372 # 2) outside transaction update changelog to content B
1373 1373 # 3) outside transaction update bookmark file referring to content B
1374 1374 # 4) bookmarks file content is read and filtered against changelog-A
1375 1375 #
1376 1376 # When this happens, bookmarks against nodes missing from A are dropped.
1377 1377 #
1378 1378 # Having this happening during read is not great, but it become worse
1379 1379 # when this happen during write because the bookmarks to the "unknown"
1380 1380 # nodes will be dropped for good. However, writes happen within locks.
1381 1381 # This locking makes it possible to have a race free consistent read.
1382 1382 # For this purpose data read from disc before locking are
1383 1383 # "invalidated" right after the locks are taken. This invalidations are
1384 1384 # "light", the `filecache` mechanism keep the data in memory and will
1385 1385 # reuse them if the underlying files did not changed. Not parsing the
1386 1386 # same data multiple times helps performances.
1387 1387 #
1388 1388 # Unfortunately in the case describe above, the files tracked by the
1389 1389 # bookmarks file cache might not have changed, but the in-memory
1390 1390 # content is still "wrong" because we used an older changelog content
1391 1391 # to process the on-disk data. So after locking, the changelog would be
1392 1392 # refreshed but `_bookmarks` would be preserved.
1393 1393 # Adding `00changelog.i` to the list of tracked file is not
1394 1394 # enough, because at the time we build the content for `_bookmarks` in
1395 1395 # (4), the changelog file has already diverged from the content used
1396 1396 # for loading `changelog` in (1)
1397 1397 #
1398 1398 # To prevent the issue, we force the changelog to be explicitly
1399 1399 # reloaded while computing `_bookmarks`. The data race can still happen
1400 1400 # without the lock (with a narrower window), but it would no longer go
1401 1401 # undetected during the lock time refresh.
1402 1402 #
1403 1403 # The new schedule is as follow
1404 1404 #
1405 1405 # 1) filecache logic detect that `_bookmarks` needs to be computed
1406 1406 # 2) cachestat for `bookmarks` and `changelog` are captured (for book)
1407 1407 # 3) We force `changelog` filecache to be tested
1408 1408 # 4) cachestat for `changelog` are captured (for changelog)
1409 1409 # 5) `_bookmarks` is computed and cached
1410 1410 #
1411 1411 # The step in (3) ensure we have a changelog at least as recent as the
1412 1412 # cache stat computed in (1). As a result at locking time:
1413 1413 # * if the changelog did not changed since (1) -> we can reuse the data
1414 1414 # * otherwise -> the bookmarks get refreshed.
1415 1415 self._refreshchangelog()
1416 1416 return bookmarks.bmstore(self)
1417 1417
1418 1418 def _refreshchangelog(self):
1419 1419 """make sure the in memory changelog match the on-disk one"""
1420 1420 if 'changelog' in vars(self) and self.currenttransaction() is None:
1421 1421 del self.changelog
1422 1422
1423 1423 @property
1424 1424 def _activebookmark(self):
1425 1425 return self._bookmarks.active
1426 1426
1427 1427 # _phasesets depend on changelog. what we need is to call
1428 1428 # _phasecache.invalidate() if '00changelog.i' was changed, but it
1429 1429 # can't be easily expressed in filecache mechanism.
1430 1430 @storecache(b'phaseroots', b'00changelog.i')
1431 1431 def _phasecache(self):
1432 1432 return phases.phasecache(self, self._phasedefaults)
1433 1433
1434 1434 @storecache(b'obsstore')
1435 1435 def obsstore(self):
1436 1436 return obsolete.makestore(self.ui, self)
1437 1437
1438 1438 @storecache(b'00changelog.i')
1439 1439 def changelog(self):
1440 1440 return self.store.changelog(txnutil.mayhavepending(self.root))
1441 1441
1442 1442 @storecache(b'00manifest.i')
1443 1443 def manifestlog(self):
1444 1444 return self.store.manifestlog(self, self._storenarrowmatch)
1445 1445
1446 1446 @repofilecache(b'dirstate')
1447 1447 def dirstate(self):
1448 1448 return self._makedirstate()
1449 1449
1450 1450 def _makedirstate(self):
1451 1451 """Extension point for wrapping the dirstate per-repo."""
1452 1452 sparsematchfn = lambda: sparse.matcher(self)
1453 1453
1454 1454 return dirstate.dirstate(
1455 1455 self.vfs, self.ui, self.root, self._dirstatevalidate, sparsematchfn
1456 1456 )
1457 1457
1458 1458 def _dirstatevalidate(self, node):
1459 1459 try:
1460 1460 self.changelog.rev(node)
1461 1461 return node
1462 1462 except error.LookupError:
1463 1463 if not self._dirstatevalidatewarned:
1464 1464 self._dirstatevalidatewarned = True
1465 1465 self.ui.warn(
1466 1466 _(b"warning: ignoring unknown working parent %s!\n")
1467 1467 % short(node)
1468 1468 )
1469 1469 return nullid
1470 1470
1471 1471 @storecache(narrowspec.FILENAME)
1472 1472 def narrowpats(self):
1473 1473 """matcher patterns for this repository's narrowspec
1474 1474
1475 1475 A tuple of (includes, excludes).
1476 1476 """
1477 1477 return narrowspec.load(self)
1478 1478
1479 1479 @storecache(narrowspec.FILENAME)
1480 1480 def _storenarrowmatch(self):
1481 1481 if repository.NARROW_REQUIREMENT not in self.requirements:
1482 1482 return matchmod.always()
1483 1483 include, exclude = self.narrowpats
1484 1484 return narrowspec.match(self.root, include=include, exclude=exclude)
1485 1485
1486 1486 @storecache(narrowspec.FILENAME)
1487 1487 def _narrowmatch(self):
1488 1488 if repository.NARROW_REQUIREMENT not in self.requirements:
1489 1489 return matchmod.always()
1490 1490 narrowspec.checkworkingcopynarrowspec(self)
1491 1491 include, exclude = self.narrowpats
1492 1492 return narrowspec.match(self.root, include=include, exclude=exclude)
1493 1493
1494 1494 def narrowmatch(self, match=None, includeexact=False):
1495 1495 """matcher corresponding the the repo's narrowspec
1496 1496
1497 1497 If `match` is given, then that will be intersected with the narrow
1498 1498 matcher.
1499 1499
1500 1500 If `includeexact` is True, then any exact matches from `match` will
1501 1501 be included even if they're outside the narrowspec.
1502 1502 """
1503 1503 if match:
1504 1504 if includeexact and not self._narrowmatch.always():
1505 1505 # do not exclude explicitly-specified paths so that they can
1506 1506 # be warned later on
1507 1507 em = matchmod.exact(match.files())
1508 1508 nm = matchmod.unionmatcher([self._narrowmatch, em])
1509 1509 return matchmod.intersectmatchers(match, nm)
1510 1510 return matchmod.intersectmatchers(match, self._narrowmatch)
1511 1511 return self._narrowmatch
1512 1512
1513 1513 def setnarrowpats(self, newincludes, newexcludes):
1514 1514 narrowspec.save(self, newincludes, newexcludes)
1515 1515 self.invalidate(clearfilecache=True)
1516 1516
1517 1517 def __getitem__(self, changeid):
1518 1518 # dealing with special cases
1519 1519 if changeid is None:
1520 1520 return context.workingctx(self)
1521 1521 if isinstance(changeid, context.basectx):
1522 1522 return changeid
1523 1523
1524 1524 # dealing with multiple revisions
1525 1525 if isinstance(changeid, slice):
1526 1526 # wdirrev isn't contiguous so the slice shouldn't include it
1527 1527 return [
1528 1528 self[i]
1529 1529 for i in pycompat.xrange(*changeid.indices(len(self)))
1530 1530 if i not in self.changelog.filteredrevs
1531 1531 ]
1532 1532
1533 1533 # dealing with some special values
1534 1534 if changeid == b'null' or changeid == nullrev:
1535 1535 return context.changectx(self, nullrev, nullid)
1536 1536 if changeid == b'tip':
1537 1537 node = self.changelog.tip()
1538 1538 rev = self.changelog.rev(node)
1539 1539 return context.changectx(self, rev, node)
1540 1540
1541 1541 # dealing with arbitrary values
1542 1542 try:
1543 1543 if isinstance(changeid, int):
1544 1544 node = self.changelog.node(changeid)
1545 1545 rev = changeid
1546 1546 elif changeid == b'.':
1547 1547 # this is a hack to delay/avoid loading obsmarkers
1548 1548 # when we know that '.' won't be hidden
1549 1549 node = self.dirstate.p1()
1550 1550 rev = self.unfiltered().changelog.rev(node)
1551 1551 elif len(changeid) == 20:
1552 1552 try:
1553 1553 node = changeid
1554 1554 rev = self.changelog.rev(changeid)
1555 1555 except error.FilteredLookupError:
1556 1556 changeid = hex(changeid) # for the error message
1557 1557 raise
1558 1558 except LookupError:
1559 1559 # check if it might have come from damaged dirstate
1560 1560 #
1561 1561 # XXX we could avoid the unfiltered if we had a recognizable
1562 1562 # exception for filtered changeset access
1563 1563 if (
1564 1564 self.local()
1565 1565 and changeid in self.unfiltered().dirstate.parents()
1566 1566 ):
1567 1567 msg = _(b"working directory has unknown parent '%s'!")
1568 1568 raise error.Abort(msg % short(changeid))
1569 1569 changeid = hex(changeid) # for the error message
1570 1570 raise
1571 1571
1572 1572 elif len(changeid) == 40:
1573 1573 node = bin(changeid)
1574 1574 rev = self.changelog.rev(node)
1575 1575 else:
1576 1576 raise error.ProgrammingError(
1577 1577 b"unsupported changeid '%s' of type %s"
1578 1578 % (changeid, pycompat.bytestr(type(changeid)))
1579 1579 )
1580 1580
1581 1581 return context.changectx(self, rev, node)
1582 1582
1583 1583 except (error.FilteredIndexError, error.FilteredLookupError):
1584 1584 raise error.FilteredRepoLookupError(
1585 1585 _(b"filtered revision '%s'") % pycompat.bytestr(changeid)
1586 1586 )
1587 1587 except (IndexError, LookupError):
1588 1588 raise error.RepoLookupError(
1589 1589 _(b"unknown revision '%s'") % pycompat.bytestr(changeid)
1590 1590 )
1591 1591 except error.WdirUnsupported:
1592 1592 return context.workingctx(self)
1593 1593
1594 1594 def __contains__(self, changeid):
1595 1595 """True if the given changeid exists
1596 1596
1597 1597 error.AmbiguousPrefixLookupError is raised if an ambiguous node
1598 1598 specified.
1599 1599 """
1600 1600 try:
1601 1601 self[changeid]
1602 1602 return True
1603 1603 except error.RepoLookupError:
1604 1604 return False
1605 1605
1606 1606 def __nonzero__(self):
1607 1607 return True
1608 1608
1609 1609 __bool__ = __nonzero__
1610 1610
1611 1611 def __len__(self):
1612 1612 # no need to pay the cost of repoview.changelog
1613 1613 unfi = self.unfiltered()
1614 1614 return len(unfi.changelog)
1615 1615
1616 1616 def __iter__(self):
1617 1617 return iter(self.changelog)
1618 1618
1619 1619 def revs(self, expr, *args):
1620 1620 '''Find revisions matching a revset.
1621 1621
1622 1622 The revset is specified as a string ``expr`` that may contain
1623 1623 %-formatting to escape certain types. See ``revsetlang.formatspec``.
1624 1624
1625 1625 Revset aliases from the configuration are not expanded. To expand
1626 1626 user aliases, consider calling ``scmutil.revrange()`` or
1627 1627 ``repo.anyrevs([expr], user=True)``.
1628 1628
1629 1629 Returns a revset.abstractsmartset, which is a list-like interface
1630 1630 that contains integer revisions.
1631 1631 '''
1632 1632 tree = revsetlang.spectree(expr, *args)
1633 1633 return revset.makematcher(tree)(self)
1634 1634
1635 1635 def set(self, expr, *args):
1636 1636 '''Find revisions matching a revset and emit changectx instances.
1637 1637
1638 1638 This is a convenience wrapper around ``revs()`` that iterates the
1639 1639 result and is a generator of changectx instances.
1640 1640
1641 1641 Revset aliases from the configuration are not expanded. To expand
1642 1642 user aliases, consider calling ``scmutil.revrange()``.
1643 1643 '''
1644 1644 for r in self.revs(expr, *args):
1645 1645 yield self[r]
1646 1646
1647 1647 def anyrevs(self, specs, user=False, localalias=None):
1648 1648 '''Find revisions matching one of the given revsets.
1649 1649
1650 1650 Revset aliases from the configuration are not expanded by default. To
1651 1651 expand user aliases, specify ``user=True``. To provide some local
1652 1652 definitions overriding user aliases, set ``localalias`` to
1653 1653 ``{name: definitionstring}``.
1654 1654 '''
1655 if specs == [b'null']:
1656 return revset.baseset([nullrev])
1655 1657 if user:
1656 1658 m = revset.matchany(
1657 1659 self.ui,
1658 1660 specs,
1659 1661 lookup=revset.lookupfn(self),
1660 1662 localalias=localalias,
1661 1663 )
1662 1664 else:
1663 1665 m = revset.matchany(None, specs, localalias=localalias)
1664 1666 return m(self)
1665 1667
1666 1668 def url(self):
1667 1669 return b'file:' + self.root
1668 1670
1669 1671 def hook(self, name, throw=False, **args):
1670 1672 """Call a hook, passing this repo instance.
1671 1673
1672 1674 This a convenience method to aid invoking hooks. Extensions likely
1673 1675 won't call this unless they have registered a custom hook or are
1674 1676 replacing code that is expected to call a hook.
1675 1677 """
1676 1678 return hook.hook(self.ui, self, name, throw, **args)
1677 1679
1678 1680 @filteredpropertycache
1679 1681 def _tagscache(self):
1680 1682 '''Returns a tagscache object that contains various tags related
1681 1683 caches.'''
1682 1684
1683 1685 # This simplifies its cache management by having one decorated
1684 1686 # function (this one) and the rest simply fetch things from it.
1685 1687 class tagscache(object):
1686 1688 def __init__(self):
1687 1689 # These two define the set of tags for this repository. tags
1688 1690 # maps tag name to node; tagtypes maps tag name to 'global' or
1689 1691 # 'local'. (Global tags are defined by .hgtags across all
1690 1692 # heads, and local tags are defined in .hg/localtags.)
1691 1693 # They constitute the in-memory cache of tags.
1692 1694 self.tags = self.tagtypes = None
1693 1695
1694 1696 self.nodetagscache = self.tagslist = None
1695 1697
1696 1698 cache = tagscache()
1697 1699 cache.tags, cache.tagtypes = self._findtags()
1698 1700
1699 1701 return cache
1700 1702
1701 1703 def tags(self):
1702 1704 '''return a mapping of tag to node'''
1703 1705 t = {}
1704 1706 if self.changelog.filteredrevs:
1705 1707 tags, tt = self._findtags()
1706 1708 else:
1707 1709 tags = self._tagscache.tags
1708 1710 rev = self.changelog.rev
1709 1711 for k, v in pycompat.iteritems(tags):
1710 1712 try:
1711 1713 # ignore tags to unknown nodes
1712 1714 rev(v)
1713 1715 t[k] = v
1714 1716 except (error.LookupError, ValueError):
1715 1717 pass
1716 1718 return t
1717 1719
1718 1720 def _findtags(self):
1719 1721 '''Do the hard work of finding tags. Return a pair of dicts
1720 1722 (tags, tagtypes) where tags maps tag name to node, and tagtypes
1721 1723 maps tag name to a string like \'global\' or \'local\'.
1722 1724 Subclasses or extensions are free to add their own tags, but
1723 1725 should be aware that the returned dicts will be retained for the
1724 1726 duration of the localrepo object.'''
1725 1727
1726 1728 # XXX what tagtype should subclasses/extensions use? Currently
1727 1729 # mq and bookmarks add tags, but do not set the tagtype at all.
1728 1730 # Should each extension invent its own tag type? Should there
1729 1731 # be one tagtype for all such "virtual" tags? Or is the status
1730 1732 # quo fine?
1731 1733
1732 1734 # map tag name to (node, hist)
1733 1735 alltags = tagsmod.findglobaltags(self.ui, self)
1734 1736 # map tag name to tag type
1735 1737 tagtypes = dict((tag, b'global') for tag in alltags)
1736 1738
1737 1739 tagsmod.readlocaltags(self.ui, self, alltags, tagtypes)
1738 1740
1739 1741 # Build the return dicts. Have to re-encode tag names because
1740 1742 # the tags module always uses UTF-8 (in order not to lose info
1741 1743 # writing to the cache), but the rest of Mercurial wants them in
1742 1744 # local encoding.
1743 1745 tags = {}
1744 1746 for (name, (node, hist)) in pycompat.iteritems(alltags):
1745 1747 if node != nullid:
1746 1748 tags[encoding.tolocal(name)] = node
1747 1749 tags[b'tip'] = self.changelog.tip()
1748 1750 tagtypes = dict(
1749 1751 [
1750 1752 (encoding.tolocal(name), value)
1751 1753 for (name, value) in pycompat.iteritems(tagtypes)
1752 1754 ]
1753 1755 )
1754 1756 return (tags, tagtypes)
1755 1757
1756 1758 def tagtype(self, tagname):
1757 1759 '''
1758 1760 return the type of the given tag. result can be:
1759 1761
1760 1762 'local' : a local tag
1761 1763 'global' : a global tag
1762 1764 None : tag does not exist
1763 1765 '''
1764 1766
1765 1767 return self._tagscache.tagtypes.get(tagname)
1766 1768
1767 1769 def tagslist(self):
1768 1770 '''return a list of tags ordered by revision'''
1769 1771 if not self._tagscache.tagslist:
1770 1772 l = []
1771 1773 for t, n in pycompat.iteritems(self.tags()):
1772 1774 l.append((self.changelog.rev(n), t, n))
1773 1775 self._tagscache.tagslist = [(t, n) for r, t, n in sorted(l)]
1774 1776
1775 1777 return self._tagscache.tagslist
1776 1778
1777 1779 def nodetags(self, node):
1778 1780 '''return the tags associated with a node'''
1779 1781 if not self._tagscache.nodetagscache:
1780 1782 nodetagscache = {}
1781 1783 for t, n in pycompat.iteritems(self._tagscache.tags):
1782 1784 nodetagscache.setdefault(n, []).append(t)
1783 1785 for tags in pycompat.itervalues(nodetagscache):
1784 1786 tags.sort()
1785 1787 self._tagscache.nodetagscache = nodetagscache
1786 1788 return self._tagscache.nodetagscache.get(node, [])
1787 1789
1788 1790 def nodebookmarks(self, node):
1789 1791 """return the list of bookmarks pointing to the specified node"""
1790 1792 return self._bookmarks.names(node)
1791 1793
1792 1794 def branchmap(self):
1793 1795 '''returns a dictionary {branch: [branchheads]} with branchheads
1794 1796 ordered by increasing revision number'''
1795 1797 return self._branchcaches[self]
1796 1798
1797 1799 @unfilteredmethod
1798 1800 def revbranchcache(self):
1799 1801 if not self._revbranchcache:
1800 1802 self._revbranchcache = branchmap.revbranchcache(self.unfiltered())
1801 1803 return self._revbranchcache
1802 1804
1803 1805 def branchtip(self, branch, ignoremissing=False):
1804 1806 '''return the tip node for a given branch
1805 1807
1806 1808 If ignoremissing is True, then this method will not raise an error.
1807 1809 This is helpful for callers that only expect None for a missing branch
1808 1810 (e.g. namespace).
1809 1811
1810 1812 '''
1811 1813 try:
1812 1814 return self.branchmap().branchtip(branch)
1813 1815 except KeyError:
1814 1816 if not ignoremissing:
1815 1817 raise error.RepoLookupError(_(b"unknown branch '%s'") % branch)
1816 1818 else:
1817 1819 pass
1818 1820
1819 1821 def lookup(self, key):
1820 1822 node = scmutil.revsymbol(self, key).node()
1821 1823 if node is None:
1822 1824 raise error.RepoLookupError(_(b"unknown revision '%s'") % key)
1823 1825 return node
1824 1826
1825 1827 def lookupbranch(self, key):
1826 1828 if self.branchmap().hasbranch(key):
1827 1829 return key
1828 1830
1829 1831 return scmutil.revsymbol(self, key).branch()
1830 1832
1831 1833 def known(self, nodes):
1832 1834 cl = self.changelog
1833 1835 get_rev = cl.index.get_rev
1834 1836 filtered = cl.filteredrevs
1835 1837 result = []
1836 1838 for n in nodes:
1837 1839 r = get_rev(n)
1838 1840 resp = not (r is None or r in filtered)
1839 1841 result.append(resp)
1840 1842 return result
1841 1843
1842 1844 def local(self):
1843 1845 return self
1844 1846
1845 1847 def publishing(self):
1846 1848 # it's safe (and desirable) to trust the publish flag unconditionally
1847 1849 # so that we don't finalize changes shared between users via ssh or nfs
1848 1850 return self.ui.configbool(b'phases', b'publish', untrusted=True)
1849 1851
1850 1852 def cancopy(self):
1851 1853 # so statichttprepo's override of local() works
1852 1854 if not self.local():
1853 1855 return False
1854 1856 if not self.publishing():
1855 1857 return True
1856 1858 # if publishing we can't copy if there is filtered content
1857 1859 return not self.filtered(b'visible').changelog.filteredrevs
1858 1860
1859 1861 def shared(self):
1860 1862 '''the type of shared repository (None if not shared)'''
1861 1863 if self.sharedpath != self.path:
1862 1864 return b'store'
1863 1865 return None
1864 1866
1865 1867 def wjoin(self, f, *insidef):
1866 1868 return self.vfs.reljoin(self.root, f, *insidef)
1867 1869
1868 1870 def setparents(self, p1, p2=nullid):
1869 1871 with self.dirstate.parentchange():
1870 1872 copies = self.dirstate.setparents(p1, p2)
1871 1873 pctx = self[p1]
1872 1874 if copies:
1873 1875 # Adjust copy records, the dirstate cannot do it, it
1874 1876 # requires access to parents manifests. Preserve them
1875 1877 # only for entries added to first parent.
1876 1878 for f in copies:
1877 1879 if f not in pctx and copies[f] in pctx:
1878 1880 self.dirstate.copy(copies[f], f)
1879 1881 if p2 == nullid:
1880 1882 for f, s in sorted(self.dirstate.copies().items()):
1881 1883 if f not in pctx and s not in pctx:
1882 1884 self.dirstate.copy(None, f)
1883 1885
1884 1886 def filectx(self, path, changeid=None, fileid=None, changectx=None):
1885 1887 """changeid must be a changeset revision, if specified.
1886 1888 fileid can be a file revision or node."""
1887 1889 return context.filectx(
1888 1890 self, path, changeid, fileid, changectx=changectx
1889 1891 )
1890 1892
1891 1893 def getcwd(self):
1892 1894 return self.dirstate.getcwd()
1893 1895
1894 1896 def pathto(self, f, cwd=None):
1895 1897 return self.dirstate.pathto(f, cwd)
1896 1898
1897 1899 def _loadfilter(self, filter):
1898 1900 if filter not in self._filterpats:
1899 1901 l = []
1900 1902 for pat, cmd in self.ui.configitems(filter):
1901 1903 if cmd == b'!':
1902 1904 continue
1903 1905 mf = matchmod.match(self.root, b'', [pat])
1904 1906 fn = None
1905 1907 params = cmd
1906 1908 for name, filterfn in pycompat.iteritems(self._datafilters):
1907 1909 if cmd.startswith(name):
1908 1910 fn = filterfn
1909 1911 params = cmd[len(name) :].lstrip()
1910 1912 break
1911 1913 if not fn:
1912 1914 fn = lambda s, c, **kwargs: procutil.filter(s, c)
1913 1915 fn.__name__ = 'commandfilter'
1914 1916 # Wrap old filters not supporting keyword arguments
1915 1917 if not pycompat.getargspec(fn)[2]:
1916 1918 oldfn = fn
1917 1919 fn = lambda s, c, oldfn=oldfn, **kwargs: oldfn(s, c)
1918 1920 fn.__name__ = 'compat-' + oldfn.__name__
1919 1921 l.append((mf, fn, params))
1920 1922 self._filterpats[filter] = l
1921 1923 return self._filterpats[filter]
1922 1924
1923 1925 def _filter(self, filterpats, filename, data):
1924 1926 for mf, fn, cmd in filterpats:
1925 1927 if mf(filename):
1926 1928 self.ui.debug(
1927 1929 b"filtering %s through %s\n"
1928 1930 % (filename, cmd or pycompat.sysbytes(fn.__name__))
1929 1931 )
1930 1932 data = fn(data, cmd, ui=self.ui, repo=self, filename=filename)
1931 1933 break
1932 1934
1933 1935 return data
1934 1936
1935 1937 @unfilteredpropertycache
1936 1938 def _encodefilterpats(self):
1937 1939 return self._loadfilter(b'encode')
1938 1940
1939 1941 @unfilteredpropertycache
1940 1942 def _decodefilterpats(self):
1941 1943 return self._loadfilter(b'decode')
1942 1944
1943 1945 def adddatafilter(self, name, filter):
1944 1946 self._datafilters[name] = filter
1945 1947
1946 1948 def wread(self, filename):
1947 1949 if self.wvfs.islink(filename):
1948 1950 data = self.wvfs.readlink(filename)
1949 1951 else:
1950 1952 data = self.wvfs.read(filename)
1951 1953 return self._filter(self._encodefilterpats, filename, data)
1952 1954
1953 1955 def wwrite(self, filename, data, flags, backgroundclose=False, **kwargs):
1954 1956 """write ``data`` into ``filename`` in the working directory
1955 1957
1956 1958 This returns length of written (maybe decoded) data.
1957 1959 """
1958 1960 data = self._filter(self._decodefilterpats, filename, data)
1959 1961 if b'l' in flags:
1960 1962 self.wvfs.symlink(data, filename)
1961 1963 else:
1962 1964 self.wvfs.write(
1963 1965 filename, data, backgroundclose=backgroundclose, **kwargs
1964 1966 )
1965 1967 if b'x' in flags:
1966 1968 self.wvfs.setflags(filename, False, True)
1967 1969 else:
1968 1970 self.wvfs.setflags(filename, False, False)
1969 1971 return len(data)
1970 1972
1971 1973 def wwritedata(self, filename, data):
1972 1974 return self._filter(self._decodefilterpats, filename, data)
1973 1975
1974 1976 def currenttransaction(self):
1975 1977 """return the current transaction or None if non exists"""
1976 1978 if self._transref:
1977 1979 tr = self._transref()
1978 1980 else:
1979 1981 tr = None
1980 1982
1981 1983 if tr and tr.running():
1982 1984 return tr
1983 1985 return None
1984 1986
1985 1987 def transaction(self, desc, report=None):
1986 1988 if self.ui.configbool(b'devel', b'all-warnings') or self.ui.configbool(
1987 1989 b'devel', b'check-locks'
1988 1990 ):
1989 1991 if self._currentlock(self._lockref) is None:
1990 1992 raise error.ProgrammingError(b'transaction requires locking')
1991 1993 tr = self.currenttransaction()
1992 1994 if tr is not None:
1993 1995 return tr.nest(name=desc)
1994 1996
1995 1997 # abort here if the journal already exists
1996 1998 if self.svfs.exists(b"journal"):
1997 1999 raise error.RepoError(
1998 2000 _(b"abandoned transaction found"),
1999 2001 hint=_(b"run 'hg recover' to clean up transaction"),
2000 2002 )
2001 2003
2002 2004 idbase = b"%.40f#%f" % (random.random(), time.time())
2003 2005 ha = hex(hashlib.sha1(idbase).digest())
2004 2006 txnid = b'TXN:' + ha
2005 2007 self.hook(b'pretxnopen', throw=True, txnname=desc, txnid=txnid)
2006 2008
2007 2009 self._writejournal(desc)
2008 2010 renames = [(vfs, x, undoname(x)) for vfs, x in self._journalfiles()]
2009 2011 if report:
2010 2012 rp = report
2011 2013 else:
2012 2014 rp = self.ui.warn
2013 2015 vfsmap = {b'plain': self.vfs, b'store': self.svfs} # root of .hg/
2014 2016 # we must avoid cyclic reference between repo and transaction.
2015 2017 reporef = weakref.ref(self)
2016 2018 # Code to track tag movement
2017 2019 #
2018 2020 # Since tags are all handled as file content, it is actually quite hard
2019 2021 # to track these movement from a code perspective. So we fallback to a
2020 2022 # tracking at the repository level. One could envision to track changes
2021 2023 # to the '.hgtags' file through changegroup apply but that fails to
2022 2024 # cope with case where transaction expose new heads without changegroup
2023 2025 # being involved (eg: phase movement).
2024 2026 #
2025 2027 # For now, We gate the feature behind a flag since this likely comes
2026 2028 # with performance impacts. The current code run more often than needed
2027 2029 # and do not use caches as much as it could. The current focus is on
2028 2030 # the behavior of the feature so we disable it by default. The flag
2029 2031 # will be removed when we are happy with the performance impact.
2030 2032 #
2031 2033 # Once this feature is no longer experimental move the following
2032 2034 # documentation to the appropriate help section:
2033 2035 #
2034 2036 # The ``HG_TAG_MOVED`` variable will be set if the transaction touched
2035 2037 # tags (new or changed or deleted tags). In addition the details of
2036 2038 # these changes are made available in a file at:
2037 2039 # ``REPOROOT/.hg/changes/tags.changes``.
2038 2040 # Make sure you check for HG_TAG_MOVED before reading that file as it
2039 2041 # might exist from a previous transaction even if no tag were touched
2040 2042 # in this one. Changes are recorded in a line base format::
2041 2043 #
2042 2044 # <action> <hex-node> <tag-name>\n
2043 2045 #
2044 2046 # Actions are defined as follow:
2045 2047 # "-R": tag is removed,
2046 2048 # "+A": tag is added,
2047 2049 # "-M": tag is moved (old value),
2048 2050 # "+M": tag is moved (new value),
2049 2051 tracktags = lambda x: None
2050 2052 # experimental config: experimental.hook-track-tags
2051 2053 shouldtracktags = self.ui.configbool(
2052 2054 b'experimental', b'hook-track-tags'
2053 2055 )
2054 2056 if desc != b'strip' and shouldtracktags:
2055 2057 oldheads = self.changelog.headrevs()
2056 2058
2057 2059 def tracktags(tr2):
2058 2060 repo = reporef()
2059 2061 oldfnodes = tagsmod.fnoderevs(repo.ui, repo, oldheads)
2060 2062 newheads = repo.changelog.headrevs()
2061 2063 newfnodes = tagsmod.fnoderevs(repo.ui, repo, newheads)
2062 2064 # notes: we compare lists here.
2063 2065 # As we do it only once buiding set would not be cheaper
2064 2066 changes = tagsmod.difftags(repo.ui, repo, oldfnodes, newfnodes)
2065 2067 if changes:
2066 2068 tr2.hookargs[b'tag_moved'] = b'1'
2067 2069 with repo.vfs(
2068 2070 b'changes/tags.changes', b'w', atomictemp=True
2069 2071 ) as changesfile:
2070 2072 # note: we do not register the file to the transaction
2071 2073 # because we needs it to still exist on the transaction
2072 2074 # is close (for txnclose hooks)
2073 2075 tagsmod.writediff(changesfile, changes)
2074 2076
2075 2077 def validate(tr2):
2076 2078 """will run pre-closing hooks"""
2077 2079 # XXX the transaction API is a bit lacking here so we take a hacky
2078 2080 # path for now
2079 2081 #
2080 2082 # We cannot add this as a "pending" hooks since the 'tr.hookargs'
2081 2083 # dict is copied before these run. In addition we needs the data
2082 2084 # available to in memory hooks too.
2083 2085 #
2084 2086 # Moreover, we also need to make sure this runs before txnclose
2085 2087 # hooks and there is no "pending" mechanism that would execute
2086 2088 # logic only if hooks are about to run.
2087 2089 #
2088 2090 # Fixing this limitation of the transaction is also needed to track
2089 2091 # other families of changes (bookmarks, phases, obsolescence).
2090 2092 #
2091 2093 # This will have to be fixed before we remove the experimental
2092 2094 # gating.
2093 2095 tracktags(tr2)
2094 2096 repo = reporef()
2095 2097
2096 2098 singleheadopt = (b'experimental', b'single-head-per-branch')
2097 2099 singlehead = repo.ui.configbool(*singleheadopt)
2098 2100 if singlehead:
2099 2101 singleheadsub = repo.ui.configsuboptions(*singleheadopt)[1]
2100 2102 accountclosed = singleheadsub.get(
2101 2103 b"account-closed-heads", False
2102 2104 )
2103 2105 scmutil.enforcesinglehead(repo, tr2, desc, accountclosed)
2104 2106 if hook.hashook(repo.ui, b'pretxnclose-bookmark'):
2105 2107 for name, (old, new) in sorted(
2106 2108 tr.changes[b'bookmarks'].items()
2107 2109 ):
2108 2110 args = tr.hookargs.copy()
2109 2111 args.update(bookmarks.preparehookargs(name, old, new))
2110 2112 repo.hook(
2111 2113 b'pretxnclose-bookmark',
2112 2114 throw=True,
2113 2115 **pycompat.strkwargs(args)
2114 2116 )
2115 2117 if hook.hashook(repo.ui, b'pretxnclose-phase'):
2116 2118 cl = repo.unfiltered().changelog
2117 2119 for rev, (old, new) in tr.changes[b'phases'].items():
2118 2120 args = tr.hookargs.copy()
2119 2121 node = hex(cl.node(rev))
2120 2122 args.update(phases.preparehookargs(node, old, new))
2121 2123 repo.hook(
2122 2124 b'pretxnclose-phase',
2123 2125 throw=True,
2124 2126 **pycompat.strkwargs(args)
2125 2127 )
2126 2128
2127 2129 repo.hook(
2128 2130 b'pretxnclose', throw=True, **pycompat.strkwargs(tr.hookargs)
2129 2131 )
2130 2132
2131 2133 def releasefn(tr, success):
2132 2134 repo = reporef()
2133 2135 if repo is None:
2134 2136 # If the repo has been GC'd (and this release function is being
2135 2137 # called from transaction.__del__), there's not much we can do,
2136 2138 # so just leave the unfinished transaction there and let the
2137 2139 # user run `hg recover`.
2138 2140 return
2139 2141 if success:
2140 2142 # this should be explicitly invoked here, because
2141 2143 # in-memory changes aren't written out at closing
2142 2144 # transaction, if tr.addfilegenerator (via
2143 2145 # dirstate.write or so) isn't invoked while
2144 2146 # transaction running
2145 2147 repo.dirstate.write(None)
2146 2148 else:
2147 2149 # discard all changes (including ones already written
2148 2150 # out) in this transaction
2149 2151 narrowspec.restorebackup(self, b'journal.narrowspec')
2150 2152 narrowspec.restorewcbackup(self, b'journal.narrowspec.dirstate')
2151 2153 repo.dirstate.restorebackup(None, b'journal.dirstate')
2152 2154
2153 2155 repo.invalidate(clearfilecache=True)
2154 2156
2155 2157 tr = transaction.transaction(
2156 2158 rp,
2157 2159 self.svfs,
2158 2160 vfsmap,
2159 2161 b"journal",
2160 2162 b"undo",
2161 2163 aftertrans(renames),
2162 2164 self.store.createmode,
2163 2165 validator=validate,
2164 2166 releasefn=releasefn,
2165 2167 checkambigfiles=_cachedfiles,
2166 2168 name=desc,
2167 2169 )
2168 2170 tr.changes[b'origrepolen'] = len(self)
2169 2171 tr.changes[b'obsmarkers'] = set()
2170 2172 tr.changes[b'phases'] = {}
2171 2173 tr.changes[b'bookmarks'] = {}
2172 2174
2173 2175 tr.hookargs[b'txnid'] = txnid
2174 2176 tr.hookargs[b'txnname'] = desc
2175 2177 # note: writing the fncache only during finalize mean that the file is
2176 2178 # outdated when running hooks. As fncache is used for streaming clone,
2177 2179 # this is not expected to break anything that happen during the hooks.
2178 2180 tr.addfinalize(b'flush-fncache', self.store.write)
2179 2181
2180 2182 def txnclosehook(tr2):
2181 2183 """To be run if transaction is successful, will schedule a hook run
2182 2184 """
2183 2185 # Don't reference tr2 in hook() so we don't hold a reference.
2184 2186 # This reduces memory consumption when there are multiple
2185 2187 # transactions per lock. This can likely go away if issue5045
2186 2188 # fixes the function accumulation.
2187 2189 hookargs = tr2.hookargs
2188 2190
2189 2191 def hookfunc():
2190 2192 repo = reporef()
2191 2193 if hook.hashook(repo.ui, b'txnclose-bookmark'):
2192 2194 bmchanges = sorted(tr.changes[b'bookmarks'].items())
2193 2195 for name, (old, new) in bmchanges:
2194 2196 args = tr.hookargs.copy()
2195 2197 args.update(bookmarks.preparehookargs(name, old, new))
2196 2198 repo.hook(
2197 2199 b'txnclose-bookmark',
2198 2200 throw=False,
2199 2201 **pycompat.strkwargs(args)
2200 2202 )
2201 2203
2202 2204 if hook.hashook(repo.ui, b'txnclose-phase'):
2203 2205 cl = repo.unfiltered().changelog
2204 2206 phasemv = sorted(tr.changes[b'phases'].items())
2205 2207 for rev, (old, new) in phasemv:
2206 2208 args = tr.hookargs.copy()
2207 2209 node = hex(cl.node(rev))
2208 2210 args.update(phases.preparehookargs(node, old, new))
2209 2211 repo.hook(
2210 2212 b'txnclose-phase',
2211 2213 throw=False,
2212 2214 **pycompat.strkwargs(args)
2213 2215 )
2214 2216
2215 2217 repo.hook(
2216 2218 b'txnclose', throw=False, **pycompat.strkwargs(hookargs)
2217 2219 )
2218 2220
2219 2221 reporef()._afterlock(hookfunc)
2220 2222
2221 2223 tr.addfinalize(b'txnclose-hook', txnclosehook)
2222 2224 # Include a leading "-" to make it happen before the transaction summary
2223 2225 # reports registered via scmutil.registersummarycallback() whose names
2224 2226 # are 00-txnreport etc. That way, the caches will be warm when the
2225 2227 # callbacks run.
2226 2228 tr.addpostclose(b'-warm-cache', self._buildcacheupdater(tr))
2227 2229
2228 2230 def txnaborthook(tr2):
2229 2231 """To be run if transaction is aborted
2230 2232 """
2231 2233 reporef().hook(
2232 2234 b'txnabort', throw=False, **pycompat.strkwargs(tr2.hookargs)
2233 2235 )
2234 2236
2235 2237 tr.addabort(b'txnabort-hook', txnaborthook)
2236 2238 # avoid eager cache invalidation. in-memory data should be identical
2237 2239 # to stored data if transaction has no error.
2238 2240 tr.addpostclose(b'refresh-filecachestats', self._refreshfilecachestats)
2239 2241 self._transref = weakref.ref(tr)
2240 2242 scmutil.registersummarycallback(self, tr, desc)
2241 2243 return tr
2242 2244
2243 2245 def _journalfiles(self):
2244 2246 return (
2245 2247 (self.svfs, b'journal'),
2246 2248 (self.svfs, b'journal.narrowspec'),
2247 2249 (self.vfs, b'journal.narrowspec.dirstate'),
2248 2250 (self.vfs, b'journal.dirstate'),
2249 2251 (self.vfs, b'journal.branch'),
2250 2252 (self.vfs, b'journal.desc'),
2251 2253 (bookmarks.bookmarksvfs(self), b'journal.bookmarks'),
2252 2254 (self.svfs, b'journal.phaseroots'),
2253 2255 )
2254 2256
2255 2257 def undofiles(self):
2256 2258 return [(vfs, undoname(x)) for vfs, x in self._journalfiles()]
2257 2259
2258 2260 @unfilteredmethod
2259 2261 def _writejournal(self, desc):
2260 2262 self.dirstate.savebackup(None, b'journal.dirstate')
2261 2263 narrowspec.savewcbackup(self, b'journal.narrowspec.dirstate')
2262 2264 narrowspec.savebackup(self, b'journal.narrowspec')
2263 2265 self.vfs.write(
2264 2266 b"journal.branch", encoding.fromlocal(self.dirstate.branch())
2265 2267 )
2266 2268 self.vfs.write(b"journal.desc", b"%d\n%s\n" % (len(self), desc))
2267 2269 bookmarksvfs = bookmarks.bookmarksvfs(self)
2268 2270 bookmarksvfs.write(
2269 2271 b"journal.bookmarks", bookmarksvfs.tryread(b"bookmarks")
2270 2272 )
2271 2273 self.svfs.write(b"journal.phaseroots", self.svfs.tryread(b"phaseroots"))
2272 2274
2273 2275 def recover(self):
2274 2276 with self.lock():
2275 2277 if self.svfs.exists(b"journal"):
2276 2278 self.ui.status(_(b"rolling back interrupted transaction\n"))
2277 2279 vfsmap = {
2278 2280 b'': self.svfs,
2279 2281 b'plain': self.vfs,
2280 2282 }
2281 2283 transaction.rollback(
2282 2284 self.svfs,
2283 2285 vfsmap,
2284 2286 b"journal",
2285 2287 self.ui.warn,
2286 2288 checkambigfiles=_cachedfiles,
2287 2289 )
2288 2290 self.invalidate()
2289 2291 return True
2290 2292 else:
2291 2293 self.ui.warn(_(b"no interrupted transaction available\n"))
2292 2294 return False
2293 2295
2294 2296 def rollback(self, dryrun=False, force=False):
2295 2297 wlock = lock = dsguard = None
2296 2298 try:
2297 2299 wlock = self.wlock()
2298 2300 lock = self.lock()
2299 2301 if self.svfs.exists(b"undo"):
2300 2302 dsguard = dirstateguard.dirstateguard(self, b'rollback')
2301 2303
2302 2304 return self._rollback(dryrun, force, dsguard)
2303 2305 else:
2304 2306 self.ui.warn(_(b"no rollback information available\n"))
2305 2307 return 1
2306 2308 finally:
2307 2309 release(dsguard, lock, wlock)
2308 2310
2309 2311 @unfilteredmethod # Until we get smarter cache management
2310 2312 def _rollback(self, dryrun, force, dsguard):
2311 2313 ui = self.ui
2312 2314 try:
2313 2315 args = self.vfs.read(b'undo.desc').splitlines()
2314 2316 (oldlen, desc, detail) = (int(args[0]), args[1], None)
2315 2317 if len(args) >= 3:
2316 2318 detail = args[2]
2317 2319 oldtip = oldlen - 1
2318 2320
2319 2321 if detail and ui.verbose:
2320 2322 msg = _(
2321 2323 b'repository tip rolled back to revision %d'
2322 2324 b' (undo %s: %s)\n'
2323 2325 ) % (oldtip, desc, detail)
2324 2326 else:
2325 2327 msg = _(
2326 2328 b'repository tip rolled back to revision %d (undo %s)\n'
2327 2329 ) % (oldtip, desc)
2328 2330 except IOError:
2329 2331 msg = _(b'rolling back unknown transaction\n')
2330 2332 desc = None
2331 2333
2332 2334 if not force and self[b'.'] != self[b'tip'] and desc == b'commit':
2333 2335 raise error.Abort(
2334 2336 _(
2335 2337 b'rollback of last commit while not checked out '
2336 2338 b'may lose data'
2337 2339 ),
2338 2340 hint=_(b'use -f to force'),
2339 2341 )
2340 2342
2341 2343 ui.status(msg)
2342 2344 if dryrun:
2343 2345 return 0
2344 2346
2345 2347 parents = self.dirstate.parents()
2346 2348 self.destroying()
2347 2349 vfsmap = {b'plain': self.vfs, b'': self.svfs}
2348 2350 transaction.rollback(
2349 2351 self.svfs, vfsmap, b'undo', ui.warn, checkambigfiles=_cachedfiles
2350 2352 )
2351 2353 bookmarksvfs = bookmarks.bookmarksvfs(self)
2352 2354 if bookmarksvfs.exists(b'undo.bookmarks'):
2353 2355 bookmarksvfs.rename(
2354 2356 b'undo.bookmarks', b'bookmarks', checkambig=True
2355 2357 )
2356 2358 if self.svfs.exists(b'undo.phaseroots'):
2357 2359 self.svfs.rename(b'undo.phaseroots', b'phaseroots', checkambig=True)
2358 2360 self.invalidate()
2359 2361
2360 2362 has_node = self.changelog.index.has_node
2361 2363 parentgone = any(not has_node(p) for p in parents)
2362 2364 if parentgone:
2363 2365 # prevent dirstateguard from overwriting already restored one
2364 2366 dsguard.close()
2365 2367
2366 2368 narrowspec.restorebackup(self, b'undo.narrowspec')
2367 2369 narrowspec.restorewcbackup(self, b'undo.narrowspec.dirstate')
2368 2370 self.dirstate.restorebackup(None, b'undo.dirstate')
2369 2371 try:
2370 2372 branch = self.vfs.read(b'undo.branch')
2371 2373 self.dirstate.setbranch(encoding.tolocal(branch))
2372 2374 except IOError:
2373 2375 ui.warn(
2374 2376 _(
2375 2377 b'named branch could not be reset: '
2376 2378 b'current branch is still \'%s\'\n'
2377 2379 )
2378 2380 % self.dirstate.branch()
2379 2381 )
2380 2382
2381 2383 parents = tuple([p.rev() for p in self[None].parents()])
2382 2384 if len(parents) > 1:
2383 2385 ui.status(
2384 2386 _(
2385 2387 b'working directory now based on '
2386 2388 b'revisions %d and %d\n'
2387 2389 )
2388 2390 % parents
2389 2391 )
2390 2392 else:
2391 2393 ui.status(
2392 2394 _(b'working directory now based on revision %d\n') % parents
2393 2395 )
2394 2396 mergemod.mergestate.clean(self, self[b'.'].node())
2395 2397
2396 2398 # TODO: if we know which new heads may result from this rollback, pass
2397 2399 # them to destroy(), which will prevent the branchhead cache from being
2398 2400 # invalidated.
2399 2401 self.destroyed()
2400 2402 return 0
2401 2403
2402 2404 def _buildcacheupdater(self, newtransaction):
2403 2405 """called during transaction to build the callback updating cache
2404 2406
2405 2407 Lives on the repository to help extension who might want to augment
2406 2408 this logic. For this purpose, the created transaction is passed to the
2407 2409 method.
2408 2410 """
2409 2411 # we must avoid cyclic reference between repo and transaction.
2410 2412 reporef = weakref.ref(self)
2411 2413
2412 2414 def updater(tr):
2413 2415 repo = reporef()
2414 2416 repo.updatecaches(tr)
2415 2417
2416 2418 return updater
2417 2419
2418 2420 @unfilteredmethod
2419 2421 def updatecaches(self, tr=None, full=False):
2420 2422 """warm appropriate caches
2421 2423
2422 2424 If this function is called after a transaction closed. The transaction
2423 2425 will be available in the 'tr' argument. This can be used to selectively
2424 2426 update caches relevant to the changes in that transaction.
2425 2427
2426 2428 If 'full' is set, make sure all caches the function knows about have
2427 2429 up-to-date data. Even the ones usually loaded more lazily.
2428 2430 """
2429 2431 if tr is not None and tr.hookargs.get(b'source') == b'strip':
2430 2432 # During strip, many caches are invalid but
2431 2433 # later call to `destroyed` will refresh them.
2432 2434 return
2433 2435
2434 2436 if tr is None or tr.changes[b'origrepolen'] < len(self):
2435 2437 # accessing the 'ser ved' branchmap should refresh all the others,
2436 2438 self.ui.debug(b'updating the branch cache\n')
2437 2439 self.filtered(b'served').branchmap()
2438 2440 self.filtered(b'served.hidden').branchmap()
2439 2441
2440 2442 if full:
2441 2443 unfi = self.unfiltered()
2442 2444 rbc = unfi.revbranchcache()
2443 2445 for r in unfi.changelog:
2444 2446 rbc.branchinfo(r)
2445 2447 rbc.write()
2446 2448
2447 2449 # ensure the working copy parents are in the manifestfulltextcache
2448 2450 for ctx in self[b'.'].parents():
2449 2451 ctx.manifest() # accessing the manifest is enough
2450 2452
2451 2453 # accessing fnode cache warms the cache
2452 2454 tagsmod.fnoderevs(self.ui, unfi, unfi.changelog.revs())
2453 2455 # accessing tags warm the cache
2454 2456 self.tags()
2455 2457 self.filtered(b'served').tags()
2456 2458
2457 2459 # The `full` arg is documented as updating even the lazily-loaded
2458 2460 # caches immediately, so we're forcing a write to cause these caches
2459 2461 # to be warmed up even if they haven't explicitly been requested
2460 2462 # yet (if they've never been used by hg, they won't ever have been
2461 2463 # written, even if they're a subset of another kind of cache that
2462 2464 # *has* been used).
2463 2465 for filt in repoview.filtertable.keys():
2464 2466 filtered = self.filtered(filt)
2465 2467 filtered.branchmap().write(filtered)
2466 2468
2467 2469 def invalidatecaches(self):
2468 2470
2469 2471 if '_tagscache' in vars(self):
2470 2472 # can't use delattr on proxy
2471 2473 del self.__dict__['_tagscache']
2472 2474
2473 2475 self._branchcaches.clear()
2474 2476 self.invalidatevolatilesets()
2475 2477 self._sparsesignaturecache.clear()
2476 2478
2477 2479 def invalidatevolatilesets(self):
2478 2480 self.filteredrevcache.clear()
2479 2481 obsolete.clearobscaches(self)
2480 2482
2481 2483 def invalidatedirstate(self):
2482 2484 '''Invalidates the dirstate, causing the next call to dirstate
2483 2485 to check if it was modified since the last time it was read,
2484 2486 rereading it if it has.
2485 2487
2486 2488 This is different to dirstate.invalidate() that it doesn't always
2487 2489 rereads the dirstate. Use dirstate.invalidate() if you want to
2488 2490 explicitly read the dirstate again (i.e. restoring it to a previous
2489 2491 known good state).'''
2490 2492 if hasunfilteredcache(self, 'dirstate'):
2491 2493 for k in self.dirstate._filecache:
2492 2494 try:
2493 2495 delattr(self.dirstate, k)
2494 2496 except AttributeError:
2495 2497 pass
2496 2498 delattr(self.unfiltered(), 'dirstate')
2497 2499
2498 2500 def invalidate(self, clearfilecache=False):
2499 2501 '''Invalidates both store and non-store parts other than dirstate
2500 2502
2501 2503 If a transaction is running, invalidation of store is omitted,
2502 2504 because discarding in-memory changes might cause inconsistency
2503 2505 (e.g. incomplete fncache causes unintentional failure, but
2504 2506 redundant one doesn't).
2505 2507 '''
2506 2508 unfiltered = self.unfiltered() # all file caches are stored unfiltered
2507 2509 for k in list(self._filecache.keys()):
2508 2510 # dirstate is invalidated separately in invalidatedirstate()
2509 2511 if k == b'dirstate':
2510 2512 continue
2511 2513 if (
2512 2514 k == b'changelog'
2513 2515 and self.currenttransaction()
2514 2516 and self.changelog._delayed
2515 2517 ):
2516 2518 # The changelog object may store unwritten revisions. We don't
2517 2519 # want to lose them.
2518 2520 # TODO: Solve the problem instead of working around it.
2519 2521 continue
2520 2522
2521 2523 if clearfilecache:
2522 2524 del self._filecache[k]
2523 2525 try:
2524 2526 delattr(unfiltered, k)
2525 2527 except AttributeError:
2526 2528 pass
2527 2529 self.invalidatecaches()
2528 2530 if not self.currenttransaction():
2529 2531 # TODO: Changing contents of store outside transaction
2530 2532 # causes inconsistency. We should make in-memory store
2531 2533 # changes detectable, and abort if changed.
2532 2534 self.store.invalidatecaches()
2533 2535
2534 2536 def invalidateall(self):
2535 2537 '''Fully invalidates both store and non-store parts, causing the
2536 2538 subsequent operation to reread any outside changes.'''
2537 2539 # extension should hook this to invalidate its caches
2538 2540 self.invalidate()
2539 2541 self.invalidatedirstate()
2540 2542
2541 2543 @unfilteredmethod
2542 2544 def _refreshfilecachestats(self, tr):
2543 2545 """Reload stats of cached files so that they are flagged as valid"""
2544 2546 for k, ce in self._filecache.items():
2545 2547 k = pycompat.sysstr(k)
2546 2548 if k == 'dirstate' or k not in self.__dict__:
2547 2549 continue
2548 2550 ce.refresh()
2549 2551
2550 2552 def _lock(
2551 2553 self,
2552 2554 vfs,
2553 2555 lockname,
2554 2556 wait,
2555 2557 releasefn,
2556 2558 acquirefn,
2557 2559 desc,
2558 2560 inheritchecker=None,
2559 2561 parentenvvar=None,
2560 2562 ):
2561 2563 parentlock = None
2562 2564 # the contents of parentenvvar are used by the underlying lock to
2563 2565 # determine whether it can be inherited
2564 2566 if parentenvvar is not None:
2565 2567 parentlock = encoding.environ.get(parentenvvar)
2566 2568
2567 2569 timeout = 0
2568 2570 warntimeout = 0
2569 2571 if wait:
2570 2572 timeout = self.ui.configint(b"ui", b"timeout")
2571 2573 warntimeout = self.ui.configint(b"ui", b"timeout.warn")
2572 2574 # internal config: ui.signal-safe-lock
2573 2575 signalsafe = self.ui.configbool(b'ui', b'signal-safe-lock')
2574 2576
2575 2577 l = lockmod.trylock(
2576 2578 self.ui,
2577 2579 vfs,
2578 2580 lockname,
2579 2581 timeout,
2580 2582 warntimeout,
2581 2583 releasefn=releasefn,
2582 2584 acquirefn=acquirefn,
2583 2585 desc=desc,
2584 2586 inheritchecker=inheritchecker,
2585 2587 parentlock=parentlock,
2586 2588 signalsafe=signalsafe,
2587 2589 )
2588 2590 return l
2589 2591
2590 2592 def _afterlock(self, callback):
2591 2593 """add a callback to be run when the repository is fully unlocked
2592 2594
2593 2595 The callback will be executed when the outermost lock is released
2594 2596 (with wlock being higher level than 'lock')."""
2595 2597 for ref in (self._wlockref, self._lockref):
2596 2598 l = ref and ref()
2597 2599 if l and l.held:
2598 2600 l.postrelease.append(callback)
2599 2601 break
2600 2602 else: # no lock have been found.
2601 2603 callback()
2602 2604
2603 2605 def lock(self, wait=True):
2604 2606 '''Lock the repository store (.hg/store) and return a weak reference
2605 2607 to the lock. Use this before modifying the store (e.g. committing or
2606 2608 stripping). If you are opening a transaction, get a lock as well.)
2607 2609
2608 2610 If both 'lock' and 'wlock' must be acquired, ensure you always acquires
2609 2611 'wlock' first to avoid a dead-lock hazard.'''
2610 2612 l = self._currentlock(self._lockref)
2611 2613 if l is not None:
2612 2614 l.lock()
2613 2615 return l
2614 2616
2615 2617 l = self._lock(
2616 2618 vfs=self.svfs,
2617 2619 lockname=b"lock",
2618 2620 wait=wait,
2619 2621 releasefn=None,
2620 2622 acquirefn=self.invalidate,
2621 2623 desc=_(b'repository %s') % self.origroot,
2622 2624 )
2623 2625 self._lockref = weakref.ref(l)
2624 2626 return l
2625 2627
2626 2628 def _wlockchecktransaction(self):
2627 2629 if self.currenttransaction() is not None:
2628 2630 raise error.LockInheritanceContractViolation(
2629 2631 b'wlock cannot be inherited in the middle of a transaction'
2630 2632 )
2631 2633
2632 2634 def wlock(self, wait=True):
2633 2635 '''Lock the non-store parts of the repository (everything under
2634 2636 .hg except .hg/store) and return a weak reference to the lock.
2635 2637
2636 2638 Use this before modifying files in .hg.
2637 2639
2638 2640 If both 'lock' and 'wlock' must be acquired, ensure you always acquires
2639 2641 'wlock' first to avoid a dead-lock hazard.'''
2640 2642 l = self._wlockref and self._wlockref()
2641 2643 if l is not None and l.held:
2642 2644 l.lock()
2643 2645 return l
2644 2646
2645 2647 # We do not need to check for non-waiting lock acquisition. Such
2646 2648 # acquisition would not cause dead-lock as they would just fail.
2647 2649 if wait and (
2648 2650 self.ui.configbool(b'devel', b'all-warnings')
2649 2651 or self.ui.configbool(b'devel', b'check-locks')
2650 2652 ):
2651 2653 if self._currentlock(self._lockref) is not None:
2652 2654 self.ui.develwarn(b'"wlock" acquired after "lock"')
2653 2655
2654 2656 def unlock():
2655 2657 if self.dirstate.pendingparentchange():
2656 2658 self.dirstate.invalidate()
2657 2659 else:
2658 2660 self.dirstate.write(None)
2659 2661
2660 2662 self._filecache[b'dirstate'].refresh()
2661 2663
2662 2664 l = self._lock(
2663 2665 self.vfs,
2664 2666 b"wlock",
2665 2667 wait,
2666 2668 unlock,
2667 2669 self.invalidatedirstate,
2668 2670 _(b'working directory of %s') % self.origroot,
2669 2671 inheritchecker=self._wlockchecktransaction,
2670 2672 parentenvvar=b'HG_WLOCK_LOCKER',
2671 2673 )
2672 2674 self._wlockref = weakref.ref(l)
2673 2675 return l
2674 2676
2675 2677 def _currentlock(self, lockref):
2676 2678 """Returns the lock if it's held, or None if it's not."""
2677 2679 if lockref is None:
2678 2680 return None
2679 2681 l = lockref()
2680 2682 if l is None or not l.held:
2681 2683 return None
2682 2684 return l
2683 2685
2684 2686 def currentwlock(self):
2685 2687 """Returns the wlock if it's held, or None if it's not."""
2686 2688 return self._currentlock(self._wlockref)
2687 2689
2688 2690 def _filecommit(
2689 2691 self,
2690 2692 fctx,
2691 2693 manifest1,
2692 2694 manifest2,
2693 2695 linkrev,
2694 2696 tr,
2695 2697 changelist,
2696 2698 includecopymeta,
2697 2699 ):
2698 2700 """
2699 2701 commit an individual file as part of a larger transaction
2700 2702 """
2701 2703
2702 2704 fname = fctx.path()
2703 2705 fparent1 = manifest1.get(fname, nullid)
2704 2706 fparent2 = manifest2.get(fname, nullid)
2705 2707 if isinstance(fctx, context.filectx):
2706 2708 node = fctx.filenode()
2707 2709 if node in [fparent1, fparent2]:
2708 2710 self.ui.debug(b'reusing %s filelog entry\n' % fname)
2709 2711 if (
2710 2712 fparent1 != nullid
2711 2713 and manifest1.flags(fname) != fctx.flags()
2712 2714 ) or (
2713 2715 fparent2 != nullid
2714 2716 and manifest2.flags(fname) != fctx.flags()
2715 2717 ):
2716 2718 changelist.append(fname)
2717 2719 return node
2718 2720
2719 2721 flog = self.file(fname)
2720 2722 meta = {}
2721 2723 cfname = fctx.copysource()
2722 2724 if cfname and cfname != fname:
2723 2725 # Mark the new revision of this file as a copy of another
2724 2726 # file. This copy data will effectively act as a parent
2725 2727 # of this new revision. If this is a merge, the first
2726 2728 # parent will be the nullid (meaning "look up the copy data")
2727 2729 # and the second one will be the other parent. For example:
2728 2730 #
2729 2731 # 0 --- 1 --- 3 rev1 changes file foo
2730 2732 # \ / rev2 renames foo to bar and changes it
2731 2733 # \- 2 -/ rev3 should have bar with all changes and
2732 2734 # should record that bar descends from
2733 2735 # bar in rev2 and foo in rev1
2734 2736 #
2735 2737 # this allows this merge to succeed:
2736 2738 #
2737 2739 # 0 --- 1 --- 3 rev4 reverts the content change from rev2
2738 2740 # \ / merging rev3 and rev4 should use bar@rev2
2739 2741 # \- 2 --- 4 as the merge base
2740 2742 #
2741 2743
2742 2744 cnode = manifest1.get(cfname)
2743 2745 newfparent = fparent2
2744 2746
2745 2747 if manifest2: # branch merge
2746 2748 if fparent2 == nullid or cnode is None: # copied on remote side
2747 2749 if cfname in manifest2:
2748 2750 cnode = manifest2[cfname]
2749 2751 newfparent = fparent1
2750 2752
2751 2753 # Here, we used to search backwards through history to try to find
2752 2754 # where the file copy came from if the source of a copy was not in
2753 2755 # the parent directory. However, this doesn't actually make sense to
2754 2756 # do (what does a copy from something not in your working copy even
2755 2757 # mean?) and it causes bugs (eg, issue4476). Instead, we will warn
2756 2758 # the user that copy information was dropped, so if they didn't
2757 2759 # expect this outcome it can be fixed, but this is the correct
2758 2760 # behavior in this circumstance.
2759 2761
2760 2762 if cnode:
2761 2763 self.ui.debug(
2762 2764 b" %s: copy %s:%s\n" % (fname, cfname, hex(cnode))
2763 2765 )
2764 2766 if includecopymeta:
2765 2767 meta[b"copy"] = cfname
2766 2768 meta[b"copyrev"] = hex(cnode)
2767 2769 fparent1, fparent2 = nullid, newfparent
2768 2770 else:
2769 2771 self.ui.warn(
2770 2772 _(
2771 2773 b"warning: can't find ancestor for '%s' "
2772 2774 b"copied from '%s'!\n"
2773 2775 )
2774 2776 % (fname, cfname)
2775 2777 )
2776 2778
2777 2779 elif fparent1 == nullid:
2778 2780 fparent1, fparent2 = fparent2, nullid
2779 2781 elif fparent2 != nullid:
2780 2782 # is one parent an ancestor of the other?
2781 2783 fparentancestors = flog.commonancestorsheads(fparent1, fparent2)
2782 2784 if fparent1 in fparentancestors:
2783 2785 fparent1, fparent2 = fparent2, nullid
2784 2786 elif fparent2 in fparentancestors:
2785 2787 fparent2 = nullid
2786 2788
2787 2789 # is the file changed?
2788 2790 text = fctx.data()
2789 2791 if fparent2 != nullid or flog.cmp(fparent1, text) or meta:
2790 2792 changelist.append(fname)
2791 2793 return flog.add(text, meta, tr, linkrev, fparent1, fparent2)
2792 2794 # are just the flags changed during merge?
2793 2795 elif fname in manifest1 and manifest1.flags(fname) != fctx.flags():
2794 2796 changelist.append(fname)
2795 2797
2796 2798 return fparent1
2797 2799
2798 2800 def checkcommitpatterns(self, wctx, match, status, fail):
2799 2801 """check for commit arguments that aren't committable"""
2800 2802 if match.isexact() or match.prefix():
2801 2803 matched = set(status.modified + status.added + status.removed)
2802 2804
2803 2805 for f in match.files():
2804 2806 f = self.dirstate.normalize(f)
2805 2807 if f == b'.' or f in matched or f in wctx.substate:
2806 2808 continue
2807 2809 if f in status.deleted:
2808 2810 fail(f, _(b'file not found!'))
2809 2811 # Is it a directory that exists or used to exist?
2810 2812 if self.wvfs.isdir(f) or wctx.p1().hasdir(f):
2811 2813 d = f + b'/'
2812 2814 for mf in matched:
2813 2815 if mf.startswith(d):
2814 2816 break
2815 2817 else:
2816 2818 fail(f, _(b"no match under directory!"))
2817 2819 elif f not in self.dirstate:
2818 2820 fail(f, _(b"file not tracked!"))
2819 2821
2820 2822 @unfilteredmethod
2821 2823 def commit(
2822 2824 self,
2823 2825 text=b"",
2824 2826 user=None,
2825 2827 date=None,
2826 2828 match=None,
2827 2829 force=False,
2828 2830 editor=False,
2829 2831 extra=None,
2830 2832 ):
2831 2833 """Add a new revision to current repository.
2832 2834
2833 2835 Revision information is gathered from the working directory,
2834 2836 match can be used to filter the committed files. If editor is
2835 2837 supplied, it is called to get a commit message.
2836 2838 """
2837 2839 if extra is None:
2838 2840 extra = {}
2839 2841
2840 2842 def fail(f, msg):
2841 2843 raise error.Abort(b'%s: %s' % (f, msg))
2842 2844
2843 2845 if not match:
2844 2846 match = matchmod.always()
2845 2847
2846 2848 if not force:
2847 2849 match.bad = fail
2848 2850
2849 2851 # lock() for recent changelog (see issue4368)
2850 2852 with self.wlock(), self.lock():
2851 2853 wctx = self[None]
2852 2854 merge = len(wctx.parents()) > 1
2853 2855
2854 2856 if not force and merge and not match.always():
2855 2857 raise error.Abort(
2856 2858 _(
2857 2859 b'cannot partially commit a merge '
2858 2860 b'(do not specify files or patterns)'
2859 2861 )
2860 2862 )
2861 2863
2862 2864 status = self.status(match=match, clean=force)
2863 2865 if force:
2864 2866 status.modified.extend(
2865 2867 status.clean
2866 2868 ) # mq may commit clean files
2867 2869
2868 2870 # check subrepos
2869 2871 subs, commitsubs, newstate = subrepoutil.precommit(
2870 2872 self.ui, wctx, status, match, force=force
2871 2873 )
2872 2874
2873 2875 # make sure all explicit patterns are matched
2874 2876 if not force:
2875 2877 self.checkcommitpatterns(wctx, match, status, fail)
2876 2878
2877 2879 cctx = context.workingcommitctx(
2878 2880 self, status, text, user, date, extra
2879 2881 )
2880 2882
2881 2883 # internal config: ui.allowemptycommit
2882 2884 allowemptycommit = (
2883 2885 wctx.branch() != wctx.p1().branch()
2884 2886 or extra.get(b'close')
2885 2887 or merge
2886 2888 or cctx.files()
2887 2889 or self.ui.configbool(b'ui', b'allowemptycommit')
2888 2890 )
2889 2891 if not allowemptycommit:
2890 2892 return None
2891 2893
2892 2894 if merge and cctx.deleted():
2893 2895 raise error.Abort(_(b"cannot commit merge with missing files"))
2894 2896
2895 2897 ms = mergemod.mergestate.read(self)
2896 2898 mergeutil.checkunresolved(ms)
2897 2899
2898 2900 if editor:
2899 2901 cctx._text = editor(self, cctx, subs)
2900 2902 edited = text != cctx._text
2901 2903
2902 2904 # Save commit message in case this transaction gets rolled back
2903 2905 # (e.g. by a pretxncommit hook). Leave the content alone on
2904 2906 # the assumption that the user will use the same editor again.
2905 2907 msgfn = self.savecommitmessage(cctx._text)
2906 2908
2907 2909 # commit subs and write new state
2908 2910 if subs:
2909 2911 uipathfn = scmutil.getuipathfn(self)
2910 2912 for s in sorted(commitsubs):
2911 2913 sub = wctx.sub(s)
2912 2914 self.ui.status(
2913 2915 _(b'committing subrepository %s\n')
2914 2916 % uipathfn(subrepoutil.subrelpath(sub))
2915 2917 )
2916 2918 sr = sub.commit(cctx._text, user, date)
2917 2919 newstate[s] = (newstate[s][0], sr)
2918 2920 subrepoutil.writestate(self, newstate)
2919 2921
2920 2922 p1, p2 = self.dirstate.parents()
2921 2923 hookp1, hookp2 = hex(p1), (p2 != nullid and hex(p2) or b'')
2922 2924 try:
2923 2925 self.hook(
2924 2926 b"precommit", throw=True, parent1=hookp1, parent2=hookp2
2925 2927 )
2926 2928 with self.transaction(b'commit'):
2927 2929 ret = self.commitctx(cctx, True)
2928 2930 # update bookmarks, dirstate and mergestate
2929 2931 bookmarks.update(self, [p1, p2], ret)
2930 2932 cctx.markcommitted(ret)
2931 2933 ms.reset()
2932 2934 except: # re-raises
2933 2935 if edited:
2934 2936 self.ui.write(
2935 2937 _(b'note: commit message saved in %s\n') % msgfn
2936 2938 )
2937 2939 raise
2938 2940
2939 2941 def commithook():
2940 2942 # hack for command that use a temporary commit (eg: histedit)
2941 2943 # temporary commit got stripped before hook release
2942 2944 if self.changelog.hasnode(ret):
2943 2945 self.hook(
2944 2946 b"commit", node=hex(ret), parent1=hookp1, parent2=hookp2
2945 2947 )
2946 2948
2947 2949 self._afterlock(commithook)
2948 2950 return ret
2949 2951
2950 2952 @unfilteredmethod
2951 2953 def commitctx(self, ctx, error=False, origctx=None):
2952 2954 """Add a new revision to current repository.
2953 2955 Revision information is passed via the context argument.
2954 2956
2955 2957 ctx.files() should list all files involved in this commit, i.e.
2956 2958 modified/added/removed files. On merge, it may be wider than the
2957 2959 ctx.files() to be committed, since any file nodes derived directly
2958 2960 from p1 or p2 are excluded from the committed ctx.files().
2959 2961
2960 2962 origctx is for convert to work around the problem that bug
2961 2963 fixes to the files list in changesets change hashes. For
2962 2964 convert to be the identity, it can pass an origctx and this
2963 2965 function will use the same files list when it makes sense to
2964 2966 do so.
2965 2967 """
2966 2968
2967 2969 p1, p2 = ctx.p1(), ctx.p2()
2968 2970 user = ctx.user()
2969 2971
2970 2972 if self.filecopiesmode == b'changeset-sidedata':
2971 2973 writechangesetcopy = True
2972 2974 writefilecopymeta = True
2973 2975 writecopiesto = None
2974 2976 else:
2975 2977 writecopiesto = self.ui.config(b'experimental', b'copies.write-to')
2976 2978 writefilecopymeta = writecopiesto != b'changeset-only'
2977 2979 writechangesetcopy = writecopiesto in (
2978 2980 b'changeset-only',
2979 2981 b'compatibility',
2980 2982 )
2981 2983 p1copies, p2copies = None, None
2982 2984 if writechangesetcopy:
2983 2985 p1copies = ctx.p1copies()
2984 2986 p2copies = ctx.p2copies()
2985 2987 filesadded, filesremoved = None, None
2986 2988 with self.lock(), self.transaction(b"commit") as tr:
2987 2989 trp = weakref.proxy(tr)
2988 2990
2989 2991 if ctx.manifestnode():
2990 2992 # reuse an existing manifest revision
2991 2993 self.ui.debug(b'reusing known manifest\n')
2992 2994 mn = ctx.manifestnode()
2993 2995 files = ctx.files()
2994 2996 if writechangesetcopy:
2995 2997 filesadded = ctx.filesadded()
2996 2998 filesremoved = ctx.filesremoved()
2997 2999 elif ctx.files():
2998 3000 m1ctx = p1.manifestctx()
2999 3001 m2ctx = p2.manifestctx()
3000 3002 mctx = m1ctx.copy()
3001 3003
3002 3004 m = mctx.read()
3003 3005 m1 = m1ctx.read()
3004 3006 m2 = m2ctx.read()
3005 3007
3006 3008 # check in files
3007 3009 added = []
3008 3010 changed = []
3009 3011 removed = list(ctx.removed())
3010 3012 linkrev = len(self)
3011 3013 self.ui.note(_(b"committing files:\n"))
3012 3014 uipathfn = scmutil.getuipathfn(self)
3013 3015 for f in sorted(ctx.modified() + ctx.added()):
3014 3016 self.ui.note(uipathfn(f) + b"\n")
3015 3017 try:
3016 3018 fctx = ctx[f]
3017 3019 if fctx is None:
3018 3020 removed.append(f)
3019 3021 else:
3020 3022 added.append(f)
3021 3023 m[f] = self._filecommit(
3022 3024 fctx,
3023 3025 m1,
3024 3026 m2,
3025 3027 linkrev,
3026 3028 trp,
3027 3029 changed,
3028 3030 writefilecopymeta,
3029 3031 )
3030 3032 m.setflag(f, fctx.flags())
3031 3033 except OSError:
3032 3034 self.ui.warn(
3033 3035 _(b"trouble committing %s!\n") % uipathfn(f)
3034 3036 )
3035 3037 raise
3036 3038 except IOError as inst:
3037 3039 errcode = getattr(inst, 'errno', errno.ENOENT)
3038 3040 if error or errcode and errcode != errno.ENOENT:
3039 3041 self.ui.warn(
3040 3042 _(b"trouble committing %s!\n") % uipathfn(f)
3041 3043 )
3042 3044 raise
3043 3045
3044 3046 # update manifest
3045 3047 removed = [f for f in removed if f in m1 or f in m2]
3046 3048 drop = sorted([f for f in removed if f in m])
3047 3049 for f in drop:
3048 3050 del m[f]
3049 3051 if p2.rev() != nullrev:
3050 3052
3051 3053 @util.cachefunc
3052 3054 def mas():
3053 3055 p1n = p1.node()
3054 3056 p2n = p2.node()
3055 3057 cahs = self.changelog.commonancestorsheads(p1n, p2n)
3056 3058 if not cahs:
3057 3059 cahs = [nullrev]
3058 3060 return [self[r].manifest() for r in cahs]
3059 3061
3060 3062 def deletionfromparent(f):
3061 3063 # When a file is removed relative to p1 in a merge, this
3062 3064 # function determines whether the absence is due to a
3063 3065 # deletion from a parent, or whether the merge commit
3064 3066 # itself deletes the file. We decide this by doing a
3065 3067 # simplified three way merge of the manifest entry for
3066 3068 # the file. There are two ways we decide the merge
3067 3069 # itself didn't delete a file:
3068 3070 # - neither parent (nor the merge) contain the file
3069 3071 # - exactly one parent contains the file, and that
3070 3072 # parent has the same filelog entry as the merge
3071 3073 # ancestor (or all of them if there two). In other
3072 3074 # words, that parent left the file unchanged while the
3073 3075 # other one deleted it.
3074 3076 # One way to think about this is that deleting a file is
3075 3077 # similar to emptying it, so the list of changed files
3076 3078 # should be similar either way. The computation
3077 3079 # described above is not done directly in _filecommit
3078 3080 # when creating the list of changed files, however
3079 3081 # it does something very similar by comparing filelog
3080 3082 # nodes.
3081 3083 if f in m1:
3082 3084 return f not in m2 and all(
3083 3085 f in ma and ma.find(f) == m1.find(f)
3084 3086 for ma in mas()
3085 3087 )
3086 3088 elif f in m2:
3087 3089 return all(
3088 3090 f in ma and ma.find(f) == m2.find(f)
3089 3091 for ma in mas()
3090 3092 )
3091 3093 else:
3092 3094 return True
3093 3095
3094 3096 removed = [f for f in removed if not deletionfromparent(f)]
3095 3097
3096 3098 files = changed + removed
3097 3099 md = None
3098 3100 if not files:
3099 3101 # if no "files" actually changed in terms of the changelog,
3100 3102 # try hard to detect unmodified manifest entry so that the
3101 3103 # exact same commit can be reproduced later on convert.
3102 3104 md = m1.diff(m, scmutil.matchfiles(self, ctx.files()))
3103 3105 if not files and md:
3104 3106 self.ui.debug(
3105 3107 b'not reusing manifest (no file change in '
3106 3108 b'changelog, but manifest differs)\n'
3107 3109 )
3108 3110 if files or md:
3109 3111 self.ui.note(_(b"committing manifest\n"))
3110 3112 # we're using narrowmatch here since it's already applied at
3111 3113 # other stages (such as dirstate.walk), so we're already
3112 3114 # ignoring things outside of narrowspec in most cases. The
3113 3115 # one case where we might have files outside the narrowspec
3114 3116 # at this point is merges, and we already error out in the
3115 3117 # case where the merge has files outside of the narrowspec,
3116 3118 # so this is safe.
3117 3119 mn = mctx.write(
3118 3120 trp,
3119 3121 linkrev,
3120 3122 p1.manifestnode(),
3121 3123 p2.manifestnode(),
3122 3124 added,
3123 3125 drop,
3124 3126 match=self.narrowmatch(),
3125 3127 )
3126 3128
3127 3129 if writechangesetcopy:
3128 3130 filesadded = [
3129 3131 f for f in changed if not (f in m1 or f in m2)
3130 3132 ]
3131 3133 filesremoved = removed
3132 3134 else:
3133 3135 self.ui.debug(
3134 3136 b'reusing manifest from p1 (listed files '
3135 3137 b'actually unchanged)\n'
3136 3138 )
3137 3139 mn = p1.manifestnode()
3138 3140 else:
3139 3141 self.ui.debug(b'reusing manifest from p1 (no file change)\n')
3140 3142 mn = p1.manifestnode()
3141 3143 files = []
3142 3144
3143 3145 if writecopiesto == b'changeset-only':
3144 3146 # If writing only to changeset extras, use None to indicate that
3145 3147 # no entry should be written. If writing to both, write an empty
3146 3148 # entry to prevent the reader from falling back to reading
3147 3149 # filelogs.
3148 3150 p1copies = p1copies or None
3149 3151 p2copies = p2copies or None
3150 3152 filesadded = filesadded or None
3151 3153 filesremoved = filesremoved or None
3152 3154
3153 3155 if origctx and origctx.manifestnode() == mn:
3154 3156 files = origctx.files()
3155 3157
3156 3158 # update changelog
3157 3159 self.ui.note(_(b"committing changelog\n"))
3158 3160 self.changelog.delayupdate(tr)
3159 3161 n = self.changelog.add(
3160 3162 mn,
3161 3163 files,
3162 3164 ctx.description(),
3163 3165 trp,
3164 3166 p1.node(),
3165 3167 p2.node(),
3166 3168 user,
3167 3169 ctx.date(),
3168 3170 ctx.extra().copy(),
3169 3171 p1copies,
3170 3172 p2copies,
3171 3173 filesadded,
3172 3174 filesremoved,
3173 3175 )
3174 3176 xp1, xp2 = p1.hex(), p2 and p2.hex() or b''
3175 3177 self.hook(
3176 3178 b'pretxncommit',
3177 3179 throw=True,
3178 3180 node=hex(n),
3179 3181 parent1=xp1,
3180 3182 parent2=xp2,
3181 3183 )
3182 3184 # set the new commit is proper phase
3183 3185 targetphase = subrepoutil.newcommitphase(self.ui, ctx)
3184 3186 if targetphase:
3185 3187 # retract boundary do not alter parent changeset.
3186 3188 # if a parent have higher the resulting phase will
3187 3189 # be compliant anyway
3188 3190 #
3189 3191 # if minimal phase was 0 we don't need to retract anything
3190 3192 phases.registernew(self, tr, targetphase, [n])
3191 3193 return n
3192 3194
3193 3195 @unfilteredmethod
3194 3196 def destroying(self):
3195 3197 '''Inform the repository that nodes are about to be destroyed.
3196 3198 Intended for use by strip and rollback, so there's a common
3197 3199 place for anything that has to be done before destroying history.
3198 3200
3199 3201 This is mostly useful for saving state that is in memory and waiting
3200 3202 to be flushed when the current lock is released. Because a call to
3201 3203 destroyed is imminent, the repo will be invalidated causing those
3202 3204 changes to stay in memory (waiting for the next unlock), or vanish
3203 3205 completely.
3204 3206 '''
3205 3207 # When using the same lock to commit and strip, the phasecache is left
3206 3208 # dirty after committing. Then when we strip, the repo is invalidated,
3207 3209 # causing those changes to disappear.
3208 3210 if '_phasecache' in vars(self):
3209 3211 self._phasecache.write()
3210 3212
3211 3213 @unfilteredmethod
3212 3214 def destroyed(self):
3213 3215 '''Inform the repository that nodes have been destroyed.
3214 3216 Intended for use by strip and rollback, so there's a common
3215 3217 place for anything that has to be done after destroying history.
3216 3218 '''
3217 3219 # When one tries to:
3218 3220 # 1) destroy nodes thus calling this method (e.g. strip)
3219 3221 # 2) use phasecache somewhere (e.g. commit)
3220 3222 #
3221 3223 # then 2) will fail because the phasecache contains nodes that were
3222 3224 # removed. We can either remove phasecache from the filecache,
3223 3225 # causing it to reload next time it is accessed, or simply filter
3224 3226 # the removed nodes now and write the updated cache.
3225 3227 self._phasecache.filterunknown(self)
3226 3228 self._phasecache.write()
3227 3229
3228 3230 # refresh all repository caches
3229 3231 self.updatecaches()
3230 3232
3231 3233 # Ensure the persistent tag cache is updated. Doing it now
3232 3234 # means that the tag cache only has to worry about destroyed
3233 3235 # heads immediately after a strip/rollback. That in turn
3234 3236 # guarantees that "cachetip == currenttip" (comparing both rev
3235 3237 # and node) always means no nodes have been added or destroyed.
3236 3238
3237 3239 # XXX this is suboptimal when qrefresh'ing: we strip the current
3238 3240 # head, refresh the tag cache, then immediately add a new head.
3239 3241 # But I think doing it this way is necessary for the "instant
3240 3242 # tag cache retrieval" case to work.
3241 3243 self.invalidate()
3242 3244
3243 3245 def status(
3244 3246 self,
3245 3247 node1=b'.',
3246 3248 node2=None,
3247 3249 match=None,
3248 3250 ignored=False,
3249 3251 clean=False,
3250 3252 unknown=False,
3251 3253 listsubrepos=False,
3252 3254 ):
3253 3255 '''a convenience method that calls node1.status(node2)'''
3254 3256 return self[node1].status(
3255 3257 node2, match, ignored, clean, unknown, listsubrepos
3256 3258 )
3257 3259
3258 3260 def addpostdsstatus(self, ps):
3259 3261 """Add a callback to run within the wlock, at the point at which status
3260 3262 fixups happen.
3261 3263
3262 3264 On status completion, callback(wctx, status) will be called with the
3263 3265 wlock held, unless the dirstate has changed from underneath or the wlock
3264 3266 couldn't be grabbed.
3265 3267
3266 3268 Callbacks should not capture and use a cached copy of the dirstate --
3267 3269 it might change in the meanwhile. Instead, they should access the
3268 3270 dirstate via wctx.repo().dirstate.
3269 3271
3270 3272 This list is emptied out after each status run -- extensions should
3271 3273 make sure it adds to this list each time dirstate.status is called.
3272 3274 Extensions should also make sure they don't call this for statuses
3273 3275 that don't involve the dirstate.
3274 3276 """
3275 3277
3276 3278 # The list is located here for uniqueness reasons -- it is actually
3277 3279 # managed by the workingctx, but that isn't unique per-repo.
3278 3280 self._postdsstatus.append(ps)
3279 3281
3280 3282 def postdsstatus(self):
3281 3283 """Used by workingctx to get the list of post-dirstate-status hooks."""
3282 3284 return self._postdsstatus
3283 3285
3284 3286 def clearpostdsstatus(self):
3285 3287 """Used by workingctx to clear post-dirstate-status hooks."""
3286 3288 del self._postdsstatus[:]
3287 3289
3288 3290 def heads(self, start=None):
3289 3291 if start is None:
3290 3292 cl = self.changelog
3291 3293 headrevs = reversed(cl.headrevs())
3292 3294 return [cl.node(rev) for rev in headrevs]
3293 3295
3294 3296 heads = self.changelog.heads(start)
3295 3297 # sort the output in rev descending order
3296 3298 return sorted(heads, key=self.changelog.rev, reverse=True)
3297 3299
3298 3300 def branchheads(self, branch=None, start=None, closed=False):
3299 3301 '''return a (possibly filtered) list of heads for the given branch
3300 3302
3301 3303 Heads are returned in topological order, from newest to oldest.
3302 3304 If branch is None, use the dirstate branch.
3303 3305 If start is not None, return only heads reachable from start.
3304 3306 If closed is True, return heads that are marked as closed as well.
3305 3307 '''
3306 3308 if branch is None:
3307 3309 branch = self[None].branch()
3308 3310 branches = self.branchmap()
3309 3311 if not branches.hasbranch(branch):
3310 3312 return []
3311 3313 # the cache returns heads ordered lowest to highest
3312 3314 bheads = list(reversed(branches.branchheads(branch, closed=closed)))
3313 3315 if start is not None:
3314 3316 # filter out the heads that cannot be reached from startrev
3315 3317 fbheads = set(self.changelog.nodesbetween([start], bheads)[2])
3316 3318 bheads = [h for h in bheads if h in fbheads]
3317 3319 return bheads
3318 3320
3319 3321 def branches(self, nodes):
3320 3322 if not nodes:
3321 3323 nodes = [self.changelog.tip()]
3322 3324 b = []
3323 3325 for n in nodes:
3324 3326 t = n
3325 3327 while True:
3326 3328 p = self.changelog.parents(n)
3327 3329 if p[1] != nullid or p[0] == nullid:
3328 3330 b.append((t, n, p[0], p[1]))
3329 3331 break
3330 3332 n = p[0]
3331 3333 return b
3332 3334
3333 3335 def between(self, pairs):
3334 3336 r = []
3335 3337
3336 3338 for top, bottom in pairs:
3337 3339 n, l, i = top, [], 0
3338 3340 f = 1
3339 3341
3340 3342 while n != bottom and n != nullid:
3341 3343 p = self.changelog.parents(n)[0]
3342 3344 if i == f:
3343 3345 l.append(n)
3344 3346 f = f * 2
3345 3347 n = p
3346 3348 i += 1
3347 3349
3348 3350 r.append(l)
3349 3351
3350 3352 return r
3351 3353
3352 3354 def checkpush(self, pushop):
3353 3355 """Extensions can override this function if additional checks have
3354 3356 to be performed before pushing, or call it if they override push
3355 3357 command.
3356 3358 """
3357 3359
3358 3360 @unfilteredpropertycache
3359 3361 def prepushoutgoinghooks(self):
3360 3362 """Return util.hooks consists of a pushop with repo, remote, outgoing
3361 3363 methods, which are called before pushing changesets.
3362 3364 """
3363 3365 return util.hooks()
3364 3366
3365 3367 def pushkey(self, namespace, key, old, new):
3366 3368 try:
3367 3369 tr = self.currenttransaction()
3368 3370 hookargs = {}
3369 3371 if tr is not None:
3370 3372 hookargs.update(tr.hookargs)
3371 3373 hookargs = pycompat.strkwargs(hookargs)
3372 3374 hookargs['namespace'] = namespace
3373 3375 hookargs['key'] = key
3374 3376 hookargs['old'] = old
3375 3377 hookargs['new'] = new
3376 3378 self.hook(b'prepushkey', throw=True, **hookargs)
3377 3379 except error.HookAbort as exc:
3378 3380 self.ui.write_err(_(b"pushkey-abort: %s\n") % exc)
3379 3381 if exc.hint:
3380 3382 self.ui.write_err(_(b"(%s)\n") % exc.hint)
3381 3383 return False
3382 3384 self.ui.debug(b'pushing key for "%s:%s"\n' % (namespace, key))
3383 3385 ret = pushkey.push(self, namespace, key, old, new)
3384 3386
3385 3387 def runhook():
3386 3388 self.hook(
3387 3389 b'pushkey',
3388 3390 namespace=namespace,
3389 3391 key=key,
3390 3392 old=old,
3391 3393 new=new,
3392 3394 ret=ret,
3393 3395 )
3394 3396
3395 3397 self._afterlock(runhook)
3396 3398 return ret
3397 3399
3398 3400 def listkeys(self, namespace):
3399 3401 self.hook(b'prelistkeys', throw=True, namespace=namespace)
3400 3402 self.ui.debug(b'listing keys for "%s"\n' % namespace)
3401 3403 values = pushkey.list(self, namespace)
3402 3404 self.hook(b'listkeys', namespace=namespace, values=values)
3403 3405 return values
3404 3406
3405 3407 def debugwireargs(self, one, two, three=None, four=None, five=None):
3406 3408 '''used to test argument passing over the wire'''
3407 3409 return b"%s %s %s %s %s" % (
3408 3410 one,
3409 3411 two,
3410 3412 pycompat.bytestr(three),
3411 3413 pycompat.bytestr(four),
3412 3414 pycompat.bytestr(five),
3413 3415 )
3414 3416
3415 3417 def savecommitmessage(self, text):
3416 3418 fp = self.vfs(b'last-message.txt', b'wb')
3417 3419 try:
3418 3420 fp.write(text)
3419 3421 finally:
3420 3422 fp.close()
3421 3423 return self.pathto(fp.name[len(self.root) + 1 :])
3422 3424
3423 3425
3424 3426 # used to avoid circular references so destructors work
3425 3427 def aftertrans(files):
3426 3428 renamefiles = [tuple(t) for t in files]
3427 3429
3428 3430 def a():
3429 3431 for vfs, src, dest in renamefiles:
3430 3432 # if src and dest refer to a same file, vfs.rename is a no-op,
3431 3433 # leaving both src and dest on disk. delete dest to make sure
3432 3434 # the rename couldn't be such a no-op.
3433 3435 vfs.tryunlink(dest)
3434 3436 try:
3435 3437 vfs.rename(src, dest)
3436 3438 except OSError: # journal file does not yet exist
3437 3439 pass
3438 3440
3439 3441 return a
3440 3442
3441 3443
3442 3444 def undoname(fn):
3443 3445 base, name = os.path.split(fn)
3444 3446 assert name.startswith(b'journal')
3445 3447 return os.path.join(base, name.replace(b'journal', b'undo', 1))
3446 3448
3447 3449
3448 3450 def instance(ui, path, create, intents=None, createopts=None):
3449 3451 localpath = util.urllocalpath(path)
3450 3452 if create:
3451 3453 createrepository(ui, localpath, createopts=createopts)
3452 3454
3453 3455 return makelocalrepository(ui, localpath, intents=intents)
3454 3456
3455 3457
3456 3458 def islocal(path):
3457 3459 return True
3458 3460
3459 3461
3460 3462 def defaultcreateopts(ui, createopts=None):
3461 3463 """Populate the default creation options for a repository.
3462 3464
3463 3465 A dictionary of explicitly requested creation options can be passed
3464 3466 in. Missing keys will be populated.
3465 3467 """
3466 3468 createopts = dict(createopts or {})
3467 3469
3468 3470 if b'backend' not in createopts:
3469 3471 # experimental config: storage.new-repo-backend
3470 3472 createopts[b'backend'] = ui.config(b'storage', b'new-repo-backend')
3471 3473
3472 3474 return createopts
3473 3475
3474 3476
3475 3477 def newreporequirements(ui, createopts):
3476 3478 """Determine the set of requirements for a new local repository.
3477 3479
3478 3480 Extensions can wrap this function to specify custom requirements for
3479 3481 new repositories.
3480 3482 """
3481 3483 # If the repo is being created from a shared repository, we copy
3482 3484 # its requirements.
3483 3485 if b'sharedrepo' in createopts:
3484 3486 requirements = set(createopts[b'sharedrepo'].requirements)
3485 3487 if createopts.get(b'sharedrelative'):
3486 3488 requirements.add(b'relshared')
3487 3489 else:
3488 3490 requirements.add(b'shared')
3489 3491
3490 3492 return requirements
3491 3493
3492 3494 if b'backend' not in createopts:
3493 3495 raise error.ProgrammingError(
3494 3496 b'backend key not present in createopts; '
3495 3497 b'was defaultcreateopts() called?'
3496 3498 )
3497 3499
3498 3500 if createopts[b'backend'] != b'revlogv1':
3499 3501 raise error.Abort(
3500 3502 _(
3501 3503 b'unable to determine repository requirements for '
3502 3504 b'storage backend: %s'
3503 3505 )
3504 3506 % createopts[b'backend']
3505 3507 )
3506 3508
3507 3509 requirements = {b'revlogv1'}
3508 3510 if ui.configbool(b'format', b'usestore'):
3509 3511 requirements.add(b'store')
3510 3512 if ui.configbool(b'format', b'usefncache'):
3511 3513 requirements.add(b'fncache')
3512 3514 if ui.configbool(b'format', b'dotencode'):
3513 3515 requirements.add(b'dotencode')
3514 3516
3515 3517 compengine = ui.config(b'format', b'revlog-compression')
3516 3518 if compengine not in util.compengines:
3517 3519 raise error.Abort(
3518 3520 _(
3519 3521 b'compression engine %s defined by '
3520 3522 b'format.revlog-compression not available'
3521 3523 )
3522 3524 % compengine,
3523 3525 hint=_(
3524 3526 b'run "hg debuginstall" to list available '
3525 3527 b'compression engines'
3526 3528 ),
3527 3529 )
3528 3530
3529 3531 # zlib is the historical default and doesn't need an explicit requirement.
3530 3532 elif compengine == b'zstd':
3531 3533 requirements.add(b'revlog-compression-zstd')
3532 3534 elif compengine != b'zlib':
3533 3535 requirements.add(b'exp-compression-%s' % compengine)
3534 3536
3535 3537 if scmutil.gdinitconfig(ui):
3536 3538 requirements.add(b'generaldelta')
3537 3539 if ui.configbool(b'format', b'sparse-revlog'):
3538 3540 requirements.add(SPARSEREVLOG_REQUIREMENT)
3539 3541
3540 3542 # experimental config: format.exp-use-side-data
3541 3543 if ui.configbool(b'format', b'exp-use-side-data'):
3542 3544 requirements.add(SIDEDATA_REQUIREMENT)
3543 3545 # experimental config: format.exp-use-copies-side-data-changeset
3544 3546 if ui.configbool(b'format', b'exp-use-copies-side-data-changeset'):
3545 3547 requirements.add(SIDEDATA_REQUIREMENT)
3546 3548 requirements.add(COPIESSDC_REQUIREMENT)
3547 3549 if ui.configbool(b'experimental', b'treemanifest'):
3548 3550 requirements.add(b'treemanifest')
3549 3551
3550 3552 revlogv2 = ui.config(b'experimental', b'revlogv2')
3551 3553 if revlogv2 == b'enable-unstable-format-and-corrupt-my-data':
3552 3554 requirements.remove(b'revlogv1')
3553 3555 # generaldelta is implied by revlogv2.
3554 3556 requirements.discard(b'generaldelta')
3555 3557 requirements.add(REVLOGV2_REQUIREMENT)
3556 3558 # experimental config: format.internal-phase
3557 3559 if ui.configbool(b'format', b'internal-phase'):
3558 3560 requirements.add(b'internal-phase')
3559 3561
3560 3562 if createopts.get(b'narrowfiles'):
3561 3563 requirements.add(repository.NARROW_REQUIREMENT)
3562 3564
3563 3565 if createopts.get(b'lfs'):
3564 3566 requirements.add(b'lfs')
3565 3567
3566 3568 if ui.configbool(b'format', b'bookmarks-in-store'):
3567 3569 requirements.add(bookmarks.BOOKMARKS_IN_STORE_REQUIREMENT)
3568 3570
3569 3571 return requirements
3570 3572
3571 3573
3572 3574 def filterknowncreateopts(ui, createopts):
3573 3575 """Filters a dict of repo creation options against options that are known.
3574 3576
3575 3577 Receives a dict of repo creation options and returns a dict of those
3576 3578 options that we don't know how to handle.
3577 3579
3578 3580 This function is called as part of repository creation. If the
3579 3581 returned dict contains any items, repository creation will not
3580 3582 be allowed, as it means there was a request to create a repository
3581 3583 with options not recognized by loaded code.
3582 3584
3583 3585 Extensions can wrap this function to filter out creation options
3584 3586 they know how to handle.
3585 3587 """
3586 3588 known = {
3587 3589 b'backend',
3588 3590 b'lfs',
3589 3591 b'narrowfiles',
3590 3592 b'sharedrepo',
3591 3593 b'sharedrelative',
3592 3594 b'shareditems',
3593 3595 b'shallowfilestore',
3594 3596 }
3595 3597
3596 3598 return {k: v for k, v in createopts.items() if k not in known}
3597 3599
3598 3600
3599 3601 def createrepository(ui, path, createopts=None):
3600 3602 """Create a new repository in a vfs.
3601 3603
3602 3604 ``path`` path to the new repo's working directory.
3603 3605 ``createopts`` options for the new repository.
3604 3606
3605 3607 The following keys for ``createopts`` are recognized:
3606 3608
3607 3609 backend
3608 3610 The storage backend to use.
3609 3611 lfs
3610 3612 Repository will be created with ``lfs`` requirement. The lfs extension
3611 3613 will automatically be loaded when the repository is accessed.
3612 3614 narrowfiles
3613 3615 Set up repository to support narrow file storage.
3614 3616 sharedrepo
3615 3617 Repository object from which storage should be shared.
3616 3618 sharedrelative
3617 3619 Boolean indicating if the path to the shared repo should be
3618 3620 stored as relative. By default, the pointer to the "parent" repo
3619 3621 is stored as an absolute path.
3620 3622 shareditems
3621 3623 Set of items to share to the new repository (in addition to storage).
3622 3624 shallowfilestore
3623 3625 Indicates that storage for files should be shallow (not all ancestor
3624 3626 revisions are known).
3625 3627 """
3626 3628 createopts = defaultcreateopts(ui, createopts=createopts)
3627 3629
3628 3630 unknownopts = filterknowncreateopts(ui, createopts)
3629 3631
3630 3632 if not isinstance(unknownopts, dict):
3631 3633 raise error.ProgrammingError(
3632 3634 b'filterknowncreateopts() did not return a dict'
3633 3635 )
3634 3636
3635 3637 if unknownopts:
3636 3638 raise error.Abort(
3637 3639 _(
3638 3640 b'unable to create repository because of unknown '
3639 3641 b'creation option: %s'
3640 3642 )
3641 3643 % b', '.join(sorted(unknownopts)),
3642 3644 hint=_(b'is a required extension not loaded?'),
3643 3645 )
3644 3646
3645 3647 requirements = newreporequirements(ui, createopts=createopts)
3646 3648
3647 3649 wdirvfs = vfsmod.vfs(path, expandpath=True, realpath=True)
3648 3650
3649 3651 hgvfs = vfsmod.vfs(wdirvfs.join(b'.hg'))
3650 3652 if hgvfs.exists():
3651 3653 raise error.RepoError(_(b'repository %s already exists') % path)
3652 3654
3653 3655 if b'sharedrepo' in createopts:
3654 3656 sharedpath = createopts[b'sharedrepo'].sharedpath
3655 3657
3656 3658 if createopts.get(b'sharedrelative'):
3657 3659 try:
3658 3660 sharedpath = os.path.relpath(sharedpath, hgvfs.base)
3659 3661 except (IOError, ValueError) as e:
3660 3662 # ValueError is raised on Windows if the drive letters differ
3661 3663 # on each path.
3662 3664 raise error.Abort(
3663 3665 _(b'cannot calculate relative path'),
3664 3666 hint=stringutil.forcebytestr(e),
3665 3667 )
3666 3668
3667 3669 if not wdirvfs.exists():
3668 3670 wdirvfs.makedirs()
3669 3671
3670 3672 hgvfs.makedir(notindexed=True)
3671 3673 if b'sharedrepo' not in createopts:
3672 3674 hgvfs.mkdir(b'cache')
3673 3675 hgvfs.mkdir(b'wcache')
3674 3676
3675 3677 if b'store' in requirements and b'sharedrepo' not in createopts:
3676 3678 hgvfs.mkdir(b'store')
3677 3679
3678 3680 # We create an invalid changelog outside the store so very old
3679 3681 # Mercurial versions (which didn't know about the requirements
3680 3682 # file) encounter an error on reading the changelog. This
3681 3683 # effectively locks out old clients and prevents them from
3682 3684 # mucking with a repo in an unknown format.
3683 3685 #
3684 3686 # The revlog header has version 2, which won't be recognized by
3685 3687 # such old clients.
3686 3688 hgvfs.append(
3687 3689 b'00changelog.i',
3688 3690 b'\0\0\0\2 dummy changelog to prevent using the old repo '
3689 3691 b'layout',
3690 3692 )
3691 3693
3692 3694 scmutil.writerequires(hgvfs, requirements)
3693 3695
3694 3696 # Write out file telling readers where to find the shared store.
3695 3697 if b'sharedrepo' in createopts:
3696 3698 hgvfs.write(b'sharedpath', sharedpath)
3697 3699
3698 3700 if createopts.get(b'shareditems'):
3699 3701 shared = b'\n'.join(sorted(createopts[b'shareditems'])) + b'\n'
3700 3702 hgvfs.write(b'shared', shared)
3701 3703
3702 3704
3703 3705 def poisonrepository(repo):
3704 3706 """Poison a repository instance so it can no longer be used."""
3705 3707 # Perform any cleanup on the instance.
3706 3708 repo.close()
3707 3709
3708 3710 # Our strategy is to replace the type of the object with one that
3709 3711 # has all attribute lookups result in error.
3710 3712 #
3711 3713 # But we have to allow the close() method because some constructors
3712 3714 # of repos call close() on repo references.
3713 3715 class poisonedrepository(object):
3714 3716 def __getattribute__(self, item):
3715 3717 if item == 'close':
3716 3718 return object.__getattribute__(self, item)
3717 3719
3718 3720 raise error.ProgrammingError(
3719 3721 b'repo instances should not be used after unshare'
3720 3722 )
3721 3723
3722 3724 def close(self):
3723 3725 pass
3724 3726
3725 3727 # We may have a repoview, which intercepts __setattr__. So be sure
3726 3728 # we operate at the lowest level possible.
3727 3729 object.__setattr__(repo, '__class__', poisonedrepository)
@@ -1,53 +1,52 b''
1 1 ===================================
2 2 Test repository filtering avoidance
3 3 ===================================
4 4
5 5 This test file is a bit special as he does not check feature, but performance related internal code path.
6 6
7 7 Right now, filtering a repository comes with a cost that might be significant.
8 8 Until this get better, ther are various operation that try hard not to trigger
9 9 a filtering computation. This test file make sure we don't reintroduce code that trigger the filtering for these operation:
10 10
11 11 Setup
12 12 -----
13 13 $ hg init test-repo
14 14 $ cd test-repo
15 15 $ echo "some line" > z
16 16 $ echo a > a
17 17 $ hg commit -Am a
18 18 adding a
19 19 adding z
20 20 $ echo "in a" >> z
21 21 $ echo b > b
22 22 $ hg commit -Am b
23 23 adding b
24 24 $ echo "file" >> z
25 25 $ echo c > c
26 26 $ hg commit -Am c
27 27 adding c
28 28 $ hg rm a
29 29 $ echo c1 > c
30 30 $ hg add c
31 31 c already tracked!
32 32 $ echo d > d
33 33 $ hg add d
34 34 $ rm b
35 35
36 36 $ cat << EOF >> $HGRCPATH
37 37 > [devel]
38 38 > debug.repo-filters = yes
39 39 > [ui]
40 40 > debug = yes
41 41 > EOF
42 42
43 43
44 44 tests
45 45 -----
46 46
47 47 Getting the node of `null`
48 48
49 49 $ hg init test-repo
50 50 $ cd test-repo
51 51 $ hg log -r null -T "{node}\n"
52 debug.filters: computing revision filter for "visible"
53 52 0000000000000000000000000000000000000000
General Comments 0
You need to be logged in to leave comments. Login now