| quark |
| Restricted Project |
| Lint Skipped |
| Unit Tests Skipped |
| from mercurial import wireproto as wireprotov1server | from mercurial import wireproto as wireprotov1server | ||||
| def setupserver(ui, repo): | def setupserver(ui, repo): | ||||
| """Sets up a normal Mercurial repo so it can serve files to shallow repos. | """Sets up a normal Mercurial repo so it can serve files to shallow repos. | ||||
| """ | """ | ||||
| onetimesetup(ui) | onetimesetup(ui) | ||||
| # don't send files to shallow clients during pulls | # don't send files to shallow clients during pulls | ||||
| def generatefiles(orig, self, changedfiles, linknodes, commonrevs, source): | def generatefiles(orig, self, changedfiles, linknodes, commonrevs, source, | ||||
| *args, **kwargs): | |||||
| caps = self._bundlecaps or [] | caps = self._bundlecaps or [] | ||||
| if shallowrepo.requirement in caps: | if shallowrepo.requirement in caps: | ||||
| # only send files that don't match the specified patterns | # only send files that don't match the specified patterns | ||||
| includepattern = None | includepattern = None | ||||
| excludepattern = None | excludepattern = None | ||||
| for cap in (self._bundlecaps or []): | for cap in (self._bundlecaps or []): | ||||
| if cap.startswith("includepattern="): | if cap.startswith("includepattern="): | ||||
| includepattern = cap[len("includepattern="):].split('\0') | includepattern = cap[len("includepattern="):].split('\0') | ||||
| elif cap.startswith("excludepattern="): | elif cap.startswith("excludepattern="): | ||||
| excludepattern = cap[len("excludepattern="):].split('\0') | excludepattern = cap[len("excludepattern="):].split('\0') | ||||
| m = match.always(repo.root, '') | m = match.always(repo.root, '') | ||||
| if includepattern or excludepattern: | if includepattern or excludepattern: | ||||
| m = match.match(repo.root, '', None, | m = match.match(repo.root, '', None, | ||||
| includepattern, excludepattern) | includepattern, excludepattern) | ||||
| changedfiles = list([f for f in changedfiles if not m(f)]) | changedfiles = list([f for f in changedfiles if not m(f)]) | ||||
| return orig(self, changedfiles, linknodes, commonrevs, source) | return orig(self, changedfiles, linknodes, commonrevs, source, | ||||
| *args, **kwargs) | |||||
| if util.safehasattr(changegroup, 'cgpacker'): | if util.safehasattr(changegroup, 'cgpacker'): | ||||
| wrapfunction(changegroup.cgpacker, 'generatefiles', generatefiles) | wrapfunction(changegroup.cgpacker, 'generatefiles', generatefiles) | ||||
| else: | else: | ||||
| wrapfunction(changegroup.cg1packer, 'generatefiles', generatefiles) | wrapfunction(changegroup.cg1packer, 'generatefiles', generatefiles) | ||||
| onetime = False | onetime = False | ||||
| def onetimesetup(ui): | def onetimesetup(ui): | ||||