#!/bin/sh -eux

. /etc/os-release

if [ "$(id -u)" -eq 0 ]; then
        alias sudo=
fi

install_ximea() {
        filename=XIMEA.tgz
        curl -L "$XIMEA_DOWNLOAD_URL" -o "$filename"
        tar xzf $filename
        cd package
        sudo ./install -noudev
}

install_gpujpeg() {(
        curl -LO https://github.com/CESNET/GPUJPEG/releases/download/\
continuous/GPUJPEG-Linux.tar.xz
        tar xaf GPUJPEG-Linux.tar.xz
        sudo cp -r GPUJPEG/* /usr/local/
        sudo ldconfig
)}

# Install NDI
install_ndi() {(
        tar -xzf Install_NDI_SDK_Linux.tar.gz
        # shellcheck disable=SC2125
        installer=./Install*NDI*sh
        yes | PAGER="cat" $installer
        sudo cp -r NDI\ SDK\ for\ Linux/include/* /usr/local/include/
)}

# TODO: currently needed for Debian 11, which is used for ARM builds
#       remove when not needed
install_pipewire() {(
        if { [ "$ID" = ubuntu ] && [ "$VERSION_ID" = 20.04 ]; } ||
                { [ "${ID_LIKE-$ID}" = debian ] && [ "$VERSION_ID" -le 11 ]; }
        then
                sudo apt -y install libdbus-1-dev meson
                git clone https://github.com/PipeWire/pipewire
                cd pipewire
                git checkout 19bcdaebe29b95edae2b285781dab1cc841be638 # last one supporting meson 0.53.2 in U20.04
                ./autogen.sh -Dtests=disabled
                make -j "$(nproc)"
                sudo make install
        else
                sudo apt -y install libpipewire-0.3-dev
        fi
)}

install_rav1e() {(
        # TODO: use avx2 later
        if expr "${UG_ARCH-}" : '.*avx' >/dev/null; then
                avx2=avx2
        fi
        fpattern="librav1e.*linux-${avx2-sse4}.tar.gz"
        "${GITHUB_WORKSPACE-.}"/.github/scripts/download-gh-asset.sh xiph/rav1e \
                "$fpattern" librav1e.tar.gz
        sudo tar xaf librav1e.tar.gz -C /usr/local
        sudo rm -rf /usr/local/lib/librav1e.so*
        sudo sed -i -e 's-prefix=dist-prefix=/usr/local-' \
                -e 's/-lrav1e/-lrav1e -lm -pthread/' \
                /usr/local/lib/pkgconfig/rav1e.pc
)}

install_svt_jpegxs() {(
        git clone --depth 1 https://github.com/OpenVisualCloud/SVT-JPEG-XS
        # when built in U22.04, the stack is set as executable for some reason,
        # not in Arch, eg.
        export LDFLAGS="-Wl,-z,noexecstack"
        cmake -B SVT-JPEG-XS/build SVT-JPEG-XS
        cmake --build SVT-JPEG-XS/build --parallel "$(nproc)"
        sudo cmake --install SVT-JPEG-XS/build
)}

# FFmpeg master needs at least v1.3.277 as for 6th Mar '25
install_vulkan() {(
        sudo apt build-dep libvulkan1
        git clone --depth 1 https://github.com/KhronosGroup/Vulkan-Headers
        mkdir Vulkan-Headers/build
        cd Vulkan-Headers/build
        cmake ..
        sudo make install
        cd ../..
        git clone --depth 1 https://github.com/KhronosGroup/Vulkan-Loader
        mkdir Vulkan-Loader/build
        cd Vulkan-Loader/build
        cmake ..
        cmake --build . --parallel "$(nproc)"
        sudo make install
)}

show_help=
if [ $# -eq 1 ] && { [ "$1" = -h ] || [ "$1" = --help ] || [ "$1" = help ]; }; then
        show_help=1
fi

if [ $# -eq 0 ] || [ $show_help ]; then
        set -- gpujpeg ndi pipewire rav1e svt_jpegxs vulkan ximea
fi

if [ $show_help ]; then
        set +x
        printf "Usage:\n"
        printf "\t%s [<features>] | [ -h | --help | help ]\n" "$0"
        printf "\nInstall all aditional dependencies (without arguments) or install one explicitly.\n"
        printf "\nAvailable ones: %s%s%s\n" "$(tput bold)" "$*" "$(tput sgr0)"
        return 0
fi

cd /var/tmp

while [ $# -gt 0 ]; do
        install_"$1"
        shift
done