WooCommerce：实现基于用户角色和配送地址的动态附加费

/**
 * 前端JavaScript代码：监听“配送到不同地址”复选框状态
 * 仅在结账页面且用户已登录时执行
 */
function action_wp_footer_for_shipping_fee() {
    // 仅对已登录用户生效
    if ( ! is_user_logged_in() ) {
        return;
    }

    // 仅在结账页面执行，排除WooCommerce端点URL
    if ( is_checkout() && ! is_wc_endpoint_url() ) :

    // 页面加载时清除自定义的“ship_different”会话变量，确保状态刷新
    if ( WC()->session->get( 'ship_different' ) ) {
        WC()->session->__unset( 'ship_different' );
    }

    ?>
    <script type="text/javascript">
    jQuery( function($) {
        // 确保WooCommerce checkout参数已定义
        if ( typeof wc_checkout_params === 'undefined' )
            return false;

        var checkboxId = '#ship-to-different-address-checkbox'; // “配送到不同地址”复选框的ID
        var checkboxState = '';

        // Ajax函数：将复选框状态发送到后端
        function triggerShippingAddressFeeUpdate( value ) {
             $.ajax({
                type: 'POST',
                url: wc_checkout_params.ajax_url, // WooCommerce的Ajax URL
                data: {
                    'action': 'ship_different_address', // Ajax动作名称
                    'ship_different': value, // 复选框状态 ('yes'/'no')
                },
                success: function (result) {
                    $( 'body' ).trigger( 'update_checkout' ); // 触发结账页面更新，重新计算费用
                    // console.log( result ); // 用于测试，生产环境可移除
                }
            });
        }

        // 监听复选框的“change”事件
        $( checkboxId ).on( 'change', function() {
            checkboxState = $( this ).prop('checked') === true ? 'yes' : 'no';
            triggerShippingAddressFeeUpdate( checkboxState );
        });
    });
    </script>
    <?php
    endif;
}
add_action( 'wp_footer', 'action_wp_footer_for_shipping_fee' );

/**
 * 后端Ajax处理器：接收前端状态并存储到WooCommerce会话
 */
function get_ajax_ship_different_address() {
    if ( isset( $_POST['ship_different'] ) ) {
        // 将前端发送的状态存储到WooCommerce会话，并进行安全转义
        WC()->session->set( 'ship_different', esc_attr( $_POST['ship_different'] ) );
        echo esc_attr( $_POST['ship_different'] ); // 返回状态给前端（可选，用于调试）
    }

    die(); // 终止后续PHP执行
}
// 注册Ajax处理器，同时支持登录用户和非登录用户（尽管前端已限制为登录用户）
add_action( 'wp_ajax_ship_different_address', 'get_ajax_ship_different_address' );
add_action( 'wp_ajax_nopriv_ship_different_address', 'get_ajax_ship_different_address' );

/**
 * 购物车费用计算：根据用户角色和会话状态添加/移除附加费
 */
function action_woocommerce_cart_calculate_fees_for_shipping_fee( $cart ) {
    // 避免在后台管理界面或购物车页面执行此逻辑
    if ( ( is_admin() && ! defined( 'DOING_AJAX' ) ) || is_cart() ) {
        return;
    }

    // 仅对已登录用户生效
    if ( ! is_user_logged_in() ) {
        return;
    }

    // 获取当前用户对象
    $user = wp_get_current_user();

    // 定义需要收取附加费的用户角色，请根据您的实际需求进行调整
    $roles_to_check = array( 'team', 'team2', 'administrator' ); // 示例：'team', 'team2' 和 'administrator'

    // 获取当前用户的所有角色
    $user_roles = ( array ) $user->roles;

    // 检查用户是否拥有需要收取费用的角色之一
    // array_intersect 查找两个数组的交集，如果交集不为空，则表示用户拥有目标角色
    $has_target_role = ! empty( array_intersect( $user_roles, $roles_to_check ) );

    // 如果用户拥有目标角色
    if ( $has_target_role ) {
        // 定义附加费金额
        $fee_amount = 5; // $5.00

        // 检查WooCommerce会话中“ship_different”的值是否为“yes”
        if ( WC()->session->get( 'ship_different' ) == 'yes' ) {
            // 如果满足条件，添加附加费
            $cart->add_fee( __( '配送附加费', 'woocommerce'), $fee_amount );
        }
    }
}
add_action( 'woocommerce_cart_calculate_fees', 'action_woocommerce_cart_calculate_fees_for_shipping_fee', 10, 1 );