#!/bin/bash

# Unified build script for YAMS
#
# Usage: ./setup.sh [Debug|Release|Profiling|Fuzzing] [--coverage] [--tsan] [--no-tsan]
#   build_type: Release (default), Debug, Profiling, or Fuzzing (TODO)
#   --coverage: Enable code coverage instrumentation (Debug builds only)
#   --tsan: Enable ThreadSanitizer for race detection (default for Debug builds)
#   --no-tsan: Disable ThreadSanitizer (overrides default)
#
# Environment variables (for CI/advanced use):
#   YAMS_CONAN_HOST_PROFILE  - Path to Conan host profile (bypasses auto-detection)
#   YAMS_CONAN_ARCH          - Target architecture (x86_64, armv8, etc.)
#   YAMS_EXTRA_MESON_FLAGS   - Additional Meson setup flags
#   YAMS_ENABLE_MOBILE_BINDINGS - Build mobile C ABI shared library (true/false)
#   YAMS_COMPILER            - Force compiler (clang or gcc)
#   YAMS_CPPSTD              - C++ standard (17, 20, 23)
#   YAMS_ENABLE_MODULES      - Enable C++20 modules (true/false, auto-detected if not set)
#   YAMS_LIBCXX_HARDENING    - libc++ hardening mode (none, fast, extensive, debug)
#   YAMS_INSTALL_PREFIX      - Installation prefix (default: /usr/local or Homebrew)
#   YAMS_ONNX_GPU            - GPU provider for ONNX (auto|cuda|coreml|migraphx|none, default: auto)
#                              auto: macOS=coreml, Linux+NVIDIA=cuda, Linux+ROCm=migraphx, otherwise=none
#                              migraphx: AMD ROCm GPU acceleration (builds ONNX Runtime from source)
#   YAMS_ROCM_PATH           - ROCm installation path for MIGraphX builds (default: /opt/rocm)
#   YAMS_ROCM_GPU_TARGETS    - Semicolon-separated AMD GPU architectures for MIGraphX builds
#                              (default: gfx900;gfx906;gfx908;gfx90a;gfx1030;gfx1100)
#                              Common targets:
#                                gfx900  = Vega 10 (MI25, Vega 56/64)
#                                gfx906  = Vega 20 (MI50/60, Radeon VII)
#                                gfx908  = CDNA (MI100)
#                                gfx90a  = CDNA2 (MI200 series: MI210, MI250, MI250X)
#                                gfx942  = CDNA3 (MI300 series: MI300A, MI300X, MI325X)
#                                gfx1030 = RDNA2 (RX 6800/6900 series)
#                                gfx1100 = RDNA3 (RX 7900 series)
#                                gfx1101 = RDNA3 (RX 7700/7800 series)
#                                gfx1200 = RDNA4 (RX 9070 series)
#
# The script prefers Clang when available, falling back to GCC otherwise. It
# ensures Conan is given a concrete C++ standard so dependencies resolve cleanly
# and keeps Meson in sync with the generated toolchain file.

set -euo pipefail

ENABLE_COVERAGE=false
ENABLE_TSAN="${ENABLE_TSAN:-}"  # Preserve environment variable if set
BUILD_TYPE_INPUT=""

while [[ $# -gt 0 ]]; do
  case "$1" in
    --coverage)
      ENABLE_COVERAGE=true
      shift
      ;;
    --tsan)
      ENABLE_TSAN=true
      shift
      ;;
    --no-tsan)
      ENABLE_TSAN=false
      shift
      ;;
    Debug|Release|Profiling|Fuzzing|debug|release|profiling|fuzzing)
      if [[ -n "${BUILD_TYPE_INPUT}" ]]; then
        echo "Error: Build type specified multiple times" >&2
        exit 1
      fi
      BUILD_TYPE_INPUT="$1"
      shift
      ;;
    *)
      echo "Usage: $0 [Debug|Release|Profiling|Fuzzing] [--coverage] [--tsan] [--no-tsan]" >&2
      exit 1
      ;;
  esac
done

BUILD_TYPE_INPUT=${BUILD_TYPE_INPUT:-Release}
BUILD_TYPE_INPUT_LOWER=$(echo "${BUILD_TYPE_INPUT}" | tr '[:upper:]' '[:lower:]')

case "${BUILD_TYPE_INPUT_LOWER}" in
  debug)
    BUILD_TYPE="Debug"
    ;;
  release)
    BUILD_TYPE="Release"
    ;;
  profiling)
    BUILD_TYPE="Debug"  # Use Debug as base for Conan/Meson
    ENABLE_PROFILING=true
    ;;
  fuzzing)
    BUILD_TYPE="Debug"  # Use Debug as base for Conan/Meson
    ENABLE_FUZZING=true
    echo "Fuzzing build: AFL++/libFuzzer instrumentation enabled"
    ;;
  *)
    echo "Unknown build type: ${BUILD_TYPE_INPUT}. Expected Debug, Release, Profiling, or Fuzzing." >&2
    exit 1
    ;;
