#!/bin/sh
set -eu

SCRIPT_DIR=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
ROOT_DIR=$(CDPATH= cd -- "$SCRIPT_DIR/../.." && pwd)
SOURCE_DIR=${1:-"$ROOT_DIR/audio_prompts/customer_fs"}
OUTPUT_IMAGE=${2:-"$ROOT_DIR/output/customer_fs/customer_fs.bin"}
VENV_DIR=${CUSTOMER_FS_VENV:-"$ROOT_DIR/.venv-customer-fs"}
PYTHON_BOOTSTRAP=${PYTHON3:-python3}
TEMP_IMAGE="${OUTPUT_IMAGE}.tmp"

cleanup() {
    rm -f -- "$TEMP_IMAGE"
}
trap cleanup EXIT INT TERM

if [ ! -x "$VENV_DIR/bin/python" ]; then
    "$PYTHON_BOOTSTRAP" -m venv "$VENV_DIR"
fi

if ! "$VENV_DIR/bin/python" -c 'import littlefs; raise SystemExit(0 if littlefs.__version__ == "0.19.0" else 1)' 2>/dev/null; then
    "$VENV_DIR/bin/python" -m pip install --requirement "$SCRIPT_DIR/requirements.txt"
fi

mkdir -p -- "$(dirname -- "$OUTPUT_IMAGE")"
rm -f -- "$TEMP_IMAGE"

"$VENV_DIR/bin/python" "$SCRIPT_DIR/generate_macos.py" \
    --source "$SOURCE_DIR" \
    --output "$TEMP_IMAGE"

"$VENV_DIR/bin/python" "$SCRIPT_DIR/verify_customer_fs.py" \
    --image "$TEMP_IMAGE" \
    --source "$SOURCE_DIR"

mv -f -- "$TEMP_IMAGE" "$OUTPUT_IMAGE"
trap - EXIT INT TERM
printf 'customer_fs ready: %s\n' "$OUTPUT_IMAGE"