#!/bin/bash
#BASE_DIR=$(cd $(dirname $0) && pwd)
BASE_DIR="$(dirname "$(pwd)")"
# Check if required tools are installed
if ! command -v java &> /dev/null; then
  echo "Error: Java is not installed."
  exit 1
fi

# Define common variables
E2E_MOCKDATA_GENERATOR_PATH="$BASE_DIR/e2e-mockdata-generator"
CONFIG_PATH="$BASE_DIR/config"
COLLECTION="tsg-olap-e2e-test-collection.json"
ENVIRONMENT="$BASE_DIR/environment.json"
DATA_CENTER="tsg_olap_dc_a" # Default Data Center 
OTHER_VAR=""

# data_center_name:kafka_server
declare -A KAFKA_SERVERS=(
  ["tsg_olap_dc_a"]="192.168.44.11:9092"
  ["tsg_olap_dc_b"]="192.168.44.11:9092"
)

# data_center_name:hos_endpoint
declare -A HOS_ENDPOINTS=(
  ["tsg_olap_dc_a"]="192.168.44.11"
  ["tsg_olap_dc_b"]="192.168.44.11"
)

log_message() {
  echo "$(date '+%Y-%m-%d %H:%M:%S') - $1"
}

generate_logs() {
  log_message "Generating logs for $1 $2 ..."
  local data_center=$1
  local bootstrap_server=$2
  cd $E2E_MOCKDATA_GENERATOR_PATH || { echo "Error: Cannot change directory to $E2E_MOCKDATA_GENERATOR_PATH"; exit 1; }
  declare -A LOGS=(
    ["SESSION-RECORD"]="session_record.dat"
    ["VOIP-RECORD"]="voip_record.dat"
    ["PROXY-EVENT"]="proxy_event.dat"
  )
    for topic in "${!LOGS[@]}"; do
      # Replace the "data_center" value in the file
      sed -i "s/\"data_center\":\"[^\"]*\"/\"data_center\":\"$data_center\"/g" ./datasets/logs/${LOGS[$topic]}
      java -cp e2e-mockdata-generator.jar com.geedgenetworks.LogGenerator --bootstrap_server $bootstrap_server   --topic $topic -f ./datasets/logs/${LOGS[$topic]}
    done
}

generate_metrics() {
  log_message "Generating metrics for $1 $2..."
  local data_center=$1
  local bootstrap_server=$2
  cd $E2E_MOCKDATA_GENERATOR_PATH || { echo "Error: Cannot change directory to $E2E_MOCKDATA_GENERATOR_PATH"; exit 1; }
  declare -A METRICS=(
    ["NETWORK-TRAFFIC-METRIC"]="network_traffic_metric.dat"
    ["POLICY-RULE-METRIC"]="policy_rule_metric.dat"
    ["OBJECT-STATISTICS-METRIC"]="object_statistics_metric.dat"
    ["STATISTICS-RULE-METRIC"]="statistics_rule_metric.dat"
  )
    for topic in "${!METRICS[@]}"; do
      # Replace the "data_center" value in the file
      sed -i "s/\"data_center\":\"[^\"]*\"/\"data_center\":\"$data_center\"/g" ./datasets/metrics/${METRICS[$topic]}
      java -cp e2e-mockdata-generator.jar com.geedgenetworks.LogGenerator --bootstrap_server $bootstrap_server --topic $topic -f ./datasets/metrics/${METRICS[$topic]}
    done
}


