reinstall_python_sdk.sh

#!/usr/bin/env bash
# ======================
# reinstall_python_sdk.sh
# ======================
#
# Purpose
# -------
# This script reinstalls the Summoner Python SDK inside a virtual environment, and (as part of the workflow)
# also triggers a Rust-side reinstall via `reinstall_rust_server.sh`.
#
# The goal is to make "reinstall everything I need to run Summoner" a single command, while remaining compatible
# with different virtualenv naming conventions (venv/.venv/custom) and different Python installers (pip vs uv).
#
# What it does, step by step
# --------------------------
# 1) Selects which virtual environment to use (in priority order):
#    - If --venv <path> is provided: uses that venv.
#    - Else if $VIRTUAL_ENV is set (i.e., you already activated a venv): uses the active venv.
#    - Else:
#        - Default:       <repo-root>/venv
#        - With --dev-core: <this-script-dir>/venv
#
# 2) Activates that virtual environment (it must already exist).
#
# 3) Reinstalls Rust components by calling:
#    - ./reinstall_rust_server.sh [<optional-prefix>] [--dev-core]
#
# 4) Reinstalls the Python package named `summoner`:
#    - If `summoner` is already installed, it is uninstalled first.
#    - Then it is installed again from this directory:
#        - Default: installs normally   (pkg install .)
#        - With --dev-core: editable    (pkg install -e .)
#
# Arguments
# ---------
# Usage:
#   bash reinstall_python_sdk.sh [<optional-prefix>] [--dev-core] [--uv] [--venv <path>]
#
# - <optional-prefix>
#   An optional string forwarded to the Rust reinstall script. If you do not need it, omit it.
#
# - --dev-core
#   Switches the default venv location (only used when neither --venv nor $VIRTUAL_ENV is provided),
#   and also switches Python installation mode:
#     - Default venv becomes: <this-script-dir>/venv
#     - Installs `summoner` in editable mode (-e), which is useful when you are actively developing the SDK.
#
# - --uv
#   Uses `uv` as the package manager instead of `pip` for Python install/uninstall/show operations.
#   This is useful in repos that standardize on uv and do not want direct `pip ...` commands in their workflow.
#
#   Important details:
#     - You must have `uv` installed and available on PATH.
#     - The script still activates a venv; `--uv` only changes how packages are installed inside it.
#     - In `--uv` mode, the script sets VIRTUAL_ENV to the chosen venv path so `uv pip ...` targets the correct environment.
#
# - --venv <path>
#   Explicitly selects which venv to use (supports venv/.venv/myvenv/etc).
#   This has priority over both $VIRTUAL_ENV and the default conventions.
#
#   Examples:
#     bash reinstall_python_sdk.sh --venv .venv
#     bash reinstall_python_sdk.sh --venv myvenv --uv
#
# Environment expectations
# ------------------------
# - A virtual environment must already exist at the selected location.
#   If it does not exist, create it first (examples below).
#
# - Default mode requires `pip` to exist inside the venv.
#   If your venv was created in a way that does not include pip, either:
#     - recreate the venv with pip available, or
#     - rerun this script with --uv.
#
# Examples
# --------
# 1) Typical usage (pip, normal install into <repo-root>/venv unless $VIRTUAL_ENV is set):
#    bash reinstall_python_sdk.sh
#
# 2) With a Rust prefix forwarded to reinstall_rust_server.sh:
#    bash reinstall_python_sdk.sh rust_server_v1_0_0
#
# 3) Developer workflow (editable install, venv next to this script unless $VIRTUAL_ENV/--venv is provided):
#    bash reinstall_python_sdk.sh --dev-core
#
# 4) uv workflow (uses uv pip instead of pip):
#    bash reinstall_python_sdk.sh --uv
#    bash reinstall_python_sdk.sh rust_server_v1_0_0 --uv
#    bash reinstall_python_sdk.sh --dev-core --uv
#
# 5) Flexible venv naming:
#    python3 -m venv .venv
#    source .venv/bin/activate
#    bash reinstall_python_sdk.sh --uv
#
#    # or explicitly:
#    bash reinstall_python_sdk.sh --venv .venv --uv
#
# Creating the venv (if missing)
# ------------------------------
# Default venv:
#   python3 -m venv venv
#   source venv/bin/activate
#
# uv venv (if your repo standardizes on uv):
#   uv venv venv
#   source venv/bin/activate
#
# Troubleshooting
# ---------------
# - "Virtualenv not found": create the venv at the expected location (see above), or pass --venv <path>,
#   or activate your venv first so $VIRTUAL_ENV is set.
# - "--uv was set but 'uv' is not on PATH": install uv and ensure it is available in your shell.
# - "'pip' not found in this environment": use --uv or recreate the venv with pip.
# ======================
set -euo pipefail

