#!/bin/bash
# Builds the required IDB Frameworks for the AXe project.

set -e
set -o pipefail

# Environment and Configuration
IDB_CHECKOUT_DIR="${IDB_CHECKOUT_DIR:-./idb_checkout}"
IDB_GIT_REF="${IDB_GIT_REF:-76639e4d0e1741adf391cab36f19fbc59378153e}"
BUILD_OUTPUT_DIR="${BUILD_OUTPUT_DIR:-./build_products}"
DERIVED_DATA_PATH="${DERIVED_DATA_PATH:-./build_derived_data}"
BUILD_XCFRAMEWORK_DIR="${BUILD_XCFRAMEWORK_DIR:-${BUILD_OUTPUT_DIR}/XCFrameworks}"
FBSIMCONTROL_PROJECT="${IDB_CHECKOUT_DIR}/FBSimulatorControl.xcodeproj"
TEMP_DIR="${TEMP_DIR:-$(mktemp -d)}"

FRAMEWORK_SDK="macosx"
FRAMEWORK_CONFIGURATION="Release"

# Notarization Configuration
NOTARIZATION_API_KEY_PATH="${NOTARIZATION_API_KEY_PATH:-./keys/AuthKey_8TJYVXVDQ6.p8}"
NOTARIZATION_KEY_ID="${NOTARIZATION_KEY_ID:-8TJYVXVDQ6}"
NOTARIZATION_ISSUER_ID="${NOTARIZATION_ISSUER_ID:-69a6de8e-e388-47e3-e053-5b8c7c11a4d1}"

# --- Helper Functions ---

# Temporarily disable xcpretty to see actual errors in CI
# if hash xcpretty 2>/dev/null; then
#   HAS_XCPRETTY=true
# fi

# Function to print a section header with emoji
function print_section() {
  local emoji="$1"
  local title="$2"
  echo ""
  echo ""
  echo "${emoji} ${title}"
  echo "$(printf '·%.0s' {1..60})"
}

# Function to print a subsection header
function print_subsection() {
  local emoji="$1"
  local title="$2"
  echo ""
  echo "${emoji} ${title}"
}

# Function to print success message
function print_success() {
  local message="$1"
  echo "✅ ${message}"
}

# Function to print info message
function print_info() {
  local message="$1"
  echo "ℹ️  ${message}"
}

# Function to print warning message
function print_warning() {
  local message="$1"
  echo "⚠️  ${message}"
}

# Function to invoke xcodebuild, optionally with xcpretty
function invoke_xcodebuild() {
  local arguments=$@
  print_info "Executing: xcodebuild ${arguments[*]}"

  local exit_code
  if [[ -n $HAS_XCPRETTY ]]; then
    NSUnbufferedIO=YES xcodebuild $arguments | xcpretty -c
    exit_code=${PIPESTATUS[0]}
  else
    xcodebuild $arguments 2>&1
    exit_code=$?
  fi

  return $exit_code
}

function clone_idb_repo() {
  if [ ! -d $IDB_CHECKOUT_DIR ]; then
    print_info "Creating $IDB_DIRECTORY directory and cloning idb repository..."
    git clone https://github.com/facebook/idb.git $IDB_CHECKOUT_DIR
    (cd $IDB_CHECKOUT_DIR && git checkout "$IDB_GIT_REF")
    print_success "idb repository cloned at $IDB_GIT_REF."
  else
    print_info "Updating idb repository to $IDB_GIT_REF..."
    (cd $IDB_CHECKOUT_DIR && git fetch --all --tags --prune && git reset --hard "$IDB_GIT_REF")
    print_success "idb repository updated to $IDB_GIT_REF."
  fi
}

