#!/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 .