#!/bin/bash

# 🟢 小綠重置工具 v1.2
OPENCLAW_DIR="$HOME/.openclaw"
WORKING="$OPENCLAW_DIR/golden-working.json"
CONFIG="$OPENCLAW_DIR/openclaw.json"

get_model() { grep -A2 '"primary"' "$CONFIG" 2>/dev/null | head -1 | sed 's/.*"primary": "\(.*\)".*/\1/'; }

echo "========================================"
echo "  🟢 小綠重置工具 v1.2"
echo "========================================"
echo "  目前：$(get_model)"
echo ""
echo "  [1] 儲存還原點"
echo "  [2] 還原 DeepSeek（穩定）"
echo "  [3] 切換到 Mimo（OpenRouter）"
echo "  [4] 重啟 Gateway"
echo "  [q] 離開"
echo ""
read -p "  請選擇: " choice

case $choice in
  1) cp "$CONFIG" "$WORKING"
     TS=$(date +%Y%m%d-%H%M)
     cp "$CONFIG" "$OPENCLAW_DIR/snapshot-${TS}.json"
     echo "✅ 已儲存 snapshot-${TS}.json" ;;
  2) [ -f "$WORKING" ] && cp "$WORKING" "$CONFIG" && echo "✅ 還原 DeepSeek，重啟中..." && openclaw gateway restart || echo "❌ 無還原點" ;;
  3) python3 -c "import json; c=json.load(open('$CONFIG')); c['agents']['defaults']['model']={'primary':'openrouter/xiaomi/mimo-v2.5-pro','fallbacks':['openrouter/xiaomi/mimo-v2.5','openrouter/auto']}; json.dump(c,open('$CONFIG','w'),indent=2)"
     echo "✅ 切到 Mimo，重啟中..."
     openclaw gateway restart ;;
  4) echo "重啟中..." && openclaw gateway restart ;;
  q) echo "掰掰 👋" ;;
  *) echo "❌ 錯誤" ;;
esac
echo "按 Enter 離開..."
read