#!/usr/bin/env bash
set -euo pipefail

########################################
# Usage / 參數
#
# $1: ENVIRONMENT (test|live)
# $2: FILE_NAME
# $3+: [MODULE] [removebackup_target] [SetID...]
#
# 說明：
# - MODULE 可省略，預設 exec_merge
# - removebackup 模組後可加 pre/post；不加則預設 both（同時刪 Pre+Post）
# - SetID filter 形式為 S11 / S21 / S31 ...（可多個）
#   未加：針對 input file 內所有 set_id 全部執行
#   有加：僅針對指定 set_id 的 world_id_lists 執行（套用至所有模組含 removebackup）
#
# 單一基礎功能 (single module):
#   generate                      : 產生合併目錄與檔案 (echo)
#   reward_before_merge           : 合併前發送獎勵 fastcgi (echo)
#   dbcheck                       : 檢查 DB 連線 (echo)
#   prebackup                     : 合併前 DB 備份 (echo)
#   restore_prebackup             : 由 Pre 備份 DB 還原回原 DB (echo)
#   merge                         : 執行合併 (echo)
#   logcheck                      : 初次檢查 log (echo)
#   waitdone                      : 等待各 List 出現 ,Done (echo)
#   postbackup                    : 合併後 DB 備份 (echo)
#   check                         : merge_check 比對差異 (echo)
#   update_db_setup               : 更新 db_setup_for_merge (echo, 僅 live 產生指令)
#   move_old_server_directory     : 移動舊 server 目錄 (echo)
#   remove_source_wids_on_hosts   : hosts 移除 world (echo, 僅 live)
#   removebackup                  : 刪除 Pre/Post 備份 DB (echo，可加 pre/post)
#   merge_rollback_partial_worlds : AccountDB 部分回滾 (echo)
#
# 群組功能 (group module) 與組成:
#   exec_merge  =
#       generate
#       + dbcheck
#       + prebackup
#       + merge
#       + logcheck
#       + waitdone
#       + postbackup
#       + check
#
#   before_merge =
#       generate
#       + reward_before_merge
#
#   after_merge =
#       update_db_setup
#       + move_old_server_directory
#       + remove_source_wids_on_hosts
#
#   end_merge =
#       removebackup
########################################

print_usage() {
  cat <<EOF >&2
Usage:
  $0 <ENVIRONMENT: test|live> <FILE_NAME> [MODULE] [removebackup_target] [SetID...]

ENVIRONMENT:
  test | live

FILE_NAME:
  input file = \$HOME/bin/merge_n1/<ENVIRONMENT>/<FILE_NAME>

單一基礎功能 (single module):
  generate                      : 產生合併目錄與檔案 (echo)
  reward_before_merge           : 合併前發送獎勵 fastcgi (echo)
  dbcheck                       : 檢查 DB 連線 (echo)
  prebackup                     : 合併前 DB 備份 (echo)
  restore_prebackup             : 由 Pre 備份 DB 還原回原 DB (echo)
  merge                         : 執行合併 (echo)
  logcheck                      : 初次檢查 log (echo)
  waitdone                      : 等待各 List 出現 ,Done (echo)
  postbackup                    : 合併後 DB 備份 (echo)
  check                         : merge_check 比對差異 (echo)
  update_db_setup               : 更新 db_setup_for_merge (echo, 僅 live 產生指令)
  move_old_server_directory     : 移動舊 server 目錄 (echo)
  remove_source_wids_on_hosts   : hosts 移除 world (echo, 僅 live)
  removebackup                  : 刪除 Pre/Post 備份 DB (echo，可加 pre/post)
  merge_rollback_partial_worlds : AccountDB 部分回滾 (echo)

群組功能 (group module) 與組成:
  exec_merge  =
      generate
      + dbcheck
      + prebackup
      + merge
      + logcheck
      + waitdone
      + postbackup
      + check

  before_merge =
      generate
      + reward_before_merge

  after_merge =
      update_db_setup
      + move_old_server_directory
      + remove_source_wids_on_hosts

  end_merge =
      removebackup

removebackup_target（僅 removebackup 模組可用，可省略）：
  pre  : 只刪 _<FILE_NAME>_Pre
  post : 只刪 _<FILE_NAME>_Post
  (未提供) : 同時刪 _<FILE_NAME>_Pre & _<FILE_NAME>_Post

SetID 過濾參數（可省略，可多個）：
  形式：S11 / S21 / S31 ...
  - 未提供：處理 input file 內全部 set_id
  - 有提供：只處理指定 set_id 的 world_id_lists（套用至所有模組，含 removebackup）
EOF
}

