atrun.sh

#!/bin/sh

# 2020-09-13 19:00 2020-09-14 00:00 3 /opt/ioi/sbin/contest.sh prep 0

SCHEDULE="/opt/ioi/misc/schedule2.txt"

TIMENOW=$(date +"%Y-%m-%d %H:%M")

schedule()
{
	JOBID=$1

	if [ $(wc -l $SCHEDULE | cut -d\  -f1) -lt "$JOBID" ]; then
		logger -p local0.info "ATRUN: Scheduling job $JOBID does not exist"
		return 1
	fi

	# Get details of job to schedule
	JOB=$(cat $SCHEDULE | head -$JOBID | tail -1)
	ATTIME=$(echo "$JOB" | awk '{print($2, $1)}')
	NEXTTIME=$(echo "$JOB" | awk '{print($1, $2)}')
	CMD=$(echo "$JOB" | cut -d\  -f6-)

	# Check if the job is over
	if [ "$TIMENOW" \> "$NEXTTIME" -o "$TIMENOW" = "$NEXTTIME" ]; then
		logger -p local0.info "ATRUN: Scheduling job $JOBID at $NEXTTIME is in the past"
		execute $JOBID
		return
	fi

	# Remove existing jobs that were created by this script
	for i in `atq | cut -f1`; do
		if at -c $i | grep -q '# AUTO-CONTEST-SCHEDULE'; then
			atrm $i
		fi
	done

	cat - <<EOM | at -M "$ATTIME" 2> /dev/null
# AUTO-CONTEST-SCHEDULE
/opt/ioi/sbin/atrun.sh exec $JOBID
EOM
	#echo $date, $time, $cmd

	logger -p local0.info "ATRUN: Scheduling next job $JOBID at $NEXTTIME"
}

execute()
{
	JOBID=$1
	JOB=$(cat $SCHEDULE | head -$JOBID | tail -1)
	ENDTIME=$(echo "$JOB" | cut -d\  -f3,4)
	if ! echo "$ENDTIME" | grep -q '[0-9]\{4\}-[0-9]\{2\}-[0-9]\{2\}\s[0-9]\{2\}:[0-9]\{2\}'; then
		ENDTIME="9999-12-31 23:59"
	fi
	if [ "$TIMENOW" \> "$ENDTIME" ]; then
		NEWJOBID=$(echo "$JOB" | cut -d\  -f5)
		logger -p local0.info "ATRUN: Job $JOBID is over, jumping to $NEWJOBID"
		schedule $NEWJOBID
	else
		CMD=$(echo "$JOB" | cut -d\  -f6-)
		logger -p local0.info "ATRUN: Run job $JOBID now: $CMD"
		$CMD
		NEWJOBID=$((JOBID+1))
		schedule $NEWJOBID
	fi
}

case "$1" in
	schedule)
		logger -p local0.info "ATRUN: Restart scheduling"
		schedule 1
		;;
	next)
		;;
	exec)
		execute $2
		;;
	*)
esac
# vim: ft=sh ts=8 noet