#!/bin/bash
# 
#   Copyright (C) 2013, 2014, 2015, 2016 Linaro, Inc
# 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
# 
set -e
set -o pipefail

# Improve debug logs
PRGNAME=$(basename $0)
PS4='+ $PRGNAME: ${FUNCNAME+"$FUNCNAME : "}$LINENO: '

usage()
{
    ret=1
    [ x"$1" != x ] && ret=$1
    cat << EOF
  jenkins.sh [--help] [-s snapshot dir] [g git reference dir] [--abedir path] [-w workspace]
EOF
    exit $ret
}

read_var()
{
   local artifact_list=$1
   local var=$2
   grep "^${var}=" "${artifact_list}" | cut -d = -f 2-
}

if test $# -lt 1; then
    echo "ERROR: No options for build!"
    usage
fi

# Directory of ABE source files
abe_dir=

# This is where all the builds go
user_workspace="${WORKSPACE}"

# The files in this directory are shared across all platforms 
shared="${HOME}/workspace/shared"

# This is an optional directory for the reference copy of the git repositories.
git_reference="${HOME}/snapshots-ref"

# Override default versions of components
change=""

# Server to store results on.
logserver=""

# Template of logs' directory name
logname=""

# Compiler languages to build
languages=default

# Whether to run tests
runtests=false

# Target to build for
target=""

# File with value for --testcontainer= option
testcontainer_file=""

# Whether attempt bootstrap
try_bootstrap=false

# The release version string, usually a date
releasestr=

# This is a string of optional extra arguments to pass to abe at runtime
user_options=""

# Return status
status=0

# Whether to exclude some component from 'make check'
excludecheck_opt=""

# Whether to rebuild the toolchain even if logs are already present.
# Note that the check is done on logserver/logname pair, so logname should not
# be relying on variables that this script sets for match to succeed.
# In practice, --norebuild option should be accompanied by something like
# --logname gcc-<sha1>
rebuild=true

orig_parameters="$@"

OPTS="$(getopt -o s:g:w:o:l:rt:b:h -l override:,gcc-branch:,snapshots:,gitrepo:,abedir:,workspace:,options:,logserver:,logname:,languages:,runtests,target:,testcontainerfile:,bootstrap,help,excludecheck:,norebuild,extraconfig: -- "$@")"
while test $# -gt 0; do
    case $1 in
	--gcc-branch) change="$change gcc=$2"; shift ;;
	--override) change="$change $2"; shift ;;
	--extraconfig) change="${change} --extraconfig $2"; shift ;;
        -s|--snapshots) user_snapshots=$2; shift ;;
        -g|--gitrepo) git_reference=$2; shift ;;
        --abedir) abe_dir=$2; shift ;;
	-t|--target) target=$2; shift ;;
	--testcontainerfile) testcontainer_file=$2; shift ;;
        -w|--workspace) user_workspace=$2; shift ;;
        -o|--options) user_options=$2; shift ;;
        --logserver) logserver=$2; shift ;;
        --logname) logname=$2; shift ;;
        -l|--languages) languages=$2; shift ;;
        -r|--runtests) runtests="true" ;;
        -b|--bootstrap) try_bootstrap="true" ;;
	--excludecheck) excludecheck_opt="$excludecheck_opt --excludecheck $2"; shift ;;
	--norebuild) rebuild=false ;;
	-h|--help) usage 0 ;;
        *) usage ;;
    esac
    shift
done

if test x"${abe_dir}" = x; then
    echo "Error: --abedir missing"
    usage
fi

if test x"${user_workspace}" = x; then
    echo "Error: user_workspace is not defined. Make sure WORKSPACE is defined or use --workspace option"
    usage
fi

# set default values for options to make life easier
user_snapshots="${user_workspace}/snapshots"

# Non matrix builds use node_selector, but matrix builds use NODE_NAME
if test x"${node_selector}" != x; then
    node="$(echo ${node_selector} | tr '-' '_')"
    job=${JOB_NAME}
