#!/usr/bin/env bash
set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
source "$PLUGIN_AVAILABLE_PATH/config/functions"

get_phase_script() {
  declare desc="extracts app.json from app image and returns the appropriate json key/value"
  local IMAGE="$1"; local PHASE_SCRIPT_KEY="$2"
  local GET_PHASE_SCRIPT_TMP_WORK_DIR=$(mktemp -d "/tmp/dokku_get_phase_script.XXXX")
  local APP_JSON_FILE="$GET_PHASE_SCRIPT_TMP_WORK_DIR/app.json"
  trap 'rm -rf "$GET_PHASE_SCRIPT_TMP_WORK_DIR" > /dev/null' RETURN INT TERM

  copy_from_image "$IMAGE" "/app/app.json" "$GET_PHASE_SCRIPT_TMP_WORK_DIR" 2>/dev/null || true

  if [[ -f "$APP_JSON_FILE" ]];then
    local VALUE=$(get_json_value "scripts.dokku.${PHASE_SCRIPT_KEY}" < "$APP_JSON_FILE")
  else
    return 0
  fi

  echo "$VALUE"
}

execute_script() {
  declare desc="executes appropriate phase script key from app.json"
  local APP="$1"; local IMAGE="$2"; local PHASE_SCRIPT_KEY="$3"
  local SCRIPT_CMD=$(get_phase_script "$IMAGE" "$PHASE_SCRIPT_KEY" 2>/dev/null)
  if [[ -n "$SCRIPT_CMD" ]];then
    dokku_log_info1 "Running '$SCRIPT_CMD' in app container"
    local COMMAND
    COMMAND="set -eo pipefail; [[ \$DOKKU_TRACE ]] && set -x ; "
    COMMAND+=" if [[ -d '/app' ]]; then "
    COMMAND+="   export HOME=/app ; "
    COMMAND+="   cd \$HOME ; "
    COMMAND+=" fi ; "
    COMMAND+=" if [[ -d '/app/.profile.d' ]]; then "
    COMMAND+="   for file in /app/.profile.d/*; do source \$file; done ; "
    COMMAND+=" fi ; "
    COMMAND+=" if [[ -d '/cache' ]]; then "
    COMMAND+="   echo restoring installation cache... ; "
    COMMAND+="   rm -rf /tmp/cache ; "
    COMMAND+="   ln -sf /cache /tmp/cache ; "
    COMMAND+=" fi ; "
    COMMAND+=" $SCRIPT_CMD ;"
    COMMAND+=" if [[ -d '/cache' ]]; then "
    COMMAND+="   echo removing installation cache... ; "
    COMMAND+="   rm -f /tmp/cache ; "
    COMMAND+=" fi "

    local CACHE_DIR="$DOKKU_ROOT/$APP/cache"
    local DOCKER_ARGS=$(: | plugn trigger docker-args-deploy "$APP")
    # strip --restart args from DOCKER_ARGS
    local DOCKER_ARGS=$(sed -e "s/--restart=[[:graph:]]\+[[:blank:]]\?//g" <<< "$DOCKER_ARGS")

    # shellcheck disable=SC2086
    local id=$(docker run "$DOKKU_GLOBAL_RUN_ARGS" -e DOKKU_TRACE="$DOKKU_TRACE" --label=dokku_phase_script="${PHASE_SCRIPT_KEY}" -d -v "$CACHE_DIR:/cache" $DOCKER_ARGS "$IMAGE" /bin/bash -c "$COMMAND")
    if test "$(docker wait "$id")" -eq 0; then
      dokku_container_log_verbose_quiet "$id"
      if [[ "$PHASE_SCRIPT_KEY" != "postdeploy" ]];then
        if ! is_image_herokuish_based "$IMAGE"; then
          local DOKKU_DOCKERFILE_ENTRYPOINT=$(config_get "$APP" DOKKU_DOCKERFILE_ENTRYPOINT)
          local DOKKU_DOCKERFILE_CMD=$(config_get "$APP" DOKKU_DOCKERFILE_CMD)
          [[ -n "$DOKKU_DOCKERFILE_ENTRYPOINT" ]] && local DOCKER_COMMIT_ENTRYPOINT_CHANGE_ARG="--change='$DOKKU_DOCKERFILE_ENTRYPOINT'"
          [[ -n "$DOKKU_DOCKERFILE_CMD" ]] && local DOCKER_COMMIT_CMD_CHANGE_ARG="--change='$DOKKU_DOCKERFILE_CMD'"

          local DOCKER_COMMIT_ARGS="$DOCKER_COMMIT_ENTRYPOINT_CHANGE_ARG $DOCKER_COMMIT_CMD_CHANGE_ARG"
        fi
        # shellcheck disable=SC2086
        eval docker commit $DOCKER_COMMIT_ARGS "$id" "$IMAGE" > /dev/null
      fi
    else
      dokku_container_log_verbose_quiet "$id"
      dokku_log_fail "execution of '$SCRIPT_CMD' failed!"
    fi
  fi
}

exec_app_json_scripts() {
  declare desc="core app.json scripts execution"
  local trigger="$0 app_json_scripts"
  local APP="$1"

  case "$0" in
    *pre-deploy)
      local IMAGE_TAG="$2"
      local IMAGE=$(get_app_image_name "$APP" "$IMAGE_TAG")
      local PHASE_SCRIPT_KEY="predeploy"
      ;;
    *post-deploy)
      local IMAGE_TAG="$4"
      local IMAGE=$(get_app_image_name "$APP" "$IMAGE_TAG")
      local PHASE_SCRIPT_KEY="postdeploy"
      ;;
  esac

  dokku_log_info1 "Attempting to run scripts.dokku.$PHASE_SCRIPT_KEY from app.json (if defined)"
  execute_script "$APP" "$IMAGE" "$PHASE_SCRIPT_KEY"
}

exec_app_json_scripts "$@"