generate_files() {
  log_message "Generating files for $1 $2 ..."
  local data_center=$1
  local bootstrap_server=$2
  cd $E2E_MOCKDATA_GENERATOR_PATH || { echo "Error: Cannot change directory to $E2E_MOCKDATA_GENERATOR_PATH"; exit 1; }
  
 declare -A FILES=(
    ["123e4567-e89b-12d3-a456-426614174001"]="traffic_pcapng"
    ["123e4567-e89b-12d3-a456-426614174002"]="html"
    ["123e4567-e89b-12d3-a456-426614174003"]="html"
    ["123e4567-e89b-12d3-a456-426614174004"]="eml"
    ["123e4567-e89b-12d3-a456-426614174005"]="traffic_pcapng"
  )

 declare -A TROBULESHOOTING_FILES=(
    ["123e4567-e89b-12d3-a456-426614174006"]="troubleshooting_pcapng"
 )
    for uuid in "${!FILES[@]}"; do
      java -cp e2e-mockdata-generator.jar com.geedgenetworks.FileChunkGenerator --bootstrap_server $bootstrap_server --topic TRAFFIC-FILE-STREAM-RECORD -n $uuid --file_type ${FILES[$uuid]}
    done
   for uuid in "${!TROBULESHOOTING_FILES[@]}"; do
      java -cp e2e-mockdata-generator.jar com.geedgenetworks.FileChunkGenerator --bootstrap_server $bootstrap_server --topic TROUBLESHOOTING-FILE-STREAM-RECORD -n $uuid --file_type ${TROBULESHOOTING_FILES[$uuid]}
   done
}


# Define function to run diagnostic report
  # -folder logs ：对日志进行故障诊断，输出诊断明细；指定data_center。
  # -folder metrics：对Metrics进行故障诊断，输出诊断明细；指定data_center。
  # -folder files：对文件进行故障诊断，输出诊断明细；指定分中心HOS访问地址。
  # -folder logs：对日志进行故障诊断，通过表情形式输出测试结果
  # -folder files：对文件进行故障诊断，通过表情形式输出测试结果
  #  newman run ./tsg-olap-e2e-test-collection.json -n 1 --delay-request 500 -e $ENVIRONMENT --env-var "data_center=$DATA_CENTER" --ignore-redirects --folder logs -r emojitrain
  #  newman run ./tsg-olap-e2e-test-collection.json -n 1 --delay-request 500 -e $ENVIRONMENT --env-var "hos_ip=$HOS_IP" --ignore-redirects --folder files -r emojitrain
run_diagnostic() {
  log_message "Running diagnostic for $1..."
  cd $BASE_DIR || { echo "Error: Cannot change directory to $BASE_DIR"; exit 1; }
  local folder=$1
  local env_var=$2
  local verbose=""
  local emojitrain=""
  if $verbose_flag; then
    verbose="--verbose"
  fi

  if $emojitrain_flag; then
    emojitrain="-r emojitrain"
  fi

  # Split the env_var string by comma and prepare the env_var_str
  IFS=',' read -ra env_vars <<< "$env_var"
  local env_var_str=""
  for var in "${env_vars[@]}"; do
    env_var_str+="--env-var $var "
  done
  
   newman run $CONFIG_PATH/tsg-olap-e2e-test-collection.json -g $CONFIG_PATH/globals.json -n 1 -e $CONFIG_PATH/environment.json --delay-request 500 --timeout-script 10000 --timeout-request 300000 --timeout 3600000 --insecure $verbose --ignore-redirects  $env_var_str --folder $folder $emojitrain
}


clear_data() {
  log_message "Clearing test data..."
  newman run $CONFIG_PATH/tsg-olap-e2e-test-collection.json -g $CONFIG_PATH/globals.json -n 1 --delay-request 500 -e $CONFIG_PATH/environment.json --ignore-redirects --folder clear_test_data -r emojitrain
}

