#!/bin/bash

set -e  # Exit immediately if a command exits with a non-zero status
set -u  # Treat unset variables as an error
set -o pipefail  # Catch errors in pipelines

# Ensure the script is running in the correct directory
NAME="Notedeck"
REQUIRED_DIR="notedeck"
ARCH=${ARCH:-"aarch64"}
TARGET=${TARGET:-${ARCH}-apple-darwin}
CURRENT_DIR=$(basename "$PWD")

if [ "$CURRENT_DIR" != "$REQUIRED_DIR" ]; then
    echo "Error: This script must be run from the '$REQUIRED_DIR' directory."
    exit 1
fi

# Ensure all required variables are set
REQUIRED_VARS=(NOTEDECK_APPLE_RELEASE_CERT_ID NOTEDECK_RELEASE_APPLE_ID NOTEDECK_APPLE_APP_SPECIFIC_PW NOTEDECK_APPLE_TEAM_ID)
for VAR in "${REQUIRED_VARS[@]}"; do
    if [ -z "${!VAR:-}" ]; then
        echo "Error: Required variable '$VAR' is not set." >&2
        exit 1
    fi
done

# Ensure required tools are installed
if ! command -v cargo &> /dev/null; then
    echo "Error: cargo is not installed." >&2
    exit 1
fi

if ! command -v xcrun &> /dev/null; then
    echo "Error: xcrun is not installed." >&2
    exit 1
fi

if ! command -v create-dmg &> /dev/null; then
    echo "Error: create-dmg is not installed." >&2
    exit 1
fi

# Build the .app bundle
echo "Building .app bundle..."
cargo bundle --release --target $TARGET

# Sign the app
echo "Codesigning the app..."
codesign \
  --deep \
  --force \
  --verify \
  --options runtime \
  --entitlements entitlements.plist \
  --sign "$NOTEDECK_APPLE_RELEASE_CERT_ID" \
  target/${TARGET}/release/bundle/osx/$NAME.app

# Create a zip for notarization
echo "Creating zip for notarization..."
zip -r notedeck.zip target/${TARGET}/release/bundle/osx/$NAME.app

# Submit for notarization
echo "Submitting for notarization..."
xcrun notarytool submit \
  --apple-id "$NOTEDECK_RELEASE_APPLE_ID" \
  --password "$NOTEDECK_APPLE_APP_SPECIFIC_PW" \
  --team-id "$NOTEDECK_APPLE_TEAM_ID" \
  --wait \
  notedeck.zip

# Staple the notarization
echo "Stapling notarization to the app..."
xcrun stapler staple target/${TARGET}/release/bundle/osx/$NAME.app

echo "Removing notedeck.zip"
rm notedeck.zip

# Create the .dmg package
echo "Creating .dmg package..."
mkdir -p packages
create-dmg \
  --window-size 600 400 \
  --app-drop-link 400 100 \
  packages/$NAME-${ARCH}.dmg \
  target/${TARGET}/release/bundle/osx/$NAME.app

echo "Build and signing process completed successfully."
