class Kitsune::Kit::Commands::SetupSwap

def perform_rollback(ssh)
  script = <<~EOH
    set -e
    BACKUP_DIR="/usr/local/backups"
    MARKER_FILE="${BACKUP_DIR}/setup_swap.after"
    SWAPPINESS_BEFORE_FILE="${BACKUP_DIR}/setup_swap.swappiness.before"
    if [ ! -f "$MARKER_FILE" ]; then
      echo "💡 No swap marker found, skipping rollback."
      exit 0
    fi
    echo "🧹 Removing swap..."
    sudo swapoff /swapfile || true
    sudo rm -f /swapfile
    sudo sed -i '/\\/swapfile none swap sw 0 0/d' /etc/fstab
    sudo rm -f "$MARKER_FILE"
    if [ -f "$SWAPPINESS_BEFORE_FILE" ]; then
      ORIGINAL=$(cat "$SWAPPINESS_BEFORE_FILE")
      echo "🔁 Restoring vm.swappiness: $ORIGINAL"
      sudo sysctl vm.swappiness=$ORIGINAL
      sudo sed -i '/vm.swappiness=/d' /etc/sysctl.conf
      echo "vm.swappiness=$ORIGINAL" | sudo tee -a /etc/sysctl.conf
      sudo rm -f "$SWAPPINESS_BEFORE_FILE"
    else
      echo "💡 No swappiness backup found, skipping restore."
    fi
    echo "✅ Swap rollback completed"
    free -h
  EOH
  say ssh.exec!(script)
end