如何修改代码使WooCommerce新客首单自定义费用适配最新版本

原代码的核心问题 #

你的代码在新版WooCommerce失效主要是因为订单查询方式过时、钩子使用不规范，以及边界场景处理缺失。以下是针对性修改：

具体修改点 #

• 替换过时的订单查询方式：弃用get_posts，改用WooCommerce官方提供的WC_Order_Query，兼容最新版订单数据结构，性能更优
• 处理未登录用户场景：未登录用户get_current_user_id()返回0，会误判为新客户，需增加登录判断
• 避免重复添加费用：woocommerce_before_calculate_totals会多次触发，需先移除同名费用再添加
• 扩大有效订单状态范围：仅判断completed会遗漏processing等已支付的有效订单，需覆盖所有已完成/处理中的状态
• 使用钩子传递的cart对象：新版中直接调用WC()->cart可能出现上下文错误，改用钩子参数传入的cart实例

修正后的完整代码 #

function custom_add_first_order_fee( $cart ) {
    // 避免递归调用
    if ( doing_action( 'woocommerce_before_calculate_totals' ) ) {
        return;
    }

    // 仅针对登录用户（未登录用户逻辑可按需调整）
    $user_id = get_current_user_id();
    if ( $user_id === 0 ) {
        return;
    }

    // 使用WC_Order_Query查询用户有效订单
    $order_query = new WC_Order_Query( array(
        'customer_id' => $user_id,
        'status'      => array( 'completed', 'processing' ), // 覆盖所有已支付有效状态
        'limit'       => 1, // 查到1条即停止，提升性能
        'return'      => 'ids', // 仅返回订单ID，减少数据加载
    ) );
    $user_orders = $order_query->get_orders();

    // 用户无有效订单时添加费用
    if ( empty( $user_orders ) ) {
        // 移除已存在的同名费用，避免重复添加
        foreach ( $cart->get_fees() as $key => $fee ) {
            if ( $fee->name === __( 'Joining Fee', 'your-plugin' ) ) {
                $cart->remove_fee( $key );
            }
        }
        // 添加自定义费用，第三个参数控制是否含税，第四个参数可指定税类
        $cart->add_fee( __( 'Joining Fee', 'your-plugin' ), 100, false, '' );
    }
}
add_action( 'woocommerce_before_calculate_totals', 'custom_add_first_order_fee', 10, 1 );

额外说明 #

• 若需支持未登录用户的首单判断，可通过用户邮箱或会话ID追踪，但实现更复杂，需结合订单创建时的邮箱匹配逻辑
• 代码中的your-plugin需替换为实际的文本域，用于多语言支持

内容的提问来源于stack exchange，提问作者hacker