vps.sh

#!/usr/bin/env bash

# Repository bootstrap.
# Compatibility fallback is handled below when local files are incomplete.
# Compatibility marker for legacy updater: VPS 全能控制面板

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
RELEASE_URL="https://raw.githubusercontent.com/Chunlion/VPS-Optimize/main/dist/vps.sh"
MODULE_LIST="$SCRIPT_DIR/scripts/modules.list"
MODULES=()

download_release_script() {
    local output_file="$1"
    if command -v curl >/dev/null 2>&1; then
        curl -fsSL --connect-timeout 10 --max-time 90 --retry 2 --retry-delay 1 --retry-connrefused "$RELEASE_URL" -o "$output_file"
    elif command -v wget >/dev/null 2>&1; then
        wget -q --timeout=15 --tries=3 -O "$output_file" "$RELEASE_URL"
    else
        echo "缺少 curl/wget，无法下载脚本。" >&2
        return 1
    fi
    [[ -s "$output_file" ]]
}

switch_to_release_script() {
    local tmp_file self_path
    tmp_file=$(mktemp /tmp/vps-optimize-release.XXXXXX.sh) || {
        echo "创建临时脚本失败。" >&2
        return 1
    }

    echo "当前脚本文件不完整，正在下载可运行脚本..." >&2
    if ! download_release_script "$tmp_file"; then
        rm -f "$tmp_file"
        echo "下载脚本失败：$RELEASE_URL" >&2
        return 1
    fi
    if ! bash -n "$tmp_file" >/dev/null 2>&1; then
        rm -f "$tmp_file"
        echo "下载的脚本未通过语法检查。" >&2
        return 1
    fi
    if ! grep -q 'func_sni_stack_quick_menu' "$tmp_file" || ! grep -q 'main_menu' "$tmp_file"; then
        rm -f "$tmp_file"
        echo "下载的脚本内容不完整。" >&2
        return 1
    fi

    chmod +x "$tmp_file" 2>/dev/null || true
    self_path="$0"
    if [[ -f "$self_path" && -w "$self_path" ]]; then
        mv "$tmp_file" "$self_path"
        chmod +x "$self_path" 2>/dev/null || true
        exec bash "$self_path" "$@"
    fi

    exec bash "$tmp_file" "$@"
}

load_source_modules() {
    local raw module
    MODULES=()
    [[ -f "$MODULE_LIST" ]] || {
        echo "Missing module list: scripts/modules.list" >&2
        return 1
    }
    while IFS= read -r raw || [[ -n "$raw" ]]; do
        module="${raw%%#*}"
        module="${module#"${module%%[![:space:]]*}"}"
        module="${module%"${module##*[![:space:]]}"}"
        [[ -n "$module" ]] || continue
        if [[ "$module" == *.sh ]]; then
            echo "Invalid module list entry, omit .sh: ${module}" >&2
            return 1
        fi
        MODULES+=("$module")
    done < "$MODULE_LIST"
    [[ ${#MODULES[@]} -gt 0 ]] || {
        echo "Module list is empty: scripts/modules.list" >&2
        return 1
    }
}

missing_module=0
if ! load_source_modules; then
    missing_module=1
fi
for module in "${MODULES[@]}"; do
    if [[ ! -f "$SCRIPT_DIR/src/${module}.sh" ]]; then
        echo "Missing source module: src/${module}.sh" >&2
        missing_module=1
    fi
done

if [[ "$missing_module" -ne 0 ]]; then
    switch_to_release_script "$@"
    exit 1
fi

for module in "${MODULES[@]}"; do
    # shellcheck source=/dev/null
    . "$SCRIPT_DIR/src/${module}.sh"
done