start.sh

#!/usr/bin/env bash
set -euo pipefail

# Run openWakeWord (Linux/MacOS/Windows/Android Termux).
# - Resolves model path
# - Starts microphone capture and pipes PCM16 16kHz mono to stdin
#
# Usage:
#   ./start.sh [--model <name|path>] [--threshold <0..1>] [--trigger-level <int>] [--optimizeUnstableSystems] [--] [extra args...]

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

MODEL_NAME="alexa_v0.1"
THRESHOLD=""
TRIGGER_LEVEL=""
OPTIMIZE_UNSTABLE_SYSTEMS=0

is_termux() {
    if [[ -n "${TERMUX_VERSION:-}" ]]; then
        return 0
    fi

    if [[ "${PREFIX:-}" == "/data/data/com.termux/files/usr" ]]; then
        return 0
    fi

    return 1
}

usage() {
    cat <<EOF
Usage: $(basename "$0") [--model <name|path>] [--threshold <0..1>] [--trigger-level <int>] [--optimizeUnstableSystems] [--] [extra args...]

Examples:
  $(basename "$0") --model alexa_v0.1
  $(basename "$0") --model alexa_v0.1 --threshold 0.5 --trigger-level 4
  $(basename "$0") --model /absolute/path/to/model.onnx
  $(basename "$0") --model alexa_v0.1 --optimizeUnstableSystems

Defaults:
  --trigger-level 4
  --threshold 0.5
  --model alexa_v0.1

Model lookup:
  - $SCRIPT_DIR/models
  - $SCRIPT_DIR/../models (if exists)

Environment overrides:
  OPENWAKEWORD_BIN=<path-to-openwakeword-or-openwakeword.exe>
  OPENWAKEWORD_ORT_LIB_DIR=<path-to-onnxruntime-pc/<version>/<os-arch>/lib>
  OPENWAKEWORD_WIN_MIC=<Windows mic device for ffmpeg dshow, e.g. Microphone (USB Audio Device)>
  ORT_LIB_DIR=<path-to-onnxruntime-android/<version>/jni/<abi>> (Termux)

Termux:
  - Uses PulseAudio module-sles-source + SoX rec.
  - Installs missing dependencies (pulseaudio, sox) via pkg.
EOF
}

install_if_missing() {
    local cmd="$1"
    local pkg_name="$2"

    if ! command -v "$cmd" >/dev/null 2>&1; then
        echo "[start] (setup) installing package: $pkg_name (missing $cmd)"
        pkg install -y "$pkg_name"
    fi
}

install_pc_tools_if_missing() {
    local os_raw="$1"

    case "$os_raw" in
        Linux*)
            if command -v arecord >/dev/null 2>&1 || command -v rec >/dev/null 2>&1; then
                return 0
            fi

            echo "[start] (setup) no Linux mic tool found, trying to install arecord/rec"
            if command -v apt-get >/dev/null 2>&1; then
                if command -v sudo >/dev/null 2>&1; then
                    sudo apt-get update || true
                    sudo apt-get install -y alsa-utils sox || true
                else
                    apt-get update || true
                    apt-get install -y alsa-utils sox || true
                fi
            elif command -v dnf >/dev/null 2>&1; then
                if command -v sudo >/dev/null 2>&1; then
                    sudo dnf install -y alsa-utils sox || true
                else
                    dnf install -y alsa-utils sox || true
                fi
            elif command -v yum >/dev/null 2>&1; then
                if command -v sudo >/dev/null 2>&1; then
                    sudo yum install -y alsa-utils sox || true
                else
                    yum install -y alsa-utils sox || true
                fi
            elif command -v pacman >/dev/null 2>&1; then
                if command -v sudo >/dev/null 2>&1; then
                    sudo pacman -Sy --noconfirm alsa-utils sox || true
                else
                    pacman -Sy --noconfirm alsa-utils sox || true
                fi
            fi
            ;;
        Darwin*)
            if command -v rec >/dev/null 2>&1; then
                return 0
            fi

            echo "[start] (setup) SoX not found on macOS, trying brew install sox"
            if command -v brew >/dev/null 2>&1; then
                brew install sox || true
            fi
            ;;
        MINGW*|MSYS*|CYGWIN*)
            if command -v ffmpeg >/dev/null 2>&1 || command -v rec >/dev/null 2>&1; then
                return 0
            fi

            echo "[start] (setup) no Windows mic tool found, trying winget/choco"
            if command -v winget >/dev/null 2>&1; then
                winget install -e --id Gyan.FFmpeg --accept-package-agreements --accept-source-agreements || true
                winget install -e --id SoX.SoX --accept-package-agreements --accept-source-agreements || true
            elif command -v choco >/dev/null 2>&1; then
                choco install -y ffmpeg sox.portable || true
            fi
            ;;
    esac
}

