#!/usr/bin/env bash

set -o errexit -o pipefail -o nounset

if ! [ -x "$(command -v parallel)" ]; then
	echo "parallel is required please install it"
	echo "Debian: sudo apt-get install parallel"
	echo "Osx:    brew install parallel"
	exit 1
fi

function usage() {
        echo "Usage: $0 route_request_file conf [concurrency] [outDir]"
        echo "Example: $0 ../test_requests/demo_routes.txt"
        echo "Example: $0 ../test_requests/demo_routes.txt ~/valhalla.json"
        echo "Example: $0 ../test_requests/demo_routes.txt ~/valhalla.json 8"
        echo "Example: $0 ../test_requests/demo_routes.txt ~/valhalla.json 8 my_special_dir"
        exit 1
}

#set the input file
if [ -z "${1:-}" ]; then
	usage
elif [ ! -f "${1:-}" ]; then
	usage
else
	readonly INPUT="${1}"
fi

#set config file
readonly CONF="${2:-}"

#how many threads do you want, default to max
readonly CONCURRENCY="${3:-$(nproc)}"

#where do you want the output, default to current time
readonly DEFAULT_OUTDIR=$(date +%Y%m%d_%H%M%S)_$(basename "${INPUT%.*}")
readonly RESULTS_OUTDIR="results/${4:-$DEFAULT_OUTDIR}"
mkdir --parent "${RESULTS_OUTDIR}"

#turn the nice input into something parallel can parse for args
TMP="$(mktemp)"
cp --recursive --preserve=mode,ownership,timestamps "${INPUT}" "${TMP}"
for arg in $(valhalla_run_route --help | grep -o '\-[a-z\-]\+' | sort | uniq); do
    sed -i -e "s/^${arg}[ ]\+/${arg}|/g" "${TMP}"
    sed -i -e "s/[ ]\+${arg}[ ]\+/|${arg}|/g" "${TMP}"
done
# Strip single-quotes from json string (arg to -j option in INPUT} to keep
# valhalla's json parser happy
sed -i -e "s/'//g" "${TMP}"

#run all of the paths, make sure to cut off the timestamps
#from the log messages otherwise every line will be a diff
#TODO: add leading zeros to output files so they sort nicely
echo -e "\x1b[32;1mWriting routes from ${INPUT} with a concurrency of ${CONCURRENCY} into ${RESULTS_OUTDIR}\x1b[0m"
cat "${TMP}" | parallel --progress -k -C '\|' -P "${CONCURRENCY}" "valhalla_run_route {} --config ${CONF} 2>&1 | tee -a ${RESULTS_OUTDIR}/{#}.tmp | grep -F NARRATIVE | sed -e 's/^[^\[]*\[NARRATIVE\] //' &> ${RESULTS_OUTDIR}/{#}.txt; grep -F STATISTICS ${RESULTS_OUTDIR}/{#}.tmp | sed -e 's/^[^\[]*\[STATISTICS\] //' &>> ${RESULTS_OUTDIR}/{#}_statistics.csv; rm -f ${RESULTS_OUTDIR}/{#}.tmp"
# On macos, sed inline replace (-i opt) ends up creating a backup file with
# "-e" suffix, even though -i arg is empty. The wildcard below will ensure that
# that file is deleted as well.
rm -f ${TMP}*
echo "orgLat, orgLng, destLat, destLng, result, #Passes, runtime, trip time, length, arcDistance, #Maneuvers, elapsedCostSeconds, elapsedCostCost" > ${RESULTS_OUTDIR}/statistics.csv
cat `ls -1v ${RESULTS_OUTDIR}/*_statistics.csv` >> ${RESULTS_OUTDIR}/statistics.csv
rm -f ${RESULTS_OUTDIR}/*_statistics.csv

echo ${RESULTS_OUTDIR} > outdir.txt