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

_screen_sessions()
{
local sessions=( $( command screen -ls | 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 > 2)); then
case ${words[cword-2]} in
-[dD])
_screen_sessions
return 0
;;
esac
fi

local i
for (( i=1; i <= cword; i++ )); do
case ${words[i]} in
-r|-R|-d|-D|-x|-s|-c|-T|-e|-h|-p|-S|-t)
(( i++ ))
continue
;;
-*)
continue
;;
esac

_command_offset $i
return
done

case $prev in
-[rR])
# list detached
_screen_sessions 'Detached'
return 0
;;
-[dD])
# list attached
_screen_sessions 'Attached'
return 0
;;
-x)
# list both
_screen_sessions
return 0
;;
-s)
_shells
return 0
;;
-c)
_filedir
return 0
;;
-T)
_terms
return 0
;;
-e|-h|-p|-S|-t)
return 0
;;
esac

if [[ "$cur" == -* ]]; then
COMPREPLY=( $( compgen -W '-a -A -c -d -D -e -f -fn -fa -h -i -ln -list
-L -m -O -p -q -r -R -s -S -t -T -U -v -wipe -x -X --help
--version' -- "$cur" ) )
fi
} &&
complete -F _screen screen

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