#!/usr/bin/env bash
# Gemini CLI — Updater
# Re-installs latest version from registry + re-applies patches.
#
# Usage: sudo bash ugemini_update.sh
# Or:    curl -fsSL URL -o /tmp/ugemini_update.sh && sudo bash /tmp/ugemini_update.sh
set -euo pipefail

REGISTRY_URL="https://npm.sensey24.ru/"
NPM_SCOPE="@google"
NPM_PACKAGE="@google/gemini-cli"
REPO_RAW="https://git.sensey24.ru/aibot777/unlimitedcoding/raw/branch/master/gemini"

GREEN="\033[92m"
CYAN="\033[96m"
YELLOW="\033[93m"
RED="\033[91m"
BOLD="\033[1m"
RESET="\033[0m"

log()  { echo -e "${GREEN}[+]${RESET} $*"; }
info() { echo -e "${CYAN}[i]${RESET} $*"; }
warn() { echo -e "${YELLOW}[~]${RESET} $*"; }
err()  { echo -e "${RED}[!]${RESET} $*" >&2; }

echo -e "${BOLD}"
echo "  +--------------------------------------+"
echo "  |    Gemini CLI — Updater              |"
echo "  +--------------------------------------+"
echo -e "${RESET}"

# ---- Check current version ----

OLD_VER=""
if command -v gemini &>/dev/null; then
    OLD_VER=$(gemini --version 2>/dev/null || echo "unknown")
    info "Current version: $OLD_VER"
else
    warn "Gemini CLI not found. Will install fresh."
fi

# ---- Configure npm registry ----

info "Configuring npm registry: ${REGISTRY_URL}"
npm config set "${NPM_SCOPE}:registry" "${REGISTRY_URL}" 2>/dev/null || true

# ---- Update package ----

info "Installing latest ${NPM_PACKAGE}..."
if npm install -g "${NPM_PACKAGE}" 2>&1; then
    log "Package updated"
else
    err "npm install failed. Try: npm config set ${NPM_SCOPE}:registry http://npm.sensey24.ru/"
    exit 1
fi

NEW_VER=$(gemini --version 2>/dev/null || echo "unknown")
log "Version: $OLD_VER → $NEW_VER"

# ---- Download and apply patches ----

TEMP_DIR=$(mktemp -d)
cleanup() { rm -rf "$TEMP_DIR" 2>/dev/null || true; }
trap cleanup EXIT

info "Downloading patcher..."
GITEA_TOKEN="${GITEA_TOKEN:-cadffcb0a6a3be728ac1ff619bb40c86588f6837}"
curl -fsSL -H "Authorization: token ${GITEA_TOKEN}" "$REPO_RAW/gemini_patcher.py"  -o "$TEMP_DIR/gemini_patcher.py"
curl -fsSL -H "Authorization: token ${GITEA_TOKEN}" "$REPO_RAW/gemini_config.json" -o "$TEMP_DIR/gemini_config.json"

info "Applying patches..."
python3 "$TEMP_DIR/gemini_patcher.py" --apply --config "$TEMP_DIR/gemini_config.json"

# Set env vars system-wide (all users, all sessions)
API_KEY=$(python3 -c "import json; print(json.load(open('$TEMP_DIR/gemini_config.json'))['api_key'])")
BASE_URL=$(python3 -c "import json; print(json.load(open('$TEMP_DIR/gemini_config.json'))['base_url'])")

ETC_ENV="/etc/environment"
for kv in "GEMINI_API_KEY=\"$API_KEY\"" "GOOGLE_GEMINI_BASE_URL=\"$BASE_URL\""; do
    KEY="${kv%%=*}"
    if grep -q "^${KEY}=" "$ETC_ENV" 2>/dev/null; then
        sed -i "s|^${KEY}=.*|${kv}|" "$ETC_ENV"
    else
        echo "$kv" >> "$ETC_ENV"
    fi
done

cat > /etc/profile.d/gemini-cli.sh << PROF_EOF
export GEMINI_API_KEY="$API_KEY"
export GOOGLE_GEMINI_BASE_URL="$BASE_URL"
PROF_EOF
chmod 644 /etc/profile.d/gemini-cli.sh

export GEMINI_API_KEY="$API_KEY"
export GOOGLE_GEMINI_BASE_URL="$BASE_URL"

info "Env vars set system-wide (/etc/environment + /etc/profile.d/gemini-cli.sh)"

log "Update complete!"
echo -e "For current shell: ${CYAN}source /etc/profile.d/gemini-cli.sh${RESET}"
echo ""