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

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
# shellcheck source=tools/mcc-env.sh
source "$REPO_ROOT/tools/mcc-env.sh"
# shellcheck source=.skills/mcc-integration-testing/scripts/common.sh
source "$REPO_ROOT/.skills/mcc-integration-testing/scripts/common.sh"

usage() {
    cat <<'EOF'
Usage: tools/run-creative-e2e.sh <server-dir> <mc-version> <legacy|modern>

Examples:
  env -u MCC_SERVERS tools/run-creative-e2e.sh 1.8 1.8 legacy
  MCC_SERVERS=/home/anon/Minecraft/Servers tools/run-creative-e2e.sh 1.20.6-Vanilla 1.20.6 modern
EOF
}

SERVER_DIR="${1:-}"
MC_VERSION="${2:-}"
PROFILE="${3:-}"
SERVER_PORT=""

if [[ -z "$SERVER_DIR" || -z "$MC_VERSION" || -z "$PROFILE" ]]; then
    usage >&2
    exit 1
fi

if [[ "$PROFILE" != "legacy" && "$PROFILE" != "modern" ]]; then
    echo "Unsupported profile: $PROFILE" >&2
    exit 1
fi

SESSION_NAME="mc-${SERVER_DIR//./_}"
TEST_ROOT="${TMPDIR:-/tmp}/mcc-creative-e2e/${SERVER_DIR//\//_}"
CFG="$TEST_ROOT/MinecraftClient.$MC_VERSION.ini"
MCC_SESSION="creative-e2e-${SERVER_DIR//[^a-zA-Z0-9]/_}-${PROFILE}"
TEST_USERNAME="$(_mcc_resolve_username "$MCC_SESSION")"
MCC_LOG="$(_mcc_session_log_file "$MCC_SESSION")"
PID_FILE="$(_mcc_session_pid_file "$MCC_SESSION")"
MCC_TMUX_SESSION="$(_mcc_tmux_session_name "$MCC_SESSION")"
SERVER_LOG_FILE="$MCC_SERVERS/$SERVER_DIR/logs/latest.log"
INPUT_FILE="$(_mcc_session_input_file "$MCC_SESSION")"
SERVER_PORT="25565"

mkdir -p "$TEST_ROOT"

wait_for_file_pattern() {
    local file="$1"
    local pattern="$2"
    local description="$3"
    local timeout="${4:-60}"
    local elapsed=0

    while (( elapsed < timeout )); do
        if [[ -f "$file" ]] && grep -Fq "$pattern" "$file"; then
            return 0
        fi
        sleep 1
        ((elapsed += 1))
    done

    echo "Timed out waiting for: $description" >&2
    return 1
}

port_is_listening() {
    local port="$1"

    if ! command -v python3 >/dev/null 2>&1; then
        return 1
    fi

    python3 - "$port" <<'PY'
import socket
import sys

port = int(sys.argv[1])

for addrinfo in socket.getaddrinfo("localhost", port, 0, socket.SOCK_STREAM):
    family, socktype, proto, _, sockaddr = addrinfo
    try:
        sock = socket.socket(family, socktype, proto)
        sock.settimeout(0.2)
        if sock.connect_ex(sockaddr) == 0:
            sock.close()
            raise SystemExit(0)
        sock.close()
    except OSError:
        continue

raise SystemExit(1)
PY
}

wait_for_rcon_port_free() {
    local timeout="${1:-30}"
    local elapsed=0

    while (( elapsed < timeout )); do
        if ! port_is_listening 25575; then
            return 0
        fi
        sleep 1
        ((elapsed += 1))
    done

    echo "Timed out waiting for RCON port 25575 to become free" >&2
    return 1
}
cleanup() {
    mcc-cmd --session "$MCC_SESSION" "quit" >/dev/null 2>&1 || true
    sleep 2
    mcc-kill --session "$MCC_SESSION" >/dev/null 2>&1 || true

    if tmux has-session -t "$SESSION_NAME" 2>/dev/null; then
        tmux send-keys -t "$SESSION_NAME" "stop" C-m >/dev/null 2>&1 || true
        wait_for_server_stop "$SERVER_DIR" 20 >/dev/null 2>&1 || true
    fi

    rm -f "$MCC_SERVERS/$SERVER_DIR/stdin.pipe"
    tmux kill-session -t "$SESSION_NAME" 2>/dev/null || true
    wait_for_rcon_port_free 30 || true
}

trap cleanup EXIT

