package.sh

#!/usr/bin/env bash
set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

# 模式判断：是否开启全静态编译 (--full-static)
FULL_STATIC_MODE=0
for arg in "$@"; do
    if [ "$arg" = "--full-static" ]; then
        FULL_STATIC_MODE=1
    fi
done
if [ "${FULL_STATIC:-0}" = "1" ]; then
    FULL_STATIC_MODE=1
fi

# Resolve PHP executable (handle environments where php is php85/php84, e.g. Alpine)
if command -v php &> /dev/null; then
    PHP_EXECUTABLE="php"
elif command -v php85 &> /dev/null; then
    PHP_EXECUTABLE="php85"
elif command -v php84 &> /dev/null; then
    PHP_EXECUTABLE="php84"
else
    PHP_EXECUTABLE="php"
fi

if [ -f "./vendor/bin/tpc.php" ]; then
    TPC_BIN="./vendor/bin/tpc.php"
elif command -v tpc &> /dev/null; then
    TPC_BIN="tpc"
else
    echo "[ERROR] tpc compiler not found! Make sure swoole/typephp is installed or tpc is in PATH."
    exit 1
fi

PHPX_DIR="$SCRIPT_DIR/vendor/swoole/phpx"
export PHPX_HOME="$PHPX_DIR"

if [ "$FULL_STATIC_MODE" = "1" ]; then
    # 全静态编译前置检查
    if ! command -v clang &> /dev/null; then
        echo "[ERROR] Clang compiler is required for --full-static build, but 'clang' was not found in PATH."
        exit 1
    fi
    SDK_DIR="$PHPX_DIR/full-static/sdk"
    if [ ! -d "$SDK_DIR" ]; then
        echo "[ERROR] --full-static requires bundled SDK at $SDK_DIR, but directory was not found!"
        exit 1
    fi
    if [ ! -f "$SDK_DIR/lib/musl/crt1.o" ]; then
        echo "[ERROR] musl startup file crt1.o not found at $SDK_DIR/lib/musl/crt1.o!"
        exit 1
    fi
    if [ ! -f "$SDK_DIR/lib/libphpx.a" ]; then
        echo "[ERROR] static libphpx.a not found at $SDK_DIR/lib/libphpx.a!"
        exit 1
    fi
    if [ ! -f "$SDK_DIR/lib/libphp.a" ]; then
        echo "[ERROR] static libphp.a not found at $SDK_DIR/lib/libphp.a!"
        exit 1
    fi
else
    # 常规动态编译流程：确认 PHP_HOME 并编译动态 PHPX
    if [ -z "$PHP_HOME" ]; then
        PHP_PREFIX=$(php-config --prefix 2>/dev/null || echo "/usr")
        export PHP_HOME="$PHP_PREFIX"
    fi

    if [ -d "$PHPX_DIR" ] && [ ! -f "$PHPX_DIR/lib/libphpx.so" ] && [ ! -f "$PHPX_DIR/lib/libphpx.a" ]; then
        echo "[INFO] Building PHPX library in $PHPX_DIR ..."
        cd "$PHPX_DIR"
        
        # 修复 C++ mpfr.h / gmp.h 搜索路径
        EXTRA_INC=""
        for d in /usr/include /usr/local/include /usr/include/x86_64-linux-gnu; do
            [ -d "$d" ] && EXTRA_INC="$EXTRA_INC -I$d"
        done
        
        # 在 CMakeLists.txt 的 include_directories 中注入
        sed -i 's/include_directories(include tests\/include src\/misc)/include_directories(include tests\/include src\/misc \/usr\/include \/usr\/local\/include \/usr\/include\/x86_64-linux-gnu)/g' CMakeLists.txt
        
        cmake . -Dphp_dir="$PHP_HOME" -DBUILD_TESTS=OFF -DBUILD_EXT=OFF -DCMAKE_CXX_FLAGS="$EXTRA_INC" -DCMAKE_C_FLAGS="$EXTRA_INC"
        make phpx -j$(nproc)
        cd "$SCRIPT_DIR"
    fi
fi

# 3. Ensure build directory and php.ini
mkdir -p "$SCRIPT_DIR/build"
if [ -f "$SCRIPT_DIR/php.ini" ]; then
    cp -f "$SCRIPT_DIR/php.ini" "$SCRIPT_DIR/build/php.ini"
