#!/usr/bin/env bash
# Copyright (c) Edgeless Systems GmbH
#
# SPDX-License-Identifier: AGPL-3.0-only

set -euo pipefail
shopt -s inherit_errexit

# Show progress on pipes if `pv` is installed
# Otherwise use plain cat
if ! command -v pv &> /dev/null; then
  PV="cat"
else
  PV="pv"
fi

pack() {
  local cloudprovider=$1
  local unpacked_image=$2
  local packed_image=$3
  local unpacked_image_dir
  unpacked_image_dir=$(mktemp -d)
  local unpacked_image_filename
  unpacked_image_filename=disk.raw
  local tmp_tar_file
  tmp_tar_file=$(mktemp -t verity.XXXXXX.tar)
  cp "${unpacked_image}" "${unpacked_image_dir}/${unpacked_image_filename}"

  case ${cloudprovider} in

  gcp)
    echo "📥 Packing GCP image..."
    tar --owner=0 --group=0 -C "${unpacked_image_dir}" -Sch --format=oldgnu -f "${tmp_tar_file}" "${unpacked_image_filename}"
    "${PV}" "${tmp_tar_file}" | pigz -9c > "${packed_image}"
    rm "${tmp_tar_file}"
    echo "  Repacked image stored in ${packed_image}"
    ;;

  azure)
    echo "📥 Packing Azure image..."
    truncate -s %1MiB "${unpacked_image_dir}/${unpacked_image_filename}"
    qemu-img convert -p -f raw -O vpc -o force_size,subformat=fixed "${unpacked_image_dir}/${unpacked_image_filename}" "${packed_image}"
    echo "  Repacked image stored in ${packed_image}"
    ;;

  *)
    echo "unknown cloud provider"
    exit 1
    ;;
  esac

  rm -r "${unpacked_image_dir}"

}

if [[ $# -ne 3 ]]; then
  echo "Usage: $0 <cloudprovider> <unpacked_image> <packed_image>"
  exit 1
fi

pack "${1}" "${2}" "${3}"