comparison hgext/rebase.py @ 39100:e9e742bd0501

rebase: use action variable to select things to do

39099:3f9a5d746cf6

    """
    opts = pycompat.byteskwargs(opts)
    inmemory = ui.configbool('rebase', 'experimental.inmemory')
    dryrun = opts.get('dry_run')
    confirm = opts.get('confirm')
    stop = opts.get('stop')
    selactions = [k for k in ['abort', 'stop', 'continue'] if opts.get(k)]
    if len(selactions) > 1:
        raise error.Abort(_('cannot use --%s with --%s')
                          % tuple(selactions[:2]))
    action = selactions[0] if selactions else None

    if confirm and action:
        raise error.Abort(_('cannot specify both --confirm and --%s') % action)
    if dryrun and confirm:
        raise error.Abort(_('cannot specify both --confirm and --dry-run'))

    if (opts.get('continue') or opts.get('abort') or
        repo.currenttransaction() is not None):
        # in-memory rebase is not compatible with resuming rebases.
        # (Or if it is run within a transaction, since the restart logic can
        # fail the entire transaction.)
        inmemory = False

        userrevs = list(repo.revs(opts.get('auto_orphans')))
        opts['rev'] = [revsetlang.formatspec('%ld and orphan()', userrevs)]
        opts['dest'] = '_destautoorphanrebase(SRC)'

    if dryrun or confirm:
        return _dryrunrebase(ui, repo, opts)
    elif stop:
        rbsrt = rebaseruntime(repo, ui)
        rbsrt.restorestatus()

        if rbsrt.collapsef:
            raise error.Abort(_("cannot stop in --collapse session"))

        try:
            # in-memory merge doesn't support conflicts, so if we hit any, abort
            # and re-run as an on-disk merge.
            overrides = {('rebase', 'singletransaction'): True}
            with ui.configoverride(overrides, 'rebase'):
                return _dorebase(ui, repo, opts, inmemory=inmemory)
        except error.InMemoryMergeConflictsError:
            ui.warn(_('hit merge conflicts; re-running rebase without in-memory'
                      ' merge\n'))
            _dorebase(ui, repo, {'abort': True})
            return _dorebase(ui, repo, opts, inmemory=False)
    else:
        return _dorebase(ui, repo, opts)

def _dryrunrebase(ui, repo, opts):
    rbsrt = rebaseruntime(repo, ui, inmemory=True, opts=opts)
    confirm = opts.get('confirm')
    if confirm:
        ui.status(_('starting in-memory rebase\n'))
    else:

    with repo.wlock(), repo.lock():
        needsabort = True
        try:
            overrides = {('rebase', 'singletransaction'): True}
            with ui.configoverride(overrides, 'rebase'):
                _origrebase(ui, repo, opts, rbsrt, inmemory=True,
                            leaveunfinished=True)
        except error.InMemoryMergeConflictsError:
            ui.status(_('hit a merge conflict\n'))
            return 1
        else:

            if needsabort:
                # no need to store backup in case of dryrun
                rbsrt._prepareabortorcontinue(isabort=True, backup=False,
                                              suppwarns=True)

def _dorebase(ui, repo, opts, inmemory=False):
    rbsrt = rebaseruntime(repo, ui, inmemory, opts)
    return _origrebase(ui, repo, opts, rbsrt, inmemory=inmemory)

def _origrebase(ui, repo, opts, rbsrt, inmemory=False, leaveunfinished=False):
    with repo.wlock(), repo.lock():
        # Validate input and define rebasing points
        destf = opts.get('dest', None)
        srcf = opts.get('source', None)
        basef = opts.get('base', None)
        revf = opts.get('rev', [])
        # search default destination in this space
        # used in the 'hg pull --rebase' case, see issue 5214.
        destspace = opts.get('_destspace')
        contf = opts.get('continue')
        abortf = opts.get('abort')
        if opts.get('interactive'):
            try:
                if extensions.find('histedit'):
                    enablehistedit = ''
            except KeyError:

        if rbsrt.collapsemsg and not rbsrt.collapsef:
            raise error.Abort(
                _('message can only be specified with collapse'))

        if contf or abortf:
            if rbsrt.collapsef:
                raise error.Abort(
                    _('cannot use collapse with continue or abort'))
            if srcf or basef or destf:
                raise error.Abort(
                    _('abort and continue do not allow specifying revisions'))
            if abortf and opts.get('tool', False):
                ui.warn(_('tool option will be ignored\n'))
            if contf:
                ms = mergemod.mergestate.read(repo)
                mergeutil.checkunresolved(ms)

            retcode = rbsrt._prepareabortorcontinue(abortf)
            if retcode is not None:
                return retcode
        else:
            destmap = _definedestmap(ui, repo, inmemory, destf, srcf, basef,
                                     revf, destspace=destspace)

39100:e9e742bd0501

    """
    opts = pycompat.byteskwargs(opts)
    inmemory = ui.configbool('rebase', 'experimental.inmemory')
    dryrun = opts.get('dry_run')
    confirm = opts.get('confirm')
    selactions = [k for k in ['abort', 'stop', 'continue'] if opts.get(k)]
    if len(selactions) > 1:
        raise error.Abort(_('cannot use --%s with --%s')
                          % tuple(selactions[:2]))
    action = selactions[0] if selactions else None

    if confirm and action:
        raise error.Abort(_('cannot specify both --confirm and --%s') % action)
    if dryrun and confirm:
        raise error.Abort(_('cannot specify both --confirm and --dry-run'))

    if action in {'abort', 'continue'} or repo.currenttransaction() is not None:
        # in-memory rebase is not compatible with resuming rebases.
        # (Or if it is run within a transaction, since the restart logic can
        # fail the entire transaction.)
        inmemory = False

        userrevs = list(repo.revs(opts.get('auto_orphans')))
        opts['rev'] = [revsetlang.formatspec('%ld and orphan()', userrevs)]
        opts['dest'] = '_destautoorphanrebase(SRC)'

    if dryrun or confirm:
        return _dryrunrebase(ui, repo, action, opts)
    elif action == 'stop':
        rbsrt = rebaseruntime(repo, ui)
        rbsrt.restorestatus()

        if rbsrt.collapsef:
            raise error.Abort(_("cannot stop in --collapse session"))

        try:
            # in-memory merge doesn't support conflicts, so if we hit any, abort
            # and re-run as an on-disk merge.
            overrides = {('rebase', 'singletransaction'): True}
            with ui.configoverride(overrides, 'rebase'):
                return _dorebase(ui, repo, action, opts, inmemory=inmemory)
        except error.InMemoryMergeConflictsError:
            ui.warn(_('hit merge conflicts; re-running rebase without in-memory'
                      ' merge\n'))
            _dorebase(ui, repo, action='abort')
            return _dorebase(ui, repo, action, opts, inmemory=False)
    else:
        return _dorebase(ui, repo, action, opts)

