#!/usr/bin/env bash
set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
source "$(dirname $0)/../common/functions"

release_and_deploy() {
  source "$(dirname $0)/../common/functions"
  local APP="$1"; local IMAGE="dokku/$APP"

  if verify_app_image "$APP"; then
    if is_image_buildstep_based "$IMAGE"; then
      IMAGE_SOURCE_TYPE="buildstep"
    else
      IMAGE_SOURCE_TYPE="dockerfile"
    fi

    dokku_log_info1 "Releasing $APP ..."
    dokku release "$APP" "$IMAGE_SOURCE_TYPE"
    dokku_log_info1 "Deploying $APP ..."
    dokku deploy "$APP"
    dokku_log_info2 "Application deployed:"
         dokku urls $APP | sed "s/^/       /"
    echo
  fi
}

generate_scale_file() {
  local APP="$1"; local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
  copy_from_image "$APP" "/app/DOKKU_SCALE" "$DOKKU_ROOT/$APP" 2>/dev/null || dokku_log_info1_quiet "DOKKU_SCALE not found in app image"
  if [[ ! -f $DOKKU_SCALE_FILE ]]; then
    local TMP_WORK_DIR=$(mktemp --tmpdir=/tmp -d DOKKU_SCALE.XXXXX)
    trap 'rm -rf $TMP_WORK_DIR' EXIT


    if is_image_buildstep_based "$APP"; then
      copy_from_image "$APP" /app/Procfile $TMP_WORK_DIR 2>/dev/null || true
      copy_from_image "$APP" /app/.release $TMP_WORK_DIR 2>/dev/null || true

      if [[ -f $TMP_WORK_DIR/Procfile ]];then
        local PROCFILE="$TMP_WORK_DIR/Procfile"
      elif [[ -f $TMP_WORK_DIR/.release ]];then
        local PROCFILE="$TMP_WORK_DIR/.release"
      else
        echo "web=1" >> $DOKKU_SCALE_FILE
      fi
    else
      echo "web=1" >> $DOKKU_SCALE_FILE
    fi

    if [[ -f $PROCFILE ]]; then
      while read line || [ -n "$line" ]
      do
        local PROC_NAME=${line%%:*}
        local PROC_CMD=${line#*:}
        [[ "$PROC_NAME" =~ -.* ]] && continue
        [[ -n "$PROC_CMD" ]] && echo "$PROC_NAME=1" >> $DOKKU_SCALE_FILE
      done < "$PROCFILE"
    fi

    dokku_log_info1_quiet "New DOKKU_SCALE file generated"
    while read line || [ -n "$line" ]
    do
      dokku_log_info2_quiet "$line"
    done < "$DOKKU_SCALE_FILE"
  fi
}

set_scale() {
  local APP="$1"; local DOKKU_SCALE_FILE="$DOKKU_ROOT/$APP/DOKKU_SCALE"
  shift 1
  local SCALE_SETTINGS=("$@")
  for procscale in "${SCALE_SETTINGS[@]}"; do
    PROC_NAME=${procscale%%=*}
    PROC_COUNT=${procscale#*=}
    is_number $PROC_COUNT || dokku_log_fail "ps:scale $PROC_COUNT is not a number"
    dokku_log_info1_quiet "Scaling $APP:$PROC_NAME to $PROC_COUNT"
    if (egrep -q ^${PROC_NAME}= $DOKKU_SCALE_FILE > /dev/null 2>&1);then
      sed --in-place "s:^${PROC_NAME}=.*:$PROC_NAME=$PROC_COUNT:g" $DOKKU_SCALE_FILE
    else
      echo "$PROC_NAME=$PROC_COUNT" >> $DOKKU_SCALE_FILE
    fi
  done
}