# Help（可不帶任何其他參數）
if [ "${1-}" = "-h" ] || [ "${1-}" = "--help" ] || [ "${1-}" = "help" ]; then
  print_usage
  exit 0
fi

KNOWN_MODULES=(
  exec_merge before_merge after_merge end_merge
  generate reward_before_merge dbcheck prebackup restore_prebackup merge logcheck waitdone postbackup check
  update_db_setup move_old_server_directory remove_source_wids_on_hosts removebackup
  merge_rollback_partial_worlds
)

is_known_module() {
  local m="$1"
  local x
  for x in "${KNOWN_MODULES[@]}"; do
    if [ "$x" = "$m" ]; then
      return 0
    fi
  done
  return 1
}

if [ "$#" -lt 2 ] || [ "$#" -gt 20 ]; then
  print_usage
  exit 1
fi

ENVIRONMENT="$1"
FILE_NAME="$2"
shift 2

case "$ENVIRONMENT" in
  test|live) ;;
  *)
    echo "Error: ENVIRONMENT must be 'test' or 'live', got '$ENVIRONMENT'." >&2
    print_usage
    exit 1
    ;;
esac

MODULE="exec_merge"
BACKUP_TARGET="both"
declare -a SET_FILTER_TOKENS=()

if [ "$#" -ge 1 ] && is_known_module "$1"; then
  MODULE="$1"
  shift 1
fi

while [ "$#" -gt 0 ]; do
  arg="$1"
  shift 1

  if [ "$MODULE" = "removebackup" ] && { [ "$arg" = "pre" ] || [ "$arg" = "post" ]; }; then
    BACKUP_TARGET="$arg"
    continue
  fi

  if [[ "$arg" =~ ^S[0-9]+$ ]]; then
    SET_FILTER_TOKENS+=( "$arg" )
    continue
  fi

  echo "Error: unknown argument '$arg'." >&2
  print_usage
  exit 1
done

declare -A SELECTED_SETIDS=()
if [ "${#SET_FILTER_TOKENS[@]}" -gt 0 ]; then
  for t in "${SET_FILTER_TOKENS[@]}"; do
    sid="${t#S}"
    SELECTED_SETIDS["$sid"]=1
  done
fi

INPUT_FILE="$HOME/bin/merge_n1/${ENVIRONMENT}/${FILE_NAME}"
if [ ! -f "$INPUT_FILE" ]; then
  echo "Error: input file not found: $INPUT_FILE" >&2
  exit 1
fi
if [ ! -s "$INPUT_FILE" ]; then
  echo "Error: input file is empty: $INPUT_FILE" >&2
  exit 1
fi

START_TS=$(date +%s)
START_TIME=$(date '+%Y-%m-%d %H:%M:%S')
echo "== run_merge_n1_echo start at ${START_TIME} (ENV=${ENVIRONMENT}, FILE_NAME=${FILE_NAME}, MODULE=${MODULE}) =="

declare -A list_worlds
declare -A list_setid
declare -A set_lists
declare -A set_first_list
declare -a all_lists

