cursor-api/scripts/build.sh

#!/bin/bash
set -euo pipefail

# 颜色输出函数
info() { echo -e "\033[1;34m[INFO]\033[0m $*"; }
warn() { echo -e "\033[1;33m[WARN]\033[0m $*"; }
error() { echo -e "\033[1;31m[ERROR]\033[0m $*" >&2; exit 1; }

# 检查是否在 Linux 环境
is_linux() {
    [ "$(uname -s)" = "Linux" ]
}

# 检查必要的工具
check_requirements() {
    local missing_tools=()

    # 基础工具检查
    for tool in cargo protoc npm node; do
        if ! command -v "$tool" &>/dev/null; then
            missing_tools+=("$tool")
        fi
    done

    # cross 工具检查（仅在 Linux 上需要）
    if [[ "$OS" == "Linux" ]] && ! command -v cross &>/dev/null; then
        missing_tools+=("cross")
    fi

    if [[ ${#missing_tools[@]} -gt 0 ]]; then
        error "缺少必要工具: ${missing_tools[*]}"
    fi
}

# 帮助信息
show_help() {
    cat << EOF
用法: $(basename "$0") [选项]

选项:
  --cross         使用 cross 进行交叉编译（仅在 Linux 上有效）
  --static        使用静态链接（默认动态链接）
  --help          显示此帮助信息

不带参数时只编译当前平台
EOF
}

# 判断是否使用 cross
should_use_cross() {
    local target=$1
    # 如果不是 Linux 环境，直接返回 false
    if [[ "$OS" != "Linux" ]]; then
        return 1
    fi
    
    # 在 Linux 环境下，以下目标不使用 cross：
    # 1. Linux 上的 x86_64-unknown-linux-gnu
    if [[ "$target" == "x86_64-unknown-linux-gnu" ]]; then
        return 1
    fi
    return 0
}

# 并行构建函数
build_target() {
    local target=$1
    local extension=""
    local rustflags="${2:-}"

    info "正在构建 $target..."

    # 确定文件后缀
    [[ $target == *"windows"* ]] && extension=".exe"

    # 判断是否使用 cross
    if should_use_cross "$target"; then
        env RUSTFLAGS="$rustflags" cross build --target "$target" --release
    else
        if [[ $target != "$CURRENT_TARGET" ]]; then
            env RUSTFLAGS="$rustflags" cargo build --target "$target" --release
        else
            env RUSTFLAGS="$rustflags" cargo build --release
        fi
    fi

    # 移动编译产物到 release 目录
    local binary_name="cursor-api"
    [[ $USE_STATIC == true ]] && binary_name+="-static"

    local binary_path
    if [[ $target == "$CURRENT_TARGET" ]]; then
        binary_path="target/release/cursor-api$extension"
    else
        binary_path="target/$target/release/cursor-api$extension"
    fi

    if [[ -f "$binary_path" ]]; then
        cp "$binary_path" "release/${binary_name}-$target$extension"
        info "完成构建 $target"
    else
        warn "构建产物未找到: $target"
        warn "查找路径: $binary_path"
        warn "当前目录内容:"
        ls -R target/
        return 1
    fi
}

# 获取 CPU 架构和操作系统
ARCH=$(uname -m | sed 's/^aarch64\|arm64$/aarch64/;s/^x86_64\|x86-64\|x64\|amd64$/x86_64/')
OS=$(uname -s)

# 确定当前系统的目标平台
get_target() {
    local arch=$1
    local os=$2
    case "$os" in
        "Darwin") echo "${arch}-apple-darwin" ;;
        "Linux") echo "${arch}-unknown-linux-gnu" ;;
        "MINGW"*|"MSYS"*|"CYGWIN"*|"Windows_NT") echo "${arch}-pc-windows-msvc" ;;
        "FreeBSD") echo "${arch}-unknown-freebsd" ;;
        *) error "不支持的系统: $os" ;;
    esac
}

# 设置当前目标平台
CURRENT_TARGET=$(get_target "$ARCH" "$OS")

# 检查是否成功获取目标平台
[ -z "$CURRENT_TARGET" ] && error "无法确定当前系统的目标平台"

# 获取系统对应的所有目标
get_targets() {
    case "$1" in
        "linux")
            # Linux 构建所有 Linux 目标和 FreeBSD 目标
            echo "x86_64-unknown-linux-gnu x86_64-unknown-freebsd"
            ;;
        "freebsd")
            # FreeBSD 只构建当前架构的 FreeBSD 目标
            echo "${ARCH}-unknown-freebsd"
            ;;
        "windows")
            # Windows 构建所有 Windows 目标
            echo "x86_64-pc-windows-msvc"
            ;;
        "macos")
            # macOS 构建所有 macOS 目标
            echo "x86_64-apple-darwin aarch64-apple-darwin"
            ;;
        *) error "不支持的系统组: $1" ;;
    esac
}

# 解析参数
USE_STATIC=false

while [[ $# -gt 0 ]]; do
    case $1 in
        --static) USE_STATIC=true ;;
        --help) show_help; exit 0 ;;
        *) error "未知参数: $1" ;;
    esac
    shift
done

# 检查依赖
check_requirements

# 确定要构建的目标
case "$OS" in
    "Darwin") 
        TARGETS=($(get_targets "macos"))
        ;;
    "Linux")
        TARGETS=($(get_targets "linux"))
        ;;
    "FreeBSD")
        TARGETS=($(get_targets "freebsd"))
        ;;
    "MINGW"*|"MSYS"*|"CYGWIN"*|"Windows_NT")
        TARGETS=($(get_targets "windows"))
        ;;
    *) error "不支持的系统: $OS" ;;
esac

# 创建 release 目录
mkdir -p release

# 设置静态链接标志
RUSTFLAGS=""
[[ $USE_STATIC == true ]] && RUSTFLAGS="-C target-feature=+crt-static"

# 并行构建所有目标
info "开始构建..."
for target in "${TARGETS[@]}"; do
    build_target "$target" "$RUSTFLAGS" &
done

# 等待所有构建完成
wait

# 为 macOS 平台创建通用二进制
if [[ "$OS" == "Darwin" ]] && [[ ${#TARGETS[@]} -gt 1 ]]; then
    binary_suffix=""
    [[ $USE_STATIC == true ]] && binary_suffix="-static"

    if [[ -f "release/cursor-api${binary_suffix}-x86_64-apple-darwin" ]] && \
       [[ -f "release/cursor-api${binary_suffix}-aarch64-apple-darwin" ]]; then
        info "创建 macOS 通用二进制..."
        lipo -create \
            "release/cursor-api${binary_suffix}-x86_64-apple-darwin" \
            "release/cursor-api${binary_suffix}-aarch64-apple-darwin" \
            -output "release/cursor-api${binary_suffix}-universal-apple-darwin"
    fi
fi

info "构建完成！"