#!/bin/bash

# HTTP File Runner - Installation Script
# This script downloads and installs the latest release of httprunner

set -euo pipefail

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Configuration
GITHUB_REPO="christianhelle/httprunner"
INSTALL_DIR="${INSTALL_DIR:-/usr/local/bin}"
BINARY_NAME="httprunner"

# Functions
log_info() {
    echo -e "${BLUE}ℹ️  $1${NC}" >&2
}

log_success() {
    echo -e "${GREEN}✅ $1${NC}" >&2
}

log_warning() {
    echo -e "${YELLOW}⚠️  $1${NC}" >&2
}

log_error() {
    echo -e "${RED}❌ $1${NC}" >&2
}

detect_platform() {
    local os=$(uname -s | tr '[:upper:]' '[:lower:]')
    local arch=$(uname -m)
    
    case "$os" in
        linux*)
            os="linux"
            ;;
        darwin*)
            os="macos"
            ;;
        *)
            log_error "Unsupported operating system: $os"
            exit 1
            ;;
    esac
    
    case "$arch" in
        x86_64|amd64)
            arch="x86_64"
            ;;
        aarch64|arm64)
            arch="aarch64"
            ;;
    esac
    
    echo "${os}-${arch}"
}

check_dependencies() {
    local deps=("curl" "tar")
    
    for dep in "${deps[@]}"; do
        if ! command -v "$dep" >/dev/null 2>&1; then
            log_error "Required dependency '$dep' not found. Please install it first."
            exit 1
        fi
    done
}

get_latest_release() {
    log_info "Fetching latest release information..."
    local api_url="https://api.github.com/repos/$GITHUB_REPO/releases/latest"
    
    if ! curl -s "$api_url" | grep -o '"tag_name": "[^"]*' | grep -o '[^"]*$'; then
        log_error "Failed to fetch release information"
        exit 1
    fi
}

download_and_install() {
    local platform="$1"
    local version="$2"
    local archive_name="httprunner-${platform}.tar.gz"
    local download_url="https://github.com/$GITHUB_REPO/releases/download/$version/$archive_name"
    local temp_dir=$(mktemp -d)
    
    log_info "Downloading httprunner $version for $platform..."
    
    if ! curl -L -o "$temp_dir/$archive_name" "$download_url"; then
        log_error "Failed to download httprunner"
        rm -rf "$temp_dir"
        exit 1
    fi
    
    log_info "Extracting archive..."
    if ! tar -xzf "$temp_dir/$archive_name" -C "$temp_dir"; then
        log_error "Failed to extract archive"
        rm -rf "$temp_dir"
        exit 1
    fi
    
    log_info "Installing to $INSTALL_DIR..."
    
    # Check if we need sudo
    if [ ! -w "$INSTALL_DIR" ]; then
        if command -v sudo >/dev/null 2>&1; then
            log_warning "Installing with sudo (directory not writable by current user)"
            sudo cp "$temp_dir/$BINARY_NAME" "$INSTALL_DIR/"
            sudo chmod +x "$INSTALL_DIR/$BINARY_NAME"
        else
            log_error "Cannot write to $INSTALL_DIR and sudo is not available"
            log_info "Try setting INSTALL_DIR to a writable directory:"
            log_info "  INSTALL_DIR=\$HOME/.local/bin curl -fsSL https://christianhelle.com/httprunner/install | bash"
            rm -rf "$temp_dir"
            exit 1
        fi
    else
        cp "$temp_dir/$BINARY_NAME" "$INSTALL_DIR/"
        chmod +x "$INSTALL_DIR/$BINARY_NAME"
    fi
    
    # Cleanup
    rm -rf "$temp_dir"
    
    log_success "httprunner $version installed successfully!"
}

verify_installation() {
    if command -v "$BINARY_NAME" >/dev/null 2>&1; then
        local installed_version=$($BINARY_NAME --version 2>/dev/null | head -n1 || echo "unknown")
        log_success "Installation verified: $installed_version"
        log_info "You can now run: $BINARY_NAME --help"
    else
        log_warning "Binary installed but not found in PATH"
        log_info "Make sure $INSTALL_DIR is in your PATH"
        log_info "Add this to your shell profile: export PATH=\"$INSTALL_DIR:\$PATH\""
    fi
}

show_usage() {
    echo "HTTP File Runner Installation Script"
    echo ""
    echo "Usage: $0 [OPTIONS]"
    echo ""
    echo "Options:"
    echo "  -h, --help      Show this help message"
    echo "  -d, --dir DIR   Set installation directory (default: /usr/local/bin)"
    echo ""
    echo "Environment variables:"
    echo "  INSTALL_DIR     Installation directory (default: /usr/local/bin)"
    echo ""
    echo "Examples:"
    echo "  # Install to default location"
    echo "  curl -fsSL https://christianhelle.com/httprunner/install | bash"
    echo ""
    echo "  # Install to custom directory"
    echo "  INSTALL_DIR=\$HOME/.local/bin curl -fsSL https://christianhelle.com/httprunner/install | bash"
    echo ""
    echo "  # Install to custom directory using flag"
    echo "  curl -fsSL https://christianhelle.com/httprunner/install | bash -s -- --dir \$HOME/.local/bin"
}

main() {
    # Parse arguments
    while [[ $# -gt 0 ]]; do
        case $1 in
            -h|--help)
                show_usage
                exit 0
                ;;
            -d|--dir)
                INSTALL_DIR="$2"
                shift 2
                ;;
            *)
                log_error "Unknown option: $1"
                show_usage
                exit 1
                ;;
        esac
    done
    
    log_info "Starting HTTP File Runner installation..."
    log_info "Target directory: $INSTALL_DIR"
    
    # Detect platform
    local platform=$(detect_platform)
    log_info "Detected platform: $platform"
    
    # Check dependencies
    check_dependencies
    
    # Get latest release
    local version=$(get_latest_release)
    log_info "Latest version: $version"
    
    # Create install directory if it doesn't exist
    if [ ! -d "$INSTALL_DIR" ]; then
        log_info "Creating installation directory: $INSTALL_DIR"
        if ! mkdir -p "$INSTALL_DIR" 2>/dev/null; then
            if command -v sudo >/dev/null 2>&1; then
                sudo mkdir -p "$INSTALL_DIR"
            else
                log_error "Cannot create directory $INSTALL_DIR"
                exit 1
            fi
        fi
    fi
    
    # Download and install
    download_and_install "$platform" "$version"
    
    # Verify installation
    verify_installation
    
    echo ""
    log_success "🎉 Installation complete!"
    log_info "Get started with: $BINARY_NAME --help"
    log_info "Documentation: https://christianhelle.com/httprunner/"
}

# Run main function with all arguments
main "$@"