summaryrefslogtreecommitdiff
path: root/completions/_modules
blob: 36fffdf60c2f85c842c41e79b9cdc6347e162069 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
# 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.
[ -f /etc/profile.d/modules.sh ] || return 1

_module_list ()
{
    local modules="$(command sed 's/:/ /g' <<<$LOADEDMODULES | sort)"
    compgen -W "$modules" -- $1
}

_module_path ()
{
    local modules="$(command sed 's/:/ /g' <<<$MODULEPATH | sort)"
    compgen -W "$modules" -- $1
}

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

    compgen -W "$modules" -- $1
}

# A completion function for the module alias
_module ()
{
    local cur prev words cword
    _init_completion || return

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

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

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

    elif [[ $cword -eq 2 ]]; then
        case $prev in
            add|display|help|load|show|whatis)
                COMPREPLY=( $(_module_avail "$cur") )
                ;;
            rm|switch|swap|unload|update)
                COMPREPLY=( $(_module_list "$cur") )
                ;;
            unuse)
                COMPREPLY=( $(_module_path "$cur") )
                ;;
        esac
    elif [[ $cword -eq 3 ]]; then
        case ${words[1]} in
            swap|switch)
                COMPREPLY=( $(_module_avail "$cur") )
                ;;
        esac
    fi

} &&
complete -F _module -o default module

# ex: filetype=sh