fi

# 4. Compile project via TPC
if [ "$FULL_STATIC_MODE" = "1" ]; then
    echo "[INFO] Running TPC in --full-static mode with Clang compiler ($PHP_EXECUTABLE)..."
    if ! "$PHP_EXECUTABLE" "$TPC_BIN" "$SCRIPT_DIR/project.linux.yml" --full-static --compiler=clang; then
        echo "[ERROR] TPC --full-static compilation failed!"
        ls -la "$SCRIPT_DIR/build" || true
        exit 1
    fi
else
    echo "[INFO] Running TPC compiler with PHP_HOME=$PHP_HOME PHPX_HOME=$PHPX_HOME ..."
    if ! "$PHP_EXECUTABLE" "$TPC_BIN" "$SCRIPT_DIR/project.linux.yml"; then
        echo "[ERROR] TPC compilation failed!"
        echo "[DEBUG] Inspecting generated C++ files in build directory..."
        ls -la "$SCRIPT_DIR/build" || true
        exit 1
    fi
fi

echo "[INFO] Step 2: Packaging into dist directory..."
rm -rf "$SCRIPT_DIR/dist"
mkdir -p "$SCRIPT_DIR/dist"

# Copy executable
if [ "$FULL_STATIC_MODE" = "1" ]; then
    if [ -f "$SCRIPT_DIR/build/webman-server" ]; then
        cp -f "$SCRIPT_DIR/build/webman-server" "$SCRIPT_DIR/dist/webman-server"
        chmod +x "$SCRIPT_DIR/dist/webman-server"
    elif [ -f "$SCRIPT_DIR/build/webman_server" ]; then
        cp -f "$SCRIPT_DIR/build/webman_server" "$SCRIPT_DIR/dist/webman-server"
        chmod +x "$SCRIPT_DIR/dist/webman-server"
    fi
    [ -d "$SCRIPT_DIR/config" ] && cp -r "$SCRIPT_DIR/config" "$SCRIPT_DIR/dist/"
    [ -d "$SCRIPT_DIR/public" ] && cp -r "$SCRIPT_DIR/public" "$SCRIPT_DIR/dist/"
    if [ -d "$SCRIPT_DIR/app/view" ]; then
        mkdir -p "$SCRIPT_DIR/dist/app"
        cp -r "$SCRIPT_DIR/app/view" "$SCRIPT_DIR/dist/app/"
    fi
    [ -f "$SCRIPT_DIR/start.sh" ] && cp -f "$SCRIPT_DIR/start.sh" "$SCRIPT_DIR/dist/" && chmod +x "$SCRIPT_DIR/dist/start.sh"
    if command -v strip &> /dev/null; then
        echo "[INFO] Stripping debug symbols from static binary..."
        strip --strip-all "$SCRIPT_DIR/dist/webman-server" 2>/dev/null || true
    fi
    echo "[INFO] Full-static packaging completed successfully! Dist path: $SCRIPT_DIR/dist"
    ls -la "$SCRIPT_DIR/dist"
    exit 0
fi

# 常规动态打包流程：
if [ -f "$SCRIPT_DIR/build/webman-server" ]; then
    cp -f "$SCRIPT_DIR/build/webman-server" "$SCRIPT_DIR/dist/webman-server.bin"
    chmod +x "$SCRIPT_DIR/dist/webman-server.bin"
elif [ -f "$SCRIPT_DIR/build/webman_server" ]; then
    cp -f "$SCRIPT_DIR/build/webman_server" "$SCRIPT_DIR/dist/webman-server.bin"
    chmod +x "$SCRIPT_DIR/dist/webman-server.bin"
fi

# Copy PHPX runtime library
if [ -f "$PHPX_DIR/lib/libphpx.so" ]; then
    cp -f "$PHPX_DIR/lib/libphpx.so" "$SCRIPT_DIR/dist/"
elif [ -f "$PHPX_DIR/libphpx.so" ]; then
    cp -f "$PHPX_DIR/libphpx.so" "$SCRIPT_DIR/dist/"
fi

