#!/bin/bash -x

if [[ $# -ne 1 ]]; then
    echo "Illegal number of parameters" >&2
    echo "Usage: $0 <version_string>" >&2
    exit 2
fi

pip_installed_version=$(pip list | grep klvm_tools_rs | awk '{print $2}')
python_import_version=$(python -c 'import klvm_tools_rs; print(klvm_tools_rs.get_version())')

expected_version="$1"

if [ "$expected_version" == "$pip_installed_version" ] && [ "$expected_version" == "$python_import_version" ]; then
    exit 0
else
    echo "klvm_tools_rs VERSIONS does not match expected version"
    echo "PIP INSTALLED VERSION: $pip_installed_version"
    echo "PYTHON IMPORTED VERSION: $python_import_version"
    echo "EXPECTED VERSION: $expected_version"
    exit 1
fi