#!/bin/bash

if [ $# -ne 1 ]; then
  echo "Incorrect number of arguments"
fi

NODE=$1

echo "Resetting ${NODE}"

NODEIP=$(kubectl get nodes -o wide | grep ${NODE} | grep -v "INTERNAL-IP" | awk '{print $6}')
DISKS=$(talosctl -n ${NODEIP} disks | awk '{print $2}' | grep -v DEV | grep -v "/dev/sdg" | grep -v "/dev/mmc")

echo "Disks: $DISKS"
echo "Node IP: $NODEIP"

for d in $DISKS ; do

echo "Creating disk-wipe pod to clear $d on $NODE (${NODEIP})"
cat <<EOF | kubectl apply -f -
apiVersion: v1
kind: Pod
metadata:
  name: disk-wipe
spec:
  restartPolicy: Never
  nodeName: ${NODE}
  containers:
  - name: disk-wipe
    image: busybox
    securityContext:
      privileged: true
    command: ["/bin/sh", "-c", "dd if=/dev/zero bs=1M count=100 oflag=direct of=${d}"]
EOF

kubectl wait --timeout=900s --for=jsonpath='{.status.phase}=Succeeded' pod disk-wipe

kubectl delete pod disk-wipe

done

talosctl -n ${NODEIP} reboot