#!/usr/bin/env bash
# McpMux Installer
# Usage: curl -fsSL https://install.mcpmux.com | bash
#    or: curl -fsSL https://install.mcpmux.com | bash -s -- --version 0.0.12
set -euo pipefail

GITHUB_REPO="mcpmux/mcp-mux"
VERSION=""
SKIP_VERIFY=false

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

info()  { printf '\033[1;34m%s\033[0m\n' "$*"; }
ok()    { printf '\033[1;32m%s\033[0m\n' "$*"; }
warn()  { printf '\033[1;33m%s\033[0m\n' "$*" >&2; }
fail()  { printf '\033[1;31mError: %s\033[0m\n' "$*" >&2; exit 1; }

need() {
  command -v "$1" &>/dev/null || fail "'$1' is required but not installed."
}

# ── Parse args ───────────────────────────────────────────────────────

while [[ $# -gt 0 ]]; do
  case "$1" in
    --version|-v)  VERSION="$2"; shift 2 ;;
    --skip-verify) SKIP_VERIFY=true; shift ;;
    --help|-h)
      echo "Usage: install.sh [--version VERSION] [--skip-verify]"
      exit 0
      ;;
    *) fail "Unknown option: $1" ;;
  esac
done

# ── Prerequisites ────────────────────────────────────────────────────

need curl

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

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

info "Detected architecture: $ARCH"

# ── Resolve version ─────────────────────────────────────────────────

if [[ -z "$VERSION" ]]; then
  info "Fetching latest release..."
  VERSION=$(curl -fsSL "https://api.github.com/repos/${GITHUB_REPO}/releases/latest" \
    | grep '"tag_name"' | head -1 | sed 's/.*"v\([^"]*\)".*/\1/')
  [[ -n "$VERSION" ]] || fail "Could not determine latest version"
fi

info "Installing McpMux v${VERSION}"

# ── GPG verification helper ─────────────────────────────────────────

verify_signature() {
  local file="$1"
  if [[ "$SKIP_VERIFY" == "true" ]]; then
    warn "Skipping signature verification (--skip-verify)"
    return 0
  fi
  if ! command -v gpg &>/dev/null; then
    warn "gpg not found — skipping signature verification"
    warn "Install gnupg and re-run, or use --skip-verify to suppress this warning"
    return 0
  fi

  local sig_url="https://github.com/${GITHUB_REPO}/releases/download/v${VERSION}/$(basename "$file").sig"
  local sig_file="${file}.sig"

  if curl -fsSL -o "$sig_file" "$sig_url" 2>/dev/null; then
    # Import McpMux signing key if not already present
    if ! gpg --list-keys hello@mcpmux.com &>/dev/null 2>&1; then
      curl -fsSL https://apt.mcpmux.com/key.gpg | gpg --import --quiet 2>/dev/null || true
    fi
    if gpg --verify "$sig_file" "$file" 2>/dev/null; then
      ok "Signature verified"
    else
      warn "Signature verification failed — package may not be signed yet"
    fi
    rm -f "$sig_file"
  else
    warn "No signature file found — skipping verification"
  fi
}

# ── Install via detected package manager ─────────────────────────────

install_deb() {
  local deb_name="mcpmux_${VERSION}_${ARCH}.deb"
  local deb_url="https://github.com/${GITHUB_REPO}/releases/download/v${VERSION}/${deb_name}"
  local tmp="/tmp/mcpmux.deb"

  info "Downloading .deb package..."
  curl -fsSL -o "$tmp" "$deb_url" || fail "Download failed: $deb_url"
  verify_signature "$tmp"

  info "Installing..."
  sudo apt-get install -y "$tmp"
  rm -f "$tmp"

  echo ""
  info "Tip: For automatic updates via apt, add the McpMux repository:"
  echo "  curl -fsSL https://install.mcpmux.com/apt | sudo bash"
}

install_rpm() {
  local rpm_name="mcpmux-${VERSION}-1.${ARCH}.rpm"
  local rpm_url="https://github.com/${GITHUB_REPO}/releases/download/v${VERSION}/${rpm_name}"
  local tmp="/tmp/mcpmux.rpm"

  info "Downloading .rpm package..."
  curl -fsSL -o "$tmp" "$rpm_url" || fail "Download failed: $rpm_url"
  verify_signature "$tmp"

  info "Installing..."
  sudo dnf install -y "$tmp"
  rm -f "$tmp"
}

install_pacman() {
  if command -v yay &>/dev/null; then
    info "Installing from AUR via yay..."
    yay -S mcpmux-bin
  elif command -v paru &>/dev/null; then
    info "Installing from AUR via paru..."
    paru -S mcpmux-bin
  else
    warn "No AUR helper found (yay or paru)."
    echo ""
    echo "Option 1: Install an AUR helper first:"
    echo "  sudo pacman -S --needed git base-devel"
    echo "  git clone https://aur.archlinux.org/yay-bin.git && cd yay-bin && makepkg -si"
    echo "  yay -S mcpmux-bin"
    echo ""
    echo "Option 2: Download the AppImage:"
    echo "  https://github.com/${GITHUB_REPO}/releases/latest"
    exit 1
  fi
}

install_appimage() {
  local appimage_name="McpMux_${VERSION}_${ARCH}.AppImage"
  local appimage_url="https://github.com/${GITHUB_REPO}/releases/download/v${VERSION}/${appimage_name}"
  local install_dir="${HOME}/.local/bin"
  local install_path="${install_dir}/McpMux.AppImage"

  info "No supported package manager found — installing AppImage..."
  mkdir -p "$install_dir"

  info "Downloading AppImage..."
  curl -fsSL -o "$install_path" "$appimage_url" || fail "Download failed: $appimage_url"
  verify_signature "$install_path"
  chmod +x "$install_path"

  ok "Installed to ${install_path}"

  # Check if ~/.local/bin is in PATH
  if ! echo "$PATH" | tr ':' '\n' | grep -qx "$install_dir"; then
    echo ""
    warn "~/.local/bin is not in your PATH. Add it with:"
    echo '  echo '\''export PATH="$HOME/.local/bin:$PATH"'\'' >> ~/.bashrc'
    echo '  source ~/.bashrc'
  fi
}

# ── Main ─────────────────────────────────────────────────────────────

# Check if APT repo is already configured (prefer managed updates)
if [[ -f /etc/apt/sources.list.d/mcpmux.list ]]; then
  info "McpMux APT repository detected — using apt..."
  sudo apt-get update && sudo apt-get install -y mcpmux
elif command -v apt-get &>/dev/null; then
  install_deb
elif command -v dnf &>/dev/null; then
  install_rpm
elif command -v pacman &>/dev/null; then
  install_pacman
else
  install_appimage
fi

echo ""
ok "McpMux installed successfully!"
echo "Run 'mcpmux' to start."