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

########################################
# Usage / 參數
#
# $1: ENVIRONMENT (test|live)
# $2: FILE_NAME
# $3+: [MODULE] [removebackup_target] [SetID...]
#
# 說明：
# - 同 run_merge_n1，但只輸出預計執行的指令（dry-run）
# - MODULE 可省略，預設 exec_merge
# - SetID filter 形式為 S11 / S21 / S31 ...（可多個）
#   未加：針對 FILE_NAME 內所有 world_id_lists 都全部產生指令
#   有加：僅針對指定 set_id 的 world_id_lists 產生指令（套用至所有模組，含 removebackup*）
# - removebackup / removebackup_wb / removebackup_wob 模組後可接 pre/post；不加則預設 both（同時刪 Pre+Post）
#
# 單一基礎功能 (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)
#   update_guild_race_data        : 更新公會競賽資料（merge_update_guild_race_data）(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；不帶第3參數，預設 backup_flag=1)
#   removebackup_wb               : 刪除 Pre/Post 備份 DB (echo，可加 pre/post；第3參數=1)
#   removebackup_wob              : 刪除 Pre/Post 備份 DB (echo，可加 pre/post；第3參數=0)
#   merge_rollback_partial_worlds : AccountDB 部分回滾 (echo，以「過濾後第一個 list」的 merge_setup.ini 作為來源)
#
# 群組功能 (group module) 與組成:
#   exec_merge  =
#       generate
#       + dbcheck
#       + prebackup
#       + merge
#       + logcheck
#       + waitdone
#       + update_guild_race_data
#       + 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
#
#   re_run_exec_merge =
#       removebackup_wob post
#       + restore_prebackup
#       + merge_rollback_partial_worlds
#       + merge
#       + logcheck
#       + waitdone
#       + update_guild_race_data
#       + postbackup
#       + check
#
#   re_run_pre_merge =
#       removebackup_wob post
#       + restore_prebackup
#       + merge_rollback_partial_worlds
#
# 注意：
# - re_run_pre_merge：固定流程，不接受 pre/post 參數；若使用者帶入 pre/post，會直接忽略（不中止）
########################################

hr() { printf '%s\n' "============================================================"; }
step() { hr; printf '[STEP] %s\n' "$1"; hr; }
info() { printf '[INFO] %s\n' "$1"; }
warn() { printf '[WARN] %s\n' "$1" >&2; }
err()  { printf '[ERROR] %s\n' "$1" >&2; }

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

說明：同 run_merge_n1，但只輸出預計執行的指令（dry-run）。

單一基礎功能 / 群組功能 / removebackup_target / SetID filter 與 run_merge_n1 相同。

範例：
  run_merge_n1_echo live myfile
  run_merge_n1_echo live myfile exec_merge S21 S31
  run_merge_n1_echo live myfile removebackup post S21
  run_merge_n1_echo live myfile removebackup_wb post S21
  run_merge_n1_echo live myfile removebackup_wob post S21
  run_merge_n1_echo live myfile re_run_exec_merge S21
  run_merge_n1_echo live myfile re_run_pre_merge S21
  run_merge_n1_echo live myfile re_run_pre_merge pre S21 S31   # pre/post 會被忽略（不中止）
EOF
}

if [ "${1-}" = "-h" ] || [ "${1-}" = "--help" ] || [ "${1-}" = "help" ]; then
  print_usage
  exit 0
fi

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

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

is_backup_module() {
  case "$1" in
    removebackup|removebackup_wb|removebackup_wob) return 0 ;;
    *) return 1 ;;
  esac
}

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

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

case "$ENVIRONMENT" in
  test|live) ;;
  *)
    err "ENVIRONMENT must be 'test' or 'live', got '$ENVIRONMENT'."
    print_usage
    exit 1
    ;;
esac

MODULE="exec_merge"
BACKUP_TARGET="both"   # both|pre|post
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 is_backup_module "$MODULE" && { [ "$arg" = "pre" ] || [ "$arg" = "post" ]; }; then
    BACKUP_TARGET="$arg"
    continue
  fi

  if [ "$MODULE" = "re_run_pre_merge" ] && { [ "$arg" = "pre" ] || [ "$arg" = "post" ]; }; then
    warn "re_run_pre_merge ignores argument '$arg' (fixed pipeline)."
    continue
  fi

  if [ "$MODULE" = "re_run_exec_merge" ] && { [ "$arg" = "pre" ] || [ "$arg" = "post" ]; }; then
    if [ "$arg" = "post" ]; then
      warn "re_run_exec_merge already includes 'removebackup_wob post'; ignore redundant 'post' argument."
      continue
    fi
    err "re_run_exec_merge does not accept 'pre'. It always uses removebackup_wob post."
    exit 1
  fi

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

  err "Unknown argument '$arg'."
  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
  err "input file not found: $INPUT_FILE"
  exit 1
fi
if [ ! -s "$INPUT_FILE" ]; then
  err "input file is empty: $INPUT_FILE"
  exit 1
fi

