#!/bin/bash
# Drift CLI Installer - Hosted on godrift.ai
# This script installs the latest Drift CLI from GitHub releases

set -e

# Configuration
REPO="drift-tech/drift-releases"
CLI_NAME="drift"
GITHUB_API="https://api.github.com/repos/${REPO}/releases/latest"
ANALYTICS_ENDPOINT="https://analytics-service-267976266267.us-central1.run.app"

# Colors
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m'

# Generate unique session ID for this installation
INSTALL_SESSION_ID="install_$(date +%s%N)_$RANDOM"

# Function to track analytics events (non-blocking, silent)
track_event() {
    local event_name="$1"
    local success="$2"
    local error_msg="$3"

    local payload=""

    if [ -n "$success" ] && [ -n "$error_msg" ]; then
        payload="{\"action\": \"track\", \"distinct_id\": \"$INSTALL_SESSION_ID\", \"event\": \"$event_name\", \"properties\": {\"os\": \"Linux\", \"arch\": \"${ARCH_DEB:-unknown}\", \"version\": \"${VERSION:-unknown}\", \"success\": $success, \"error\": \"$error_msg\"}}"
    elif [ -n "$success" ]; then
        payload="{\"action\": \"track\", \"distinct_id\": \"$INSTALL_SESSION_ID\", \"event\": \"$event_name\", \"properties\": {\"os\": \"Linux\", \"arch\": \"${ARCH_DEB:-unknown}\", \"version\": \"${VERSION:-unknown}\", \"success\": $success}}"
    else
        payload="{\"action\": \"track\", \"distinct_id\": \"$INSTALL_SESSION_ID\", \"event\": \"$event_name\", \"properties\": {\"os\": \"Linux\", \"arch\": \"${ARCH_DEB:-unknown}\", \"version\": \"${VERSION:-unknown}\"}}"
    fi

    # Send to analytics endpoint (fire and forget)
    (curl -s -X POST "$ANALYTICS_ENDPOINT" \
        -H "Content-Type: application/json" \
        -d "$payload" \
        >/dev/null 2>&1) &
}

# ASCII Art
echo ""
echo "  ╔══════════════════════════════════════╗"
echo "  ║         DRIFT CLI INSTALLER          ║"
echo "  ║     AI Copilot for Simulations       ║"
echo "  ║           godrift.ai                 ║"
echo "  ╚══════════════════════════════════════╝"
echo ""

echo "Installing Drift CLI..."

# Track installation started
track_event "install_started" "" ""

# Check if Linux
if [ "$(uname)" != "Linux" ]; then
    echo -e "${RED}Error: This installer is only for Linux systems${NC}"
    echo "For other platforms, visit: https://godrift.ai"
    track_event "install_incomplete" "false" "not_linux"
    exit 1
fi

# Detect architecture
ARCH=$(uname -m)
case $ARCH in
    x86_64)
        ARCH_DEB="amd64"
        ;;
    aarch64|arm64)
        ARCH_DEB="arm64"
        ;;
    *)
        echo -e "${RED}Error: Unsupported architecture: $ARCH${NC}"
        echo "Supported: x64, arm64"
        track_event "install_incomplete" "false" "unsupported_arch"
        exit 1
        ;;
esac

echo "Detected: Linux $ARCH_DEB"

# Check for curl or wget
if command -v curl >/dev/null 2>&1; then
    DOWNLOAD_CMD="curl -fsSL"
    DOWNLOAD_OUTPUT="-o"
elif command -v wget >/dev/null 2>&1; then
    DOWNLOAD_CMD="wget -q"
    DOWNLOAD_OUTPUT="-O"
else
    echo -e "${RED}Error: curl or wget is required${NC}"
    echo "Install with: sudo apt-get install curl"
    track_event "install_incomplete" "false" "no_curl_wget"
    exit 1
fi

# Get latest release from GitHub
echo "Fetching latest version..."
if command -v curl >/dev/null 2>&1; then
    RELEASE_DATA=$(curl -s "$GITHUB_API")
