#!/bin/bash

if [[ $1 == "Debian" ]]; then
    if cat /etc/apt/sources.list | grep -E "sid|unstable";  then
        if [[ $3 == "xorg" ]]; then
            echo "apt-get -y install xorg"
        fi
        if [[ $3 == "wayland" ]]; then
            echo "Error: Wayland not available on Debian"
            exit 2
        fi
        if [[ $2 == "vm" ]]; then
            echo "Skipping: No further action required"
        fi
        if [[ $2 == "nvidia" ]]; then
            echo "apt-get -y install nvidia-driver firmware-misc-nonfree"
        fi
    else
        if [[$2 == "nvidia" ]]; then
            echo "apt-get -y install nvidia-driver firmware-misc-nonfree"
        fi
    fi
fi

if [[ $1 == "Arch" ]]; then
    echo "gpu setup"
fi