start_all_services.sh

#!/bin/bash

# LangGraph 代码助手 - 启动所有服务脚本

echo "╔══════════════════════════════════════════════════════════╗"
echo "║                                                          ║"
echo "║         启动 LangGraph 代码助手所有服务                   ║"
echo "║                                                          ║"
echo "╚══════════════════════════════════════════════════════════╝"
echo ""

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 日志目录
LOG_DIR="logs"
mkdir -p $LOG_DIR

# 解析命令行参数
LLM_MODE="real"  # 默认使用真实 LLM (real_llm_server.py)

while [[ $# -gt 0 ]]; do
    case $1 in
        --mock)
            LLM_MODE="mock"
            shift
            ;;
        --buggy)
            LLM_MODE="buggy"
            shift
            ;;
        --help)
            echo "用法: $0 [选项]"
            echo ""
            echo "选项:"
            echo "  --mock     使用 simple_llm_server.py (模拟 LLM，不调用真实 API)"
            echo "  --buggy    使用 buggy_llm_server.py (故意返回错误代码，用于测试)"
            echo "  默认       使用 real_llm_server.py (调用真实 Kimi API)"
            echo ""
            echo "示例:"
            echo "  $0              # 使用真实 LLM (Kimi API)"
            echo "  $0 --mock       # 使用模拟 LLM"
            echo "  $0 --buggy      # 使用 Buggy LLM (测试错误处理)"
            exit 0
            ;;
        *)
            echo "未知选项: $1"
            echo "使用 --help 查看帮助"
            exit 1
            ;;
    esac
done

# 检查端口是否被占用
check_port() {
    local port=$1
    if lsof -Pi :$port -sTCP:LISTEN -t >/dev/null 2>&1 ; then
        return 0  # 端口被占用
    else
        return 1  # 端口空闲
    fi
}

# 等待服务启动
wait_for_service() {
    local host=$1
    local port=$2
    local service_name=$3
    local max_wait=30
    local count=0
    
    echo -n "  等待 $service_name 启动..."
    while [ $count -lt $max_wait ]; do
        if nc -z $host $port 2>/dev/null || curl -s http://$host:$port/health > /dev/null 2>&1; then
            echo -e " ${GREEN}✓${NC}"
            return 0
        fi
        sleep 1
        count=$((count + 1))
        echo -n "."
    done
    echo -e " ${RED}✗ 超时${NC}"
    return 1
}

# 检查 Redis
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "1️⃣  检查 Redis 服务"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 6379; then
    echo -e "  ${GREEN}✓${NC} Redis 已在运行 (localhost:6379)"
else
    echo -e "  ${YELLOW}⚠${NC} Redis 未运行，尝试启动..."
    
    # 检查是否有 Docker
    if command -v docker &> /dev/null; then
        echo "    使用 Docker 启动 Redis..."
        docker run -d --name redis-langgraph -p 6379:6379 redis:latest > /dev/null 2>&1
        if [ $? -eq 0 ]; then
            wait_for_service localhost 6379 "Redis"
        else
            echo -e "    ${RED}✗${NC} Docker 启动失败，请手动启动 Redis"
            echo "    命令: redis-server 或 docker run -d -p 6379:6379 redis:latest"
        fi
    else
        echo -e "    ${RED}✗${NC} 请手动启动 Redis"
        echo "    命令: redis-server"
    fi
fi

# 启动 LLM 服务器
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "2️⃣  启动 LLM 服务器 (localhost:8000)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 8000; then
    echo -e "  ${YELLOW}⚠${NC} 端口 8000 已被占用，跳过启动"
else
    case $LLM_MODE in
        mock)
            echo -e "  ${BLUE}使用模式: Mock LLM (simple_llm_server.py)${NC}"
            python simple_llm_server.py > $LOG_DIR/llm_server.log 2>&1 &
            ;;
        buggy)
            echo -e "  ${YELLOW}使用模式: Buggy LLM (buggy_llm_server.py) - 用于测试${NC}"
            python buggy_llm_server.py > $LOG_DIR/llm_server.log 2>&1 &
            ;;
        *)
            echo -e "  ${GREEN}使用模式: Real LLM (real_llm_server.py - Kimi API)${NC}"
            python real_llm_server.py > $LOG_DIR/llm_server.log 2>&1 &
            ;;
    esac
    LLM_PID=$!
    echo $LLM_PID > $LOG_DIR/llm_server.pid
    wait_for_service localhost 8000 "LLM Server"
fi

# 启动 Coder 服务
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "3️⃣  启动 Coder 服务 (localhost:8001)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 8001; then
    echo -e "  ${YELLOW}⚠${NC} 端口 8001 已被占用，跳过启动"
