WooCommerce结账页：为特定用户角色在选择不同收货地址时添加额外费用

/**
 * 客户端脚本：监听“配送到不同地址”复选框状态变化并发送AJAX请求
 */
function custom_shipping_fee_script() {
    // 仅在已登录用户且在结账页面执行
    if ( ! is_user_logged_in() || ! ( is_checkout() && ! is_wc_endpoint_url() ) ) {
        return;
    }

    // 页面加载时，清除可能存在的旧会话数据，确保状态同步
    if ( WC()->session->get( 'ship_different' ) ) {
        WC()->session->__unset( 'ship_different' );
    }

    ?>
    <script type="text/javascript">
    jQuery( function($) {
        if ( typeof wc_checkout_params === 'undefined' ) {
            return false;
        }

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

        // AJAX函数：发送复选框状态到服务器
        function triggerShippingAddressChange( 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事件
        $( checkbox ).on( 'change', function() {
            status = $( this ).prop('checked') === true ? 'yes' : 'no';
            triggerShippingAddressChange( status );
        });
    });
    </script>
    <?php
}
add_action( 'wp_footer', 'custom_shipping_fee_script' );

/**
 * 服务器端AJAX处理器：接收客户端状态并存储到WooCommerce会话
 */
function handle_ajax_ship_different_address() {
    if ( isset( $_POST['ship_different'] ) ) {
        // 将接收到的状态存储到WooCommerce会话中
        WC()->session->set( 'ship_different', esc_attr( $_POST['ship_different'] ) );
        echo $_POST['ship_different']; // 返回状态作为响应 (可选)
    }

    die(); // 终止AJAX请求
}
// 注册AJAX动作，分别处理登录用户和非登录用户
add_action( 'wp_ajax_ship_different_address', 'handle_ajax_ship_different_address' );
add_action( 'wp_ajax_nopriv_ship_different_address', 'handle_ajax_ship_different_address' );

/**
 * 购物车费用计算：根据用户角色和会话状态添加或移除费用
 */
function add_conditional_shipping_fee( $cart ) {
    // 仅在结账页且非后台AJAX请求时执行
    if ( ( is_admin() && ! defined( 'DOING_AJAX' ) ) || is_cart() ) {
        return;
    }

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

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

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

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

    // 检查当前用户是否拥有目标角色中的任意一个
    // array_intersect 返回两个数组的交集，如果交集非空，则表示用户拥有目标角色
    $has_target_role = ! empty( array_intersect( $user_roles, $target_roles ) );

    // 如果用户拥有目标角色
    if ( $has_target_role ) {
        $fee_amount = 5; // 定义费用金额，可根据需要修改

        // 从会话中获取“配送到不同地址”的状态
        if ( WC()->session->get( 'ship_different' ) == 'yes' ) {
            // 如果状态为'yes'，则添加费用
            $cart->add_fee( __( '配送操作费', 'woocommerce'), $fee_amount );
        }
    }
}
add_action( 'woocommerce_cart_calculate_fees', 'add_conditional_shipping_fee', 10, 1 );