# Copy PHP embed runtime library (libphp.so) if present
PHP_LIB_DIR=$(php-config --prefix 2>/dev/null)/lib
for candidate in "$PHP_LIB_DIR/libphp.so" "$PHP_LIB_DIR/libphp8.so" /usr/lib/libphp.so /usr/lib/libphp8.4.so /usr/lib/x86_64-linux-gnu/libphp8.4.so /usr/lib/x86_64-linux-gnu/libphp.so; do
    if [ -f "$candidate" ]; then
        echo "[INFO] Found PHP runtime library: $candidate, copying to dist/ ..."
        cp -f "$candidate" "$SCRIPT_DIR/dist/libphp.so"
        break
    fi
done
if [ ! -f "$SCRIPT_DIR/dist/libphp.so" ]; then
    FOUND_LIBPHP=$(find /usr -name "libphp*.so" 2>/dev/null | head -n 1 || true)
    if [ -n "$FOUND_LIBPHP" ] && [ -f "$FOUND_LIBPHP" ]; then
        echo "[INFO] Found PHP runtime library via find: $FOUND_LIBPHP, copying to dist/ ..."
        cp -f "$FOUND_LIBPHP" "$SCRIPT_DIR/dist/libphp.so"
    fi
fi

[ -d "$SCRIPT_DIR/config" ] && cp -r "$SCRIPT_DIR/config" "$SCRIPT_DIR/dist/"
[ -d "$SCRIPT_DIR/public" ] && cp -r "$SCRIPT_DIR/public" "$SCRIPT_DIR/dist/"
if [ -d "$SCRIPT_DIR/app/view" ]; then
    mkdir -p "$SCRIPT_DIR/dist/app"
    cp -r "$SCRIPT_DIR/app/view" "$SCRIPT_DIR/dist/app/"
fi

