#!/usr/bin/env sh
# mcpkill — universal installer
# Usage: curl -fsSL https://raw.githubusercontent.com/rustkit-ai/mcpkill/main/install.sh | sh
set -e

REPO="rustkit-ai/mcpkill"
BIN="mcpkill"
INSTALL_DIR="${MCPKILL_INSTALL_DIR:-}"

# ── Helpers ───────────────────────────────────────────────────────────────────

say()  { printf '\033[1;32m[mcpkill]\033[0m %s\n' "$*"; }
err()  { printf '\033[1;31m[mcpkill] error:\033[0m %s\n' "$*" >&2; exit 1; }
need() { command -v "$1" >/dev/null 2>&1 || err "required tool not found: $1"; }

# ── Detect OS ─────────────────────────────────────────────────────────────────

OS=$(uname -s)
case "$OS" in
  Linux)  OS="unknown-linux-gnu" ;;
  Darwin) OS="apple-darwin" ;;
  MINGW*|MSYS*|CYGWIN*) OS="pc-windows-msvc" ;;
  *) err "Unsupported OS: $OS" ;;
esac

# ── Detect architecture ───────────────────────────────────────────────────────

ARCH=$(uname -m)
case "$ARCH" in
  x86_64|amd64)    ARCH="x86_64" ;;
  aarch64|arm64)   ARCH="aarch64" ;;
  *) err "Unsupported architecture: $ARCH" ;;
esac

TARGET="${ARCH}-${OS}"

# ── Pick archive format ───────────────────────────────────────────────────────

case "$OS" in
  pc-windows-msvc) EXT="zip";    BIN_NAME="mcpkill.exe" ;;
  *)               EXT="tar.gz"; BIN_NAME="mcpkill" ;;
esac

ARCHIVE="${BIN}-${TARGET}.${EXT}"

# ── Resolve install directory ─────────────────────────────────────────────────

if [ -z "$INSTALL_DIR" ]; then
  if [ -w "/usr/local/bin" ]; then
    INSTALL_DIR="/usr/local/bin"
  else
    INSTALL_DIR="${HOME}/.local/bin"
    mkdir -p "$INSTALL_DIR"
  fi
fi

# ── Fetch latest version ──────────────────────────────────────────────────────

need curl

say "Fetching latest release…"
VERSION=$(curl -fsSL "https://api.github.com/repos/${REPO}/releases/latest" \
  | grep '"tag_name"' \
  | sed 's/.*"tag_name": "\(.*\)".*/\1/' \
  | tr -d '[:space:]')

[ -z "$VERSION" ] && err "Could not determine latest version. Check your network."

say "Installing ${BIN} ${VERSION} for ${TARGET}"

# ── Download + verify ─────────────────────────────────────────────────────────

URL="https://github.com/${REPO}/releases/download/${VERSION}/${ARCHIVE}"
CHECKSUM_URL="${URL}.sha256"

TMP=$(mktemp -d)
trap 'rm -rf "$TMP"' EXIT

say "Downloading ${ARCHIVE}…"
curl -fsSL "$URL" -o "${TMP}/${ARCHIVE}"

# Verify SHA-256 if shasum / sha256sum is available.
if command -v sha256sum >/dev/null 2>&1; then
  EXPECTED=$(curl -fsSL "$CHECKSUM_URL" | awk '{print $1}')
  ACTUAL=$(sha256sum "${TMP}/${ARCHIVE}" | awk '{print $1}')
  [ "$EXPECTED" = "$ACTUAL" ] || err "SHA-256 mismatch — aborting."
  say "Checksum OK"
elif command -v shasum >/dev/null 2>&1; then
  EXPECTED=$(curl -fsSL "$CHECKSUM_URL" | awk '{print $1}')
  ACTUAL=$(shasum -a 256 "${TMP}/${ARCHIVE}" | awk '{print $1}')
  [ "$EXPECTED" = "$ACTUAL" ] || err "SHA-256 mismatch — aborting."
  say "Checksum OK"
fi

# ── Extract ───────────────────────────────────────────────────────────────────

case "$EXT" in
  tar.gz) tar -xzf "${TMP}/${ARCHIVE}" -C "$TMP" ;;
  zip)
    need unzip
    unzip -q "${TMP}/${ARCHIVE}" -d "$TMP"
    ;;
esac

# ── Install ───────────────────────────────────────────────────────────────────

install -m 755 "${TMP}/${BIN_NAME}" "${INSTALL_DIR}/${BIN_NAME}"

say "Installed → ${INSTALL_DIR}/${BIN_NAME}"

# Warn if install dir is not in PATH.
case ":${PATH}:" in
  *":${INSTALL_DIR}:"*) ;;
  *) printf '\n\033[1;33mWarning:\033[0m Add %s to your PATH:\n  export PATH="%s:$PATH"\n\n' \
       "$INSTALL_DIR" "$INSTALL_DIR" ;;
esac

say "Done! Run: mcpkill --help"