ray/ci/travis/build-autoscaler-images.sh

#!/usr/bin/env bash

# This script build docker images for autoscaler.
# For now, we only build python3.6 images.
set -e
set -x

SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
ROOT_DIR=$(cd "$SCRIPT_DIR"/../../; pwd)
DOCKER_USERNAME="raytravisbot"

# We will only build and push when we are building branch build.
if [[ "$TRAVIS" == "true" && "$TRAVIS_PULL_REQUEST" == "false" ]]; then
    echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin

    wheel="$(basename "$ROOT_DIR"/.whl/*cp36m-manylinux*)"
    commit_sha=$(echo "$TRAVIS_COMMIT" | head -c 6)
    cp -r "$ROOT_DIR"/.whl "$ROOT_DIR"/docker/autoscaler/.whl

    docker build -q -t rayproject/base-deps docker/base-deps

    docker build \
        --build-arg WHEEL_PATH=".whl/$wheel" \
        --build-arg WHEEL_NAME="$wheel" \
        -t rayproject/autoscaler:"$commit_sha" \
        "$ROOT_DIR"/docker/autoscaler

    docker tag rayproject/base-deps rayproject/base-deps:"$commit_sha" 
    docker push rayproject/base-deps:"$commit_sha"
    docker push rayproject/autoscaler:"$commit_sha"


    # We have a branch build, e.g. release/v0.7.0
    if [[ "$TRAVIS_BRANCH" != "master" ]]; then
       # Replace / in branch name to - so it is legal tag name
       normalized_branch_name=$(echo "$TRAVIS_BRANCH" | sed -e "s/\//-/")
       docker tag rayproject/autoscaler:"$commit_sha" rayproject/autoscaler:"$normalized_branch_name"
       docker tag rayproject/base-deps:"$commit_sha" rayproject/base-deps:"$normalized_branch_name"
       docker push rayproject/autoscaler:"$normalized_branch_name"
       docker push rayproject/base-deps:"$normalized_branch_name"
    else
       docker tag rayproject/autoscaler:"$commit_sha" rayproject/autoscaler:latest
       docker tag rayproject/base-deps:"$commit_sha" rayproject/base-deps:latest
       docker push rayproject/autoscaler:latest
       docker push rayproject/base-deps:latest
    fi
fi