#!/bin/bash

API_BASE="http://localhost:3000/api"

echo "=========================================="
echo "  多智能体任务管理系统 - 功能演示"
echo "=========================================="
echo ""

# 1. 注册智能体
echo "📝 步骤 1: 注册智能体..."
AGENT_RESPONSE=$(curl -s -X POST $API_BASE/agents/register \
  -H "Content-Type: application/json" \
  -d '{
    "name": "演示智能体",
    "type": "general",
    "capabilities": ["数据分析", "任务处理"]
  }')

AGENT_ID=$(echo $AGENT_RESPONSE | grep -o '"id":"[^"]*"' | cut -d'"' -f4)
echo "✅ 智能体已注册"
echo "   ID: $AGENT_ID"
echo ""

# 2. 查看所有智能体
echo "📋 步骤 2: 查看所有智能体..."
curl -s $API_BASE/agents | jq '.[0] | {id, name, type, status}'
echo ""

# 3. 创建任务
echo "📌 步骤 3: 创建任务..."
TASK_RESPONSE=$(curl -s -X POST $API_BASE/tasks \
  -H "Content-Type: application/json" \
  -d '{
    "title": "测试任务：数据分析",
    "description": "分析用户行为数据并生成报告",
    "priority": "high",
    "created_by": "human"
  }')

TASK_ID=$(echo $TASK_RESPONSE | grep -o '"id":"[^"]*"' | cut -d'"' -f4)
echo "✅ 任务已创建"
echo "   ID: $TASK_ID"
echo ""

# 4. 查看待处理任务
echo "📋 步骤 4: 查看待处理任务..."
curl -s "$API_BASE/tasks?status=pending" | jq '.[0] | {id, title, priority, status}'
echo ""

# 5. 智能体领取任务
echo "🤖 步骤 5: 智能体领取任务..."
curl -s -X POST $API_BASE/tasks/$TASK_ID/claim \
  -H "Content-Type: application/json" \
  -d "{\"agent_id\": \"$AGENT_ID\"}" | jq '.'
echo ""

# 6. 发送心跳（标记为忙碌）
echo "💓 步骤 6: 发送心跳（忙碌状态）..."
curl -s -X POST $API_BASE/agents/$AGENT_ID/heartbeat \
  -H "Content-Type: application/json" \
  -d '{"status": "busy"}' | jq '.'
echo ""

# 7. 模拟任务执行
echo "⏳ 步骤 7: 模拟任务执行（3秒）..."
sleep 3
echo ""

# 8. 完成任务
echo "✅ 步骤 8: 标记任务完成..."
curl -s -X PUT $API_BASE/tasks/$TASK_ID \
  -H "Content-Type: application/json" \
  -d '{
    "status": "completed",
    "result": "数据分析完成，发现3个关键趋势：1) 用户活跃度提升20% 2) 转化率增长15% 3) 留存率稳定在85%"
  }' | jq '.'
echo ""

# 9. 查看任务详情
echo "📊 步骤 9: 查看完成的任务详情..."
curl -s $API_BASE/tasks/$TASK_ID | jq '{id, title, status, result, completed_at}'
echo ""

# 10. 查看统计信息
echo "📈 步骤 10: 查看系统统计..."
curl -s $API_BASE/stats | jq '.'
echo ""

echo "=========================================="
echo "  演示完成！"
echo "=========================================="
echo ""
echo "💡 提示："
echo "  - Web 界面: http://localhost:3000"
echo "  - 查看所有智能体: curl $API_BASE/agents | jq"
echo "  - 查看所有任务: curl $API_BASE/tasks | jq"