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

# désactiver le verrouillage
export LOCKPRG=/bin/true

COULEUR_NORMALE=$'\e[0m'
COULEUR_ROUGE=$'\e[31;1m'
COULEUR_VERTE=$'\e[32;1m'
COULEUR_JAUNE=$'\e[33;1m'
COULEUR_BLEUE=$'\e[34;1m'
COULEUR_BLANCHE=$'\e[37;1m'

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

    # Si screen pas installé, ne rien faire
    local dir screen="$(which screen 2>/dev/null)"
    if [ -z "$screen" ]; then
        for dir in /usr/local/bin /opt/bin /usr/bin /bin; do
            if [ -x "$dir/screen" ]; then
                screen="$dir/screen"
                PATH="$PATH:$dir"
                break
            fi
        done
    fi
    if [ -z "$screen" ]; then
        echo "${COULEUR_JAUNE}WARNING${COULEUR_NORMALE} screen introuvable. une session bash classique sera lancée"
        exec /bin/bash -l
    fi

    # corriger TERM pour les vieilles versions de Linux
    case "$TERM" in
    xterm*) TERM=xterm;;
    screen*) TERM=screen;;
    esac
    export TERM

    local msgprefix
    local screens count
    local session_name="nutools-$USER"

    screens="$(LANG=C screen -ls | grep -Ei "${session_name}.*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 -S "$session_name"
        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 -S "$session_name"
    else
        exec /bin/bash -l
    fi
}

if [ $# -gt 0 ]; then
    # lancer les commandes fournies avant de lancer SCREEN
    "$@"
fi
__auto_screen