prepare_config() {
    MCC_TEST_ACCOUNT_TYPE=mojang MCC_TEST_PASSWORD=- \
        bash "$REPO_ROOT/.skills/mcc-integration-testing/scripts/prepare_offline_mcc_config.sh" \
        "$CFG" "$MC_VERSION" "$TEST_USERNAME" >/dev/null

    sed_in_place \
        -e "s#^Server = .*#Server = { Host = \"localhost\", Port = $SERVER_PORT }#" \
        -e 's/TerrainAndMovements = false/TerrainAndMovements = true/' \
        -e 's/InventoryHandling = false/InventoryHandling = true/' \
        -e 's/EntityHandling = false/EntityHandling = true/' \
        -e 's/AutoRespawn = false/AutoRespawn = true/' \
        "$CFG"
    disable_noisy_bots_in_ini "$CFG"
}

send_mcc_command() {
    local command="$1"
    local delay="${2:-2}"
    mcc-cmd --session "$MCC_SESSION" "$command"
    sleep "$delay"
}

run_server_command() {
    local command="$1"
    local attempt
    for attempt in 1 2 3 4 5; do
        if bash "$REPO_ROOT/tools/mc-rcon.sh" "$command" >/dev/null 2>&1; then
            return 0
        fi
        sleep 1
    done

    echo "Server command failed after retries: $command" >&2
    return 1
}

print_phase() {
    local name="$1"
    local status="$2"
    printf 'PHASE_%s=%s\n' "$name" "$status"
}

assert_log_contains() {
    local file="$1"
    local pattern="$2"
    local description="$3"
    local timeout="${4:-20}"
    wait_for_file_pattern "$file" "$pattern" "$description" "$timeout"
}

start_mcc_session() {
    local -a mcc_args=("$CFG" "$TEST_USERNAME" "-" "localhost:$SERVER_PORT")
    local mcc_args_cmd
    mcc_args_cmd="$(printf '%q ' "${mcc_args[@]}")"

    tmux kill-session -t "$MCC_TMUX_SESSION" 2>/dev/null || true
    rm -f "$PID_FILE"
    tmux new-session -d -s "$MCC_TMUX_SESSION" -x 160 -y 50 \
        "cd '$REPO_ROOT' && printf '%s\n' \"\$\$\" > '$PID_FILE' && exec env MCC_FILE_INPUT=1 MCC_INPUT_FILE='$INPUT_FILE' dotnet run --project MinecraftClient -c Release --no-build -- $mcc_args_cmd > '$MCC_LOG' 2>&1"

    for _ in $(seq 1 25); do
        if [[ -s "$PID_FILE" ]]; then
            return 0
        fi
        sleep 0.2
    done

    echo "Failed to capture MCC PID for session $MCC_SESSION" >&2
    return 1
}

legacy_server_setup() {
    run_server_command "gamerule sendCommandFeedback true"
    run_server_command "time set day"
    run_server_command "weather clear"
    run_server_command "gamemode creative $TEST_USERNAME"
    run_server_command "fill -2 79 -2 2 79 2 stone"
    run_server_command "tp $TEST_USERNAME 0 80 0"
}

modern_server_setup() {
    run_server_command "gamerule sendCommandFeedback true"
    run_server_command "gamerule logAdminCommands true"
    run_server_command "time set day"
    run_server_command "weather clear"
    run_server_command "gamemode creative $TEST_USERNAME"
    run_server_command "fill -2 79 -2 2 79 2 stone"
    run_server_command "tp $TEST_USERNAME 0 80 0"
}

legacy_mob_and_effects() {
    run_server_command "summon Cow 2 80 0"
    run_server_command "summon Zombie 4 80 0"
    run_server_command "summon Pig -2 80 0"
    run_server_command "effect $TEST_USERNAME 1 30 1 true"
    run_server_command "effect $TEST_USERNAME 10 10 1 true"
}

modern_mob_and_effects() {
    run_server_command "summon minecraft:cow 2 80 0"
    run_server_command "summon minecraft:zombie 4 80 0"
    run_server_command "summon minecraft:pig -2 80 0"
    run_server_command "effect give $TEST_USERNAME minecraft:speed 30 1 true"
    run_server_command "effect give $TEST_USERNAME minecraft:regeneration 10 1 true"
}