# Function to build a single framework
# $1: Scheme name
# $2: Project file path
# $3: Base output directory (for .framework and .xcframework)
function framework_build() {
  local scheme_name="$1"
  local project_file="$2"
  local output_base_dir="$3"

  print_subsection "🔨" "Building framework: ${scheme_name}"
  print_info "Project: ${project_file}"

  invoke_xcodebuild \
    -project "${project_file}" \
    -scheme "${scheme_name}" \
    -sdk "${FRAMEWORK_SDK}" \
    -destination "generic/platform=macOS" \
    -configuration "${FRAMEWORK_CONFIGURATION}" \
    -derivedDataPath "${DERIVED_DATA_PATH}" \
    build \
    SKIP_INSTALL=NO \
    ONLY_ACTIVE_ARCH=NO \
    BUILD_LIBRARY_FOR_DISTRIBUTION=YES \
    GCC_WARN_ABOUT_MISSING_FIELD_INITIALIZERS=NO \
    CLANG_WARN_DOCUMENTATION_COMMENTS=NO \
    GCC_TREAT_WARNINGS_AS_ERRORS=NO \
    SWIFT_TREAT_WARNINGS_AS_ERRORS=NO
  local build_exit_code=$?

  if [ $build_exit_code -eq 0 ]; then
    print_success "Framework ${scheme_name} built successfully!"
  else
    echo "❌ Error: Framework ${scheme_name} build failed with exit code ${build_exit_code}"
    exit $build_exit_code
  fi
}

# Function to install a single framework to Frameworks/
# $1: Scheme name (used to find the .framework in derived data)
# $2: Base output directory
function install_framework() {
  local scheme_name="$1"
  local output_base_dir="$2"
  local built_framework_path="${DERIVED_DATA_PATH}/Build/Products/${FRAMEWORK_CONFIGURATION}/${scheme_name}.framework"
  local final_framework_install_dir="${output_base_dir}/Frameworks"

  print_info "Installing framework ${scheme_name}.framework to ${final_framework_install_dir}..."
  if [[ ! -d "${built_framework_path}" ]]; then
    echo "❌ Error: Built framework not found at ${built_framework_path} for installation."
    exit 1
  fi

  mkdir -p "${final_framework_install_dir}"
  print_info "Copying ${built_framework_path} to ${final_framework_install_dir}/"
  cp -R "${built_framework_path}" "${final_framework_install_dir}/"
  print_success "Framework ${scheme_name}.framework installed to ${final_framework_install_dir}/"
}

# Function to create a single XCFramework
# $1: Scheme name
# $2: Base output directory (where XCFrameworks/ subdirectory will be created)
function create_xcframework() {
  local scheme_name="$1"
  local output_base_dir="$2"
  local signed_framework_path="${output_base_dir}/Frameworks/${scheme_name}.framework"
  local final_xcframework_output_dir="${output_base_dir}/XCFrameworks"
  local xcframework_path="${final_xcframework_output_dir}/${scheme_name}.xcframework"

  print_subsection "📦" "Creating XCFramework for ${scheme_name}"
  if [[ ! -d "${signed_framework_path}" ]]; then
    echo "❌ Error: Signed framework not found at ${signed_framework_path} for XCFramework creation."
    exit 1
  fi

  mkdir -p "${final_xcframework_output_dir}"
  rm -rf "${xcframework_path}"

  print_info "Packaging ${signed_framework_path} into ${xcframework_path}"
  invoke_xcodebuild \
    -create-xcframework \
    -framework "${signed_framework_path}" \
    -output "${xcframework_path}"
  local xcframework_exit_code=$?

  if [ $xcframework_exit_code -eq 0 ]; then
    print_success "XCFramework ${scheme_name}.xcframework created at ${xcframework_path}"
  else
    echo "❌ Error: XCFramework creation for ${scheme_name} failed with exit code ${xcframework_exit_code}"
    exit $xcframework_exit_code
  fi
}

# Function to strip a framework of nested frameworks
# $1: Base output directory
# $2: Framework path
function strip_framework() {
  local output_base_dir="$1"
  local framework_path="${output_base_dir}/Frameworks/${2}"

  if [ -d "$framework_path" ]; then
    print_info "Stripping Framework $framework_path"
    rm -r "$framework_path"
  fi
}

