#!/bin/bash

NRUNS=${NRUNS:-4}
XWORDS=${XWORDS:-"./obj_linux_memdbg/xwords"}
DICT=${DICT:-dict.xwd}
HOST=${HOST:-localhost}
PORT=${PORT:-10999}
QUIT=${QUIT:-"-q 2"}
USE_CURSES=${USE_CURSES:="yes"}
WAIT_MAX=${WAIT_MAX:-10}

RUN_NAME=$(basename $0)_$$

usage() {
    echo "usage: $0 <no params>"
    cat <<EOF
The goal of this script is to simulate real-world loads on the relay,
with games starting, stopping, moves happening, etc. over a long time.
It uses ENV variables rather than commandline parms for configuration.
EOF
    echo "    env: NRUNS: number of simultaneous games; default 4"
    echo "    env: DICT: dictionary; default: dict.xwd"
    echo "    env: HOST: remote host; default: localhost"
    echo "    env: PORT: remote port; default: 10999"
    echo "    env: WAIT_MAX: most seconds to wait between moves; default: 10"
    exit 0
}


random() {
    # RANDOM is a bashism
#     RAND=$(dd if=/dev/urandom count=1 2>/dev/null | cksum | cut -f1 -d" ")
#     echo $RAND
    echo $RANDOM
}

game_curses() {
    NAME=$1
    COOKIE=$2
    WAIT=$3
    INDEX=$4
    SERVER_PARAMS=$5
    $XWORDS -u -d $DICT -r $NAME -a $HOST -p $PORT \
        -C $COOKIE $QUIT -z 0:$WAIT >/dev/null -0 $SERVER_PARAMS \
        2>/tmp/$RUN_NAME/log_${COOKIE}_${INDEX}.txt < /dev/null &
}

check_logs() {
    COOKIE=$1
    OK=1
    for LOG in /tmp/$RUN_NAME/log_${COOKIE}_*.txt; do
        if ! grep -q XWPROTO_END_GAME $LOG; then
            echo "$LOG didn't end correctly; check it out."
            OK=0
        fi
    done

    [ 1 = $OK ] && echo "game $COOKIE ended successfully"
}

do_one() {
    CROOT=${1:-$(exec sh -c 'echo $PPID')}
    INDX=1

    while [ -d /tmp/$RUN_NAME ]; do                 # loop forever
        COOKIE="$CROOT:$INDX"
        INDX=$((INDX+1))
        unset ZERO_DONE ONE_DONE TWO_DONE THREE_DONE

        TODO=$(($CROOT % 3))
        TODO=$((TODO+2))
        COUNT=0
        for NAME in Bbbbb Aaaaa Kkkkk Eeeee; do
            [ $COUNT = $TODO ] && break
            while :; do
                RAND=$(random)
                INDEX=$(( $RAND % $TODO ))
                WAIT=$(( $RAND % $WAIT_MAX ))
                case $INDEX in
                    0)
                        if [ -z "$ZERO_DONE" ]; then
                            REMOTES=""
                            for JJ in $(seq $(($TODO-1))); do
                                REMOTES="$REMOTES -N"; 
                            done
                            ZERO_DONE=1
                            if [ "$USE_CURSES" = "yes" ]; then
                                game_curses $NAME $COOKIE $WAIT $INDEX \
                                    "-s $REMOTES"
                            else
                                $XWORDS -d $DICT -r $NAME -s $REMOTES \
                                    -a $HOST -p $PORT -C $COOKIE $QUIT &
                            fi
                            break
                        fi
                        ;;
                    1)
                        if [ -z "$ONE_DONE" ]; then
                            ONE_DONE=1
                            if [ "$USE_CURSES" = "yes" ]; then
                                game_curses $NAME $COOKIE $WAIT $INDEX
                            else
                                $XWORDS -d $DICT -r $NAME -a $HOST -p $PORT \
                                    -C $COOKIE $QUIT &
                            fi
                            break
                        fi
                        ;;
                    2)
                        if [ -z "$TWO_DONE" ]; then
                            TWO_DONE=1
                            if [ "$USE_CURSES" = "yes" ]; then
                                game_curses $NAME $COOKIE $WAIT $INDEX
                            else
                                $XWORDS -d $DICT -r $NAME -a $HOST -p $PORT \
                                    -C $COOKIE $QUIT &
                            fi
                            break
                        fi
                        ;;
                    3)
                        if [ -z "$THREE_DONE" ]; then
                            THREE_DONE=1
                            if [ "$USE_CURSES" = "yes" ]; then
                                game_curses $NAME $COOKIE $WAIT $INDEX
                            else
                                $XWORDS -d $DICT -r $NAME -a $HOST -p $PORT \
                                    -C $COOKIE $QUIT &
                            fi
                            break
                        fi
                        ;;
                esac
            done
            COUNT=$((COUNT+1))
        done

        wait

        check_logs $COOKIE
        sleep $(( $(random) % 60 ))
    done
}

############################################################################
# main

[ -n "$1" ] && usage

mkdir -p /tmp/$RUN_NAME
echo "************************************************************"
echo "* created /tmp/$RUN_NAME; delete it to stop runaway script *"
echo "************************************************************"

for II in $(seq $NRUNS); do
    do_one $II &
done

wait