else
    node="$(echo ${NODE_NAME} | tr '-' '_')"
    job="$(echo ${JOB_NAME}  | cut -d '/' -f 1)"
fi

arch="$(uname -m)"

if [ x"$logserver" = x"" -a x"$logname" != x"" ]; then
    echo "ERROR: \$logname is not provided, but \$logserver is set to $logserver"
    exit 1
fi
# Now that all variables from $logname template are known, calculate log dir.
eval dir="$logname"

# Split $logserver into "server:path".
basedir="${logserver#*:}"
logserver="${logserver%:*}"

# Check status of logs on $logserver and rebuild if appropriate.
[ x"$logserver" != x"" ] && ssh $logserver mkdir -p $(dirname $basedir/$dir)
# Loop and wait until we successfully grabbed the lock.  The while condition is,
# effectively, "while true;" with a provision to skip if $logserver is not set.
while [ x"$logserver" != x"" ]; do
    # Non-blocking read lock, and check whether logs already exist.
    log_status=$(ssh $logserver flock -ns $basedir/$dir.lock -c \
	"\"if [ -e $basedir/$dir ]; then exit 0; else exit 2; fi\""; echo $?)

    case $log_status in
	0)
	    echo "Logs are already present in $logserver:$basedir/$dir"
	    if ! $rebuild; then
		exit 0
	    fi
	    echo "But we are asked to rebuild them anyway"
	    ;;
	1)
	    echo "Can't obtain read lock; waiting for another build to finish"
	    sleep 60
	    continue
	    ;;
	2)
	    echo "Logs don't exist in $basedir/$dir, trying to rebuild"
	    ;;
	*)
	    echo "ERROR: Unexpected status of logs: $log_status"
	    exit 1
	    ;;
    esac

    # Acquire the lock for the duration of the build.  The lock is released
    # in the "trap" cleanup below on signal or normal exit.
    # Note that the ssh command will be running in the background for the
    # duration of the build (spot "&" at its end).  Ssh command will exit
    # when lock file is deleted by the "trap" cleanup.
    # We place a unique marker into the lock file to check on our side who
    # has the lock, since we can't inspect return value of the ssh command.
    #
    # Note on '-tt': We forcefully allocate pseudo-tty for the flock command
    # so that flock dies (through SIGHUP) and releases the lock when this
    # script is cancelled or terminated for whatever reason.  Without SIGHUP
    # reaching flock we risk situations when a lock will hang forever preventing
    # any subsequent builds to progress.  There are a couple of options as to
    # exactly how enable delivery of SIGHUP (e.g., set +m), and 'ssh -tt' seems
    # like the simplest one.
    ssh -tt $logserver flock -nx $basedir/$dir.lock -c \
	"\"echo $(hostname)-$$-$BUILD_URL > $basedir/$dir.lock; while [ -e $basedir/$dir.lock ]; do sleep 10; done\"" &
    pid=$!
    # This is borderline fragile, since we are giving the above ssh command
    # a fixed period of time (10sec) to connect to $logserver and populate
    # $basedir/$dir.lock.  In practice, $logserver is a fast-ish machine,
    # which serves connections quickly.  In the worst-case scenario, we will
    # just retry a couple of times in this loop.
    sleep 10

    if [ x"$(ssh $logserver cat $basedir/$dir.lock)" \
	= x"$(hostname)-$$-$BUILD_URL" ]; then
	trap "ssh $logserver rm -f $basedir/$dir.lock" 0 1 2 3 5 9 13 15
	# Hurray!  Break from the loop and go ahead with the build!
	break
    fi

    kill $pid || true
done

# Test the config parameters from the Jenkins Build Now page

# See if we're supposed to build a source tarball
if test x"${tarsrc}" = xtrue -o "$(echo $user_options | grep -c -- --tarsrc)" -gt 0; then
    tars="--tarsrc"
fi

# See if we're supposed to build a binary tarball
if test x"${tarbin}" = xtrue -o "$(echo $user_options | grep -c -- --tarbin)" -gt 0; then
    tars="${tars} --tarbin "
fi

