#!/bin/sh

# Copyright 2012-2017 Edward W. Koenig, Vancouver, WA, USA
# All rights reserved.
#
# Redistribution and use of this script, with or without modification, is
# permitted provided that the following conditions are met:
#
# 1. Redistributions of this script must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ''AS IS'' AND ANY EXPRESS OR IMPLIED
# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
# nvidia-switch utility 3.0
#
# A tool to install and cleanly remove the nvidia driver without stomping on the xorg files.
# Note: this tool "should" allow switching on the fly, but why? Pick a driver! You will still
# need to edit a bunch of other config files.

# TO-DO: needs more cleanup; still way too brittle.
# might be better to move mesa and xorg conflicts to /var/lib/nvidia 


ROOT="${ROOT:-/}"
CWD=$(pwd)
LIBSUFFIX="LIBDIRSUFFIX" # This will be replaced in the build script
COMPAT32="LIB32FLAG" # This will be replaced in the build script (yes | no)
INC="${ROOT}usr/include/GL"
LIB="${ROOT}usr/lib${LIBSUFFIX}"
LIB32="${ROOT}usr/lib"
XMOD="${LIB}/xorg/modules"
XLIB="$XMOD/extensions"
NV_VERSION="PKGVERSION" # This will be replaced in the build script
GL_VERSION="1.2.0"
GLVND="GLVNDFLAG" # This will be replaced in the build script (yes | no)
NV_CONF="/usr/share/X11/xorg.conf.d/10-nvidia.conf"

save_GL(){
  cd $2
  case "$1" in
    "nvidia")
      mv libGL.so.$GL_VERSION xorg-libGL.so.$GL_VERSION
      mv libGL.la xorg-libGL.la
      mv libGL.la-nvidia libGL.la
      mv libEGL.so.1.0.0 xorg-libEGL.so.1.0.0

      if [ "$GLVND" = "yes" ]; then
        ln -sf libGL.so.1.0.0 libGL.so.1
        ln -sf libGL.so.1 libGL.so
        ln -sf libEGL.so.1-nvidia libEGL.so.1
        ln -sf libEGL.so.1 libEGL.so
      else
        ln -sf libGL.so.$NV_VERSION libGL.so.1
        ln -sf libGL.so.1 libGL.so
        ln -sf libEGL.so.$NV_VERSION libEGL.so.1
        ln -sf libEGL.so.1 libEGL.so
      fi

      mv libGLESv1_CM.so.1.1.0 xorg-libGLESv1_CM.so.1.1.0
      ln -sf libGLESv1_CM.so.1-nvidia libGLESv1_CM.so.1
      ln -sf libGLESv1_CM.so.1 libGLESv1_CM.so
      mv libGLESv2.so.2.0.0 xorg-libGLESv2.so.2.0.0
      ln -sf libGLESv2.so.2-nvidia libGLESv2.so.2
      ln -sf libGLESv2.so.2 libGLESv2.so
    ;;

    "xorg")
       mv libGL.la libGL.la-nvidia
       mv xorg-libGL.la libGL.la
       mv xorg-libEGL.so.1.0.0 libEGL.so.1.0.0
       mv xorg-libGL.so.$GL_VERSION libGL.so.$GL_VERSION

       ln -sf libGL.so.$GL_VERSION libGL.so.1
       ln -sf libGL.so.$GL_VERSION libGL.so
       ln -sf libEGL.so.1.0.0 libEGL.so.1
       ln -sf libEGL.so.1.0.0 libEGL.so

       mv xorg-libGLESv1_CM.so.1.1.0 libGLESv1_CM.so.1.1.0
       ln -sf libGLESv1_CM.so.1.1.0 libGLESv1_CM.so.1
       ln -sf libGLESv1_CM.so.1.1.0 libGLESv1_CM.so
       mv xorg-libGLESv2.so.2.0.0 libGLESv2.so.2.0.0
       ln -sf libGLESv2.so.2.0.0 libGLESv2.so.2
       ln -sf libGLESv2.so.2.0.0 libGLESv2.so
  ;;
  *)
        echo "nothind to do for save_gl?"
        exit 1
  ;;
  esac
  cd "$CWD"
}

save_glx(){
  cd "$XLIB"
  case "$1" in
      "nvidia")
        if [ -e libglx.so ]; then
          mv libglx.so xorg-libglx.so
        fi
	  ln -sf libglx.so.$NV_VERSION libglx.so
      ;;
      "xorg")
        if [ -e xorg-libglx.so ]; then
          rm -f libglx.so
          mv xorg-libglx.so libglx.so
        else
          rm -f libglx.so
        fi
      ;;
        *)
          echo "nothind to do for save_glx?"
          exit 1
      ;;
  esac
  cd "$CWD"
}

