# Module dedicated to host utility code dedicated to changeset rewrite

#

# Copyright 2017 Octobus <contact@octobus.net>

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

# Status: Stabilization of the API in progress

#

#   The content of this module should move into core incrementally once we are

#   happy one piece of it (and hopefully, able to reuse it in other core

#   commands).

from mercurial import (

    obsolete,

    revset,

)

from . import (

    compat,

)

def bookmarksupdater(repo, oldid, tr):

    """Return a callable update(newid) updating the current bookmark

    and bookmarks bound to oldid to newid.

    """

    def updatebookmarks(newid):

        oldbookmarks = repo.nodebookmarks(oldid)

        bmchanges = [(b, newid) for b in oldbookmarks]

        if bmchanges:

            compat.bookmarkapplychanges(repo, tr, bmchanges)

    return updatebookmarks

def disallowednewunstable(repo, revs):

    """Check that editing <revs> will not create disallowed unstable

    (unstable creation is controled by some special config).

    """

    allowunstable = obsolete.isenabled(repo, obsolete.allowunstableopt)

    if allowunstable:

        return revset.baseset()

    return repo.revs("(%ld::) - %ld", revs, revs)