#!/usr/bin/env bash
# Tool: pnpm
# DisplayName: Pnpm
# Description: Install pnpm package manager
#
# pnpm Installer Script
#
# Installs pnpm (fast, disk-space-efficient package manager for Node.js).
# Supports glibc and musl (Alpine) Linux on x86_64 and arm64.
#
# Linux runtime requirements:
#   - glibc 2.27+ and libatomic.so.1 (for glibc builds)
#   - Debian/Ubuntu: apt-get install -y libatomic1
#   - Fedora/RHEL:   dnf install -y libatomic
#
# Docker usage:
#   curl -fsSL https://get.pnpm.io/install.sh | ENV="$HOME/.bashrc" SHELL="$(which bash)" bash -
#

# Run metascript to check if the shell is bash and load libraries
PARENT_DIR="$(dirname "$0")/.."
METASCRIPT_LOCAL="$PARENT_DIR/bootstrap.sh"
METASCRIPT_URL="https://git.adityagupta.dev/sortedcord/bootstrap/raw/branch/master/bootstrap.sh"

if [ -f "$METASCRIPT_LOCAL" ]; then
    . "$METASCRIPT_LOCAL"
else
    if command -v curl >/dev/null 2>&1; then
        eval "$(curl -fsSL "$METASCRIPT_URL")"
    else
        echo "Error: curl is not installed to fetch bootstrap.sh." >&2
        exit 1
    fi
fi

set -euo pipefail

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

# ─── Helper Functions ─────────────────────────────────────────────────

download() {
        curl -fsSL "$1"
}

is_glibc_compatible() {
    getconf GNU_LIBC_VERSION >/dev/null 2>&1 || ldd --version >/dev/null 2>&1 || return 1
}

# Detect libc suffix — empty for glibc, "-musl" for musl-based distros
detect_libc_suffix() {
    if ! is_glibc_compatible; then
        printf -- '-musl'
    fi
}

# pnpm v11.0.0-rc.3 renamed release assets. Older versions use legacy names.
use_legacy_assets() {
    local version="$1"
    local major
    major="$(echo "$version" | cut -d. -f1)"
    if [ "$major" -lt 11 ] 2>/dev/null; then
        return 0
    fi
    case "$version" in
        11.0.0-rc.1|11.0.0-rc.2) return 0 ;;
        *) return 1 ;;
    esac
}

# Legacy asset basename for pre-v11.0.0-rc.3 releases
legacy_asset_basename() {
    local arch libc_suffix
    arch="$1"
    libc_suffix="$2"
    if [ -n "$libc_suffix" ]; then
        printf 'pnpm-linuxstatic-%s' "$arch"
    else
        printf 'pnpm-linux-%s' "$arch"
    fi
}

# Release-page asset basename (without extension)
asset_basename() {
    local version arch libc_suffix
    version="$1"
    arch="$2"
    libc_suffix="$3"
    if use_legacy_assets "$version"; then
        legacy_asset_basename "$arch" "$libc_suffix"
    else
        printf 'pnpm-linux-%s%s' "$arch" "$libc_suffix"
    fi
}

# Map system arch to pnpm's naming (x64 / arm64)
detect_pnpm_arch() {
    local arch
    arch="$(uname -m | tr '[:upper:]' '[:lower:]')"

    case "${arch}" in
        x86_64 | amd64) arch="x64" ;;
        arm64 | aarch64) arch="arm64" ;;
        *) return 1 ;;
    esac

    # Double check 32-bit OS reported as 64-bit
    if [ "${arch}" = "x64" ] && [ "$(getconf LONG_BIT)" -eq 32 ]; then
        return 1
    elif [ "${arch}" = "arm64" ] && [ "$(getconf LONG_BIT)" -eq 32 ]; then
        return 1
    fi

    printf '%s' "${arch}"
}

# ─── Installation Logic ──────────────────────────────────────────────

