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

cd "$(dirname "$0")"

linuxcnc_root=${LINUXCNC_ROOT:-../linuxcnc}
manifest=${1:-linuxcnc-rs274-wasm-source-files.txt}
cxx=${CXX:-g++}
build_dir=$(mktemp -d "${TMPDIR:-/tmp}/cnc_sim_linuxcnc_wasm_safe_probe.XXXXXX")

if [[ ! -f "$manifest" ]]; then
    echo "missing manifest: $manifest" >&2
    exit 1
fi

if [[ ! -d "$linuxcnc_root" ]]; then
    echo "missing LinuxCNC root: $linuxcnc_root" >&2
    exit 1
fi
trap 'rm -rf "$build_dir"' EXIT

shim_list="$build_dir/wasm_safe_shims.txt"
./list-linuxcnc-wasm-safe-shims.sh > "$shim_list"
mapfile -t shim_sources < "$shim_list"

common_flags=(
    -std=c++20
    -DULAPI
    -include wctype.h
    -I "$(pwd)/core/include"
    -I "$(pwd)/core/src"
    -I "$(pwd)/core/wasm_shims"
    -I "$linuxcnc_root/src"
    -I "$linuxcnc_root/src/emc"
    -I "$linuxcnc_root/src/emc/nml_intf"
    -I "$linuxcnc_root/src/emc/rs274ngc"
    -I "$linuxcnc_root/src/emc/motion"
    -I "$linuxcnc_root/include"
)

source_list="$build_dir/wasm_core_sources.txt"
LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-wasm-manifest-sources.sh "$manifest" core full > "$source_list"
mapfile -t sources < "$source_list"

shim_index=0
for source in "${shim_sources[@]}"; do
    obj="$build_dir/shim_${shim_index}.o"
    "$cxx" "${common_flags[@]}" -c "$source" -o "$obj"
    shim_index=$((shim_index + 1))
done

source_index=0
for source in "${sources[@]}"; do
    obj="$build_dir/linuxcnc_${source_index}.o"
    "$cxx" "${common_flags[@]}" -c "$source" -o "$obj"
    source_index=$((source_index + 1))
done

echo "linuxcnc wasm-safe source object probe passed (${#sources[@]} linuxcnc objects + ${#shim_sources[@]} shims)"