bash "$REPO_ROOT/.skills/mcc-integration-testing/scripts/preflight_test_env.sh" "$SERVER_DIR" >/dev/null
bash "$REPO_ROOT/.skills/mcc-integration-testing/scripts/reset_shared_test_state.sh" "$SERVER_DIR" >/dev/null
mcc-reset-session --session "$MCC_SESSION" >/dev/null
wait_for_rcon_port_free 30 || true
mkdir -p "$(dirname "$MCC_LOG")" "$(dirname "$INPUT_FILE")"
rm -f "$MCC_LOG" "$INPUT_FILE"

bash "$REPO_ROOT/.skills/mcc-integration-testing/scripts/ensure_offline_server.sh" "$SERVER_DIR" >/dev/null
SERVER_PORT="$(bash "$REPO_ROOT/.skills/mcc-integration-testing/scripts/get_server_port.sh" "$SERVER_DIR")"
if [[ -f "$MCC_SERVERS/$SERVER_DIR/server.properties" ]]; then
    sed_in_place 's/^use-native-transport=.*/use-native-transport=false/' "$MCC_SERVERS/$SERVER_DIR/server.properties"
fi

mc-start "$SERVER_DIR" >/dev/null
wait_for_server_ready "$SERVER_DIR" || exit 1
prepare_config

: > "$INPUT_FILE"

start_mcc_session

assert_log_contains "$MCC_LOG" "Server was successfully joined." "MCC join success" 90
assert_log_contains "$SERVER_LOG_FILE" "$TEST_USERNAME joined the game" "server join entry" 30
print_phase "CONNECT" "PASS"

run_server_command "op $TEST_USERNAME"
sleep 1

if [[ "$PROFILE" == "legacy" ]]; then
    legacy_server_setup
else
    modern_server_setup
fi
sleep 2

chat_token="creative_e2e_chat_${MC_VERSION//./_}"
cmd_token="creative_e2e_cmd_${MC_VERSION//./_}"
broadcast_token="server_broadcast_${MC_VERSION//./_}"
whisper_token="server_whisper_${MC_VERSION//./_}"

send_mcc_command "$chat_token" 2
assert_log_contains "$SERVER_LOG_FILE" "$chat_token" "client chat on server" 20

send_mcc_command "/say $cmd_token" 2
assert_log_contains "$SERVER_LOG_FILE" "$cmd_token" "client command on server" 20
print_phase "SEND" "PASS"

run_server_command "say $broadcast_token"
run_server_command "tell $TEST_USERNAME $whisper_token"
assert_log_contains "$MCC_LOG" "$broadcast_token" "server broadcast in MCC" 20
assert_log_contains "$MCC_LOG" "$whisper_token" "server whisper in MCC" 20
print_phase "RECEIVE" "PASS"

send_mcc_command "look east" 2
send_mcc_command "move east -f" 2
send_mcc_command "move west -f" 2
send_mcc_command "move down -f" 2
send_mcc_command "move get" 2
assert_log_contains "$MCC_LOG" "[FileInput] > look east" "look command" 20
assert_log_contains "$MCC_LOG" "[FileInput] > move east -f" "move east command" 20
assert_log_contains "$MCC_LOG" "[FileInput] > move west -f" "move west command" 20
assert_log_contains "$MCC_LOG" "[FileInput] > move down -f" "move down command" 20
assert_log_contains "$MCC_LOG" "[FileInput] > move get" "move get command" 20
print_phase "MOVEMENT" "PASS"
print_phase "PHYSICS" "PASS"

if [[ "$PROFILE" == "legacy" ]]; then
    legacy_mob_and_effects
else
    modern_mob_and_effects
fi
sleep 2

send_mcc_command "entity" 3
assert_log_contains "$MCC_LOG" "[FileInput] > entity" "entity command" 20
print_phase "MOBS" "PASS"

send_mcc_command "health" 2
assert_log_contains "$MCC_LOG" "[FileInput] > health" "health command after effects" 20
print_phase "EFFECTS" "PASS"

send_mcc_command "inventory player list" 3
send_mcc_command "inventory creativegive 36 Diamond 16" 3
if [[ "$PROFILE" == "modern" ]]; then
    send_mcc_command "inventory creativedelete 36" 3
fi
send_mcc_command "inventory player list" 3
assert_log_contains "$MCC_LOG" "[FileInput] > inventory player list" "inventory list command" 20
assert_log_contains "$MCC_LOG" "Requested Diamond x16 in slot #36" "creative give result" 20
if [[ "$PROFILE" == "modern" ]]; then
    assert_log_contains "$MCC_LOG" "Requested to clear slot #36" "creative delete result" 20
fi
print_phase "INVENTORY" "PASS"

printf 'LOG_DIR=%s\n' "$TEST_ROOT"