/brz/remove-bazaar

#!/usr/bin/python

from bzrlib import plugin

from bzrlib import commands

head="""\

# Programmable completion for the Bazaar-NG bzr command under bash. Source

# this file (or on some systems add it to ~/.bash_completion and start a new

# shell) and bash's completion mechanism will know all about bzr's options!

# Known to work with bash 2.05a with programmable completion and extended

# pattern matching enabled (use 'shopt -s extglob progcomp' to enable

# these if they are not already enabled).

# Based originally on the svn bash completition script.

# Customized by Sven Wilhelm/Icecrash.com

# Adjusted for automatic generation by Martin von Gagern

if shopt -q extglob; then

	_tmp_unset_extglob=""

else

	_tmp_unset_extglob="shopt -u extglob"

fi

shopt -s extglob progcomp

"""

fun="""\

%(function_name)s ()

{

	local cur cmds cmdOpts opt helpCmds optBase i

	COMPREPLY=()

	cur=${COMP_WORDS[COMP_CWORD]}

	cmds='%(cmds)s'

	if [[ $COMP_CWORD -eq 1 ]] ; then

		COMPREPLY=( $( compgen -W "$cmds" -- $cur ) )

		return 0

	fi

	# if not typing an option, or if the previous option required a

	# parameter, then fallback on ordinary filename expansion

	helpCmds='help|--help|h|\?'

	if [[ ${COMP_WORDS[1]} != @($helpCmds) ]] && \

	   [[ "$cur" != -* ]] ; then

		return 0

	fi

	cmdOpts=

	case ${COMP_WORDS[1]} in

%(cases)s\

	*)

		cmdOpts='--help -h'

		;;

	esac

	cmdOpts=" $cmdOpts "

	# take out options already given

	for (( i=2; i<=$COMP_CWORD-1; ++i )) ; do

		opt=${COMP_WORDS[$i]}

		case $opt in

		--*)    optBase=${opt/=*/} ;;

		-*)     optBase=${opt:0:2} ;;

		esac

		cmdOpts=" $cmdOpts "

		cmdOpts=${cmdOpts/ ${optBase} / }

		# take out some alternatives

		case $optBase in

%(optalt)s\

		esac

		# skip next option if this one requires a parameter

		if [[ $opt == @($optsParam) ]] ; then

			((++i))

		fi

	done

	COMPREPLY=( $( compgen -W "$cmdOpts" -- $cur ) )

	return 0

}

"""

tail="""\

complete -F %(function_name)s -o default bzr

$_tmp_unset_extglob

unset _tmp_unset_extglob

"""

def bash_completion_function(out, function_name="_bzr", function_only=False):

    aliases = []

    cases = ""

    optaliases = {}

    reqarg = {}

    for name in sorted(commands.all_command_names()):

        cmd = commands.get_cmd_object(name)

        cases += "\t" + name

        aliases.append(name)

        for alias in cmd.aliases:

            cases += "|" + alias

            aliases.append(alias)

        cases += ")\n"

        plugin = cmd.plugin_name()

        if plugin is not None:

            cases += "\t\t# plugin \"%s\"\n" % plugin

        opts = cmd.options()

        optnames = []

        for optname in sorted(cmd.options()):

            opt = opts[optname]

            optset = set()

            for (name, short_name, optname, help) in opt.iter_switches():

                if short_name is not None:

                    optset.add("-" + short_name)

                if name is not None:

                    optset.add("--" + name)

            for optname in optset:

                if optname not in optaliases:

                    optaliases[optname] = optset

                else:

                    optaliases[optname] &= optset

            optnames.extend(sorted(optset))

        cases += "\t\tcmdOpts='" + " ".join(optnames) + "'\n\t\t;;\n"

    optalt = ""

    for opt1 in sorted(optaliases):

        optset = optaliases[opt1]

        if len(optset) == 1:

            continue

        optalt += "\t\t" + opt1 + ")\n"

        for opt2 in sorted(optset):

            if opt1 != opt2:

                optalt += "\t\t\tcmdOpts=${cmdOpts/ " + opt2 + " / }\n"

        optalt += "\t\t\t;;\n"

    if function_only:

        template = fun

    else:

        template = head + fun + tail

    out.write(template % {"cmds": " ".join(aliases),

                          "cases": cases,

                          "optalt": optalt,

                          "function_name": function_name,

                          })

if __name__ == '__main__':

    import sys

    import locale

    locale.setlocale(locale.LC_ALL, '')

    plugin.load_plugins()

    commands.install_bzr_command_hooks()

    bash_completion_function(sys.stdout)