stop_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'
NC='\033[0m' # No Color

LOG_DIR="logs"

# 停止进程
stop_service() {
    local pid_file=$1
    local service_name=$2
    
    if [ -f "$pid_file" ]; then
        local pid=$(cat $pid_file)
        if ps -p $pid > /dev/null 2>&1; then
            echo -n "  停止 $service_name (PID: $pid)..."
            kill $pid 2>/dev/null
            sleep 1
            if ps -p $pid > /dev/null 2>&1; then
                kill -9 $pid 2>/dev/null
            fi
            echo -e " ${GREEN}✓${NC}"
        else
            echo -e "  ${YELLOW}⚠${NC} $service_name 未运行"
        fi
        rm -f $pid_file
    else
        echo -e "  ${YELLOW}⚠${NC} 未找到 $service_name 的 PID 文件"
    fi
}

# 通过端口停止进程
stop_by_port() {
    local port=$1
    local service_name=$2
    
    local pid=$(lsof -ti:$port 2>/dev/null)
    if [ -n "$pid" ]; then
        echo -n "  停止 $service_name (端口: $port, PID: $pid)..."
        kill $pid 2>/dev/null
        sleep 1
        if ps -p $pid > /dev/null 2>&1; then
            kill -9 $pid 2>/dev/null
        fi
        echo -e " ${GREEN}✓${NC}"
    else
        echo -e "  ${YELLOW}⚠${NC} 端口 $port 没有进程运行"
    fi
}

# 通过进程名停止
stop_by_name() {
    local process_name=$1
    local display_name=$2
    
    local pids=$(pgrep -f "$process_name")
    if [ -n "$pids" ]; then
        echo -n "  停止 $display_name..."
        echo "$pids" | xargs kill 2>/dev/null
        sleep 1
        # 强制停止残留进程
        pgrep -f "$process_name" | xargs kill -9 2>/dev/null
        echo -e " ${GREEN}✓${NC}"
    else
        echo -e "  ${YELLOW}⚠${NC} 未找到 $display_name 进程"
    fi
}

echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "1️⃣  通过 PID 文件停止服务"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

# 通过 PID 文件停止服务
if [ -d "$LOG_DIR" ]; then
    stop_service "$LOG_DIR/tool.pid" "Tool Service"
    stop_service "$LOG_DIR/debugger.pid" "Debugger Service"
    stop_service "$LOG_DIR/checker.pid" "Checker Service"
    stop_service "$LOG_DIR/coder.pid" "Coder Service"
    stop_service "$LOG_DIR/llm_server.pid" "LLM Server"
fi

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "2️⃣  通过端口检查并停止残留进程"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

stop_by_port 8004 "Tool Service"
stop_by_port 8003 "Debugger Service"
stop_by_port 8002 "Checker Service"
stop_by_port 8001 "Coder Service"
stop_by_port 8000 "LLM Server"

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "3️⃣  通过进程名停止特定服务"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

stop_by_name "real_llm_server.py" "Real LLM Server"
stop_by_name "buggy_llm_server.py" "Buggy LLM Server"
stop_by_name "simple_llm_server.py" "Simple LLM Server"
stop_by_name "local_llm_service_restful.py" "LLM Services"

# 清理旧的 agent_tool_clients.py 进程（如果有）
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "4️⃣  清理旧服务进程"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

stop_by_name "agent_tool_clients.py" "旧版 Agent Tool Clients"

# 停止 Redis Docker 容器（如果是通过脚本启动的）
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "5️⃣  检查 Redis Docker 容器"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

if command -v docker &> /dev/null; then
    if docker ps | grep -q redis-langgraph; then
        echo -n "  停止 Redis Docker 容器..."
        docker stop redis-langgraph > /dev/null 2>&1
        docker rm redis-langgraph > /dev/null 2>&1
        echo -e " ${GREEN}✓${NC}"
    else
        echo -e "  ${YELLOW}⚠${NC} Redis Docker 容器未运行"
    fi
else
    echo -e "  ${YELLOW}⚠${NC} Docker 未安装，跳过"
fi

# 验证所有服务是否已停止
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "6️⃣  验证服务状态"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

ports=(8000 8001 8002 8003 8004)
all_stopped=true

for port in "${ports[@]}"; do
    if lsof -Pi :$port -sTCP:LISTEN -t >/dev/null 2>&1; then
        echo -e "  ${RED}✗${NC} 端口 $port 仍有进程运行"
        all_stopped=false
    else
        echo -e "  ${GREEN}✓${NC} 端口 $port 已清空"
    fi
done

# 清理日志（可选）
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "7️⃣  日志文件管理"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""

read -p "是否清理日志文件? (y/N): " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    if [ -d "$LOG_DIR" ]; then
        echo -n "  清理日志文件..."
        rm -f $LOG_DIR/*.log
        rm -f $LOG_DIR/*.pid
        echo -e " ${GREEN}✓${NC}"
    fi
else
    echo -e "  ${YELLOW}保留日志文件${NC}"
    if [ -d "$LOG_DIR" ]; then
        echo ""
        echo "  可用日志文件:"
        ls -lh $LOG_DIR/*.log 2>/dev/null | awk '{print "    " $9 " (" $5 ")"}'
    fi
fi

echo ""
echo "╔══════════════════════════════════════════════════════════╗"
if [ "$all_stopped" = true ]; then
    echo "║                   ✅ 停止完成                             ║"
else
    echo "║                   ⚠️  部分服务未停止                      ║"
fi
echo "╚══════════════════════════════════════════════════════════╝"
echo ""

if [ "$all_stopped" = true ]; then
    echo "所有服务已成功停止。"
else
    echo "部分服务未能停止，请手动检查："
    echo "  ps aux | grep -E '(llm_server|local_llm_service)' | grep -v grep"
    echo "  lsof -i :8000-8004"
fi

echo ""
echo "💡 提示:"
echo "  • 重新启动: ./start_all_services.sh"
echo "  • 查看日志: ls -lh logs/"
echo ""