# Set the release string if specefied
if ! test x"${release}" = xsnapshot -o x"${release}"; then
    releasestr="--release ${release}"
fi
if test "$(echo $user_options | grep -c -- --release)" -gt 0; then
    release="$(echo  $user_options | grep -o -- "--release [a-zA-Z0-9]* " | cut -d ' ' -f 2)"
    releasestr="--release ${release}"
fi

# Get the versions of dependant components to use
if test x"${gmp_snapshot}" != x"latest" -a x"${gmp_snapshot}" != x; then
    change="${change} gmp=${gmp_snapshot}"
fi
if test x"${mpc_snapshot}" != x"latest" -a x"${mpc_snapshot}" != x; then
    change="${change} mpc=${mpc_snapshot}"
fi
if test x"${mpfr_snapshot}" != x"latest" -a x"${mpfr_snapshot}" != x; then
    change="${change} mpfr=${mpfr_snapshot}"
fi

if test x"${binutils_snapshot}" != x"latest" -a x"${binutils_snapshot}" != x; then
    change="${change} binutils=${binutils_snapshot}"
fi
if test x"${linux_snapshot}" != x"latest" -a x"${linux_snapshot}" != x; then
    change="${change} linux-${linux_snapshot}"
fi

if test x"${target}" != x"native" -a x"${target}" != x; then
    platform="--target ${target}"
fi

if test x"${libc}" != x; then
    # ELF based targets are bare metal only
    case ${target} in
	arm*-none-*)
	    change="${change} --set libc=newlib"
	    ;;
	*)
	    change="${change} --set libc=${libc}"
	    ;;
    esac
fi

# Create a build directory
if test -d ${user_workspace}/_build; then
    rm -fr ${user_workspace}/_build
fi
mkdir -p ${user_workspace}/_build

# Use the newly created build directory
pushd ${user_workspace}/_build

# Configure Abe itself. Force the use of bash instead of the Ubuntu
# default of dash as some configure scripts go into an infinite loop with
# dash. Not good...
export CONFIG_SHELL="/bin/bash"
if test x"${debug}" = x"true"; then
    export CONFIG_SHELL="/bin/bash -x"
fi

# Download QEMU provided by Peter Maydell.
# The tarball has README with version information.
if [ x"$(uname -m)" = x"x86_64" ]; then
    wget --progress=dot:giga http://people.linaro.org/~maxim.kuvyrkov/qemu-20160707.tgz
    tar xf qemu-20160707.tgz
    export PATH="$(pwd)/qemu-wip:$PATH"
    for i in aarch64 arm armeb; do
	qemu-$i --version
    done
fi

# Print some information about the build machine
echo Running on $(hostname)
uname -a
lsb_release -a
lscpu
cat /proc/meminfo

$CONFIG_SHELL ${abe_dir}/configure --with-local-snapshots=${user_snapshots} --with-git-reference-dir=${git_reference} --with-languages=${languages} --enable-schroot-test

# load commonly used varibles set by configure
if test -e "${PWD}/host.conf"; then
    . "${PWD}/host.conf"
fi

# Delete the previous test result files to avoid problems.
find ${user_workspace} -name \*.sum -exec rm {} \;  2>&1 > /dev/null

if test x"${try_bootstrap}" = xtrue; then
    # Attempt to bootstrap GCC is build and target are compatible
    build1="$(grep "^build=" host.conf | sed -e "s/build=\(.*\)-\(.*\)-\(.*\)-\(.*\)/\1-\3-\4/")"
    target1="$(echo ${target} | sed -e "s/\(.*\)-\(.*\)-\(.*\)-\(.*\)/\1-\3-\4/")"
    if test x"${build1}" = x"${target1}" -o x"${platform}" = x""; then
	try_bootstrap="--enable bootstrap"
    else
	try_bootstrap="--disable bootstrap"
    fi
else
    try_bootstrap=""
fi

