#!/bin/bash

debian_sources="/etc/apt/sources.list.d/debian.sources"
sources_list="/etc/apt/sources.list"

add_components() {
    local file="$1"
    if grep -q "contrib" "$file" && grep -q "non-free" "$file" && grep -q "non-free-firmware" "$file"; then
        echo ""
    else
       sed -i 's/main/main contrib non-free non-free-firmware/' "$file"
    fi
}

if [ -f "$debian_sources" ]; then
    sudo cp /etc/apt/sources.list.d/debian.sources debian.sources.bak
    add_components "$debian_sources"
elif [ -f "$sources_list" ]; then
    sudo cp /etc/apt/sources.list sources.list.bak
    add_components "$sources_list"
else
    echo ""
fi

sudo apt update
sudo apt install -y nvidia-driver firmware-misc-nonfree