# Function to resign a framework with Developer ID
# $1: Base output directory
# $2: Framework name (e.g., "FBSimulatorControl.framework")
function resign_framework() {
  local output_base_dir="$1"
  local framework_name="$2"
  local framework_path="${output_base_dir}/Frameworks/${framework_name}"

  if [ -d "$framework_path" ]; then
    print_info "Resigning framework: ${framework_name}"

    # First, sign all dynamic libraries and binaries inside the framework
    print_info "Signing embedded binaries in ${framework_name}..."

    # Find and sign all .dylib files recursively
    find "$framework_path" -name "*.dylib" -type f | while read -r dylib_path; do
      print_info "  Signing dylib: $(basename "$dylib_path")"
      codesign --force \
        --sign "Apple Distribution: Wei Wang (A4YJ9MRZ66)" \
        --options runtime \
        --timestamp \
        --verbose \
        "$dylib_path"

      if [ $? -ne 0 ]; then
        echo "❌ Error: Failed to sign dylib: $dylib_path"
        exit 1
      fi
    done

    # Remove any existing signature from the main framework binary first
    print_info "Removing existing signature from ${framework_name}..."
    codesign --remove-signature "$framework_path" 2>/dev/null || true

    # Sign the main framework bundle with specific notarization-compatible options
    print_info "Signing main framework bundle: ${framework_name}"
    codesign --force \
      --sign "Apple Distribution: Wei Wang (A4YJ9MRZ66)" \
      --options runtime \
      --entitlements entitlements.plist \
      --timestamp \
      --verbose \
      "$framework_path"

    if [ $? -eq 0 ]; then
      print_success "Framework ${framework_name} resigned successfully"

      # Verify the signature with strictest verification
      print_info "Performing strict verification for ${framework_name}..."
      codesign -vvv --strict "$framework_path"

      if [ $? -eq 0 ]; then
        print_success "Signature verification passed for ${framework_name}"

        # Display signature details
        print_info "Signature details for ${framework_name}:"
        codesign -dv "$framework_path" 2>&1 | grep -E "(Identifier|TeamIdentifier|Authority|Timestamp)" || true
      else
        echo "❌ Error: Signature verification failed for ${framework_name}"
        exit 1
      fi
    else
      echo "❌ Error: Failed to resign framework ${framework_name}"
      exit 1
    fi
  else
    print_warning "Framework not found: $framework_path"
  fi
}

# Function to resign an XCFramework with Developer ID
# $1: Base output directory
# $2: XCFramework name (e.g., "FBSimulatorControl.xcframework")
function resign_xcframework() {
  local output_base_dir="$1"
  local xcframework_name="$2"
  local xcframework_path="${output_base_dir}/XCFrameworks/${xcframework_name}"

  if [ -d "$xcframework_path" ]; then
    print_info "Resigning XCFramework: ${xcframework_name}"

    # Sign XCFramework with Developer ID and runtime hardening
    codesign --force \
      --sign "Apple Distribution: Wei Wang (A4YJ9MRZ66)" \
      --options runtime \
      --deep \
      --timestamp \
      "$xcframework_path"

    if [ $? -eq 0 ]; then
      print_success "XCFramework ${xcframework_name} resigned successfully"

      # Verify the signature with strictest verification and deep checking
      print_info "Performing strict verification for XCFramework ${xcframework_name}..."
      codesign -vvv --deep "$xcframework_path"

      if [ $? -eq 0 ]; then
        print_success "XCFramework signature verification passed for ${xcframework_name}"

        # Display signature details
        print_info "XCFramework signature details for ${xcframework_name}:"
        codesign -dv --deep "$xcframework_path" 2>&1 | grep -E "(Identifier|TeamIdentifier|Authority)" || true
      else
        echo "❌ Error: XCFramework signature verification failed for ${xcframework_name}"
        exit 1
      fi
    else
      echo "❌ Error: Failed to resign XCFramework ${xcframework_name}"
      exit 1
    fi
  else
    print_warning "XCFramework not found: $xcframework_path"
  fi
}

