#!/bin/bash
# Build script for Windows ARM64 using MinGW-w64

set -e

# Add ARM64 toolchain to PATH
export PATH="/clangarm64/bin:$PATH"


echo "Building Vectorscan for Windows ARM64..."

# Check if we're running in the right environment
if [[ "$MSYSTEM" != "MINGW64" && "$MSYSTEM" != "UCRT64" ]]; then
    echo "Warning: This script should be run in MSYS2 MINGW64 or UCRT64 environment"
    echo "Current MSYSTEM: $MSYSTEM"
fi

# Set environment variables for cross-compilation
export CMAKE_SYSTEM_NAME=Windows
export CMAKE_SYSTEM_PROCESSOR=ARM64

# Try to find the ARM64 cross-compiler
ARM64_GCC=""
ARM64_GPP=""
ARM64_AR=""
ARM64_RANLIB=""

# Check common locations for ARM64 cross-compiler
# First check if we have the ARM64 CLANG tools available
if command -v /clangarm64/bin/aarch64-w64-mingw32-gcc.exe &> /dev/null; then
    ARM64_GCC="/clangarm64/bin/aarch64-w64-mingw32-gcc.exe"
    ARM64_GPP="/clangarm64/bin/aarch64-w64-mingw32-g++.exe"
    # Use LLVM archiver since it's available
    ARM64_AR="/clangarm64/bin/llvm-ar.exe"
    ARM64_RANLIB="/clangarm64/bin/llvm-ranlib.exe"
    echo "Using CLANGARM64 ARM64 cross-compilation toolchain"
elif [[ "$MSYSTEM" == "CLANGARM64" ]] && command -v clang &> /dev/null && command -v clang++ &> /dev/null; then
    ARM64_GCC="/clangarm64/bin/clang.exe"
    ARM64_GPP="/clangarm64/bin/clang++.exe"
    ARM64_AR="/clangarm64/bin/llvm-ar.exe"
    ARM64_RANLIB="/clangarm64/bin/llvm-ranlib.exe"
    echo "Using CLANGARM64 native toolchain"
elif command -v aarch64-w64-mingw32-gcc &> /dev/null; then
    ARM64_GCC="aarch64-w64-mingw32-gcc"
    ARM64_GPP="aarch64-w64-mingw32-g++"
    ARM64_AR="aarch64-w64-mingw32-ar"
    ARM64_RANLIB="aarch64-w64-mingw32-ranlib"
    echo "Using mingw-w64 GCC toolchain"
elif command -v /mingw64/bin/aarch64-w64-mingw32-gcc &> /dev/null; then
    ARM64_GCC="/mingw64/bin/aarch64-w64-mingw32-gcc"
    ARM64_GPP="/mingw64/bin/aarch64-w64-mingw32-g++"
    ARM64_AR="/mingw64/bin/aarch64-w64-mingw32-ar"
    ARM64_RANLIB="/mingw64/bin/aarch64-w64-mingw32-ranlib"
    echo "Using mingw64 GCC toolchain"
else
    echo "Error: ARM64 cross-compiler not found!"
    echo "Please install mingw-w64 ARM64 toolchain:"
    echo "  pacman -S mingw-w64-clang-aarch64-gcc mingw-w64-clang-aarch64-g++"
    exit 1
fi

echo "Using ARM64 cross-compiler: $ARM64_GCC"

# Set compiler environment
export CC="$ARM64_GCC"
export CXX="$ARM64_GPP"
export AR="$ARM64_AR"
export RANLIB="$ARM64_RANLIB"

# Create build directory
BUILD_DIR="build-windows-arm64"
echo "Creating build directory: $BUILD_DIR"
mkdir -p "$BUILD_DIR"
cd "$BUILD_DIR"

# Configure with CMake
echo "Configuring with CMake..."
cmake .. \
    -G "MinGW Makefiles" -DCMAKE_MAKE_PROGRAM="mingw32-make" \
    -DCMAKE_SYSTEM_NAME=Windows \
    -DCMAKE_SYSTEM_PROCESSOR=ARM64 \
    -DCMAKE_C_COMPILER="$CC" \
    -DCMAKE_CXX_COMPILER="$CXX" \
    -DCMAKE_AR="$AR" \
    -DCMAKE_RANLIB="$RANLIB" \
    -DCMAKE_FIND_ROOT_PATH_MODE_PROGRAM=NEVER \
    -DCMAKE_FIND_ROOT_PATH_MODE_LIBRARY=ONLY \
    -DCMAKE_FIND_ROOT_PATH_MODE_INCLUDE=ONLY \
    -DBUILD_STATIC_LIBS=ON \
    -DBUILD_SHARED_LIBS=ON \
    -DFAT_RUNTIME=ON \
    -DBUILD_AVX2=OFF \
    -DBUILD_AVX512=OFF \
    -DBUILD_SVE=OFF \
    -DBUILD_SVE2=OFF \
    -DSIMDE_BACKEND=OFF \
    -DSIMDE_NATIVE=OFF \
    -DBUILD_UNIT=ON \
    -DCMAKE_BUILD_TYPE=Release \
    -DCMAKE_C_FLAGS="-D_WIN32_WINNT=0x0A00 -DWIN32_LEAN_AND_MEAN" \
    -DCMAKE_CXX_FLAGS="-D_WIN32_WINNT=0x0A00 -DWIN32_LEAN_AND_MEAN -Wno-deprecated-declarations -Wno-error=deprecated-declarations"

# Build
echo "Building..."
NPROC=$(nproc 2>/dev/null || echo 4)
mingw32-make -j$NPROC

echo ""
echo "Build completed successfully!"
echo ""

echo "libc++.dll to be copied to bin directory..."
cp /clangarm64/bin/libc++.dll "$BUILD_DIR"/bin/libc++.dll

# check the architecture of the built libraries
ldd -p lib/libhs.a | grep "file format" || echo "lib/libhs.a is not a valid static library"

# Check if we can run the tests (only if not cross-compiling)
echo "Checking if tests can be run..."
if file ./bin/unit-hyperscan.exe | grep -q "$(uname -m)"; then
    echo "Running tests..."
    echo "Running unit-hyperscan tests..."
    if ./bin/unit-hyperscan.exe; then
        echo "✓ unit-hyperscan tests passed"
    else
        echo "✗ unit-hyperscan tests failed"
    fi

    echo ""
    echo "Running unit-internal tests..."
    if ./bin/unit-internal.exe; then
        echo "✓ unit-internal tests passed"
    else
        echo "✗ unit-internal tests failed"
    fi

    echo ""
    echo "All tests completed!"
else
    echo "Tests built for different architecture (ARM64), cannot run on current system."
    echo "Tests need to be run on a Windows ARM64 target system."
    echo ""
    echo "Built test executables:"
    echo "  $(pwd)/bin/unit-hyperscan.exe"
    echo "  $(pwd)/bin/unit-internal.exe"
fi
echo ""
echo "Output files:"
echo "  Static libraries:"
echo "    $(pwd)/lib/libhs.a"
echo "    $(pwd)/lib/libhs_runtime.a"
echo "  Shared libraries:"
echo "    $(pwd)/bin/hs.dll"
echo "    $(pwd)/bin/hs_runtime.dll"
echo "  Import libraries:"
echo "    $(pwd)/lib/libhs.dll.a"
echo "    $(pwd)/lib/libhs_runtime.dll.a"
echo ""
echo "To use these libraries, copy them to your target Windows ARM64 system."