#!/usr/bin/bash

# This script can be invoked as follows:

#

# glibc-bench-compare [options] <BUILD> [BUILD]

#

# Options may be one of the following:

#

# -t		The BUILD arguments are task ids and not a version-release string

# -a ARCH	Do comparison for ARCH architecture

#

# If any of the above options are given, both BUILD arguments must be given.

# Otherwise, if only one BUILD is specified, then it is compared against the

# installed glibc.

# Silence the pushd/popd messages

pushd() {

	command pushd "$@" > /dev/null 2>&1

}

popd() {

	command popd "$@" > /dev/null 2>&1

}

# Clean up any downloaded files before we exit

trap "rm -rf /tmp/glibc-bench-compare.$BASHPID.*" EXIT

task=0

arch=$(uname -i)

options=0

path=0

installed=

# Look for any commandline options

while getopts ":tpa:" opt; do

	case $opt in

		p)

		path=1

		;;

		t)

		task=1

		options=1

		echo "Not implemented."

		exit 1

		;;

		a)

		arch=$OPTARG

		options=1

		;;

		*)

		;;

	esac

done

# Done, now shift all option arguments out.

shift $((OPTIND-1))

if [ $# -gt 2 ] || [ $# -eq 0 ] || [ $# -lt 2 -a $options -eq 1 ]; then

	echo "Usage: $0 [OPTIONS] <old> [new]"

	echo

	echo "OPTIONS:"

	echo -e "\t-t\tCompare two brew tasks"

	echo -e "\t-a ARCH\tGet rpms for the ARCH architecture"

	echo -e "\t-p\tCompare built rpms in two paths."

	echo -e "\t\tThis minimally needs glibc, glibc-common and glibc-benchtests"

	exit 1

fi

if [ -z $2 ]; then

	new="$1"

	old=$(rpm --queryformat "%{VERSION}-%{RELEASE}\n" -q glibc | head -1)

	installed=$old

else

	new="$2"

	old="$1"

fi

decompress_rpms() {

	# We were given a path to the rpms.  Figure out the version-release and

	# decompress the rpms.

	if [ -n $1 ]; then

		vr=$(rpm --queryformat="%{VERSION}-%{RELEASE}" -qp $1/glibc-2*.rpm | head -1)

		mkdir $vr && pushd $vr

	fi

	for r in $1*.rpm; do

		( rpm2cpio $r | cpio -di ) > /dev/null

	done

	if [ -n $1 ]; then

		popd

		echo $vr

	fi

}

# Get rpms for a build and decompress them

get_build() {

	echo "Processing build $1"

	mkdir $1 && pushd $1

	brew buildinfo "glibc-$1" |

	sed -n -e "s|/mnt/koji\(.\+$arch.\+\)|http://kojipkgs.fedoraproject.org\1|p" |

	while read url; do

		echo "Downloading $url"

		wget -q $url

	done

	decompress_rpms

	echo "Removing rpms"

	rm -f $1/*.rpm

	popd

}

# Run benchmarks for a build

run_bench() {

	if [ -z $1 ]; then

		make DETAILED=1 ver=$installed prefix= -f /usr/libexec/glibc-benchtests/bench.mk bench

	else

		make DETAILED=1 ver=$1 prefix=$PWD -f $1/usr/libexec/glibc-benchtests/bench.mk bench

	fi

}

# Get absolute paths if needed, since we will change into the working directory

# next.

if [ $path -eq 1 ]; then

	old_path=$(realpath $old)/

	new_path=$(realpath $new)/

fi

tmpdir=$(mktemp -p /tmp -d glibc-bench-compare.$$.XXXX)

pushd $tmpdir

# Get both builds.

if [ $path -eq 0 ]; then

	if [ -z $installed ]; then

		get_build $old

	fi

	get_build $new

else

	old=$(decompress_rpms $old_path)

	new=$(decompress_rpms $new_path)

fi

# make bench for each of those.

if [ -z $installed ]; then

	run_bench $old

else

	run_bench

fi

run_bench $new

# Now run the comparison script.

$old/usr/libexec/glibc-benchtests/compare_bench.py $old/usr/libexec/glibc-benchtests/benchout.schema.json \

	bench.$old.out bench.$new.out