#!/bin/bash

# File containing allowed drivers (from safe mode)
SAFE_DRIVERS="/home/$(whoami)/safe_mode_drivers.txt"
BLACKLIST_FILE="/etc/modprobe.d/blacklist-custom.conf"

# Verify safe drivers file exists
if [ ! -f "$SAFE_DRIVERS" ]; then
echo "Error: $SAFE_DRIVERS not found!"
exit 1
fi

# Get all kernel modules for current kernel
MODULE_DIR="/lib/modules/$(uname -r)"
if [ ! -d "$MODULE_DIR" ]; then
echo "Error: Kernel modules directory $MODULE_DIR not found!"
exit 1
fi

# Clear old blacklist
echo "# Auto-generated blacklist" | sudo tee "$BLACKLIST_FILE"

# Find all .ko files and process them
find "$MODULE_DIR" -name '.ko' -type f | while read -r module_path; do
module_name=$(basename "$module_path" | cut -d. -f1)
if ! grep -q "^$module_name$" "$SAFE_DRIVERS"; then
echo "blacklist $module_name" | sudo tee -a "$BLACKLIST_FILE"
fi
done

echo "Blacklist updated. Running update-initramfs..."
sudo update-initramfs -u