START_TS=$(date +%s)
START_TIME=$(date '+%Y-%m-%d %H:%M:%S')

step "run_merge_n1_echo START"
info "Start time   : ${START_TIME}"
info "ENV          : ${ENVIRONMENT}"
info "FILE_NAME    : ${FILE_NAME}"
info "MODULE       : ${MODULE}"
info "INPUT_FILE   : ${INPUT_FILE}"
if is_backup_module "$MODULE"; then
  info "BACKUP_TARGET: ${BACKUP_TARGET}"
fi
if [ "${#SET_FILTER_TOKENS[@]}" -gt 0 ]; then
  info "SetID filter : ${SET_FILTER_TOKENS[*]}"
else
  info "SetID filter : (none, all set_id)"
fi

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
    err "line $LINE_NO has no WorldID: '$line'"
    exit 1
  fi

  LINE_SET_ID=""
  for wid in "${WORLDS[@]}"; do
    if ! [[ "$wid" =~ ^[0-9]+$ ]]; then
      err "line $LINE_NO has non-numeric WorldID '$wid'."
      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
        err "line $LINE_NO has mixed SET_ID: expected $LINE_SET_ID, got $wid_set_id (WorldID: $wid)."
        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
  err "no valid WorldID lists found in file (after SetID filter, if any): $INPUT_FILE"
  exit 1
fi

mapfile -t SORTED_SET_IDS < <(printf '%s\n' "${!set_lists[@]}" | sort -n)

step "Parsed Lists Summary"
info "Total lists (after filter): ${#all_lists[@]}"
info "SetID groups              : ${#SORTED_SET_IDS[@]}"
info "Lists by SET_ID:"
for set_id in "${SORTED_SET_IDS[@]}"; do
  info "  SET_ID ${set_id} => ${set_lists[$set_id]}"
done

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="" machine="" 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
    err "unsupported SET_ID '$sid' (only 10~39 supported)."
    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() {
  step "MODULE: generate (echo)"
  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]}"

    info "SET_ID=${set_id} AREA=${AREA} MACHINE=${MACHINE} GENERATOR=${GENERATOR}"
    for lid in "${LIDS[@]}"; do
      WORLD_ID_LISTS="${list_worlds[$lid]}"
      printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/${GENERATOR} ${DIRECTORY_NAME} ${WORLD_ID_LISTS}"
    done
  done
}

run_reward_before_merge() {
  step "MODULE: reward_before_merge (echo)"
  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    printf '%s\n' "\$HOME/bin/send_fastcgi_all_once_n1_by_world_reward_merge ${ENVIRONMENT} ${WORLD_ID_LISTS}"
  done
}

run_db_connect_verify() {
  step "MODULE: dbcheck (echo)"
  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"
      printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_connect_verify \"${REMOTE_MERGE_INI}\""
    done
  done
}

run_pre_backup() {
  step "MODULE: prebackup (echo)"

  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"
  printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Pre AccountDB"

  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"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Pre GameDB"
  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}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre WorldDB"
  done
}

run_restore_prebackup() {
  step "MODULE: restore_prebackup (echo)"

  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"
  printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Pre AccountDB"

  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"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Pre GameDB"
  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}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_restore \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre WorldDB"
  done
}

run_merge_exec() {
  step "MODULE: merge (echo)"
  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}"
    printf 'ssh %s "%s"\n' "$MACHINE" "cd ${REMOTE_BASE_DIR}; nohup ./merge_exec &"
  done
}

run_log_check() {
  step "MODULE: logcheck (echo)"
  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}"
    printf 'ssh %s "%s"\n' "$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
}

run_wait_done() {
  step "MODULE: waitdone (echo)"
  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}"
    printf 'ssh %s "%s"\n' "$MACHINE" "cd ${REMOTE_BASE_DIR}; grep -q ',Done' Test.log* 2>/dev/null"
  done
}

run_update_guild_race_data() {
  step "MODULE: update_guild_race_data (echo)"
  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"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_update_guild_race_data \"${REMOTE_MERGE_INI}\""
  done
}

run_post_backup() {
  step "MODULE: postbackup (echo)"

  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"
  printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\" _${FILE_NAME}_Post AccountDB"

  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"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI_FIRST_SET}\" _${FILE_NAME}_Post GameDB"
  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}"
    REMOTE_MERGE_INI="${REMOTE_BASE_DIR}/merge_setup.ini"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_db_backup \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Post WorldDB"
  done
}

run_check_diff() {
  step "MODULE: check (echo)"
  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"
    printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_check \"${REMOTE_MERGE_INI}\" _${FILE_NAME}_Pre _${FILE_NAME}_Post $ENVIRONMENT"
  done
}

run_update_db_setup() {
  step "MODULE: update_db_setup (echo)"
  if [ "$ENVIRONMENT" != "live" ]; then
    info "update_db_setup is only allowed in live environment. (No commands generated.)"
    return
  fi
  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    printf '%s\n' "\$HOME/bin/update_db_setup_for_merge ${WORLD_ID_LISTS}"
  done
}