LINE_NO=0
while IFS= read -r line || [ -n "$line" ]; do
  LINE_NO=$((LINE_NO + 1))

  line_trimmed="${line#"${line%%[![:space:]]*}"}"
  line_trimmed="${line_trimmed%"${line_trimmed##*[![:space:]]}"}"

  if [ -z "$line_trimmed" ] || [[ "$line_trimmed" =~ ^# ]]; then
    continue
  fi

  read -r -a WORLDS <<< "$line_trimmed"
  if [ "${#WORLDS[@]}" -lt 1 ]; then
    echo "Error: line $LINE_NO has no WorldID: '$line'" >&2
    exit 1
  fi

  LINE_SET_ID=""
  for wid in "${WORLDS[@]}"; do
    if ! [[ "$wid" =~ ^[0-9]+$ ]]; then
      echo "Error: line $LINE_NO has non-numeric WorldID '$wid'." >&2
      exit 1
    fi
    wid_set_id=$(( wid / 100 ))
    if [ -z "$LINE_SET_ID" ]; then
      LINE_SET_ID="$wid_set_id"
    else
      if (( wid_set_id != LINE_SET_ID )); then
        echo "Error: line $LINE_NO has mixed SET_ID: expected $LINE_SET_ID, got $wid_set_id (WorldID: $wid)." >&2
        exit 1
      fi
    fi
  done

  if [ "${#SET_FILTER_TOKENS[@]}" -gt 0 ] && [[ -z "${SELECTED_SETIDS[$LINE_SET_ID]+x}" ]]; then
    continue
  fi

  LID="L${LINE_NO}"
  list_worlds["$LID"]="$line_trimmed"
  list_setid["$LID"]="$LINE_SET_ID"
  all_lists+=( "$LID" )

  if [[ -z "${set_lists[$LINE_SET_ID]+x}" ]]; then
    set_lists["$LINE_SET_ID"]="$LID"
    set_first_list["$LINE_SET_ID"]="$LID"
  else
    set_lists["$LINE_SET_ID"]+=" $LID"
  fi
done < "$INPUT_FILE"

if [ "${#all_lists[@]}" -eq 0 ]; then
  echo "Error: no valid WorldID lists found in file (after SetID filter, if any): $INPUT_FILE" >&2
  exit 1
fi

echo "Input file      : $INPUT_FILE"
echo "Environment     : $ENVIRONMENT"
echo "FILE_NAME       : $FILE_NAME"
if [ "${#SET_FILTER_TOKENS[@]}" -gt 0 ]; then
  echo "SetID filter    : ${SET_FILTER_TOKENS[*]}"
else
  echo "SetID filter    : (none, all set_id)"
fi
echo "Lists by SET_ID :"
mapfile -t SORTED_SET_IDS < <(printf '%s\n' "${!set_lists[@]}" | sort -n)
for set_id in "${SORTED_SET_IDS[@]}"; do
  echo "  SET_ID $set_id => ${set_lists[$set_id]}"
done
echo

FIRST_LID_GLOBAL="${all_lists[0]}"
FIRST_SETID_GLOBAL="${list_setid[$FIRST_LID_GLOBAL]}"

get_area_machine_and_generator() {
  local env="$1"
  local sid="$2"
  local area=""
  local machine=""
  local generator=""

  if (( sid >= 10 && sid < 20 )); then
    area="asia"
  elif (( sid >= 20 && sid < 30 )); then
    area="us"
  elif (( sid >= 30 && sid < 40 )); then
    area="eu"
  else
    echo "Error: unsupported SET_ID '$sid' (only 10~39 supported)." >&2
    return 1
  fi

  if [ "$env" = "live" ]; then
    machine="MS${sid}"
    generator="generate_merge_preparation"
  else
    case "$area" in
      asia) machine="MERGE_ASIA" ;;
      us)   machine="MERGE_US" ;;
      eu)   machine="MERGE_EU" ;;
    esac
    generator="generate_merge_preparation_test"
  fi

  printf '%s %s %s\n' "$area" "$machine" "$generator"
}

run_generate() {
  echo ">>> MODULE: generate (echo create merge directories and files)"
  for set_id in "${SORTED_SET_IDS[@]}"; do
    read AREA MACHINE GENERATOR < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    read -r -a LIDS <<< "${set_lists[$set_id]}"

    echo "==== [GENERATE] SET_ID ${set_id} / AREA=${AREA} / MACHINE=${MACHINE} / GEN=${GENERATOR} ===="
    for lid in "${LIDS[@]}"; do
      WORLD_ID_LISTS="${list_worlds[$lid]}"
      echo "# [${GENERATOR}] $MACHINE $DIRECTORY_NAME :: $WORLD_ID_LISTS"
      echo ssh "$MACHINE" "\$HOME/bin/${GENERATOR} ${DIRECTORY_NAME} ${WORLD_ID_LISTS}"
    done
  done
  echo
}

run_reward_before_merge() {
  echo ">>> MODULE: reward_before_merge (echo send_fastcgi_all_once_n1_by_world_reward_merge)"
  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    echo "# [reward_before_merge] LID=${lid} :: ${WORLD_ID_LISTS}"
    echo "\$HOME/bin/send_fastcgi_all_once_n1_by_world_reward_merge ${ENVIRONMENT} ${WORLD_ID_LISTS}"
  done
  echo
}