# ─────────────────────────────────────────────────────────────
# Resolve absolute paths
# ─────────────────────────────────────────────────────────────
THIS_SCRIPT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ROOT_DIR="$(cd "$THIS_SCRIPT/.." && pwd)"
RUST_SCRIPT="$THIS_SCRIPT/reinstall_rust_server.sh"

# ─────────────────────────────────────────────────────────────
# Parse args: <optional-prefix> [--dev-core] [--uv] [--venv <path>]
# ─────────────────────────────────────────────────────────────
DEV_CORE=false
USE_UV=false
PREFIX_FILTER=""
VENV_OVERRIDE=""

echo "🔍 Raw arguments: $*"

prev=""
for arg in "$@"; do
  if [[ "$arg" == "--dev-core" ]]; then
    DEV_CORE=true
  elif [[ "$arg" == "--uv" ]]; then
    USE_UV=true
  elif [[ "$prev" == "--venv" ]]; then
    VENV_OVERRIDE="$arg"
    prev=""
    continue
  elif [[ "$arg" == "--venv" ]]; then
    prev="--venv"
  elif [[ -z "$PREFIX_FILTER" && "$arg" != --* ]]; then
    PREFIX_FILTER="$arg"
  fi
done

if [[ "$prev" == "--venv" ]]; then
  echo "❌ --venv requires a value, e.g. --venv .venv" >&2
  exit 1
fi

echo "✅ Final values: DEV_CORE=$DEV_CORE, USE_UV=$USE_UV, PREFIX_FILTER=$PREFIX_FILTER, VENV_OVERRIDE=$VENV_OVERRIDE"

# ─────────────────────────────────────────────────────────────
# Select venv location (flexible)
# Priority:
#   1) --venv <path>
#   2) active venv via $VIRTUAL_ENV
#   3) default convention (dev-core vs parent venv)
# ─────────────────────────────────────────────────────────────
if [[ -n "$VENV_OVERRIDE" ]]; then
  VENV_DIR="$VENV_OVERRIDE"
elif [[ -n "${VIRTUAL_ENV:-}" ]]; then
  VENV_DIR="$VIRTUAL_ENV"
elif [[ "$DEV_CORE" == true ]]; then
  VENV_DIR="$THIS_SCRIPT/venv"
else
  VENV_DIR="$ROOT_DIR/venv"
fi

