#!/usr/bin/with-contenv bash

# Register exit handler
trap scriptExitHandler EXIT

function scriptExitHandler() {
  LAST_EXIT_CODE=$?
  if [ "${LAST_EXIT_CODE}" = "0" ]; then
    echo "> Script finished successfully"
    exit "${LAST_EXIT_CODE}"
  fi

  echo "> Script finished with an error"
  exit "${LAST_EXIT_CODE}"
}

# Check if plugin installed. This is very basic check that doesn't involve database
function isPluginInstalled() {
  PLUGIN="${1:?PLUGIN is required}"

  if [ -d "${PLUGIN_PATH}" ] && [ -f "${PLUGIN_PATH}/${PLUGIN}.php" ]; then
    return 0
  fi
  return 1
}

# Install plugin
function installPlugin() {
  wp plugin install "${@}" >/dev/null 2>&1
  sleep 0.5
}

# Main function
function main() {
  PLUGIN_LIST="${WORDPRESS_PLUGIN_LIST:-}"
  PLUGIN_STRICT_INSTALL="${WORDPRESS_PLUGIN_INSTALL_STRICT:-false}"
  WP_CONTENT_PATH="/var/www/html/wp-content"

  echo "> Automated WordPress Plugin Installer"
  if [ -z "${PLUGIN_LIST}" ]; then
    echo "> No plugins defined. Skipping installation."
    return 0
  fi

  echo "> About to install defined plugins"
  for PLUGIN_EXPR in ${PLUGIN_LIST}; do
    IFS=':' read -ra PLUGIN <<<"${PLUGIN_EXPR}"

    PLUGIN_NAME="${PLUGIN[0]}"
    PLUGIN_PATH="${WP_CONTENT_PATH}/plugins/${PLUGIN_NAME}"
    export PLUGIN_PATH

    if isPluginInstalled "${PLUGIN_NAME}"; then
      echo "> Plugin '${PLUGIN_NAME}' already installed and will be skipped."
      continue
    fi

    WP_PLUGIN_INSTALL_ARGS="${PLUGIN_NAME}"

    if [ -n "${PLUGIN[1]}" ]; then
      WP_PLUGIN_INSTALL_ARGS="${WP_PLUGIN_INSTALL_ARGS} --version=${PLUGIN[1]}"
    fi

    echo "> Installing plugin '${PLUGIN_NAME}' version '${PLUGIN[1]}'"
    installPlugin "${WP_PLUGIN_INSTALL_ARGS}" &
  done

  echo "> Waiting for all plugins to install..."
  wait

  # Plugins are installed concurrently, so we need to verify if installed, separately
  echo "> About to verify install of defined plugins"

  FAILED_COUNT=0

  for PLUGIN_EXPR in ${PLUGIN_LIST}; do
    IFS=':' read -ra PLUGIN <<<"${PLUGIN_EXPR}"

    PLUGIN_NAME="${PLUGIN[0]}"
    PLUGIN_PATH="${WP_CONTENT_PATH}/plugins/${PLUGIN_NAME}"
    export PLUGIN_PATH

    if isPluginInstalled "${PLUGIN_NAME}"; then
      echo "> Plugin '${PLUGIN_NAME}' installed"
      continue
    fi

    ((FAILED_COUNT = FAILED_COUNT + 1))
    echo "> Warning: Plugin '${PLUGIN_NAME}' failed to install"
  done

  echo "> Total of ${FAILED_COUNT} plugins failed to install"

  if [ "${PLUGIN_STRICT_INSTALL}" = "true" ] && [ "${FAILED_COUNT}" != "0" ]; then
    echo "> WORDPRESS_PLUGIN_INSTALL_STRICT is set to true. Terminating with non-zero exit code"
    return 1
  fi
}

main "${@}"
exit $?