#!/usr/bin/env bash

. $(dirname $0)/util
set -eu -o pipefail

: ${TEST_INTEGRATION=}
: ${TEST_GATEWAY=}
: ${TEST_DOCKERFILE=}
: ${DOCKERFILE_RELEASES=}
: ${CONTINUOUS_INTEGRATION=}
: ${BUILDKIT_REGISTRY_MIRROR_DIR=}

progressFlag=""
if [ "$CONTINUOUS_INTEGRATION" == "true" ]; then progressFlag="--progress=plain"; fi

if [ "$#" == 0 ]; then TEST_INTEGRATION=1; fi

while test $# -gt 0
  do
    case "$1" in
        gateway)
          TEST_GATEWAY=1
        ;;
        dockerfile)
          TEST_DOCKERFILE=1
        ;;
        integration)
          TEST_INTEGRATION=1
        ;;
        *)
          echo "unknown arg $1"
        ;;
    esac
    shift
done

mirrorCreated=""
if [ -z $BUILDKIT_REGISTRY_MIRROR_DIR ]; then
  BUILDKIT_REGISTRY_MIRROR_DIR=$(mktemp -t buildkit.mirror.XXXXXXXXXX)
  mirrorCreated="1"
fi

iid="buildkit-tests"
iidfile=$(mktemp -t docker-iidfile.XXXXXXXXXX)
set -x

case $buildmode in
"buildkit")
  tmpfile=$(mktemp -t docker-iidfile.XXXXXXXXXX)
  buildctl build $progressFlag --frontend=dockerfile.v0 --local context=. --local dockerfile=. \
    --frontend-opt target=integration-tests \
    --frontend-opt filename=./hack/dockerfiles/test.buildkit.Dockerfile \
    --exporter=docker --exporter-opt name=$iid --exporter-opt output=$tmpfile
  docker load -i $tmpfile
  rm $tmpfile
  ;;
*)
  case $buildmode in
  "docker-buildkit")
    export DOCKER_BUILDKIT=1
    docker build --iidfile $iidfile -f ./hack/dockerfiles/test.buildkit.Dockerfile --target integration-tests --force-rm .
    ;;
  *)
    docker build --iidfile $iidfile -f ./hack/dockerfiles/test.Dockerfile --target integration-tests --force-rm .
    ;;
  esac
  iid=$(cat $iidfile)
  ;;
esac

if [ "$TEST_INTEGRATION" == 1 ]; then
  docker run --rm -v /tmp --privileged $iid go test ${TESTFLAGS:--v} ${TESTPKGS:-./...}
fi


if [ "$TEST_GATEWAY" == 1 ]; then
  docker run --rm $iid go build ./frontend/gateway/client
fi


if [ "$TEST_DOCKERFILE" == 1 ]; then
  if [ -z $DOCKERFILE_RELEASES ]; then
    DOCKERFILE_RELEASES="mainline experimental mounts secrets ssh"
  fi
  
  
  for release in $DOCKERFILE_RELEASES; do
    buildtags=$(cat ./frontend/dockerfile/release/$release/tags)
    tarout=$(mktemp -t dockerfile-frontend.XXXXXXXXXX)
    case $buildmode in
    "buildkit")
      buildctl build $progressFlag --frontend=dockerfile.v0 --local context=. --local dockerfile=. \
        --frontend-opt filename=./frontend/dockerfile/cmd/dockerfile-frontend/Dockerfile \
        --frontend-opt build-arg:BUILDTAGS="$buildtags" \
        --exporter=oci --exporter-opt output=$tarout
      ;;
    "docker-buildkit")
      dfiidfile=$(mktemp -t docker-iidfile.XXXXXXXXXX)
      docker build --iidfile=$dfiidfile -f ./frontend/dockerfile/cmd/dockerfile-frontend/Dockerfile --build-arg BUILDTAGS="$buildtags" .
      dfiid=$(cat $dfiidfile)
      docker save -o $tarout $dfiid
      docker rmi $dfiid
      rm $dfiidfile
      ;;
    esac

    if [ -s $tarout ]; then
      cid=$(docker create -v /tmp --rm --privileged -e BUILDKIT_WORKER_RANDOM=1 -e FRONTEND_GATEWAY_ONLY=local:/$release.tar -e EXTERNAL_DF_FRONTEND=/dockerfile-frontend $iid go test -tags "$buildtags" ${TESTFLAGS:--v} ./frontend/dockerfile)
      docker cp $tarout $cid:/$release.tar
      docker start -a $cid
    fi
    rm $tarout
  done
fi

case $buildmode in
"docker-buildkit")
  rm "$iidfile"
  docker rmi $iid
  ;;
"legacy")
  rm "$iidfile"
  ;;
esac

if [ ! -z $mirrorCreated ] && [ ! -z $BUILDKIT_REGISTRY_MIRROR_DIR ]; then
  rm -rf $BUILDKIT_REGISTRY_MIRROR_DIR
fi