run_termux_pulseaudio() {
    if ! pulseaudio --check >/dev/null 2>&1; then
        echo "[start] (setup) starting pulseaudio"
        pulseaudio -D --exit-idle-time=-1
        sleep 0.5
    fi
}

load_termux_pulseaudio_module() {
    local module_name="$1"

    if pactl list short modules 2>/dev/null | awk '{print $2}' | grep -qx "$module_name"; then
        return 0
    fi

    echo "[start] (setup) loading pulseaudio module: $module_name"
    pactl load-module "$module_name" >/dev/null
}

pick_termux_mic_source() {
    local source_name

    source_name="$(pactl list short sources 2>/dev/null | awk '{print $2}' | grep -E '^OpenSL_ES_source$' || true)"
    if [[ -n "$source_name" ]]; then
        printf '%s' "$source_name"
        return 0
    fi

    source_name="$(pactl list short sources 2>/dev/null | awk '{print $2}' | grep -v '\.monitor$' | head -n 1 || true)"
    if [[ -n "$source_name" ]]; then
        printf '%s' "$source_name"
        return 0
    fi

    return 1
}

REMAINING_ARGS=()
while [[ $# -gt 0 ]]; do
    case "$1" in
        -h|--help)
            usage
            exit 0
            ;;
        --model)
            shift
            [[ $# -gt 0 ]] || { echo "[start] Missing value for --model" >&2; exit 2; }
            MODEL_NAME="$1"
            shift
            ;;
        --threshold)
            shift
            [[ $# -gt 0 ]] || { echo "[start] Missing value for --threshold" >&2; exit 2; }
            THRESHOLD="$1"
            shift
            ;;
        --trigger-level)
            shift
            [[ $# -gt 0 ]] || { echo "[start] Missing value for --trigger-level" >&2; exit 2; }
            TRIGGER_LEVEL="$1"
            shift
            ;;
        --optimizeUnstableSystems)
            OPTIMIZE_UNSTABLE_SYSTEMS=1
            shift
            ;;
        --)
            shift
            REMAINING_ARGS+=("$@")
            break
            ;;
        *)
            REMAINING_ARGS+=("$1")
            shift
            ;;
    esac
done

if [[ -n "$THRESHOLD" ]]; then
    if ! [[ "$THRESHOLD" =~ ^([0-9]+([.][0-9]+)?|[.][0-9]+)$ ]]; then
        echo "[start] [ERROR] Invalid --threshold value: '$THRESHOLD'" >&2
        echo "    Expected decimal number in range <0..1>, e.g. 0.5" >&2
        exit 2
    fi

    if ! awk -v threshold="$THRESHOLD" 'BEGIN { exit !(threshold >= 0 && threshold <= 1) }'; then
        echo "[start] [ERROR] --threshold out of range: '$THRESHOLD'" >&2
        echo "    Allowed range is <0..1>." >&2
        exit 2
    fi
fi

OPTIMIZE_UNSTABLE_SYSTEMS_ARGS=()
if [[ "$OPTIMIZE_UNSTABLE_SYSTEMS" -eq 1 ]]; then
    OPTIMIZE_UNSTABLE_SYSTEMS_ARGS=("--optimizeUnstableSystems")
fi

TRIGGER_LEVEL_ARGS=()
if [[ -n "$TRIGGER_LEVEL" ]]; then
    TRIGGER_LEVEL_ARGS=("--trigger-level" "$TRIGGER_LEVEL")
fi

THRESHOLD_ARGS=()
if [[ -n "$THRESHOLD" ]]; then
    THRESHOLD_ARGS=("--threshold" "$THRESHOLD")
fi

MODEL_DIR_LOCAL="$SCRIPT_DIR/models"
MODEL_DIR_SHARED="$SCRIPT_DIR/../models"

resolve_model_path() {
    local model="$1"

    if [[ -f "$model" ]]; then
        printf '%s' "$model"
        return 0
    fi

    local candidate="$model"
    if [[ "$candidate" != *.onnx ]]; then
        candidate="${candidate}.onnx"
    fi

    if [[ -f "$MODEL_DIR_LOCAL/$candidate" ]]; then
        printf '%s' "$MODEL_DIR_LOCAL/$candidate"
        return 0
    fi

    if [[ -d "$MODEL_DIR_SHARED" && -f "$MODEL_DIR_SHARED/$candidate" ]]; then
        printf '%s' "$MODEL_DIR_SHARED/$candidate"
        return 0
    fi

    return 1
}

model_path=""
if ! model_path="$(resolve_model_path "$MODEL_NAME")"; then
    echo "[start] [ERROR] Model not found: $MODEL_NAME" >&2
    echo "[start] Looked in:" >&2
    echo "  - $MODEL_DIR_LOCAL" >&2
    if [[ -d "$MODEL_DIR_SHARED" ]]; then
        echo "  - $MODEL_DIR_SHARED" >&2
    fi
    exit 2
fi

if is_termux; then
    arch_raw="$(uname -m)"
    abi=""
    case "$arch_raw" in
        aarch64) abi="arm64-v8a" ;;
        armv7l|armv8l|armv7*) abi="armeabi-v7a" ;;
        x86_64) abi="x86_64" ;;
        i686|x86) abi="x86" ;;
        *)
            echo "[start] [ERROR] Unsupported architecture in Termux: $arch_raw" >&2
            exit 2
            ;;
    esac

    ROOT_PARENT="$SCRIPT_DIR/.."
    BUILD_ABI_DIR="$ROOT_PARENT/build-android/$abi"
    MARKER_FILE="$BUILD_ABI_DIR/onnxruntime-version"

    ort_version_default="1.23.2"
    ort_version_from_marker=""
    if [[ -f "$MARKER_FILE" ]]; then
        ort_version_from_marker="$(tr -d '\r\n' < "$MARKER_FILE")"
    fi

    ort_version_selected=""
    if [[ -n "$ort_version_from_marker" ]]; then
        ort_version_selected="$ort_version_from_marker"
    elif [[ -d "$ROOT_PARENT/onnxruntime-android" ]]; then
        latest_ort_version="$(ls -1 "$ROOT_PARENT/onnxruntime-android" 2>/dev/null | sort -V | tail -n 1 || true)"
        if [[ -n "$latest_ort_version" ]]; then
            ort_version_selected="$latest_ort_version"
        fi
    fi

    if [[ -z "$ort_version_selected" ]]; then
        ort_version_selected="$ort_version_default"
    fi

    OPENWAKEWORD_BIN="${OPENWAKEWORD_BIN:-$BUILD_ABI_DIR/openwakeword}"
    ORT_LIB_DIR="${ORT_LIB_DIR:-$ROOT_PARENT/onnxruntime-android/$ort_version_selected/jni/$abi}"

    echo "[start] runtime=termux"
    echo "[start] arch=$arch_raw abi=$abi"
    echo "[start] model=$model_path"
    if [[ -n "$THRESHOLD" ]]; then
        echo "[start] threshold=$THRESHOLD"
    fi
    if [[ -n "$TRIGGER_LEVEL" ]]; then
        echo "[start] trigger-level=$TRIGGER_LEVEL"
    fi
    if [[ "$OPTIMIZE_UNSTABLE_SYSTEMS" -eq 1 ]]; then
        echo "[start] optimizeUnstableSystems=enabled"
    fi
    echo "[start] binary=$OPENWAKEWORD_BIN"
    echo "[start] onnxruntime_lib_dir=$ORT_LIB_DIR"

    if [[ ! -x "$OPENWAKEWORD_BIN" ]]; then
        echo "[start] [ERROR] openwakeword binary not found/executable: $OPENWAKEWORD_BIN" >&2
        echo "    Expected: <wakeWord>/build-android/<abi>/openwakeword" >&2
        echo "    Set OPENWAKEWORD_BIN=... to override." >&2
        exit 2
    fi

    if [[ ! -d "$ORT_LIB_DIR" ]]; then
        echo "[start] [ERROR] ORT lib dir not found: $ORT_LIB_DIR" >&2
        echo "    Expected: <wakeWord>/onnxruntime-android/<version>/jni/<abi>" >&2
        echo "    Set ORT_LIB_DIR=... to override." >&2
        exit 2
    fi

    export LD_LIBRARY_PATH="$ORT_LIB_DIR:$(dirname "$OPENWAKEWORD_BIN")${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"

    install_if_missing pulseaudio pulseaudio
    install_if_missing pactl pulseaudio
    install_if_missing rec sox

    run_termux_pulseaudio
    load_termux_pulseaudio_module module-sles-source

    mic_source=""
    if ! mic_source="$(pick_termux_mic_source)"; then
        echo "[start] [ERROR] No PulseAudio sources found (microphone not available)." >&2
        echo "    Check Android permission: Settings -> Apps -> Termux -> Microphone" >&2
        exit 2
    fi

    echo "[start] (mic) pulseaudio source=$mic_source"
    rec -q -t pulseaudio "$mic_source" -c 1 -r 16000 -b 16 -e signed-integer -t raw - \
        | "$OPENWAKEWORD_BIN" \
            --model "$model_path" \
            "${THRESHOLD_ARGS[@]-}" \
            "${TRIGGER_LEVEL_ARGS[@]-}" \
            "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
            "${REMAINING_ARGS[@]-}"
    exit $?
