#!/bin/bash
set -e

# Get the target triple from environment or use default
TARGET="${TARGET:-$(rustc -vV 2>/dev/null | sed -n 's|host: ||p' || echo "unknown")}"
MANIFEST_DIR="$(dirname "$0")"

# Determine source path
HOST_TARGET=$(rustc -vV 2>/dev/null | sed -n 's|host: ||p' || echo "$TARGET")
if [[ "$TARGET" == "$HOST_TARGET" ]] || [[ "$TARGET" == "unknown" ]]; then
  # Native target - use debug or release based on profile
  if [[ "${PROFILE:-debug}" == "release" ]]; then
    SRC_DIR="$MANIFEST_DIR/target/release"
  else
    SRC_DIR="$MANIFEST_DIR/target/debug"
  fi
else
  # Cross-compilation target
  if [[ "${PROFILE:-debug}" == "release" ]]; then
    SRC_DIR="$MANIFEST_DIR/target/$TARGET/release"
  else
    SRC_DIR="$MANIFEST_DIR/target/$TARGET/debug"
  fi
fi

DEST_DIR="$MANIFEST_DIR/binaries"
# Create binaries directory if it doesn't exist
mkdir -p "$DEST_DIR"

# Function to copy a binary
copy_binary() {
  local BIN_BASE_NAME="$1"

  # Determine binary name based on target
  if [[ "$TARGET" == *"windows"* ]]; then
    BIN_NAME="${BIN_BASE_NAME}.exe"
  else
    BIN_NAME="$BIN_BASE_NAME"
  fi

  SOURCE="$SRC_DIR/$BIN_NAME"

  # Tauri expects the format: binary-{target} with hyphens
  DEST_NAME="${BIN_BASE_NAME}-$TARGET"
  if [[ "$TARGET" == *"windows"* ]]; then
    DEST_NAME="$DEST_NAME.exe"
  fi
  DEST="$DEST_DIR/$DEST_NAME"

  # Copy the binary if it exists
  if [[ -f "$SOURCE" ]]; then
    cp "$SOURCE" "$DEST"
    echo "Copied $BIN_NAME to $DEST"
  else
    echo "Warning: Binary not found at $SOURCE"
    echo "Building $BIN_BASE_NAME binary..."
    cd "$MANIFEST_DIR"
    BUILD_ARGS=("build" "--bin" "$BIN_BASE_NAME")
    if [[ -n "$PROFILE" ]] && [[ "$PROFILE" == "release" ]]; then
      BUILD_ARGS+=("--release")
    fi
    if [[ -n "$TARGET" ]] && [[ "$TARGET" != "unknown" ]] && [[ "$TARGET" != "$HOST_TARGET" ]]; then
      BUILD_ARGS+=("--target" "$TARGET")
    fi
    cargo "${BUILD_ARGS[@]}"
    if [[ -f "$SOURCE" ]]; then
      cp "$SOURCE" "$DEST"
      echo "Built and copied $BIN_NAME to $DEST"
    else
      echo "Error: Failed to build $BIN_BASE_NAME binary"
      exit 1
    fi
  fi
}

# Copy donut-proxy binary
copy_binary "donut-proxy"

# Copy donut-daemon binary
copy_binary "donut-daemon"