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

cd "$(dirname "$0")"

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

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

common_flags=(
    -std=c++20
    -DULAPI
    -ffunction-sections
    -fdata-sections
    -I "$(pwd)/core/src"
    -include wctype.h
    -I "$(pwd)/core/include"
    -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"
)

shim_list="$build_dir/wasm_safe_shims.txt"
./list-linuxcnc-wasm-safe-shims.sh > "$shim_list"
mapfile -t sources < "$shim_list"
project_source_list="$build_dir/wasm_safe_project_sources.txt"
./list-linuxcnc-wasm-safe-project-sources.sh > "$project_source_list"
mapfile -t project_sources < "$project_source_list"
sources+=("${project_sources[@]}")

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 linuxcnc_sources < "$source_list"
for source in "${linuxcnc_sources[@]}"; do
    sources+=("$source")
done

sources+=(core/wasm_shims/rs274ngc_pre_probe_main.cc)

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

"$cxx" "${objects[@]}" \
    -Wl,--gc-sections \
    -lfmt \
    -o "$build_dir/rs274ngc_pre_probe"

"$build_dir/rs274ngc_pre_probe"

echo "linuxcnc wasm rs274ngc_pre link probe passed"