#!/bin/bash
# This file is part of curtin. See LICENSE file for copyright and license info.

topdir="${CURTIN_VMTEST_TOPDIR:-${WORKSPACE:-$PWD}/output}"
pkeep=${CURTIN_VMTEST_KEEP_DATA_PASS:-logs,collect}
fkeep=${CURTIN_VMTEST_KEEP_DATA_FAIL:-logs,collect}
reuse=${CURTIN_VMTEST_REUSE_TOPDIR:-0}
export CURTIN_VMTEST_TAR_DISKS=${CURTIN_VMTEST_TAR_DISKS:-0}
export CURTIN_VMTEST_REUSE_TOPDIR=$reuse
export CURTIN_VMTEST_IMAGE_SYNC=${CURTIN_VMTEST_IMAGE_SYNC:-0}
export CURTIN_VMTEST_KEEP_DATA_PASS=$pkeep
export CURTIN_VMTEST_KEEP_DATA_FAIL=$fkeep
export CURTIN_VMTEST_TOPDIR="$topdir"
export CURTIN_VMTEST_LOG="${CURTIN_VMTEST_LOG:-$topdir/debug.log}"
export CURTIN_VMTEST_PARALLEL=${CURTIN_VMTEST_PARALLEL:-0}
export IMAGE_DIR=${IMAGE_DIR:-/srv/images}

# empty TGT_* variables in current env to avoid killing a pid we didn't start.
TGT_PID=""
TGT_LOG_D=""

error() { echo "$@" 1>&2; }
fail() { [ $# -eq 0 ] || error "$@"; exit 1; }
cleanup() {
    local ret=$?
    local keep_rules
    [ "$ret" -eq 0 ] && keep_rules="$pkeep" || keep_rules="$fkeep"
    # kill a tgtd pid that was started here.
    if [ -n "$TGT_PID" ]; then
        kill -9 ${TGT_PID};
        if [ -n "${TGT_IPC_SOCKET}" ]; then
            # var is <path>/socket but the actual socket is <path>/socket.0
            rm -f "${TGT_IPC_SOCKET}" "${TGT_IPC_SOCKET}".* ||
                error "WARN: failed removal of $TGT_IPC_SOCKET"
        fi
    fi
    if [ -n "$TGT_LOG_D" ]; then
        case ",${keep_rules}," in
            *,logs,*|*,all,*) :;;
            *) rm -Rf "${TGT_LOG_D}";;
        esac
    fi
}

if [ "$reuse" != "1" ]; then
    if [ -d "$topdir" ]; then
        fail "topdir '$topdir' existed."
    fi
    mkdir -p "$topdir" || fail "failed mkdir $topdir"
fi

start_s=$(date +%s)
parallel=${CURTIN_VMTEST_PARALLEL}
ntargs=( )
while [ $# -ne 0 ]; do
    case "$1" in
        -p|--parallel) parallel="$2"; shift;;
        --parallel=*) parallel=${1#*=};;
        -p[0-9]|-p-1|-p[0-9][0-9]) parallel=${1#-p};;
        --)
            shift
            break
            ;;
        *) ntargs[${#ntargs[@]}]="$1";;
    esac
    shift;
done

CURTIN_VMTEST_PARALLEL=$parallel

if [ ${#ntargs[@]} -eq 0 ]; then
   set -- -vv --nologcapture tests/vmtests/
fi

trap cleanup EXIT

ntargs=( "${ntargs[@]}" "$@" )

pargs=( )
if [ -n "$parallel" -a "$parallel" != "0" -a "$parallel" != "1" ]; then
    pargs=( --process-timeout=86400 "--processes=$parallel" )
fi

if [ -n "$TGT_IPC_SOCKET" ]; then
    error "existing TGT_IPC_SOCKET=${TGT_IPC_SOCKET}"
elif command -v tgtd >/dev/null 2>&1; then
    tgtdir="$topdir/tgt.d"
    mkdir -p "$tgtdir" || fail "failed to create $tgtdir"
    rm -f "$tgtdir/info" || fail "failed to remove $tgtdir/info"
    ./tools/find-tgt "$tgtdir" >"${tgtdir}/find-tgt.log" 2>&1 || {
        cat "${tgtdir}/find-tgt.log" 1>&2
        fail "could not start a tgt service"
    }
    TGT_LOG_D="$tgtdir"
    . "$tgtdir/info" >"$tgtdir/source-output.txt" 2>&1
    [ -n "$TGT_PID" ] || fail "find-tgt did not write TGT_PID"
    [ -d "/proc/${TGT_PID}" ] || fail "no tgtd process in /proc/${TGT_PID}"
else
    error "no tgtd command, iscsi tests will be skipped"
fi

# dump CURTIN_* and TGT_* and proxy variables just for info
for v in ${!CURTIN_*} ${!TGT_*} http_proxy https_proxy no_proxy; do
   echo "$v=${!v}"
done

# avoid LOG info by running python3 tests/vmtests/image_sync.py
# rather than python3 -m tests.vmtests.image_sync (LP: #1594465)
echo "Quering synced ephemeral images/kernels in $IMAGE_DIR"
printf "%.s=" {1..86}; echo
printf " %6s %8s %-10s %s/%-14s  %s\n" "Release" "Codename" "ImageDate" "Arch" "SubArch" "Path"
printf "%.s-" {1..86}; echo
fmt="  %(version)6s %(release)8s %(version_name)-10s %(arch)s/%(subarch)-14s %(path)s"
PYTHONPATH="$PWD" python3 tests/vmtests/image_sync.py query \
    --output-format="$fmt" "$IMAGE_DIR" "kflavor=generic" "ftype~(root-image.gz|squashfs)" ||
    { ret=$?; echo "FATAL: error querying images in $IMAGE_DIR" 1>&2;
      exit $ret; }
printf "%.s=" {1..86}; printf "\n\n"

echo "$(date -R): vmtest start: nosetests3 ${pargs[*]} ${ntargs[*]}"
nosetests3 "${pargs[@]}" "${ntargs[@]}"
ret=$?
end_s=$(date +%s)
echo "$(date -R): vmtest end [$ret] in $(($end_s-$start_s))s"
exit $ret

# vi: ts=4 expandtab syntax=sh
