#!/usr/bin/env bash
# nvidia/nodewright v0.17.1 - base variant: aicr-eks-training
# Path: fetch this entry's retained objects, check their SHA-256, then apply
# them with kubectl. It uses curl, shasum, awk and kubectl. It needs no cub
# CLI and no ConfigHub account.
# Generated by scripts/generate-public-site.mjs from this entry's listing:
# https://confighub.github.io/helm-expt/site/listings/nvidia-nodewright-v0-17-1-aicr-eks-training.json
# Entry page: https://confighub.github.io/helm-expt/site/charts/nvidia-nodewright-v0-17-1.html
set -euo pipefail

say() { printf '\n>> %s\n' "$*"; }

for tool in curl awk kubectl; do
  if ! command -v "$tool" >/dev/null 2>&1; then
    printf '%s is required and was not found.\n' "$tool" >&2
    exit 1
  fi
done
if command -v shasum >/dev/null 2>&1; then
  sha256_of() { shasum -a 256 "$1" | awk '{print $1}'; }
elif command -v sha256sum >/dev/null 2>&1; then
  sha256_of() { sha256sum "$1" | awk '{print $1}'; }
else
  printf 'shasum is required and was not found.\n' >&2
  exit 1
fi

work=./nvidia-nodewright-v0-17-1-aicr-eks-training-objects
objects_url=https://raw.githubusercontent.com/confighub/helm-expt/main/recipes/nvidia/nodewright/v0.17.1/revisions/aicr-eks-training/r001/rendered/release-objects.yaml
objects_sha256=6b25ebafd5569b97caf3dab889873641f69ec5595fa7d2ff5262042d8927b3ed

say "Fetch the retained objects into $work/objects.yaml"
mkdir -p "$work"
curl -fsSL -o "$work/objects.yaml" "$objects_url"

say "Check the file against the SHA-256 in the listing"
fetched_sha256="$(sha256_of "$work/objects.yaml")"
if [ "$fetched_sha256" != "$objects_sha256" ]; then
  printf 'The file differs from the listing, so the script stops. Nothing was applied.\n  listing: %s\n  fetched: %s\n' "$objects_sha256" "$fetched_sha256" >&2
  exit 1
fi
printf 'The file matches: sha256 %s\n' "$fetched_sha256"

say "Split the file into its 18 objects. The cluster is untouched so far"
rm -rf "$work/docs" "$work/crds" "$work/rest" "$work/unplaced"
mkdir -p "$work/docs" "$work/crds" "$work/rest" "$work/unplaced"
awk -v dir="$work/docs" 'BEGIN { n = 0; f = sprintf("%s/doc-%04d.yaml", dir, n) } /^---[ \t]*$/ { close(f); n++; f = sprintf("%s/doc-%04d.yaml", dir, n); next } { print > f }' "$work/objects.yaml"
for n in 0005 0006; do mv "$work/docs/doc-$n.yaml" "$work/crds/"; done
for n in 0001 0002 0004 0007 0008 0009 0010 0011 0012 0013 0014 0015; do mv "$work/docs/doc-$n.yaml" "$work/rest/"; done
for n in 0016 0017 0018; do mv "$work/docs/doc-$n.yaml" "$work/rest/"; done
for n in 0003; do mv "$work/docs/doc-$n.yaml" "$work/unplaced/"; done
ls "$work/crds" "$work/rest" "$work/unplaced"

say "Check that kubectl reaches a cluster"
if ! kubectl get --raw /version >/dev/null 2>&1; then
  printf 'kubectl cannot reach a cluster. Point it at a throwaway cluster and re-run this script.\n' >&2
  exit 1
fi
kubectl config current-context 2>/dev/null || true

say "Create the skyhook namespace if it does not exist"
if ! kubectl get namespace skyhook >/dev/null 2>&1; then
  kubectl create namespace skyhook
fi

wait_for_crd() {
  crd_name="$1"
  deadline=$(( $(date +%s) + 180 ))
  until kubectl get "crd/${crd_name}" >/dev/null 2>&1; do
    if [ "$(date +%s)" -ge "$deadline" ]; then
      printf "CRD %s did not appear within 180 seconds.\n" "$crd_name" >&2
      return 1
    fi
    sleep 2
  done
  kubectl wait --for=condition=Established --timeout=120s "crd/${crd_name}"
}

say "Check the 2 CRDs this file holds"
missing_crds=0
if ! kubectl get crd/deploymentpolicies.skyhook.nvidia.com >/dev/null 2>&1; then
  missing_crds=1
fi
if ! kubectl get crd/skyhooks.skyhook.nvidia.com >/dev/null 2>&1; then
  missing_crds=1
fi
if [ "$missing_crds" -eq 1 ]; then
  say "A CRD is missing, so apply the file's CRDs first"
  kubectl apply --server-side -f "$work/crds"
  crd_outcome="were applied first"
elif kubectl diff --server-side -f "$work/crds" >/dev/null 2>&1; then
  say "The CRDs already exist and match this file. The script leaves them under their current owner"
  crd_outcome="already existed and matched, so the script left them alone"
elif [ "${APPLY_CRDS:-0}" = "1" ]; then
  say "The existing CRDs differ from this file. APPLY_CRDS=1 is set, so the script applies the file's CRDs"
  kubectl apply --server-side -f "$work/crds"
  crd_outcome="differed and were applied, because APPLY_CRDS=1 was set"
else
  say "The CRDs already exist and differ from this file, or another tool owns them"
  cat <<'EOF_CRDS'
This script does not change CRDs it did not just create.
If this is an upgrade, the objects that follow may need the newer CRDs.
To let this script apply the file's CRDs, re-run with:
  APPLY_CRDS=1 bash try-objects.sh
EOF_CRDS
  exit 1
fi
wait_for_crd deploymentpolicies.skyhook.nvidia.com
wait_for_crd skyhooks.skyhook.nvidia.com

say "Apply the other 15 objects"
kubectl apply -f "$work/rest"

say "Apply 1 object that names no namespace into skyhook, the namespace this entry was built for"
kubectl apply -n skyhook -f "$work/unplaced"

say "Done. The 2 CRDs $crd_outcome. kubectl applied the other 16 objects of this entry's retained file."
say "The script stops here. It did not wait for any workload to start"
say "Not run by this script: 2 steps of this entry's plan. The entry page says who runs each one."
cat <<'EOF_NOT_RUN'
   - Before an upgrade, delete the operator Deployment if its selector would change. A Deployment's selector is immutable. The listing gives the rest.
   - Before a delete, remove the custom resources and the webhook configurations. Two Jobs run while the operator is still there. The listing gives the rest.
EOF_NOT_RUN