fi

OS_RAW="$(uname -s)"
ARCH_RAW="$(uname -m)"

OS_ID=""
case "$OS_RAW" in
    Linux*) OS_ID="linux" ;;
    Darwin*) OS_ID="macos" ;;
    MINGW*|MSYS*|CYGWIN*) OS_ID="windows" ;;
    *) OS_ID="$OS_RAW" ;;
esac

ARCH_ID=""
case "$ARCH_RAW" in
    x86_64|amd64) ARCH_ID="x64" ;;
    aarch64|arm64) ARCH_ID="arm64" ;;
    *) ARCH_ID="$ARCH_RAW" ;;
esac

install_pc_tools_if_missing "$OS_RAW"

OPENWAKEWORD_BIN="${OPENWAKEWORD_BIN:-$SCRIPT_DIR/../build-pc/${OS_ID}-${ARCH_ID}/openwakeword}"
BUILD_DIR="$(dirname "$OPENWAKEWORD_BIN")"

ORT_BASE_DIR="$SCRIPT_DIR/../onnxruntime-pc"
ORT_LIB_DIR_DEFAULT=""
marker_ort_version=""
MARKER_FILE="$BUILD_DIR/onnxruntime-version"
if [[ -f "$MARKER_FILE" ]]; then
    marker_ort_version="$(tr -d '\r\n' < "$MARKER_FILE")"
