entrypoint.sh

#!/bin/sh
set -e

function main() {
  echo "" # see https://github.com/actions/toolkit/issues/168

  sanitize "${INPUT_NAME}" "name"
  sanitize "${INPUT_USERNAME}" "username"
  sanitize "${INPUT_PASSWORD}" "password"

  # Set global booleans and values for use throughout script
  REGISTRY_NO_PROTOCOL=$(echo "${INPUT_REGISTRY}" | sed -e 's/^https:\/\///g')
  if uses "${INPUT_REGISTRY}" && ! isPartOfTheName "${REGISTRY_NO_PROTOCOL}"; then
    INPUT_NAME="${REGISTRY_NO_PROTOCOL}/${INPUT_NAME}"
  fi

  SHORT_SHA=$(echo "${GITHUB_SHA}" | cut -c1-7)
  isPullRequest=$(! [ "${GITHUB_REF/refs\/pull\//}" = "${GITHUB_REF}" ] && echo "true" || echo "false")
  if $isPullRequest; then
    BRANCH=$GITHUB_HEAD_REF
    PR_NAME="PR$(echo $GITHUB_REF_NAME | cut -d '/' -f1)"
  else
    BRANCH=$GITHUB_REF_NAME
  fi
  BRANCH=$(echo $BRANCH | sed -e "s/\//-/g")
  isOnMaster=$([ "${BRANCH}" == "master" ] && echo "true" || echo "false")
  isReleaseBranch=$(! [ "${BRANCH/release-/}" = "${BRANCH}" ] && echo "true" || echo "false")
  isGitTag=$([ "${GITHUB_REF_TYPE}" == "tag" ] && echo "true" || echo "false")
  hasCustomTag=$([ $(echo "${INPUT_NAME}" | sed -e "s/://g") != "${INPUT_NAME}" ] && echo "true" || echo "false")
  # End globals

  if uses "${INPUT_TAGS}"; then
    TAGS=$(echo "${INPUT_TAGS}" | sed "s/,/ /g")
  else 
    translateDockerTag
  fi

  if uses "${INPUT_WORKDIR}"; then
    changeWorkingDirectory
  fi

  echo ${INPUT_PASSWORD} | docker login -u ${INPUT_USERNAME} --password-stdin ${INPUT_REGISTRY}

  FIRST_TAG=$(echo $TAGS | cut -d ' ' -f1)
  DOCKERNAME="${INPUT_NAME}:${FIRST_TAG}"
  BUILDPARAMS=""
  CONTEXT="."

  if uses "${INPUT_DOCKERFILE}"; then
    useCustomDockerfile
  fi
  if uses "${INPUT_BUILDARGS}"; then
    addBuildArgs
  fi
  if uses "${INPUT_CONTEXT}"; then
    CONTEXT="${INPUT_CONTEXT}"
  fi
  if usesBoolean "${INPUT_CACHE}"; then
    useBuildCache
  fi
  if usesBoolean "${INPUT_SNAPSHOT}"; then
    useSnapshot
  fi
  if usesBoolean "${INPUT_ADDFLUXTAG}"; then
    addFluxTag
  fi
  if usesBoolean "${INPUT_ADDTOMTAG}"; then
    addTomTag
  fi

  push

  echo "::set-output name=tag::${FIRST_TAG}"
  DIGEST=$(docker inspect --format='{{index .RepoDigests 0}}' ${DOCKERNAME})
  echo "::set-output name=digest::${DIGEST}"

  docker logout
}

function sanitize() {
  if [ -z "${1}" ]; then
    >&2 echo "Unable to find the ${2}. Did you set with.${2}?"
    exit 1
  fi
}

function isPartOfTheName() {
  [ $(echo "${INPUT_NAME}" | sed -e "s/${1}//g") != "${INPUT_NAME}" ]
}

function translateDockerTag() {
  if $isPullRequest; then
    TAGS="${GITHUB_SHA}"
  elif $hasCustomTag; then
    TAGS=$(echo ${INPUT_NAME} | cut -d':' -f2)
    INPUT_NAME=$(echo ${INPUT_NAME} | cut -d':' -f1)
  elif $isOnMaster; then
    TAGS="latest"
  elif $isGitTag; then
    if usesBoolean "${INPUT_TAG_NAMES}"; then
      TAGS=$GITHUB_REF_NAME
    else
      TAGS="latest"
    fi;
  else
    TAGS="${BRANCH}"
  fi;
}

function changeWorkingDirectory() {
  cd "${INPUT_WORKDIR}"
}

function useCustomDockerfile() {
  BUILDPARAMS="${BUILDPARAMS} -f ${INPUT_DOCKERFILE}"
}

function addBuildArgs() {
  for ARG in $(echo "${INPUT_BUILDARGS}" | tr ',' '\n'); do
    BUILDPARAMS="${BUILDPARAMS} --build-arg ${ARG}"
    echo "::add-mask::${ARG}"
  done
}

function useBuildCache() {
  if docker pull ${DOCKERNAME} 2>/dev/null; then
    BUILDPARAMS="$BUILDPARAMS --cache-from ${DOCKERNAME}"
  fi
}

function uses() {
  [ ! -z "${1}" ]
}

function usesBoolean() {
  [ ! -z "${1}" ] && [ "${1}" = "true" ]
}

function useSnapshot() {
  local TIMESTAMP=`date +%Y%m%d%H%M%S`
  local VSHORT_SHA=$(echo "${GITHUB_SHA}" | cut -c1-6)
  local SNAPSHOT_TAG="${TIMESTAMP}${VSHORT_SHA}"
  TAGS="${TAGS} ${SNAPSHOT_TAG}"
  echo ::set-output name=snapshot-tag::"${SNAPSHOT_TAG}"
}

function addFluxTag() {
  local FLUX_TAG="${BRANCH}-${SHORT_SHA}"
  TAGS="${TAGS} ${FLUX_TAG}"
  echo ::set-output name=flux-tag::"${FLUX_TAG}"
}

function addTomTag() {
  if $isOnMaster; then
    local DATESTAMP=$(TZ=UTC git show --quiet HEAD --date='format-local:%y-%m-%d' --format="%cd")
    local TOM_TAG="${DATESTAMP}.${GITHUB_RUN_NUMBER}"
  elif $isPullRequest; then
    local TOM_TAG="${PR_NAME}.${GITHUB_RUN_NUMBER}"
  elif $isReleaseBranch; then
    local TOM_TAG="${BRANCH/release-/}-hotfix"
  else
    local TOM_TAG="${BRANCH}.${SHORT_SHA}"
  fi
  TAGS="${TAGS} ${TOM_TAG}"
  echo ::set-output name=tom-tag::"${TOM_TAG}"
}

function push() {
  local BUILD_TAGS=""
  for TAG in ${TAGS}
  do
    BUILD_TAGS="${BUILD_TAGS}-t ${INPUT_NAME}:${TAG} "
  done
  docker build ${INPUT_BUILDOPTIONS} ${BUILDPARAMS} ${BUILD_TAGS} ${CONTEXT}

  for TAG in ${TAGS}
  do
    docker push "${INPUT_NAME}:${TAG}"
  done
}

main