#!/bin/bash

set -e

if [ $# -eq 0 ]; then
    echo "Usage: $0 <image-name>"
    echo "Example: $0 base-debian"
    exit 1
fi

IMAGE_NAME=${1:-base-debian}
echo "Building Docker image: $IMAGE_NAME"

if ! command -v dpkg &> /dev/null; then
    echo "❌ Error: dpkg not found. This script only works on Debian-based systems."
    exit 1
fi

ARCH=$(dpkg --print-architecture)  # Outputs amd64, arm64, armhf, etc.
echo "🛠 Detected architecture: $ARCH"

REPO="registry.tomastm.com"
TAG_LATEST="latest"
BUILD_DIR="./${IMAGE_NAME}"

if [ ! -d "${BUILD_DIR}" ]; then
  echo "❌ Error: Directory '${BUILD_DIR}' does not exist."
  exit 1
fi

_UID=$(id -u)
_GID=$(id -g)

COMMIT_ID=$(git rev-parse --short HEAD)

IMAGE_ARCH="${REPO}/${IMAGE_NAME}:${COMMIT_ID}-${ARCH}"
IMAGE_LATEST="${REPO}/${IMAGE_NAME}:${TAG_LATEST}-${ARCH}"

docker build \
  -t ${IMAGE_ARCH} \
  --build-arg UID=${_UID} \
  --build-arg GID=${_GID} \
  ${BUILD_DIR}

docker tag ${IMAGE_ARCH} ${IMAGE_LATEST}

echo "✅ Successfully built and tagged:"
echo "   📌 ${IMAGE_ARCH}"
echo "   🔄 ${IMAGE_LATEST}"

read -p "🚀 Do you want to push the image to ${REPO}? (y/N): " PUSH_CONFIRM
if [[ "$PUSH_CONFIRM" =~ ^[Yy]$ ]]; then
  echo "📤 Pushing images..."
  docker push ${IMAGE_ARCH}
  docker push ${IMAGE_LATEST}
  echo "✅ Successfully pushed images to ${REPO}"

  echo "ℹ️ Next step: If you have built and pushed both amd64 and arm64 versions, run the following command to create a multi-arch manifest:"
  echo "   docker manifest create ${REPO}/${IMAGE_NAME}:${COMMIT_ID} \\"
  echo "      ${REPO}/${IMAGE_NAME}:${COMMIT_ID}-amd64 \\"
  echo "      ${REPO}/${IMAGE_NAME}:${COMMIT_ID}-arm64"
  echo "   docker manifest push ${REPO}/${IMAGE_NAME}:${COMMIT_ID}"
else
  echo "ℹ️ Skipping push. You can manually push later with:"
  echo "   docker push ${IMAGE_ARCH}"
  echo "   docker push ${IMAGE_LATEST}"
fi