#!/bin/bash

 # tests/exec.sh - Test DevStack run_process() and stop_process()

 #
 # exec.sh start|stop|status
 #

 # Set USE_SCREEN True|False to change use of screen.

 #
 # This script emulates the basic exec envirnment in ``stack.sh`` to test
 # the process spawn and kill operations.
 
 if [[ -z $1 ]]; then
     echo "$0 start|stop"
     exit 1
 fi
 
 TOP_DIR=$(cd $(dirname "$0")/.. && pwd)
 source $TOP_DIR/functions
 
 USE_SCREEN=${USE_SCREEN:-False}
 
 ENABLED_SERVICES=fake-service
 
 SERVICE_DIR=/tmp
 SCREEN_NAME=test
 SCREEN_LOGDIR=${SERVICE_DIR}/${SCREEN_NAME}
 
 
 # Kill background processes on exit
 trap clean EXIT
 clean() {
     local r=$?
     jobs -p
     kill >/dev/null 2>&1 $(jobs -p)
     exit $r
 }
 
 
 # Exit on any errors so that errors don't compound
 trap failed ERR
 failed() {
     local r=$?
     jobs -p
     kill >/dev/null 2>&1 $(jobs -p)
     set +o xtrace
     [ -n "$LOGFILE" ] && echo "${0##*/} failed: full log in $LOGFILE"
     exit $r
 }
 
 function status {
     if [[ -r $SERVICE_DIR/$SCREEN_NAME/fake-service.pid ]]; then
         pstree -pg $(cat $SERVICE_DIR/$SCREEN_NAME/fake-service.pid)
     fi
     ps -ef | grep fake
 }
 
 function setup_screen {
 if [[ ! -d $SERVICE_DIR/$SCREEN_NAME ]]; then
     rm -rf $SERVICE_DIR/$SCREEN_NAME
     mkdir -p $SERVICE_DIR/$SCREEN_NAME
 fi
 
 if [[ "$USE_SCREEN" == "True" ]]; then
     # Create a new named screen to run processes in
     screen -d -m -S $SCREEN_NAME -t shell -s /bin/bash
     sleep 1
 
     # Set a reasonable status bar
     if [ -z "$SCREEN_HARDSTATUS" ]; then
         SCREEN_HARDSTATUS='%{= .} %-Lw%{= .}%> %n%f %t*%{= .}%+Lw%< %-=%{g}(%{d}%H/%l%{g})'
     fi
     screen -r $SCREEN_NAME -X hardstatus alwayslastline "$SCREEN_HARDSTATUS"
 fi
 
 # Clear screen rc file
 SCREENRC=$TOP_DIR/tests/$SCREEN_NAME-screenrc
 if [[ -e $SCREENRC ]]; then
     echo -n > $SCREENRC
 fi
 }
 
 # Mimic logging
     # Set up output redirection without log files
     # Copy stdout to fd 3
     exec 3>&1
     if [[ "$VERBOSE" != "True" ]]; then
         # Throw away stdout and stderr
         #exec 1>/dev/null 2>&1
         :
     fi
     # Always send summary fd to original stdout
     exec 6>&3
 
 
 if [[ "$1" == "start" ]]; then
     echo "Start service"
     setup_screen

     run_process fake-service "$TOP_DIR/tests/fake-service.sh"

     sleep 1
     status
 elif [[ "$1" == "stop" ]]; then
     echo "Stop service"

     stop_process fake-service

     status
 elif [[ "$1" == "status" ]]; then
     status
 else
     echo "Unknown command"
     exit 1
 fi