#!/usr/bin/env bash
set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../.." && pwd)"
STACK_ENV="${STACK_ENV:-$ROOT_DIR/stack.env}"
ENSURE_XMPP_SECRET_SCRIPT="$ROOT_DIR/utilities/prosody/ensure_xmpp_secret.sh"

if [[ -f "$STACK_ENV" ]]; then
  set -a
  . "$STACK_ENV"
  set +a
fi

if [[ -x "$ENSURE_XMPP_SECRET_SCRIPT" ]]; then
  XMPP_SECRET="$("$ENSURE_XMPP_SECRET_SCRIPT" "$STACK_ENV")"
  export XMPP_SECRET
fi

STACK_ID="${GIA_STACK_ID:-${STACK_ID:-}}"
STACK_ID="$(echo "$STACK_ID" | tr -cs 'a-zA-Z0-9._-' '-' | sed 's/^-*//; s/-*$//')"

name_with_stack() {
  local base="$1"
  if [[ -n "$STACK_ID" ]]; then
    echo "${base}_${STACK_ID}"
  else
    echo "$base"
  fi
}

POD_NAME="$(name_with_stack "gia")"
PROSODY_CONTAINER="$(name_with_stack "prosody_gia")"

PROSODY_CONFIG_FILE="${QUADLET_PROSODY_CONFIG_FILE:-$ROOT_DIR/utilities/prosody/prosody.cfg.lua}"
PROSODY_CERTS_DIR="${QUADLET_PROSODY_CERTS_DIR:-$ROOT_DIR/.podman/gia_prosody_certs}"
PROSODY_DATA_DIR="${QUADLET_PROSODY_DATA_DIR:-$ROOT_DIR/.podman/gia_prosody_data}"
PROSODY_LOGS_DIR="${QUADLET_PROSODY_LOGS_DIR:-$ROOT_DIR/.podman/gia_prosody_logs}"
PROSODY_IMAGE="${PROSODY_IMAGE:-docker.io/prosody/prosody-alpine:latest}"

mkdir -p "$PROSODY_CERTS_DIR" "$PROSODY_DATA_DIR" "$PROSODY_LOGS_DIR"

up() {
  local run_args=()
  local pod_state=""
  if podman pod exists "$POD_NAME"; then
    pod_state="$(podman pod inspect "$POD_NAME" --format '{{.State}}' 2>/dev/null || true)"
    if [[ "$pod_state" == "Running" ]]; then
      run_args+=(--pod "$POD_NAME")
    else
      echo "Warning: pod '$POD_NAME' state is '$pod_state'; starting $PROSODY_CONTAINER standalone with explicit ports." >&2
      run_args+=(-p 5222:5222 -p 5269:5269 -p 5280:5280 -p 8888:8888)
    fi
  else
    echo "Warning: pod '$POD_NAME' not found; starting $PROSODY_CONTAINER standalone with explicit ports." >&2
    run_args+=(-p 5222:5222 -p 5269:5269 -p 5280:5280 -p 8888:8888)
  fi
  podman run -d \
    --replace \
    --name "$PROSODY_CONTAINER" \
    "${run_args[@]}" \
    --env-file "$STACK_ENV" \
    -v "$PROSODY_CONFIG_FILE:/etc/prosody/prosody.cfg.lua:ro" \
    -v "$PROSODY_CERTS_DIR:/etc/prosody/certs" \
    -v "$PROSODY_DATA_DIR:/var/lib/prosody" \
    -v "$PROSODY_LOGS_DIR:/var/log/prosody" \
    -v "$ROOT_DIR:/code" \
    "$PROSODY_IMAGE" >/dev/null
  if [[ " ${run_args[*]} " == *" --pod "* ]]; then
    echo "Started $PROSODY_CONTAINER in pod $POD_NAME"
  else
    echo "Started $PROSODY_CONTAINER standalone (not attached to pod $POD_NAME)"
  fi
}

down() {
  podman rm -f "$PROSODY_CONTAINER" >/dev/null 2>&1 || true
  echo "Stopped $PROSODY_CONTAINER"
}

status() {
  podman ps --format "table {{.Names}}\t{{.Status}}" | grep -E "^$PROSODY_CONTAINER\b" || true
}

logs() {
  podman logs -f "$PROSODY_CONTAINER"
}

case "${1:-}" in
  up)
    up
    ;;
  down)
    down
    ;;
  restart)
    down
    up
    ;;
  status)
    status
    ;;
  logs)
    logs
    ;;
  *)
    echo "Usage: $0 {up|down|restart|status|logs}" >&2
    exit 2
    ;;
esac