# Function to build the AXe executable using Swift Package Manager
# $1: Base output directory
function build_axe_executable() {
  local output_base_dir="$1"
  local build_config="release"
  local executable_source=".build/arm64-apple-macosx/${build_config}/axe"
  local executable_dest="${output_base_dir}/axe"

  print_subsection "⚡" "Building AXe executable"
  print_info "Using Swift Package Manager to build AXe..."

  # Clean any existing build products to ensure fresh build
  print_info "Cleaning previous build products..."
  swift package clean

  # Build using Swift Package Manager (rely on environment variables for cache control)
  swift build --configuration ${build_config}
  local build_exit_code=$?

  if [ $build_exit_code -eq 0 ]; then
    print_success "AXe executable built successfully!"

    # Copy executable to build products directory
    print_info "Installing executable to ${executable_dest}"
    cp "${executable_source}" "${executable_dest}"
    print_success "AXe executable installed to ${executable_dest}"

    # Configure rpath for organized framework loading
    print_info "Configuring executable rpath for organized framework loading..."

    # Remove any existing rpaths first
    install_name_tool -delete_rpath "@executable_path/Frameworks" "${executable_dest}" 2>/dev/null || true
    install_name_tool -delete_rpath "@loader_path/Frameworks" "${executable_dest}" 2>/dev/null || true

    # Add primary rpath: look for frameworks in Frameworks/ subdirectory relative to executable
    install_name_tool -add_rpath "@executable_path/Frameworks" "${executable_dest}"
    print_success "Added rpath: @executable_path/Frameworks"

    # Add fallback rpath: look for frameworks in Frameworks/ relative to current library
    install_name_tool -add_rpath "@loader_path/Frameworks" "${executable_dest}"
    print_success "Added rpath: @loader_path/Frameworks"

    # Verify rpath configuration
    print_info "Verifying rpath configuration..."
    local rpath_output=$(otool -l "${executable_dest}" | grep -A2 LC_RPATH | grep path | awk '{print $2}')
    if [[ -n "$rpath_output" ]]; then
      print_success "Executable rpath configuration verified:"
      echo "$rpath_output" | while read -r path; do
        print_info "  → ${path}"
      done
    else
      print_warning "No rpath entries found in executable"
    fi

    print_success "Executable rpath configured for organized framework deployment"
  else
    echo "❌ Error: AXe executable build failed with exit code ${build_exit_code}"
    exit $build_exit_code
  fi
}

# Function to sign the AXe executable with Developer ID
# $1: Base output directory
function sign_axe_executable() {
  local output_base_dir="$1"
  local executable_path="${output_base_dir}/axe"

  if [ -f "$executable_path" ]; then
    print_info "Signing AXe executable: ${executable_path}"

    # Sign with Developer ID and runtime hardening
    codesign --force \
      --sign "Apple Distribution: Wei Wang (A4YJ9MRZ66)" \
      --options runtime \
      --entitlements entitlements.plist \
      --timestamp \
      "$executable_path"

    if [ $? -eq 0 ]; then
      print_success "AXe executable signed successfully"

      # Verify the signature with strictest verification
      print_info "Performing strict verification for AXe executable..."
      codesign -vvv "$executable_path"

      if [ $? -eq 0 ]; then
        print_success "AXe executable signature verification passed"

        # Display signature details
        print_info "AXe executable signature details:"
        codesign -dv "$executable_path" 2>&1 | grep -E "(Identifier|TeamIdentifier|Authority)" || true
      else
        echo "❌ Error: AXe executable signature verification failed"
        exit 1
      fi
    else
      echo "❌ Error: Failed to sign AXe executable"
      exit 1
    fi
  else
    print_warning "AXe executable not found: $executable_path"
  fi
}

# Function to create a package for notarization
# $1: Base output directory
function package_for_notarization() {
  local output_base_dir="$1"
  local package_name="AXe-$(date +%Y%m%d-%H%M%S)"
  local package_dir="${output_base_dir}/${package_name}"
  local package_zip="${output_base_dir}/${package_name}.zip"

  print_subsection "📦" "Creating notarization package" >&2
  print_info "Package name: ${package_name}" >&2

  # Create temporary package directory
  rm -rf "${package_dir}" "${package_zip}"
  mkdir -p "${package_dir}"

  # Copy executable to package directory
  print_info "Copying executable to package..." >&2
  cp "${output_base_dir}/axe" "${package_dir}/"

  # Create zip package (redirect zip output to stderr)
  print_info "Creating zip package: ${package_zip}" >&2
  (cd "${output_base_dir}" && zip -r "${package_name}.zip" "${package_name}/") >&2

  # Clean up temporary directory
  rm -rf "${package_dir}"

  if [ -f "${package_zip}" ]; then
    print_success "Notarization package created: ${package_zip}" >&2
    # Store the clean absolute path
    local clean_path="$(cd "$(dirname "${package_zip}")" && pwd)/$(basename "${package_zip}")"
    # Only echo the path to stdout for capture
    echo "${clean_path}"
  else
    echo "❌ Error: Failed to create notarization package" >&2
    exit 1
  fi
}

