#!/bin/bash

set -x
set -e 

export H5_LDFLAGS=-L/usr/local/lib
export H5_CFLAGS=-I/usr/local/include


# build wheel directly from developer-provided .tar.gz of TREXIO (generated with `python setup.py sdist`)
# note: trexio-VERSION.tar.gz has to be in the root directory of the host machine and provided as an argument to this script

# process input: first argument is the name of the .tar.gz with the source code of the Python API
if [[ -z "$1" ]]; then
    echo "Please specify the name of the TREXIO source code distribution (with .tar.gz suffix)"
    exit 1
fi

# derive TREXIO version from the file name
TREXIO_SOURCE=${1}
# remove prefix that ends with "-"
tmp=${TREXIO_SOURCE#*-}
# remove suffix that ends with ".tar.gz"
TR_VERSION=${tmp%.tar.gz*}
# print the computed version
echo "TREXIO VERSION:" ${TR_VERSION}

# unzip and enter the folder with TREXIO Python API
gzip -cd /tmp/trexio-${TR_VERSION}.tar.gz | tar xvf -
cd trexio-${TR_VERSION}

# the function below build manylinux wheels based on the provided version of python (e.g. build_wheel_for_py 36)
function build_wheel_for_py() 
{

   if [[ -z "$1" ]]; then
      echo "Empty string provided instead of the Python version"
      exit 1
   fi

   # derive PYVERSION from the input argument
   PYVERSION=${1}

   # python versions <= 3.7 required additional "m" in the platform tag, e.g. cp37-cp37m
   if [[ ${PYVERSION} -eq 36 ]] || [[ ${PYVERSION} -eq 37 ]]; then
       PYM="m"
   else
       PYM=""
   fi

   CPYTHON="cp${PYVERSION}-cp${PYVERSION}${PYM}"

   # create and activate a virtual environment based on CPython version ${PYVERSION}
   /opt/python/${CPYTHON}/bin/python3 -m venv --clear trexio-manylinux-py${PYVERSION}
   source trexio-manylinux-py${PYVERSION}/bin/activate 
   python3 --version

   # upgrade pip, otherwise it complains that manylinux wheel is "...not supported wheel on this platform"
   pip install --upgrade pip 
   # install dependencies needed to build manylinux wheel
   pip install --upgrade setuptools wheel auditwheel numpy

   # set an environment variable needed to locate numpy header files
   source tools/set_NUMPY_INCLUDEDIR.sh

   # produce conventional (non-manylinux) wheel
   python3 setup.py bdist_wheel

   # use auditwheel from PyPA to repair all wheels and make them manylinux-compatible
   auditwheel repair dist/trexio-${TR_VERSION}-${CPYTHON}-*.whl

   # install the produced manylinux wheel in the virtual environment
   python3 -m pip install wheelhouse/trexio-${TR_VERSION}-${CPYTHON}-manylinux*.whl

   # run test script
   cd test && python3 test_api.py && cd ..

   # cleaning
   rm -rf -- dist/ build/ trexio.egg-info/

   # deactivate the current environment
   deactivate

   # remove the virtual environment
   rm -rf -- trexio-manylinux-py${PYVERSION}

}


# build wheels for all versions of CPython in this container 
for CPYVERSION in 36 37 38 39
do
  build_wheel_for_py ${CPYVERSION}
done