#!/bin/sh
# Stoffel CLI installer.
#
#   curl -fsSL https://get.stoffelmpc.com | sh
#   curl -fsSL https://get.stoffelmpc.com | sh -s -- --version 0.1.0
#   curl -fsSL https://get.stoffelmpc.com | sh -s -- --runner-only
#
# Env overrides:
#   STOFFEL_VERSION       pin a version (same as --version)
#   STOFFEL_INSTALL_DIR   install location (default: ~/.local/bin)
#   STOFFEL_COMPONENT     install "cli" (default) or "runner"
set -eu

REPO="Stoffel-Labs/stoffel"
BIN="stoffel"
RUNNER="stoffel-run"
INSTALL_DIR="${STOFFEL_INSTALL_DIR:-$HOME/.local/bin}"
VERSION="${STOFFEL_VERSION:-}"
COMPONENT="${STOFFEL_COMPONENT:-cli}"

err()  { echo "stoffel-install: error: $*" >&2; exit 1; }
info() { echo "stoffel-install: $*"; }

# --- args ---
while [ $# -gt 0 ]; do
  case "$1" in
    --version)   VERSION="${2:-}"; shift 2 ;;
    --version=*) VERSION="${1#--version=}"; shift ;;
    --runner-only) COMPONENT="runner"; shift ;;
    --component)   COMPONENT="${2:-}"; shift 2 ;;
    --component=*) COMPONENT="${1#--component=}"; shift ;;
    *) err "unknown argument: $1" ;;
  esac
done

case "$COMPONENT" in
  cli|runner) ;;
  *) err "unsupported component '$COMPONENT' (expected 'cli' or 'runner')" ;;
esac

# --- downloader (curl or wget) ---
if command -v curl >/dev/null 2>&1; then
  dl()    { curl -fsSL "$1" -o "$2"; }
  dlout() { curl -fsSL "$1"; }
elif command -v wget >/dev/null 2>&1; then
  dl()    { wget -qO "$2" "$1"; }
  dlout() { wget -qO- "$1"; }
else
  err "need curl or wget installed"
fi

# --- platform detection ---
os="$(uname -s)"; arch="$(uname -m)"
case "$os" in
  Linux)  os_part="unknown-linux-gnu" ;;
  Darwin) os_part="apple-darwin" ;;
  *)      err "unsupported OS '$os' (Linux and macOS are supported)" ;;
esac
case "$arch" in
  x86_64|amd64)  arch_part="x86_64" ;;
  arm64|aarch64) arch_part="aarch64" ;;
  *)             err "unsupported architecture '$arch'" ;;
esac
TARGET="${arch_part}-${os_part}"

# --- resolve release tag/version ---
case "$COMPONENT" in
  cli)
    prefix="cli-v"
    archive_name="stoffel"
    ;;
  runner)
    prefix="stoffel-run-v"
    archive_name="$RUNNER"
    ;;
esac

if [ -n "$VERSION" ]; then
  TAG="${prefix}${VERSION}"
else
  info "resolving latest ${COMPONENT} release..."
  TAG="$(dlout "https://api.github.com/repos/${REPO}/releases" 2>/dev/null \
    | grep '"tag_name"' \
    | sed -E 's/.*"tag_name":[[:space:]]*"([^"]+)".*/\1/' \
    | grep "^${prefix}" | head -n 1 || true)"
  [ -n "$TAG" ] || err "no ${prefix}* release found for ${REPO} (has a release been cut yet?)"
  VERSION="${TAG#$prefix}"
fi

TARBALL="${archive_name}-${VERSION}-${TARGET}.tar.gz"
BASE="https://github.com/${REPO}/releases/download/${TAG}"
info "installing ${COMPONENT} ${VERSION} (${TARGET})"

# --- temp workspace ---
tmp="$(mktemp -d)"
trap 'rm -rf "$tmp"' EXIT INT TERM
cd "$tmp"

# --- download ---
dl "${BASE}/${TARBALL}" "$TARBALL" \
  || err "no prebuilt binary for ${TARGET} in ${TAG} (your platform may not be built yet — try building from source)"
dl "${BASE}/SHA256SUMS" "SHA256SUMS" || err "could not download SHA256SUMS"

# --- checksum verification ---
if command -v sha256sum >/dev/null 2>&1; then sha="sha256sum"
elif command -v shasum >/dev/null 2>&1; then sha="shasum -a 256"
else err "need sha256sum or shasum to verify the download"
fi
expected="$(grep " ${TARBALL}\$" SHA256SUMS | awk '{print $1}' | head -n 1)"
[ -n "$expected" ] || err "no checksum listed for ${TARBALL}"
actual="$($sha "$TARBALL" | awk '{print $1}')"
[ "$expected" = "$actual" ] || err "checksum mismatch for ${TARBALL}"
info "checksum verified"

# --- extract & install ---
tar xzf "$TARBALL"
stage="${archive_name}-${VERSION}-${TARGET}"

mkdir -p "$INSTALL_DIR"

# install_binary <name> <required:0|1>
install_binary() {
  name="$1"; required="$2"
  src="${stage}/${name}"
  [ -f "$src" ] || src="$(find . -type f -name "$name" | head -n 1)"
  if [ -z "${src:-}" ] || [ ! -f "$src" ]; then
    if [ "$required" -eq 1 ]; then
      err "binary '${name}' not found in archive"
    fi
    info "note: '${name}' not bundled in this release; skipping"
    return 0
  fi
  install -m 0755 "$src" "$INSTALL_DIR/${name}" 2>/dev/null \
    || { cp "$src" "$INSTALL_DIR/${name}" && chmod 0755 "$INSTALL_DIR/${name}"; }
  info "installed to ${INSTALL_DIR}/${name}"
}

# The CLI is required; the runner powers local MPC execution (`stoffel run`,
# `stoffel dev`). `stoffel` discovers `stoffel-run` as a sibling on disk.
case "$COMPONENT" in
  cli)
    install_binary "$BIN" 1
    install_binary "$RUNNER" 1
    ;;
  runner)
    install_binary "$RUNNER" 1
    ;;
esac

# --- PATH hint ---
case ":${PATH}:" in
  *":${INSTALL_DIR}:"*) : ;;
  *)
    echo ""
    echo "  ${INSTALL_DIR} is not on your PATH. Add it with:"
    echo "    echo 'export PATH=\"${INSTALL_DIR}:\$PATH\"' >> ~/.profile && . ~/.profile"
    echo ""
    ;;
esac

# --- confirm ---
case "$COMPONENT" in
  cli)
    if "$INSTALL_DIR/${BIN}" --version >/dev/null 2>&1; then
      info "$("$INSTALL_DIR/${BIN}" --version)"
    fi
    info "done — run '${BIN} --help' to get started"
    ;;
  runner)
    if "$INSTALL_DIR/${RUNNER}" --help >/dev/null 2>&1; then
      info "${RUNNER} installed"
    elif runner_check_output="$("$INSTALL_DIR/${RUNNER}" 2>&1)"; then
      info "${RUNNER} installed"
    else
      status=$?
      # With no arguments, stoffel-run prints usage and exits 1. That still
      # proves the installed binary starts without requiring a positional $1.
      if [ "$status" -eq 1 ] && printf '%s\n' "$runner_check_output" | grep -q "Stoffel VM Runner"; then
        info "${RUNNER} installed"
      else
        err "${RUNNER} failed to start after install (exit ${status})"
      fi
    fi
    info "done — run '${RUNNER} --help' to get started"
    ;;
esac