# If user passed a relative path via --venv, resolve it relative to ROOT_DIR for stability.
# (If it's already absolute, keep it as-is.)
if [[ "$VENV_DIR" != /* ]]; then
  VENV_DIR="$(cd "$ROOT_DIR" && cd "$VENV_DIR" && pwd)"
fi

# ─────────────────────────────────────────────────────────────
# Activate virtualenv
# ─────────────────────────────────────────────────────────────
if [ -f "$VENV_DIR/bin/activate" ]; then
  echo "✅ Activating venv from: $VENV_DIR"
  # shellcheck disable=SC1090
  . "$VENV_DIR/bin/activate"
else
  echo "❌ Virtualenv not found at: $VENV_DIR" >&2
  echo "   Tips:" >&2
  echo "     - pass --venv .venv (or your venv path)" >&2
  echo "     - or activate your venv first so \$VIRTUAL_ENV is set" >&2
  exit 1
fi

# ─────────────────────────────────────────────────────────────
# Package manager shim (pip default, --uv switches to uv pip)
# ─────────────────────────────────────────────────────────────
pkg() {
  if [[ "$USE_UV" == "true" ]]; then
    command -v uv >/dev/null 2>&1 || { echo "❌ --uv was set but 'uv' is not on PATH"; exit 1; }
    # Force uv to target this exact venv, even if activation is flaky.
    VIRTUAL_ENV="$VENV_DIR" uv pip "$@"
  else
    command -v pip >/dev/null 2>&1 || {
      echo "❌ 'pip' not found in this environment. Either recreate the venv with pip, or rerun with --uv."
      exit 1
    }
    pip "$@"
  fi
}

# Diagnostic: show interpreter + tool in use
PY=$(command -v python 2>/dev/null || echo 'not found')
echo "🐍 Using Python: $PY"
echo "🧪 Selected venv: $VENV_DIR"
if [[ -n "${VIRTUAL_ENV:-}" ]]; then
  echo "🧪 Active \$VIRTUAL_ENV: $VIRTUAL_ENV"
fi

if [[ "$USE_UV" == "true" ]]; then
  UV=$(command -v uv 2>/dev/null || echo 'not found')
  UVV=$(uv --version 2>/dev/null || echo 'not found')
  echo "📦 Using uv:     $UV"
  echo "🔧 uv version:   $UVV"
else
  PIP=$(command -v pip 2>/dev/null || echo 'not found')
  PV=$(pip --version 2>/dev/null || echo 'not found')
  echo "📦 Using Pip:    $PIP"
  echo "🔧 Pip version:  $PV"
fi

# ─────────────────────────────────────────────────────────────
# Reinstall Rust crates (optional prefix)
# ─────────────────────────────────────────────────────────────
echo "🔁 Reinstalling Rust crates via: $RUST_SCRIPT"
if [ ! -f "$RUST_SCRIPT" ]; then
  echo "❌ Missing script: $RUST_SCRIPT"
  exit 1
fi

if [ "$DEV_CORE" = true ]; then
  if [[ -n "${VENV_OVERRIDE:-}" ]]; then
    bash "$RUST_SCRIPT" "$PREFIX_FILTER" --dev-core --venv "$VENV_DIR"
  else
    bash "$RUST_SCRIPT" "$PREFIX_FILTER" --dev-core
  fi
else
  if [[ -n "${VENV_OVERRIDE:-}" ]]; then
    bash "$RUST_SCRIPT" "$PREFIX_FILTER" --venv "$VENV_DIR"
  else
    bash "$RUST_SCRIPT" "$PREFIX_FILTER"
  fi
fi


# ─────────────────────────────────────────────────────────────
# Reinstall Python package in editable mode
# ─────────────────────────────────────────────────────────────
echo "🔁 Reinstalling Python package: summoner"

cd "$THIS_SCRIPT"

if pkg show summoner > /dev/null 2>&1; then
  echo "🗑️  Uninstalling existing 'summoner' package..."
  pkg uninstall -y summoner
else
  echo "ℹ️  'summoner' not currently installed — skipping uninstall."
fi

echo "📦 Installing 'summoner'..."
if [ "$DEV_CORE" = true ]; then
  echo "   (dev-core: editable install)"
  pkg install -e .
else
  echo "   (non-dev-core: regular install)"
  pkg install .
fi

echo "✅ Python SDK reinstalled successfully with prefix: '$PREFIX_FILTER'"

if [ "$DEV_CORE" = true ]; then
  echo "   (used --dev-core → default venv would be $THIS_SCRIPT/venv if not overridden)"
else
  echo "   (used default → default venv would be $ROOT_DIR/venv if not overridden)"
fi

if [[ -n "$VENV_OVERRIDE" ]]; then
  echo "   (used --venv → venv at $VENV_DIR)"
elif [[ -n "${VIRTUAL_ENV:-}" ]]; then
  echo "   (used active \$VIRTUAL_ENV → venv at $VENV_DIR)"
else
  echo "   (used convention → venv at $VENV_DIR)"
fi

if [[ "$USE_UV" == "true" ]]; then
  echo "   (used --uv → uv pip targeting $VENV_DIR)"
else
  echo "   (used default → pip)"
fi