fi
if [[ -n "$marker_ort_version" && -d "$ORT_BASE_DIR/$marker_ort_version/${OS_ID}-${ARCH_ID}/lib" ]]; then
    ORT_LIB_DIR_DEFAULT="$ORT_BASE_DIR/$marker_ort_version/${OS_ID}-${ARCH_ID}/lib"
elif [[ -d "$ORT_BASE_DIR" ]]; then
    latest_ort_version="$(ls -1 "$ORT_BASE_DIR" 2>/dev/null | sort -V | tail -n 1 || true)"
    if [[ -n "$latest_ort_version" && -d "$ORT_BASE_DIR/$latest_ort_version/${OS_ID}-${ARCH_ID}/lib" ]]; then
        ORT_LIB_DIR_DEFAULT="$ORT_BASE_DIR/$latest_ort_version/${OS_ID}-${ARCH_ID}/lib"
    fi
fi

OPENWAKEWORD_ORT_LIB_DIR="${OPENWAKEWORD_ORT_LIB_DIR:-$ORT_LIB_DIR_DEFAULT}"

if [[ "$OS_RAW" == MINGW* || "$OS_RAW" == MSYS* || "$OS_RAW" == CYGWIN* ]]; then
    if [[ "$OPENWAKEWORD_BIN" != *.exe ]]; then
        OPENWAKEWORD_BIN="${OPENWAKEWORD_BIN}.exe"
    fi