run_db_connect_verify() {
  echo ">>> MODULE: dbcheck (echo merge_db_connect_verify)"
  for set_id in "${SORTED_SET_IDS[@]}"; do
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    read -r -a LIDS <<< "${set_lists[$set_id]}"

    for lid in "${LIDS[@]}"; do
      WORLD_ID_LISTS="${list_worlds[$lid]}"
      WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
      REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"
      REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"

      echo "# [merge_db_connect_verify] $MACHINE $REMOTE_MERGE_INI"
      echo ssh "$MACHINE" "\$HOME/bin/merge_db_connect_verify \"${REMOTE_MERGE_INI}\""
    done
  done
  echo
}

run_pre_backup() {
  echo ">>> MODULE: prebackup (echo Pre DB backup)"

  read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$FIRST_SETID_GLOBAL")
  DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
  WORLD_ID_LISTS_FIRST_GLOBAL="${list_worlds[$FIRST_LID_GLOBAL]}"
  WORLD_ID_LISTS_STR_FIRST_GLOBAL=${WORLD_ID_LISTS_FIRST_GLOBAL// /_}
  REMOTE_BASE_DIR_FIRST_GLOBAL="\$HOME/servers${FIRST_SETID_GLOBAL}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_GLOBAL}"
  REMOTE_MERGE_INI_FIRST_GLOBAL="${REMOTE_BASE_DIR_FIRST_GLOBAL}/merge_setup.ini"

  echo "==== [PRE BACKUP] Global First List (AccountDB) ===="
  echo "# [Pre-AccountDB] $FIRST_LID_GLOBAL ($WORLD_ID_LISTS_FIRST_GLOBAL) on $MACHINE"
  echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Pre AccountDB"
  echo

  echo "==== [PRE BACKUP] Per SET_ID First List (GameDB) ===="
  for set_id in "${SORTED_SET_IDS[@]}"; do
    FIRST_LID="${set_first_list[$set_id]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS_FIRST_SET="${list_worlds[$FIRST_LID]}"
    WORLD_ID_LISTS_STR_FIRST_SET=${WORLD_ID_LISTS_FIRST_SET// /_}
    REMOTE_BASE_DIR_FIRST_SET="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_SET}"
    REMOTE_MERGE_INI_FIRST_SET="${REMOTE_BASE_DIR_FIRST_SET}/merge_setup.ini"

    echo "# [Pre-GameDB] SET_ID=${set_id} first list: $FIRST_LID ($WORLD_ID_LISTS_FIRST_SET) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Pre GameDB"
  done
  echo

  echo "==== [PRE BACKUP] Each List (WorldDB) ===="
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"

    echo "# [Pre-WorldDB] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre WorldDB"
  done
  echo
}

run_restore_prebackup() {
  echo ">>> MODULE: restore_prebackup (echo restore DB from Pre backup)"

  read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$FIRST_SETID_GLOBAL")
  DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
  WORLD_ID_LISTS_FIRST_GLOBAL="${list_worlds[$FIRST_LID_GLOBAL]}"
  WORLD_ID_LISTS_STR_FIRST_GLOBAL=${WORLD_ID_LISTS_FIRST_GLOBAL// /_}
  REMOTE_BASE_DIR_FIRST_GLOBAL="\$HOME/servers${FIRST_SETID_GLOBAL}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_GLOBAL}"
  REMOTE_MERGE_INI_FIRST_GLOBAL="${REMOTE_BASE_DIR_FIRST_GLOBAL}/merge_setup.ini"

  echo "==== [RESTORE PRE] Global First List (AccountDB) ===="
  echo "# [Restore-AccountDB] $FIRST_LID_GLOBAL ($WORLD_ID_LISTS_FIRST_GLOBAL) on $MACHINE"
  echo ssh "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Pre AccountDB"
  echo

  echo "==== [RESTORE PRE] Per SET_ID First List (GameDB) ===="
  for set_id in "${SORTED_SET_IDS[@]}"; do
    FIRST_LID="${set_first_list[$set_id]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS_FIRST_SET="${list_worlds[$FIRST_LID]}"
    WORLD_ID_LISTS_STR_FIRST_SET=${WORLD_ID_LISTS_FIRST_SET// /_}
    REMOTE_BASE_DIR_FIRST_SET="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_SET}"
    REMOTE_MERGE_INI_FIRST_SET="${REMOTE_BASE_DIR_FIRST_SET}/merge_setup.ini"

    echo "# [Restore-GameDB] SET_ID=${set_id} first list: $FIRST_LID ($WORLD_ID_LISTS_FIRST_SET) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Pre GameDB"
  done
  echo

  echo "==== [RESTORE PRE] Each List (WorldDB) ===="
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"

    echo "# [Restore-WorldDB] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre WorldDB"
  done
  echo
}

run_merge_exec() {
  echo ">>> MODULE: merge (echo merge_exec)"
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"

    echo "# [merge_exec] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "cd ${REMOTE_BASE_DIR}; nohup ./merge_exec &"
  done
  echo
}

run_log_check() {
  echo ">>> MODULE: logcheck (echo first grep on Test.log*)"
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"

    echo "# [log check] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "cd ${REMOTE_BASE_DIR}; grep -E ',WolrdDB|,WorldDB|,GameDB|,AccountDB|,Done| autosend_info| redenvelope_info' Test.log* || echo 'No match in Test.log*'"
  done
  echo
}

run_wait_done() {
  echo ">>> MODULE: waitdone (echo wait commands for ',Done')"
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"

    echo "# [wait-check] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "cd ${REMOTE_BASE_DIR}; grep -q ',Done' Test.log* 2>/dev/null"
  done
  echo
}

run_post_backup() {
  echo ">>> MODULE: postbackup (echo Post DB backup)"

  read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$FIRST_SETID_GLOBAL")
  DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
  WORLD_ID_LISTS_FIRST_GLOBAL="${list_worlds[$FIRST_LID_GLOBAL]}"
  WORLD_ID_LISTS_STR_FIRST_GLOBAL=${WORLD_ID_LISTS_FIRST_GLOBAL// /_}
  REMOTE_BASE_DIR_FIRST_GLOBAL="\$HOME/servers${FIRST_SETID_GLOBAL}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_GLOBAL}"
  REMOTE_MERGE_INI_FIRST_GLOBAL="${REMOTE_BASE_DIR_FIRST_GLOBAL}/merge_setup.ini"

  echo "==== [POST BACKUP] Global First List (AccountDB) ===="
  echo "# [Post-AccountDB] First global list: $FIRST_LID_GLOBAL ($WORLD_ID_LISTS_FIRST_GLOBAL) on $MACHINE"
  echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Post AccountDB"
  echo

  echo "==== [POST BACKUP] Per SET_ID First List (GameDB) ===="
  for set_id in "${SORTED_SET_IDS[@]}"; do
    FIRST_LID="${set_first_list[$set_id]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS_FIRST_SET="${list_worlds[$FIRST_LID]}"
    WORLD_ID_LISTS_STR_FIRST_SET=${WORLD_ID_LISTS_FIRST_SET// /_}
    REMOTE_BASE_DIR_FIRST_SET="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_SET}"
    REMOTE_MERGE_INI_FIRST_SET="${REMOTE_BASE_DIR_FIRST_SET}/merge_setup.ini"

    echo "# [Post-GameDB] SET_ID=${set_id} first list: $FIRST_LID ($WORLD_ID_LISTS_FIRST_SET) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Post GameDB"
  done
  echo

  echo "==== [POST BACKUP] Each List (WorldDB) ===="
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"

    echo "# [Post-WorldDB] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE"
    echo ssh "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Post WorldDB"
  done
  echo
}

run_check_diff() {
  echo ">>> MODULE: check (echo merge_check)"
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    WORLD_ID_LISTS_STR=${WORLD_ID_LISTS// /_}
    REMOTE_BASE_DIR="\$HOME/servers${set_id}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"

    echo "# [merge_check] LID=${lid}, SET_ID=${set_id} ($WORLD_ID_LISTS) on $MACHINE (ENV=${ENVIRONMENT})"
    echo ssh "$MACHINE" "\$HOME/bin/merge_check \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre _${FILE_NAME}_Post $ENVIRONMENT"
  done
  echo
}

run_update_db_setup() {
  echo ">>> MODULE: update_db_setup (echo update_db_setup_for_merge, only ENV=live)"
  if [ "$ENVIRONMENT" != "live" ]; then
    echo "# update_db_setup_for_merge is only allowed in live environment. Current ENV=${ENVIRONMENT}."
    echo "# (No commands generated.)"
    echo
    return
  fi

  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    echo "# [update_db_setup_for_merge] LID=${lid} :: ${WORLD_ID_LISTS}"
    echo "\$HOME/bin/update_db_setup_for_merge ${WORLD_ID_LISTS}"
  done
  echo
}

run_move_old_server_directory() {
  echo ">>> MODULE: move_old_server_directory (echo move_source_wids_directory_after_merge)"
  for lid in "${all_lists[@]}"; do
    set_id="${list_setid[$lid]}"
    read AREA _ _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
    DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"
    WORLD_ID_LISTS="${list_worlds[$lid]}"

    echo "# [move_source_wids_directory_after_merge] LID=${lid}, SET_ID=${set_id}, ENV=${ENVIRONMENT}, DIR=${DIRECTORY_NAME} :: ${WORLD_ID_LISTS}"
    echo "\$HOME/bin/move_source_wids_directory_after_merge ${ENVIRONMENT} ${DIRECTORY_NAME} ${WORLD_ID_LISTS}"
  done
  echo
}

run_remove_source_wids_on_hosts() {
  echo ">>> MODULE: remove_source_wids_on_hosts (echo backup hosts and call remove_source_wids_on_hosts_after_merge, ENV=live only)"

  HOSTS_FILE="\$HOME/bin/hosts"
  HOSTS_BAK="\$HOME/bin/hosts.old"

  if [ "$ENVIRONMENT" != "live" ]; then
    echo "# remove_source_wids_on_hosts is only allowed in live environment. Current ENV=${ENVIRONMENT}."
    echo "# (No commands generated.)"
    echo
    return
  fi

  echo "==== [remove_source_wids_on_hosts] ENV=live ===="
  echo "# Backup hosts:"
  echo "cp -arf ${HOSTS_FILE} ${HOSTS_BAK}"

  echo "# Call remove_source_wids_on_hosts_after_merge for each WORLD_ID_LISTS:"
  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    echo "# [remove_source_wids_on_hosts_after_merge] LID=${lid} :: ${WORLD_ID_LISTS}"
    echo "\$HOME/bin/remove_source_wids_on_hosts_after_merge ${WORLD_ID_LISTS}"
  done
  echo
}

run_remove_backup() {
  echo ">>> MODULE: removebackup (echo drop backup databases; target=${BACKUP_TARGET})"

  drop_cmds=""
  case "$BACKUP_TARGET" in
    pre)
      drop_cmds="\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Pre"
      ;;
    post)
      drop_cmds="\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Post"
      ;;
    both)
      drop_cmds="\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Pre; \
\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Post"
      ;;
    *)
      echo "Error: invalid removebackup target '${BACKUP_TARGET}' (expected pre|post|both)." >&2
      print_usage
      exit 1
      ;;
  esac

  if [ "$ENVIRONMENT" = "test" ]; then
    echo "==== [removebackup] ENV=test ===="

    declare -a MACHINES=()
    if [ "${#SET_FILTER_TOKENS[@]}" -eq 0 ]; then
      MACHINES=(MERGE_COMMON MERGE_ASIA MERGE_US MERGE_EU)
    else
      MACHINES=(MERGE_COMMON)
      for set_id in "${SORTED_SET_IDS[@]}"; do
        read AREA _ _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
        case "$AREA" in
          asia) MACHINES+=(MERGE_ASIA) ;;
          us)   MACHINES+=(MERGE_US) ;;
          eu)   MACHINES+=(MERGE_EU) ;;
        esac
      done
    fi

    declare -A SEEN=()
    for m in "${MACHINES[@]}"; do
      if [[ -z "${SEEN[$m]+x}" ]]; then
        SEEN["$m"]=1
        echo "# [removebackup] test env on ${m} for FILE_NAME=${FILE_NAME} target=${BACKUP_TARGET}"
        echo ssh "$m" "$drop_cmds"
      fi
    done
  else
    echo "==== [removebackup] ENV=live ===="

    if [ "${#SET_FILTER_TOKENS[@]}" -eq 0 ]; then
      echo "# [removebackup] live env via sendscript alldb for FILE_NAME=${FILE_NAME} target=${BACKUP_TARGET}"
      echo sendscript alldb "$drop_cmds"
      echo
      return
    fi

    declare -A LIVE_SEEN=()
    for set_id in "${SORTED_SET_IDS[@]}"; do
      read _ MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$set_id")
      if [[ -z "${LIVE_SEEN[$MACHINE]+x}" ]]; then
        LIVE_SEEN["$MACHINE"]=1
        echo "# [removebackup] live env on ${MACHINE} for SET_ID=${set_id}, FILE_NAME=${FILE_NAME} target=${BACKUP_TARGET}"
        echo ssh "$MACHINE" "$drop_cmds"
      fi
    done
  fi

  echo
}

