install.sh

#!/bin/bash

set -e

PREFIXES="${NUGETURIPREFIXES:-"https://pkgs.dev.azure.com/"}"
USENET6="${DOTNET6:-"false"}"
WRAPPERTYPE="${WRAPPERTYPE:-"SHELL_FUNCTION"}"
ALIAS_DOTNET="${DOTNETALIAS:-"true"}"
ALIAS_NUGET="${NUGETALIAS:-"true"}"
ALIAS_NPM="${NPMALIAS:-"true"}"
ALIAS_YARN="${YARNALIAS:-"true"}"
ALIAS_NPX="${NPXALIAS:-"true"}"
ALIAS_RUSH="${RUSHALIAS:-"true"}"
ALIAS_PNPM="${PNPMALIAS:-"true"}"
INSTALL_PIP_HELPER="${PYTHON:-"false"}"
COMMA_SEP_TARGET_FILES="${TARGETFILES:-"DEFAULT"}"

# Destination to install wrappers for the `EXECUTABLE` wrapper type.
# This path must take precedence over /usr/local/bin and the original tools,
# which is ensured via `containerEnv.PATH` in devcontainer-feature.json.
EXECUTABLES_TARGET_DIR='/usr/local/share/codespace-features'

WRAPPED_BINS_ARR=()

if [ "${ALIAS_DOTNET}" = "true" ]; then
    WRAPPED_BINS_ARR+=('dotnet')
fi
if [ "${ALIAS_NUGET}" = "true" ]; then
    WRAPPED_BINS_ARR+=('nuget')
fi
if [ "${ALIAS_NPM}" = "true" ]; then
    WRAPPED_BINS_ARR+=('npm')
fi
if [ "${ALIAS_YARN}" = "true" ]; then
    WRAPPED_BINS_ARR+=('yarn')
fi
if [ "${ALIAS_NPX}" = "true" ]; then
    WRAPPED_BINS_ARR+=('npx')
fi
if [ "${ALIAS_RUSH}" = "true" ]; then
    WRAPPED_BINS_ARR+=('rush')
    WRAPPED_BINS_ARR+=('rush-pnpm')
fi
if [ "${ALIAS_PNPM}" = "true" ]; then
    WRAPPED_BINS_ARR+=('pnpm')
    WRAPPED_BINS_ARR+=('pnpx')
fi

# Source /etc/os-release to get OS info
. /etc/os-release

if [ "$(id -u)" -ne 0 ]; then
    echo -e 'Script must be run as root. Use sudo, su, or add "USER root" to your Dockerfile before running this script.'
    exit 1
fi

apt_get_update() {
    if [ "$(find /var/lib/apt/lists/* | wc -l)" = "0" ]; then
        echo "Running apt-get update..."
        apt-get update -y
    fi
}

