"bundle" was appropriate for the bundle1 days where a bundle
was a changegroup. In a bundle2 world, changegroup readers
are referred to as "changegroup unpackers."
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| phases, | phases, | ||||
| pycompat, | pycompat, | ||||
| revlog, | revlog, | ||||
| util, | util, | ||||
| vfs as vfsmod, | vfs as vfsmod, | ||||
| ) | ) | ||||
| class bundlerevlog(revlog.revlog): | class bundlerevlog(revlog.revlog): | ||||
| def __init__(self, opener, indexfile, bundle, linkmapper): | def __init__(self, opener, indexfile, cgunpacker, linkmapper): | ||||
| # How it works: | # How it works: | ||||
| # To retrieve a revision, we need to know the offset of the revision in | # To retrieve a revision, we need to know the offset of the revision in | ||||
| # the bundle (an unbundle object). We store this offset in the index | # the bundle (an unbundle object). We store this offset in the index | ||||
| # (start). The base of the delta is stored in the base field. | # (start). The base of the delta is stored in the base field. | ||||
| # | # | ||||
| # To differentiate a rev in the bundle from a rev in the revlog, we | # To differentiate a rev in the bundle from a rev in the revlog, we | ||||
| # check revision against repotiprev. | # check revision against repotiprev. | ||||
| opener = vfsmod.readonlyvfs(opener) | opener = vfsmod.readonlyvfs(opener) | ||||
| revlog.revlog.__init__(self, opener, indexfile) | revlog.revlog.__init__(self, opener, indexfile) | ||||
| self.bundle = bundle | self.bundle = cgunpacker | ||||
| n = len(self) | n = len(self) | ||||
| self.repotiprev = n - 1 | self.repotiprev = n - 1 | ||||
| self.bundlerevs = set() # used by 'bundle()' revset expression | self.bundlerevs = set() # used by 'bundle()' revset expression | ||||
| for deltadata in bundle.deltaiter(): | for deltadata in cgunpacker.deltaiter(): | ||||
| node, p1, p2, cs, deltabase, delta, flags = deltadata | node, p1, p2, cs, deltabase, delta, flags = deltadata | ||||
| size = len(delta) | size = len(delta) | ||||
| start = bundle.tell() - size | start = cgunpacker.tell() - size | ||||
| link = linkmapper(cs) | link = linkmapper(cs) | ||||
| if node in self.nodemap: | if node in self.nodemap: | ||||
| # this can happen if two branches make the same change | # this can happen if two branches make the same change | ||||
| self.bundlerevs.add(self.nodemap[node]) | self.bundlerevs.add(self.nodemap[node]) | ||||
| continue | continue | ||||
| for p in (p1, p2): | for p in (p1, p2): | ||||
| def strip(self, *args, **kwargs): | def strip(self, *args, **kwargs): | ||||
| raise NotImplementedError | raise NotImplementedError | ||||
| def checksize(self): | def checksize(self): | ||||
| raise NotImplementedError | raise NotImplementedError | ||||
| class bundlechangelog(bundlerevlog, changelog.changelog): | class bundlechangelog(bundlerevlog, changelog.changelog): | ||||
| def __init__(self, opener, bundle): | def __init__(self, opener, cgunpacker): | ||||
| changelog.changelog.__init__(self, opener) | changelog.changelog.__init__(self, opener) | ||||
| linkmapper = lambda x: x | linkmapper = lambda x: x | ||||
| bundlerevlog.__init__(self, opener, self.indexfile, bundle, | bundlerevlog.__init__(self, opener, self.indexfile, cgunpacker, | ||||
| linkmapper) | linkmapper) | ||||
| def baserevision(self, nodeorrev): | def baserevision(self, nodeorrev): | ||||
| # Although changelog doesn't override 'revision' method, some extensions | # Although changelog doesn't override 'revision' method, some extensions | ||||
| # may replace this class with another that does. Same story with | # may replace this class with another that does. Same story with | ||||
| # manifest and filelog classes. | # manifest and filelog classes. | ||||
| # This bypasses filtering on changelog.node() and rev() because we need | # This bypasses filtering on changelog.node() and rev() because we need | ||||
| # revision text of the bundle base even if it is hidden. | # revision text of the bundle base even if it is hidden. | ||||
| oldfilter = self.filteredrevs | oldfilter = self.filteredrevs | ||||
| try: | try: | ||||
| self.filteredrevs = () | self.filteredrevs = () | ||||
| return changelog.changelog.revision(self, nodeorrev, raw=True) | return changelog.changelog.revision(self, nodeorrev, raw=True) | ||||
| finally: | finally: | ||||
| self.filteredrevs = oldfilter | self.filteredrevs = oldfilter | ||||
| class bundlemanifest(bundlerevlog, manifest.manifestrevlog): | class bundlemanifest(bundlerevlog, manifest.manifestrevlog): | ||||
| def __init__(self, opener, bundle, linkmapper, dirlogstarts=None, dir=''): | def __init__(self, opener, cgunpacker, linkmapper, dirlogstarts=None, | ||||
| dir=''): | |||||
| manifest.manifestrevlog.__init__(self, opener, dir=dir) | manifest.manifestrevlog.__init__(self, opener, dir=dir) | ||||
| bundlerevlog.__init__(self, opener, self.indexfile, bundle, | bundlerevlog.__init__(self, opener, self.indexfile, cgunpacker, | ||||
| linkmapper) | linkmapper) | ||||
| if dirlogstarts is None: | if dirlogstarts is None: | ||||
| dirlogstarts = {} | dirlogstarts = {} | ||||
| if self.bundle.version == "03": | if self.bundle.version == "03": | ||||
| dirlogstarts = _getfilestarts(self.bundle) | dirlogstarts = _getfilestarts(self.bundle) | ||||
| self._dirlogstarts = dirlogstarts | self._dirlogstarts = dirlogstarts | ||||
| self._linkmapper = linkmapper | self._linkmapper = linkmapper | ||||
| if d in self._dirlogstarts: | if d in self._dirlogstarts: | ||||
| self.bundle.seek(self._dirlogstarts[d]) | self.bundle.seek(self._dirlogstarts[d]) | ||||
| return bundlemanifest( | return bundlemanifest( | ||||
| self.opener, self.bundle, self._linkmapper, | self.opener, self.bundle, self._linkmapper, | ||||
| self._dirlogstarts, dir=d) | self._dirlogstarts, dir=d) | ||||
| return super(bundlemanifest, self).dirlog(d) | return super(bundlemanifest, self).dirlog(d) | ||||
| class bundlefilelog(bundlerevlog, filelog.filelog): | class bundlefilelog(bundlerevlog, filelog.filelog): | ||||
| def __init__(self, opener, path, bundle, linkmapper): | def __init__(self, opener, path, cgunpacker, linkmapper): | ||||
| filelog.filelog.__init__(self, opener, path) | filelog.filelog.__init__(self, opener, path) | ||||
| bundlerevlog.__init__(self, opener, self.indexfile, bundle, | bundlerevlog.__init__(self, opener, self.indexfile, cgunpacker, | ||||
| linkmapper) | linkmapper) | ||||
| def baserevision(self, nodeorrev): | def baserevision(self, nodeorrev): | ||||
| return filelog.filelog.revision(self, nodeorrev, raw=True) | return filelog.filelog.revision(self, nodeorrev, raw=True) | ||||
| class bundlepeer(localrepo.localpeer): | class bundlepeer(localrepo.localpeer): | ||||
| def canpush(self): | def canpush(self): | ||||
| return False | return False | ||||
| def _write(self, fp): | def _write(self, fp): | ||||
| raise NotImplementedError | raise NotImplementedError | ||||
| def _updateroots(self, phase, newroots, tr): | def _updateroots(self, phase, newroots, tr): | ||||
| self.phaseroots[phase] = newroots | self.phaseroots[phase] = newroots | ||||
| self.invalidate() | self.invalidate() | ||||
| self.dirty = True | self.dirty = True | ||||
| def _getfilestarts(bundle): | |||||
| bundlefilespos = {} | bundlefilespos = {} | ||||
| for chunkdata in iter(bundle.filelogheader, {}): | def _getfilestarts(cgunpacker): | ||||
| for chunkdata in iter(cgunpacker.filelogheader, {}): | |||||
| fname = chunkdata['filename'] | fname = chunkdata['filename'] | ||||
| bundlefilespos[fname] = bundle.tell() | bundlefilespos[fname] = bundle.tell() | ||||
| for chunk in iter(lambda: bundle.deltachunk(None), {}): | for chunk in iter(lambda: cgunpacker.deltachunk(None), {}): | ||||
| pass | pass | ||||
| return bundlefilespos | return bundlefilespos | ||||
| class bundlerepository(localrepo.localrepository): | class bundlerepository(localrepo.localrepository): | ||||
| """A repository instance that is a union of a local repo and a bundle. | """A repository instance that is a union of a local repo and a bundle. | ||||
| Instances represent a read-only repository composed of a local repository | Instances represent a read-only repository composed of a local repository | ||||
| with the contents of a bundle file applied. The repository instance is | with the contents of a bundle file applied. The repository instance is | ||||