#!/usr/bin/env bash
#   Copyright (C) 2024 John Törnblom
#
# This file is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING. If not see
# <http://www.gnu.org/licenses/>.

SCRIPT_PATH="${BASH_SOURCE[0]}"
SCRIPT_NAME="$(basename "${SCRIPT_PATH}")"
SCRIPT_DIR="$(dirname "${SCRIPT_PATH}")"

PS5_PAYLOAD_SDK="$(realpath "${SCRIPT_DIR}/..")"
PS5_PAYLOAD_HBROOT="${PS5_PAYLOAD_SDK}/target/user/homebrew"

LIBS_CRT="${PS5_PAYLOAD_SDK}/target/lib/crt1.o"
LIBS_C="-lc"
LIBS_CXX="-lunwind -lc++abi -lc++"
LIBS_DEPS="-lSceLibcInternal -lSceNet"
LIBS_KERN="-lkernel_web"
TU=""

for ARG in "$@"; do
    if [[ "$ARG" == "-nostdlib" ||
          "$ARG" == "-nodefaultlibs" ]]; then
	LIBS_C=""
	LIBS_CXX=""
	LIBS_DEPS=""
	LIBS_KERN=""
    fi
    if [[ "$ARG" == "-nostdlib++" ]]; then
	LIBS_CXX=""
    fi
    if [[ "$ARG" == "-lkernel" ||
          "$ARG" == "-lkernel_sys" ||
          "$ARG" == "-lkernel_web" ]]; then
	LIBS_KERN=""
    fi
    if [[ "$ARG" == "-nostartfiles" ||
	  "$ARG" == "-c" ||
	  "$ARG" == "-shared" ]]; then
	LIBS_CRT=""
    fi
    if [[ ! "${ARG:0:1}" == "-" ]]; then
	TU="${ARG}"
    fi
done
if [[ "${TU}" == "" ]]; then
    LIBS_CRT=""
    LIBS_C=""
    LIBS_CXX=""
    LIBS_DEPS=""
    LIBS_KERN=""
fi

# clang++-20 started appending crt objects when linking
if [[ $($SCRIPT_DIR/prospero-llvm-config --version | cut -d. -f1) -ge 20 ]]; then
    export SCE_PROSPERO_SDK_DIR="${PS5_PAYLOAD_SDK}/"
    LIBS_CRT=""
fi

# ensure clang++ knows where prospero-lld is at
export PATH="${SCRIPT_DIR}:${PATH}"

exec -a "${SCRIPT_NAME}" "${SCRIPT_DIR}/clang++" \
     --start-no-unused-arguments \
     -target x86_64-sie-ps5 \
     -fvisibility-nodllstorageclass=default \
     -isysroot "${PS5_PAYLOAD_SDK}" \
     -stdlib++-isystem "${PS5_PAYLOAD_SDK}/target/include/c++/v1" \
     -L "${PS5_PAYLOAD_SDK}/target/lib" \
     -L "${PS5_PAYLOAD_HBROOT}/lib" \
     -fno-stack-protector -fno-plt -femulated-tls \
     -frtti -fexceptions \
     $LIBS_CXX $LIBS_C \
     --end-no-unused-arguments \
     "$@" \
     --start-no-unused-arguments \
     --sysroot "${PS5_PAYLOAD_SDK}" \
     $LIBS_CRT $LIBS_KERN $LIBS_DEPS \
     --end-no-unused-arguments
