#!/bin/bash

vdir=`dirname $0`
. $vdir/vasm-functions
check_root

if [ $DISPLAY ]; then
  DIALOG=Xdialog
else
  DIALOG=dialog
fi
kernel=`uname -r`
if ( lspci | grep -i vga | grep -iq " ati " ) then
       if ( ls /var/log/packages | grep -i fglrx ) then
       removepkg ati* >/dev/null
    fi
     j="0"
     for i in $drivers; do
         driv[$j]="$i"
         j=`expr $j + 1`
     done

 if [ ! "x$driv[0]" = "x" ];then
  $DCMD --backtitle "$BACKTITLE" --title "Driver Selection" --menu \
   "     Which Driver would you like to install?\n
    Choose the Driver you would like to use!" 20 71 4\
    "FGLRX" "This is the Proprietary fglrx driver" \
    "radeon" "This is the opensource radeon driver" \
    "ati" "This is the opensource ati driver" \
    "Nvidia-legacy" "This is the Proprietary Legacy driver for older cards" \
    "Nvidia-latest" "This is the newest Proprietary driver for newer cards" \
    "nv" "This is the opensource nvidia driver" \
    2> $freply
     result=$?
     if [ $result != 0 ];then
       exit 0
     fi
     DRIVER="`cat $freply`"
     #echo $DRIVER
     case "$DRIVER" in
      "ati")
        #cat /etc/X11/xorg.conf.ati > $HOME/xorg.conf.new
       exit 23
       ;;
      "radeon") 
         sed s'/\"ati\"/\"radeon\"/'g /etc/X11/xorg.conf > /etc/X11/xorg.conf.new && \
         mv /etc/X11/xorg.conf.new /etc/X11/xorg.conf
         exit 0
       ;;
      "FGLRX")
       /sbin/installpkg /usr/src/vl-pkgs/fglrx*$kernel*tlz
       result=$?
       if [ $result = 0 ];then
         sed s'/\"ati\"/\"fglrx\"/'g /etc/X11/xorg.conf > /etc/X11/xorg.conf.new && \
         mv /etc/X11/xorg.conf.new /etc/X11/xorg.conf
         depmod -aq
         exit 0
       fi
       exit 1
       ;;
      esac
  fi
kernel=`uname -r`
if ( lspci | grep -i vga | grep -iq "geforce " ) then
       if ( ls /var/log/packages | grep -i nvidia ) then
         removepkg nvidia* >/dev/null
       fi
       /sbin/installpkg /usr/src/vl-pkgs/nvidia-9755*$kernel*tlz
       result=$?
       if [ $result = 0 ];then
         sed s'/vesa/nvidia/'g /etc/X11/xorg.conf > /etc/X11/xorg.conf.new && \
         mv /etc/X11/xorg.conf.new /etc/X11/xorg.conf
         depmod -aq
         exit 0
       fi
       exit 1 
else
       if ( ls /var/log/packages | grep -i nvidia ) then
         removepkg nvidia* >/dev/null
       fi
       /sbin/installpkg /usr/src/vl-pkgs/nvidia-7184*$kernel*tlz
       result=$?
       if [ $result = 0 ];then
         sed s'/vesa/nvidia/'g /etc/X11/xorg.conf > /etc/X11/xorg.conf.new && \
         mv /etc/X11/xorg.conf.new /etc/X11/xorg.conf
         depmod -aq
         exit 0
       fi
       exit 1
fi