#!/usr/bin/env bash

# This script automatically download ray and run the sanity check for Windows
# in various Python version. This script requires conda command to exist.

set -euo pipefail

export RAY_HASH="${RAY_HASH:-}"
export RAY_VERSION="${RAY_VERSION:-}"

if [[ -z "$RAY_HASH" ]]; then
    echo "RAY_HASH env var should be provided"
    exit 1
fi

if [[ -z "$RAY_VERSION" ]]; then
    echo "RAY_VERSION env var should be provided"
    exit 1
fi

if [[ ! -x "$(command -v conda)" ]]; then
    echo "conda doesn't exist. Please download conda for this machine"
    exit 1
else
    echo "conda exists"
fi

echo "Start downloading Ray version ${RAY_VERSION} of commit ${RAY_HASH}"

python -m pip install --upgrade pip

# This is required to use conda activate
source "$(conda info --base)/etc/profile.d/conda.sh"

PYTHON_VERSIONS=( "3.10" "3.11" "3.12" )

for PYTHON_VERSION in "${PYTHON_VERSIONS[@]}"; do
    ENV_NAME="${RAY_VERSION}-${PYTHON_VERSION}-env"
    conda create -y -n "${ENV_NAME}" python="${PYTHON_VERSION}"
    conda activate "${ENV_NAME}"

    FAILED="false"

    GOT_PYTHON_VERSION="$(python -c 'import sys; v=sys.version_info; print(f"{v.major}.{v.minor}")')"
    if [[ "${GOT_PYTHON_VERSION}" != "${PYTHON_VERSION}" ]]; then
        echo "Python version incorrect, got ${GOT_PYTHON_VERSION}, want ${PYTHON_VERSION}"
        FAILED="true"
    else
        pip install --index-url 'https://test.pypi.org/simple/' --extra-index-url 'https://pypi.org/simple' "ray[cpp]==${RAY_VERSION}"

        printf "\n\n\n"
        echo "========================================================="
        python sanity_check.py --ray_version="$RAY_VERSION" --ray_commit="$RAY_HASH"
        echo "========================================================="
        printf "\n\n\n"
    fi

    conda deactivate
    conda remove -y --name "${ENV_NAME}" --all

    if [[ "$FAILED" == "true" ]]; then
        echo "PYTHON ${PYTHON_VERSION} failed sanity check."
        exit 1
    else
        echo "PYTHON ${PYTHON_VERSION} succeed sanity check."
    fi
done