#!/usr/bin/env bash
# Tool: uv
# DisplayName: uv
# Description: Fast Python package installer and resolver
# Strategy: binary
#
# uv Installer Script
#

set -euo pipefail

TMP_DIR="$(make_temp_dir)"
cleanup() {
    rm -rf "$TMP_DIR"
}
trap cleanup EXIT

install_uv() {
    if has_command uv || [ -f "$HOME/.local/bin/uv" ]; then
        if ! confirm "uv is already installed. Reinstall/Upgrade?"; then
            log_info "Skipping uv installation."
            return
        fi
    fi

    # Detect architecture
    local raw_arch
    raw_arch=$(detect_arch)
    local arch=""
    case "$raw_arch" in
        x86_64) arch="x86_64" ;;
        arm64)  arch="aarch64" ;;
        *)      log_error "Unsupported Linux architecture: $raw_arch"; exit 1 ;;
    esac

    # Determine target based on libc
    local target=""
    if ldd --version 2>&1 | grep -q "GLIBC"; then
        target="${arch}-unknown-linux-gnu"
    else
        target="${arch}-unknown-linux-musl"
    fi

    log_info "Fetching latest uv version from GitHub..."
    local latest_tag=""
    latest_tag=$(github_get_latest_release "astral-sh/uv")

    if [ -z "$latest_tag" ]; then
        latest_tag="latest"
    fi

    log_info "Downloading uv ${latest_tag}..."
    local archive="$TMP_DIR/uv.tar.gz"
    github_download_asset "astral-sh/uv" "$latest_tag" "uv-${target}\.tar\.gz" "$archive"

    # Extract the binaries
    log_info "Extracting uv binaries..."
    tar -xzf "$archive" --strip-components 1 -C "$TMP_DIR"

    # Install to ~/.local/bin
    local target_dir="$HOME/.local/bin"
    mkdir -p "$target_dir"
    log_info "Installing uv and uvx to $target_dir..."
    cp "$TMP_DIR/uv" "$target_dir/uv"
    cp "$TMP_DIR/uvx" "$target_dir/uvx"
    chmod +x "$target_dir/uv" "$target_dir/uvx"
    track_file "$target_dir/uv"
    track_file "$target_dir/uvx"
    register_tool "uv" "binary" "$latest_tag" "github:astral-sh/uv"
}

configure_shell() {
    # Add ~/.local/bin to PATH for the current process
    export PATH="$HOME/.local/bin:$PATH"


    write_env_snippet "local-bin" 'export PATH="$HOME/.local/bin:$PATH"'
    write_env_snippet "uv" 'eval "$(uv generate-shell-completion bash)"'
}

main() {
    install_uv
    configure_shell

    echo
    log_success "uv installation and configuration complete."
}

main "$@"