#!/bin/bash

#
# Copyright (C) 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#

readonly MODE=$1
readonly TARGET=$2

readonly WORK_DIR=$PWD
readonly PLATFORM=$(uname -s | tr A-Z a-z)
readonly IWASM_CMD="${WORK_DIR}/../../../../product-mini/platforms/${PLATFORM}/build/iwasm"
readonly WAMRC_CMD="${WORK_DIR}/../../../../wamr-compiler/build/wamrc"

if [[ $MODE != "aot" ]];then
    python3 -m venv wasi-env && source wasi-env/bin/activate
    python3 -m pip install -r test-runner/requirements.txt
    TEST_RUNTIME_EXE="${IWASM_CMD}" python3 test-runner/wasi_test_runner.py \
                -r adapters/wasm-micro-runtime.py \
                -t \
                    tests/c/testsuite/ \
                    tests/assemblyscript/testsuite/ \
                    tests/proposals/wasi-threads/
    exit_code=${PIPESTATUS[0]}
    deactivate
else
    target_option=""
    if [[ $TARGET == "X86_32" ]];then
        target_option="--target=i386"
    fi

    # Run WASI thread proposal tests
    exit_code=0
    wasm_tests=$(ls tests/proposals/wasi-threads/*.wasm)
    for test_wasm in ${wasm_tests}; do
        test_aot="${test_wasm%.wasm}.aot"
        test_json="${test_wasm%.wasm}.json"

        echo "Compiling $test_wasm to $test_aot"
        ${WAMRC_CMD} --enable-multi-thread ${target_option} \
            -o $test_aot $test_wasm

        echo "Running $test_aot"
        expected=$(jq .exit_code ${test_json})
        ${IWASM_CMD} $test_aot
        ret=${PIPESTATUS[0]}

        echo "expected=$expected, actual=$ret"
        if [[ $expected != "" ]] && [[ $expected != $ret ]];then
            exit_code=1
        fi
    done
fi

exit ${exit_code}