comparison mercurial/dispatch.py @ 35224:6e6d0a5b88e6

dispatch: replace _earlyreq*() with new fancyopts-based parser

35223:4edd2202f7d7

        # input/output/error streams
        self.fin = fin
        self.fout = fout
        self.ferr = ferr

        # remember options pre-parsed by _earlyreqopt*()
        self.earlyoptions = {}

        # reposetups which run before extensions, useful for chg to pre-fill
        # low-level repo state (for example, changelog) before extensions.
        self.prereposetups = prereposetups or []

        ferr = util.stderr

    try:
        if not req.ui:
            req.ui = uimod.ui.load()
        if req.ui.plain('strictflags'):
            req.earlyoptions.update(_earlyparseopts(req.args))
        if _earlyreqoptbool(req, 'traceback', ['--traceback']):
            req.ui.setconfig('ui', 'traceback', 'on', '--traceback')

        # set ui streams from the request
        if req.fin:
            req.ui.fin = req.fin

                'pdb': pdb.post_mortem
            }

            # read --config before doing anything else
            # (e.g. to change trust settings for reading .hg/hgrc)
            cfgs = _parseconfig(req.ui,
                                _earlyreqopt(req, 'config', ['--config']))

            if req.repo:
                # copy configs that were passed on the cmdline (--config) to
                # the repo ui
                for sec, name, val in cfgs:

            debugger = ui.config("ui", "debugger")
            debugmod = pdb
            if not debugger or ui.plain():
                # if we are in HGPLAIN mode, then disable custom debugging
                debugger = 'pdb'
            elif _earlyreqoptbool(req, 'debugger', ['--debugger']):
                # This import can be slow for fancy debuggers, so only
                # do it when absolutely necessary, i.e. when actual
                # debugging has been requested
                with demandimport.deactivated():
                    try:

            debugtrace[debugger] = debugmod.set_trace
            debugmortem[debugger] = debugmod.post_mortem

            # enter the debugger before command execution
            if _earlyreqoptbool(req, 'debugger', ['--debugger']):
                ui.warn(_("entering debugger - "
                        "type c to continue starting hg or h for help\n"))

                if (debugger != 'pdb' and
                    debugtrace[debugger] == debugtrace['pdb']):

                return _dispatch(req)
            finally:
                ui.flush()
        except: # re-raises
            # enter the debugger when we hit an exception
            if _earlyreqoptbool(req, 'debugger', ['--debugger']):
                traceback.print_exc()
                debugmortem[debugger](sys.exc_info()[2])
            raise

    return _callcatch(ui, _runcatchfunc)

            raise error.Abort(_('malformed --config option: %r '
                               '(use --config section.name=value)') % cfg)

    return configs

def _earlyparseopts(args):
    options = {}
    fancyopts.fancyopts(args, commands.globalopts, options,
                        gnu=False, early=True,
                        optaliases={'repository': ['repo']})
    return options

