#!/bin/bash

set -e

usage() {
  echo "$0 [--no-pull] <--all | --only DEVICE_TYPE | --prebuilt-image DEVICE_TYPE IMAGE_NAME>"
  exit 1
}

root_dir=$( cd "$( dirname "${BASH_SOURCE[0]}" )/../../" && pwd )
if [ "${root_dir}" != "${PWD}" ]; then
  echo "You must execute $(basename $0) from the root directory: ${root_dir}"
  exit 1
fi

WORKSPACE=./tests

BBB_DEBIAN_IMAGE="bone-debian-9.5-iot-armhf-2018-08-30-4gb.img"
BBB_DEBIAN_IMAGE_URL="http://debian.beagleboard.org/images/${BBB_DEBIAN_IMAGE}.xz"

RASPBIAN_IMAGE="2019-09-26-raspbian-buster-lite"
RASPBIAN_IMAGE_URL="http://downloads.raspberrypi.org/raspbian_lite/images/raspbian_lite-2019-09-30/2019-09-26-raspbian-buster-lite.zip"

TINKER_IMAGE="20170417-tinker-board-linaro-stretch-alip-v1.8"
TINKER_IMAGE_URL="http://dlcdnet.asus.com/pub/ASUS/mb/Linux/Tinker_Board_2GB/${TINKER_IMAGE}.zip"

UBUNTU_IMAGE="Ubuntu-Bionic-x86-64.img"
UBUNTU_IMAGE_URL="https://d1b0l86ne08fsf.cloudfront.net/mender-convert/images/${UBUNTU_IMAGE}.gz"

UBUNTU_SERVER_RPI_IMAGE="ubuntu-18.04.4-preinstalled-server-armhf+raspi3.img"
UBUNTU_SERVER_RPI_IMAGE_URL="http://cdimage.ubuntu.com/ubuntu/releases/bionic/release/${UBUNTU_SERVER_RPI_IMAGE}.xz"

# Keep common function declarations in separate utils script
UTILS_PATH=${0/$(basename $0)/test-utils.sh}
source $UTILS_PATH

# Some distros do not have /sbin in path for "normal users"
export PATH="${PATH}:/sbin"

if [ "$1" = "--no-pull" ]; then
  shift
else
  echo "Automatically pulling submodules. Use --no-pull to disable"
  git submodule update --init --remote
fi

mkdir -p ${WORKSPACE}

get_pytest_files

if ! [ "$1" == "--all" -o "$1" == "--only" -a -n "$2" -o "$1" == "--prebuilt-image" -a -n "$3" ]; then
  usage
fi

test_result=0

if [ "$1" == "--prebuilt-image" ]; then
  run_tests "$2" "$3" || test_result=$?
  exit $test_result

else
  if [ "$1" == "--all" -o "$1" == "--only" -a "$2" == "qemux86_64" ]; then
    convert_and_test "qemux86_64" \
                     "release-1" \
                     "${UBUNTU_IMAGE_URL}" \
                     "${UBUNTU_IMAGE}" \
                     "${UBUNTU_IMAGE}.gz" \
                     "configs/qemux86-64_config" || test_result=$?
  fi

  if [ "$1" == "--all" -o "$1" == "--only" -a "$2" == "raspberrypi3" ]; then
    convert_and_test "raspberrypi3" \
                     "release-1" \
                     "${RASPBIAN_IMAGE_URL}" \
                     "${RASPBIAN_IMAGE}.img" \
                     "${RASPBIAN_IMAGE}.zip" \
                     "configs/raspberrypi3_config" || test_result=$?
  fi

  if [ "$1" == "--all" -o "$1" == "--only" -a "$2" == "linaro-alip" ]; then
    # MEN-2809: Disabled due broken download link
    #convert_and_test "linaro-alip" \
    #                 "release-1" \
    #                 "${TINKER_IMAGE_URL}" \
    #                 "${TINKER_IMAGE}.img" \
    #                 "${TINKER_IMAGE}.zip" || test_result=$?
    true
  fi

  if [ "$1" == "--all" -o "$1" == "--only" -a "$2" == "beaglebone" ]; then
    convert_and_test "beaglebone" \
                     "release-1" \
                     "${BBB_DEBIAN_IMAGE_URL}" \
                     "${BBB_DEBIAN_IMAGE}" \
                     "${BBB_DEBIAN_IMAGE}.xz" || test_result=$?
  fi

  if [ "$1" == "--all" -o "$1" == "--only" -a "$2" == "ubuntu" ]; then
    convert_and_test "raspberrypi3" \
                     "release-1" \
                     "${UBUNTU_SERVER_RPI_IMAGE_URL}" \
                     "${UBUNTU_SERVER_RPI_IMAGE}" \
                     "${UBUNTU_SERVER_RPI_IMAGE}.xz" \
                     "configs/raspberrypi3_config" || test_result=$?
  fi

  exit $test_result
fi