#!/usr/bin/env bash

set -e

if [ "$#" -ne 1 ]; then
    echo "Expected a top-level directory name for unsigned artifacts."
    exit 1
fi

if [[ -z "$C3R_ALL_Y_TEST_COLLABORATION" ]]; then
    echo "Expected env var C3R_ALL_Y_TEST_COLLABORATION to be set"
    echo "to a collaboration ID where all cryptographic parameters"
    echo "are set to 'Yes'."
    exit 1
fi

if [[ -z "$C3R_ALL_N_TEST_COLLABORATION" ]]; then
    echo "Expected env var C3R_ALL_N_TEST_COLLABORATION to be set"
    echo "to a collaboration ID where all cryptographic parameters"
    echo "are set to 'No'."
    exit 1
fi

ARTIFACT_DIR=$1

SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )

pushd $SCRIPT_DIR/.. > /dev/null

C3R_CLI_DIR=c3r-cli/build/libs
C3R_CLI_SPARK_DIR=c3r-cli-spark/build/libs

# Build everything
./gradlew build -x checkFileUtilTest --parallel

# Get the version from the CLI
C3R_VERSION=$(java -jar $C3R_CLI_DIR/c3r-cli-all.jar --version)
echo "JAR version found: $C3R_VERSION"

# Test CLI JAR
python3 c3r-cli/src/integration-test/integration_tests.py \
  --id-y $C3R_ALL_Y_TEST_COLLABORATION \
  --id-n $C3R_ALL_N_TEST_COLLABORATION \
  $C3R_CLI_DIR/c3r-cli-all.jar

# Test CLI Spark JAR
python3 c3r-cli-spark/src/integration-test/spark_integration_tests.py \
  --id-y $C3R_ALL_Y_TEST_COLLABORATION \
  --id-n $C3R_ALL_N_TEST_COLLABORATION \
  $C3R_CLI_SPARK_DIR/c3r-cli-spark-all.jar


mkdir -p $ARTIFACT_DIR

# Move CLI jar to the artifact directory
mv $C3R_CLI_DIR/c3r-cli-all.jar $ARTIFACT_DIR/c3r-cli-$C3R_VERSION.jar
mv $C3R_CLI_SPARK_DIR/c3r-cli-spark-all.jar $ARTIFACT_DIR/c3r-cli-spark-$C3R_VERSION.jar

# Given an SDK project name, collect it's JAR, javadocs JAR, and sources JAR
# and move them to the artifact directory with a version number applied
function move_sdk_jars() {
    local sdk_name=$1
    local sdk_jar_dir=$sdk_name/build/libs
    # Move sdk JARs to the artifact directory
    mv $sdk_jar_dir/$sdk_name.jar $ARTIFACT_DIR/$sdk_name-$C3R_VERSION.jar
    mv $sdk_jar_dir/$sdk_name-javadoc.jar $ARTIFACT_DIR/$sdk_name-javadoc-$C3R_VERSION.jar
    mv $sdk_jar_dir/$sdk_name-sources.jar $ARTIFACT_DIR/$sdk_name-sources-$C3R_VERSION.jar
}

# Move sdk JARs to the artifact directory
move_sdk_jars c3r-sdk-core
move_sdk_jars c3r-sdk-parquet

echo "JAR files tagged with version $C3R_VERSION and placed in $ARTIFACT_DIR/"

popd > /dev/null # $SCRIPT_DIR/..