# Help message
show_help() {
  echo ""
  echo "Usage: $0 [options]"
  echo ""
  echo "Options:"
  echo "  -g <type>       Generate data (logs, metrics, files)"
  echo "  -d <type>       Run diagnostic report (logs, metrics, files)"
  echo "  -c              Clear test data"
  echo "  -a              Perform all operations: generate data, run diagnostics, and clear data"
  echo "  -i <key=value,...>"
  echo "                  Set environment variable (data_center, hos_ip)"
  echo "  -v              Enable verbose reporting"
  echo "  -e              Enable emojitrain reporting"
  echo "  -h              Show this help message"
  echo ""
  echo "Examples:"
  echo "  $0 -a                                      Perform all operations"
  echo "  $0 -g logs                                 Generate log data"
  echo "  $0 -d metrics -v                           Run diagnostics on metrics data with verbose reporting"
  echo "  $0 -c                                      Clear test data"
  echo "  $0 -g logs -i data_center=my_data_center   Generate log data in my_data_center"
  echo ""
}

# Initialize flags
generate_flag=false
diagnostic_flag=false
verbose_flag=false
emojitrain_flag=false
clear_flag=false
all_flag=false
generate_type=""
diagnostic_type=""


# Parse command-line arguments
while getopts ":g:d:i:acveh" opt; do
  case $opt in
    g)
      generate_type=$OPTARG
      generate_flag=true
      ;;
    d)
      diagnostic_type=$OPTARG
      diagnostic_flag=true
      ;;    
    c)
      clear_flag=true
      ;;
    a)
      all_flag=true
      ;;
    i)
      # Split the input based on whitespace, then further split by '=' to assign key-value pairs
      IFS=',' read -ra vars <<< "$OPTARG"
      for var in "${vars[@]}"; do
        # Remove leading and trailing whitespaces
        var=$(echo "$var" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')
        IFS='=' read -r key value <<< "$var"
        case $key in
          data_center) DATA_CENTER=$value ;;
          other_var) OTHER_VAR=$value ;;
          *) echo "Invalid environment variable: $key"; show_help; exit 1 ;;
        esac
      done
      ;;
    v)
      verbose_flag=true
      ;; 
    e)
      emojitrain_flag=true
      ;;
    h)
      show_help
      exit 0
      ;;
    \?)
      echo "Invalid option: -$OPTARG" >&2
      show_help
      exit 1
      ;;
  esac
done


if $generate_flag; then
 case $generate_type in
    logs) generate_logs "$DATA_CENTER" "${KAFKA_SERVERS[$DATA_CENTER]}" ;;
    metrics) generate_metrics "$DATA_CENTER" "${KAFKA_SERVERS[$DATA_CENTER]}" ;;
    files) generate_files "$DATA_CENTER" "${KAFKA_SERVERS[$DATA_CENTER]}" ;;
    *) echo "Invalid generate type"; show_help; exit 1 ;;
  esac
fi

if $diagnostic_flag; then
  case $diagnostic_type in
    logs) run_diagnostic "logs" "data_center=$DATA_CENTER" ;;
    metrics) run_diagnostic "metrics" "data_center=$DATA_CENTER" ;;
    files) run_diagnostic "files" "data_center=$DATA_CENTER,hos_ip=${HOS_ENDPOINTS[$DATA_CENTER]}" ;;
    *) echo "Invalid diagnostic type"; show_help; exit 1 ;;
  esac
fi

if $clear_flag; then
  clear_data
fi

if $all_flag; then
 
  for data_center_var in "${!KAFKA_SERVERS[@]}"; do
     generate_logs "$data_center_var" "${KAFKA_SERVERS[$data_center_var]}"
     generate_metrics "$data_center_var" "${KAFKA_SERVERS[$data_center_var]}"
     generate_files "$data_center_var" "${KAFKA_SERVERS[$data_center_var]}"
     echo "Wait 30 Seconds..."
     sleep 30
     run_diagnostic "logs" "data_center=$data_center_var"
     run_diagnostic "metrics" "data_center=$data_center_var"
     run_diagnostic "files" "data_center=$data_center_var,hos_ip=${HOS_ENDPOINTS[$data_center_var]}" 
     clear_data
 done  
fi

# Display help message if no arguments are passed
if ! $generate_flag && ! $diagnostic_flag && ! $clear_flag && ! $all_flag; then
  show_help
  exit 1
fi

log_message "E2E test execution completed."