WooCommerce 结账：根据用户角色和不同收货地址添加条件费用

/**
 * 客户端 JavaScript：检测“配送到不同地址”复选框状态并发送 AJAX 请求。
 */
function action_wp_footer_for_shipping_fee() {
    // 仅对已登录用户和结账页面执行
    if ( ! is_user_logged_in() || ! ( is_checkout() && ! is_wc_endpoint_url() ) ) {
        return;
    }

    // 在页面加载时清除自定义 WooCommerce 会话变量，防止数据残留
    if ( WC()->session->get( 'ship_different' ) ) {
        WC()->session->__unset( 'ship_different' );
    }

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

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

        // AJAX 函数：发送复选框状态到服务器
        function triggerShippingFeeAjax( value ) {
             $.ajax({
                type: 'POST',
                url: wc_checkout_params.ajax_url, // WordPress 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() {
            checkbox_state = $( this ).prop('checked') === true ? 'yes' : 'no';
            triggerShippingFeeAjax( checkbox_state );
        });
    });
    </script>
    <?php
}
add_action( 'wp_footer', 'action_wp_footer_for_shipping_fee' );

/**
 * WordPress AJAX 处理器：接收客户端发送的复选框状态并存储到会话。
 */
function get_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 请求
}
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' ); // 示例角色：团队成员、团队2成员、管理员

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

    // 检查用户是否拥有任何一个目标角色
    $has_required_role = false;
    foreach ( $roles_to_check as $role_to_check ) {
        if ( in_array( $role_to_check, $user_roles ) ) {
            $has_required_role = true;
            break;
        }
    }

    // 如果用户拥有指定角色且“配送到不同地址”被选中
    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', 'action_woocommerce_cart_calculate_fees_for_shipping_fee', 10, 1 );