comparison hgext/inhibit.py @ 1241:3625d006e81b

inhibit: Add -D option to the bookmark command

The -D option for bookmark is similar to the -B option for strip.
It deletes the bookmark and prunes the changes underneath it that are
not reachable.

1240:e1347ce2f954

from mercurial import repoview
from mercurial import revset
from mercurial import error
from mercurial import commands
from mercurial import bookmarks

cmdtable = {}
command = cmdutil.command(cmdtable)

def reposetup(ui, repo):

    """ Add inhibition markers to every obsolete bookmarks """
    repo = bkmstoreinst._repo
    bkmstorenodes = [repo[v].node() for v in bkmstoreinst.values()]
    _inhibitmarkers(repo, bkmstorenodes)
    return orig(bkmstoreinst, *args, **kwargs)

# obsolescence inhibitor
########################

def _schedulewrite(tr, obsinhibit):

    extensions.wrapcommand(commands.table, 'update', _update)
    # There are two ways to save bookmark changes during a transation, we
    # wrap both to add inhibition markers.
    extensions.wrapfunction(bookmarks.bmstore, 'recordchange', _bookmarkchanged)
    extensions.wrapfunction(bookmarks.bmstore, 'write', _bookmarkchanged)


def gethashsymbols(tree):
    # Returns the list of symbols of the tree that look like hashes
    # for example for the revset 3::abe3ff it will return ('abe3ff')

1241:3625d006e81b

from mercurial import repoview
from mercurial import revset
from mercurial import error
from mercurial import commands
from mercurial import bookmarks
from mercurial.i18n import _

cmdtable = {}
command = cmdutil.command(cmdtable)

def reposetup(ui, repo):

    """ Add inhibition markers to every obsolete bookmarks """
    repo = bkmstoreinst._repo
    bkmstorenodes = [repo[v].node() for v in bkmstoreinst.values()]
    _inhibitmarkers(repo, bkmstorenodes)
    return orig(bkmstoreinst, *args, **kwargs)

def _bookmark(orig, ui, repo, *bookmarks, **opts):
    """ Add a -D option to the bookmark command, map it to prune -B """
    def getdefaultopts(module, command):
        """ Get default options of a command from a module """
        cmds = [v for k,v in module.cmdtable.items() if command in k]
        assert len(cmds) == 1, "Ambiguous command"
        # Options of the first command that matched
        cmdopts = cmds[0][1]
        optsdict = {}
        for d in cmdopts:
            optsdict[d[1]] = d[2]
        return optsdict

    haspruneopt = opts.get('prune', False)

    if not haspruneopt:
        return orig(ui, repo, *bookmarks, **opts)
    # Call prune -B
    evolve = extensions.find('evolve')
    optsdict = getdefaultopts(evolve, 'prune|obsolete')
    optsdict['bookmark'] = bookmarks[0]
    evolve.cmdprune(ui, repo, **optsdict)

# obsolescence inhibitor
########################

def _schedulewrite(tr, obsinhibit):

    extensions.wrapcommand(commands.table, 'update', _update)
    # There are two ways to save bookmark changes during a transation, we
    # wrap both to add inhibition markers.
    extensions.wrapfunction(bookmarks.bmstore, 'recordchange', _bookmarkchanged)
    extensions.wrapfunction(bookmarks.bmstore, 'write', _bookmarkchanged)
    # Add bookmark -D option
    entry = extensions.wrapcommand(commands.table, 'bookmark', _bookmark)
    entry[1].append(('D','prune',None,
                    _('delete the bookmark and prune the commits underneath')))



def gethashsymbols(tree):
    # Returns the list of symbols of the tree that look like hashes
    # for example for the revset 3::abe3ff it will return ('abe3ff')