#!/usr/bin/env bash
EXECFOLDER=$1 # First argument is the working folder as this is launched with cwd of /root
TIMEOUT_SEC=$2
OKSIGNALFILE="$EXECFOLDER/ok.txt"
COMMANDTORUNFILE="$EXECFOLDER/command.txt"
EXITFILE="$EXECFOLDER/exit.txt"
#OUTPUTFILE="/home/test_output_.txt"
end=$((SECONDS+3600))

function finish {
  rm -f "$COMMANDTORUNFILE"
  rm -f "$OKSIGNALFILE"
}
trap finish EXIT

export DEBIAN_FRONTEND=noninteractive

while true
do
        stop=false
        while [ $SECONDS -lt $end ];
        do
            if test -f "$COMMANDTORUNFILE"; then
                # echo "COMMAND FILE FOUND" >> "$OUTPUTFILE" # Leave this here as an example of debugging
                COMMAND="$(cat "$COMMANDTORUNFILE" | awk '{$1=$1;print}')"
                for validCmd in "${@:3}"
                do
                    if [ "$COMMAND" == "$validCmd" ]; then
                        # Eventually we should split the cmd file to be line by line instead of space separated,
                        # but it works for now because the commands are running under sudo
                        IFS=' ' read -ra COMMANDARGS <<< "$COMMAND"
                    fi
                done
                if [ -z "$COMMANDARGS" ]; then
                    rm "$COMMANDTORUNFILE"
                    exit 111777 # Special exit code - arbitrarily picked for when the command is not expected
                fi
                sudo "${COMMANDARGS[@]}" 2> "$EXECFOLDER/stderr.txt" 1> "$EXECFOLDER/stdout.txt"
                STATUSCODE=$?
                echo $STATUSCODE > "$EXECFOLDER/status.txt"
                rm "$COMMANDTORUNFILE"
                touch "$EXECFOLDER/output.txt"
            fi
            if test -f "$OKSIGNALFILE"; then
                rm "$OKSIGNALFILE"
            fi
            if test -f "$EXITFILE"; then
                rm "$EXITFILE"
                exit 0
            fi
            sleep 5
        done
done