# Checkout all sources now to avoid grabbing lock for 1-2h while building and
# testing runs.  We configure ABE to use reference snapshots, which are shared
# across all builds and are updated by an external process.  The lock protects
# us from looking into an inconsistent state of reference snapshots.
(
    flock -s 9
    $CONFIG_SHELL ${abe_dir}/abe.sh ${platform} ${change} --checkout all
) 9>${git_reference}.lock

# Now we build the cross compiler, for a native compiler this becomes
# the stage2 bootstrap build.
ret=0
$CONFIG_SHELL ${abe_dir}/abe.sh --list-artifacts build-artifacts.txt --disable update ${tars} ${releasestr} ${platform} ${change} ${try_bootstrap} --timeout 100 --build all --disable make_docs > build.out 2> >(tee build.err >&2) || ret=$?

# If abe returned an error, make jenkins see this as a build failure
if test $ret -gt 0; then
    echo "================= TAIL OF LOG: BEGIN ================="
    tail -n 50 build.out
    echo "================= TAIL OF LOG: FINISH ================="
    exit 1
fi

# if runtests is true, then run make check after the build completes
if $runtests; then
    # check that expect is working, and dump some debug info then exit if not
    if ! echo "spawn true" | /usr/bin/expect -f - >/dev/null; then
        echo "expect cannot spawn processes. Aborting make check."
        echo "some debug info follows..."
        echo "running: ls -l /dev/ptmx"
        ls -l /dev/ptmx
        echo "running: ls -l /dev/pts"
        ls -l /dev/pts
        echo "running: grep devpts /proc/mounts"
        grep devpts /proc/mounts
        exit 1
    fi

    check="--check all"
    check="${check}${excludecheck_opt}"

    if [ x"$testcontainer_file" != x"" ]; then
        if [ ! -f "$testcontainer_file" ]; then
            echo "ERROR: Cannot start test container"
            exit 1
        fi

        check="$check --testcontainer $(cat "$testcontainer_file")"
    fi


    ret=0
    $CONFIG_SHELL ${abe_dir}/abe.sh --list-artifacts check-artifacts.txt --disable update ${check} ${tars} ${releasestr} ${platform} ${change} ${try_bootstrap} --timeout 100 --build all --disable make_docs > check.out 2> >(tee check.err >&2) || ret=$?

    # If abe returned an error, make jenkins see this as a build failure
    if test $ret -gt 0; then
	echo "================= TAIL OF LOG: BEGIN ================="
	tail -n 50 check.out
	echo "================= TAIL OF LOG: FINISH ================="
	exit 1
    fi
fi

# Create the BUILD-INFO file for Jenkins.
cat << EOF > ${user_workspace}/BUILD-INFO.txt
Format-Version: 0.5

Files-Pattern: *
License-Type: open
EOF

if test x"${tars}" = x; then
    # date="$(${gcc} --version | head -1 | cut -d ' ' -f 4 | tr -d ')')"
    date="$(date +%Y%m%d)"
else
    date=${release}
fi

# This is the remote directory for tcwgweb where all test results and log
# files get copied too.

# These fields are enabled by the buikd-user-vars plugin.
if test x"${BUILD_USER_FIRST_NAME}" != x; then
    requestor="-${BUILD_USER_FIRST_NAME}"
fi
if test x"${BUILD_USER_LAST_NAME}" != x; then
    requestor="${requestor}.${BUILD_USER_LAST_NAME}"
fi

echo "Build by ${requestor} on ${NODE_NAME} with parameters: $orig_parameters"

manifest="$(read_var build-artifacts.txt manifest)"
if test x"${manifest}" != x; then
    echo "node=${node}" >> ${manifest}
    echo "requestor=${requestor}" >> ${manifest}
    if test x"${BUILD_USER_ID}" != x; then
	echo "email=${BUILD_USER_ID}" >> ${manifest}
    fi
    echo "build_url=${BUILD_URL}" >> ${manifest}
else
    echo "ERROR: No manifest file, build probably failed!"
fi

