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

# Build the Rust FFI bridge for React Native targets.
#
# Usage:
#   ./scripts/build-rust.sh ios       # Build for iOS device + simulator
#   ./scripts/build-rust.sh android   # Build for Android architectures
#   ./scripts/build-rust.sh all       # Build for all targets

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
RUST_DIR="$SCRIPT_DIR/../rust"
IOS_XCFRAMEWORK_OUT="$SCRIPT_DIR/../ios/rust/OndeReactNative.xcframework"
ANDROID_OUT="$SCRIPT_DIR/../android/src/main/jniLibs"
IOS_DEPLOYMENT_TARGET="${IOS_DEPLOYMENT_TARGET:-16.0}"

build_ios() {
    echo "=== Building Rust for iOS ==="

    # Device (arm64)
    echo "  → aarch64-apple-ios"
    IPHONEOS_DEPLOYMENT_TARGET="$IOS_DEPLOYMENT_TARGET" \
    cargo build --manifest-path "$RUST_DIR/Cargo.toml" \
        --target aarch64-apple-ios --release

    # Simulator (arm64 Apple Silicon)
    echo "  → aarch64-apple-ios-sim"
    IPHONEOS_DEPLOYMENT_TARGET="$IOS_DEPLOYMENT_TARGET" \
    cargo build --manifest-path "$RUST_DIR/Cargo.toml" \
        --target aarch64-apple-ios-sim --release

    # Build the XCFramework directly from the cargo output — no staging dirs needed.
    # CocoaPods picks the correct slice (device vs simulator) automatically.
    echo "  → Creating XCFramework"
    rm -rf "$IOS_XCFRAMEWORK_OUT"
    mkdir -p "$(dirname "$IOS_XCFRAMEWORK_OUT")"
    xcodebuild -create-xcframework \
        -library "$RUST_DIR/target/aarch64-apple-ios/release/libonde_react_native.a" \
        -library "$RUST_DIR/target/aarch64-apple-ios-sim/release/libonde_react_native.a" \
        -output "$IOS_XCFRAMEWORK_OUT"

    echo "  ✓ iOS XCFramework created at ios/rust/OndeReactNative.xcframework"
    echo "  ✓ Built with IPHONEOS_DEPLOYMENT_TARGET=$IOS_DEPLOYMENT_TARGET"
}

build_android() {
    echo "=== Building Rust for Android ==="

    # Verify NDK is available
    if [ -z "${ANDROID_NDK_HOME:-}" ]; then
        echo "Error: ANDROID_NDK_HOME is not set."
        echo "Set it to your Android NDK installation path, e.g.:"
        echo "  export ANDROID_NDK_HOME=\$HOME/Library/Android/sdk/ndk/<version>"
        exit 1
    fi

    TARGETS=(
        "aarch64-linux-android:arm64-v8a"
        "armv7-linux-androideabi:armeabi-v7a"
        "x86_64-linux-android:x86_64"
        "i686-linux-android:x86"
    )

    for entry in "${TARGETS[@]}"; do
        RUST_TARGET="${entry%%:*}"
        ABI="${entry##*:}"

        echo "  → $RUST_TARGET ($ABI)"
        cargo build --manifest-path "$RUST_DIR/Cargo.toml" \
            --target "$RUST_TARGET" --release

        mkdir -p "$ANDROID_OUT/$ABI"
        cp "$RUST_DIR/target/$RUST_TARGET/release/libonde_react_native.so" \
           "$ANDROID_OUT/$ABI/libonde_react_native.so"
    done

    echo "  ✓ Android shared libraries copied to android/src/main/jniLibs/"
}

case "${1:-all}" in
    ios)     build_ios ;;
    android) build_android ;;
    all)     build_ios; build_android ;;
    *)
        echo "Usage: $0 {ios|android|all}"
        exit 1
        ;;
esac

echo ""
echo "Build complete."