timeout/timeout
Simon Vieille 0ee1015dcb
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
add verbose option
2023-01-01 12:07:17 +01:00

79 lines
1.2 KiB
Bash
Executable file

#!/bin/sh
usage() {
printf "Usage: %s [-t TIMEOUT] [-c COMMAND]\n\n" "$0"
printf "TIMEOUT and COMMAND can be set as env vars.\n"
}
on_interrupt() {
print "Process aborted!\n"
exit 130
}
create_script() {
printf "#!/bin/sh\n%s\n" "$1" > "$2"
chmod +x "$SCRIPT_FILE"
}
run_script() {
if [ "$2" -eq 0 ]; then
"$SCRIPT_FILE" >/dev/null 2>&1 &
else
"$SCRIPT_FILE" >&2 &
fi
printf "%d" $!
}
stop_delete_script() {
kill -9 "$1"
rm -f "$2"
}
check_pid() {
PID=$1
LOOP=$2
for _ in $(seq 1 "$((LOOP*2))"); do
test -d "/proc/$PID/" && sleep 0.5 || STATUS=0
done
printf "%d" ${STATUS:-1}
}
main() {
while getopts "ht:c:v" option; do
case "${option}" in
h) usage; exit 0;;
t) TIMEOUT="$OPTARG";;
c) COMMAND="$OPTARG";;
v) VERBOSE=1;;
*) usage; exit 1;;
esac
done
TIMEOUT="${TIMEOUT:-10}"
VERBOSE="${VERBOSE:-0}"
if [ -z "$COMMAND" ]; then
printf "Command is required!\n"
exit 1
fi
SCRIPT_FILE="$(mktemp)"
create_script "$COMMAND" "$SCRIPT_FILE"
SCRIPT_PID=$(run_script "$SCRIPT_FILE" "$VERBOSE")
EXIT_STATUS=$(check_pid "$SCRIPT_PID" "$TIMEOUT")
stop_delete_script "$SCRIPT_PID" "$SCRIPT_FILE"
exit "$EXIT_STATUS"
}
trap on_interrupt INT
main "$@"