#! /bin/sh -e
SCRATCH=scratch/$(basename "$0" .sh)

rm -rf "${SCRATCH}"
mkdir -p "${SCRATCH}"
mkdir -p "${SCRATCH}/libsA"
mkdir -p "${SCRATCH}/libsB"

cp main "${SCRATCH}"/
cp libfoo.so libbar.so "${SCRATCH}/libsA/"
cp libfoo.so libbar.so "${SCRATCH}/libsB/"

oldRPath=$(../src/patchelf --print-rpath "${SCRATCH}/main")
if test -z "$oldRPath"; then oldRPath="/oops"; fi
pathA="$(pwd)/${SCRATCH}/libsA"
pathB="$(pwd)/${SCRATCH}/libsB"
../src/patchelf --force-rpath --set-rpath "$oldRPath:$pathA:$pathB" "${SCRATCH}/main"

cp "${SCRATCH}"/main "${SCRATCH}/mainA"
cp "${SCRATCH}"/main "${SCRATCH}/mainB"

../src/patchelf --shrink-rpath "${SCRATCH}/main"
../src/patchelf --shrink-rpath --allowed-rpath-prefixes "$oldRPath:$pathA" "${SCRATCH}/mainA"
../src/patchelf --shrink-rpath --allowed-rpath-prefixes "$oldRPath:$pathB" "${SCRATCH}/mainB"

check() {
    exe=$1
    mustContain=$2
    mustNotContain=$3

    rpath=$(../src/patchelf --print-rpath "$exe")
    echo "RPATH of $exe after: $rpath"

    if ! echo "$rpath" | grep -q "$mustContain"; then
        echo "RPATH didn't contain '$mustContain' when it should have"
        exit 1
    fi

    if echo "$rpath" | grep -q "$mustNotContain"; then
        echo "RPATH contained '$mustNotContain' when it shouldn't have"
        exit 1
    fi
}

check "${SCRATCH}/main"  "$pathA" "$pathB"
check "${SCRATCH}/mainA" "$pathA" "$pathB"
check "${SCRATCH}/mainB" "$pathB" "$pathA"