# Function to submit package for notarization
# $1: Package zip path
function notarize_package() {
  local package_zip="$1"

  print_subsection "🍎" "Submitting for Apple notarization"

  # Check if API key exists
  if [ ! -f "${NOTARIZATION_API_KEY_PATH}" ]; then
    echo "❌ Error: Notarization API key not found at ${NOTARIZATION_API_KEY_PATH}"
    print_info "Please ensure the API key file exists or set NOTARIZATION_API_KEY_PATH environment variable"
    exit 1
  fi

  print_info "API Key: ${NOTARIZATION_API_KEY_PATH}"
  print_info "Key ID: ${NOTARIZATION_KEY_ID}"
  print_info "Issuer ID: ${NOTARIZATION_ISSUER_ID}"
  print_info "Package: ${package_zip}"
  print_info "Temporary directory: ${TEMP_DIR}"

  # Submit for notarization
  print_info "Submitting package for notarization..."
  local submit_output=$(xcrun notarytool submit "${package_zip}" \
    --key "${NOTARIZATION_API_KEY_PATH}" \
    --key-id "${NOTARIZATION_KEY_ID}" \
    --issuer "${NOTARIZATION_ISSUER_ID}" \
    --wait 2>&1)
  local submit_exit_code=$?

  echo "${submit_output}"

  if [ $submit_exit_code -eq 0 ] && echo "${submit_output}" | grep -q "status: Accepted"; then
    # Extract submission ID from output
    local submission_id=$(echo "${submit_output}" | grep "id:" | head -1 | awk '{print $2}')
    print_success "Notarization completed successfully!"
    print_info "Submission ID: ${submission_id}"

    # Extract notarized executable from package and replace original
    print_info "Extracting notarized executable to replace original..."
    local temp_extract_dir="${BUILD_OUTPUT_DIR}/temp_notarized"
    rm -rf "${temp_extract_dir}"
    mkdir -p "${temp_extract_dir}"

    # Extract the notarized package
    unzip -q "${package_zip}" -d "${temp_extract_dir}"

    # Find the extracted executable
    local extracted_executable=$(find "${temp_extract_dir}" -name "axe" -type f | head -1)

    if [ -f "${extracted_executable}" ]; then
      # Replace the original executable with the notarized one
      cp "${extracted_executable}" "${BUILD_OUTPUT_DIR}/axe"
      print_success "Original executable replaced with notarized version"

      # Verify notarization status using spctl
      print_info "Verifying notarization with spctl assessment..."
      spctl -a -v "${BUILD_OUTPUT_DIR}/axe" 2>&1 | grep -q "accepted" || {
        print_info "Note: spctl shows 'not an app' for command-line tools - this is expected"
        print_info "Notarized command-line tools are validated differently by macOS"
      }

      # Check if the executable has the notarization signature
      print_info "Checking code signature details..."
      local sig_info=$(codesign -dv "${BUILD_OUTPUT_DIR}/axe" 2>&1)
      if echo "$sig_info" | grep -q "runtime"; then
        print_success "Executable has runtime hardening enabled (required for notarization)"
      else
        print_warning "Runtime hardening not detected in signature"
      fi

      print_success "Notarized executable is ready for distribution"

      # Ensure Frameworks directory exists for final package
      print_info "Preparing Frameworks directory for final package..."
      if [ ! -d "${BUILD_OUTPUT_DIR}/Frameworks" ]; then
        print_info "Frameworks directory not found, recreating from XCFrameworks..."
        mkdir -p "${BUILD_OUTPUT_DIR}/Frameworks"

        # Extract frameworks from XCFrameworks for deployment
        for xcframework in "${BUILD_OUTPUT_DIR}/XCFrameworks"/*.xcframework; do
          if [ -d "$xcframework" ]; then
            local framework_name=$(basename "$xcframework" .xcframework)
            print_info "Extracting ${framework_name}.framework from XCFramework..."

            # Find the macOS framework inside the XCFramework
            local macos_framework=$(find "$xcframework" -name "${framework_name}.framework" -path "*/macos-*" | head -1)
            if [ -d "$macos_framework" ]; then
              cp -R "$macos_framework" "${BUILD_OUTPUT_DIR}/Frameworks/"
              print_info "Copied ${framework_name}.framework to Frameworks directory"
            else
              print_warning "Could not find macOS framework in ${xcframework}"
            fi
          fi
        done
        print_success "Frameworks directory recreated from XCFrameworks"
      else
        print_info "Frameworks directory already exists"
      fi

      # Create final deployment package in temporary directory
      print_info "Creating final deployment package..."
      local final_package_name="AXe-Final-$(date +%Y%m%d-%H%M%S)"
      local final_package_dir="${TEMP_DIR}/${final_package_name}"
      local final_package_zip="${TEMP_DIR}/${final_package_name}.zip"

      # Create final package directory
      mkdir -p "${final_package_dir}"

      # Copy notarized executable and frameworks to final package
      cp "${BUILD_OUTPUT_DIR}/axe" "${final_package_dir}/"
      if [ -d "${BUILD_OUTPUT_DIR}/Frameworks" ]; then
        cp -R "${BUILD_OUTPUT_DIR}/Frameworks" "${final_package_dir}/"
        print_info "Included Frameworks directory in final package"
      else
        print_info "No Frameworks directory found - creating executable-only package"
      fi

      # Create final zip package
      print_info "Creating final package: ${final_package_zip}"
      (cd "${TEMP_DIR}" && zip -r "${final_package_name}.zip" "${final_package_name}/")

      # Clean up temporary package directory
      rm -rf "${final_package_dir}"

      if [ -f "${final_package_zip}" ]; then
        print_success "Final deployment package created: ${final_package_zip}"

        # Clean up build artifacts (axe executable and Frameworks, keep XCFrameworks)
        print_info "Cleaning up build artifacts..."
        rm -f "${BUILD_OUTPUT_DIR}/axe"
        rm -rf "${BUILD_OUTPUT_DIR}/Frameworks"
        print_success "Cleaned up axe executable and Frameworks directory"
        print_info "Preserved XCFrameworks directory for Swift package builds"

        # Output the final package path
        echo ""
        echo "📦 Final Package Location:"
        echo "${final_package_zip}"
        echo ""

        # Update the global PACKAGE_ZIP variable
        PACKAGE_ZIP="${final_package_zip}"
      else
        echo "❌ Error: Failed to create final deployment package"
        exit 1
      fi

      # Clean up temporary extraction directory and original notarization package
      rm -rf "${temp_extract_dir}"
      rm -f "${package_zip}"
      print_info "Cleaned up temporary notarization files"
    else
      echo "❌ Error: Could not find notarized executable in package"
      exit 1
    fi
  else
    echo "❌ Error: Notarization failed"

    # Extract submission ID for log fetching
    local submission_id=$(echo "${submit_output}" | grep "id:" | head -1 | awk '{print $2}')

    if [ -n "${submission_id}" ]; then
      print_info "Submission ID: ${submission_id}"
      print_info "Fetching notarization log for detailed error information..."

      # Fetch the notary log using notarytool
      echo ""
      echo "📋 Notarization Log:"
      echo "$(printf '·%.0s' {1..60})"
      xcrun notarytool log \
        --key "${NOTARIZATION_API_KEY_PATH}" \
        --key-id "${NOTARIZATION_KEY_ID}" \
        --issuer "${NOTARIZATION_ISSUER_ID}" \
        "${submission_id}"
      echo "$(printf '·%.0s' {1..60})"
    else
      print_info "Could not extract submission ID from notarization output"
    fi

    exit 1
  fi
}

