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

fn-domains-app-enabled() {
  declare APP="$1"
  local DOMAINS_APP_ENABLED=false
  if [[ "$(is_app_vhost_enabled "$APP")" == "true" ]]; then
    DOMAINS_APP_ENABLED=true
  fi
  echo "$DOMAINS_APP_ENABLED"
}

fn-domains-app-vhosts() {
  declare APP="$1"
  local APP_VHOST_PATH="$DOKKU_ROOT/$APP/VHOST"
  if [[ -f "$APP_VHOST_PATH" ]]; then
    tr '\n' ' ' <"$APP_VHOST_PATH"
  fi
}

fn-domains-global-enabled() {
  local DOMAINS_GLOBAL_ENABLED=false
  if [[ "$(is_global_vhost_enabled)" == "true" ]]; then
    DOMAINS_GLOBAL_ENABLED=true
  fi
  echo "$DOMAINS_GLOBAL_ENABLED"
}

fn-domains-global-vhosts() {
  if [[ "$(is_global_vhost_enabled)" == "true" ]]; then
    get_global_vhosts | tr '\n' ' '
  fi
}

fn-domains-generate-urls() {
  declare APP="$1" DEFAULT_SCHEME="$2" DEFAULT_LISTEN_PORT="$3"

  local app_vhosts="$(plugn trigger domains-list "$APP")"
  if [[ -n "$app_vhosts" ]]; then
    for app_vhost in $app_vhosts; do
      fn-domains-generate-urls-from-config "$APP" "$DEFAULT_SCHEME" "$app_vhost" "$DEFAULT_LISTEN_PORT"
    done
  else
    if [[ -s "$DOKKU_ROOT/VHOST" ]]; then
      while read -r VHOST || [[ -n "$VHOST" ]]; do
        fn-domains-generate-urls-from-config "$APP" "$DEFAULT_SCHEME" "$VHOST" "$DEFAULT_LISTEN_PORT"
      done <"$DOKKU_ROOT/VHOST"
    else
      fn-domains-generate-urls-from-config "$APP" "$DEFAULT_SCHEME" "$(hostname -f)" "$DEFAULT_LISTEN_PORT"
    fi
  fi
}

fn-domains-generate-urls-from-config() {
  declare APP="$1" DEFAULT_SCHEME="$2" VHOST="$3" DEFAULT_LISTEN_PORT="$4"
  local APP_PORT_MAP="$(plugn trigger ports-get "$APP")"

  if [[ "$(plugn trigger proxy-is-enabled "$APP")" == "false" ]]; then
    local DOKKU_APP_WEB_LISTENERS PORT
    DOKKU_APP_WEB_LISTENERS="$(plugn trigger network-get-listeners "$APP" "web" | xargs)"
    for DOKKU_APP_WEB_LISTENER in $DOKKU_APP_WEB_LISTENERS; do
      listen_port="$(echo "$DOKKU_APP_WEB_LISTENER" | cut -d ':' -f2)"
      fn-domains-generate-url "$DEFAULT_SCHEME" "$VHOST" "$listen_port"
    done
    shopt -u nullglob
  elif [[ -n "$APP_PORT_MAP" ]]; then
    CERTS_EXIST="$(plugn trigger certs-exists "$APP")"
    local port_map
    while IFS= read -r port_map; do
      local scheme="$(awk -F ':' '{ print $1 }' <<<"$port_map")"
      local listen_port="$(awk -F ':' '{ print $2 }' <<<"$port_map")"
      if [[ "$CERTS_EXIST" != "true" ]] && [[ "$scheme" == "https" ]]; then
        continue
      fi
      fn-domains-generate-url "$scheme" "$VHOST" "$listen_port"
    done <<<"$APP_PORT_MAP"
  else
    fn-domains-generate-url "$DEFAULT_SCHEME" "$VHOST" "$DEFAULT_LISTEN_PORT"
  fi
}

fn-domains-generate-url() {
  declare SCHEME="$1" VHOST="$2" PORT="$3"
  if [[ "$PORT" == "80" ]]; then
    echo "http://$VHOST"
  elif [[ "$PORT" == "443" ]]; then
    if [[ "$SCHEME" == "https" ]]; then
      echo "https://$VHOST"
    else
      echo "$SCHEME://$VHOST:$PORT"
    fi
  else
    echo "$SCHEME://$VHOST:$PORT"
  fi
}