#!/bin/sh -eux

. /etc/os-release

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

install_ximea() {
        if [ "$(dpkg --print-architecture | cut -b1-3)" = arm ]; then
                filename=XIMEA_Linux_ARM_SP.tgz
        else
                filename=XIMEA_Linux_SP.tgz
        fi
        curl -LO https://www.ximea.com/downloads/recent/$filename
        tar xzf $filename
        cd package
        touch bin/streamViewer.64 # TOREMOVE
        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: needed only for U20.04, remove after upgrading to U22.04
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
)}

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 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