def _dryrunrebase(ui, repo, action, opts):
    rbsrt = rebaseruntime(repo, ui, inmemory=True, opts=opts)
    confirm = opts.get('confirm')
    if confirm:
        ui.status(_('starting in-memory rebase\n'))
    else:

    with repo.wlock(), repo.lock():
        needsabort = True
        try:
            overrides = {('rebase', 'singletransaction'): True}
            with ui.configoverride(overrides, 'rebase'):
                _origrebase(ui, repo, action, opts, rbsrt, inmemory=True,
                            leaveunfinished=True)
        except error.InMemoryMergeConflictsError:
            ui.status(_('hit a merge conflict\n'))
            return 1
        else:

            if needsabort:
                # no need to store backup in case of dryrun
                rbsrt._prepareabortorcontinue(isabort=True, backup=False,
                                              suppwarns=True)

def _dorebase(ui, repo, action, opts, inmemory=False):
    rbsrt = rebaseruntime(repo, ui, inmemory, opts)
    return _origrebase(ui, repo, action, opts, rbsrt, inmemory=inmemory)

def _origrebase(ui, repo, action, opts, rbsrt, inmemory=False,
                leaveunfinished=False):
    assert action != 'stop'
    with repo.wlock(), repo.lock():
        # Validate input and define rebasing points
        destf = opts.get('dest', None)
        srcf = opts.get('source', None)
        basef = opts.get('base', None)
        revf = opts.get('rev', [])
        # search default destination in this space
        # used in the 'hg pull --rebase' case, see issue 5214.
        destspace = opts.get('_destspace')
        if opts.get('interactive'):
            try:
                if extensions.find('histedit'):
                    enablehistedit = ''
            except KeyError:

        if rbsrt.collapsemsg and not rbsrt.collapsef:
            raise error.Abort(
                _('message can only be specified with collapse'))

        if action:
            if rbsrt.collapsef:
                raise error.Abort(
                    _('cannot use collapse with continue or abort'))
            if srcf or basef or destf:
                raise error.Abort(
                    _('abort and continue do not allow specifying revisions'))
            if action == 'abort' and opts.get('tool', False):
                ui.warn(_('tool option will be ignored\n'))
            if action == 'continue':
                ms = mergemod.mergestate.read(repo)
                mergeutil.checkunresolved(ms)

            retcode = rbsrt._prepareabortorcontinue(isabort=(action == 'abort'))
            if retcode is not None:
                return retcode
        else:
            destmap = _definedestmap(ui, repo, inmemory, destf, srcf, basef,
                                     revf, destspace=destspace)