install.sh

#!/bin/sh
# Sparsify installer — one command from checkout to `sparsify run`.
#
#   ./install.sh                       (from a cloned repo)
#   curl -fsSL <repo>/install.sh | sh  (standalone; set SPARSIFY_REPO first
#                                       if the default below is wrong)
#
# What it does, transparently:
#   1. verifies macOS on Apple Silicon and Python >= 3.10
#   2. installs Sparsify into its own venv at ~/.sparsify/venv
#   3. links the `sparsify` command into ~/.local/bin (or /usr/local/bin)
#   4. leaves models in ~/.sparsify/models (override: SPARSIFY_MODELS_DIR)
#
# Knobs (env): SPARSIFY_HOME, SPARSIFY_BIN_DIR, SPARSIFY_NO_SERVICE=1
set -eu

SPARSIFY_HOME="${SPARSIFY_HOME:-$HOME/.sparsify}"
SPARSIFY_REPO="${SPARSIFY_REPO:-https://github.com/daylinkltd/sparsify}"

say()  { printf '\033[1;36m>>\033[0m %s\n' "$1"; }
fail() { printf '\033[1;31mERROR:\033[0m %s\n' "$1" >&2; exit 1; }

# 1 ── platform checks ------------------------------------------------------
SYSTEM="$(uname -s)"
MACHINE="$(uname -m)"
IS_MACOS_ARM64=0

if [ "$SYSTEM" = "Darwin" ] && [ "$MACHINE" = "arm64" ]; then
  IS_MACOS_ARM64=1
fi

if [ "$IS_MACOS_ARM64" = "1" ]; then
  say "Platform: macOS (Apple Silicon arm64). Installing with MLX backend."
  EXTRA_FEATURES="[mlx,viz,dev]"
else
  say "Platform: $SYSTEM ($MACHINE). Installing with PyTorch backend (CUDA/CPU)."
  EXTRA_FEATURES="[torch,viz,dev]"
fi

PY=""
for cand in python3.13 python3.12 python3.11 python3.10 python3; do
  if command -v "$cand" >/dev/null 2>&1; then
    if "$cand" -c 'import sys; sys.exit(0 if sys.version_info >= (3,10) else 1)' 2>/dev/null; then
      PY="$cand"; break
    fi
  fi
done
[ -n "$PY" ] || fail "Python >= 3.10 not found. Install it (e.g. 'brew install python') and re-run."
say "Using $($PY --version 2>&1)"

# 2 ── source tree ----------------------------------------------------------
if [ -f "pyproject.toml" ] && grep -q '^name = "sparsify"' pyproject.toml 2>/dev/null; then
  SRC_DIR="$(pwd)"
  say "Installing from checkout: $SRC_DIR"
else
  command -v git >/dev/null 2>&1 || fail "git not found (needed to fetch Sparsify)."
  SRC_DIR="$SPARSIFY_HOME/src"
  if [ -d "$SRC_DIR/.git" ]; then
    say "Updating existing source in $SRC_DIR"
    git -C "$SRC_DIR" pull --ff-only
  else
    say "Cloning $SPARSIFY_REPO"
    mkdir -p "$SPARSIFY_HOME"
    git clone --depth 1 "$SPARSIFY_REPO" "$SRC_DIR"
  fi
fi

# 3 ── venv -----------------------------------------------------------------
say "Creating venv at $SPARSIFY_HOME/venv"
mkdir -p "$SPARSIFY_HOME"
"$PY" -m venv "$SPARSIFY_HOME/venv"
"$SPARSIFY_HOME/venv/bin/pip" install --quiet --upgrade pip
say "Installing Sparsify dependencies (this may take a few minutes on the first run)"
"$SPARSIFY_HOME/venv/bin/pip" install --quiet "$SRC_DIR$EXTRA_FEATURES" huggingface_hub hf_transfer

# Brand the interpreter so Activity Monitor / ps show "sparsify-runtime",
# not "python3.12": copy the venv's interpreter stub and point the console
# script at it.
cp -L "$SPARSIFY_HOME/venv/bin/python3" "$SPARSIFY_HOME/venv/bin/sparsify-runtime"
# Mac uses -i '', Linux/others use -i without space.
if [ "$SYSTEM" = "Darwin" ]; then
  sed -i '' "1s|.*|#!$SPARSIFY_HOME/venv/bin/sparsify-runtime|" "$SPARSIFY_HOME/venv/bin/sparsify"
else
  sed -i "1s|.*|#!$SPARSIFY_HOME/venv/bin/sparsify-runtime|" "$SPARSIFY_HOME/venv/bin/sparsify"
fi

# 4 ── launcher -------------------------------------------------------------
BIN_DIR="${SPARSIFY_BIN_DIR:-$HOME/.local/bin}"
if [ -z "${SPARSIFY_BIN_DIR:-}" ]; then
  case ":$PATH:" in *":$BIN_DIR:"*) ;; *)
    if [ -w /usr/local/bin ]; then BIN_DIR=/usr/local/bin; fi ;;
  esac
fi
mkdir -p "$BIN_DIR"
ln -sf "$SPARSIFY_HOME/venv/bin/sparsify" "$BIN_DIR/sparsify"
say "Linked sparsify -> $BIN_DIR/sparsify"
case ":$PATH:" in
  *":$BIN_DIR:"*) ;;
  *) printf '\033[1;33mNOTE:\033[0m add %s to your PATH, e.g.:\n  echo '\''export PATH="%s:$PATH"'\'' >> ~/.zshrc\n' "$BIN_DIR" "$BIN_DIR" ;;
esac

# 5 ── smoke check + background service --------------------------------------
"$SPARSIFY_HOME/venv/bin/sparsify" --version >/dev/null || fail "installed CLI failed to run"

if [ "$SYSTEM" != "Darwin" ]; then
  SERVICE_MSG='Run `sparsify serve` directly in your terminal to start the API service (automatic start is macOS-only).'
elif [ -n "${SPARSIFY_NO_SERVICE:-}" ]; then
  SERVICE_MSG='Service not installed (SPARSIFY_NO_SERVICE set) - run `sparsify serve` or `sparsify start` yourself'
else
  say "Starting the Sparsify API service on http://localhost:7777"
  if SPARSIFY_HOME="$SPARSIFY_HOME" "$SPARSIFY_HOME/venv/bin/sparsify" start; then
    SERVICE_MSG='API running on http://localhost:7777 (starts at login; sparsify stop to remove)'
  else
    SERVICE_MSG='Service not started - run `sparsify serve` manually (see message above)'
  fi
fi

printf '\n\033[1;32mSparsify installed.\033[0m %s\n\nTry:\n\n' "$SERVICE_MSG"
printf '  sparsify models              # browse the catalog\n'
printf '  sparsify pull olmoe:1b-7b    # 3.9 GB starter MoE\n'
printf '  sparsify run  olmoe:1b-7b    # chat, auto RAM budget\n\n'
printf '  curl http://localhost:7777/v1/chat/completions \\\n'
printf '    -d '"'"'{"model":"olmoe:1b-7b","messages":[{"role":"user","content":"hi"}]}'"'"'\n\n'
MODELS_DIR=$("$SPARSIFY_HOME/venv/bin/python" -c \
  "from sparsify.runtime.model_registry import MODELS_DIR; print(MODELS_DIR)" \
  2>/dev/null || printf '%s' "${SPARSIFY_MODELS_DIR:-$SPARSIFY_HOME/models}")
printf 'Models live in %s\n' "$MODELS_DIR"
printf '  (change with SPARSIFY_MODELS_DIR, or ~/.sparsify/config.json {"models_dir": "..."})\n'