#!/bin/sh
# Verify command-line parsing and validation through the compiler driver.
set -eu
work=$(mktemp -d)
trap 'rm -rf "$work"' EXIT HUP INT TERM
checks=0

# Use the compiler's standard Make memory and stack settings.
command() {
    timeout 30 "${MAKE:-make}" --no-print-directory driver-command \
        RAD_BIN="${RAD_BIN:-bin/radiance.rv64.dev}" \
        RAD_EMULATOR="${RAD_EMULATOR:-emulator}" \
        DRIVER_ARGUMENTS="$1" > "$work/command.log" 2>&1
}

# Require a specific diagnostic, including checks that precede file loading.
reject() {
    if command "$1"; then
        echo "command: unexpected success: $1"
        exit 1
    fi
    if ! grep -Fq -- "$2" "$work/command.log"; then
        cat "$work/command.log"
        echo "command: missing diagnostic: $2"
        exit 1
    fi
    checks=$((checks + 1))
}

# Require successful compilation or dumping.
accept() {
    if ! command "$1"; then
        cat "$work/command.log"
        exit 1
    fi
    checks=$((checks + 1))
}

while IFS='|' read -r arguments expected; do
    reject "$arguments" "$expected"
done <<'CASES'
|usage: radiance
-pkg|`-pkg` requires a package name
-mod|`-mod` requires a module path
-start|`-start` requires an assembly path
-entry|`-entry` requires a package name
-o|`-o` requires an output path
-ril|`-ril` requires an output directory
-dump|`-dump` requires a mode
-mod absent.rad|`-mod` must follow a `-pkg` argument
-start absent.ras|`-start` must follow a `-pkg` argument
-dump invalid|unknown dump mode invalid
-unknown|unknown argument -unknown
-test|no package specified
-pkg p|package p has no Radiance modules specified
-pkg p -mod absent.ras|package p has no Radiance modules specified
-pkg p -mod p.rad -start x.rad|`-start` requires a `.ras` assembly file
-pkg p -mod p.rad -start x.ras -start y.ras|has more than one startup file
-pkg p -mod p.rad -pkg q -mod q.rad|`-entry` required when multiple packages specified
-pkg p -mod p.rad -pkg q -mod q.rad -entry missing|entry package missing not found
-pkg p -mod p.rad -start x.ras -pkg q -mod q.rad -entry q|`-start` is only supported on the entry package
-pkg p -mod p.rad -ril out -o out.rv64|`-ril` requires a separate invocation
-pkg p -mod p.rad -ril out -dump ast|`-ril` requires a separate invocation
CASES

# At each path limit, parsing reaches the deliberately absent source file.
arguments='-pkg p'
for count in $(seq 1 129); do
    arguments="$arguments -mod $work/absent.rad"
    if [ "$count" -eq 128 ]; then reject "$arguments" 'error reading file'; fi
done
reject "$arguments" 'too many modules specified for package'
arguments="-pkg p -mod $work/absent.rad"
for count in $(seq 1 65); do
    arguments="$arguments -mod $work/absent.ras"
    if [ "$count" -eq 64 ]; then reject "$arguments" 'error reading file'; fi
done
reject "$arguments" 'too many assembly modules specified'
reject '-pkg a -pkg b -pkg c -pkg d -pkg e' 'too many packages specified'

printf '%s\n' '/// Executable command fixture.' '@default fn main() -> u32 { return 0; }' > "$work/p.rad"
for mode in ast graph il asm; do
    accept "-pkg p -mod $work/p.rad -dump $mode"
done
# A single package selects itself; the final output option selects the path.
accept "-pkg p -mod $work/p.rad -entry unused -debug -o $work/first.rv64 -o $work/final.rv64"
test ! -e "$work/first.rv64"
test -s "$work/final.rv64"
timeout 10 "${RAD_EMULATOR:-emulator}" -run "$work/final.rv64"

arguments=''
for name in a b c; do
    printf '%s\n' '/// Package fixture value.' 'constant value: u32 = 0;' > "$work/$name.rad"
    arguments="$arguments -pkg $name -mod $work/$name.rad"
done
accept "$arguments -pkg p -mod $work/p.rad -entry p -o $work/multiple.rv64"
timeout 10 "${RAD_EMULATOR:-emulator}" -run "$work/multiple.rv64"

# Binary package output preserves bytes across destinations and replacements.
mkdir "$work/packages"
accept "$arguments -pkg p -mod $work/p.rad -entry p -ril $work/packages"
for name in a b c p; do
    test -s "$work/packages/$name.ril"
done
cp "$work/packages/p.ril" "$work/expected.ril"
printf '%s\n' 'stale output bytes' >> "$work/packages/p.ril"
accept "-pkg p -mod $work/p.rad -ril $work/packages"
cmp "$work/expected.ril" "$work/packages/p.ril"

# MAX_PATH_LEN includes the terminating zero byte.
for length in 249 250 255 256; do
    leaf=$(awk -v n="$((length - ${#work} - 1))" 'BEGIN { for (i = 0; i < n; i++) printf "x" }')
    directory="$work/$leaf"
    if [ "$length" -eq 249 ]; then
        mkdir "$directory"
        accept "-pkg p -mod $work/p.rad -ril $directory"
        cmp "$work/expected.ril" "$directory/p.ril"
        reject "-pkg package -mod $work/p.rad -ril $directory" 'binary RIL output path is too long'
    else
        reject "-pkg p -mod $work/p.rad -ril $directory" 'binary RIL output path is too long'
    fi
done
reject "-pkg p -mod $work/p.rad -ril $work/missing" 'cannot write binary RIL package p'
mkdir -p "$work/blocked/p.ril"
reject "-pkg p -mod $work/p.rad -ril $work/blocked" 'cannot write binary RIL package p'
reject "-pkg p -mod $work/p.rad -mod absent.ras -ril $work/packages" 'binary RIL output requires Radiance source modules'
reject "-pkg p -mod $work/p.rad -start absent.ras -ril $work/packages" 'binary RIL output requires Radiance source modules'
echo "command tests: $checks passed"