comparison hgext/shelve.py @ 30522:7b3136bc7bfd

shelve: make --keep option survive user intervention (issue5431)

Currently if user runs 'hg unshelve --keep' and merge conflicts
occur, the information about --keep provided by user is lost and
shelf is deleted after 'hg unshelve --continue'. This is obviously
not desired, so this patch fixes it.

30521:86cd09bc13ba

    Handles saving and restoring a shelved state. Ensures that different
    versions of a shelved state are possible and handles them appropriately.
    """
    _version = 1
    _filename = 'shelvedstate'

    @classmethod
    def load(cls, repo):
        fp = repo.vfs(cls._filename)
        try:

            wctx = nodemod.bin(fp.readline().strip())
            pendingctx = nodemod.bin(fp.readline().strip())
            parents = [nodemod.bin(h) for h in fp.readline().split()]
            stripnodes = [nodemod.bin(h) for h in fp.readline().split()]
            branchtorestore = fp.readline().strip()
        except (ValueError, TypeError) as err:
            raise error.CorruptedState(str(err))
        finally:
            fp.close()

            obj.wctx = repo[wctx]
            obj.pendingctx = repo[pendingctx]
            obj.parents = parents
            obj.stripnodes = stripnodes
            obj.branchtorestore = branchtorestore
        except error.RepoLookupError as err:
            raise error.CorruptedState(str(err))

        return obj

    @classmethod
    def save(cls, repo, name, originalwctx, pendingctx, stripnodes,
             branchtorestore):
        fp = repo.vfs(cls._filename, 'wb')
        fp.write('%i\n' % cls._version)
        fp.write('%s\n' % name)
        fp.write('%s\n' % nodemod.hex(originalwctx.node()))
        fp.write('%s\n' % nodemod.hex(pendingctx.node()))
        fp.write('%s\n' %
                 ' '.join([nodemod.hex(p) for p in repo.dirstate.parents()]))
        fp.write('%s\n' %
                 ' '.join([nodemod.hex(n) for n in stripnodes]))
        fp.write('%s\n' % branchtorestore)
        fp.close()

    @classmethod
    def clear(cls, repo):
        util.unlinkpath(repo.join(cls._filename), ignoremissing=True)

        tr.close()

        stripnodes = [repo.changelog.node(rev)
                      for rev in xrange(oldtiprev, len(repo))]
        shelvedstate.save(repo, basename, pctx, tmpwctx, stripnodes,
                          branchtorestore)

        util.rename(repo.join('rebasestate'),
                    repo.join('unshelverebasestate'))
        raise error.InterventionRequired(
            _("unresolved conflicts (see 'hg resolve', then "

        if abortf and opts.get('tool', False):
            ui.warn(_('tool option will be ignored\n'))

        try:
            state = shelvedstate.load(repo)
        except IOError as err:
            if err.errno != errno.ENOENT:
                raise
            cmdutil.wrongtooltocontinue(repo, _('unshelve'))
        except error.CorruptedState as err:

30522:7b3136bc7bfd

    Handles saving and restoring a shelved state. Ensures that different
    versions of a shelved state are possible and handles them appropriately.
    """
    _version = 1
    _filename = 'shelvedstate'
    _keep = 'keep'
    _nokeep = 'nokeep'

    @classmethod
    def load(cls, repo):
        fp = repo.vfs(cls._filename)
        try:

            wctx = nodemod.bin(fp.readline().strip())
            pendingctx = nodemod.bin(fp.readline().strip())
            parents = [nodemod.bin(h) for h in fp.readline().split()]
            stripnodes = [nodemod.bin(h) for h in fp.readline().split()]
            branchtorestore = fp.readline().strip()
            keep = fp.readline().strip() == cls._keep
        except (ValueError, TypeError) as err:
            raise error.CorruptedState(str(err))
        finally:
            fp.close()

            obj.wctx = repo[wctx]
            obj.pendingctx = repo[pendingctx]
            obj.parents = parents
            obj.stripnodes = stripnodes
            obj.branchtorestore = branchtorestore
            obj.keep = keep
        except error.RepoLookupError as err:
            raise error.CorruptedState(str(err))

        return obj

    @classmethod
    def save(cls, repo, name, originalwctx, pendingctx, stripnodes,
             branchtorestore, keep=False):
        fp = repo.vfs(cls._filename, 'wb')
        fp.write('%i\n' % cls._version)
        fp.write('%s\n' % name)
        fp.write('%s\n' % nodemod.hex(originalwctx.node()))
        fp.write('%s\n' % nodemod.hex(pendingctx.node()))
        fp.write('%s\n' %
                 ' '.join([nodemod.hex(p) for p in repo.dirstate.parents()]))
        fp.write('%s\n' %
                 ' '.join([nodemod.hex(n) for n in stripnodes]))
        fp.write('%s\n' % branchtorestore)
        fp.write('%s\n' % (cls._keep if keep else cls._nokeep))
        fp.close()

    @classmethod
    def clear(cls, repo):
        util.unlinkpath(repo.join(cls._filename), ignoremissing=True)

        tr.close()

        stripnodes = [repo.changelog.node(rev)
                      for rev in xrange(oldtiprev, len(repo))]
        shelvedstate.save(repo, basename, pctx, tmpwctx, stripnodes,
                          branchtorestore, opts.get('keep'))

        util.rename(repo.join('rebasestate'),
                    repo.join('unshelverebasestate'))
        raise error.InterventionRequired(
            _("unresolved conflicts (see 'hg resolve', then "

        if abortf and opts.get('tool', False):
            ui.warn(_('tool option will be ignored\n'))

        try:
            state = shelvedstate.load(repo)
            if opts.get('keep') is None:
                opts['keep'] = state.keep
        except IOError as err:
            if err.errno != errno.ENOENT:
                raise
            cmdutil.wrongtooltocontinue(repo, _('unshelve'))
        except error.CorruptedState as err: