prepare-machine

#!/bin/bash
set -euo pipefail

usage() {
    echo "Usage: ./prepare-machine [ ARGS ] <ROLE>"
    echo
    echo "ARGS:"
    echo "  -h, --help    ... Print help"
    echo "  -v, --verbose ... Verbose mode"
    echo
    echo "ROLE: dev-docker | dev-machine"
}

ROLE=dev-machine
while [ $# != 0 ]; do
    case "$1" in
    dev-docker)
        ROLE=dev-docker
        shift
        ;;
    dev-machine)
        ROLE=dev-machine
        shift
        ;;
    -h|--help)
        usage
        exit 0
        ;;
    -v|--verbose)
        set -x
        shift
        ;;
    *)
        echo "Unknown parameter '$1'"
        usage
        exit 1
    esac
done


case "$ROLE" in
dev-machine)
    ;;
dev-docker)
    GCC_TOOLSET_AUTOENABLE=${GCC_TOOLSET_AUTOENABLE-true}
    INSTALL_NIX=${INSTALL_NIX-true}
    ;;
esac

if [ "$EUID" = 0 ]; then
    SUDO=${SUDO-}
else
    SUDO=${SUDO-sudo}
fi

OS_ID="$(grep ^ID= /etc/os-release | cut -d'=' -f2 | tr -d '"')"
OS_ID_LIKE="$(grep ^ID_LIKE= /etc/os-release | cut -d'=' -f2 | tr -d '"')"
OS_VERSION="$(grep ^VERSION= /etc/os-release | cut -d'=' -f2 | tr -d '"')"
OS_VERSION_MAJOR="${OS_VERSION%%.*}"

echo "=== Setting up development environment ==="
echo "Selected role: $ROLE"

echo
echo "*** Installing common tools ***"
$SUDO dnf install -y sudo git

echo
echo "*** Enabling extra repositories (epel/crb) ***"
$SUDO dnf install -y 'dnf-command(config-manager)'

if [[ "$OS_ID" =~ fedora|rhel || "$OS_ID_LIKE" =~ fedora|rhel ]] && [ "$OS_VERSION_MAJOR" = 8 ] && [ "$OS_ID" != ol ]; then
    # PowerTools repository is enabled (CentOS/Rocky/etc. 8)
    echo "Detected OS: RHEL-like 8"
    $SUDO dnf install -y epel-release
    $SUDO dnf config-manager --set-enabled powertools
elif [[ "$OS_ID" =~ fedora|rhel || "$OS_ID_LIKE" =~ fedora|rhel ]] && [ "$OS_VERSION_MAJOR" = 9 ] && [ "$OS_ID" != ol ]; then
    # CodeReady Builder repository is enabled (CentOS/Rocky/etc. 9+)
    echo "Detected OS: RHEL-like 9"
    $SUDO dnf install -y epel-release
    $SUDO dnf config-manager --set-enabled crb
elif [ "$OS_ID" = ol ] && [[ "$OS_VERSION_MAJOR" =~ 8|9 ]]; then
    # CodeReady Builder repository is enabled (Oracle Linux 8, 9)
    echo "Detected OS: Oracle Linux ${OS_VERSION_MAJOR}"
    $SUDO dnf install -y epel-release
    $SUDO dnf config-manager --set-enabled "ol${OS_VERSION_MAJOR}_codeready_builder"
elif [ "$OS_ID" = ol ] && [ "$OS_VERSION_MAJOR" = 10 ]; then
    # CodeReady Builder repository is enabled (Oracle Linux 10)
    echo "Detected OS: Oracle Linux 10"
    $SUDO dnf install -y oracle-epel-release-el10
    $SUDO dnf config-manager --set-enabled "ol10_codeready_builder"
    # OL10 comes with gcc 14 by default, but misses libasan-static package
    GCC_TOOLSET_VERSION=${GCC_TOOLSET_VERSION-15}
    USE_GCC_TOOLSET=${USE_GCC_TOOLSET-true}
    INSTALL_UV=${INSTALL_UV-true}
