在 WooCommerce 结账时为特定用户角色添加“配送到不同地址”费用

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

    // 在页面加载时清除自定义的 WooCommerce 会话变量
    // 这确保了页面刷新时，费用状态不会因旧的会话数据而错误显示
    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 checkboxSelector = '#ship-to-different-address-checkbox'; // “配送到不同地址”复选框的 ID
            var checkboxState = '';

            // 定义 AJAX 函数，用于发送复选框状态
            function triggerShipToDifferentAddress( 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' 事件
            $( checkboxSelector ).on( 'change', function() {
                checkboxState = $( this ).prop('checked') === true ? 'yes' : 'no';
                triggerShipToDifferentAddress( checkboxState );
            });
        });
    </script>
    <?php
}
add_action( 'wp_footer', 'custom_ship_to_different_address_script' );

/**
 * 后端 PHP：处理 AJAX 请求并将会话数据存储到 WooCommerce
 */
function custom_get_ajax_ship_different_address() {
    if ( isset( $_POST['ship_different'] ) ) {
        // 将前端发送的复选框状态存储到 WooCommerce 会话中
        WC()->session->set( 'ship_different', sanitize_text_field( $_POST['ship_different'] ) );
        echo sanitize_text_field( $_POST['ship_different'] ); // 返回状态给前端（可选）
    }
    // 终止 AJAX 请求的执行
    wp_die();
}
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' );

/**
 * WooCommerce 费用计算：根据用户角色和“配送到不同地址”状态添加费用
 */
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;
    }

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

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

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

    // 检查用户是否拥有任何一个目标角色
    // array_intersect 返回两个数组的交集，如果结果不为空，则表示用户拥有至少一个目标角色
    $common_roles = array_intersect( $user_roles, $target_roles );

    // 如果用户拥有至少一个目标角色
    if ( ! empty( $common_roles ) ) {
        $fee_amount = 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', 'custom_add_remove_ship_to_different_address_fee', 10, 1 );