# This becomes the path on the remote file server    
if test x"${logserver}" != x""; then
    # Re-eval $dir as we now have full range of variables available.
    eval dir="$logname"
    ssh ${logserver} mkdir -p ${basedir}/${dir}
    if test x"${manifest}" != x; then
	scp ${manifest} ${logserver}:${basedir}/${dir}/
    fi

# If 'make check' works, we get .sum files with the results. These we
# convert to JUNIT format, which is what Jenkins wants it's results
# in. We then cat them to the console, as that seems to be the only
# way to get the results into Jenkins.
#if test x"${sums}" != x; then
#    for i in ${sums}; do
#	name="$(basename $i)"
#	${abe_dir}/sum2junit.sh $i $user_workspace/${name}.junit
#	cp $i ${user_workspace}/results/${dir}
#    done
#    junits="$(find ${user_workspace} -name *.junit)"
#    if test x"${junits}" = x; then
#	echo "Bummer, no junit files yet..."
#    fi
#else
#    echo "Bummer, no test results yet..."
#fi
#touch $user_workspace/*.junit
fi

# Find all the test result files.
sums="$(find ${user_workspace} -name \*.sum -not -path "*/gdb/testsuite/outputs/*")"

# Canadian Crosses are a win32 hosted cross toolchain built on a Linux
# machine.
if test x"${canadian}" = x"true"; then
    $CONFIG_SHELL ${abe_dir}/abe.sh --list-artifacts canadian-build1.txt --disable update ${change} ${platform} --build all
    distro="$(lsb_release -sc)"
    # Ubuntu Lucid uses an older version of Mingw32
    if test x"${distro}" = x"lucid"; then
	$CONFIG_SHELL ${abe_dir}/abe.sh --list-artifacts canadian-build2.txt --disable update ${change} ${tars} --host=i586-mingw32msvc ${platform} --build all
    else
	$CONFIG_SHELL ${abe_dir}/abe.sh --list-artifacts canadian-build2.txt --disable update ${change} ${tars} --host=i686-w64-mingw32 ${platform} --build all
    fi
fi

# This setups all the files needed by tcwgweb
if test x"${logserver}" != x"" && test x"${sums}" != x -o x"${runtests}" != x"true"; then
    logs_dir=$(mktemp -d)

    if test x"${sums}" != x; then
	test_logs=""
	for s in ${sums}; do
	    test_logs="$test_logs ${s%.sum}.log"
	done

	cp ${sums} ${test_logs} ${logs_dir}/ || status=1
	
	# Copy over the logs from make check, which we need to find testcase errors.
        for check in $(read_var check-artifacts.txt 'log_check_[^=]*'); do
	    cp ${check} ${logs_dir}/ || status=1
        done
    fi

    # Copy over the build logs
    for makelog in $(read_var check-artifacts.txt 'log_make_[^=]*'); do
        cp ${makelog} ${logs_dir}/ || status=1
    done

    # Copy stdout and stderr output from abe.
    cp build.out build.err ${logs_dir}/ || status=1
    if $runtests; then
	cp check.out check.err ${logs_dir}/ || status=1
    fi

    xz ${logs_dir}/* || status=1
    scp ${logs_dir}/* ${logserver}:${basedir}/${dir}/ || status=1
    rm -rf ${logs_dir} || status=1

    echo "Uploaded test results and build logs to ${logserver}:${basedir}/${dir}/ with status: $status"

    if test x"${tarsrc}" = xtrue -a x"${release}" != x; then
	allfiles="$(ls ${user_snapshots}/*${release}*.xz)"
	srcfiles="$(echo ${allfiles} | egrep -v "arm|aarch")"
	scp ${srcfiles} ${logserver}:/home/abe/var/snapshots/ || status=1
	rm -f ${srcfiles} || status=1
    fi

    if test x"${tarbin}" = xtrue -a x"${release}" != x; then
	allfiles="$(ls ${user_snapshots}/*${release}*.xz)"
	binfiles="$(echo ${allfiles} | egrep "arm|aarch")"
	scp ${binfiles} ${logserver}:/work/space/binaries/ || status=1
	rm -f ${binfiles} || status=1
    fi

fi

exit $status