setup_snapshot.sh

#!/usr/bin/env bash

set -euo pipefail

# Synthesize the directory name for the snapshot build.
function cli_snapshot_directory() {
    dir="cli"

    # Append correct os name.
    case $RUNNER_OS in
    Linux)
        dir="${dir}_linux"
        ;;
    Windows)
        dir="${dir}_windows"
        ;;
    macOS)
        dir="${dir}_darwin"
        ;;
    esac

    # Append correct arch name.
    case $RUNNER_ARCH in
    X86)
        dir="${dir}_386"
        ;;
    X64)
        dir="${dir}_amd64_v1"
        ;;
    ARM)
        dir="${dir}_arm_6"
        ;;
    ARM64)
        dir="${dir}_arm64"
        ;;
    esac

    echo $dir
}

# Default to main branch if branch is not specified.
if [ -z "$BRANCH" ]; then
  BRANCH=main
fi

# Find last successful build on $BRANCH.
last_successful_run_id=$(
  gh run list -b "$BRANCH" -w release-snapshot --json 'databaseId,conclusion' |
      jq 'limit(1; .[] | select(.conclusion == "success")) | .databaseId'
)
if [ -z "$last_successful_run_id" ]; then
  echo "Unable to find last successful build"
  exit 1
fi

# Determine artifact name with the right binaries for this runner.
case $RUNNER_OS in
Linux)
    artifact="cli_linux_snapshot"
    ;;
Windows)
    artifact="cli_windows_snapshot"
    ;;
macOS)
    artifact="cli_darwin_snapshot"
    ;;
esac

# Change into temporary directory.
cd "$RUNNER_TEMP"

gh run download "$last_successful_run_id" -n "$artifact" -D .bin

dir="$PWD/.bin/$(cli_snapshot_directory)"

if [ ! -d "$dir" ]; then
    echo "Directory does not exist: $dir"
    exit 1
fi

if [ "$RUNNER_OS" == "Windows" ]; then
    (
        cd "$dir"
        mv ./databricks.exe ./databricks
    )
fi

# Add databricks to path.
chmod +x "$dir/databricks"
echo "$dir" >> "$GITHUB_PATH"