#!/bin/bash

# 设置颜色输出
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${YELLOW}停止开发环境服务...${NC}"

# 从PID文件读取进程ID并停止
if [ -f frontend.pid ]; then
    FRONTEND_PID=$(cat frontend.pid)
    if ps -p $FRONTEND_PID > /dev/null; then
        kill $FRONTEND_PID
        echo -e "${GREEN}前端服务已停止${NC}"
    fi
    rm -f frontend.pid
fi

if [ -f backend.pid ]; then
    BACKEND_PID=$(cat backend.pid)
    if ps -p $BACKEND_PID > /dev/null; then
        kill $BACKEND_PID
        echo -e "${GREEN}后端服务已停止${NC}"
    fi
    rm -f backend.pid
fi

if [ -f collector.pid ]; then
    COLLECTOR_PID=$(cat collector.pid)
    if ps -p $COLLECTOR_PID > /dev/null; then
        kill $COLLECTOR_PID
        echo -e "${GREEN}数据采集服务已停止${NC}"
    fi
    rm -f collector.pid
fi

if [ -f spark.pid ]; then
    SPARK_PID=$(cat spark.pid)
    if ps -p $SPARK_PID > /dev/null; then
        kill $SPARK_PID
        echo -e "${GREEN}Spark处理器已停止${NC}"
    fi
    rm -f spark.pid
fi

# 额外检查：强制停止相关进程
echo -e "${YELLOW}检查是否有残留进程...${NC}"

# 停止可能的残留进程
pkill -f "vite"
pkill -f "spring-boot:run"
pkill -f "stock_crawler.py"
pkill -f "StockDataProcessor"

echo -e "${GREEN}所有开发服务已停止！${NC}"