else
    echo "  启动 Coder 服务..."
    python local_llm_service_restful.py --service coder --port 8001 --llm-url http://localhost:8000 > $LOG_DIR/coder.log 2>&1 &
    CODER_PID=$!
    echo $CODER_PID > $LOG_DIR/coder.pid
    wait_for_service localhost 8001 "Coder Service"
fi

# 启动 Checker 服务
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "4️⃣  启动 Checker 服务 (localhost:8002)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 8002; then
    echo -e "  ${YELLOW}⚠${NC} 端口 8002 已被占用，跳过启动"
else
    echo "  启动 Checker 服务..."
    python local_llm_service_restful.py --service checker --port 8002 --llm-url http://localhost:8000 > $LOG_DIR/checker.log 2>&1 &
    CHECKER_PID=$!
    echo $CHECKER_PID > $LOG_DIR/checker.pid
    wait_for_service localhost 8002 "Checker Service"
fi

# 启动 Debugger 服务
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "5️⃣  启动 Debugger 服务 (localhost:8003)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 8003; then
    echo -e "  ${YELLOW}⚠${NC} 端口 8003 已被占用，跳过启动"
else
    echo "  启动 Debugger 服务..."
    python local_llm_service_restful.py --service debugger --port 8003 --llm-url http://localhost:8000 > $LOG_DIR/debugger.log 2>&1 &
    DEBUGGER_PID=$!
    echo $DEBUGGER_PID > $LOG_DIR/debugger.pid
    wait_for_service localhost 8003 "Debugger Service"
fi

# 启动 Tool 服务
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "6️⃣  启动 Tool 服务 (localhost:8004)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

if check_port 8004; then
    echo -e "  ${YELLOW}⚠${NC} 端口 8004 已被占用，跳过启动"
else
    echo "  启动 Tool 服务..."
    python local_llm_service_restful.py --service tool --port 8004 --llm-url http://localhost:8000 > $LOG_DIR/tool.log 2>&1 &
    TOOL_PID=$!
    echo $TOOL_PID > $LOG_DIR/tool.pid
    wait_for_service localhost 8004 "Tool Service"
fi

# 总结
echo ""
echo "╔══════════════════════════════════════════════════════════╗"
echo "║                   🎉 启动完成                             ║"
echo "╚══════════════════════════════════════════════════════════╝"
echo ""
echo "📋 服务状态:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

services=(
    "Redis:6379"
    "LLM_Server:8000"
    "Coder:8001"
    "Checker:8002"
    "Debugger:8003"
    "Tool:8004"
)

for service in "${services[@]}"; do
    IFS=':' read -r name port <<< "$service"
    if check_port $port; then
        echo -e "  ${GREEN}✓${NC} $name (localhost:$port)"
    else
        echo -e "  ${RED}✗${NC} $name (localhost:$port)"
    fi
done

echo ""
echo "🌐 LLM 模式:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
case $LLM_MODE in
    mock)
        echo -e "  ${BLUE}Mock LLM${NC} - 使用预定义响应，不调用 API"
        ;;
    buggy)
        echo -e "  ${YELLOW}Buggy LLM${NC} - 故意返回错误代码，用于测试错误处理"
        ;;
    *)
        echo -e "  ${GREEN}Real LLM${NC} - 调用 Moonshot Kimi API (kimi-k2-turbo-preview)"
        echo "  API Key: sk-4VF7qW5HsuXjgOwiCl4boalx8HzY5U2HBSF8mIFPD8Hz8SLM"
        ;;
esac

echo ""
echo "📝 日志文件:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  • LLM Server: $LOG_DIR/llm_server.log"
echo "  • Coder: $LOG_DIR/coder.log"
echo "  • Checker: $LOG_DIR/checker.log"
echo "  • Debugger: $LOG_DIR/debugger.log"
echo "  • Tool: $LOG_DIR/tool.log"

echo ""
echo "🧪 测试命令:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  # 健康检查所有服务"
echo "  for port in 8000 8001 8002 8003 8004; do"
echo "    curl -s http://localhost:\$port/health | python -m json.tool"
echo "  done"
echo ""
echo "  # 运行端到端测试"
echo "  python test/test_real_llm.py --mode e2e"
echo ""
echo "  # 测试单个服务"
echo "  python test/test_real_llm.py --mode single"
echo ""
echo "  # 测试错误处理（需要使用 --buggy 模式启动）"
echo "  python test/test_with_buggy_llm.py"
echo ""

echo "⚠️  停止所有服务:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  ./stop_all_services.sh"
echo ""
echo "💡 提示:"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  • 查看实时日志: tail -f logs/[service_name].log"
echo "  • 检查服务状态: curl http://localhost:[port]/health"
echo "  • 重置 Buggy LLM 计数器: curl -X POST http://localhost:8000/reset"
echo ""