#!/bin/bash
# 重建 Kafka 容器以应用新的配置

GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
BLUE='\033[0;34m'
NC='\033[0m'

echo -e "${YELLOW}🔄 重建 Kafka 容器...${NC}"
echo ""

# 1. 停止 Blockchain Service (如果在运行)
echo -e "${BLUE}步骤 1: 停止 Blockchain Service${NC}"
PID=$(lsof -ti :3012 2>/dev/null)
if [ ! -z "$PID" ]; then
    echo "停止 Blockchain Service (PID: $PID)..."
    kill $PID
    sleep 2
    echo -e "${GREEN}✓ Blockchain Service 已停止${NC}"
else
    echo -e "${YELLOW}⚠️  Blockchain Service 未在运行${NC}"
fi
echo ""

# 2. 停止并删除 Kafka 容器
echo -e "${BLUE}步骤 2: 停止并删除旧容器${NC}"
docker compose stop kafka 2>/dev/null || true
docker compose rm -f kafka 2>/dev/null || true
echo -e "${GREEN}✓ 旧容器已删除${NC}"
echo ""

# 3. 重新创建容器
echo -e "${BLUE}步骤 3: 创建新容器${NC}"
docker compose up -d kafka
echo "等待 Kafka 启动..."
sleep 20
echo -e "${GREEN}✓ Kafka 容器已创建${NC}"
echo ""

# 4. 验证配置
echo -e "${BLUE}步骤 4: 验证配置${NC}"
CONTAINER_NAME=$(docker compose ps -q kafka 2>/dev/null)
if [ ! -z "$CONTAINER_NAME" ]; then
    ADVERTISED=$(docker inspect "$CONTAINER_NAME" 2>/dev/null | grep -A 1 "KAFKA_ADVERTISED_LISTENERS" | head -1)
    if echo "$ADVERTISED" | grep -q "localhost:9092"; then
        echo -e "${GREEN}✓ Kafka 配置已更新！${NC}"
        echo "$ADVERTISED"
    else
        echo -e "${YELLOW}⚠ Kafka 配置可能需要检查${NC}"
    fi
else
    echo -e "${YELLOW}⚠ 未找到 Kafka 容器${NC}"
fi
echo ""

# 5. 测试连接
echo -e "${BLUE}步骤 5: 测试 Kafka 连接${NC}"
if nc -zv localhost 9092 2>&1 | grep -q "succeeded\|Connected"; then
    echo -e "${GREEN}✓ Kafka 端口可访问${NC}"
else
    echo -e "${RED}✗ Kafka 端口不可访问${NC}"
    exit 1
fi
echo ""

echo -e "${YELLOW}======================================${NC}"
echo -e "${GREEN}✓ Kafka 重建完成！${NC}"
echo -e "${YELLOW}======================================${NC}"
echo ""
echo -e "${BLUE}下一步：${NC}"
echo "1. 启动 Blockchain Service: npm run start:dev"
echo "2. 运行健康检查: ./scripts/health-check.sh"
echo "3. 运行快速测试: ./scripts/quick-test.sh"