#!/bin/bash
# Tier-1 dispatch for lemmy component operations.
# Delegates to generated run-{op}.sh scripts when present (manifest-plan workflow),
# or falls back to inline _do_{op} functions for direct invocation.
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

[ -f "${SCRIPT_DIR}/env-lemmy" ] && source "${SCRIPT_DIR}/env-lemmy"

LEMMY_NAME="${LEMMY_NAME:-lemmy}"
LEMMY_NAMESPACE="${LEMMY_NAMESPACE:-lemmy}"

CMD_TSK="${CMD_TSK:-${1:-install}}"

source "${SCRIPT_DIR}/lemmy-lib.sh"

_resolve_kubeconfig() {
  if [ -n "${KUBECONFIG:-}" ] && [ -f "$KUBECONFIG" ]; then
    return
  fi
  for candidate in /var/lib/k0s/pki/admin.conf /etc/kubernetes/admin.conf /root/.kube/config; do
    if [ -f "$candidate" ]; then
      export KUBECONFIG="$candidate"
      return
    fi
  done
  echo "ERROR: kubeconfig not found — set KUBECONFIG explicitly" >&2; exit 1
}

_do_install() {
  for manifest in namespace.yaml certificate.yaml referencegrant.yaml \
                  pictrs-pvc.yaml nginx-configmap.yaml lemmy-configmap.yaml \
                  pictrs-deployment.yaml pictrs-service.yaml \
                  lemmy-deployment.yaml lemmy-service.yaml \
                  lemmy-ui-deployment.yaml lemmy-ui-service.yaml \
                  proxy-deployment.yaml proxy-service.yaml \
                  httproute.yaml; do
    local path="$SCRIPT_DIR/manifests/${manifest}"
    [ -f "$path" ] && _kubectl apply -f "$path"
  done
  echo "Waiting for lemmy-proxy pod to be ready..."
  _wait_for_pod 300
  echo "${LEMMY_NAME} installed in namespace ${LEMMY_NAMESPACE}"
}

_do_update() {
  for manifest in certificate.yaml referencegrant.yaml nginx-configmap.yaml \
                  httproute.yaml pictrs-deployment.yaml \
                  lemmy-ui-deployment.yaml proxy-deployment.yaml \
                  lemmy-deployment.yaml; do
    local path="$SCRIPT_DIR/manifests/${manifest}"
    [ -f "$path" ] && _kubectl apply -f "$path"
  done
  _kubectl rollout restart "deployment/lemmy-backend" --namespace "$LEMMY_NAMESPACE"
  echo "Waiting for lemmy-proxy to be ready..."
  _wait_for_pod 180
  echo "${LEMMY_NAME} updated in namespace ${LEMMY_NAMESPACE}"
}

_do_delete() {
  for manifest in httproute.yaml proxy-deployment.yaml proxy-service.yaml \
                  lemmy-ui-deployment.yaml lemmy-ui-service.yaml \
                  lemmy-deployment.yaml lemmy-service.yaml \
                  pictrs-deployment.yaml pictrs-service.yaml; do
    local path="$SCRIPT_DIR/manifests/${manifest}"
    [ -f "$path" ] && _kubectl delete -f "$path" --ignore-not-found
  done
  echo "${LEMMY_NAME} deleted from namespace ${LEMMY_NAMESPACE}"
}

_do_restart() {
  _kubectl rollout restart "deployment/lemmy-backend" --namespace "$LEMMY_NAMESPACE"
  _kubectl rollout restart "deployment/lemmy-ui" --namespace "$LEMMY_NAMESPACE"
  _kubectl rollout restart "deployment/lemmy-proxy" --namespace "$LEMMY_NAMESPACE"
  _wait_for_pod 180
  echo "${LEMMY_NAME} restarted"
}

_do_health() {
  local pod
  pod=$(_kubectl get pod \
    --namespace "$LEMMY_NAMESPACE" \
    --selector "app.kubernetes.io/name=lemmy-proxy" \
    -o jsonpath='{.items[0].metadata.name}' 2>/dev/null)
  if [ -z "$pod" ]; then
    echo "UNHEALTHY: no lemmy-proxy pod found in ${LEMMY_NAMESPACE}" >&2; exit 1
  fi
  if _kubectl exec --namespace "$LEMMY_NAMESPACE" "$pod" -- \
      wget -qO- "http://localhost:80/" >/dev/null 2>&1; then
    echo "HEALTHY: ${LEMMY_NAME} responding"
  else
    echo "UNHEALTHY: ${LEMMY_NAME} did not respond" >&2; exit 1
  fi
}

_resolve_kubeconfig

case "$CMD_TSK" in
  install)  [ -f "$SCRIPT_DIR/run-init.sh" ]    && exec bash "$SCRIPT_DIR/run-init.sh";    _do_install  ;;
  update)   [ -f "$SCRIPT_DIR/run-update.sh" ]  && exec bash "$SCRIPT_DIR/run-update.sh";  _do_update   ;;
  delete)   [ -f "$SCRIPT_DIR/run-delete.sh" ]  && exec bash "$SCRIPT_DIR/run-delete.sh";  _do_delete   ;;
  restart)  [ -f "$SCRIPT_DIR/run-restart.sh" ] && exec bash "$SCRIPT_DIR/run-restart.sh"; _do_restart  ;;
  health)   _do_health ;;
  *)        echo "ERROR: unknown CMD_TSK='${CMD_TSK}'. Valid: install|update|delete|restart|health" >&2; exit 1 ;;
esac