#!/bin/bash
set -e

usage() {
cat << EOF
Usage: $0 [OPTIONS]
Build a new Python Docker image.

Options:
    -g, --gpu                 Build an image with GPU support.
    -c, --use-cache           Use layer cache when building a new image.
EOF
}

CACHE_FLAG='--no-cache'
DOCKERFILE='Dockerfile'
ACCELERATOR='none'
IMAGE_TAG='kaggle/python-build'
BUILD_ARGS=''

while :; do
    case "$1" in 
        -h|--help)
            usage
            exit
            ;;
        -g|--gpu)
            IMAGE_TAG='kaggle/python-gpu-build'
            DOCKERFILE='gpu.Dockerfile'
            ACCELERATOR='gpu'
            ;;
        -c|--use-cache)
            CACHE_FLAG=''
            ;;
        -?*)
            usage
            printf 'ERROR: Unknown option: %s\n' "$1" >&2
            exit
            ;;
        *)            
            break
    esac

    shift
done

BUILD_ARGS+=" --build-arg GIT_COMMIT=$(git rev-parse HEAD)"
BUILD_ARGS+=" --build-arg BUILD_DATE=$(date '+%Y%m%d-%H%M%S')"

readonly CACHE_FLAG
readonly DOCKERFILE
readonly ACCELERATOR
readonly IMAGE_TAG
readonly BUILD_ARGS

SRCDIR=$(dirname "${BASH_SOURCE[0]}")
DOCKERFILE_OUTDIR="${SRCDIR}/.generated"
mkdir -p $DOCKERFILE_OUTDIR
DOCKERFILE_PATH="$DOCKERFILE_OUTDIR/$DOCKERFILE"

# Generate Dockerfile from template.
echo "Generating Dockerfile from template..."
docker run --rm -v $PWD:/input:ro gcr.io/kaggle-images/go-renderizer:latest --ACCELERATOR=$ACCELERATOR /input/Dockerfile.tmpl > $DOCKERFILE_PATH
echo "==================== $DOCKERFILE START ===================="
cat $DOCKERFILE_PATH
echo "==================== $DOCKERFILE  END  ===================="

set -x
docker build --rm --pull $CACHE_FLAG -t "$IMAGE_TAG" -f "$DOCKERFILE_PATH" $BUILD_ARGS .
