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

cd "$(dirname "$0")"

# LinuxCNC source basis: this syntax probe checks the native RS274 manifest's
# core, binding, and tooldata entries from LinuxCNC src/emc/rs274ngc and
# src/emc/tooldata before source-object and source-link tests consume them.
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
self_checks=${CNC_SIM_SOURCE_SYNTAX_SELF_CHECKS:-0}
if [[ "$self_checks" != "0" && "$self_checks" != "1" ]]; then
    echo "CNC_SIM_SOURCE_SYNTAX_SELF_CHECKS must be 0 or 1: $self_checks" >&2
    exit 1
fi
build_dir=${CNC_SIM_SOURCE_SYNTAX_BUILD_DIR-build/source-syntax-test}
if [[ -z "$build_dir" ]]; then
    echo "CNC_SIM_SOURCE_SYNTAX_BUILD_DIR must not be empty" >&2
    exit 1
fi
if [[ "$build_dir" =~ [[:space:]] ]]; then
    echo "CNC_SIM_SOURCE_SYNTAX_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_SOURCE_SYNTAX_BUILD_DIR must not be the filesystem root" >&2
    exit 1
fi
exec 9>"$build_dir/source-syntax.lock"
flock 9
manifest=${1:-linuxcnc-rs274-source-files.txt}
preflight_cache_dir="$build_dir/preflight-cache"

LINUXCNC_ROOT="$linuxcnc_root" ./check-linuxcnc-inputs-cached.sh --cache-dir "$preflight_cache_dir" "$manifest"

missing_source_manifest=$(mktemp "${TMPDIR:-/tmp}/cnc_sim_missing_rs274_manifest_source.XXXXXX.txt")
missing_source_log=$(mktemp "${TMPDIR:-/tmp}/cnc_sim_missing_rs274_manifest_source.XXXXXX.log")
source_list="$build_dir/linuxcnc_source_syntax_sources.txt"
cleanup_source_syntax_temps() {
    rm -f "$missing_source_manifest" "$missing_source_log"
}
trap cleanup_source_syntax_temps EXIT
grep -Fx 'trap cleanup_source_syntax_temps EXIT' test-linuxcnc-source-syntax.sh >/dev/null
if awk '/^rm -f "\$missing_source_manifest" "\$missing_source_log"$/{ found = 1 } END { exit !found }' test-linuxcnc-source-syntax.sh >/dev/null; then
    echo "linuxcnc source syntax probe still relies on manual cleanup for missing-source temporaries" >&2
    exit 1
fi
CNC_SIM_PREFLIGHT_CACHE_DIR="$preflight_cache_dir" LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-source-manifest-sources.sh "$manifest" core full > "$source_list"
CNC_SIM_PREFLIGHT_CACHE_DIR="$preflight_cache_dir" LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-source-manifest-sources.sh "$manifest" binding full >> "$source_list"
CNC_SIM_PREFLIGHT_CACHE_DIR="$preflight_cache_dir" LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-source-manifest-sources.sh "$manifest" tooldata full >> "$source_list"

if [[ "$self_checks" == "1" ]]; then

printf 'core:src/emc/rs274ngc/does_not_exist.cc:test missing source\n' >"$missing_source_manifest"
if ./test-linuxcnc-source-objects.sh "$missing_source_manifest" >"$missing_source_log" 2>&1; then
    echo "linuxcnc source object probe accepted missing manifest source" >&2
    exit 1
fi
if ! grep -F "missing manifest source: src/emc/rs274ngc/does_not_exist.cc" "$missing_source_log" >/dev/null; then
    echo "linuxcnc source object probe did not report missing manifest source clearly" >&2
    sed -n '1,20p' "$missing_source_log" >&2
    exit 1
fi
if ./test-linuxcnc-source-link.sh "$missing_source_manifest" >"$missing_source_log" 2>&1; then
    echo "linuxcnc source link probe accepted missing manifest source" >&2
    exit 1
fi
if ! grep -F "missing manifest source: src/emc/rs274ngc/does_not_exist.cc" "$missing_source_log" >/dev/null; then
    echo "linuxcnc source link probe did not report missing manifest source clearly" >&2
    sed -n '1,20p' "$missing_source_log" >&2
    exit 1
fi

fi

common_flag_output=$(LINUXCNC_ROOT="$linuxcnc_root" ./list-linuxcnc-source-common-cxxflags.sh --python --syntax-only)
mapfile -t common_flags <<<"$common_flag_output"

source_syntax_signature="$build_dir/source-syntax.signature"
source_syntax_next_signature="$build_dir/source-syntax.signature.next"
{
    printf 'CXX=%s\n' "$cxx"
    printf 'PWD=%s\n' "$PWD"
    printf 'LINUXCNC_ROOT=%s\n' "$(cd "$linuxcnc_root" && pwd)"
    printf 'MANIFEST=%s\n' "$(cd "$(dirname "$manifest")" && pwd)/$(basename "$manifest")"
    printf 'BUILD_RULE_VERSION=1\n'
    printf 'COMMON_FLAGS='
    printf ' %s' "${common_flags[@]}"
    printf '\n'
    printf 'SOURCES:\n'
    cat "$source_list"
} > "$source_syntax_next_signature"
if [[ ! -f "$source_syntax_signature" ]] || ! cmp -s "$source_syntax_signature" "$source_syntax_next_signature"; then
    rm -f "$build_dir"/source_*.stamp "$build_dir"/source_*.d
fi
mv "$source_syntax_next_signature" "$source_syntax_signature"

source_syntax_makefile="$build_dir/source-syntax.mk"
{
    printf 'CXX := %s\n' "$cxx"
    printf 'COMMON_FLAGS :='
    printf ' %s' "${common_flags[@]}"
    printf '\n\n'
    printf 'STAMPS :=\n'
    source_index=0
    while IFS= read -r source; do
        [[ -z "$source" ]] && continue
        printf 'STAMPS += %s/source_%s.stamp\n' "$build_dir" "$source_index"
        source_index=$((source_index + 1))
    done < "$source_list"
    printf '\n.PHONY: all\n'
    printf 'all: $(STAMPS)\n\n'

    source_index=0
    while IFS= read -r source; do
        [[ -z "$source" ]] && continue
        printf '%s/source_%s.stamp: %s\n' "$build_dir" "$source_index" "$source"
        printf '\t@$(CXX) $(COMMON_FLAGS) -MMD -MP -MF %s/source_%s.d %s\n' "$build_dir" "$source_index" "$source"
        printf '\t@touch %s/source_%s.stamp\n\n' "$build_dir" "$source_index"
        source_index=$((source_index + 1))
    done < "$source_list"

    printf '%s' '-include'
    source_index=0
    while IFS= read -r source; do
        [[ -z "$source" ]] && continue
        printf ' %s/source_%s.d' "$build_dir" "$source_index"
        source_index=$((source_index + 1))
    done < "$source_list"
    printf '\n'
} > "$source_syntax_makefile"
make --output-sync=target -j"$build_jobs" -f "$source_syntax_makefile"

echo "linuxcnc rs274 source syntax probe passed"