#!/bin/sh

# Copyright 2012-2019 Edward W. Koenig, Vancouver, WA, USA
# Copyright 2020 Lenard Spencer, Orlando, FL, 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 4.7.2
#
# A tool to install and cleanly remove the nvidia driver without stomping
# on the mesa files.  Note: this tool "should" allow switching on the fly,
# but why? Pick a driver! You will stilll need to edit a bunch of other
# config files.

# When ROOT=<PATH> is specified when using {install,upgrade,remove}pkg,
# PATH is usually specified without a trailing "/", so we need to add it
if [ "$ROOT" = "" ]; then
  ROOT="${ROOT:-"/"}"
else
  ROOT="${ROOT}/"
fi
CWD=$(pwd)
COMPAT32="LIB32FLAG" # This will be replaced in the build script (yes | no)
LIBSUFFIX="LIBDIRSUFFIX" # This will be replaced in the build script
INC="${ROOT}usr/include/GL"
LIB="${ROOT}usr/lib${LIBSUFFIX}"
LIB32="${ROOT}usr/lib"

save_GL(){
# backup original mesa conflicts
  cd $2
  case "$1" in
    "nvidia")
      cp libGL{,ES*}.la /var/log/nvidia/$3
      mv libGL{.so.1.2.0,ESv1_CM.so.1.1.0,ESv2.so.2.0.0} /var/log/nvidia/$3
      sed -i s/1.2.0/1.7.0/g libGL.la
      sed -i s/1.1.0/1.2.0/g libGLESv1_CM.la
      sed -i s/2.0.0/2.1.0/g libGLESv2.la
    ;;

    "xorg")
      mv /var/log/nvidia/$3/* .
 ;;

  *)
        echo "nothing to do for save_gl?"
        exit 1
  ;;
  esac
  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/or in either /usr/share/X11/xorg.conf.d or"
  echo "/etc/X11/xorg.conf.d.  Otherwise, this may lead to improperly"
  echo  -e "working drivers.\n"

  if [ "$LIBSUFFIX" = "64" ]; then
    save_GL "nvidia" $LIB "64"
    # Check for multilib configuration
    if [ "$COMPAT32" = "yes" ]; then
      save_GL "nvidia" $LIB32 "32"
    fi
  else
    save_GL "nvidia" $LIB "32"
  fi
}

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

  if [ "$LIBSUFFIX" = "64" ]; then
    save_GL "xorg" $LIB "64"
    # Check for multilib configuration
    if [ "$COMPAT32" = "yes" ]; then
      save_GL "xorg" $LIB32 "32"
    fi
  else
    save_GL "xorg" $LIB "32"
  fi
}

usage(){
  echo "Usage:"
  echo " --install    Set up nvidia driver files and update symlinks"
  echo " --remove     Return to stock xorg files and restore symlinks"
  echo " --help	      Show this help message"
}

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