summaryrefslogtreecommitdiffstats
path: root/completions/dmypy
blob: 7fab74f50d8a8bdb97b402524defa809b16449ac (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
# dmypy completion                                         -*- shell-script -*-

_comp_cmd_dmypy()
{
    local cur prev words cword comp_args
    _comp_initialize -- "$@" || return

    case $prev in
        --help | --version | -[hV])
            return
            ;;
        --status-file)
            _comp_compgen_filedir
            return
            ;;
    esac

    local cmd="" has_cmd="" i
    for ((i = 1; i < cword; i++)); do
        if [[ ${words[i]} != -* && ${words[i - 1]} != --status-file ]]; then
            cmd=${words[i]}
            has_cmd=set
            break
        fi
    done

    case ${cmd-} in
        check | run)
            _comp_compgen_filedir '@(py|pyi)'
            return
            ;;
    esac

    if [[ $cur == -* ]]; then
        _comp_compgen_help
        return
    fi

    if [[ ! $has_cmd ]]; then
        local cmds=$("$1" --help 2>&1 |
            command sed -ne '/positional arguments/{p;n;p;q;}' |
            command sed -ne 's/{\(.*\)}/\1/p')
        _comp_compgen -F , -- -W '$cmds'
        return
    fi
} &&
    complete -F _comp_cmd_dmypy dmypy

# ex: filetype=sh