comparison hgext/rebase.py @ 35388:dd11df900f7f

rebase: replace --inmemory flag with rebase.experimental.inmemory config

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

35387:c0c6df81c9bb

        desc += ' (%s)' % ' '.join(names)
    return desc

class rebaseruntime(object):
    """This class is a container for rebase runtime state"""
    def __init__(self, repo, ui, opts=None):
        if opts is None:
            opts = {}

        # prepared: whether we have rebasestate prepared or not. Currently it
        # decides whether "self.repo" is unfiltered or not.

        # keepopen is not meant for use on the command line, but by
        # other extensions
        self.keepopen = opts.get('keepopen', False)
        self.obsoletenotrebased = {}
        self.obsoletewithoutsuccessorindestination = set()
        self.inmemory = opts.get('inmemory', False)

    @property
    def repo(self):
        if self.prepared:
            return self._repo.unfiltered()

    ('', 'keepbranches', False, _('keep original branch names')),
    ('D', 'detach', False, _('(DEPRECATED)')),
    ('i', 'interactive', False, _('(DEPRECATED)')),
    ('t', 'tool', '', _('specify merge tool')),
    ('c', 'continue', False, _('continue an interrupted rebase')),
    ('',  'inmemory', False, _('run rebase in-memory (EXPERIMENTAL)')),
    ('a', 'abort', False, _('abort an interrupted rebase'))] +
    cmdutil.formatteropts,
    _('[-s REV | -b REV] [-d REV] [OPTION]'))
def rebase(ui, repo, **opts):
    """move changeset (and descendants) to a different branch

    unexpectedly::

      [rebase]
      singletransaction = True

    Return Values:

    Returns 0 on success, 1 if nothing to rebase or there are
    unresolved conflicts.

    """
    if opts.get('continue') or opts.get('abort'):
        # in-memory rebase is not compatible with resuming rebases.
        opts['inmemory'] = False

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

def _origrebase(ui, repo, **opts):
    opts = pycompat.byteskwargs(opts)
    if 'inmemory' not in opts:
        opts['inmemory'] = False
    rbsrt = rebaseruntime(repo, ui, opts)

    with repo.wlock(), repo.lock():
        # Validate input and define rebasing points
        destf = opts.get('dest', None)
        srcf = opts.get('source', None)

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

        tr = None

            with util.acceptintervention(dsguard):
                rbsrt._performrebase(tr)

        rbsrt._finishrebase()

def _definedestmap(ui, repo, destf=None, srcf=None, basef=None, revf=None,
                   destspace=None, opts=None):
    """use revisions argument to define destmap {srcrev: destrev}"""
    if revf is None:
        revf = []

    # destspace is here to work around issues with `hg pull --rebase` see

    if revf and basef:
        raise error.Abort(_('cannot specify both a revision and a base'))
    if revf and srcf:
        raise error.Abort(_('cannot specify both a revision and a source'))

    if not opts['inmemory']:
        cmdutil.checkunfinished(repo)
        cmdutil.bailifchanged(repo)

    if ui.configbool('commands', 'rebase.requiredest') and not destf:
        raise error.Abort(_('you must specify a destination'),

    #
    # Note that there are cases where this isn't true -- e.g., rebasing large
    # stacks that include the WCP. However, I'm not yet sure where the cutoff
    # is.
    rebasingwcp = repo['.'].rev() in rebaseset
    if opts['inmemory'] and rebasingwcp:
        opts['inmemory'] = False
        # Check these since we did not before.
        cmdutil.checkunfinished(repo)
        cmdutil.bailifchanged(repo)

    if not destf:

35388:dd11df900f7f

        desc += ' (%s)' % ' '.join(names)
    return desc

class rebaseruntime(object):
    """This class is a container for rebase runtime state"""
    def __init__(self, repo, ui, inmemory=False, opts=None):
        if opts is None:
            opts = {}

        # prepared: whether we have rebasestate prepared or not. Currently it
        # decides whether "self.repo" is unfiltered or not.

        # keepopen is not meant for use on the command line, but by
        # other extensions
        self.keepopen = opts.get('keepopen', False)
        self.obsoletenotrebased = {}
        self.obsoletewithoutsuccessorindestination = set()
        self.inmemory = inmemory

    @property
    def repo(self):
        if self.prepared:
            return self._repo.unfiltered()

    ('', 'keepbranches', False, _('keep original branch names')),
    ('D', 'detach', False, _('(DEPRECATED)')),
    ('i', 'interactive', False, _('(DEPRECATED)')),
    ('t', 'tool', '', _('specify merge tool')),
    ('c', 'continue', False, _('continue an interrupted rebase')),
    ('a', 'abort', False, _('abort an interrupted rebase'))] +
    cmdutil.formatteropts,
    _('[-s REV | -b REV] [-d REV] [OPTION]'))
def rebase(ui, repo, **opts):
    """move changeset (and descendants) to a different branch

    unexpectedly::

      [rebase]
      singletransaction = True

    By default, rebase writes to the working copy, but you can configure it to
    run in-memory for for better performance, and to allow it to run if the
    working copy is dirty::

      [rebase]
      experimental.inmemory = True

    Return Values:

    Returns 0 on success, 1 if nothing to rebase or there are
    unresolved conflicts.

    """
    inmemory = ui.configbool('rebase', 'experimental.inmemory')
    if opts.get('continue') or opts.get('abort'):
        # in-memory rebase is not compatible with resuming rebases.
        inmemory = False

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

def _origrebase(ui, repo, inmemory=False, **opts):
    opts = pycompat.byteskwargs(opts)
    rbsrt = rebaseruntime(repo, ui, inmemory, opts)

    with repo.wlock(), repo.lock():
        # Validate input and define rebasing points
        destf = opts.get('dest', None)
        srcf = opts.get('source', None)

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

        tr = None

            with util.acceptintervention(dsguard):
                rbsrt._performrebase(tr)

        rbsrt._finishrebase()

def _definedestmap(ui, repo, rbsrt, destf=None, srcf=None, basef=None,
                   revf=None, destspace=None):
    """use revisions argument to define destmap {srcrev: destrev}"""
    if revf is None:
        revf = []

    # destspace is here to work around issues with `hg pull --rebase` see

    if revf and basef:
        raise error.Abort(_('cannot specify both a revision and a base'))
    if revf and srcf:
        raise error.Abort(_('cannot specify both a revision and a source'))

    if not rbsrt.inmemory:
        cmdutil.checkunfinished(repo)
        cmdutil.bailifchanged(repo)

    if ui.configbool('commands', 'rebase.requiredest') and not destf:
        raise error.Abort(_('you must specify a destination'),

    #
    # Note that there are cases where this isn't true -- e.g., rebasing large
    # stacks that include the WCP. However, I'm not yet sure where the cutoff
    # is.
    rebasingwcp = repo['.'].rev() in rebaseset
    if rbsrt.inmemory and rebasingwcp:
        rbsrt.inmemory = False
        # Check these since we did not before.
        cmdutil.checkunfinished(repo)
        cmdutil.bailifchanged(repo)

    if not destf: