#!/bin/bash

# Slackware build script for gpt4all

# rizitis 30/10/2023 <rizitis@gmail.com>
# updated 21/04/2024
# You are free to maintain it in the SBo way and simply remove my name, no copyrights ;)
#
# Current maintaining from Condor 16/08/2024 <condor _AT_ stz-bg _DOT_ com> until author return
#
# 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.


cd $(dirname $0) ; CWD=$(pwd)

CHECKOUT=228d5379cfb54e966449e153082c74c66b27c6c9
PRGNAM=gpt4all
VERSION=${VERSION:-3.10.0}
BUILD=${BUILD:-1}
TAG=${TAG:-_condor}
PKGTYPE=${PKGTYPE:-txz}
JOBS=-j$(getconf _NPROCESSORS_ONLN)

if [ -z "$ARCH" ]; then
  case "$( uname -m )" in
    i?86) ARCH=i586 ;;
    arm*) ARCH=arm ;;
       *) ARCH=$( uname -m ) ;;
  esac
fi


if [ ! -z "${PRINT_PACKAGE_NAME}" ]; then
  echo "$PRGNAM-$VERSION-$ARCH-$BUILD$TAG.$PKGTYPE"
  exit 0
fi

TMP=${TMP:-/tmp/rtz}
PKG=$TMP/package-$PRGNAM
OUTPUT=${OUTPUT:-/tmp}

if [ "$ARCH" = "i586" ]; then
  SLKCFLAGS="-O2 -march=i586 -mtune=i686"
  LIBDIRSUFFIX=""
elif [ "$ARCH" = "i686" ]; then
  SLKCFLAGS="-O2 -march=i686 -mtune=i686"
  LIBDIRSUFFIX=""
elif [ "$ARCH" = "x86_64" ]; then
  SLKCFLAGS="-O2 -fPIC"
  LIBDIRSUFFIX="64"
else
  SLKCFLAGS="-O2"
  LIBDIRSUFFIX=""
fi

set -e 

rm -rf $PKG
mkdir -p $TMP $PKG $OUTPUT
cd $TMP
rm -rf $PRGNAM

git clone https://github.com/nomic-ai/gpt4all
cd $PRGNAM/gpt4all-chat
git checkout "$CHECKOUT"
git submodule deinit -f --all
git submodule update --init --recursive
#sed -i '/CMAKE_INSTALL_PREFIX/d' CMakeLists.txt

chown -R root:root .
find -L . \
 \( -perm 777 -o -perm 775 -o -perm 750 -o -perm 711 -o -perm 555 \
  -o -perm 511 \) -exec chmod 755 {} \; -o \
 \( -perm 666 -o -perm 664 -o -perm 640 -o -perm 600 -o -perm 444 \
  -o -perm 440 -o -perm 400 \) -exec chmod 644 {} \;

#( cd $TMP/$PRGNAM/gpt4all-backend/deps/llama.cpp-mainline/ggml/src && cp $CWD/ggml_vk_generate_shaders.py . && python3 ggml_vk_generate_shaders.py )

read -p "The project depency specailly llama and kompute are 2 years old, don't expect something special"

# You can enable a lot of optional features by passing
# variables to the script (VAR=yes/no ./gpt4all.SlackBuild).
# for example: VULKAN=yes ROCM=yes BLAS=yes ./gpt4all.SlackBuild
# Vulkan not working btw

vulkan="-DGGML_VULKAN=OFF -DLLMODEL_VULKAN=OFF" ; [ "${VULKAN:-no}" != "no" ] && vulkan="-DGGML_VULKAN=ON -DLLMODEL_VULKAN=ON"
rocm="-DGGML_HIP_UMA=OFF -DLLMODEL_ROCM=OFF"    ; [ "${ROCM:-no}" != "no" ]   && rocm="-DGGML_HIP_UMA=ON -DLLMODEL_ROCM=ON"
cuda="-DGGML_CUDA=OFF -DLLMODEL_CUDA=OFF"       ; [ "${CUDA:-no}" != "no" ]   && cuda="-DGGML_CUDA=ON -DLLMODEL_CUDA=ON"
blas="-DGGML_BLAS=OFF -DLLAMA_BLAS=OFF"         ; [ "${BLAS:-no}" != "no" ]   && blas="-DGGML_BLAS=ON -DLLAMA_BLAS=ON"

mkdir -p build
cd build
 cmake \
    -Wno-dev \
    -DCMAKE_C_FLAGS:STRING="$SLKCFLAGS" \
    -DCMAKE_CXX_FLAGS:STRING="$SLKCFLAGS" \
    -DCMAKE_INSTALL_PREFIX=/opt/GPT4All \
    -DLIB_SUFFIX=${LIBDIRSUFFIX} \
    $blas \
    $rocm \
    $cuda \
    $vulkan \
    -DKOMPUTE_OPT_BUILD_SHADERS=ON \
    -DKOMPUTE_OPT_DISABLE_VULKAN_VERSION_CHECK=ON \
    -DKOMPUTE_OPT_USE_BUILT_IN_FMT=OFF \
    -DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=OFF \
    -DKOMPUTE_OPT_USE_BUILT_IN_SPDLOG=OFF \
    -DLLMODEL_KOMPUTE=ON \
    -DCMAKE_BUILD_TYPE=Release ..
cmake --build . $JOBS
make install/strip DESTDIR=$PKG

cd $TMP/$PRGNAM

rm -r $PKG/opt/GPT4All/bin
cp -R gpt4all-chat/build/bin $PKG/opt/GPT4All/
mv gpt4all-* $PKG/opt/GPT4All/


# Don't ship .la files:
rm -f $PKG/{,usr/}lib${LIBDIRSUFFIX}/*.la

# Strip binaries and libraries - this can be done with 'make install-strip'
# in many source trees, and that's usually acceptable, if not, use this:
find $PKG -print0 | xargs -0 file | grep -e "executable" -e "shared object" | grep ELF \
  | cut -f 1 -d : | xargs strip --strip-unneeded 2> /dev/null || true
find $PKG -name perllocal.pod -o -name ".packlist" -o -name "*.bs" | xargs rm -f || true

cd $TMP/$PRGNAM
mkdir -p $PKG/usr/doc/$PRGNAM-$VERSION
cp -a LICENSE* README* \
  $PKG/usr/doc/$PRGNAM-$VERSION
cat $CWD/$PRGNAM.SlackBuild > $PKG/usr/doc/$PRGNAM-$VERSION/$PRGNAM.SlackBuild


mkdir -p $PKG/usr/share/applications/
cat $CWD/GPT4All.desktop > $PKG/usr/share/applications/GPT4All.desktop
mkdir -p $PKG/usr/share/GPT4All/icons/{32x32,48x48}
cp $CWD/logo-32.png $PKG/usr/share/GPT4All/icons/32x32/
cp $CWD/logo-48.png $PKG/usr/share/GPT4All/icons/48x48/

mkdir -p $PKG/install
cat $CWD/slack-desc > $PKG/install/slack-desc
cat $CWD/doinst.sh > $PKG/install/doinst.sh
cat $CWD/slack-required > $PKG/install/slack-required

cd $PKG
/sbin/makepkg -l n -c n $OUTPUT/$PRGNAM-$VERSION-$ARCH-$BUILD$TAG.$PKGTYPE