save_wfb(){
  cd "$XMOD"
  if [ -e libwfb.so ]; then
    echo "Using existing Xorg libwfb..."
  else
    if [ "$1" = "nvidia" ]; then
	ln -sf libwfb.so.$NV_VERSION libwfb.so.1
	ln -sf libwfb.so.1 libwfb.so
    fi
    if [ "$1" = "xorg" ]; then
	rm -f libwfb.so
	rm -f libwfb.so.1
    fi
  fi
  cd "$CWD"
}

nvidia_install(){
  echo $'Installing to nvidia-driver files!\n'
  echo "Make sure the nvidia driver is ENABLED in /etc/X11/xorg.conf"
  echo "and in /etc/X11/xorg.conf.d."
  echo "Otherwise, this may lead to improperly working drivers."

  save_glx "nvidia"
  save_GL "nvidia" $LIB
  save_wfb "nvidia"
# Check for multilib configuration
  if [ "$COMPAT32" = "yes" ]; then
    if [ -e $LIB32/libGL.so.$GL_VERSION ]; then
      save_GL "nvidia" $LIB32
    else
      echo "mesa-compat32 is not installed, skipping..."
    fi
  fi
}

nvidia_remove(){
  echo $'Returning to stock xorg files!\n'
  echo "Make sure the nvidia driver is DISABLED in /usr/share/X11/xorg.conf.d"
  echo "/etc/X11/xorg.conf and in /etc/X11/xorg.conf.d."
  echo "Otherwise, this may lead to improperly working drivers."

  save_glx "xorg"
  save_GL "xorg" $LIB
  save_wfb "xorg"
# remove dangling sylinks
  rm -f $LIB/libnvcuvid.so.1
  rm -f $LIB/libnvidia-cfg.so.1
  rm -f $LIB/libnvidia-encode.so.1
  rm -f $LIB/libnvidia-fbc.so.1
  rm -f $LIB/libnvidia-ifr.so.1
  rm -f $LIB/libnvidia-ml.so.1
  rm -f $LIB/libnvidia-opencl.so.1
  rm -f $LIB/libcuda.so.1
  rm -f $LIB/libOpenCL.so.1
# Check for multilib configuration
  if [ "$COMPAT32" = "yes" ]; then
    if [ -e $LIB32/xorg-libGL.so.$GL_VERSION ]; then
      save_GL "xorg" $LIB32
    else
      echo "mesa-compat32 is not installed, skipping..."
    fi
# remove dangling sylinks
    rm -f $LIB32/libnvcuvid.so.1
    rm -f $LIB32/libnvidia-encode.so.1
    rm -f $LIB32/libnvidia-fbc.so.1
    rm -f $LIB32/libnvidia-ifr.so.1
    rm -f $LIB32/libnvidia-ml.so.1
    rm -f $LIB32/libnvidia-opencl.so.1
    rm -f $LIB32/libcuda.so.1
    rm -f $LIB32/libOpenCL.so.1
  fi

  if [ -e $NV_CONF ]; then
    echo "removing $NV_CONF: you will need to provide your own"
    echo "or reinstall nvidia-driver."
    rm -f $NV_CONF
  fi
}

restore(){
# Attempt to reinstall damaged stock packages if something went wrong.
# Obviously, you will need to run this after '--remove' and before
# 'removepkg nvidia-driver'.
# User must:
# 1. remove any dangling symlinks and xorg-*, *-nvidia files in /usr/lib*/,
#    and any other housekeeping tasks.
# 2. copy stock Slackware packages to /root/Slackware
# 3. be running STOCK Slackware-14.2
# 4. realize that COMPAT32 is not supported ;-)

if [ "$LIBSUFFIX" = "64" ]; then
  ARCH="x86_64"
else
  ARCH="x86"
fi

PKGPATH="/root/Slackware"
MESA="mesa-11.2.2-$ARCH-1.txz"
MESA32="mesa-compat32-11.2.2-$ARCH-1compat32.txz"
XORG="xorg-server-1.18.3-$ARCH-2.txz"
# Note: The above is for stock 14.2; modify as needed for /patches or
# multilib updates
if [ -d $PKGPATH ]; then
  if [ `find $PKGPATH -prune -empty -type d` ]; then
    echo "Your $PKGPATH is empty!"
  else
    upgradepkg --reinstall $MESA $XORG
    if [ "COMPAT32" = "yes" ]; then
      upgradepkg --reinstall $MESA32
    fi
  fi
else
  echo "Please create $PKGPATH containing the appropriate packages"
fi
}

usage(){
  echo "Usage:"
  echo " --install		Set up nvidia driver files (only during pkg install)"
  echo " --remove		Return to stock xorg files and remove all symlinks"
  echo " --restore		Reinstall stock Mesa and xorg-server [EXPERIMENTAL]"
  echo " --help			Show this help message"
}

case "$1" in
  "--install")
    nvidia_install
    ;;
  "--remove")
    nvidia_remove
    ;;
  "--restore")
    restore
    ;;
  "--help")
    usage
    ;;
  *)
    usage
    ;;
esac