#!/usr/bin/env bash
# Install the latest pVisor nightly wheel from GitHub Releases (tag: nightly).
#
# Usage:
#   curl -fsSL https://raw.githubusercontent.com/DeepLink-org/pvisor/main/scripts/install-nightly.sh | bash
#   PVISOR_GITHUB_REPO=DeepLink-org/pvisor PVISOR_NIGHTLY_TAG=nightly ./scripts/install-nightly.sh
#
# Environment:
#   PYTHON                   Python interpreter (default: python3)
#   PVISOR_GITHUB_REPO   owner/repo (default: DeepLink-org/pvisor)
#   PVISOR_NIGHTLY_TAG   release tag (default: nightly)

set -euo pipefail

REPO="${PVISOR_GITHUB_REPO:-DeepLink-org/pvisor}"
TAG="${PVISOR_NIGHTLY_TAG:-nightly}"
PYTHON="${PYTHON:-python3}"

if ! command -v "$PYTHON" >/dev/null 2>&1; then
  echo "error: Python not found ($PYTHON)" >&2
  exit 1
fi

case "$(uname -s)-$(uname -m)" in
  Linux-x86_64) platform_re='manylinux.*x86_64' ;;

  Darwin-arm64) platform_re='macosx.*arm64' ;;
  *)
    echo "error: unsupported platform $(uname -s)-$(uname -m)" >&2
    exit 1
    ;;
esac

if ! "$PYTHON" -c 'import sys; sys.exit(sys.version_info < (3, 10))'; then
  echo "error: pVisor wheels require Python 3.10+; got $("$PYTHON" --version)" >&2
  exit 1
fi
api="https://api.github.com/repos/${REPO}/releases/tags/${TAG}"

echo "Fetching nightly release assets from ${REPO} (tag=${TAG})..." >&2

# Write the asset picker to a helper file instead of embedding a here-document
# inside a command substitution: bash 5.2 mis-parses long heredocs whose body
# contains a line beginning with the delimiter (for example `PY3_RE`) in that
# position.
asset_helper="$(mktemp)"
trap 'rm -f "$asset_helper"' EXIT
cat > "$asset_helper" <<'PY'
import json
import re
import sys
import urllib.error
import urllib.request

api, platform_pattern, repo, tag = sys.argv[1:]
platform_re = re.compile(platform_pattern)

# Wheels contain native CLIs but only pure Python modules, so one py3-none wheel
# is published per supported OS/architecture.
WHEEL_RE = re.compile(r"-py3-none-")

req = urllib.request.Request(api, headers={"Accept": "application/vnd.github+json"})
try:
    with urllib.request.urlopen(req, timeout=60) as resp:
        data = json.load(resp)
except urllib.error.HTTPError as e:
    if e.code == 404:
        sys.exit(
            "nightly release not found — wait for the Nightly Build workflow on main, "
            f"or open https://github.com/{repo}/actions/workflows/nightly.yml"
        )
    raise

for asset in data.get("assets", []):
    name = asset.get("name", "")
    if not name.endswith(".whl") or not name.startswith("pvisor_sandbox-"):
        continue
    if not WHEEL_RE.search(name):
        continue
    if platform_re.search(name):
        print(asset["browser_download_url"])
        break
else:
    sys.exit(
        f"no platform wheel for {platform_re.pattern} in nightly release — "
        f"check https://github.com/{repo}/releases/tag/{tag}"
    )
PY

url="$("$PYTHON" - "$api" "$platform_re" "$REPO" "$TAG" < "$asset_helper")"

echo "Installing ${url}" >&2
"$PYTHON" -m pip install --upgrade pip
"$PYTHON" -m pip install --force-reinstall "$url"
"$PYTHON" -c "import pvisor; print('pVisor', pvisor.__version__)"

scripts_dir="$("$PYTHON" -c 'import sysconfig; print(sysconfig.get_path("scripts"))')"
binaries=(pvisor pvisor-tui pvisor-replay pvisor-daemon)
for binary in "${binaries[@]}"; do
  if [ ! -x "$scripts_dir/$binary" ]; then
    echo "error: wheel did not install executable $scripts_dir/$binary" >&2
    exit 1
  fi
  "$scripts_dir/$binary" --version
done
echo "CLI component set installed in $scripts_dir" >&2
case ":$PATH:" in
  *":$scripts_dir:"*) ;;
  *) echo "Add it to PATH: export PATH=\"$scripts_dir:\$PATH\"" >&2 ;;
esac