run_merge_rollback_partial_worlds() {
  echo ">>> MODULE: merge_rollback_partial_worlds (echo rollback AccountDB partial worlds)"

  read AREA MACHINE _ < <(get_area_machine_and_generator "$ENVIRONMENT" "$FIRST_SETID_GLOBAL")
  DIRECTORY_NAME="merge_${AREA}_${FILE_NAME}_${ENVIRONMENT}"

  WORLD_ID_LISTS_FIRST_GLOBAL="${list_worlds[$FIRST_LID_GLOBAL]}"
  WORLD_ID_LISTS_STR_FIRST_GLOBAL=${WORLD_ID_LISTS_FIRST_GLOBAL// /_}
  REMOTE_BASE_DIR_FIRST_GLOBAL="\$HOME/servers${FIRST_SETID_GLOBAL}/${DIRECTORY_NAME}/${WORLD_ID_LISTS_STR_FIRST_GLOBAL}"
  REMOTE_MERGE_INI_FIRST_GLOBAL="${REMOTE_BASE_DIR_FIRST_GLOBAL}/merge_setup.ini"

  echo "# [merge_rollback_partial_worlds] First global list: $FIRST_LID_GLOBAL ($WORLD_ID_LISTS_FIRST_GLOBAL) on $MACHINE"
  echo "# [merge_rollback_partial_worlds] INI: $REMOTE_MERGE_INI_FIRST_GLOBAL"
  echo ssh "$MACHINE" "\$HOME/bin/merge_rollback_partial_worlds \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\""
  echo
}

case "$MODULE" in
  exec_merge)
    run_generate
    run_db_connect_verify
    run_pre_backup
    run_merge_exec
    run_log_check
    run_wait_done
    run_post_backup
    run_check_diff
    ;;
  before_merge)
    run_generate
    run_reward_before_merge
    ;;
  after_merge)
    run_update_db_setup
    run_move_old_server_directory
    run_remove_source_wids_on_hosts
    ;;
  end_merge)
    run_remove_backup
    ;;

  generate) run_generate ;;
  reward_before_merge) run_reward_before_merge ;;
  dbcheck) run_db_connect_verify ;;
  prebackup) run_pre_backup ;;
  restore_prebackup) run_restore_prebackup ;;
  merge) run_merge_exec ;;
  logcheck) run_log_check ;;
  waitdone) run_wait_done ;;
  postbackup) run_post_backup ;;
  check) run_check_diff ;;
  update_db_setup) run_update_db_setup ;;
  move_old_server_directory) run_move_old_server_directory ;;
  remove_source_wids_on_hosts) run_remove_source_wids_on_hosts ;;
  removebackup) run_remove_backup ;;
  merge_rollback_partial_worlds) run_merge_rollback_partial_worlds ;;

  *)
    echo "Error: unknown MODULE '${MODULE}'." >&2
    print_usage
    exit 1
    ;;
esac

END_TS=$(date +%s)
END_TIME=$(date '+%Y-%m-%d %H:%M:%S')
ELAPSED=$((END_TS - START_TS))
ELAPSED_H=$((ELAPSED / 3600))
ELAPSED_M=$(((ELAPSED % 3600) / 60))
ELAPSED_S=$((ELAPSED % 60))

echo "== run_merge_n1_echo finished (MODULE=${MODULE}) =="
echo "Start time : ${START_TIME}"
echo "End time   : ${END_TIME}"
printf 'Elapsed    : %02d 小時 %02d 分 %02d 秒\n' "$ELAPSED_H" "$ELAPSED_M" "$ELAPSED_S"
echo "ENV=${ENVIRONMENT}, FILE_NAME=${FILE_NAME}"
echo "Dry-run (echo) 完成，以上為所有預計執行的指令。"

