#!/usr/bin/env bash
# BibiGPT Installer — universal install script
# Usage: curl -fsSL https://bibigpt.co/install.sh | bash
set -euo pipefail

LATEST_JSON_CN="https://bibigpt-apps.oss-cn-beijing.aliyuncs.com/desktop-releases/latest.json"
LATEST_JSON_INTL="https://bibigpt-apps.chatvid.ai/desktop-releases/latest.json"
BIN_DIR="${BIBI_INSTALL_DIR:-/usr/local/bin}"

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' "$*"; }
err()   { printf '\033[1;31mError: %s\033[0m\n' "$*" >&2; exit 1; }

# ── Detect OS & arch ──────────────────────────────────────────────────
OS="$(uname -s)"
ARCH="$(uname -m)"

case "$OS" in
  Darwin)
    case "$ARCH" in
      arm64)  PLATFORM="darwin-aarch64" ;;
      x86_64) PLATFORM="darwin-x86_64" ;;
      *) err "Unsupported macOS architecture: $ARCH" ;;
    esac
    ;;
  Linux)
    case "$ARCH" in
      x86_64|amd64) PLATFORM="linux-x86_64" ;;
      *) err "Unsupported Linux architecture: $ARCH (only x86_64 is supported)" ;;
    esac
    ;;
  MINGW*|MSYS*|CYGWIN*)
    PLATFORM="windows-x86_64"
    ;;
  *) err "Unsupported OS: $OS" ;;
esac

info "Detected platform: $PLATFORM"

# ── Fetch latest version ─────────────────────────────────────────────
info "Fetching latest version info..."