# Function to print usage information
function print_usage() {
cat <<EOF
./build.sh usage:
  ./build.sh [<command>] [<options>]*

Commands:
  help
    Print this usage information.

  setup
    Clone the IDB repository and set up directories.

  clean
    Clean previous build products and derived data.

  frameworks
    Build all IDB frameworks (FBControlCore, XCTestBootstrap, FBSimulatorControl, FBDeviceControl).

  install
    Install built frameworks to the Frameworks directory.

  strip
    Strip nested frameworks from the built frameworks.

  sign-frameworks
    Code sign all frameworks with Developer ID.

  xcframeworks
    Create XCFrameworks from the built frameworks.

  sign-xcframeworks
    Code sign all XCFrameworks with Developer ID.

  executable
    Build the AXe executable using Swift Package Manager.

  sign-executable
    Code sign the AXe executable with Developer ID.

  package
    Create a notarization package (zip file).

  notarize
    Submit package for Apple notarization and replace original executable.

  build (default)
    Run all steps from setup through notarization.

Environment Variables:
  IDB_CHECKOUT_DIR       Directory for IDB repository (default: ./idb_checkout)
  BUILD_OUTPUT_DIR       Directory for build outputs (default: ./build_products)
  DERIVED_DATA_PATH      Directory for derived data (default: ./build_derived_data)
  TEMP_DIR               Temporary directory for final packages (default: system temp)
  NOTARIZATION_API_KEY_PATH  Path to notarization API key (default: ./keys/AuthKey_8TJYVXVDQ6.p8)
  NOTARIZATION_KEY_ID    Notarization key ID (default: 8TJYVXVDQ6)
  NOTARIZATION_ISSUER_ID Notarization issuer ID (default: 69a6de8e-e388-47e3-e053-5b8c7c11a4d1)

Examples:
  ./build.sh                    # Build everything (default)
  ./build.sh help               # Show this help
  ./build.sh frameworks         # Only build frameworks
  ./build.sh sign-frameworks    # Only sign frameworks
  ./build.sh notarize           # Only run notarization step
EOF
}

