#!/bin/bash
# -*- coding: utf-8 mode: sh -*- vim:sw=4:sts=4:et:ai:si:sta:fenc=utf-8
source "$(dirname "$0")/lib/ulib/ulib" || exit 1
urequire DEFAULTS

function display_help() {
    uecho "$scriptname: Faire une connexion ssh en lançant automatiquement un screen sur l'hôte distant

USAGE
    $scriptname [user@]host [options]

En principe, hormis l'argument user@host, il ne faudrait spécifier que des
options. Dans le cas où d'autres arguments seraient spécifiés, les meilleurs
efforts sont faits pour lancer ces commandes avant screen."
}

if [ $# -eq 1 -a "$1" == --help ]; then
    display_help
    exit 0
fi

vars="$("$scriptdir/ussh" --parse "$@")" || die
eval "$vars"
[ "${#hosts[*]}" -gt 0 ] || die "Vous devez spécifier l'hôte sur lequel se connecter"

cmd="'$(quoted_args "${args[@]}")
"'
function __ask() {
    local r
    read -p "$1" -t 2 r
    [ $? -gt 128 -o -z "$r" -o "$r" == "o" -o "$r" == "O" -o "$r" == "y" -o "$r" == "Y" ]
}

function __auto_screen() {
    # Si on est déjà dans screen, ne rien faire
    [ -z "$STY" ] || return

    local msgprefix
    local screens count

    screens="$(LANG=C screen -ls | grep -Ei "attached|detached")"
    if [ -n "$screens" ]; then
        count="$(echo "$screens" | wc -l)"
    else
        count=0
    fi
    if [ $count -gt 0 ]; then
        if [ $count -eq 1 ]; then
            msgprefix="Il y a '"$COULEUR_BLEUE"'1 session screen en cours'"$COULEUR_NORMALE"'
Cette session"
        else
            msgprefix="Il y a '"$COULEUR_ROUGE"'$count sessions screen en cours'"$COULEUR_NORMALE"':
$screens
La première session"
        fi
        if __ask "
$msgprefix sera reconnectée automatiquent dans 2 secondes
Voulez-vous reconnecter la session screen? [On] "; then
            exec screen -q -s -/bin/bash -xRR
        else
            exec /bin/bash -l
        fi
    elif __ask "
Une '"$COULEUR_VERTE"'nouvelle session screen'"$COULEUR_NORMALE"' sera lancée automatiquement dans 2 secondes
Voulez-vous lancer une session screen? [On] "; then
        exec screen -q -s -/bin/bash -RR
    else
        exec /bin/bash -l
    fi
}

__auto_screen
'\'

for host in "${hosts[@]}"; do
    ${exec:+exec} "$ssh" "${options[@]}" -t "$host" -- /bin/bash -c "$cmd"
done