fi

echo "[start] runtime=pc"
echo "[start] model=$model_path"
if [[ -n "$THRESHOLD" ]]; then
    echo "[start] threshold=$THRESHOLD"
fi
if [[ -n "$TRIGGER_LEVEL" ]]; then
    echo "[start] trigger-level=$TRIGGER_LEVEL"
fi
if [[ "$OPTIMIZE_UNSTABLE_SYSTEMS" -eq 1 ]]; then
    echo "[start] optimizeUnstableSystems=enabled"
fi
echo "[start] binary=$OPENWAKEWORD_BIN"
if [[ -n "$OPENWAKEWORD_ORT_LIB_DIR" ]]; then
    echo "[start] onnxruntime_lib_dir=$OPENWAKEWORD_ORT_LIB_DIR"
fi

if [[ ! -x "$OPENWAKEWORD_BIN" ]]; then
    echo "[start] [ERROR] Executable not found/executable: $OPENWAKEWORD_BIN" >&2
    echo "    Run ./build.sh first or set OPENWAKEWORD_BIN=..." >&2
    exit 2
fi

case "$OS_RAW" in
    Linux*)
        if [[ -n "$OPENWAKEWORD_ORT_LIB_DIR" ]]; then
            export LD_LIBRARY_PATH="$OPENWAKEWORD_ORT_LIB_DIR${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
        fi
        if command -v arecord >/dev/null 2>&1; then
            echo "[start] Linux microphone via arecord"
            arecord -r 16000 -c 1 -f S16_LE -t raw - \
                | exec "$OPENWAKEWORD_BIN" \
                    --model "$model_path" \
                    "${THRESHOLD_ARGS[@]-}" \
                    "${TRIGGER_LEVEL_ARGS[@]-}" \
                    "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
                    "${REMAINING_ARGS[@]-}"
        elif command -v rec >/dev/null 2>&1; then
            echo "[start] Linux microphone via SoX rec"
            rec -q -c 1 -b 16 -e signed-integer -t raw - rate 16000 \
                | exec "$OPENWAKEWORD_BIN" \
                    --model "$model_path" \
                    "${THRESHOLD_ARGS[@]-}" \
                    "${TRIGGER_LEVEL_ARGS[@]-}" \
                    "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
                    "${REMAINING_ARGS[@]-}"
        else
            echo "[start] [ERROR] No microphone capture tool found (install arecord or sox)." >&2
            exit 2
        fi
        ;;
    Darwin*)
        if [[ -n "$OPENWAKEWORD_ORT_LIB_DIR" ]]; then
            export DYLD_LIBRARY_PATH="$OPENWAKEWORD_ORT_LIB_DIR${DYLD_LIBRARY_PATH:+:$DYLD_LIBRARY_PATH}"
        fi
        if command -v rec >/dev/null 2>&1; then
            echo "[start] macOS microphone via SoX rec"
            rec -q -c 1 -b 16 -e signed-integer -t raw - rate 16000 \
                | exec "$OPENWAKEWORD_BIN" \
                    --model "$model_path" \
                    "${THRESHOLD_ARGS[@]-}" \
                    "${TRIGGER_LEVEL_ARGS[@]-}" \
                    "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
                    "${REMAINING_ARGS[@]-}"
        else
            echo "[start] [ERROR] SoX (rec) not found. Install sox for microphone capture." >&2
            exit 2
        fi
        ;;
    MINGW*|MSYS*|CYGWIN*)
        if [[ -n "$OPENWAKEWORD_ORT_LIB_DIR" ]]; then
            export PATH="$OPENWAKEWORD_ORT_LIB_DIR:$PATH"
        else
            echo "[start] [ERROR] ONNX Runtime lib dir not found for Windows." >&2
            echo "    Set OPENWAKEWORD_ORT_LIB_DIR=..." >&2
            exit 2
        fi
        if command -v ffmpeg >/dev/null 2>&1; then
            WIN_MIC="${OPENWAKEWORD_WIN_MIC:-}"
            if [[ -z "$WIN_MIC" ]]; then
                WIN_MIC="default"
                if ! ffmpeg -hide_banner -loglevel error -f dshow -i "audio=$WIN_MIC" -t 0.2 -f null - >/dev/null 2>&1; then
                    WIN_MIC="virtual-audio-capturer"
                    if ! ffmpeg -hide_banner -loglevel error -f dshow -i "audio=$WIN_MIC" -t 0.2 -f null - >/dev/null 2>&1; then
                        detected_mic="$(ffmpeg -hide_banner -list_devices true -f dshow -i dummy 2>&1 | awk -F'"' '/\(audio\)/ { print $2; exit }')"
                        if [[ -n "$detected_mic" ]]; then
                            WIN_MIC="$detected_mic"
                        else
                            echo "[start] [ERROR] Could not auto-detect Windows microphone for ffmpeg dshow." >&2
                            echo "    Try one of:" >&2
                            echo "      OPENWAKEWORD_WIN_MIC=\"Microphone (...)\" ./start.sh" >&2
                            echo "      OPENWAKEWORD_WIN_MIC=\"default\" ./start.sh" >&2
                            echo "    Device list command:" >&2
                            echo "      ffmpeg -list_devices true -f dshow -i dummy" >&2
                            exit 2
                        fi
                    fi
                fi
            fi

            echo "[start] Windows microphone via ffmpeg dshow (device: $WIN_MIC)"
            ffmpeg -hide_banner -loglevel error \
                -f dshow -i "audio=$WIN_MIC" \
                -ac 1 -ar 16000 -f s16le -acodec pcm_s16le - \
                | exec "$OPENWAKEWORD_BIN" \
                    --model "$model_path" \
                    "${THRESHOLD_ARGS[@]-}" \
                    "${TRIGGER_LEVEL_ARGS[@]-}" \
                    "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
                    "${REMAINING_ARGS[@]-}"
        elif command -v rec >/dev/null 2>&1; then
            echo "[start] Windows microphone via SoX rec"
            rec -q -c 1 -b 16 -e signed-integer -t raw - rate 16000 \
                | exec "$OPENWAKEWORD_BIN" \
                    --model "$model_path" \
                    "${THRESHOLD_ARGS[@]-}" \
                    "${TRIGGER_LEVEL_ARGS[@]-}" \
                    "${OPTIMIZE_UNSTABLE_SYSTEMS_ARGS[@]-}" \
                    "${REMAINING_ARGS[@]-}"
        else
            echo "[start] [ERROR] No microphone capture tool found on Windows (install ffmpeg or sox)." >&2
            echo "    For ffmpeg device list: ffmpeg -list_devices true -f dshow -i dummy" >&2
            echo "    Optionally set OPENWAKEWORD_WIN_MIC to the exact device name." >&2
            exit 2
        fi
        ;;
    *)
        echo "[start] [ERROR] Unsupported OS: $OS_RAW" >&2
        exit 2
        ;;
esac