# Individual command functions
function cmd_setup() {
  print_section "📥" "Repository Setup"
  clone_idb_repo
}

function cmd_clean() {
  print_section "🧹" "Cleaning Previous Build Products"
  print_info "Cleaning previous build products and derived data..."
  rm -rf "${BUILD_OUTPUT_DIR}"
  rm -rf "${DERIVED_DATA_PATH}"
  mkdir -p "${BUILD_OUTPUT_DIR}"
  mkdir -p "${BUILD_XCFRAMEWORK_DIR}"
  mkdir -p "${DERIVED_DATA_PATH}"
  print_success "Build directories cleaned and recreated"
}

function cmd_frameworks() {
  print_section "🔧" "Building Frameworks"
  framework_build "FBControlCore" "${FBSIMCONTROL_PROJECT}" "${BUILD_OUTPUT_DIR}"
  framework_build "XCTestBootstrap" "${FBSIMCONTROL_PROJECT}" "${BUILD_OUTPUT_DIR}"
  framework_build "FBSimulatorControl" "${FBSIMCONTROL_PROJECT}" "${BUILD_OUTPUT_DIR}"
  framework_build "FBDeviceControl" "${FBSIMCONTROL_PROJECT}" "${BUILD_OUTPUT_DIR}"
}

function cmd_install() {
  print_section "📦" "Installing Frameworks"
  install_framework "FBControlCore" "${BUILD_OUTPUT_DIR}"
  install_framework "XCTestBootstrap" "${BUILD_OUTPUT_DIR}"
  install_framework "FBSimulatorControl" "${BUILD_OUTPUT_DIR}"
  install_framework "FBDeviceControl" "${BUILD_OUTPUT_DIR}"
}

function cmd_strip() {
  print_section "✂️" "Stripping Nested Frameworks"
  strip_framework "${BUILD_OUTPUT_DIR}" "FBSimulatorControl.framework/Versions/Current/Frameworks/XCTestBootstrap.framework"
  strip_framework "${BUILD_OUTPUT_DIR}" "FBSimulatorControl.framework/Versions/Current/Frameworks/FBControlCore.framework"
  strip_framework "${BUILD_OUTPUT_DIR}" "FBDeviceControl.framework/Versions/Current/Frameworks/XCTestBootstrap.framework"
  strip_framework "${BUILD_OUTPUT_DIR}" "FBDeviceControl.framework/Versions/Current/Frameworks/FBControlCore.framework"
  strip_framework "${BUILD_OUTPUT_DIR}" "XCTestBootstrap.framework/Versions/Current/Frameworks/FBControlCore.framework"
}

function cmd_sign_frameworks() {
  print_section "🔒" "Resigning Frameworks"
  print_info "Resigning frameworks..."
  resign_framework "${BUILD_OUTPUT_DIR}" "FBSimulatorControl.framework"
  resign_framework "${BUILD_OUTPUT_DIR}" "FBDeviceControl.framework"
  resign_framework "${BUILD_OUTPUT_DIR}" "XCTestBootstrap.framework"
  resign_framework "${BUILD_OUTPUT_DIR}" "FBControlCore.framework"
  print_success "Frameworks resigned successfully"
}

function cmd_xcframeworks() {
  print_section "📦" "Creating XCFrameworks"
  create_xcframework "FBControlCore" "${BUILD_OUTPUT_DIR}"
  create_xcframework "XCTestBootstrap" "${BUILD_OUTPUT_DIR}"
  create_xcframework "FBSimulatorControl" "${BUILD_OUTPUT_DIR}"
  create_xcframework "FBDeviceControl" "${BUILD_OUTPUT_DIR}"
}

