#!/bin/bash
# 停止 Python 后端服务脚本

# 默认端口（从 server_config.py 中获取）
DEFAULT_PORT=7861

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

echo -e "${YELLOW}正在查找运行中的 Python 后端进程...${NC}"

# 方法1: 通过进程名查找
PIDS=$(ps aux | grep "[p]ython.*startup.py -a" | awk '{print $2}')

# 方法2: 如果方法1没找到，通过端口查找
if [ -z "$PIDS" ]; then
    echo -e "${YELLOW}未找到 startup.py 进程，尝试通过端口 ${DEFAULT_PORT} 查找...${NC}"
    PIDS=$(lsof -ti:${DEFAULT_PORT} 2>/dev/null)
fi

if [ -z "$PIDS" ]; then
    echo -e "${RED}未找到运行中的 Python 后端进程${NC}"
    exit 0
fi

echo -e "${GREEN}找到以下进程:${NC}"
ps aux | grep "[p]ython.*startup.py -a" | grep -v grep
if [ -n "$PIDS" ]; then
    echo -e "${YELLOW}进程 PID: $PIDS${NC}"
fi

# 询问是否确认停止
read -p "是否停止这些进程? (y/n): " -n 1 -r
echo
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
    echo -e "${YELLOW}已取消${NC}"
    exit 0
fi

# 停止进程
for PID in $PIDS; do
    if [ -n "$PID" ]; then
        echo -e "${YELLOW}正在停止进程 $PID...${NC}"
        kill $PID 2>/dev/null
        if [ $? -eq 0 ]; then
            echo -e "${GREEN}进程 $PID 已发送停止信号${NC}"
        else
            echo -e "${RED}无法停止进程 $PID${NC}"
        fi
    fi
done

# 等待进程结束
echo -e "${YELLOW}等待进程结束...${NC}"
sleep 3

# 检查是否还有进程在运行
REMAINING=$(ps aux | grep "[p]ython.*startup.py -a" | awk '{print $2}')
if [ -n "$REMAINING" ]; then
    echo -e "${YELLOW}仍有进程在运行，强制终止...${NC}"
    for PID in $REMAINING; do
        kill -9 $PID 2>/dev/null
        echo -e "${GREEN}已强制终止进程 $PID${NC}"
    done
fi

echo -e "${GREEN}所有进程已停止${NC}"