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

cd "$(dirname "$0")"

linuxcnc_root=${LINUXCNC_ROOT:-../linuxcnc}
cxx=${CXX:-g++}
if [[ -z ${CXX:-} ]] && command -v ccache >/dev/null 2>&1; then
    cxx="ccache $cxx"
fi
build_jobs=${CNC_SIM_BUILD_JOBS:-8}
if ! [[ "$build_jobs" =~ ^[1-9][0-9]*$ ]]; then
    echo "CNC_SIM_BUILD_JOBS must be a positive integer: $build_jobs" >&2
    exit 1
fi
build_dir=${CNC_SIM_WASM_INTERP_FIND_BUILD_DIR-build/wasm-interp-find-link}
if [[ -z "$build_dir" ]]; then
    echo "CNC_SIM_WASM_INTERP_FIND_BUILD_DIR must not be empty" >&2
    exit 1
fi
if [[ "$build_dir" =~ [[:space:]] ]]; then
    echo "CNC_SIM_WASM_INTERP_FIND_BUILD_DIR must not contain whitespace: $build_dir" >&2
    exit 1
fi
mkdir -p "$build_dir"
build_dir=$(cd "$build_dir" && pwd)
if [[ "$build_dir" == "/" ]]; then
    echo "CNC_SIM_WASM_INTERP_FIND_BUILD_DIR must not be the filesystem root" >&2
    exit 1
fi
exec 9>"$build_dir/interp-find.lock"
flock 9

preflight_cache_dir=${CNC_SIM_PREFLIGHT_CACHE_DIR:-"$build_dir/preflight-cache"}
LINUXCNC_ROOT="$linuxcnc_root" ./check-linuxcnc-inputs-cached.sh --cache-dir "$preflight_cache_dir" --root-only
linuxcnc_root=$(cd "$linuxcnc_root" && pwd)

tooldata_mmap_backend_shim=$(./list-linuxcnc-wasm-safe-shims.sh tooldata_mmap_backend.cc)
tooldata_runtime_stubs_shim=$(./list-linuxcnc-wasm-safe-shims.sh tooldata_runtime_stubs.cc)
rtapi_compat_shim=$(./list-linuxcnc-wasm-safe-shims.sh rtapi_compat.cc)
linuxcnc_source_output=$(LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-source-files.sh \
    src/emc/tooldata/tooldata_common.cc \
    src/emc/rs274ngc/interp_find.cc)
mapfile -t linuxcnc_sources <<<"$linuxcnc_source_output"
tooldata_common_source=${linuxcnc_sources[0]}
interp_find_source=${linuxcnc_sources[1]}
grep -F 'LinuxCNC source basis: src/emc/rs274ngc/interp_find.cc implements' core/wasm_shims/interp_find_probe_main.cc >/dev/null
grep -F 'LinuxCNC source basis: src/emc/rs274ngc/rs274ngc_interp.hh declares' core/wasm_shims/interp_find_probe_stubs.cc >/dev/null

common_flag_output=$(LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-wasm-common-cxxflags.sh --function-sections --data-sections)
mapfile -t common_flags <<<"$common_flag_output"

sources=(
    "$tooldata_common_source"
    "$tooldata_mmap_backend_shim"
    "$tooldata_runtime_stubs_shim"
    "$rtapi_compat_shim"
    core/wasm_shims/interp_find_probe_stubs.cc
    "$interp_find_source"
    core/wasm_shims/interp_find_probe_main.cc
)
objects=(
    "$build_dir/tooldata_common.o"
    "$build_dir/tooldata_mmap_backend.o"
    "$build_dir/tooldata_runtime_stubs.o"
    "$build_dir/rtapi_compat.o"
    "$build_dir/interp_find_probe_stubs.o"
    "$build_dir/interp_find.o"
    "$build_dir/interp_find_probe_main.o"
)

signature="$build_dir/interp-find.signature"
next_signature="$build_dir/interp-find.signature.next"
{
    printf 'CXX=%s\n' "$cxx"
    printf 'PWD=%s\n' "$PWD"
    printf 'LINUXCNC_ROOT=%s\n' "$linuxcnc_root"
    printf 'BUILD_RULE_VERSION=1\n'
    printf 'COMMON_FLAGS='
    printf ' %s' "${common_flags[@]}"
    printf '\n'
    printf 'SOURCES:\n'
    printf '%s\n' "${sources[@]}"
} > "$next_signature"
if [[ ! -f "$signature" ]] || ! cmp -s "$signature" "$next_signature"; then
    rm -f "$build_dir"/*.o "$build_dir"/*.d "$build_dir/interp_find_probe"
fi
mv "$next_signature" "$signature"

makefile="$build_dir/interp-find.mk"
{
    printf 'CXX := %s\n' "$cxx"
    printf 'COMMON_FLAGS :='
    printf ' %s' "${common_flags[@]}"
    printf '\n\n'
    printf 'OBJECTS :='
    printf ' %s' "${objects[@]}"
    printf '\n\n'
    printf 'DEP_FILES :=\n'
    for object in "${objects[@]}"; do
        printf 'DEP_FILES += %s.d\n' "$object"
    done
    printf '\n.PHONY: all\n'
    printf 'all: %s/interp_find_probe\n\n' "$build_dir"

    object_index=0
    for source in "${sources[@]}"; do
        printf '%s: %s\n' "${objects[$object_index]}" "$source"
        printf '\t@$(CXX) $(COMMON_FLAGS) -MMD -MP -MF %s.d -c %s -o %s\n\n' "${objects[$object_index]}" "$source" "${objects[$object_index]}"
        object_index=$((object_index + 1))
    done

    printf '%s/interp_find_probe: $(OBJECTS)\n' "$build_dir"
    printf '\t@$(CXX) $(OBJECTS) -Wl,--gc-sections -o %s/interp_find_probe\n\n' "$build_dir"
    printf '\n-include $(DEP_FILES)\n'
} > "$makefile"
make --output-sync=target -j"$build_jobs" -f "$makefile" >/dev/null

"$build_dir/interp_find_probe"

echo "linuxcnc wasm interp_find link probe passed"