function cmd_sign_xcframeworks() {
  print_section "🔒" "Resigning XCFrameworks"
  print_info "Resigning XCFrameworks with Developer ID..."
  resign_xcframework "${BUILD_OUTPUT_DIR}" "FBControlCore.xcframework"
  resign_xcframework "${BUILD_OUTPUT_DIR}" "XCTestBootstrap.xcframework"
  resign_xcframework "${BUILD_OUTPUT_DIR}" "FBSimulatorControl.xcframework"
  resign_xcframework "${BUILD_OUTPUT_DIR}" "FBDeviceControl.xcframework"
  print_success "XCFrameworks resigned successfully"
}

function cmd_executable() {
  print_section "⚡" "Building AXe Executable"
  build_axe_executable "${BUILD_OUTPUT_DIR}"
}

function cmd_sign_executable() {
  print_section "🔒" "Signing AXe Executable"
  sign_axe_executable "${BUILD_OUTPUT_DIR}"
}

function cmd_package() {
  print_section "📦" "Packaging for Notarization"
  PACKAGE_ZIP=$(package_for_notarization "${BUILD_OUTPUT_DIR}")
  print_info "Package created: ${PACKAGE_ZIP}"
}

function cmd_notarize() {
  print_section "🍎" "Apple Notarization"
  if [ -z "${PACKAGE_ZIP}" ]; then
    # Find the most recent package if PACKAGE_ZIP isn't set
    PACKAGE_ZIP=$(ls -t "${BUILD_OUTPUT_DIR}"/AXe-*.zip 2>/dev/null | head -1)
    if [ -z "${PACKAGE_ZIP}" ]; then
      echo "❌ Error: No package found. Run 'package' command first."
      exit 1
    fi
    print_info "Using package: ${PACKAGE_ZIP}"
  fi
  notarize_package "${PACKAGE_ZIP}"
}

function cmd_build() {
  print_section "🚀" "IDB Framework Builder for AXe Project"

  print_info "IDB Checkout Directory: ${IDB_CHECKOUT_DIR}"
  print_info "Build Output Directory: ${BUILD_OUTPUT_DIR}"
  print_info "Derived Data Path: ${DERIVED_DATA_PATH}"
  print_info "XCFramework Output Directory: ${BUILD_XCFRAMEWORK_DIR}"
  print_info "Temporary Directory: ${TEMP_DIR}"
  print_info "IDB Project: ${FBSIMCONTROL_PROJECT}"
  print_info "Notarization API Key: ${NOTARIZATION_API_KEY_PATH}"
  print_info "Notarization Key ID: ${NOTARIZATION_KEY_ID}"

  # Run all steps
  cmd_setup
  cmd_clean
  cmd_frameworks
  cmd_install
  cmd_strip
  cmd_sign_frameworks
  cmd_xcframeworks
  cmd_sign_xcframeworks
  cmd_executable
  cmd_sign_executable
  cmd_package
  cmd_notarize

  print_section "🎉" "Build Complete!"
  print_success "All framework builds, XCFramework creation, AXe executable, and notarization completed."
  print_info "📦 XCFrameworks are located in ${BUILD_XCFRAMEWORK_DIR}"
  print_info "📁 Final deployment package is located at ${PACKAGE_ZIP}"
  print_info "🧹 Build artifacts (axe executable and Frameworks) have been cleaned up"
  echo ""
  echo "🏁 Build process finished successfully!"
  echo ""
}

# Parse command line arguments
COMMAND="${1:-build}"

case $COMMAND in
  help)
    print_usage
    exit 0;;
  setup)
    cmd_setup;;
  clean)
    cmd_clean;;
  frameworks)
    cmd_frameworks;;
  install)
    cmd_install;;
  strip)
    cmd_strip;;
  sign-frameworks)
    cmd_sign_frameworks;;
  xcframeworks)
    cmd_xcframeworks;;
  sign-xcframeworks)
    cmd_sign_xcframeworks;;
  executable)
    cmd_executable;;
  sign-executable)
    cmd_sign_executable;;
  package)
    cmd_package;;
  notarize)
    cmd_notarize;;
  build)
    cmd_build;;
  *)
    echo "Unknown command: $COMMAND"
    echo ""
    print_usage
    exit 1;;
esac

exit 0