#!/usr/bin/env bash

# Based on the template in: https://docs.digitalocean.com/reference/api/spaces-api/
bash -euo pipefail
source script/lib/blob-store.sh

allowed_targets=("linux-targz" "macos")
is_allowed_target() {
    for val in "${allowed_targets[@]}"; do
        if [[ "$1" == "$val" ]]; then
            return 0
        fi
    done
    return 1
}

if [[ -n "${1:-}" ]]; then
    if is_allowed_target "$1"; then
        target="$1"
    else
        echo "Error: Target '$1' is not allowed"
        echo "Usage: $0 [${allowed_targets[@]}]"
        exit 1
    fi
else
echo "Error: Target is not specified"
echo "Usage: $0 [${allowed_targets[@]}]"
exit 1
fi
echo "Uploading nightly for target: $target"

bucket_name="zed-nightly-host"

sha=$(git rev-parse HEAD)
echo ${sha} > target/latest-sha
case "$target" in
    macos)
        upload_to_blob_store $bucket_name "target/aarch64-apple-darwin/release/Zed.dmg" "nightly/Zed-aarch64.dmg"
        upload_to_blob_store $bucket_name "target/x86_64-apple-darwin/release/Zed.dmg" "nightly/Zed-x86_64.dmg"
        upload_to_blob_store $bucket_name "target/release/Zed.dmg" "nightly/Zed.dmg"
        upload_to_blob_store $bucket_name "target/latest-sha" "nightly/latest-sha"
        rm -f "target/aarch64-apple-darwin/release/Zed.dmg" "target/x86_64-apple-darwin/release/Zed.dmg" "target/release/Zed.dmg"
        rm -f "target/latest-sha"
        ;;
    linux-targz)
        find . -type f -name "zed-*.tar.gz" -print0 | while IFS= read -r -d '' bundle_file; do
            upload_to_blob_store $bucket_name "$bundle_file" "nightly/$(basename "$bundle_file")"
            rm -f "$bundle_file"
        done
        upload_to_blob_store $bucket_name "target/latest-sha" "nightly/latest-sha-linux-targz"
        rm -f "target/latest-sha"
        ;;
    *)
        echo "Error: Unknown target '$target'"
        exit 1
        ;;
esac
