# bash completion for screen                               -*- shell-script -*-

_screen_sessions()
{
    local sessions=($(command screen -ls | command sed -ne \
        's|^\t\{1,\}\([0-9]\{1,\}\.[^\t]\{1,\}\).*'"$1"'.*$|\1|p'))
    if [[ $cur == +([0-9])?(.*) ]]; then
        # Complete sessions including pid prefixes
        COMPREPLY=($(compgen -W '${sessions[@]}' -- "$cur"))
    else
        # Create unique completions, dropping pids where possible
        local -A res
        local i tmp
        for i in "${sessions[@]}"; do
            res[${i/#+([0-9])./}]+=" $i"
        done
        for i in "${!res[@]}"; do
            [[ ${res[i]} == \ *\ * ]] && tmp+=" ${res[i]}" || tmp+=" $i"
        done
        COMPREPLY=($(compgen -W '$tmp' -- "$cur"))
    fi
} &&
    _screen()
    {
        local cur prev words cword
        _init_completion || return

        if ((cword == 1)); then
            if [[ $cur == /dev* ]]; then
                COMPREPLY=($(compgen -W "$(
                    shopt -s nullglob
                    printf '%s\n' \
                        /dev/serial/by-id/* /dev/ttyUSB* /dev/ttyACM* 2>/dev/null
                )" \
                    -- "$cur"))
                return
            fi
            if [[ $cur == //* ]]; then
                COMPREPLY=($(compgen -W '//telnet' -- "$cur"))
                return
            fi
        fi

        case ${words[1]} in
            /dev*)
                if ((cword == 2)); then
                    COMPREPLY=($(compgen -W '110 300 600 1200 2400 4800 9600 \
                14400 19200 38400 57600 115200 128000 256000' -- "$cur"))
                    # TODO more, comma separated options
                fi
                return
                ;;
            //telnet)
                ((cword == 2)) && _known_hosts_real -- "$cur"
                return
                ;;
        esac

        if ((cword > 2)); then
            case ${words[cword - 2]} in
                -*[dD])
                    _screen_sessions
                    return
                    ;;
            esac
        fi

        local i
        for ((i = 1; i <= cword; i++)); do
            case ${words[i]} in
                -*[rRdDxscTehpSt])
                    ((i++))
                    continue
                    ;;
                -*)
                    continue
                    ;;
            esac

            _command_offset $i
            return
        done

        case $prev in
            -*[rR])
                # list detached
                _screen_sessions 'Detached'
                return
                ;;
            -*[dD])
                # list attached
                _screen_sessions 'Attached'
                return
                ;;
            -*x)
                # list both
                _screen_sessions
                return
                ;;
            -*s)
                _shells
                return
                ;;
            -*c)
                _filedir
                return
                ;;
            -T)
                _terms
                return
                ;;
            -*[ehpSt])
                return
                ;;
        esac

        if [[ $cur == -* ]]; then
            COMPREPLY=($(compgen -W '$(_parse_help "$1")' -- "$cur"))
        fi
    } &&
        complete -F _screen screen

# ex: filetype=sh
