#!/bin/bash

## EXPERIMENT RUNNER
# exemple:  sudo ./scripts/run-3 2 . orig-client 100 100 100, orig-client 100 100 100 . orig-server, orig-server

die() { echo "$*" 1>&2 ; exit 1; }

WAITFOR=2280 # 38min

# SPLIT CLIENT AND SERVER TARGETS
OLD_IFS=$IFS; IFS="."
read -a PARTS <<< "$@"
IFS=$OLD_IFS
[[ ${#PARTS[@]} == 5 ]] || die "Separate count, clients, servers experiments and tor daemon command by dots. I expect 5 parts but got ${#PARTS[@]}"

# EXTRACT DATA
OLD_IFS=$IFS; IFS=","
read -a CONFIG <<< "${PARTS[0]}"
read -a CLIENT_TARGETS <<< "${PARTS[1]}"
read -a SERVER_TARGETS <<< "${PARTS[2]}"
read -a TOR_DAEMON_CLIENT <<< "${PARTS[3]}"
read -a TOR_DAEMON_SERVER <<< "${PARTS[4]}"
IFS=$OLD_IFS

[[ ${#CLIENT_TARGETS[@]} == ${#SERVER_TARGETS[@]} ]] \
  && [[ ${#SERVER_TARGETS[@]} == ${#TOR_DAEMON_CLIENT[@]} ]] \
  && [[ ${#TOR_DAEMON_CLIENT[@]} == ${#TOR_DAEMON_SERVER[@]} ]] \
  || die "Same number of declaration is required. Got ${#CLIENT_TARGETS[@]} clients, ${#SERVER_TARGETS[@]} servers, ${#TOR_DAEMON_CLIENT[@]} Tor daemon clients and ${#TOR_DAEMON_SERVER[@]} Tor daemon servers"
instances=${#CLIENT_TARGETS[@]}
REPEAT=${CONFIG[0]}
IDENTIFIER=${CONFIG[1]}

echo ":: Configuration Summary"
echo "repeat=$REPEAT, identifier=$IDENTIFIER"
for i in $(seq 0 $(expr $instances - 1)); do
  echo "  instance=$i - server=${SERVER_TARGETS[$i]} - client=${CLIENT_TARGETS[$i]} - tor server=${TOR_DAEMON_SERVER[$i]} - tor client=${TOR_DAEMON_CLIENT[$i]}"
done

echo ":: Clear Resources"
for i in $(seq 0 $(expr $instances - 1)); do
  echo "clear $i"
  docker container inspect donarxp_server_${i} > /dev/null 2>&1 && docker kill donarxp_server_${i}
  docker container inspect donarxp_client_${i} > /dev/null 2>&1 && docker kill donarxp_client_${i}
done

echo ":: Bootstrap"
echo "Create output folder..."
docker run \
  --rm \
  --user root \
  -v `pwd`/out:/home/donar/out \
  registry.gitlab.inria.fr/qdufour/donar \
  chown -R 1000:1000 /home/donar

for i in $(seq 0 $(expr $instances - 1)); do
echo "Spawning container for instance=$i..."
docker run \
  --rm \
  -d \
  --name "donarxp_server_${i}" \
  -e HOME='/tmp' \
  -v `pwd`/out:/home/donar/out \
  registry.gitlab.inria.fr/qdufour/donar \
  ${TOR_DAEMON_SERVER[$i]}

docker run \
  --rm \
  -d \
  --name "donarxp_client_${i}" \
  -e HOME='/tmp' \
  -v `pwd`/out:/home/donar/out \
  registry.gitlab.inria.fr/qdufour/donar \
  ${TOR_DAEMON_CLIENT[$i]}
done

sleep 10

echo ":: Start experiment..."
for j in $(seq 1 $REPEAT); do
run_fold=`mktemp -up ./out XXXXXXXXXXXXXXXX`
echo "start - batch_count=$j - folder=$run_fold"

echo "  reset containers"
for i in $(seq 0 $(expr $instances - 1)); do
  docker exec donarxp_client_${i} sh -c "mkdir -p $run_fold-$i; echo repeat=$REPEAT,identifier=$IDENTIFIER > $run_fold-$i/info.txt"
  docker exec donarxp_client_${i} sh -c "echo instance=$i - server=${SERVER_TARGETS[$i]} - client=${CLIENT_TARGETS[$i]} - tor server=${TOR_DAEMON_SERVER[$i]} - tor client=${TOR_DAEMON_CLIENT[$i]}>> $run_fold-$i/info.txt"

  docker exec donarxp_client_${i} sh -c 'killall --quiet -9 bash; killall --quiet -9 donar; killall --quiet -9 measlat; killall --quiet -9 udpecho; killall --quiet -9 torecho'
  docker exec donarxp_server_${i} sh -c 'killall --quiet -9 bash; killall --quiet -9 donar; killall --quiet -9 measlat; killall --quiet -9 udpecho; killall --quiet -9 torecho'
done

echo "  launch servers"
for i in $(seq 0 $(expr $instances - 1)); do
  echo "    instance=$i - exec=${SERVER_TARGETS[$i]}"
  docker exec donarxp_server_$i ${SERVER_TARGETS[$i]} ${run_fold}-$i &
  sleep 5
done

sleep 10
echo "  launch clients"

(subshellpid=$BASHPID 
  (sleep $WAITFOR; echo "  timeout"; kill -9 $subshellpid) & 
  (for i in $(seq 0 $(expr $instances - 1)); do
    echo "    instance=$i - exec=${CLIENT_TARGETS[$i]}"
    docker exec donarxp_client_$i ${CLIENT_TARGETS[$i]} ${run_fold}-$i &
    sleep 10
  done; 
  wait; 
  echo "  ok"; 

  for i in $(seq 0 $(expr $instances - 1)); do
    docker exec donarxp_client_${i} killall -15 donar; 
    docker exec donarxp_server_${i} killall -15 donar; 
  done; 
  
  kill -15 $subshellpid
  
)) 2>/dev/null 

echo "done"
done

echo ":: Clear Resources"
for i in $(seq 0 $(expr $instances - 1)); do
  echo "clear $i"
  docker kill "donarxp_server_${i}"
  docker kill "donarxp_client_${i}"
done