#!/usr/bin/env bash
# Smoke test for the Wyoming voice pipeline.
# Tests: STT server alive | TTS server alive | TTS generates audio
#
# Usage: ./test-pipeline.sh

set -euo pipefail

STT_HOST="${STT_HOST:-localhost}"
STT_PORT="${STT_PORT:-10300}"
TTS_HOST="${TTS_HOST:-localhost}"
TTS_PORT="${TTS_PORT:-10301}"
VENV="${VENV:-$HOME/homeai-voice-env}"

PASS=0
FAIL=0

check() {
    local name="$1"; local result="$2"
    if [[ "$result" == ok* ]]; then
        echo "  [PASS] $name${result#ok}"; PASS=$((PASS + 1))
    else
        echo "  [FAIL] $name — $result"; FAIL=$((FAIL + 1))
    fi
}

echo "=== HomeAI Voice Pipeline Smoke Test ==="
echo ""

echo "1. STT Wyoming server (port $STT_PORT)"
if nc -z -w2 "$STT_HOST" "$STT_PORT" 2>/dev/null; then
    check "STT port open" "ok"
else
    check "STT port open" "port $STT_PORT not reachable — is wyoming-stt running?"
fi

echo ""
echo "2. TTS Wyoming server (port $TTS_PORT)"
if nc -z -w2 "$TTS_HOST" "$TTS_PORT" 2>/dev/null; then
    check "TTS port open" "ok"
else
    check "TTS port open" "port $TTS_PORT not reachable — is wyoming-tts running?"
fi

echo ""
echo "3. Kokoro TTS synthesis test"
TTS_OUTPUT="/tmp/homeai-tts-test.wav"
"$VENV/bin/python3" - <<'PYEOF'
import sys, os, asyncio
import numpy as np

model_dir = os.path.expanduser("~/models/kokoro")
model_path = os.path.join(model_dir, "kokoro-v1.0.onnx")
voices_path = os.path.join(model_dir, "voices-v1.0.bin")

if not os.path.exists(model_path):
    print(f"Model not found: {model_path}")
    sys.exit(1)

from kokoro_onnx import Kokoro
tts = Kokoro(model_path, voices_path)
samples, sr = tts.create("Hello, I am your home assistant. The voice pipeline is working.", voice="af_heart", speed=1.0)

# Write WAV
import wave, struct
samples_int16 = (np.clip(samples, -1.0, 1.0) * 32767).astype(np.int16)
with wave.open("/tmp/homeai-tts-test.wav", "w") as wf:
    wf.setnchannels(1)
    wf.setsampwidth(2)
    wf.setframerate(sr)
    wf.writeframes(samples_int16.tobytes())

print(f"Generated {len(samples)/sr:.1f}s of audio at {sr}Hz")
sys.exit(0)
PYEOF
if [[ $? -eq 0 && -f "$TTS_OUTPUT" ]]; then
    size=$(wc -c < "$TTS_OUTPUT")
    check "Kokoro synthesis" "ok — ${size} bytes written to $TTS_OUTPUT"
    echo ""
    echo "  To play: afplay $TTS_OUTPUT"
else
    check "Kokoro synthesis" "synthesis failed"
fi

echo ""
echo "─────────────────────────────────"
echo "Results: $PASS passed, $FAIL failed"
[[ $FAIL -eq 0 ]]