# Checks if packages are installed and installs them if not
check_packages() {
    if ! dpkg -s "$@" > /dev/null 2>&1; then
        apt_get_update
        apt-get -y install --no-install-recommends "$@"
        rm -rf /var/lib/apt/lists/*
    fi
}

export DEBIAN_FRONTEND=noninteractive

if [ "${ID}" = "mariner" ]; then
    tdnf install -y wget ca-certificates
    tdnf clean all
else
    check_packages wget ca-certificates
fi

# Change to the directory where this script is located
cd "$(dirname "$0")"

cp ./scripts/install-provider.sh /tmp
chmod +rx /tmp/install-provider.sh
cp ./scripts/install-python-keyring.sh /tmp
chmod +rx /tmp/install-python-keyring.sh

sed "s|REPLACE_WITH_AZURE_DEVOPS_NUGET_FEED_URL_PREFIX|${PREFIXES}|g" ./scripts/run-dotnet.sh > /usr/local/bin/run-dotnet.sh
chmod +rx /usr/local/bin/run-dotnet.sh
sed "s|REPLACE_WITH_AZURE_DEVOPS_NUGET_FEED_URL_PREFIX|${PREFIXES}|g" ./scripts/run-nuget.sh > /usr/local/bin/run-nuget.sh
chmod +rx /usr/local/bin/run-nuget.sh
cp ./scripts/run-npm.sh /usr/local/bin/run-npm.sh
chmod +rx /usr/local/bin/run-npm.sh
cp ./scripts/run-yarn.sh /usr/local/bin/run-yarn.sh
chmod +rx /usr/local/bin/run-yarn.sh
cp ./scripts/write-npm.sh /usr/local/bin/write-npm.sh
chmod +rx /usr/local/bin/write-npm.sh
cp ./scripts/run-npx.sh /usr/local/bin/run-npx.sh
chmod +rx /usr/local/bin/run-npx.sh

cp ./scripts/run-rush.sh /usr/local/bin/run-rush.sh
chmod +rx /usr/local/bin/run-rush.sh
cp ./scripts/run-rush-pnpm.sh /usr/local/bin/run-rush-pnpm.sh
chmod +rx /usr/local/bin/run-rush-pnpm.sh

cp ./scripts/run-pnpm.sh /usr/local/bin/run-pnpm.sh
chmod +rx /usr/local/bin/run-pnpm.sh
cp ./scripts/run-pnpx.sh /usr/local/bin/run-pnpx.sh
chmod +rx /usr/local/bin/run-pnpx.sh

if [ "${INSTALL_PIP_HELPER}" = "true" ]; then
    USER="${_REMOTE_USER}" /tmp/install-python-keyring.sh
    rm /tmp/install-python-keyring.sh
fi

INSTALL_WITH_SUDO="false"
if command -v sudo >/dev/null 2>&1; then
    if [ "root" != "$_REMOTE_USER" ]; then
        INSTALL_WITH_SUDO="true"
    fi
fi

if [ "$WRAPPERTYPE" = "SHELL_FUNCTION" ]; then
    echo "Installing shell functions for wrapped commands."

    if [ "${COMMA_SEP_TARGET_FILES}" = "DEFAULT" ]; then
        if [ "${INSTALL_WITH_SUDO}" = "true" ]; then
            COMMA_SEP_TARGET_FILES="~/.bashrc,~/.zshrc"
        else
            COMMA_SEP_TARGET_FILES="/etc/bash.bashrc,/etc/zsh/zshrc"
        fi
    fi

    IFS=',' read -r -a TARGET_FILES_ARR <<< "$COMMA_SEP_TARGET_FILES"

    for ALIAS in "${WRAPPED_BINS_ARR[@]}"; do
        for TARGET_FILE in "${TARGET_FILES_ARR[@]}"; do
            CMD="$ALIAS() { /usr/local/bin/run-$ALIAS.sh \"\$@\"; }"

            if [ "${INSTALL_WITH_SUDO}" = "true" ]; then
                sudo -u ${_REMOTE_USER} bash -c "echo '$CMD' >> $TARGET_FILE"
            else
                echo $CMD >> $TARGET_FILE || true
            fi
        done
    done
elif [ "$WRAPPERTYPE" = "EXECUTABLE" ]; then
    echo "Installing executable scripts for wrapped commands."
    sudo mkdir -p "${EXECUTABLES_TARGET_DIR}"

    for ALIAS in "${WRAPPED_BINS_ARR[@]}"; do
        TARGET_EXECUTABLE_PATH="${EXECUTABLES_TARGET_DIR}/$ALIAS"
        TARGET_WRAPPER_PATH="/usr/local/bin/run-$ALIAS.sh"
        TARGET_WRAPPER_EXE_ENV_VAR_NAME=$(sed "s|-|_|g" <<<"${ALIAS}_EXE" | tr '[:lower:]' '[:upper:]')
        cp ./scripts/executable-wrapper-template.sh "$TARGET_EXECUTABLE_PATH"
        sed -i "s|SUBST_TARGET_WRAPPER_PATH|$TARGET_WRAPPER_PATH|g; s|SUBST_TARGET_BIN_NAME|$ALIAS|g; s|SUBST_EXE_ENV_VAR|$TARGET_WRAPPER_EXE_ENV_VAR_NAME|g" "$TARGET_EXECUTABLE_PATH"
        chmod +x "$TARGET_EXECUTABLE_PATH"
    done
else
    echo "Invalid WRAPPERTYPE specified: $WRAPPERTYPE. Must be one of SHELL_FUNCTION or EXECUTABLE."
    exit 1
fi

if [ "${INSTALL_WITH_SUDO}" = "true" ]; then
    sudo -u ${_REMOTE_USER} bash -c "/tmp/install-provider.sh ${USENET6}"
fi
rm /tmp/install-provider.sh

exit 0