#!/bin/bash
# MiMo V2.5 TTS — Hermes Agent Plugin Setup
# 用法: bash setup.sh [--uninstall]

set -euo pipefail

HERMES_HOME="${HERMES_HOME:-$HOME/.hermes}"
PLUGIN_DIR="$HERMES_HOME/plugins/tts/mimo"
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"

GREEN='\033[0;32m'; YELLOW='\033[1;33m'; RED='\033[0;31m'; NC='\033[0m'
ok()   { echo -e "${GREEN}✓${NC} $*"; }
warn() { echo -e "${YELLOW}⚠${NC} $*"; }
fail() { echo -e "${RED}✗${NC} $*" >&2; }

# ── Uninstall ──────────────────────────────────────────────────────
if [[ "${1:-}" == "--uninstall" ]]; then
    echo "Uninstalling MiMo TTS plugin..."
    command -v hermes &>/dev/null && hermes plugins disable tts/mimo 2>/dev/null || true
    rm -rf "$PLUGIN_DIR"
    rm -f "$HOME/mimo_tts.py"
    ok "Done. Restart Hermes session (/reset) to take effect."
    exit 0
fi

# ── Install ────────────────────────────────────────────────────────
echo "Installing MiMo V2.5 TTS plugin for Hermes Agent..."

# 1. Pre-flight checks
[[ -d "$HERMES_HOME" ]] || { fail "Hermes not found at $HERMES_HOME"; exit 1; }

if [[ -f "$HERMES_HOME/.env" ]]; then
    grep -q "^XIAOMI_API_KEY=" "$HERMES_HOME/.env" 2>/dev/null \
        && ok "XIAOMI_API_KEY found" \
        || warn "XIAOMI_API_KEY missing in $HERMES_HOME/.env"
else
    warn ".env not found — add XIAOMI_API_KEY before using TTS"
fi

# 2. Install plugin
mkdir -p "$PLUGIN_DIR"
cp "$SCRIPT_DIR/plugin.yaml" "$PLUGIN_DIR/plugin.yaml"
cp "$SCRIPT_DIR/__init__.py" "$PLUGIN_DIR/__init__.py"
ok "Plugin installed → $PLUGIN_DIR"

# 3. Copy standalone script
cp "$SCRIPT_DIR/mimo_tts.py" "$HOME/mimo_tts.py"
chmod +x "$HOME/mimo_tts.py"
ok "Standalone script → $HOME/mimo_tts.py"

# 4. Enable in Hermes
if command -v hermes &>/dev/null; then
    hermes plugins enable tts/mimo 2>/dev/null && ok "Enabled in Hermes" \
        || warn "Run manually: hermes plugins enable tts/mimo"
else
    warn "hermes not in PATH — run: hermes plugins enable tts/mimo"
fi

# 5. Done
echo ""
echo "── Next steps ──────────────────────────────────────"
echo "  hermes config set tts.provider mimo"
echo "  hermes config set tts.voice Mia       # or 冰糖, 茉莉, Chloe, etc."
echo "  /reset                                 # restart session"
echo ""
echo "  Standalone: python3 ~/mimo_tts.py --list-voices"
echo "────────────────────────────────────────────────────"