#!/usr/bin/env bash
#
# Package McpAutomationBridge plugin as pre-built binaries.
# Output can be distributed to Blueprint-only projects (no C++ compilation needed).
#
# Usage:
#   ./scripts/package-plugin.sh /path/to/UE_5.6
#   ./scripts/package-plugin.sh /path/to/UE_5.6 /custom/output/dir
#   ./scripts/package-plugin.sh /path/to/UE_5.6 /custom/output/dir -NoDefaultPlugins
#
# The script will:
#   1. Build the plugin via RunUAT
#   2. Set "Installed": true in the output .uplugin
#   3. Create a zip archive ready for distribution
#

set -euo pipefail

# ─── Arguments ──────────────────────────────────────────────────────────────

ENGINE_DIR="${1:?Usage: $0 <UnrealEngineDir> [OutputDir] [extra RunUAT args...]}"
shift

OUTPUT_DIR=""
EXTRA_ARGS=()
for arg in "$@"; do
    case "$arg" in
        -*) EXTRA_ARGS+=("$arg") ;;
        *)
            if [ -n "$OUTPUT_DIR" ]; then
                echo "ERROR: Unexpected extra output directory argument: $arg"
                exit 1
            fi
            OUTPUT_DIR="$arg"
            ;;
    esac
done
OUTPUT_DIR="${OUTPUT_DIR:-}"

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
REPO_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
OUTPUT_DIR="${OUTPUT_DIR:-$REPO_ROOT/build}"
mkdir -p "$OUTPUT_DIR"
OUTPUT_DIR="$(cd "$OUTPUT_DIR" && pwd)"
PLUGIN_FILE="$REPO_ROOT/plugins/McpAutomationBridge/McpAutomationBridge.uplugin"

if [ ! -f "$PLUGIN_FILE" ]; then
    echo "ERROR: Plugin file not found: $PLUGIN_FILE"
    exit 1
fi

# ─── Detect platform and RunUAT path ────────────────────────────────────────

case "$(uname -s)" in
    Darwin)
        PLATFORM="Mac"
        RUN_UAT="$ENGINE_DIR/Engine/Build/BatchFiles/RunUAT.sh"
        ;;
    Linux)
        PLATFORM="Linux"
        RUN_UAT="$ENGINE_DIR/Engine/Build/BatchFiles/RunUAT.sh"
        ;;
    MINGW*|MSYS*|CYGWIN*|Windows_NT)
        PLATFORM="Win64"
        RUN_UAT="$ENGINE_DIR/Engine/Build/BatchFiles/RunUAT.bat"
        ;;
    *)
        echo "ERROR: Unsupported platform: $(uname -s)"
        exit 1
        ;;
esac

if [ ! -f "$RUN_UAT" ]; then
    echo "ERROR: RunUAT not found: $RUN_UAT"
    echo "Make sure the first argument points to your UE installation root."
    exit 1
fi

# ─── Preflight: python3 ─────────────────────────────────────────────────────

if ! command -v python3 &>/dev/null; then
    echo "ERROR: python3 is required but not found in PATH."
    echo "Install Python 3 or ensure it is on your PATH."
    exit 1
fi

# ─── Extract version info ───────────────────────────────────────────────────

# Get UE version from the engine (paths passed via argv to avoid quoting issues)
UE_VERSION_FILE="$ENGINE_DIR/Engine/Build/Build.version"
if [ -f "$UE_VERSION_FILE" ]; then
    UE_MAJOR=$(python3 -c "import json,sys; print(json.load(open(sys.argv[1]))['MajorVersion'])" "$UE_VERSION_FILE" 2>/dev/null || echo "5")
    UE_MINOR=$(python3 -c "import json,sys; print(json.load(open(sys.argv[1]))['MinorVersion'])" "$UE_VERSION_FILE" 2>/dev/null || echo "x")
    UE_VER="${UE_MAJOR}.${UE_MINOR}"
else
    UE_VER="unknown"
fi

PLUGIN_VER=$(python3 -c "import json,sys; print(json.load(open(sys.argv[1]))['VersionName'])" "$PLUGIN_FILE" 2>/dev/null || echo "0.0.0")

ZIP_NAME="McpAutomationBridge-v${PLUGIN_VER}-UE${UE_VER}-${PLATFORM}.zip"
ZIP_PATH="$OUTPUT_DIR/$ZIP_NAME"
STAGING_DIR="$(mktemp -d "${TMPDIR:-/tmp}/McpAutomationBridge-package.XXXXXX")"
PACKAGE_DIR="$STAGING_DIR/McpAutomationBridge"
SOURCE_PLUGIN_DIR="$STAGING_DIR/source/McpAutomationBridge"
SOURCE_PLUGIN_FILE="$SOURCE_PLUGIN_DIR/McpAutomationBridge.uplugin"