else
    RELEASE_DATA=$(wget -qO- "$GITHUB_API")
fi

# Parse version
VERSION=$(echo "$RELEASE_DATA" | grep -oP '"tag_name": "\K[^"]*' | head -1)
if [ -z "$VERSION" ]; then
    echo -e "${RED}Error: Could not fetch latest version${NC}"
    echo "Please check: https://github.com/$REPO/releases"
    track_event "install_incomplete" "false" "version_fetch_failed"
    exit 1
fi

echo "Latest version: $VERSION"

# Find the correct .deb file URL
DEB_URL=$(echo "$RELEASE_DATA" | grep -oP '"browser_download_url": "\K[^"]*\.deb' | grep "$ARCH_DEB" | head -1)

if [ -z "$DEB_URL" ]; then
    echo -e "${RED}Error: No .deb package found for $ARCH_DEB${NC}"
    echo "Available releases: https://github.com/$REPO/releases"
    track_event "install_incomplete" "false" "no_deb_package"
    exit 1
fi

# Download the .deb package
echo "Downloading Drift CLI $VERSION..."
TMP_DEB="/tmp/drift_${VERSION}_${ARCH_DEB}.deb"
$DOWNLOAD_CMD "$DEB_URL" $DOWNLOAD_OUTPUT "$TMP_DEB"

if [ ! -f "$TMP_DEB" ]; then
    echo -e "${RED}Error: Download failed${NC}"
    track_event "install_incomplete" "false" "download_failed"
    exit 1
fi

# Install the package
echo "Installing package..."
if command -v dpkg >/dev/null 2>&1; then
    # Debian/Ubuntu
    sudo dpkg -i "$TMP_DEB" 2>/dev/null || {
        echo "Installing dependencies..."
        sudo apt-get update >/dev/null 2>&1
        sudo apt-get install -f -y >/dev/null 2>&1
        sudo dpkg -i "$TMP_DEB"
    }
elif command -v apt >/dev/null 2>&1; then
    # Modern Debian/Ubuntu
    sudo apt install "$TMP_DEB" -y
else
    echo -e "${RED}Error: No supported package manager found${NC}"
    echo "Manual installation required: sudo dpkg -i $TMP_DEB"
    track_event "install_incomplete" "false" "no_package_manager"
    exit 1
fi

# Cleanup
rm -f "$TMP_DEB"

# Refresh shell command cache
hash -r 2>/dev/null || true

# Verify installation
if command -v drift >/dev/null 2>&1; then
    INSTALLED_VERSION=$(drift --version 2>/dev/null || echo "unknown")
    echo ""
    echo -e "${GREEN}╔════════════════════════════════════════╗${NC}"
    echo -e "${GREEN}║   ✓ Drift CLI installed successfully!  ║${NC}"
    echo -e "${GREEN}╚════════════════════════════════════════╝${NC}"
    echo ""
    echo "Version: $INSTALLED_VERSION"
    echo "Location: $(which drift)"
    echo ""
    echo -e "${GREEN}✨ Drift is ready to use immediately!${NC}"
    echo ""
    echo "Get started with:"
    echo "  drift --help          Show help"
    echo "  drift                 Start interactive mode"
    echo ""
    # Track successful installation
    track_event "install_completed" "true" ""
else
    echo -e "${YELLOW}⚠️  Drift installed but not detected in current shell${NC}"
    echo ""
    echo "To use drift in this terminal session, run one of:"
    echo -e "  ${GREEN}hash -r${NC}                    (refresh command cache)"
    echo -e "  ${GREEN}exec \$SHELL${NC}                (restart shell)"
    echo -e "  ${GREEN}source ~/.bashrc${NC}           (reload shell config)"
    echo ""
    echo "Or simply open a new terminal - drift will work automatically!"
    echo ""
    # Track installation (success but not detected in shell)
    track_event "install_completed" "true" "not_in_path"
fi