在 WooCommerce 购物车页面添加动态折扣复选框教程

/**
 * 在购物车总计区域添加折扣复选框
 */
function add_discount_checkbox_to_cart_totals() {
    // 确保在购物车页面且购物车不为空
    if ( is_cart() && ! WC()->cart->is_empty() ) {
        // 从会话中获取复选框的当前状态
        $is_discount_applied = WC()->session->get( 'apply_custom_discount', false );
        ?>
        <tr class="discount-checkbox-row">
            <th><?php esc_html_e( '应用折扣', 'your-text-domain' ); ?></th>
            <td data-title="<?php esc_attr_e( '应用折扣', 'your-text-domain' ); ?>">
                <label for="custom_discount_checkbox">
                    <input type="checkbox" id="custom_discount_checkbox" name="custom_discount_checkbox" value="1" <?php checked( $is_discount_applied, true ); ?>>
                    <?php esc_html_e( '勾选以获得固定折扣', 'your-text-domain' ); ?>
                </label>
            </td>
        </tr>
        <?php
    }
}
add_action( 'woocommerce_cart_totals_before_shipping', 'add_discount_checkbox_to_cart_totals' );

/**
 * 前端 JavaScript 监听复选框变化并发送 AJAX 请求
 */
jQuery(function($) {
    var $checkbox = $('#custom_discount_checkbox');

    if ($checkbox.length) {
        $checkbox.on('change', function() {
            var isChecked = $(this).is(':checked');

            // 显示加载指示器 (可选)
            $('.woocommerce-cart-form, .cart_totals').block({
                message: null,
                overlayCSS: {
                    background: '#fff',
                    opacity: 0.6
                }
            });

            $.ajax({
                type: 'POST',
                url: wc_cart_params.ajax_url, // WooCommerce 提供的 AJAX URL
                data: {
                    action: 'update_custom_discount_status', // 后端 AJAX 动作
                    security: wc_cart_params.update_shipping_method_nonce, // 使用 WooCommerce 提供的 nonce
                    apply_discount: isChecked ? 1 : 0
                },
                success: function(response) {
                    if (response.success) {
                        // 成功后刷新购物车片段
                        $(document.body).trigger('update_checkout'); // 刷新结账页 (如果用户在结账页)
                        $(document.body).trigger('wc_fragment_refresh'); // 刷新迷你购物车
                        $(document.body).trigger('wc_update_cart'); // 刷新购物车总计
                    } else {
                        console.error('更新折扣状态失败:', response.data);
                        alert('更新折扣状态失败，请重试。');
                    }
                },
                error: function(xhr, status, error) {
                    console.error('AJAX 请求错误:', error);
                    alert('网络错误，请重试。');
                },
                complete: function() {
                    // 隐藏加载指示器
                    $('.woocommerce-cart-form, .cart_totals').unblock();
                }
            });
        });
    }
});

/**
 * 后端 AJAX 处理器：更新折扣复选框状态
 */
function handle_custom_discount_ajax() {
    // 验证 nonce
    if ( ! isset( $_POST['security'] ) || ! wp_verify_nonce( $_POST['security'], 'woocommerce-cart' ) ) {
        wp_send_json_error( 'Nonce 验证失败！' );
    }

    $apply_discount = isset( $_POST['apply_discount'] ) ? (bool) intval( $_POST['apply_discount'] ) : false;

    // 将状态存储到 WooCommerce Session
    WC()->session->set( 'apply_custom_discount', $apply_discount );

    // 重新计算购物车总计
    WC()->cart->calculate_totals();

    wp_send_json_success( array( 'message' => '折扣状态已更新。' ) );
}
add_action( 'wp_ajax_update_custom_discount_status', 'handle_custom_discount_ajax' );
add_action( 'wp_ajax_nopriv_update_custom_discount_status', 'handle_custom_discount_ajax' ); // 允许未登录用户使用

/**
 * 根据复选框状态动态添加或移除购物车折扣费用
 */
function apply_custom_discount_fee( $cart ) {
    if ( is_admin() && ! defined( 'DOING_AJAX' ) ) {
        return; // 避免在管理后台非 AJAX 请求时执行
    }

    // 获取折扣金额 (固定值)
    $discount_amount = 10; // 例如：固定折扣 10 元/美元

    // 从会话中获取复选框状态
    $apply_discount = WC()->session->get( 'apply_custom_discount', false );

    if ( $apply_discount ) {
        // 如果折扣已勾选，添加负值费用 (即折扣)
        // 参数: name, amount, taxable, tax_class
        $cart->add_fee( esc_html__( '自定义折扣', 'your-text-domain' ), -$discount_amount, false );
    } else {
        // 如果未勾选，确保移除任何之前可能存在的折扣 (虽然 add_fee 是动态的，但明确处理更好)
        // 实际上，如果条件不满足，add_fee 就不会被调用，也就不会添加费用。
        // 但如果需要更复杂的逻辑，这里可以进行清理。
    }
}
add_action( 'woocommerce_cart_calculate_fees', 'apply_custom_discount_fee', 10, 1 );

/**
 * 初始化 WooCommerce Session，如果尚未初始化
 * 这对于某些环境可能不是必需的，但可以作为一种健壮性措施
 */
function init_woocommerce_session_for_discount() {
    if ( ! WC()->session->has_session() ) {
        WC()->session->set_customer_session_cookie( true );
    }
}
add_action( 'init', 'init_woocommerce_session_for_discount' );