# module completion by Ted Stern <stern@cray.com>          -*- shell-script -*-
#
# Use of this file is deprecated, upstream completion is available in
# modules >= 3.2.7, use that instead.
#
# Completion for Environment Modules `module' alias.
#
# See http://sourceforge.net/projects/modules/
#     http://modules.sourceforge.net/
#
# There are several versions of modules that are commonly used.  Older
# Cray UNICOS systems and many other sites use 2.2.2b.  The latest GPL'd
# version is 3.1.6.  But the module alias is somewhat self-documenting
# via the `module help' command, so use that to print the options.
#
# Programmable completion might be more difficult under tcsh since the
# module command is an alias, and the `module avail' command returns
# its output as stderr.

# Test for existence of /etc/profile.d/modules.sh too because we may end up
# being sourced before it and thus before the `module' alias has been defined.
test -f /etc/profile.d/modules.sh || return 1

proc _module_list
{
    var modules = $[ sed 's/:/ /g' <<<$LOADEDMODULES | sort]
    compgen -W $modules -- $1
}

proc _module_path
{
    var modules = $[ sed 's/:/ /g' <<<$MODULEPATH | sort]
    compgen -W $modules -- $1
}

proc _module_avail
{
    var modules = $[ \
        module avail !2 > !1 | \
        command grep -E -v '^(-|$)' | \
        xargs printf '%s\n' | sed -e 's/(default)//g' | sort]

    compgen -W $modules -- $1
}

# A completion function for the module alias
proc _module
{
    var cur = '', prev = '', words = '', cword = ''
    _init_completion || return

    if [[ $cword -eq 1 ]] {
        # First parameter on line -- we expect it to be a mode selection

        var options = ''
        set options = $[ module help !2 > !1 | command grep -E '^[[:space:]]*\+' | \
                    awk '{print $2}' | sed -e 's/|/ /g' | sort]

        setglobal COMPREPLY = '( '$(compgen -W "$options" -- "$cur") )

    } elif [[ $cword -eq 2 ]] {
        match $prev {
            with add|display|help|load|show|whatis
                setglobal COMPREPLY = '( '$(_module_avail "$cur") )
                
            with rm|switch|swap|unload|update
                setglobal COMPREPLY = '( '$(_module_list "$cur") )
                
            with unuse
                setglobal COMPREPLY = '( '$(_module_path "$cur") )
                
        }
    } elif [[ $cword -eq 3 ]] {
        match $(words[1]) {
            with swap|switch
                setglobal COMPREPLY = '( '$(_module_avail "$cur") )
                
        }
    }
    return 0
} &&
complete -F _module -o default module

# ex: ts=4 sw=4 et filetype=sh