#!/usr/bin/env bash
set -euo pipefail

VERSION=${VERSION:-$(date +%Y%m%d%H%M%S)}
API_FLAVOR=${API_FLAVOR:-corex}
ROOT_DIR=$(cd "$(dirname "$0")/.." && pwd)

cd "$ROOT_DIR"

case "$API_FLAVOR" in
  corex)
    API_DOCKERFILE=docker/china/Dockerfile.corex
    ;;
  cuda)
    API_DOCKERFILE=docker/global/Dockerfile.cuda
    ;;
  *)
    echo "Unsupported API_FLAVOR: $API_FLAVOR. Use corex or cuda." >&2
    exit 1
    ;;
esac

echo "Building UnisMindMap API image (${API_FLAVOR}): unis-mindmap-api:${VERSION}"
sudo docker build -f "$API_DOCKERFILE" \
  -t unis-mindmap-api:${VERSION} \
  -t unis-mindmap-api:${API_FLAVOR} \
  -t unis-mindmap-api:latest \
  .

echo "Building UnisMindMap Web image: unis-mindmap-web:${VERSION}"
sudo docker build -f web_ui/Dockerfile.portable \
  -t unis-mindmap-web:${VERSION} \
  -t unis-mindmap-web:latest \
  web_ui