# 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 . 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