#!/bin/bash
#
# Build and copy Bun executable and native modules to the app bundle
# ARM64 only - VibeTunnel requires Apple Silicon
#

set -euo pipefail

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Script directory
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
PROJECT_ROOT="$SCRIPT_DIR/../.."
WEB_DIR="$PROJECT_ROOT/web"
NATIVE_DIR="$WEB_DIR/native"

# Destination from Xcode (passed as argument or use BUILT_PRODUCTS_DIR)
if [ $# -eq 0 ]; then
    if [ -z "${BUILT_PRODUCTS_DIR:-}" ]; then
        echo -e "${RED}Error: No destination path provided and BUILT_PRODUCTS_DIR not set${NC}"
        exit 1
    fi
    DEST_RESOURCES="${BUILT_PRODUCTS_DIR}/${UNLOCALIZED_RESOURCES_FOLDER_PATH}"
else
    DEST_RESOURCES="$1"
fi

echo -e "${GREEN}Building and copying Bun executable (ARM64 only)...${NC}"

# Change to web directory
cd "$WEB_DIR"

# Check if native directory exists, if not use prebuilts
if [ ! -d "$NATIVE_DIR" ] || [ ! -f "$NATIVE_DIR/vibetunnel" ]; then
    echo -e "${YELLOW}Native directory not found. Checking for prebuilt binaries...${NC}"
    
    # Use prebuilt binaries if available
    PREBUILTS_DIR="$PROJECT_ROOT/mac/Resources/BunPrebuilts"
    ARCH=$(uname -m)
    
    if [ "$ARCH" != "arm64" ]; then
        echo -e "${RED}Error: VibeTunnel requires Apple Silicon (ARM64)${NC}"
        exit 1
    fi
    
    if [ -d "$PREBUILTS_DIR/arm64" ] && [ -f "$PREBUILTS_DIR/arm64/vibetunnel" ]; then
        echo -e "${GREEN}Using prebuilt binaries for ARM64${NC}"
        mkdir -p "$NATIVE_DIR"
        cp "$PREBUILTS_DIR/arm64/vibetunnel" "$NATIVE_DIR/"
        cp "$PREBUILTS_DIR/arm64/pty.node" "$NATIVE_DIR/"
        cp "$PREBUILTS_DIR/arm64/spawn-helper" "$NATIVE_DIR/"
        chmod +x "$NATIVE_DIR/vibetunnel"
        chmod +x "$NATIVE_DIR/spawn-helper"
    else
        # Try to build if Bun is available
        echo -e "${YELLOW}Prebuilt binaries not found. Attempting to build...${NC}"
        
        # Check if build-native.js exists
        if [ -f "build-native.js" ]; then
            # Try different ways to run bun
            if command -v bun &> /dev/null; then
                echo "Using bun to build..."
                bun build-native.js
            elif command -v npx &> /dev/null; then
                echo "Bun not found, using npx bun to build..."
                npx -y bun build-native.js
            elif command -v node &> /dev/null; then
                echo "Using node to build (fallback)..."
                node build-native.js
            else
                echo -e "${RED}Error: No JavaScript runtime found (bun, npx, or node).${NC}"
                echo -e "${RED}Please ensure prebuilt binaries are available in:${NC}"
                echo -e "${RED}  $PREBUILTS_DIR/arm64/${NC}"
                exit 1
            fi
        else
            echo -e "${RED}Error: build-native.js not found and no prebuilt binaries available${NC}"
            exit 1
        fi
    fi
fi

# Verify native files exist
if [ ! -f "$NATIVE_DIR/vibetunnel" ]; then
    echo -e "${RED}Error: Bun executable not found at $NATIVE_DIR/vibetunnel${NC}"
    exit 1
fi

# Copy Bun executable
echo "Copying Bun executable to app bundle..."
cp "$NATIVE_DIR/vibetunnel" "$DEST_RESOURCES/"
chmod +x "$DEST_RESOURCES/vibetunnel"

# Copy native modules
if [ -f "$NATIVE_DIR/pty.node" ]; then
    echo "Copying pty.node..."
    cp "$NATIVE_DIR/pty.node" "$DEST_RESOURCES/"
else
    echo -e "${RED}Error: pty.node not found${NC}"
    exit 1
fi

if [ -f "$NATIVE_DIR/spawn-helper" ]; then
    echo "Copying spawn-helper..."
    cp "$NATIVE_DIR/spawn-helper" "$DEST_RESOURCES/"
    chmod +x "$DEST_RESOURCES/spawn-helper"
else
    echo -e "${RED}Error: spawn-helper not found${NC}"
    exit 1
fi

echo -e "${GREEN}✓ Bun executable and native modules copied successfully${NC}"

# Verify the files
echo "Verifying copied files:"
ls -la "$DEST_RESOURCES/vibetunnel" || echo "vibetunnel not found!"
ls -la "$DEST_RESOURCES/pty.node" || echo "pty.node not found!"
ls -la "$DEST_RESOURCES/spawn-helper" || echo "spawn-helper not found!"

echo ""
echo -e "${GREEN}Note: VibeTunnel requires Apple Silicon (M1/M2/M3) Macs.${NC}"