#!/bin/bash

set -euf -o pipefail

scripts=$(dirname $0)
# shellcheck source=jenkins-helpers.sh
. $scripts/jenkins-helpers.sh
# shellcheck source=round-robin.sh
. $scripts/round-robin.sh

convert_args_to_variables "$@"

obligatory_variables rr[ci_config]
declare -A rr

# Execution mode: baseline, bisect, jenkins-full
rr[mode]="${rr[mode]-baseline}"

# Set custom revision for one of the projects, and use baseline revisions
# for all other projects.
rr[ci_project]="${rr[ci_project]-tcwg_gnu}"
rr[baseline_branch]="${rr[baseline_branch]-linaro-local/ci/${rr[ci_project]}/${rr[ci_config]}}"
rr[update_baseline]="${rr[update_baseline]-update}"
rr[top_artifacts]="${rr[top_artifacts]-$(pwd)/artifacts}"

# Resolve top_artifacts to absolute dir because some of the subsequent
# processes work below pwd and would write their artifacts in a wrong
# location
rr[top_artifacts]=$(abs_path "${rr[top_artifacts]}")

# {toolchain_name}-{toolchain_ver}-{target}-{type_of_test}
IFS=- read -a ci_config <<EOF
${rr[ci_config]}
EOF
# type_of_test contains the type of action to perform in this test
# campaign: bootstrap, bootstrap_lto, check_binutils, ....
type_of_test=${type_of_test-${ci_config[3]}}

case "$type_of_test" in
    *_binutils)
	rr[target]="${rr[target]-native}"
	rr[components]="binutils"
	;;
    *bootstrap*)
	rr[target]="${rr[target]-native}"
	rr[components]="gcc"
	;;
    *_cross)
	rr[target]="${rr[target]-${ci_config[2]}}"
	rr[components]="binutils gcc glibc qemu"
	;;
    *) assert_with_msg "Unknown type_of_test: $type_of_test" false ;;
esac

# Use baseline branches by default.
for c in ${rr[components]}; do
    rr[${c}_branch]=${rr[${c}_branch]-baseline}
    if [ x"${rr[${c}_branch]}" != x"baseline" ]; then
	obligatory_variables rr[${c}_url]
    fi
done

start_at="${start_at-default}"
finish_at="${finish_at-default}"
verbose="${verbose-true}"
verbose2="${verbose2-false}"

if $verbose2; then set -x; fi

trap print_traceback EXIT

# Set start and finish steps for different modes.
default_start_at=""
default_finish_at=""
case "${rr[mode]}" in
    "baseline")
	default_finish_at="update_baseline"
	;;
    "bisect")
	case "$(print_single_updated_component):$type_of_test" in
	    binutils:*) default_start_at="build_abe-binutils" ;;
	    gcc:*_cross) default_start_at="build_abe-stage1" ;;
	    gcc:*) default_start_at="build_abe-${type_of_test#check_}" ;;
	    glibc:*) default_start_at="build_abe-glibc" ;;
	    qemu:*) default_start_at="build_abe-qemu" ;;
	    *) assert_with_msg "Trying to bisecting unknown component: $(print_single_updated_component)" false ;;
	esac
	default_finish_at="check_regression"
	;;
    "jenkins-full") ;;
esac
if [ x"$start_at" = x"default" ]; then
    start_at="$default_start_at"
fi
if [ x"$finish_at" = x"default" ]; then
    finish_at="$default_finish_at"
fi

run_step_init "$start_at" "$finish_at" "${rr[top_artifacts]}" "$verbose"

# Exit with code 0 if no regression compared to base-artifacts/results.
no_regression_p ()
{
    (
    set -euf -o pipefail

    no_build_regression_p "$@"

    local ref_artifacts=$1
    local new_artifacts=$2

    local sumfiles_base=$ref_artifacts/sumfiles
    local sumfiles_new=$new_artifacts/sumfiles

    if ! [ -d $sumfiles_base ]; then
	return 0
    elif ! [ -d $sumfiles_new ]; then
	return 1
    fi

    local res

    # We use our modified version of GCC's comparison script
    clone_or_update_repo gcc-compare-results master https://git.linaro.org/toolchain/gcc-compare-results.git

    # (defined by init_step in jenkins-helpers)
    # shellcheck disable=SC2154
    gcc-compare-results/compare_tests -compr none -pass-thresh 0.9 \
				      $sumfiles_base $sumfiles_new \
	| sed -e "s/^/# /" | tee $run_step_artifacts/results.regressions &
    res=0 && wait $! || res=$?

    # Handle return codes in the same way as in compare_jobs.sh from
    # gcc-compare-results
    case $res in
	# Regression, No common logs, extra logs or build failed
	2|3|4|5)
	    res=1
	    ;;
	# No change or improvement
	0|1)
	    res=0
	    ;;
	*)
	    assert_with_msg "Unknown results comparison status: $res" false
	    res=1
	    ;;
    esac

    if [ $res -ne 0 ]; then
        return 1
    fi

    return 0
    )
}

run_step stop_on_fail -10 reset_artifacts
run_step stop_on_fail x prepare_abe
case "$type_of_test" in
    build_cross)
	run_step skip_on_fail 0 true
	run_step skip_on_fail 1 build_abe binutils
	run_step skip_on_fail 2 build_abe stage1
	run_step skip_on_fail 3 build_abe linux
	run_step skip_on_fail 4 build_abe glibc
	run_step skip_on_fail 5 build_abe stage2
	run_step skip_on_fail 6 build_abe qemu
	;;
    check_cross)
	run_step skip_on_fail -8 build_abe binutils
	run_step skip_on_fail -7 build_abe stage1
	run_step skip_on_fail -6 build_abe linux
	run_step skip_on_fail -5 build_abe glibc
	run_step skip_on_fail -4 build_abe stage2
	run_step skip_on_fail -3 build_abe qemu
	run_step skip_on_fail 0 build_abe dejagnu
	run_step skip_on_fail 1 build_abe check_gcc
	;;
    check_*)
	run_step skip_on_fail 0 build_abe ${type_of_test#check_}
	run_step skip_on_fail 1 build_abe ${type_of_test}
	;;
    *)
	run_step skip_on_fail 0 true
	run_step skip_on_fail 1 build_abe ${type_of_test}
	;;
esac
run_step reset_on_fail x check_regression
run_step stop_on_fail x update_baseline
run_step stop_on_fail x push_baseline

trap "" EXIT