#!/usr/bin/env bash

# Copyright (c) 2021 Tigera, Inc. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

set -ex

FV_DIR="/home/semaphore/calico/process/testing/winfv-felix"
EXIT_CODE=0

pushd ${FV_DIR}
# Prepare local files
cp ~/.docker/config.json docker_auth.json

if [[ $FV_PROVISIONER == "aws" ]]; then
  cp ~/$SEMAPHORE_GIT_DIR/felix/bin/calico-felix.exe .
  cp ~/$SEMAPHORE_GIT_DIR/felix/fv/win-fv.exe .
  # Prepare key for windows fv.
  ssh-keygen -f master_ssh_key -N ''
  puttygen ${MASTER_CONNECT_KEY} -O private -o ${WIN_PPK_KEY}
  chmod 600 $WIN_PPK_KEY
  aws ec2 import-key-pair --key-name ${KEYPAIR_NAME} --public-key-material file://${MASTER_CONNECT_KEY_PUB}

  # Set up the cluster.
  NAME_PREFIX="$CLUSTER_NAME" KUBE_VERSION="${K8S_VERSION}" WINDOWS_KEYPAIR_NAME="$KEYPAIR_NAME" WINDOWS_PEM_FILE="$MASTER_CONNECT_KEY" WINDOWS_PPK_FILE="$WIN_PPK_KEY" WINDOWS_OS="Windows${WINDOWS_VERSION}container" BACKEND=${BACKEND} /bin/bash -x ./setup-fv.sh -q | tee fv.log

  # Run FV
  MASTER_IP=$(grep ubuntu@ fv.log | cut -d '@' -f2)
  SSH_CMD=$(echo ssh -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -i ${MASTER_CONNECT_KEY} ubuntu@${MASTER_IP})

  ${SSH_CMD} ls -ltr /home/ubuntu
  ${SSH_CMD} ls -ltr /home/ubuntu/winfv
  ${SSH_CMD} touch /home/ubuntu/winfv/file-ready
  ${SSH_CMD} time /home/ubuntu/winfv/wait-report.sh
  ${SSH_CMD} ls -ltr /home/ubuntu/report
  popd

  # Get results and logs
  SCP_CMD=$(echo scp -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -i ${FV_DIR}/${MASTER_CONNECT_KEY})
  ${SCP_CMD} -r ubuntu@${MASTER_IP}:/home/ubuntu/report /home/semaphore
elif [[ $FV_PROVISIONER == "capz" ]]; then
  export KUBE_VERSION="$K8S_VERSION"
  /bin/bash -x ./setup-fv-capz.sh -q | tee setup-fv.log; pstat=${PIPESTATUS[0]}
  if [[ $pstat != 0 ]]; then
    EXIT_CODE=$pstat
  fi
  mv ./report /home/semaphore
  mv ./setup-fv.log /home/semaphore/report
  popd
fi
ls -ltr /home/semaphore/report

# Print relevant snippets from logs
log_regexps='(?<!Decode)Failure|SUCCESS|FV-TEST-START'
compgen -G /home/semaphore/report/*.log > /dev/null && \
for log_file in /home/semaphore/report/*.log; do
    prefix="[$(basename ${log_file})]"
    cat ${log_file} | iconv -f UTF-16 -t UTF-8 | sed 's/\r$//g' | grep --line-buffered --perl ${log_regexps} -B 2 -A 15 | sed 's/.*/'"${prefix}"' &/g'
done;

# Stop for debug
echo "Check for pause file..."
while [ -f /home/semaphore/pause-for-debug ];
do
    echo "#"
    sleep 30
done

# Search for error code file
if [[ -f /home/semaphore/report/error-codes || $EXIT_CODE != 0 ]];
then
    echo "Windows FV returned error(s)."
    exit 1
fi

echo "Run Windows FV is done."