VERSION=""
DOWNLOAD_URL=""
BACKUP_URL=""
for json_url in "$LATEST_JSON_CN" "$LATEST_JSON_INTL"; do
  JSON="$(curl -sf --connect-timeout 5 "$json_url" 2>/dev/null || true)"
  if [ -n "$JSON" ]; then
    VERSION="$(echo "$JSON" | grep -oE '"version"\s*:\s*"[^"]*"' | head -1 | cut -d'"' -f4)"
    PLATFORM_BLOCK="$(echo "$JSON" | tr -d '\n' | grep -oE "\"$PLATFORM\"\s*:\s*\{[^}]*\}" || true)"
    DOWNLOAD_URL="$(echo "$PLATFORM_BLOCK" | grep -oE '"url"\s*:\s*"[^"]*"' | head -1 | cut -d'"' -f4)"
    BACKUP_URL="$(echo "$PLATFORM_BLOCK" | grep -oE '"backupUrl"\s*:\s*"[^"]*"' | head -1 | cut -d'"' -f4)"
    [ -n "$VERSION" ] && break
  fi
done

[ -z "$VERSION" ] && err "Failed to fetch version info. Check your network connection."
info "Latest version: $VERSION"

# ── Temp dir & cleanup ───────────────────────────────────────────────
WORK_DIR="$(mktemp -d)"
cleanup() { rm -rf "$WORK_DIR"; }
trap cleanup EXIT

download() {
  local url="$1" dest="$2"
  [ -n "$url" ] || return 1
  curl -fL --progress-bar "$url" -o "$dest"
}

download_with_fallback() {
  local dest="$1"
  if [ -n "$DOWNLOAD_URL" ] && download "$DOWNLOAD_URL" "$dest"; then
    return 0
  fi
  if [ -n "$BACKUP_URL" ] && [ "$BACKUP_URL" != "$DOWNLOAD_URL" ]; then
    warn "Primary download failed, trying backup..."
    if download "$BACKUP_URL" "$dest"; then
      return 0
    fi
  fi
  err "Download failed for $PLATFORM. See https://bibigpt.co/apps/desktop"
}

print_done() {
  echo ""
  echo "  bibi --version          # verify"
  echo "  bibi summarize \"<URL>\"  # summarize a video"
  echo ""
  echo "Getting started:"
  echo "  bibi auth login          # open browser to log in"
  if [[ "$OS" == "Darwin" ]]; then
    echo "  open /Applications/BibiGPT.app  # or launch the desktop app to log in"
  fi
  echo ""
  echo "Alternative: set your API token manually:"
  echo "  export BIBI_API_TOKEN=<token>  # get one at https://bibigpt.co/settings"
}

# ── Windows (Git Bash / MSYS / Cygwin) ───────────────────────────────
if [[ "$OS" == MINGW* || "$OS" == MSYS* || "$OS" == CYGWIN* ]]; then
  # Prefer winget if available
  if command -v winget &>/dev/null; then
    info "Installing via winget..."
    winget install BibiGPT --source winget
    ok "BibiGPT $VERSION installed via winget!"
    print_done
    exit 0
  fi

  # No winget — download .exe installer and run it
  warn "winget not found, downloading installer..."
  FILENAME="BibiGPT-${VERSION}-${PLATFORM}.exe"
  INSTALLER="${WORK_DIR}/${FILENAME}"

  [ -z "$DOWNLOAD_URL" ] && DOWNLOAD_URL="https://bibigpt-apps.oss-cn-beijing.aliyuncs.com/desktop-releases/${FILENAME}"

  info "Downloading ${FILENAME}..."
  download_with_fallback "$INSTALLER"

  info "Launching installer..."
  # Convert to Windows path for native execution
  WIN_PATH="$(cygpath -w "$INSTALLER" 2>/dev/null || echo "$INSTALLER")"
  cmd //c start "" "$WIN_PATH" || start "$WIN_PATH" 2>/dev/null || err "Failed to launch installer. Run manually: $WIN_PATH"

  ok "Installer launched! Follow the setup wizard to complete installation."
  echo ""
  echo "After installation, open a new terminal and run:"
  print_done
  exit 0
fi

# ── macOS ─────────────────────────────────────────────────────────────
if [[ "$OS" == "Darwin" ]]; then
  # Prefer Homebrew if available (manages updates, quarantine, cleanup)
  if command -v brew &>/dev/null; then
    if brew list --cask jimmylv/bibigpt/bibigpt &>/dev/null && [ -d "/Applications/BibiGPT.app" ]; then
      info "Upgrading via Homebrew..."
      brew upgrade --cask jimmylv/bibigpt/bibigpt
    else
      info "Installing via Homebrew..."
      brew install --cask jimmylv/bibigpt/bibigpt --force
    fi
    ok "BibiGPT $VERSION installed via Homebrew!"
    print_done
    exit 0
  fi

  # No Homebrew — direct .app install (like Deno, Bun, rustup)
  warn "Homebrew not found, installing directly..."

  FILENAME="BibiGPT-${VERSION}-${PLATFORM}.app.tar.gz"
  TARBALL="${WORK_DIR}/${FILENAME}"

  [ -z "$DOWNLOAD_URL" ] && DOWNLOAD_URL="https://bibigpt-apps.oss-cn-beijing.aliyuncs.com/desktop-releases/${FILENAME}"

  info "Downloading ${FILENAME}..."
  download_with_fallback "$TARBALL"

  info "Extracting..."
  tar -xzf "$TARBALL" -C "$WORK_DIR"

  APP_PATH="${WORK_DIR}/BibiGPT.app"
  [ -d "$APP_PATH" ] || err "Extraction failed: BibiGPT.app not found in archive"

  # Remove quarantine flag (download from internet triggers Gatekeeper)
  xattr -rd com.apple.quarantine "$APP_PATH" 2>/dev/null || true

  # Move to /Applications (overwrite if exists)
  DEST="/Applications/BibiGPT.app"
  if [ -d "$DEST" ]; then
    warn "Replacing existing ${DEST}..."
    if [ -w "$DEST" ]; then rm -rf "$DEST"; else sudo rm -rf "$DEST"; fi
  fi
  if [ -w "/Applications" ]; then mv "$APP_PATH" "$DEST"; else sudo mv "$APP_PATH" "$DEST"; fi

  # Symlink CLI binary — same path Homebrew cask uses
  CLI_BIN="${DEST}/Contents/MacOS/BibiGPT"
  if [ -x "$CLI_BIN" ]; then
    info "Creating CLI symlink: ${BIN_DIR}/bibi"
    if [ -w "$BIN_DIR" ] || mkdir -p "$BIN_DIR" 2>/dev/null; then
      ln -sf "$CLI_BIN" "${BIN_DIR}/bibi"
    else
      sudo mkdir -p "$BIN_DIR"
      sudo ln -sf "$CLI_BIN" "${BIN_DIR}/bibi"
    fi
  fi

  ok "BibiGPT $VERSION installed to /Applications/BibiGPT.app"
  echo ""
  echo "Tip: Install Homebrew (https://brew.sh) for automatic updates."
  print_done
  exit 0
fi

# ── Linux ─────────────────────────────────────────────────────────────
[ -z "$DOWNLOAD_URL" ] && [ -z "$BACKUP_URL" ] && \
  err "No Linux build is currently published. See https://bibigpt.co/apps/desktop"

FILENAME="BibiGPT-${VERSION}-${PLATFORM}.AppImage"
TMPFILE="${WORK_DIR}/${FILENAME}"

info "Downloading ${FILENAME}..."
download_with_fallback "$TMPFILE"
chmod +x "$TMPFILE"

# Install CLI binary
info "Installing to ${BIN_DIR}/bibi ..."
if [ -w "$BIN_DIR" ] || mkdir -p "$BIN_DIR" 2>/dev/null; then
  mv "$TMPFILE" "${BIN_DIR}/bibi"
else
  sudo mkdir -p "$BIN_DIR"
  sudo mv "$TMPFILE" "${BIN_DIR}/bibi"
fi

# Create .desktop entry for app launchers (freedesktop standard)
DESKTOP_DIR="${XDG_DATA_HOME:-$HOME/.local/share}/applications"
mkdir -p "$DESKTOP_DIR"
cat > "${DESKTOP_DIR}/bibigpt.desktop" <<DESKTOP
[Desktop Entry]
Name=BibiGPT
Comment=AI Video & Audio Summarizer
Exec=${BIN_DIR}/bibi %u
Icon=bibigpt
Terminal=false
Type=Application
Categories=Utility;AudioVideo;
MimeType=x-scheme-handler/bibigpt;
DESKTOP
# Update desktop database if available
command -v update-desktop-database &>/dev/null && update-desktop-database "$DESKTOP_DIR" 2>/dev/null || true

ok "BibiGPT $VERSION installed successfully!"
print_done