esac

if [[ "${ENABLE_COVERAGE}" == "true" ]] && [[ "${BUILD_TYPE}" != "Debug" ]]; then
  echo "Error: --coverage flag requires Debug build type" >&2
  exit 1
fi

# Select desired C++ standard (defaults to C++23 if compiler supports it, otherwise C++20).
# Override with YAMS_CPPSTD=20/23.
# Auto-detection: test compiler for C++23 support (constexpr containers)
if [[ -z "${YAMS_CPPSTD:-}" ]]; then
  # Test if compiler supports C++23 with constexpr containers
  if echo '#include <vector>
#if __cplusplus >= 202302L && defined(__cpp_lib_constexpr_vector) && __cpp_lib_constexpr_vector >= 201907L
#define HAS_CPP23_CONTAINERS 1
#else
#define HAS_CPP23_CONTAINERS 0
#endif
#if !HAS_CPP23_CONTAINERS
#error "No C++23 constexpr containers"
#endif' | "${CXX:-c++}" -std=c++23 -fsyntax-only -x c++ - >/dev/null 2>&1; then
    CPPSTD_INPUT="23"
    echo "Auto-detected C++23 compiler support (constexpr containers available)"
  else
    CPPSTD_INPUT="20"
    echo "C++23 not fully supported, using C++20"
  fi
else
  CPPSTD_INPUT="${YAMS_CPPSTD}"
fi

case "${CPPSTD_INPUT}" in
  17|20|23)
    CPPSTD="${CPPSTD_INPUT}"
    ;;
  c++17|c++20|c++23)
    CPPSTD="${CPPSTD_INPUT#c++}"
    ;;
  *)
    echo "Unknown C++ standard: ${CPPSTD_INPUT}. Expected 17|20|23 or c++17|c++20|c++23." >&2
    exit 1
    ;;
esac
# Value for Meson project option
MESON_CPPSTD="c++${CPPSTD}"

if [[ -z "${YAMS_ENABLE_MODULES:-}" ]]; then
  ENABLE_MODULES=false
  if echo 'export module test; export int foo() { return 42; }' | \
     "${CXX:-c++}" -std=c++20 -fmodules -fsyntax-only -x c++ - >/dev/null 2>&1; then
    ENABLE_MODULES=true
    echo "Auto-detected C++20 module support (Clang -fmodules)"
  elif echo 'export module test; export int foo() { return 42; }' | \
       "${CXX:-c++}" -std=c++20 -fmodules-ts -fsyntax-only -x c++ - >/dev/null 2>&1; then
    ENABLE_MODULES=true
    echo "Auto-detected C++20 module support (GCC -fmodules-ts)"
  else
    echo "C++20 modules not fully supported by compiler, disabled"
  fi
else
  if [[ "${YAMS_ENABLE_MODULES}" == "true" ]]; then
    ENABLE_MODULES=true
    echo "C++20 modules enabled via YAMS_ENABLE_MODULES=true"
  else
    ENABLE_MODULES=false
    echo "C++20 modules disabled via YAMS_ENABLE_MODULES=false"
  fi
fi

# libc++ hardening mode (only applies when using libc++)
# Options: none, fast, extensive, debug
# Override with YAMS_LIBCXX_HARDENING=fast|extensive|debug|none
LIBCXX_HARDENING=${YAMS_LIBCXX_HARDENING:-none}

# Check if using explicit Conan profile (CI mode)
CONAN_HOST_PROFILE=${YAMS_CONAN_HOST_PROFILE:-}
CONAN_ARCH=${YAMS_CONAN_ARCH:-}
CONAN_EXTRA_OPTIONS=${YAMS_CONAN_EXTRA_OPTIONS:-}
USE_PROFILE=false

if [[ -n "${CONAN_HOST_PROFILE}" ]]; then
  if [[ ! -f "${CONAN_HOST_PROFILE}" ]]; then
    echo "ERROR: Conan host profile not found: ${CONAN_HOST_PROFILE}" >&2
    exit 1
  fi
  USE_PROFILE=true
  echo "Using explicit Conan host profile: ${CONAN_HOST_PROFILE}"
fi

CONAN_ARGS=(-s "build_type=${BUILD_TYPE}" -b missing --update)

# Add common Conan options (sqlite3 with FTS5, etc.)
CONAN_ARGS+=(-o "sqlite3/*:fts5=True")

