#!/usr/bin/env bash
set -eo pipefail

# shellcheck source=scripts/common.sh
source "${REPO_ROOT}/scripts/common.sh"
# shellcheck source=scripts/common_cmdline.sh
source "${REPO_ROOT}/scripts/common_cmdline.sh"

function test_via_ir_equivalence()
{
    (( $# <= 2 )) || fail "This function accepts at most two arguments."
    local solidity_file="$1"
    local optimize_flag="$2"
    [[ $optimize_flag == --optimize || $optimize_flag == "" ]] || assertFail "The second argument must be --optimize if present."

    local output_file_prefix
    output_file_prefix=$(basename "$solidity_file" .sol)

    SOLTMPDIR=$(mktemp -d -t "cmdline-test-via-ir-equivalence-${output_file_prefix}-XXXXXX")
    pushd "$SOLTMPDIR" > /dev/null

    local optimizer_flags=()
    [[ $optimize_flag == "" ]] || optimizer_flags+=("$optimize_flag")
    [[ $optimize_flag == "" ]] || output_file_prefix+="_optimize"

    msg_on_error --no-stderr \
        "$SOLC" --ir-optimized --debug-info location "${optimizer_flags[@]}" "$solidity_file" |
            stripCLIDecorations |
            split_on_empty_lines_into_numbered_files "$output_file_prefix" ".yul"

    local asm_output_two_stage asm_output_via_ir

    for yul_file in $(find . -name "${output_file_prefix}*.yul" | sort -V); do
        asm_output_two_stage+=$(
            msg_on_error --no-stderr \
                "$SOLC" --strict-assembly --asm "${optimizer_flags[@]}" --no-optimize-yul "$yul_file" |
                    stripCLIDecorations
        )
    done

    asm_output_via_ir=$(
        msg_on_error --no-stderr \
            "$SOLC" --via-ir --asm --debug-info location "${optimizer_flags[@]}" "$solidity_file" |
                stripCLIDecorations
    )

    diff_values "$asm_output_two_stage" "$asm_output_via_ir" --ignore-space-change --ignore-blank-lines

    local bin_output_two_stage bin_output_via_ir

    for yul_file in $(find . -name "${output_file_prefix}*.yul" | sort -V); do
        bin_output_two_stage+=$(
            msg_on_error --no-stderr \
                "$SOLC" --strict-assembly --bin "${optimizer_flags[@]}" "$yul_file" --no-optimize-yul |
                    stripCLIDecorations
        )
    done

    bin_output_via_ir=$(
        msg_on_error --no-stderr \
            "$SOLC" --via-ir --bin "${optimizer_flags[@]}" "$solidity_file" | stripCLIDecorations
    )

    diff_values "$bin_output_two_stage" "$bin_output_via_ir" --ignore-space-change --ignore-blank-lines

    popd > /dev/null
    rm -r "$SOLTMPDIR"
}

externalContracts=(
    externalTests/solc-js/DAO/TokenCreation.sol
    libsolidity/semanticTests/externalContracts/_prbmath/PRBMathSD59x18.sol
    libsolidity/semanticTests/externalContracts/_prbmath/PRBMathUD60x18.sol
    libsolidity/semanticTests/externalContracts/_stringutils/stringutils.sol
    libsolidity/semanticTests/externalContracts/deposit_contract.sol
    libsolidity/semanticTests/externalContracts/FixedFeeRegistrar.sol
    libsolidity/semanticTests/externalContracts/snark.sol
)

requiresOptimizer=(
    externalTests/solc-js/DAO/TokenCreation.sol
    libsolidity/semanticTests/externalContracts/deposit_contract.sol
    libsolidity/semanticTests/externalContracts/FixedFeeRegistrar.sol
    libsolidity/semanticTests/externalContracts/snark.sol
)

for contractFile in "${externalContracts[@]}"
do
    if ! [[ "${requiresOptimizer[*]}" =~ $contractFile ]]
    then
        printTask "    - ${contractFile}"
        test_via_ir_equivalence "${REPO_ROOT}/test/${contractFile}"
    fi

    printTask "    - ${contractFile} (optimized)"
    test_via_ir_equivalence "${REPO_ROOT}/test/${contractFile}" --optimize
done