#!/usr/bin/env bash
# Copyright (c) 2025 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.

# This script is run locally to push the FV test artifacts to our sub-directory
# in GCS. The counterpart to this script is load-test-artifacts, which runs on
# the test VM.  It can assume that the artifacts have been downloaded to /tmp.
#
# fv-artifacts are published to GCS (GCS_WORKFLOW_DIR) rather than the build
# cache so the in-cloud GCP test VMs fetch them without cross-cloud egress. The
# source images live in whichever store the CI system provides, published there
# by the blocks that build them, so this script bridges the two backends.

set -e
set -x

# Resolve the s3-cmd wrapper before we cd out of the component directory.
S3_CMD="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)/.semaphore/s3-cmd"

cd ../..

fetch_image() {
  local img=$1
  if [ -n "${CI_ARTIFACT_STORAGE:-}" ]; then
    ( cd /tmp && artifact pull workflow "${img}" ) 2>/dev/null
    return
  fi
  "$S3_CMD" cp "${S3_WORKFLOW_DIR}/${img}" "/tmp/${img}" 2>/dev/null
}

# Fetch the pre-built images if they are available, and re-upload them into the
# GCS fv-artifacts directory for the test VM. These were built by the "Build: *"
# blocks in the Prerequisites stage.
for img in calico-image.tar.zst node-image.tar.zst whisker-image.tar.zst; do
  if fetch_image "${img}"; then
    echo "Found pre-built ${img}, including in artifacts."
    gcloud storage cp "/tmp/${img}" "${GCS_WORKFLOW_DIR}/node/fv-artifacts/${img}"
  else
    echo "WARNING: ${img} is not in the workflow cache; the test VMs will build it from source."
  fi
done

tar -czf /tmp/working-copy.tgz \
  --exclude=${CALICO_DIR_NAME}/artifacts \
  ${CALICO_DIR_NAME}
gcloud storage cp /tmp/working-copy.tgz "${GCS_WORKFLOW_DIR}/node/fv-artifacts/"