build-docker.sh

#!/bin/bash

# Copyright (c) Facebook, Inc. and its affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
set -e

CENTOS_RELEASE="8.3.2011"
UBUNTU_RELEASE="20.04"
EMP_RELEASE="0.1"
AWS_RELEASE="1.8.177"
FMT_RELEASE="7.1.3"
FOLLY_RELEASE="2021.06.28.00"
GITHUB_PACKAGES="ghcr.io/facebookresearch"

PROG_NAME=$0
usage() {
  cat <<EOF >&2
Usage: $PROG_NAME [-u | -c] [-f]

-c: builds the docker images aginsts centos
-u: builds the docker images against ubuntu (default)
-f: forces a rebuild of all docker image dependencies (even if they already exist)
EOF
  exit 1
}

IMAGE_PREFIX="ubuntu"
OS_RELEASE="${UBUNTU_RELEASE}"
DOCKER_EXTENSION=".ubuntu"
FORCE_REBUILD=false
while getopts u,c,f o; do
  case $o in
  u)
    IMAGE_PREFIX="ubuntu"
    OS_RELEASE="${UBUNTU_RELEASE}"
    DOCKER_EXTENSION=".ubuntu"
    ;;
  c)
    IMAGE_PREFIX="centos"
    OS_RELEASE="${CENTOS_RELEASE}"
    DOCKER_EXTENSION=".centos"
    ;;
  f) FORCE_REBUILD=true ;;
  *) usage ;;
  esac
done
shift "$((OPTIND - 1))"

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd)"
# Docker build must run from this script build dir, so all the relative paths work inside the Dockerfile's
cd "$SCRIPT_DIR" || exit

build_dep_image() {
  local IMAGE=$1
  local DOCKER_DIR=$2
  local BUILD_ARGS=$3
  if [ "${FORCE_REBUILD}" == false ] && docker image inspect "${IMAGE}" >/dev/null 2>&1; then
    printf "%s docker image already exists. To force a rebuild please use '-f' flag...\n" "${IMAGE}"
  else
    printf "attempting to pull image %s from %s...\n" "${IMAGE}" "${GITHUB_PACKAGES}"
    if [ "${FORCE_REBUILD}" == false ] && docker pull "${GITHUB_PACKAGES}/${IMAGE}" 2>/dev/null; then
      # Use the external ghcr.io image (if a local image doesn't exist) instead of building locally...
      IMAGE="${GITHUB_PACKAGES}/${IMAGE}"
      printf "successfully pulled image %s\n\n" "${IMAGE}"
    else
      printf "%s/%s did not exist or '-f' flag was specifed\n" "${GITHUB_PACKAGES}" "${IMAGE}"
      printf "\nBuilding %s docker image...\n" "${IMAGE}"
      # shellcheck disable=SC2086
      docker build \
        ${BUILD_ARGS} \
        -t "${IMAGE}" -f "docker/${DOCKER_DIR}/Dockerfile${DOCKER_EXTENSION}" .
    fi
  fi
  RETURN="${IMAGE}"
}

EMP_IMAGE="fbpcf/${IMAGE_PREFIX}-emp:${EMP_RELEASE}"
build_dep_image "${EMP_IMAGE}" "emp" "--build-arg os_release=${OS_RELEASE} --build-arg emp_release=${EMP_RELEASE}"
EMP_IMAGE="${RETURN}"

AWS_IMAGE="fbpcf/${IMAGE_PREFIX}-aws-s3-core:${AWS_RELEASE}"
build_dep_image "${AWS_IMAGE}" "aws-s3-core" "--build-arg os_release=${OS_RELEASE} --build-arg aws_release=${AWS_RELEASE}"
AWS_IMAGE="${RETURN}"

FOLLY_IMAGE="fbpcf/${IMAGE_PREFIX}-folly:${FOLLY_RELEASE}"
build_dep_image "${FOLLY_IMAGE}" "folly" "--build-arg os_release=${OS_RELEASE} --build-arg folly_release=${FOLLY_RELEASE} --build-arg fmt_release=${FMT_RELEASE}"
FOLLY_IMAGE="${RETURN}"

printf "\nBuilding fbpcf/%s docker image...\n" ${IMAGE_PREFIX}
docker build \
  --build-arg os_release="${OS_RELEASE}" \
  --build-arg emp_image="${EMP_IMAGE}" \
  --build-arg aws_image="${AWS_IMAGE}" \
  --build-arg folly_image="${FOLLY_IMAGE}" \
  --compress \
  -t "fbpcf/${IMAGE_PREFIX}:latest" -f "docker/Dockerfile${DOCKER_EXTENSION}" .