comparison mercurial/changegroup.py @ 39732:e03c1a63155c

changegroup: tease out a temporary prune method for manifests

It's extracted so extensions can filter manifest nodes if needed. This
is an unfortunate hack, but I think I only need it for manifests. The
long-term solution will be to rework the relationship between
changegroups and storage so that this isn't required.

Differential Revision: https://phab.mercurial-scm.org/D4685

39731:c71f80bfb414

            store = mfl.getstorage(tree)

            if not self._filematcher.visitdir(store.tree[:-1] or '.'):
                prunednodes = []
            else:
                frev, flr = store.rev, store.linkrev
                prunednodes = [n for n in nodes
                               if flr(frev(n)) not in commonrevs]

            if tree and not prunednodes:
                continue

            lookupfn = makelookupmflinknode(tree, nodes)

                clrevtolocalrev=clrevtolocalrev,
                fullclnodes=self._fullclnodes,
                precomputedellipsis=self._precomputedellipsis)

            yield tree, deltas

    # The 'source' parameter is useful for extensions
    def generatefiles(self, changedfiles, commonrevs, source,
                      mfdicts, fastpathlinkrev, fnodes, clrevs):
        changedfiles = list(filter(self._filematcher, changedfiles))

39732:e03c1a63155c

            store = mfl.getstorage(tree)

            if not self._filematcher.visitdir(store.tree[:-1] or '.'):
                prunednodes = []
            else:
                prunednodes = self._prunemanifests(store, nodes, commonrevs)
            if tree and not prunednodes:
                continue

            lookupfn = makelookupmflinknode(tree, nodes)

                clrevtolocalrev=clrevtolocalrev,
                fullclnodes=self._fullclnodes,
                precomputedellipsis=self._precomputedellipsis)

            yield tree, deltas

    def _prunemanifests(self, store, nodes, commonrevs):
        # This is split out as a separate method to allow filtering
        # commonrevs in extension code.
        #
        # TODO(augie): this shouldn't be required, instead we should
        # make filtering of revisions to send delegated to the store
        # layer.
        frev, flr = store.rev, store.linkrev
        return [n for n in nodes if flr(frev(n)) not in commonrevs]

    # The 'source' parameter is useful for extensions
    def generatefiles(self, changedfiles, commonrevs, source,
                      mfdicts, fastpathlinkrev, fnodes, clrevs):
        changedfiles = list(filter(self._filematcher, changedfiles))