run_move_old_server_directory() {
  step "MODULE: move_old_server_directory (echo)"
  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]}"
    printf '%s\n' "\$HOME/bin/move_source_wids_directory_after_merge ${ENVIRONMENT} ${DIRECTORY_NAME} ${WORLD_ID_LISTS}"
  done
}

run_remove_source_wids_on_hosts() {
  step "MODULE: remove_source_wids_on_hosts (echo)"
  if [ "$ENVIRONMENT" != "live" ]; then
    info "remove_source_wids_on_hosts is only allowed in live environment. (No commands generated.)"
    return
  fi
  printf '%s\n' "cp -arf \$HOME/bin/hosts \$HOME/bin/hosts.old"
  for lid in "${all_lists[@]}"; do
    WORLD_ID_LISTS="${list_worlds[$lid]}"
    printf '%s\n' "\$HOME/bin/remove_source_wids_on_hosts_after_merge ${WORLD_ID_LISTS}"
  done
}

drop_cmds_no_flag() {
  case "$BACKUP_TARGET" in
    pre)  printf '%s' "\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Pre" ;;
    post) printf '%s' "\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Post" ;;
    both) printf '%s' "\$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" ;;
    *) err "invalid target '$BACKUP_TARGET'"; exit 1 ;;
  esac
}

drop_cmds_with_flag() {
  local flag="$1"
  case "$BACKUP_TARGET" in
    pre)  printf '%s' "\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Pre ${flag}" ;;
    post) printf '%s' "\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Post ${flag}" ;;
    both) printf '%s' "\$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Pre ${flag}; \$HOME/bin/drop_database_for_current_machine_by_pattern postfix _${FILE_NAME}_Post ${flag}" ;;
    *) err "invalid target '$BACKUP_TARGET'"; exit 1 ;;
  esac
}

emit_removebackup_on_machines() {
  local cmd="$1"
  if [ "$ENVIRONMENT" = "test" ]; then
    declare -A SEEN=()
    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

    for m in "${MACHINES[@]}"; do
      if [[ -z "${SEEN[$m]+x}" ]]; then
        SEEN["$m"]=1
        printf 'ssh %s "%s"\n' "$m" "$cmd"
      fi
    done
    return
  fi

  if [ "${#SET_FILTER_TOKENS[@]}" -eq 0 ]; then
    printf 'sendscript alldb "%s"\n' "$cmd"
    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
      printf 'ssh %s "%s"\n' "$MACHINE" "$cmd"
    fi
  done
}

run_remove_backup() {
  step "MODULE: removebackup (echo, target=${BACKUP_TARGET}, backup_flag=default(1))"
  emit_removebackup_on_machines "$(drop_cmds_no_flag)"
}

run_remove_backup_wb() {
  step "MODULE: removebackup_wb (echo, target=${BACKUP_TARGET}, backup_flag=1)"
  emit_removebackup_on_machines "$(drop_cmds_with_flag 1)"
}

run_remove_backup_wob() {
  step "MODULE: removebackup_wob (echo, target=${BACKUP_TARGET}, backup_flag=0)"
  emit_removebackup_on_machines "$(drop_cmds_with_flag 0)"
}

run_merge_rollback_partial_worlds() {
  step "MODULE: merge_rollback_partial_worlds (echo)"

  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"
  printf 'ssh %s "%s"\n' "$MACHINE" "\$HOME/bin/merge_rollback_partial_worlds \"${REMOTE_MERGE_INI_FIRST_GLOBAL}\""
}

run_re_run_exec_merge() {
  step "GROUP MODULE: re_run_exec_merge (echo)"

  local old_target="$BACKUP_TARGET"
  BACKUP_TARGET="post"
  run_remove_backup_wob
  BACKUP_TARGET="$old_target"

  run_restore_prebackup
  run_merge_rollback_partial_worlds
  run_merge_exec
  run_log_check
  run_wait_done
  run_update_guild_race_data
  run_post_backup
  run_check_diff
}

run_re_run_pre_merge() {
  step "GROUP MODULE: re_run_pre_merge (echo)"

  local old_target="$BACKUP_TARGET"
  BACKUP_TARGET="post"
  run_remove_backup_wob
  BACKUP_TARGET="$old_target"

  run_restore_prebackup
  run_merge_rollback_partial_worlds
}

case "$MODULE" in
  exec_merge)
    run_generate
    run_db_connect_verify
    run_pre_backup
    run_merge_exec
    run_log_check
    run_wait_done
    run_update_guild_race_data
    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
    ;;
  re_run_exec_merge)
    run_re_run_exec_merge
    ;;
  re_run_pre_merge)
    run_re_run_pre_merge
    ;;

  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 ;;
  update_guild_race_data) run_update_guild_race_data ;;
  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 ;;
  removebackup_wb) run_remove_backup_wb ;;
  removebackup_wob) run_remove_backup_wob ;;
  merge_rollback_partial_worlds) run_merge_rollback_partial_worlds ;;
  *)
    err "unknown MODULE '${MODULE}'."
    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))

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