cleanup_staging() {
    rm -rf "$STAGING_DIR"
}
trap cleanup_staging EXIT

echo "============================================"
echo "  Package McpAutomationBridge Plugin"
echo "============================================"
echo "  Plugin version : $PLUGIN_VER"
echo "  UE version     : $UE_VER"
echo "  Platform        : $PLATFORM"
echo "  Engine          : $ENGINE_DIR"
echo "  Output          : $ZIP_PATH"
echo "============================================"
echo ""

# ─── Build ──────────────────────────────────────────────────────────────────

echo "Building plugin..."
rm -f "$ZIP_PATH"
mkdir -p "$(dirname "$SOURCE_PLUGIN_DIR")"
cp -R "$REPO_ROOT/plugins/McpAutomationBridge" "$SOURCE_PLUGIN_DIR"

if [ "${UE_MAJOR:-0}" -gt 5 ] || { [ "${UE_MAJOR:-0}" -eq 5 ] && [ "${UE_MINOR:-0}" -ge 2 ]; }; then
    python3 -c "
import json, sys
path = sys.argv[1]
with open(path, 'r') as handle:
    data = json.load(handle)
for dependency in data.get('Plugins', []):
    if dependency.get('Name') == 'PCG':
        dependency.pop('Optional', None)
with open(path, 'w') as handle:
    json.dump(data, handle, indent=2)
    handle.write('\n')
" "$SOURCE_PLUGIN_FILE"
fi

"$RUN_UAT" BuildPlugin \
    -Plugin="$SOURCE_PLUGIN_FILE" \
    -Package="$PACKAGE_DIR" \
    -TargetPlatforms="$PLATFORM" \
    -Rocket "${EXTRA_ARGS[@]}"

echo ""
echo "Build complete."

# ─── Post-process: set Installed=true ────────────────────────────────────────

if [ -f "$PACKAGE_DIR/McpAutomationBridge.uplugin" ]; then
    OUTPUT_PLUGIN_DIR="$PACKAGE_DIR"
elif [ -f "$PACKAGE_DIR/HostProject/Plugins/McpAutomationBridge/McpAutomationBridge.uplugin" ]; then
    OUTPUT_PLUGIN_DIR="$PACKAGE_DIR/HostProject/Plugins/McpAutomationBridge"
else
    echo "ERROR: Packaged plugin output not found under: $PACKAGE_DIR"
    exit 1
fi

OUTPUT_UPLUGIN="$OUTPUT_PLUGIN_DIR/McpAutomationBridge.uplugin"
if [ -f "$OUTPUT_UPLUGIN" ]; then
    echo "Setting Installed=true in output .uplugin..."
    python3 -c "
import json, sys
f = sys.argv[1]
with open(f, 'r') as fh:
    data = json.load(fh)
data['Installed'] = True
with open(f, 'w') as fh:
    json.dump(data, fh, indent=2)
    fh.write('\n')
" "$OUTPUT_UPLUGIN"
fi

# ─── Zip ─────────────────────────────────────────────────────────────────────

echo "Creating archive: $ZIP_NAME"
rm -f "$ZIP_PATH"
find "$OUTPUT_PLUGIN_DIR" -type d -iname "*.dSYM" -prune -exec rm -rf {} +
find "$OUTPUT_PLUGIN_DIR" -type f \( \
    -iname "*.pdb" -o \
    -iname "*.debug" -o \
    -iname "*.sym" \
\) -delete
cd "$(dirname "$OUTPUT_PLUGIN_DIR")"
zip -r "$ZIP_PATH" McpAutomationBridge/ \
    -x "*.pdb" \
    -x "*.debug" \
    -x "*.sym" \
    -x "*.dSYM/" \
    -x "*.dSYM/*" \
    -x "McpAutomationBridge/Intermediate/*"
python3 -c "
import sys
import zipfile

archive_path = sys.argv[1]
forbidden_extensions = ('.pdb', '.debug', '.sym')
with zipfile.ZipFile(archive_path) as archive:
    forbidden_entries = [
        name
        for name in archive.namelist()
        if name.lower().endswith(forbidden_extensions)
        or any(
            part.endswith('.dsym')
            for part in name.replace('\\\\', '/').lower().rstrip('/').split('/')
        )
    ]
if forbidden_entries:
    print(
        'ERROR: Distribution archive contains debug symbols: '
        + ', '.join(forbidden_entries),
        file=sys.stderr,
    )
    sys.exit(1)
" "$ZIP_PATH"
echo ""

FINAL_SIZE=$(du -sh "$ZIP_PATH" | cut -f1)
echo "============================================"
echo "  Done!"
echo "  Archive: $ZIP_PATH ($FINAL_SIZE)"
echo "============================================"
echo ""
echo "To install: unzip into YourProject/Plugins/"