# Fuzzing builds: disable programs in zstd to avoid symlink issues in Docker
if [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  CONAN_ARGS+=(-o "zstd/*:build_programs=False")
fi

# Add extra Conan options from environment (for CI)
if [[ -n "${CONAN_EXTRA_OPTIONS}" ]]; then
  # shellcheck disable=SC2086
  read -ra EXTRA_OPTS <<< "${CONAN_EXTRA_OPTIONS}"
  CONAN_ARGS+=("${EXTRA_OPTS[@]}")
  echo "Extra Conan options: ${CONAN_EXTRA_OPTIONS}"
fi

detect_version() {
  local bin="$1"
  if command -v "${bin}" >/dev/null 2>&1; then
    if "${bin}" -dumpfullversion >/dev/null 2>&1; then
      "${bin}" -dumpfullversion
    elif "${bin}" -dumpversion >/dev/null 2>&1; then
      "${bin}" -dumpversion
    else
      "${bin}" --version | head -n1 | grep -oE '[0-9]+(\.[0-9]+)*' | head -n1
    fi
  fi
}

COERCE_MAJOR() {
  echo "$1" | cut -d. -f1
}

COMPILER_OVERRIDE=${YAMS_COMPILER:-}

# When using explicit profile, skip auto-detection of compiler settings
if [[ "${USE_PROFILE}" == "true" ]]; then
  echo "Skipping compiler auto-detection (using profile: ${CONAN_HOST_PROFILE})"
  # Still set CC/CXX for Meson if not already set
  if [[ -z "${CC:-}" ]] && command -v clang >/dev/null 2>&1; then
    export CC="clang"
    export CXX="clang++"
  elif [[ -z "${CC:-}" ]] && command -v gcc >/dev/null 2>&1; then
    export CC="gcc"
    export CXX="g++"
  fi
  # Add profile to Conan args
  CONAN_ARGS+=(-pr:h "${CONAN_HOST_PROFILE}" -pr:b default)
  # Add architecture if specified
  if [[ -n "${CONAN_ARCH}" ]]; then
    CONAN_ARGS+=(-s:h "arch=${CONAN_ARCH}")
    echo "Target architecture: ${CONAN_ARCH}"
  fi
else
  # Auto-detection mode (original behavior)
  if [[ "${COMPILER_OVERRIDE}" == clang ]] || { [[ -z "${COMPILER_OVERRIDE}" ]] && command -v clang >/dev/null 2>&1 && command -v clang++ >/dev/null 2>&1; }; then
    echo "--- Using Clang toolchain ---"
    export CC="clang"
    export CXX="clang++"
    # macOS requires libc++, Linux can use libstdc++11
    if [[ "$(uname -s)" == "Darwin" ]]; then
      LIBCXX="libc++"
    else
      LIBCXX="libstdc++11"
    fi
    CLANG_VERSION=$(detect_version clang++)
    CLANG_MAJOR=$(COERCE_MAJOR "${CLANG_VERSION:-0}")
    if [[ -z "${CLANG_MAJOR}" || "${CLANG_MAJOR}" == 0 ]]; then
      echo "Unable to detect clang version." >&2
      exit 1
    fi
    if [[ "$(uname -s)" == "Darwin" ]]; then
      # On macOS use Conan's apple-clang compiler model
      CONAN_ARGS+=(
        -s "compiler=apple-clang"
        -s "compiler.version=${CLANG_MAJOR}"
        -s "compiler.libcxx=${LIBCXX}"
        -s "compiler.cppstd=${CPPSTD}"
      )
    else
      CONAN_ARGS+=(
        -s "compiler=clang"
        -s "compiler.version=${CLANG_MAJOR}"
        -s "compiler.libcxx=${LIBCXX}"
        -s "compiler.cppstd=${CPPSTD}"
      )
    fi
  else
    echo "--- Using GCC toolchain ---"
    export CC="gcc"
    export CXX="g++"
    if ! command -v g++ >/dev/null 2>&1; then
      echo "g++ not found in PATH." >&2
      exit 1
    fi
    GCC_VERSION=$(detect_version g++)
    GCC_MAJOR=$(COERCE_MAJOR "${GCC_VERSION:-0}")
    if [[ -z "${GCC_MAJOR}" || "${GCC_MAJOR}" == 0 ]]; then
      echo "Unable to detect gcc version." >&2
      exit 1
    fi
    CONAN_ARGS+=(
      -s "compiler=gcc"
      -s "compiler.version=${GCC_MAJOR}"
      -s "compiler.libcxx=libstdc++11"
      -s "compiler.cppstd=${CPPSTD}"
    )
  fi
fi

# Wrap CC/CXX with ccache if available for faster incremental builds
if command -v ccache >/dev/null 2>&1; then
  echo "--- Enabling ccache for faster incremental builds ---"
  # Only wrap if not already wrapped
  if [[ "${CC:-}" != ccache* ]]; then
    export CC="ccache ${CC:-cc}"
    export CXX="ccache ${CXX:-c++}"
  fi
  # Show ccache stats if verbose
  ccache -s 2>/dev/null | head -5 || true
fi

if [[ "${ENABLE_PROFILING:-false}" == "true" ]]; then
  BUILD_DIR="build/profiling"
  CONAN_SUBDIR="build-profiling"
  # Conan often writes profiling/debug toolchains under a nested build-debug directory
  CONAN_ALT_SUBDIR="build-debug"
  BUILD_TYPE_MESON_LOWER="debug"
elif [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  BUILD_DIR="build/fuzzing"
  CONAN_SUBDIR="build-fuzzing"
  BUILD_TYPE_MESON_LOWER="debug"
elif [[ "${BUILD_TYPE}" == "Debug" ]]; then
  BUILD_DIR="builddir"
  CONAN_SUBDIR="build-debug"
  BUILD_TYPE_MESON_LOWER="debug"
else
  BUILD_DIR="build/${BUILD_TYPE_INPUT_LOWER}"
  CONAN_SUBDIR="build-${BUILD_TYPE_INPUT_LOWER}"
  BUILD_TYPE_MESON_LOWER="${BUILD_TYPE_INPUT_LOWER}"
fi

# Some Conan generators ignore --output-folder and always emit to build-<build_type>.
# Remember locations to probe for toolchains and Meson cache files.
CONAN_GENERATED_DIR="${BUILD_DIR}/build-${BUILD_TYPE_INPUT_LOWER}"
CONAN_ALT_DIR="${BUILD_DIR}/${CONAN_ALT_SUBDIR:-}"

# Detect install prefix based on platform
if [[ "$(uname -s)" == "Darwin" ]]; then
  # macOS: check for Homebrew installation
  if command -v brew >/dev/null 2>&1; then
    BREW_PREFIX=$(brew --prefix 2>/dev/null || echo "/usr/local")
    INSTALL_PREFIX="${BREW_PREFIX}"
  else
    INSTALL_PREFIX="/usr/local"
  fi
else
  # Linux: use standard prefix
  INSTALL_PREFIX="/usr/local"
fi

# Allow override via environment
INSTALL_PREFIX="${YAMS_INSTALL_PREFIX:-${INSTALL_PREFIX}}"

echo "Build Type:        ${BUILD_TYPE_INPUT}"
if [[ "${ENABLE_PROFILING:-false}" == "true" ]]; then
  echo "Profiling:         Tracy enabled"
fi
if [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  echo "Fuzzing:           AFL++/libFuzzer enabled"
fi
echo "Build Dir:         ${BUILD_DIR}"
echo "Install Prefix:    ${INSTALL_PREFIX}"
echo "C++ Std:           ${MESON_CPPSTD} (Conan: ${CPPSTD})"
if [[ "${LIBCXX_HARDENING}" != "none" ]]; then
  echo "libc++ Hardening:  ${LIBCXX_HARDENING}"
fi

echo "--- Exporting custom Conan recipes... ---"
# qpdf export removed - PDF plugin will be updated in separate PBI

# Export custom onnxruntime recipe if it exists
if [[ -f "conan/onnxruntime/conanfile.py" ]]; then
  # Avoid sporadic Conan cache race/collision errors by retrying export once.
  echo "Exporting onnxruntime/1.23.2 from conan/onnxruntime/"
  if ! conan export conan/onnxruntime --name=onnxruntime --version=1.23.2; then
    echo "Retrying conan export onnxruntime/1.23.2..."
    conan export conan/onnxruntime --name=onnxruntime --version=1.23.2
  fi
fi

echo "--- Running conan install... ---"
# Add policy toolchain for legacy recipes if in Docker or CI
POLICY_TC=""
if [[ -n "${DOCKERFILE_CONF_REV:-}" ]] || [[ -n "${CI:-}" ]]; then
  POLICY_TC="/tmp/yams_policy_toolchain.cmake"
  echo 'cmake_policy(VERSION 3.5)' > "$POLICY_TC"
  CONAN_ARGS+=(-c "tools.cmake.cmaketoolchain:user_toolchain+=${POLICY_TC}")
fi

# Enable tests and benchmarks for Debug builds in Conan (needed for Catch2/gtest/benchmark dependencies)
if [[ "${BUILD_TYPE}" == "Debug" ]] || [[ "${ENABLE_PROFILING:-false}" == "true" ]] || [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  CONAN_ARGS+=(-o build_tests=True)
  CONAN_ARGS+=(-o build_benchmarks=True)
fi

# Enable Tracy profiling for profiling builds
if [[ "${ENABLE_PROFILING:-false}" == "true" ]]; then
  echo "Tracy profiling enabled"
  CONAN_ARGS+=(-o "tracy/*:enable=True")
fi

# Handle optional feature flags from environment
if [[ "${YAMS_DISABLE_ONNX:-}" == "true" ]]; then
  echo "ONNX support disabled (YAMS_DISABLE_ONNX=true)"
  CONAN_ARGS+=(-o "yams/*:enable_onnx=False")
else
  # Auto-detect GPU for ONNX acceleration
  # Override with YAMS_ONNX_GPU=cuda|coreml|migraphx|none
  ONNX_GPU="${YAMS_ONNX_GPU:-auto}"
  if [[ "${ONNX_GPU}" == "auto" ]]; then
    if [[ "$(uname -s)" == "Darwin" ]]; then
      # macOS: CoreML is included in standard ONNX Runtime builds
      ONNX_GPU="coreml"
      echo "macOS detected: enabling CoreML GPU acceleration (Neural Engine + GPU)"
    elif command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi >/dev/null 2>&1; then
      # NVIDIA GPU detected - now validate CUDA toolkit is actually available
      # nvidia-smi presence alone doesn't guarantee CUDA runtime is usable
      if command -v nvcc >/dev/null 2>&1; then
        # nvcc is available (CUDA toolkit installed via package manager or CUDA SDK)
        ONNX_GPU="cuda"
        echo "NVIDIA GPU + CUDA toolkit (nvcc) detected: enabling CUDA acceleration"
      elif [[ -f /usr/local/cuda/lib64/libcudart.so ]] || \
           [[ -f /usr/local/cuda/lib/libcudart.so ]] || \
           ldconfig -p 2>/dev/null | grep -q libcudart; then
        # CUDA runtime library found (sufficient for inference, even without nvcc)
        ONNX_GPU="cuda"
        echo "NVIDIA GPU + CUDA runtime (libcudart) detected: enabling CUDA acceleration"
      else
        # GPU present but CUDA toolkit/runtime missing - use CPU to avoid runtime failures
        ONNX_GPU="none"
        echo "NVIDIA GPU detected but CUDA toolkit missing: using CPU-only ONNX Runtime"
        echo "  Install CUDA toolkit for GPU acceleration: https://developer.nvidia.com/cuda-downloads"
      fi
    elif [[ -d /opt/rocm ]] || command -v rocm-smi >/dev/null 2>&1; then
      # ROCm detected - use MIGraphX (builds ONNX Runtime from source)
      # Detect ROCm path
      if [[ -d /opt/rocm ]]; then
        ROCM_PATH="/opt/rocm"
      elif command -v rocm-smi >/dev/null 2>&1; then
        ROCM_PATH="$(dirname "$(dirname "$(command -v rocm-smi)")")"
      fi
      # Check if MIGraphX is installed
      if [[ -f "${ROCM_PATH}/lib/libmigraphx.so" ]]; then
        ONNX_GPU="migraphx"
        echo "ROCm + MIGraphX detected at ${ROCM_PATH}: enabling AMD GPU acceleration"
        echo "  Note: ONNX Runtime will be built from source (takes ~20-30 minutes)"

        # Auto-detect GPU architecture using rocminfo
        if [[ -z "${YAMS_ROCM_GPU_TARGETS:-}" ]] && command -v rocminfo >/dev/null 2>&1; then
          DETECTED_TARGETS=$(rocminfo 2>/dev/null | grep -oP 'gfx[0-9a-z]+' | sort -u | tr '\n' ';' | sed 's/;$//')
          if [[ -n "${DETECTED_TARGETS}" ]]; then
            export YAMS_ROCM_GPU_TARGETS="${DETECTED_TARGETS}"
            echo "  Auto-detected GPU target(s): ${DETECTED_TARGETS}"
          fi
        fi
      else
        ONNX_GPU="none"
        echo "ROCm detected at ${ROCM_PATH} but MIGraphX not found"
        echo "  Using CPU-only ONNX Runtime"
        echo "  Install MIGraphX for GPU acceleration: https://github.com/ROCm/AMDMIGraphX"
      fi
    else
      ONNX_GPU="none"
      echo "No GPU detected: using CPU-only ONNX Runtime"
    fi
  fi

  # Handle MIGraphX-specific options
  if [[ "${ONNX_GPU}" == "migraphx" ]]; then
    # Set ROCm path for conan (use detected path or environment override)
    ROCM_PATH="${YAMS_ROCM_PATH:-${ROCM_PATH:-/opt/rocm}}"
    if [[ ! -d "${ROCM_PATH}" ]]; then
      echo "ERROR: ROCm not found at ${ROCM_PATH}" >&2
      echo "  Set YAMS_ROCM_PATH to your ROCm installation path" >&2
      exit 1
    fi
    CONAN_ARGS+=(-o "onnxruntime/*:rocm_path=${ROCM_PATH}")
    echo "ROCm path: ${ROCM_PATH}"

    # Pass GPU targets if specified
    if [[ -n "${YAMS_ROCM_GPU_TARGETS:-}" ]]; then
      echo "ROCm GPU targets: ${YAMS_ROCM_GPU_TARGETS}"
    fi
  fi

  if [[ "${ONNX_GPU}" != "none" ]]; then
    CONAN_ARGS+=(-o "onnxruntime/*:with_gpu=${ONNX_GPU}")
    echo "ONNX GPU provider: ${ONNX_GPU}"
  fi
fi

if [[ "${YAMS_DISABLE_SYMBOL_EXTRACTION:-}" == "true" ]]; then
  echo "Symbol extraction disabled (YAMS_DISABLE_SYMBOL_EXTRACTION=true)"
  CONAN_ARGS+=(-o "yams/*:enable_symbol_extraction=False")
fi

if [[ "${YAMS_DISABLE_PDF:-}" == "true" ]]; then
  echo "PDF support disabled (YAMS_DISABLE_PDF=true)"
  CONAN_ARGS+=(-o "yams/*:enable_pdf=False")
fi

# Force building missing packages to ensure ABI compatibility
# This is especially important for C++23 with Clang + libstdc++
CONAN_ARGS+=(--build=missing)

# Use runtime_deploy to copy shared libraries next to executables (mainly for Windows, no-op on Unix with RPATH)
conan install . -of "${BUILD_DIR}" "${CONAN_ARGS[@]}" --deployer=runtime_deploy --deployer-folder="${BUILD_DIR}"

# Check for either native or cross file (Conan generates cross file for cross-compilation)
# Conan 2.x sometimes nests generator output under build-<type>/conan even when -of points at
# ${BUILD_DIR}. Probe primary, then CONAN_GENERATED_DIR, then the legacy root, then CONAN_ALT_DIR.
NATIVE_FILE_PRIMARY="${BUILD_DIR}/${CONAN_SUBDIR}/conan/conan_meson_native.ini"
CROSS_FILE_PRIMARY="${BUILD_DIR}/${CONAN_SUBDIR}/conan/conan_meson_cross.ini"
NATIVE_FILE_SECONDARY="${CONAN_GENERATED_DIR}/conan/conan_meson_native.ini"
CROSS_FILE_SECONDARY="${CONAN_GENERATED_DIR}/conan/conan_meson_cross.ini"
NATIVE_FILE_TERTIARY="${BUILD_DIR}/conan/conan_meson_native.ini"
CROSS_FILE_TERTIARY="${BUILD_DIR}/conan/conan_meson_cross.ini"
NATIVE_FILE_QUAT="${CONAN_ALT_DIR}/conan/conan_meson_native.ini"
CROSS_FILE_QUAT="${CONAN_ALT_DIR}/conan/conan_meson_cross.ini"

if [[ -f "${NATIVE_FILE_PRIMARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--native-file"
  MESON_TOOLCHAIN_FILE="${NATIVE_FILE_PRIMARY}"
elif [[ -f "${CROSS_FILE_PRIMARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--cross-file"
  MESON_TOOLCHAIN_FILE="${CROSS_FILE_PRIMARY}"
elif [[ -f "${NATIVE_FILE_SECONDARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--native-file"
  MESON_TOOLCHAIN_FILE="${NATIVE_FILE_SECONDARY}"
elif [[ -f "${CROSS_FILE_SECONDARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--cross-file"
  MESON_TOOLCHAIN_FILE="${CROSS_FILE_SECONDARY}"
elif [[ -f "${NATIVE_FILE_TERTIARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--native-file"
  MESON_TOOLCHAIN_FILE="${NATIVE_FILE_TERTIARY}"
elif [[ -f "${CROSS_FILE_TERTIARY}" ]]; then
  MESON_TOOLCHAIN_ARG="--cross-file"
  MESON_TOOLCHAIN_FILE="${CROSS_FILE_TERTIARY}"
elif [[ -n "${CONAN_ALT_DIR}" && -f "${NATIVE_FILE_QUAT}" ]]; then
  MESON_TOOLCHAIN_ARG="--native-file"
  MESON_TOOLCHAIN_FILE="${NATIVE_FILE_QUAT}"
elif [[ -n "${CONAN_ALT_DIR}" && -f "${CROSS_FILE_QUAT}" ]]; then
  MESON_TOOLCHAIN_ARG="--cross-file"
  MESON_TOOLCHAIN_FILE="${CROSS_FILE_QUAT}"
else
  echo "Error: Conan meson toolchain file not found" >&2
  echo "Checked: ${NATIVE_FILE_PRIMARY}" >&2
  echo "     and: ${CROSS_FILE_PRIMARY}" >&2
  echo "     and: ${NATIVE_FILE_SECONDARY}" >&2
  echo "     and: ${CROSS_FILE_SECONDARY}" >&2
  echo "     and: ${NATIVE_FILE_TERTIARY}" >&2
  echo "     and: ${CROSS_FILE_TERTIARY}" >&2
  if [[ -n "${CONAN_ALT_DIR}" ]]; then
    echo "     and: ${NATIVE_FILE_QUAT}" >&2
    echo "     and: ${CROSS_FILE_QUAT}" >&2
  fi
  echo "Please check the output path from 'conan install'." >&2
  exit 1
fi

MESON_ARGS=(
  "${BUILD_DIR}"
  "--prefix" "${INSTALL_PREFIX}"
  "${MESON_TOOLCHAIN_ARG}" "${MESON_TOOLCHAIN_FILE}"
  "--buildtype" "${BUILD_TYPE_MESON_LOWER}"
)

# Detect previous configured cpp_std to decide on reconfigure vs wipe
PREV_CPPSTD=""
INTRO_OPTS_JSON="${BUILD_DIR}/meson-info/intro-buildoptions.json"
if [[ -f "${INTRO_OPTS_JSON}" ]]; then
  PREV_CPPSTD=$(awk '/"name"\s*:\s*"cpp_std"/{flag=1} flag && /"value"/{gsub(/.*"value"\s*:\s*"|".*/,"",$0); print; exit}' "${INTRO_OPTS_JSON}" || true)
fi

MESON_OPTIONS=("-Dbuild-cli=true" "-Dcpp_std=${MESON_CPPSTD}" "-Denable-modules=${ENABLE_MODULES}")

# Add libc++ hardening mode if specified
if [[ "${LIBCXX_HARDENING}" != "none" ]]; then
  MESON_OPTIONS+=("-Dlibcxx-hardening=${LIBCXX_HARDENING}")
fi

# Handle optional feature flags for Meson (must match Conan options)
if [[ "${YAMS_DISABLE_ONNX:-}" == "true" ]]; then
  MESON_OPTIONS+=("-Denable-onnx=disabled")
  MESON_OPTIONS+=("-Dplugin-onnx=false")
fi

if [[ "${YAMS_DISABLE_SYMBOL_EXTRACTION:-}" == "true" ]]; then
  MESON_OPTIONS+=("-Denable-symbol-extraction=false")
  MESON_OPTIONS+=("-Dplugin-symbols=false")
fi

if [[ "${YAMS_DISABLE_PDF:-}" == "true" ]]; then
  MESON_OPTIONS+=("-Dplugin-pdf=false")
fi

# libSQL backend: default to libsql but fall back to sqlite on Linux when unavailable
LIBSQL_BACKEND_INPUT=${YAMS_DATABASE_BACKEND:-libsql}
LIBSQL_BACKEND_LOWER=$(echo "${LIBSQL_BACKEND_INPUT}" | tr '[:upper:]' '[:lower:]')
if [[ "${LIBSQL_BACKEND_LOWER}" != "libsql" && "${LIBSQL_BACKEND_LOWER}" != "sqlite" ]]; then
  echo "Error: Unknown database backend: ${LIBSQL_BACKEND_INPUT}. Expected libsql or sqlite." >&2
  exit 1
fi

if [[ "${LIBSQL_BACKEND_LOWER}" == "libsql" && "$(uname -s)" != "Darwin" ]]; then
  LIBSQL_PKG_FOUND=false
  if command -v pkg-config >/dev/null 2>&1; then
    if pkg-config --exists libsql || pkg-config --exists libsql-sqlite3; then
      LIBSQL_PKG_FOUND=true
    fi
  fi

  if [[ "${LIBSQL_PKG_FOUND}" != "true" ]]; then
    if [[ -d "subprojects/libsql" ]]; then
      echo "libSQL subproject detected at subprojects/libsql; keeping libsql backend"
    elif ! command -v cargo >/dev/null 2>&1; then
      echo "libSQL not detected on Linux (pkg-config/cargo missing); falling back to sqlite" >&2
      LIBSQL_BACKEND_LOWER="sqlite"
    fi
  fi
fi

if [[ "${LIBSQL_BACKEND_LOWER}" == "libsql" ]]; then
  echo "Database backend: libsql"
  if [[ -f "subprojects/libsql.wrap" && ! -d "subprojects/libsql" ]]; then
    if command -v meson >/dev/null 2>&1; then
      echo "Fetching libSQL subproject (meson wrap)..."
      if ! meson subprojects download libsql; then
        echo "Failed to fetch libSQL subproject; falling back to sqlite" >&2
        LIBSQL_BACKEND_LOWER="sqlite"
      fi
    else
      echo "Meson not found; cannot fetch libSQL subproject. Falling back to sqlite" >&2
      LIBSQL_BACKEND_LOWER="sqlite"
    fi
  fi
else
  echo "Database backend: sqlite"
fi

MESON_OPTIONS+=("-Ddatabase-backend=${LIBSQL_BACKEND_LOWER}")

# Mobile bindings are off by default; allow CI/users to opt-in.
if [[ "${YAMS_ENABLE_MOBILE_BINDINGS:-}" == "true" ]]; then
  MESON_OPTIONS+=("-Denable-mobile-bindings=true")
  echo "Mobile bindings enabled (YAMS_ENABLE_MOBILE_BINDINGS=true)"
elif [[ "${YAMS_ENABLE_MOBILE_BINDINGS:-}" == "false" ]]; then
  MESON_OPTIONS+=("-Denable-mobile-bindings=false")
  echo "Mobile bindings disabled (YAMS_ENABLE_MOBILE_BINDINGS=false)"
fi

# ThreadSanitizer: default enabled for Debug builds, can be overridden with --tsan/--no-tsan
if [[ -z "${ENABLE_TSAN}" ]]; then
  if [[ "${BUILD_TYPE}" == "Debug" ]] || [[ "${ENABLE_PROFILING:-false}" == "true" ]] || [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
    ENABLE_TSAN=true
  else
    ENABLE_TSAN=false
  fi
fi

if [[ "${BUILD_TYPE}" == "Debug" ]] || [[ "${ENABLE_PROFILING:-false}" == "true" ]] || [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  MESON_OPTIONS+=(
    "-Dbuild-tests=true"
    "-Denable-bench-tests=true"
    "-Denable-vector-tests=true"
  )
  echo "Vector/embedding tests enabled for ${BUILD_TYPE} build"
fi

MESON_OPTIONS+=("-Denable-tsan=${ENABLE_TSAN}")
if [[ "${ENABLE_TSAN}" == "true" ]]; then
  MESON_OPTIONS+=("-Db_sanitize=thread")
  echo "ThreadSanitizer enabled (race detection)"
else
  # Ensure any previous sanitizer setting doesn't persist across reconfigure.
  # Users can override via YAMS_EXTRA_MESON_FLAGS.
  MESON_OPTIONS+=("-Db_sanitize=none")
fi

if [[ "${ENABLE_PROFILING:-false}" == "true" ]]; then
  MESON_OPTIONS+=(
    "-Denable-profiling=true"
  )
  echo "Tracy profiling enabled for Meson build"
fi

if [[ "${ENABLE_FUZZING:-false}" == "true" ]]; then
  MESON_OPTIONS+=(
    "-Dbuild-fuzzers=true"
  )
  echo "Fuzzing targets enabled for Meson build"
fi

if [[ "${ENABLE_COVERAGE}" == "true" ]]; then
  MESON_OPTIONS+=("-Db_coverage=true")
  echo "Coverage instrumentation enabled"
fi

# Add extra Meson flags from environment (for CI)
if [[ -n "${YAMS_EXTRA_MESON_FLAGS:-}" ]]; then
  # shellcheck disable=SC2086
  read -ra EXTRA_FLAGS <<< "${YAMS_EXTRA_MESON_FLAGS}"
  MESON_OPTIONS+=("${EXTRA_FLAGS[@]}")
  echo "Extra Meson flags: ${YAMS_EXTRA_MESON_FLAGS}"
fi

# Auto-enable zyp plugin if Zig 0.15.2+ is installed
if command -v zig >/dev/null 2>&1; then
  ZIG_VERSION=$(zig version 2>/dev/null || echo "0.0.0")
  # Parse major.minor.patch
  ZIG_MAJOR=$(echo "${ZIG_VERSION}" | cut -d. -f1)
  ZIG_MINOR=$(echo "${ZIG_VERSION}" | cut -d. -f2)
  ZIG_PATCH=$(echo "${ZIG_VERSION}" | cut -d. -f3 | cut -d- -f1)  # Handle -dev suffix
  # Check if >= 0.15.2
  if [[ "${ZIG_MAJOR:-0}" -gt 0 ]] || \
     { [[ "${ZIG_MAJOR:-0}" -eq 0 ]] && [[ "${ZIG_MINOR:-0}" -gt 15 ]]; } || \
     { [[ "${ZIG_MAJOR:-0}" -eq 0 ]] && [[ "${ZIG_MINOR:-0}" -eq 15 ]] && [[ "${ZIG_PATCH:-0}" -ge 2 ]]; }; then
    echo "Zig ${ZIG_VERSION} detected, enabling zyp PDF plugin"
    MESON_OPTIONS+=("-Dplugin-zyp=true")
  else
    echo "Zig ${ZIG_VERSION} found but requires 0.15.2+ for zyp plugin"
  fi
fi

# Auto-enable Glint NL entity extractor plugin (GLiNER-based)
# Requires ONNX Runtime which is already a dependency
if [[ "${YAMS_DISABLE_ONNX:-}" != "true" ]]; then
  echo "Enabling Glint NL entity extractor plugin"
  MESON_OPTIONS+=("-Dplugin-glint=true")
fi

echo "--- Running meson setup... ---"
if [[ -n "${PREV_CPPSTD}" ]]; then
  if [[ "${PREV_CPPSTD}" != "${MESON_CPPSTD}" ]]; then
    echo "cpp_std changed (${PREV_CPPSTD} -> ${MESON_CPPSTD}); wiping build directory configuration..."
    meson setup "${MESON_ARGS[@]}" "${MESON_OPTIONS[@]}" --wipe
  else
    meson setup "${MESON_ARGS[@]}" "${MESON_OPTIONS[@]}" --reconfigure
  fi
else
  meson setup "${MESON_ARGS[@]}" "${MESON_OPTIONS[@]}"
fi

echo
echo "--- Setup complete! ---"
echo "To compile, run: meson compile -C ${BUILD_DIR}"