#!/usr/bin/env bash
set -e

# sets test image and gets base url variables used in tests.
source test_common.sh

# Test pushing with containerd, and pulling with docker
ctr image pull "$FULL_TEST_IMAGE"
ctr image tag "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE"

# Push using containerd
ctr image push --platform amd64 --plain-http "$WHALESONG_IMAGE"

# Clean stored image before pulling back down again
ctr image rm "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE"

# Pull using containerd
ctr image pull --plain-http "$WHALESONG_IMAGE"

# Run image using containerd
OUTPUT=$(ctr run --rm "$WHALESONG_IMAGE" testcontainer python -c 'print("hi from ci")')

if [[ $OUTPUT == "hi from ci" ]]; then
	echo "Pull/push test successful for single platform!"
else
	echo "Pull/push test failed for single platform!"
	exit 1
fi

# Repeat test but push _all_ platforms, i.e., test creating an index (manifest list) too.
ctr image rm "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE" || true
ctr image pull --all-platforms "$FULL_TEST_IMAGE"
ctr image tag "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE"

# Push all platforms with containerd.
ctr image push --plain-http "$WHALESONG_IMAGE"

# Clean stored image before pulling back down again
ctr image rm "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE"

# Pull using containerd
ctr image pull --plain-http "$WHALESONG_IMAGE"

# Run image using containerd
OUTPUT=$(ctr run --rm "$WHALESONG_IMAGE" testcontainer python -c 'print("hi from ci")')

if [[ $OUTPUT == "hi from ci" ]]; then
	echo "Pull/push test successful for multiple platforms!"
else
	echo "Pull/push test failed for multiple platforms!"
	exit 1
fi

# Optional in CI: cleanup
set +e
ctr image rm "$FULL_TEST_IMAGE" "$WHALESONG_IMAGE" || true