#!/bin/bash
# ═══════════════════════════════════════════════════════
#   ⚔️ 幸赋链AI军团 - macOS 授权码激活程序
#   使用方法: chmod +x activate.sh && ./activate.sh
# ═══════════════════════════════════════════════════════

GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
CYAN='\033[0;36m'
NC='\033[0m'

echo -e "${CYAN}"
echo "╔══════════════════════════════════════════╗"
echo "║    ⚔️ 幸赋链AI军团 - 授权码激活程序      ║"
echo "╚══════════════════════════════════════════╝"
echo -e "${NC}"

INSTALL_DIR="$HOME/.xingfulian"
CONFIG_FILE="$INSTALL_DIR/config/config.yml"

# =============================================
# 检查安装目录
# =============================================
if [ ! -d "$INSTALL_DIR" ]; then
    echo -e "${RED}❌ 未检测到安装目录${NC}"
    echo "请先运行 install.sh 完成安装"
    exit 1
fi

# =============================================
# 检测网络连接
# =============================================
echo -e "${YELLOW}[检测网络连接...]${NC}"
if ! ping -c 1 -W 3 api.xingfulian.cn &>/dev/null; then
    echo -e "${RED}⚠️ 无法访问 api.xingfulian.cn${NC}"
    echo "请检查网络连接后重试"
    read -p "按回车重试，或输入 q 退出: " RETRY
    if [ "$RETRY" = "q" ]; then
        exit 1
    fi
    exec "$0"
fi
echo -e "${GREEN}    ✅ 网络连接正常${NC}"
echo ""

# =============================================
# 输入授权码
# =============================================
echo "请输入你的授权码"
echo "授权码格式：XFL-DXFLQM-XXXXXXXXXXXX"
echo "可在 https://app.xingfulian.cn 查看你的授权码"
echo ""
read -p "授权码: " LICENSE_KEY

if [ -z "$LICENSE_KEY" ]; then
    echo -e "${RED}❌ 授权码不能为空${NC}"
    exit 1
fi

echo ""
echo -e "${YELLOW}[正在验证授权码...]${NC}"

# =============================================
# 调用API验证授权码
# =============================================
API_RESPONSE=$(curl -s -X POST "https://api.xingfulian.cn/api/license/verify" \
    -H "Content-Type: application/json" \
    -d "{\"key\": \"$LICENSE_KEY\"}" 2>&1)

if [ $? -ne 0 ]; then
    echo -e "${RED}❌ 网络请求失败：$API_RESPONSE${NC}"
    exit 1
fi

SUCCESS=$(echo "$API_RESPONSE" | python3 -c "import json,sys; d=json.load(sys.stdin); print('true' if d.get('success') else 'false')" 2>/dev/null)

echo ""
if [ "$SUCCESS" = "true" ]; then
    echo -e "${GREEN}✅ 授权码验证成功！${NC}"
    
    # 写入配置文件
    cat > "$CONFIG_FILE" << CONFIGEOF
# 幸赋链AI军团 - 配置文件
# 授权码激活时间：$(date)

version: "1.0"
brand: "幸赋链AI军团"

server:
  host: "127.0.0.1"
  port: 7070

license:
  key: "${LICENSE_KEY}"
  status: "active"
  activated_at: "$(date)"
  verify_url: "https://api.xingfulian.cn/api/license/verify"
  activate_url: "https://api.xingfulian.cn/api/license/activate"

skills:
  dir: "skills"
  auto_update: true

logs:
  dir: "logs"
  level: "info"
CONFIGEOF

    echo ""
    echo -e "${CYAN}"
    echo "╔══════════════════════════════════════════╗"
    echo "║       ✅ 授权码激活成功！                ║"
    echo "║                                          ║"
    echo "║  现在运行 ./start.sh 启动AI军团          ║"
    echo "╚══════════════════════════════════════════╝"
    echo -e "${NC}"
else
    MESSAGE=$(echo "$API_RESPONSE" | python3 -c "import json,sys; d=json.load(sys.stdin); print(d.get('message','未知错误'))" 2>/dev/null)
    echo -e "${RED}❌ 授权码验证失败${NC}"
    echo "错误信息：$MESSAGE"
    echo ""
    echo "可能原因："
    echo "  1. 授权码输入有误"
    echo "  2. 授权码已过期或已使用"
    echo "  3. 网络连接异常"
    echo ""
    echo "请确认后重试，或前往 https://app.xingfulian.cn 查看"
fi

echo ""