#!/bin/bash
  

  # ============================================================================
  # Mock Data Script for SearchEngine
  # ============================================================================
  # 
  # 功能说明：
  # 本脚本用于构造测试数据，包含两部分：
  # 1. tenant_id=1: 自动生成的mock数据（使用 generate_test_data.py）
  # 2. tenant_id=2: 从CSV文件导入的数据（使用 import_tenant2_csv.py）
  #
  # 数据说明：
  # - 所有数据源配置（数据库地址、CSV路径、字段映射等）都写死在脚本中
  # - 这是外部系统构造测试数据，不需要配置化
  # - 脚本会自动计算起始ID，避免主键冲突
  #
  # 使用方式：
  #   ./scripts/mock_data.sh
  #
  # ============================================================================

  
  cd "$(dirname "$0")/.."
  source /home/tw/miniconda3/etc/profile.d/conda.sh
  conda activate searchengine
  
  GREEN='\033[0;32m'
  YELLOW='\033[1;33m'
  RED='\033[0;31m'
  NC='\033[0m'
  
  echo -e "${GREEN}========================================${NC}"
  echo -e "${GREEN}Mock Data Script${NC}"
  echo -e "${GREEN}========================================${NC}"
  
  # Load config from .env file if it exists
  if [ -f .env ]; then
      set -a
      source .env
      set +a
  fi
  

  # ============================================================================
  # 写死的配置参数（不需要配置化，这是测试数据构造脚本）
  # ============================================================================

  # Tenant 1: Mock数据配置
  TENANT1_NUM_SPUS=1000  # 生成的SPU数量
  
  # Tenant 2: CSV数据配置
  TENANT2_CSV_FILE="data/customer1/goods_with_pic.5years_congku.csv.shuf.1w"  # CSV文件路径
  
  # 数据库配置（写死，不需要配置化）
  DB_HOST="120.79.247.228"
  DB_PORT="3316"
  DB_DATABASE="saas"
  DB_USERNAME="saas"
  DB_PASSWORD="P89cZHS5d7dFyc9R"

  
  echo -e "\n${YELLOW}Configuration:${NC}"

  echo "  Tenant 1 (Mock): $TENANT1_NUM_SPUS SPUs"
  echo "  Tenant 2 (CSV): $TENANT2_CSV_FILE"

  echo "  MySQL: $DB_HOST:$DB_PORT/$DB_DATABASE"

  
  # Validate CSV file exists
  if [ ! -f "$TENANT2_CSV_FILE" ]; then
      echo -e "${RED}ERROR: CSV file not found: $TENANT2_CSV_FILE${NC}"
      echo "请确保CSV文件存在于: $TENANT2_CSV_FILE"
      exit 1

  fi
  

  # ============================================================================
  # Part 1: 生成并导入 tenant_id=1 的Mock数据
  # ============================================================================
  
  echo -e "\n${YELLOW}========================================${NC}"
  echo -e "${YELLOW}Part 1/2: 生成并导入 tenant_id=1 的Mock数据${NC}"
  echo -e "${YELLOW}========================================${NC}"
  
  TENANT1_SQL_FILE="test_data_tenant1.sql"
  
  echo -e "\n${YELLOW}Step 1.1: 生成Mock测试数据${NC}"
  python scripts/generate_test_data.py \
      --num-spus $TENANT1_NUM_SPUS \
      --tenant-id "1" \
      --output "$TENANT1_SQL_FILE" \
      --db-host "$DB_HOST" \
      --db-port "$DB_PORT" \
      --db-database "$DB_DATABASE" \
      --db-username "$DB_USERNAME" \
      --db-password "$DB_PASSWORD"
  
  if [ $? -ne 0 ]; then
      echo -e "${RED}✗ 生成tenant_id=1数据失败${NC}"
      exit 1

  fi

  echo -e "${GREEN}✓ 数据已生成: $TENANT1_SQL_FILE${NC}"
  
  echo -e "\n${YELLOW}Step 1.2: 导入tenant_id=1数据到MySQL${NC}"
  python scripts/import_test_data.py \
      --db-host "$DB_HOST" \
      --db-port "$DB_PORT" \
      --db-database "$DB_DATABASE" \
      --db-username "$DB_USERNAME" \
      --db-password "$DB_PASSWORD" \
      --sql-file "$TENANT1_SQL_FILE" \
      --tenant-id "1"
  

  if [ $? -ne 0 ]; then

      echo -e "${RED}✗ 导入tenant_id=1数据失败${NC}"

      exit 1
  fi
  

  echo -e "${GREEN}✓ tenant_id=1数据已导入MySQL${NC}"

  # ============================================================================
  # Part 2: 生成并导入 tenant_id=2 的CSV数据
  # ============================================================================
  
  echo -e "\n${YELLOW}========================================${NC}"
  echo -e "${YELLOW}Part 2/2: 生成并导入 tenant_id=2 的CSV数据${NC}"
  echo -e "${YELLOW}========================================${NC}"
  

  TENANT2_SQL_FILE="tenant2_data.sql"

  
  echo -e "\n${YELLOW}Step 2.1: 从CSV生成数据${NC}"
  python scripts/import_tenant2_csv.py \
      --csv-file "$TENANT2_CSV_FILE" \
      --tenant-id "2" \
      --output "$TENANT2_SQL_FILE" \
      --db-host "$DB_HOST" \
      --db-port "$DB_PORT" \
      --db-database "$DB_DATABASE" \
      --db-username "$DB_USERNAME" \
      --db-password "$DB_PASSWORD"
  
  if [ $? -ne 0 ]; then
      echo -e "${RED}✗ 生成tenant_id=2数据失败${NC}"

      exit 1
  fi
  

  echo -e "${GREEN}✓ 数据已生成: $TENANT2_SQL_FILE${NC}"
  
  echo -e "\n${YELLOW}Step 2.2: 导入tenant_id=2数据到MySQL${NC}"

  python scripts/import_test_data.py \
      --db-host "$DB_HOST" \
      --db-port "$DB_PORT" \
      --db-database "$DB_DATABASE" \
      --db-username "$DB_USERNAME" \
      --db-password "$DB_PASSWORD" \

      --sql-file "$TENANT2_SQL_FILE" \
      --tenant-id "2"

  
  if [ $? -ne 0 ]; then

      echo -e "${RED}✗ 导入tenant_id=2数据失败${NC}"

      exit 1
  fi
  

  echo -e "${GREEN}✓ tenant_id=2数据已导入MySQL${NC}"
  
  # ============================================================================
  # 完成
  # ============================================================================

  
  echo -e "\n${GREEN}========================================${NC}"

  echo -e "${GREEN}数据导入完成!${NC}"

  echo -e "${GREEN}========================================${NC}"
  echo ""
  echo -e "下一步:"

  echo -e "  ${YELLOW}./scripts/ingest.sh 1 true${NC}  - 从MySQL灌入tenant_id=1数据到ES"
  echo -e "  ${YELLOW}./scripts/ingest.sh 2 true${NC}  - 从MySQL灌入tenant_id=2数据到ES"

  echo ""