comparison mercurial/localrepo.py @ 19201:309c439cdbaa

bundle-ng: add bundlecaps argument to getbundle() command

19200:4cfdec944edf

        return self._repo.heads()

    def known(self, nodes):
        return self._repo.known(nodes)

    def getbundle(self, source, heads=None, common=None):
        return self._repo.getbundle(source, heads=heads, common=common)

    # TODO We might want to move the next two calls into legacypeer and add
    # unbundle instead.

    def lock(self):

                elif heads is None and remote.capable('changegroupsubset'):
                    # issue1320, avoid a race if remote changed after discovery
                    heads = rheads

                if remote.capable('getbundle'):
                    cg = remote.getbundle('pull', common=common,
                                          heads=heads or rheads)
                elif heads is None:
                    cg = remote.changegroup(fetch, 'pull')
                elif not remote.capable('changegroupsubset'):

                                                        ctx))
                        discovery.checkheads(unfi, remote, outgoing,
                                             remoteheads, newbranch,
                                             bool(inc))

                    # create a changegroup from local
                    if revs is None and not outgoing.excluded:
                        # push everything,
                        # use the fast path, no race possible on push
                        bundler = changegroup.bundle10()
                        cg = self._changegroup(outgoing.missing, bundler,
                                               'push')
                    else:
                        cg = self.getlocalbundle('push', outgoing)

                    # apply changegroup to remote
                    if unbundle:
                        # local repo finds heads on server, finds out what
                        # revs it must push. once revs transferred, if server

        # We assume that all ancestors of bases are known
        common = cl.ancestors([cl.rev(n) for n in bases])
        bundler = changegroup.bundle10()
        return self._changegroupsubset(common, csets, heads, bundler, source)

    def getlocalbundle(self, source, outgoing):
        """Like getbundle, but taking a discovery.outgoing as an argument.

        This is only implemented for local repos and reuses potentially
        precomputed sets in outgoing."""
        if not outgoing.missing:
            return None
        bundler = changegroup.bundle10()
        return self._changegroupsubset(outgoing.common,
                                       outgoing.missing,
                                       outgoing.missingheads,
                                       bundler,
                                       source)

    def getbundle(self, source, heads=None, common=None):
        """Like changegroupsubset, but returns the set difference between the
        ancestors of heads and the ancestors common.

        If heads is None, use the local heads. If common is None, use [nullid].

        else:
            common = [nullid]
        if not heads:
            heads = cl.heads()
        return self.getlocalbundle(source,
                                   discovery.outgoing(cl, common, heads))

    @unfilteredmethod
    def _changegroupsubset(self, commonrevs, csets, heads, bundler, source):

        cl = self.changelog

19201:309c439cdbaa

        return self._repo.heads()

    def known(self, nodes):
        return self._repo.known(nodes)

    def getbundle(self, source, heads=None, common=None, bundlecaps=None):
        return self._repo.getbundle(source, heads=heads, common=common,
                                    bundlecaps=None)

    # TODO We might want to move the next two calls into legacypeer and add
    # unbundle instead.

    def lock(self):

                elif heads is None and remote.capable('changegroupsubset'):
                    # issue1320, avoid a race if remote changed after discovery
                    heads = rheads

                if remote.capable('getbundle'):
                    # TODO: get bundlecaps from remote
                    cg = remote.getbundle('pull', common=common,
                                          heads=heads or rheads)
                elif heads is None:
                    cg = remote.changegroup(fetch, 'pull')
                elif not remote.capable('changegroupsubset'):

                                                        ctx))
                        discovery.checkheads(unfi, remote, outgoing,
                                             remoteheads, newbranch,
                                             bool(inc))

                    # TODO: get bundlecaps from remote
                    bundlecaps = None
                    # create a changegroup from local
                    if revs is None and not outgoing.excluded:
                        # push everything,
                        # use the fast path, no race possible on push
                        bundler = changegroup.bundle10(bundlecaps)
                        cg = self._changegroup(outgoing.missing, bundler,
                                               'push')
                    else:
                        cg = self.getlocalbundle('push', outgoing, bundlecaps)

                    # apply changegroup to remote
                    if unbundle:
                        # local repo finds heads on server, finds out what
                        # revs it must push. once revs transferred, if server

        # We assume that all ancestors of bases are known
        common = cl.ancestors([cl.rev(n) for n in bases])
        bundler = changegroup.bundle10()
        return self._changegroupsubset(common, csets, heads, bundler, source)

    def getlocalbundle(self, source, outgoing, bundlecaps=None):
        """Like getbundle, but taking a discovery.outgoing as an argument.

        This is only implemented for local repos and reuses potentially
        precomputed sets in outgoing."""
        if not outgoing.missing:
            return None
        bundler = changegroup.bundle10(bundlecaps)
        return self._changegroupsubset(outgoing.common,
                                       outgoing.missing,
                                       outgoing.missingheads,
                                       bundler,
                                       source)

    def getbundle(self, source, heads=None, common=None, bundlecaps=None):
        """Like changegroupsubset, but returns the set difference between the
        ancestors of heads and the ancestors common.

        If heads is None, use the local heads. If common is None, use [nullid].

        else:
            common = [nullid]
        if not heads:
            heads = cl.heads()
        return self.getlocalbundle(source,
                                   discovery.outgoing(cl, common, heads),
                                   bundlecaps=bundlecaps)

    @unfilteredmethod
    def _changegroupsubset(self, commonrevs, csets, heads, bundler, source):

        cl = self.changelog