fi

echo
echo "*** Setting up CPP development tools ***"
USE_GCC_TOOLSET=${USE_GCC_TOOLSET-true}
GCC_TOOLSET_VERSION=${GCC_TOOLSET_VERSION-14}
GCC_TOOLSET_AUTOENABLE=${GCC_TOOLSET_AUTOENABLE-false}
INSTALL_UV=${INSTALL_UV-false}

$SUDO dnf install -y \
    cmake make rpm-build \
    gcc gcc-c++ \
    clang-tools-extra \
    doxygen

if [ "$USE_GCC_TOOLSET" = true ]; then
    $SUDO dnf install -y "gcc-toolset-${GCC_TOOLSET_VERSION}"
fi

if [ "$GCC_TOOLSET_AUTOENABLE" = true ]; then
    if [ "$GCC_TOOLSET_VERSION" -lt 15 ]; then
        echo "source /opt/rh/gcc-toolset-${GCC_TOOLSET_VERSION}/enable" | $SUDO tee "/etc/profile.d/gcc-toolset-${GCC_TOOLSET_VERSION}-enable.sh"
    else
        echo "eval \"\$(gcc-toolset-${GCC_TOOLSET_VERSION}-env printenv | grep gcc-toolset)\"" | $SUDO tee "/etc/profile.d/gcc-toolset-${GCC_TOOLSET_VERSION}-enable.sh"
    fi

    $SUDO chmod 644 "/etc/profile.d/gcc-toolset-${GCC_TOOLSET_VERSION}-enable.sh"
fi

echo
echo "*** Installing library dependencies ***"
if [ "$USE_GCC_TOOLSET" = true ]; then
    $SUDO dnf install -y \
        "gcc-toolset-${GCC_TOOLSET_VERSION}-libasan-devel" \
        "gcc-toolset-${GCC_TOOLSET_VERSION}-libubsan-devel"
else
    $SUDO dnf install -y --enablerepo=devel \
        libasan-static \
        libubsan-static
fi

$SUDO dnf install -y \
    libpcap-devel \
    dpdk dpdk-devel dpdk-tools \


echo
echo "*** Installing python tools ***"
if [ "$INSTALL_UV" = true ]; then
    $SUDO dnf install -y uv
else
    echo "WARNING: Package uv is not installed"
fi


if [ "$INSTALL_NIX" = true ]; then
    echo
    echo "*** Installing nix shell ***"
    if nix --version; then
        echo "WARNING: Nix is already installed, skipping installation"

    else
        NIX_VERSION=2.35.2

        tmp="$(mktemp --tmpdir -d prepare-machine-XXXXXXXX)"
        trap "rm -rf '$tmp'" EXIT
        curl -Lo "$tmp/nix-${NIX_VERSION}-x86_64-linux.tar.xz" "https://releases.nixos.org/nix/nix-${NIX_VERSION}/nix-${NIX_VERSION}-x86_64-linux.tar.xz"
        tar xf "$tmp/nix-${NIX_VERSION}-x86_64-linux.tar.xz" -C "$tmp"

        mkdir -p ~/.config/nix/
        if [ -f ~/.config/nix/nix.conf ]; then
            mv ~/.config/nix/nix.conf ~/.config/nix/nix.conf.bak
        fi
        touch ~/.config/nix/nix.conf
        echo 'build-users-group =' >> ~/.config/nix/nix.conf
        if [ -f /.dockerenv ]; then # Are we running in docker?
            echo 'sandbox = false' >> ~/.config/nix/nix.conf
        fi

        NIX_BECOME="$SUDO" "$tmp/nix-${NIX_VERSION}-x86_64-linux/install"

        # Prefetch nix-shell packages
        (
            . ~/.nix-profile/etc/profile.d/nix.sh
            nix-shell --run echo
        )
    fi
fi


if [ -f ./adaptmon.spec ]; then
    echo
    echo "*** Installing RPM build dependencies ***"
    $SUDO dnf builddep -y --define 'builddep_only 1' "./adaptmon.spec"
fi