install_pnpm() {
    if has_command pnpm; then
        log_info "pnpm is already installed ($(pnpm --version))."
    fi

    local arch libc_suffix version_json version major_version asset_base

    arch="$(detect_pnpm_arch)" || {
        log_error "pnpm currently only provides pre-built binaries for x86_64/arm64 architectures."
        return 1
    }
    libc_suffix="$(detect_libc_suffix)"

    # Fetch the latest version from the npm registry, or use PNPM_VERSION if set
    if [ -z "${PNPM_VERSION:-}" ]; then
        log_info "Fetching latest pnpm version from npm registry..."
        version_json="$(download "https://registry.npmjs.org/@pnpm/exe")" || {
            log_error "Failed to fetch pnpm version info from npm registry."
            return 1
        }
        version="$(echo "$version_json" | grep -o '"latest":[[:space:]]*"[0-9.]*"' | grep -o '[0-9.]*')"
    else
        version="${PNPM_VERSION}"
    fi

    # Normalize major version (strip leading "v", extract digits)
    major_version="$(printf '%s' "$version" | sed -E 's/^v//; s/^([0-9]+).*/\1/')"
    if [ -z "$major_version" ]; then
        log_error "Invalid PNPM_VERSION: $version"
        return 1
    fi

    log_info "Downloading pnpm v${version} (linux-${arch}${libc_suffix})..."
    asset_base="$(asset_basename "$version" "$arch" "$libc_suffix")"

    if [ "$major_version" -ge 11 ]; then
        # v11+: distributed as tarballs containing the binary and dist/ directory
        download "https://github.com/pnpm/pnpm/releases/download/v${version}/${asset_base}.tar.gz" > "$TMP_DIR/pnpm.tar.gz" || {
            log_error "Failed to download pnpm tarball."
            return 1
        }
        tar -xzf "$TMP_DIR/pnpm.tar.gz" -C "$TMP_DIR" || {
            log_error "Failed to extract pnpm tarball."
            return 1
        }
        chmod +x "$TMP_DIR/pnpm"
        SHELL="${SHELL:-/bin/bash}" "$TMP_DIR/pnpm" setup --force || {
            log_error "pnpm setup failed."
            return 1
        }
    else
        # Older versions: distributed as a single executable binary
        download "https://github.com/pnpm/pnpm/releases/download/v${version}/${asset_base}" > "$TMP_DIR/pnpm" || {
            log_error "Failed to download pnpm binary."
            return 1
        }
        chmod +x "$TMP_DIR/pnpm"
        SHELL="${SHELL:-/bin/bash}" "$TMP_DIR/pnpm" setup --force || {
            log_error "pnpm setup failed."
            return 1
        }
    fi

    log_success "pnpm v${version} installed successfully!"
}

# ─── Shell Configuration ─────────────────────────────────────────────

configure_shell() {
    IFS=' ' read -ra target_files <<< "$(get_shell_configs)"

    # pnpm's `setup --force` configures PNPM_HOME and PATH automatically,
    # but we also add an env block to ensure PNPM_HOME is set consistently.
    local content
    content=$(cat << 'EOF'
# pnpm
export PNPM_HOME="$HOME/.local/share/pnpm"
case ":$PATH:" in
  *":$PNPM_HOME:"*) ;;
  *) export PATH="$PNPM_HOME:$PATH" ;;
esac
EOF
)

    for config_file in "${target_files[@]}"; do
        log_info "Configuring pnpm in $config_file..."
        inject_block "$config_file" "pnpm setup" "$content"

        # Source if modified (only for bashrc)
        if [ "$config_file" = "$HOME/.bashrc" ]; then
            . "$config_file" 2>/dev/null || true
        fi
    done
}

# ─── Main ─────────────────────────────────────────────────────────────

main() {
    install_pnpm
    configure_shell

    echo
    if has_command pnpm; then
        log_success "pnpm installation and configuration complete."
        log_info "Installed pnpm version: $(pnpm --version 2>/dev/null || echo 'unknown')"
    else
        log_success "Installation complete."
        log_info "Please close and reopen your terminal or run: source ~/.bashrc to verify."
    fi
}

main "$@"