| durin42 | |
| pulkit |
| hg-reviewers |
This will allow the command function in core to infer the presence
of the option without duplicating logic.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | hgext/narrow/narrowcommands.py (7 lines) |
| _("limit the history fetched by distance from heads"))) | _("limit the history fetched by distance from heads"))) | ||||
| extensions.wrapcommand(commands.table, 'archive', archivenarrowcmd) | extensions.wrapcommand(commands.table, 'archive', archivenarrowcmd) | ||||
| def clonenarrowcmd(orig, ui, repo, *args, **opts): | def clonenarrowcmd(orig, ui, repo, *args, **opts): | ||||
| """Wraps clone command, so 'hg clone' first wraps localrepo.clone().""" | """Wraps clone command, so 'hg clone' first wraps localrepo.clone().""" | ||||
| opts = pycompat.byteskwargs(opts) | opts = pycompat.byteskwargs(opts) | ||||
| wrappedextraprepare = util.nullcontextmanager() | wrappedextraprepare = util.nullcontextmanager() | ||||
| opts_narrow = opts['narrow'] | |||||
| narrowspecfile = opts['narrowspec'] | narrowspecfile = opts['narrowspec'] | ||||
| if narrowspecfile: | if narrowspecfile: | ||||
| filepath = os.path.join(pycompat.getcwd(), narrowspecfile) | filepath = os.path.join(pycompat.getcwd(), narrowspecfile) | ||||
| ui.status(_("reading narrowspec from '%s'\n") % filepath) | ui.status(_("reading narrowspec from '%s'\n") % filepath) | ||||
| try: | try: | ||||
| fdata = util.readfile(filepath) | fdata = util.readfile(filepath) | ||||
| except IOError as inst: | except IOError as inst: | ||||
| raise error.Abort(_("cannot read narrowspecs from '%s': %s") % | raise error.Abort(_("cannot read narrowspecs from '%s': %s") % | ||||
| (filepath, encoding.strtolocal(inst.strerror))) | (filepath, encoding.strtolocal(inst.strerror))) | ||||
| includes, excludes, profiles = sparse.parseconfig(ui, fdata, 'narrow') | includes, excludes, profiles = sparse.parseconfig(ui, fdata, 'narrow') | ||||
| if profiles: | if profiles: | ||||
| raise error.Abort(_("cannot specify other files using '%include' in" | raise error.Abort(_("cannot specify other files using '%include' in" | ||||
| " narrowspec")) | " narrowspec")) | ||||
| narrowspec.validatepatterns(includes) | narrowspec.validatepatterns(includes) | ||||
| narrowspec.validatepatterns(excludes) | narrowspec.validatepatterns(excludes) | ||||
| # narrowspec is passed so we should assume that user wants narrow clone | # narrowspec is passed so we should assume that user wants narrow clone | ||||
| opts_narrow = True | opts['narrow'] = True | ||||
| opts['include'].extend(includes) | opts['include'].extend(includes) | ||||
| opts['exclude'].extend(excludes) | opts['exclude'].extend(excludes) | ||||
| if opts_narrow: | if opts['narrow']: | ||||
| def pullbundle2extraprepare_widen(orig, pullop, kwargs): | def pullbundle2extraprepare_widen(orig, pullop, kwargs): | ||||
| # Create narrow spec patterns from clone flags | # Create narrow spec patterns from clone flags | ||||
| includepats = narrowspec.parsepatterns(opts['include']) | includepats = narrowspec.parsepatterns(opts['include']) | ||||
| excludepats = narrowspec.parsepatterns(opts['exclude']) | excludepats = narrowspec.parsepatterns(opts['exclude']) | ||||
| if not includepats and excludepats: | if not includepats and excludepats: | ||||
| # If nothing was included, we assume the user meant to include | # If nothing was included, we assume the user meant to include | ||||
| # everything, except what they asked to exclude. | # everything, except what they asked to exclude. | ||||
| includepats = {'path:.'} | includepats = {'path:.'} | ||||
| pullop.repo.setnarrowpats(includepats, excludepats) | pullop.repo.setnarrowpats(includepats, excludepats) | ||||
| # This will populate 'includepats' etc with the values from the | # This will populate 'includepats' etc with the values from the | ||||
| # narrowspec we just saved. | # narrowspec we just saved. | ||||
| orig(pullop, kwargs) | orig(pullop, kwargs) | ||||
| if opts.get('depth'): | if opts.get('depth'): | ||||
| kwargs['depth'] = opts['depth'] | kwargs['depth'] = opts['depth'] | ||||
| wrappedextraprepare = extensions.wrappedfunction(exchange, | wrappedextraprepare = extensions.wrappedfunction(exchange, | ||||
| '_pullbundle2extraprepare', pullbundle2extraprepare_widen) | '_pullbundle2extraprepare', pullbundle2extraprepare_widen) | ||||
| def pullnarrow(orig, repo, *args, **kwargs): | def pullnarrow(orig, repo, *args, **kwargs): | ||||
| if opts_narrow: | if opts['narrow']: | ||||
| repo.requirements.add(repository.NARROW_REQUIREMENT) | repo.requirements.add(repository.NARROW_REQUIREMENT) | ||||
| repo._writerequirements() | repo._writerequirements() | ||||
| return orig(repo, *args, **kwargs) | return orig(repo, *args, **kwargs) | ||||
| wrappedpull = extensions.wrappedfunction(exchange, 'pull', pullnarrow) | wrappedpull = extensions.wrappedfunction(exchange, 'pull', pullnarrow) | ||||
| with wrappedextraprepare, wrappedpull: | with wrappedextraprepare, wrappedpull: | ||||