From bc8ad082281ce347ef8f6c28e548c73108582222 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20=E2=80=9CNiols=E2=80=9D=20Jeannerod?=
 <nicolas.jeannerod@moduscreate.com>
Date: Mon, 24 Feb 2025 18:33:44 +0100
Subject: [PATCH] Make removal script support ids or names

---
 infra/proxmox-remove.sh | 82 ++++++++++++++++++++++++++++++-----------
 1 file changed, 61 insertions(+), 21 deletions(-)

diff --git a/infra/proxmox-remove.sh b/infra/proxmox-remove.sh
index 8ab7975..8ae410e 100755
--- a/infra/proxmox-remove.sh
+++ b/infra/proxmox-remove.sh
@@ -18,11 +18,11 @@ mkdir $tmpdir
 api_url=
 username=
 password=
-vm_ids=
+vm_ids_or_names=
 
 help () {
   cat <<EOF
-Usage: $0 [OPTION...] ID [ID...]
+Usage: $0 [OPTION...] ID_OR_NAME [ID_OR_NAME...]
 
 Options:
   --api-url STR     Base URL of the Proxmox API (required)
@@ -35,7 +35,6 @@ Options can also be provided by adding assignments to a '.proxmox' file in the
 current working directory. For instance, it could contain:
 
   api_url=https://192.168.51.81:8006/api2/json
-  cores=7
   username=mireille@pve
   debug=true
 
@@ -64,12 +63,12 @@ while [ $# -gt 0 ]; do
 
     -*) die_with_help "Unknown argument: '%s'." "$argument" ;;
 
-    *) vm_ids="$vm_ids $argument" ;;
+    *) vm_ids_or_names="$vm_ids_or_names $argument" ;;
   esac
 done
 
-if [ -z "$vm_ids" ]; then
-  die_with_help "Required: at least one VM id.\n"
+if [ -z "$vm_ids_or_names" ]; then
+  die_with_help "Required: at least one VM id or name.\n"
 fi
 
 if [ -z "$api_url" ] || [ -z "$username" ] || [ -z "$password" ]; then
@@ -129,48 +128,89 @@ proxmox_sync () (
   done
 )
 
+################################################################################
+## Grab VM options
+##
+## Takes the name of the VM, grabs `.#vmOptions.<name>` and gets the id from it.
+
+is_integer () {
+  [ "$1" -eq "$1" ] 2>/dev/null
+}
+
+grab_vm_options () {
+  local options
+
+  if is_integer "$1"; then
+    vm_id=$1
+    vm_name="#$1"
+
+  else
+    vm_name=$1
+
+    printf 'Grabing VM options for VM %s...\n' "$vm_name"
+
+    options=$(
+      nix eval \
+        --impure --raw --expr "
+          builtins.toJSON (builtins.getFlake (builtins.toString ./.)).vmOptions.$vm_name
+        " \
+      --log-format raw --quiet
+    )
+
+    proxmox=$(echo "$options" | jq -r .proxmox)
+    vm_id=$(echo "$options" | jq -r .vmId)
+
+    if [ "$proxmox" != fediversity ]; then
+      die "I do not know how to remove things that are not Fediversity VMs,
+  but I got proxmox = '%s' for VM %s." "$proxmox" "$vm_name"
+    fi
+
+    printf 'done grabing VM options for VM %s. Got %d.\n' "$vm_name" "$vm_id"
+  fi
+}
+
 ################################################################################
 ## Stop VM
 
 stop_vm () {
-  printf 'Stopping VM %d...\n' "$1"
+  printf 'Stopping VM %s...\n' "$vm_name"
 
-  proxmox_sync POST "$api_url/nodes/$node/qemu/$1/status/stop" \
+  proxmox_sync POST "$api_url/nodes/$node/qemu/$vm_id/status/stop" \
     'overrule-shutdown'==1
 
-  printf 'done stopping VM %d.\n' "$1"
+  printf 'done stopping VM %s.\n' "$vm_name"
 }
 
 ################################################################################
 ## Delete VM
 
 delete_vm () {
-  printf 'Deleting VM %d...\n' "$1"
+  printf 'Deleting VM %s...\n' "$vm_name"
 
-  proxmox_sync DELETE "$api_url/nodes/$node/qemu/$1" \
+  proxmox_sync DELETE "$api_url/nodes/$node/qemu/$vm_id" \
     'destroy-unreferenced-disks'==1 \
     'purge'==1
 
-  printf 'done deleting VM %d.\n' "$1"
+  printf 'done deleting VM %s.\n' "$vm_name"
 }
 
 ################################################################################
 ## Main loop
 
-printf 'Removing VMs%s...\n' "$vm_ids"
+printf 'Removing VMs%s...\n' "$vm_ids_or_names"
 
-remove_vm () {
-  stop_vm "$1"
-  delete_vm "$1"
-}
+remove_vm () (
+  grab_vm_options "$1"
+  stop_vm
+  delete_vm
+)
 
-for vm_id in $vm_ids; do
-  vm_id=${vm_id%:*}
-  remove_vm "$vm_id" &
+for vm_id_or_name in $vm_ids_or_names; do
+  remove_vm "$vm_id_or_name" &
 done
 wait
 
-printf 'done removing VMs%s.\n' "$vm_ids"
+printf 'done removing VMs%s.\n' "$vm_ids_or_names"
 
 ################################################################################
 ## Cleanup