#!/bin/sh

VGA=`lspci |grep -i vga `
X_DRIVER=`grep "Driver" /etc/X11/xorg.conf | grep -a "fglrx\|nvidia"`
KERNEL_VER=`uname -r`
#LIVE_CD=`zgrep "Package: xorg-driver-fglrx\|Package: nvidia-glx" /cdrom/dists/stable/main/binary-i386/Packages.gz`
DRIVERS_IN_LIVE_CD=`grep "Package: xorg-driver-fglrx\|Package: nvidia-glx" /cdrom/dists/stable/main/binary-i386/Packages`

add_cdrom () {
    if [ -z "$DRIVERS_IN_LIVE_CD" ]; then
	apt-setup
    else
	apt-cdrom -d /cdrom -m add
    fi
}

enable_nvidia () {
#ENABLE_NVIDIA=`md5sum /etc/X11/xorg.conf | sudo tee /var/lib/xfree86/xorg.conf.md5sum`
    nvidia-glx-config enable
}

enable_ati () {
#ENABLE_ATI=`aticonfig --initial=dual-head`
    aticonfig --initial
}

install_nvidia () {
    if [ -z "$DRIVERS_IN_LIVE_CD" ]; then
        apt-get -m -y --force-yes install nvidia-glx
    else
        apt-get -m -y --force-yes -o "Acquire::cdrom::Mount=/cdrom" install nvidia-glx
    fi
}

install_ati () {
    if [ -z "$DRIVERS_IN_LIVE_CD" ]; then
        apt-get -m -y --force-yes install xorg-driver-fglrx
	apt-get -m -y --force-yes install fglrx-kernel-$KERNEL_VER
    else
        apt-get -m -y --force-yes -o "Acquire::cdrom::Mount=/cdrom" install xorg-driver-fglrx
	apt-get -m -y --force-yes -o "Acquire::cdrom::Mount=/cdrom" install fglrx-kernel-$KERNEL_VER
    fi
}
#NVIDIA_AVAILABLE=`apt-cache show nvidia-glx`
if [ -z "$X_DRIVER" ]; then
 if [ -n "`echo $VGA | grep -a -i -w ati`" ]; then
    echo "ati"
    add_cdrom
    install_ati
    enable_ati
 fi
 if [ -n "`echo $VGA | grep -a -i -w nvidia`" ]; then
    echo "NVidia"
    #if ! NVIDIA_AVAILABLE; then
    add_cdrom
    install_nvidia
    enable_nvidia
 fi
 exit 0
fi
exit 1

#sed s/Driver		"nv"/Driver		"vesa"/ /etc/X11/xorg.conf