WooCommerce结账页：根据特定用户角色和收货地址选项添加额外费用

/**
 * 前端交互：监听“配送到不同地址”复选框并发送AJAX请求
 */
function custom_ship_to_different_address_script() {
    // 仅对已登录用户执行此脚本
    if ( ! is_user_logged_in() ) {
        return;
    }

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

        // 页面加载时，清除可能存在的自定义会话变量，确保状态干净
        if ( WC()->session->get( 'ship_different' ) ) {
            WC()->session->__unset( 'ship_different' );
        }
    ?>
    <script type="text/javascript">
    jQuery( function($) {
        // 确保WooCommerce结账参数已定义
        if ( typeof wc_checkout_params === 'undefined' )
            return false;

        var checkbox = '#ship-to-different-address-checkbox'; // 复选框ID
        var checkboxState = ''; // 复选框状态变量

        // AJAX函数，用于发送复选框状态到后端
        function triggerShipToDifferentAddressAjax( 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) {
                    // AJAX成功后，触发结账页面更新，以便重新计算费用
                    $( 'body' ).trigger( 'update_checkout' );
                    // console.log( result ); // 调试用，可移除
                }
            });
        }

        // 监听复选框的change事件
        $( checkbox ).on( 'change', function() {
            // 根据复选框是否被选中，设置状态为'yes'或'no'
            checkboxState = $( this ).prop('checked') === true ? 'yes' : 'no';
            // 调用AJAX函数发送状态
            triggerShipToDifferentAddressAjax( checkboxState );
        });
    });
    </script>
    <?php
    endif;
}
add_action( 'wp_footer', 'custom_ship_to_different_address_script' );

/**
 * 后端AJAX处理：接收前端状态并存储到WooCommerce会话
 */
function custom_get_ajax_ship_different_address() {
    // 检查POST请求中是否存在'ship_different'参数
    if ( isset( $_POST['ship_different'] ) ) {
        // 对接收到的数据进行安全处理
        $ship_different_status = esc_attr( $_POST['ship_different'] );
        // 将状态存储到WooCommerce会话中
        WC()->session->set( 'ship_different', $ship_different_status );
        // 返回状态（可选，用于调试）
        echo $ship_different_status;
    }

    die(); // 终止AJAX请求
}
// 注册AJAX动作，供登录用户和非登录用户使用
add_action( 'wp_ajax_ship_different_address', 'custom_get_ajax_ship_different_address' );
add_action( 'wp_ajax_nopriv_ship_different_address', 'custom_get_ajax_ship_different_address' );

/**
 * 动态费用计算：根据用户角色和“配送到不同地址”选项添加或移除费用
 */
function custom_add_remove_ship_to_different_address_fee( $cart ) {
    // 仅在非管理后台且非购物车页面执行此逻辑
    if ( ( is_admin() && ! defined( 'DOING_AJAX' ) ) || is_cart() ) {
        return;
    }

    // 仅对已登录用户执行此逻辑
    if ( ! is_user_logged_in() ) {
        return;
    }

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

    // 定义需要收取费用的用户角色，请根据您的实际需求修改
    $roles_to_charge = array( 'team', 'team2' ); 

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

    // 检查用户是否拥有需要收取费用的角色之一
    // 如果用户角色与$roles_to_charge有交集，则表示用户符合条件
    $has_required_role = ! empty( array_intersect( $user_roles, $roles_to_charge ) );

    // 如果用户拥有指定角色且选择了“配送到不同地址”
    if ( $has_required_role && WC()->session->get( 'ship_different' ) == 'yes' ) {
        $fee_amount = 5.00; // 定义额外费用金额
        // 添加费用到购物车
        $cart->add_fee( __( '配送服务费', 'woocommerce'), $fee_amount );
    }
}
add_action( 'woocommerce_cart_calculate_fees', 'custom_add_remove_ship_to_different_address_fee', 10, 1 );