#!/bin/bash

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

NODE=$1
DISK=$2

echo "Resetting ${DISK} on ${NODE}"

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

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

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

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

kubectl delete pod disk-wipe-${NODE}