install.sh

#!/bin/bash
# Install Spur — AI-native job scheduler
#
# Usage:
#   curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash
#
#   # Install nightly:
#   curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- nightly
#
#   # Install a specific version:
#   curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- v0.1.0
#
#   # Install to a custom directory:
#   curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | INSTALL_DIR=/opt/spur/bin bash
#
#   # Uninstall:
#   curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- uninstall

set -euo pipefail

REPO="ROCm/spur"
INSTALL_DIR="${INSTALL_DIR:-${HOME}/.local/bin}"

BINARIES="spur spurctld spurd spurdbd spurrestd"
SYMLINKS="sbatch srun squeue scancel sinfo sacct scontrol"

log()  { echo "==> $*"; }
err()  { echo "ERROR: $*" >&2; exit 1; }

usage() {
    cat <<'EOF'
Spur installer — AI-native job scheduler

USAGE:
    install.sh [OPTIONS] [VERSION]

VERSION:
    latest          Install the latest stable release (default)
    nightly         Install the latest nightly build
    v0.1.0          Install a specific version
    uninstall       Remove Spur binaries

OPTIONS:
    -h, --help      Show this help message
    -d, --dir DIR   Install directory (default: ~/.local/bin)

ENVIRONMENT:
    INSTALL_DIR     Override install directory (same as --dir)

EXAMPLES:
    # Install latest stable
    curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash

    # Install nightly
    curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- nightly

    # Install to /opt/spur/bin
    curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- -d /opt/spur/bin

    # Uninstall
    curl -fsSL https://raw.githubusercontent.com/ROCm/spur/main/install.sh | bash -s -- uninstall
EOF
    exit 0
}

do_uninstall() {
    log "Uninstalling Spur from ${INSTALL_DIR}/"
    local removed=0
    for f in ${BINARIES} ${SYMLINKS}; do
        if [ -e "${INSTALL_DIR}/${f}" ]; then
            rm -f "${INSTALL_DIR}/${f}"
            echo "  removed ${f}"
            removed=$((removed + 1))
        fi
    done
    if [ "${removed}" -eq 0 ]; then
        log "No Spur files found in ${INSTALL_DIR}/"
    else
        log "Removed ${removed} file(s). Spur has been uninstalled."
    fi
    exit 0
}

# --- Parse arguments ---
VERSION="latest"
while [ $# -gt 0 ]; do
    case "$1" in
        -h|--help)     usage ;;
        -d|--dir)      INSTALL_DIR="$2"; shift 2 ;;
        uninstall)     do_uninstall ;;
        *)             VERSION="$1"; shift ;;
    esac
done

# --- Platform check ---
OS=$(uname -s)
ARCH=$(uname -m)
[ "$OS" = "Linux" ] || err "Spur currently supports Linux only (got ${OS})"
[ "$ARCH" = "x86_64" ] || err "Spur currently supports x86_64 only (got ${ARCH})"

# --- glibc check (binaries built on manylinux_2_28, require glibc >= 2.28) ---
GLIBC_VER=$(ldd --version 2>&1 | head -1 | grep -oE '[0-9]+\.[0-9]+$' || echo "0")
if [ "$(printf '%s\n' "2.28" "${GLIBC_VER}" | sort -V | head -1)" != "2.28" ]; then
    err "Spur requires glibc >= 2.28 (found ${GLIBC_VER}). Supported: Ubuntu 20.04+, Debian 10+, RHEL 8+, Fedora 28+"
fi

# --- Resolve version ---
if [ "$VERSION" = "latest" ]; then
    log "Fetching latest release..."
    VERSION=$(curl -fsSL "https://api.github.com/repos/${REPO}/releases/latest" \
        | grep '"tag_name"' | head -1 | cut -d'"' -f4) \
        || err "No releases found. Create one with: gh release create v0.1.0"
fi
log "Installing Spur ${VERSION}"

# --- Download ---
TMPDIR=$(mktemp -d)
trap 'rm -rf "${TMPDIR}"' EXIT

if [ "$VERSION" = "nightly" ]; then
    # Nightly tarballs include date+sha in the name — find the .tar.gz asset
    TARBALL=$(curl -fsSL "https://api.github.com/repos/${REPO}/releases/tags/nightly" \
        | grep '"name"' | grep '\.tar\.gz"' | grep -v sha256 | head -1 | cut -d'"' -f4) \
        || err "Could not find nightly release assets"
else
    TARBALL="spur-${VERSION}-linux-amd64.tar.gz"
fi

URL="https://github.com/${REPO}/releases/download/${VERSION}/${TARBALL}"
log "Downloading ${TARBALL}..."
curl -fSL -o "${TMPDIR}/${TARBALL}" "${URL}" \
    || err "Download failed. Check that release ${VERSION} exists at https://github.com/${REPO}/releases"

# --- Verify checksum if available ---
SHA_URL="${URL}.sha256"
if curl -fsSL -o "${TMPDIR}/${TARBALL}.sha256" "${SHA_URL}" 2>/dev/null; then
    log "Verifying checksum..."
    (cd "${TMPDIR}" && sha256sum -c "${TARBALL}.sha256") || err "Checksum mismatch"
fi

# --- Extract ---
log "Extracting..."
tar xzf "${TMPDIR}/${TARBALL}" -C "${TMPDIR}"

# --- Install ---
mkdir -p "${INSTALL_DIR}"
# Find the extracted directory (name varies for nightly)
EXTRACTED=$(find "${TMPDIR}" -maxdepth 1 -type d -name 'spur-*' | head -1)
[ -n "${EXTRACTED}" ] || err "Could not find extracted directory"
cp -f "${EXTRACTED}"/bin/* "${INSTALL_DIR}/"
chmod +x "${INSTALL_DIR}/spur" "${INSTALL_DIR}/spurctld" "${INSTALL_DIR}/spurd" \
         "${INSTALL_DIR}/spurdbd" "${INSTALL_DIR}/spurrestd"

# --- Verify ---
if ! "${INSTALL_DIR}/spur" --version >/dev/null 2>&1; then
    # Binary exists but --version may not be implemented yet
    if [ -x "${INSTALL_DIR}/spur" ]; then
        log "Binaries installed (version flag not yet supported)"
    else
        err "Installation verification failed"
    fi
fi

# --- PATH hint ---
log "Installed to ${INSTALL_DIR}/"
log "Binaries: ${BINARIES}"
log "Symlinks: ${SYMLINKS}"

if ! echo "$PATH" | tr ':' '\n' | grep -qx "${INSTALL_DIR}"; then
    echo ""
    echo "Add to your PATH:"
    echo "  export PATH=\"${INSTALL_DIR}:\$PATH\""
    echo ""
    echo "Or add to ~/.bashrc:"
    echo "  echo 'export PATH=\"${INSTALL_DIR}:\$PATH\"' >> ~/.bashrc"
fi

log "Done."