#!/bin/bash

# First publish wheels to cache

COPY_FROM=${WORKSPACE}/cache_upload/
PUBLISH_TO=ldcg/python-cache/

set -ex

# Publish wheel files
test -d ${HOME}/bin || mkdir ${HOME}/bin
wget -q https://git.linaro.org/ci/publishing-api.git/blob_plain/HEAD:/linaro-cp.py -O ${HOME}/bin/linaro-cp.py

# Cache upload is not required for nightly build
if [ -e $COPY_FROM ]
then

  ls -alR $COPY_FROM

  time python3 ${HOME}/bin/linaro-cp.py \
    $COPY_FROM \
    $PUBLISH_TO || true

  echo "Python wheels cached: https://snapshots.linaro.org/${PUBLISH_TO}"
fi

# Now is time to upload tensorflow

SHORT_JOB_NAME=$(echo $JOB_NAME | cut -d'/' -f1)

case $SHORT_JOB_NAME in

  "ldcg-python-manylinux-tensorflow-nightly")
    OUTPUT_PATH="ldcg/python/tensorflow-manylinux-nightly/$(date -u +%Y%m%d)-${BUILD_NUMBER}/"
    ;;

  "ldcg-python-manylinux-tensorflow-onednn-nightly")
    OUTPUT_PATH="ldcg/python/tensorflow-onednn-manylinux-nightly/$(date -u +%Y%m%d)-${BUILD_NUMBER}/"
    ;;

  "ldcg-python-manylinux-tensorflow")
    OUTPUT_PATH="ldcg/python/tensorflow-manylinux/${BUILD_NUMBER}/"
    ;;

  "ldcg-python-manylinux-tensorflow-io")
    OUTPUT_PATH="ldcg/python/tensorflow-io-manylinux/${BUILD_NUMBER}/"
    ;;

  "ldcg-python-manylinux-tensorflow-text")
    OUTPUT_PATH="ldcg/python/tensorflow-text-manylinux/${BUILD_NUMBER}/"
    ;;

esac

ls -alR ${WORKSPACE}/wheels

time python3 ${HOME}/bin/linaro-cp.py \
  --link-latest \
  ${WORKSPACE}/wheels \
  $OUTPUT_PATH || true

echo "Python wheels: https://snapshots.linaro.org/$OUTPUT_PATH"