def _earlygetopt(aliases, args, strip=True):
    """Return list of values for an option (or aliases).

                pos += 1
        else:
            pos += 1
    return values

def _earlyreqopt(req, name, aliases):
    """Peek a list option without using a full options table"""
    if req.ui.plain('strictflags'):
        return req.earlyoptions[name]
    values = _earlygetopt(aliases, req.args, strip=False)
    req.earlyoptions[name] = values
    return values

def _earlyreqoptstr(req, name, aliases):
    """Peek a string option without using a full options table"""
    if req.ui.plain('strictflags'):
        return req.earlyoptions[name]
    value = (_earlygetopt(aliases, req.args, strip=False) or [''])[-1]
    req.earlyoptions[name] = value
    return value

def _earlyreqoptbool(req, name, aliases):
    """Peek a boolean option without using a full options table

    >>> req = request([b'x', b'--debugger'], uimod.ui())
    >>> _earlyreqoptbool(req, b'debugger', [b'--debugger'])
    True

    >>> req = request([b'x', b'--', b'--debugger'], uimod.ui())
    >>> _earlyreqoptbool(req, b'debugger', [b'--debugger'])
    """
    if req.ui.plain('strictflags'):
        return req.earlyoptions[name]
    try:
        argcount = req.args.index("--")
    except ValueError:
        argcount = len(req.args)
    value = None
    pos = 0
    while pos < argcount:
        arg = req.args[pos]
        if arg in aliases:
            value = True
        pos += 1
    req.earlyoptions[name] = value
    return value

def runcommand(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
    # run pre-hook, and abort if it fails
    hook.hook(lui, repo, "pre-%s" % cmd, True, args=" ".join(fullargs),
              pats=cmdpats, opts=cmdoptions)
    try:

def _dispatch(req):
    args = req.args
    ui = req.ui

    # check for cwd
    cwd = _earlyreqoptstr(req, 'cwd', ['--cwd'])
    if cwd:
        os.chdir(cwd)

    rpath = _earlyreqoptstr(req, 'repository', ["-R", "--repository", "--repo"])
    path, lui = _getlocal(ui, rpath)

    uis = {ui, lui}

    if req.repo:
        uis.add(req.repo.ui)

    if _earlyreqoptbool(req, 'profile', ['--profile']):
        for ui_ in uis:
            ui_.setconfig('profiling', 'enabled', 'true', '--profile')

    profile = lui.configbool('profiling', 'enabled')
    with profiling.profile(lui, enabled=profile) as profiler:

                            # try to infer -R from command args
                            repos = pycompat.maplist(cmdutil.findrepo, args)
                            guess = repos[0]
                            if guess and repos.count(guess) == len(repos):
                                req.args = ['--repository', guess] + fullargs
                                return _dispatch(req)
                        if not path:
                            raise error.RepoError(_("no repository found in"
                                                    " '%s' (.hg not found)")
                                                  % pycompat.getcwd())

35224:6e6d0a5b88e6

        # input/output/error streams
        self.fin = fin
        self.fout = fout
        self.ferr = ferr

        # remember options pre-parsed by _earlyparseopts()
        self.earlyoptions = {}

        # reposetups which run before extensions, useful for chg to pre-fill
        # low-level repo state (for example, changelog) before extensions.
        self.prereposetups = prereposetups or []

        ferr = util.stderr

    try:
        if not req.ui:
            req.ui = uimod.ui.load()
        req.earlyoptions.update(_earlyparseopts(req.ui, req.args))
        if req.earlyoptions['traceback']:
            req.ui.setconfig('ui', 'traceback', 'on', '--traceback')

        # set ui streams from the request
        if req.fin:
            req.ui.fin = req.fin

                'pdb': pdb.post_mortem
            }

            # read --config before doing anything else
            # (e.g. to change trust settings for reading .hg/hgrc)
            cfgs = _parseconfig(req.ui, req.earlyoptions['config'])

            if req.repo:
                # copy configs that were passed on the cmdline (--config) to
                # the repo ui
                for sec, name, val in cfgs:

            debugger = ui.config("ui", "debugger")
            debugmod = pdb
            if not debugger or ui.plain():
                # if we are in HGPLAIN mode, then disable custom debugging
                debugger = 'pdb'
            elif req.earlyoptions['debugger']:
                # This import can be slow for fancy debuggers, so only
                # do it when absolutely necessary, i.e. when actual
                # debugging has been requested
                with demandimport.deactivated():
                    try:

            debugtrace[debugger] = debugmod.set_trace
            debugmortem[debugger] = debugmod.post_mortem

            # enter the debugger before command execution
            if req.earlyoptions['debugger']:
                ui.warn(_("entering debugger - "
                        "type c to continue starting hg or h for help\n"))

                if (debugger != 'pdb' and
                    debugtrace[debugger] == debugtrace['pdb']):

                return _dispatch(req)
            finally:
                ui.flush()
        except: # re-raises
            # enter the debugger when we hit an exception
            if req.earlyoptions['debugger']:
                traceback.print_exc()
                debugmortem[debugger](sys.exc_info()[2])
            raise

    return _callcatch(ui, _runcatchfunc)

            raise error.Abort(_('malformed --config option: %r '
                               '(use --config section.name=value)') % cfg)

    return configs

def _earlyparseopts(ui, args):
    options = {}
    fancyopts.fancyopts(args, commands.globalopts, options,
                        gnu=not ui.plain('strictflags'), early=True,
                        optaliases={'repository': ['repo']})
    return options

def _earlygetopt(aliases, args, strip=True):
    """Return list of values for an option (or aliases).

                pos += 1
        else:
            pos += 1
    return values

def runcommand(lui, repo, cmd, fullargs, ui, options, d, cmdpats, cmdoptions):
    # run pre-hook, and abort if it fails
    hook.hook(lui, repo, "pre-%s" % cmd, True, args=" ".join(fullargs),
              pats=cmdpats, opts=cmdoptions)
    try:

def _dispatch(req):
    args = req.args
    ui = req.ui

    # check for cwd
    cwd = req.earlyoptions['cwd']
    if cwd:
        os.chdir(cwd)

    rpath = req.earlyoptions['repository']
    path, lui = _getlocal(ui, rpath)

    uis = {ui, lui}

    if req.repo:
        uis.add(req.repo.ui)

    if req.earlyoptions['profile']:
        for ui_ in uis:
            ui_.setconfig('profiling', 'enabled', 'true', '--profile')

    profile = lui.configbool('profiling', 'enabled')
    with profiling.profile(lui, enabled=profile) as profiler:

                            # try to infer -R from command args
                            repos = pycompat.maplist(cmdutil.findrepo, args)
                            guess = repos[0]
                            if guess and repos.count(guess) == len(repos):
                                req.args = ['--repository', guess] + fullargs
                                req.earlyoptions['repository'] = guess
                                return _dispatch(req)
                        if not path:
                            raise error.RepoError(_("no repository found in"
                                                    " '%s' (.hg not found)")
                                                  % pycompat.getcwd())