#!/bin/bash
  
  # Mock data script for SearchEngine
  # Generates test data and imports to MySQL

  # Supports both mock data generation and CSV import

  
  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
  

  # Parse arguments
  MODE="mock"  # "mock" or "csv"
  TENANT_ID=""
  NUM_SPUS=100
  CSV_FILE=""
  START_SPU_ID=1
  
  # Parse command line arguments
  while [[ $# -gt 0 ]]; do
      case $1 in
          --mode)
              MODE="$2"
              shift 2
              ;;
          --tenant-id)
              TENANT_ID="$2"
              shift 2
              ;;
          --num-spus)
              NUM_SPUS="$2"
              shift 2
              ;;
          --csv-file)
              CSV_FILE="$2"
              shift 2
              ;;
          --start-spu-id)
              START_SPU_ID="$2"
              shift 2
              ;;
          *)
              # Legacy support: first arg as tenant_id, second as num_spus
              if [ -z "$TENANT_ID" ]; then
                  TENANT_ID="$1"
              elif [ "$NUM_SPUS" = "100" ] && [[ "$1" =~ ^[0-9]+$ ]]; then
                  NUM_SPUS="$1"
              fi
              shift
              ;;
      esac
  done
  
  # Set default tenant_id based on mode
  if [ -z "$TENANT_ID" ]; then
      if [ "$MODE" = "csv" ]; then
          TENANT_ID="2"
      else
          TENANT_ID="1"
      fi
  fi
  
  # Database configuration

  DB_HOST=${DB_HOST:-"120.79.247.228"}
  DB_PORT=${DB_PORT:-"3316"}
  DB_DATABASE=${DB_DATABASE:-"saas"}
  DB_USERNAME=${DB_USERNAME:-"saas"}
  DB_PASSWORD=${DB_PASSWORD:-"P89cZHS5d7dFyc9R"}

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

  echo "  Mode: $MODE"

  echo "  Tenant ID: $TENANT_ID"

  if [ "$MODE" = "mock" ]; then
      echo "  Number of SPUs: $NUM_SPUS"
      SQL_FILE="test_data.sql"
  else
      echo "  CSV File: $CSV_FILE"
      echo "  Start SPU ID: $START_SPU_ID"
      SQL_FILE="customer1_data.sql"
  fi

  echo "  MySQL: $DB_HOST:$DB_PORT/$DB_DATABASE"
  echo "  SQL File: $SQL_FILE"
  

  # Validate CSV mode
  if [ "$MODE" = "csv" ]; then
      if [ -z "$CSV_FILE" ]; then
          echo -e "${RED}ERROR: CSV file is required in csv mode${NC}"
          echo "Usage: $0 --mode csv --csv-file <path> [--tenant-id <id>] [--start-spu-id <id>]"
          exit 1
      fi
      if [ ! -f "$CSV_FILE" ]; then
          echo -e "${RED}ERROR: CSV file not found: $CSV_FILE${NC}"
          exit 1
      fi
  fi
  

  # Step 1: Generate test data

  if [ "$MODE" = "mock" ]; then
      echo -e "\n${YELLOW}Step 1/2: 生成Mock测试数据${NC}"
      python scripts/generate_test_data.py \
          --num-spus $NUM_SPUS \
          --tenant-id "$TENANT_ID" \
          --start-spu-id 1 \
          --start-sku-id 1 \
          --output "$SQL_FILE"
  else
      echo -e "\n${YELLOW}Step 1/2: 从CSV生成数据${NC}"
      python scripts/import_customer1_csv.py \
          --csv-file "$CSV_FILE" \
          --tenant-id "$TENANT_ID" \
          --start-spu-id $START_SPU_ID \
          --output "$SQL_FILE"
  fi

  
  if [ $? -ne 0 ]; then

      echo -e "${RED}✗ 生成数据失败${NC}"

      exit 1
  fi
  

  echo -e "${GREEN}✓ 数据已生成: $SQL_FILE${NC}"

  # Step 2: Import data to MySQL
  echo -e "\n${YELLOW}Step 2/2: 导入数据到MySQL${NC}"

  if [ -z "$DB_PASSWORD" ]; then
      echo -e "${RED}ERROR: DB_PASSWORD未设置，请检查.env文件或环境变量${NC}"
      exit 1
  fi
  
  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 "$SQL_FILE" \
      --tenant-id "$TENANT_ID"
  
  if [ $? -ne 0 ]; then

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

      exit 1
  fi
  

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

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

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

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

  echo -e "  ${YELLOW}./scripts/ingest.sh $TENANT_ID${NC}  - 从MySQL灌入数据到ES"

  echo ""