# 1. 自动扫描并打包 PHP 扩展模块 (.so)
mkdir -p "$SCRIPT_DIR/dist/ext"
PHP_EXT_DIR=$(php-config --extension-dir 2>/dev/null || true)
if [ -d "$PHP_EXT_DIR" ]; then
    echo "[INFO] Copying PHP extensions from $PHP_EXT_DIR to dist/ext/ ..."
    cp -f "$PHP_EXT_DIR"/*.so "$SCRIPT_DIR/dist/ext/" 2>/dev/null || true
fi

# 2. 自动通过 ldd 探测并打包所有程序与扩展的底层共享库 (如 gmp, mpfr, libzip 等)
mkdir -p "$SCRIPT_DIR/dist/lib"
echo "[INFO] Scanning and bundling all shared library dependencies via ldd ..."
for bin_or_lib in "$SCRIPT_DIR/dist/webman-server.bin" "$SCRIPT_DIR/dist/libphpx.so" "$SCRIPT_DIR/dist/libphp.so" "$SCRIPT_DIR"/dist/ext/*.so; do
    if [ -f "$bin_or_lib" ]; then
        ldd "$bin_or_lib" 2>/dev/null | grep "=>" | awk '{print $3}' | while read -r libpath; do
            if [ -f "$libpath" ]; then
                libname=$(basename "$libpath")
                # 排除 Linux 最基础的 glibc 核心库以防跨发行版 ABI 冲突，收集 gmp/mpfr/phpx/libzip 等业务扩展库
                case "$libname" in
                    libc.so*|ld-linux*|libdl.so*|libpthread.so*|libm.so*|librt.so*)
                        ;;
                    *)
                        if [ ! -f "$SCRIPT_DIR/dist/$libname" ] && [ ! -f "$SCRIPT_DIR/dist/lib/$libname" ]; then
                            echo "[INFO] Bundling dependent library: $libname (from $libpath)"
                            cp -f "$libpath" "$SCRIPT_DIR/dist/lib/" || true
                        fi
                        ;;
                esac
            fi
        done
    fi
done

# 3. Set $ORIGIN RPATH to executable, php extensions, and shared libraries
if command -v patchelf &> /dev/null; then
    echo "[INFO] Setting RPATH to \$ORIGIN:\$ORIGIN/lib:\$ORIGIN/../lib for all binaries in dist ..."
    patchelf --set-rpath '$ORIGIN:$ORIGIN/lib' "$SCRIPT_DIR/dist/webman-server.bin" 2>/dev/null || true
    for solib in "$SCRIPT_DIR"/dist/*.so "$SCRIPT_DIR"/dist/lib/*.so*; do
        [ -f "$solib" ] && patchelf --set-rpath '$ORIGIN:$ORIGIN/lib' "$solib" 2>/dev/null || true
    done
    for extsolib in "$SCRIPT_DIR"/dist/ext/*.so; do
        [ -f "$extsolib" ] && patchelf --set-rpath '$ORIGIN/../lib:$ORIGIN:$ORIGIN/..' "$extsolib" 2>/dev/null || true
    done
fi

# 生成 Linux 专用的纯净自包含 php.ini
# 注意：PHP 扩展加载有顺序依赖（如 mysqlnd 必须在 pdo/mysqli 前加载，pdo 必须在 pdo_mysql 前加载）
cat > "$SCRIPT_DIR/dist/php.ini" << 'EOF'
output_buffering=0
implicit_flush=1
memory_limit=4G
opcache.enable_cli=0
extension_dir="./ext"

; 核心进程管理与网络扩展
extension=posix.so
extension=pcntl.so
extension=openssl.so
extension=mbstring.so
extension=mysqlnd.so
extension=pdo.so
extension=pdo_mysql.so
extension=mysqli.so
extension=curl.so
extension=fileinfo.so
extension=zip.so
extension=igbinary.so
extension=msgpack.so
extension=redis.so
extension=sockets.so
extension=event.so
EOF

# 如果 extensions.txt 中有其他自定义扩展，追加到 php.ini
if [ -f "$SCRIPT_DIR/extensions.txt" ]; then
    while IFS= read -r line || [ -n "$line" ]; do
        clean_ext=$(echo "$line" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//' -e 's/#.*$//')
        if [ -n "$clean_ext" ]; then
            if ! grep -q "extension=${clean_ext}\.so" "$SCRIPT_DIR/dist/php.ini"; then
                echo "extension=${clean_ext}.so" >> "$SCRIPT_DIR/dist/php.ini"
            fi
        fi
    done < "$SCRIPT_DIR/extensions.txt"
fi

# 检查所有扩展，若 ext/ 目录下没有对应 .so（说明已静态内联进内核），则注释该行以防产生 Warning
for ext_so in "$SCRIPT_DIR"/dist/ext/*.so; do
    :
done
while IFS= read -r line || [ -n "$line" ]; do
    if echo "$line" | grep -q "^extension="; then
        ext_file=$(echo "$line" | cut -d'=' -f2)
        if [ ! -f "$SCRIPT_DIR/dist/ext/$ext_file" ]; then
            sed -i "s/^extension=$ext_file$/; extension=$ext_file/" "$SCRIPT_DIR/dist/php.ini"
        fi
    fi
done < "$SCRIPT_DIR/dist/php.ini"

# Keep direct invocation self-contained: load the adjacent php.ini and libraries.
cat > "$SCRIPT_DIR/dist/webman-server" << 'EOF'
#!/usr/bin/env sh
set -eu

SCRIPT_DIR=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
cd "$SCRIPT_DIR"

export PHPRC="$SCRIPT_DIR"
export LD_LIBRARY_PATH="$SCRIPT_DIR:$SCRIPT_DIR/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
exec "$SCRIPT_DIR/webman-server.bin" "$@"
EOF
chmod +x "$SCRIPT_DIR/dist/webman-server"

[ -f "$SCRIPT_DIR/start.sh" ] && cp -f "$SCRIPT_DIR/start.sh" "$SCRIPT_DIR/dist/" && chmod +x "$SCRIPT_DIR/dist/start.sh"

# 剥离所有二进制与共享库中的调试符号（最小化包体积）
if command -v strip &> /dev/null; then
    echo "[INFO] Stripping debug symbols to minimize package size..."
    strip --strip-unneeded "$SCRIPT_DIR"/dist/*.bin 2>/dev/null || true
    strip --strip-unneeded "$SCRIPT_DIR"/dist/*.so 2>/dev/null || true
    strip --strip-unneeded "$SCRIPT_DIR"/dist/lib/*.so* 2>/dev/null || true
    strip --strip-unneeded "$SCRIPT_DIR"/dist/ext/*.so 2>/dev/null || true
fi

echo "[INFO] Packaging completed successfully! Dist path: $SCRIPT_DIR/dist"
ls -la "$SCRIPT_DIR/dist"