#!/usr/bin/env bash

# This script builds the official docker images and pushes them to dockerhub
# after checking with the user.

# bash strict mode
set -euo pipefail

# Check if exactly one argument is supplied
if [ "$#" -ne 1 ]; then
    echo "Usage: $0 <version>" >&2
    exit 1
fi

VERSION_STR=${1}


# The argument should be in the form of x.x.x where each x can be one or more digits
if [[ $1 =~ ^[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
    echo "Validated version number"
else
    echo "Argument must be in the form x.x.x, where x is one or more numbers." >&2
    exit 2
fi


echo "------------------------------------------"
echo "Building swe-agent"
echo "------------------------------------------"
docker build -t sweagent/swe-agent:${VERSION_STR} --build-arg MINICONDA_URL=https://repo.anaconda.com/miniconda/Miniconda3-py39_23.11.0-1-Linux-aarch64.sh -f docker/swe.Dockerfile .
docker build -t sweagent/swe-agent:latest --build-arg MINICONDA_URL=https://repo.anaconda.com/miniconda/Miniconda3-py39_23.11.0-1-Linux-aarch64.sh -f docker/swe.Dockerfile .
echo "------------------------------------------"
echo "Building swe-eval"
echo "------------------------------------------"
docker build -t sweagent/swe-eval:${VERSION_STR} -f docker/eval.Dockerfile .
docker build -t sweagent/swe-eval:latest -f docker/eval.Dockerfile .
echo "------------------------------------------"
echo "Building swe-agent-run"
echo "------------------------------------------"
docker build -t sweagent/swe-agent-run:${VERSION_STR} .
docker build -t sweagent/swe-agent-run:latest .
echo "------------------------------------------"
echo "Building of all images done"
echo "------------------------------------------"
read -p "Do you want to push to dockerhub and add the tag to github? (yes to proceed) " answer

if [ "$answer" != "yes" ]; then
    echo "Abort; Bye" >&2
    exit 3
fi

# Check if the tags already exist on Docker Hub
IMAGES=("sweagent/swe-agent:${VERSION_STR}" "sweagent/swe-agent:latest" "sweagent/swe-eval:${VERSION_STR}" "sweagent/swe-eval:latest" "sweagent/swe-agent-run:${VERSION_STR}" "sweagent/swe-agent-run:latest")
for image in "${IMAGES[@]}"; do
    if [[ $image == *"latest"* ]]; then
        continue
    fi
    IMAGE_NAME="${image%:*}"
    TAG="${image##*:}"
    URL="https://hub.docker.com/v2/repositories/${IMAGE_NAME}/tags/${TAG}/"
    HTTP_STATUS=$(curl -s -o /dev/null -w "%{http_code}" "${URL}")
    if [ "$HTTP_STATUS" -eq 200 ]; then
        echo "The tag '${TAG}' exists for '${IMAGE_NAME}' on Docker Hub." >&2
        echo "Abort" >&2
        exit 4
    fi
done

docker login || {
    echo "Failed to login to dockerhub" >&2
    exit 8
}

# Let's get serious
# Function to execute upon exit
on_error() {
    echo "====> ERROR!!! IMPORTANT: Make sure if you've already pushed something to dockerhub or pushed the tag to github!" >&2
}
trap on_error ERR

git tag v${VERSION_STR} || {
    echo "Failed to create a tag in git" >&2
    exit 5
}
echo "🔥 Tag v${VERSION_STR} created in git!"

git push origin v${VERSION_STR} || {
    echo "Failed to push the tag to github" >&2
    exit 6
}
echo "🔥 Tag v${VERSION_STR} pushed to github"

for image in "${IMAGES[@]}"; do
    docker push ${image}
    echo "🔥 Pushed ${image} to dockerhub"
done