#!/bin/bash
# -*- coding: utf-8 mode: sh -*- vim:sw=4:sts=4:et:ai:si:sta:fenc=utf-8

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]

OPTIONS
note: les options de ssh doivent se trouver *APRES* [user@]host, pour simplifier
leur analyse.
L'option -t de ssh est forcée, pour permettre la connexion par screen
Les options suivantes doivent se trouver *AVANT* le premier argument:
    -S SSH
        Spécifier l'exécutable à utiliser pour lancer ssh."
}

source "$(dirname "$0")/ulib/ulib" &&
urequire DEFAULTS ||
exit 1

ssh=
parse_opts + "${PRETTYOPTS[@]}" \
    --help '$exit_with display_help' \
    -S:,--ssh ssh= \
    @ args -- "$@" && set -- "${args[@]}" || die "$args"

userhost="$1"; shift
[ -n "$userhost" ] || die "Vous devez spécifier l'hôte sur lequel faire la connexion"

[ -n "$ssh" ] || ssh=ssh

exec "$ssh" -t "$@" "$userhost" /bin/bash -c \''
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
'\'