#!/bin/sh

set -o errexit

MONGOSH_RELEASES_URL=https://downloads.mongodb.com/compass/mongosh.json

say() {
    echo >&2 "$@"
}

sayf() {
    # shellcheck disable=SC2059
    printf >&2 "$@"
}

show_download_link() {
    say "Download mongosh manually from:"
    say
    sayf "\t%s\n" 'https://www.mongodb.com/try/download/shell'
}

for tool in jq curl; do
    which "$tool" >/dev/null || {
        say "This script requires '$tool'."
        exit 1
    }
done

os=$(uname -o | tr '[:upper:]' '[:lower:]')
arch=$(uname -m)

case "$os" in
    *linux)
        ext=tgz
        os=linux
        ;;
    darwin)
        ext=zip
        ;;
    *)
        say "❌ This script does not support this OS ($os)."
        show_download_link

        exit 1
esac

# normalize $arch:
case "$arch" in
    amd64|x64)
        arch=x86_64
        ;;
    aarch64)
        arch=arm64
        ;;
    *)
        # Use uname’s reported architecture in the jq query.
esac

if [ "$os" = "linux" ]; then
    if ldd $(which curl) | grep -q libssl.so.3 ; then
        openssl_query='and .sharedOpenssl == "openssl3"'
    else
        openssl_query='and (has("sharedOpenssl") | not)'
    fi
else
    openssl_query=''
fi

jq_query=$(cat <<EOF
.versions[0].downloads[] |
select(
    .distro == "$os" and
    .arch == "$arch"
    $openssl_query
) |
.archive.url
EOF
)

url=$(curl -fsSL $MONGOSH_RELEASES_URL | jq -r "$jq_query")

if [ -z "$url" ]; then
    say "❓ No download found for $os on $arch."
    show_download_link
    exit 1
fi

case "$ext" in
    zip)
        file=$(mktemp)

        say "Downloading $url to $file …"
        #trap 'rm -f "$file"' EXIT

        curl -fsSL "$url" > "$file"
        say "Downloaded $ext file; extracting mongosh …"

        unzip -j "$file" '*/bin/mongosh*'
        ;;
    tgz)
        say "Downloading & extracting from $url …"

        curl -fsSL "$url" | tar -xzf - \
            --transform "s/.*\///" \
            --wildcards "**/bin/mongosh*" \
        | sed -E 's/^.*[/]//'

        ;;
    *)
        say "Bad file extension: $ext"
        show_download_link
        exit 1
esac

./mongosh --build-info >/dev/null 2>&1 || {
    say "❌ Downloaded mongosh is not executable."
    ./mongosh --build-info
    exit 1
}

say "✅ Success! 'mongosh' and its crypto library are now saved in this directory."